| -rw-r--r-- | src/db.cc | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -1,99 +1,101 @@ #include <unistd.h> #include <sys/stat.h> #include <sys/types.h> +#include <stdlib.h> +#include <string.h> #include <cassert> #include <napkin/exception.h> #include "db.h" #include "config.h" namespace napkin { extern const char *sql_bootstrap; db_t::db_t() { const char *h = getenv("HOME"); if(h) { datadir = h; datadir += "/."PACKAGE_NAME"/"; }else{ #if defined(HAVE_GET_CURRENT_DIR_NAME) char *cwd = get_current_dir_name(); if(!cwd) throw napkin::exception("failed to get_current_dir_name()"); datadir = cwd; free(cwd); #elif defined(HAVE_GETCWD) { char cwd[ # if defined(MAXPATH) MAXPATH # elif defined(MAXPATHLEN) MAXPATHLEN # else /* maxpath */ 512 # endif /* maxpath */ ]; if(!getcwd(cwd,sizeof(cwd))) throw napkin::exception("failed to getcwd()"); datadir = cwd; } #else /* get cwd */ # error dunno how to get current workdir #endif /* get cwd */ datadir += "/."PACKAGE_NAME"/"; } if(access(datadir.c_str(),R_OK|W_OK) && mkdir(datadir.c_str(),0700)) throw napkin::exception("no access to '"+datadir+"' directory"); open((datadir+PACKAGE_NAME".db").c_str()); assert(_D); char **resp; int nr,nc; char *errm; if(sqlite3_get_table( _D, "SELECT s_tobed FROM sleeps LIMIT 0", &resp,&nr,&nc,&errm)!=SQLITE_OK) { if(sqlite3_exec(_D,sql_bootstrap,NULL,NULL,&errm)!=SQLITE_OK) throw napkin::exception(string("failed to bootstrap sqlite database: ")+errm); }else sqlite3_free_table(resp); } void db_t::store(const hypnodata_t& hd) { sqlite::mem_t<char*> S = sqlite3_mprintf( "INSERT INTO sleeps (" "s_tobed,s_alarm," "s_window,s_data_a," "s_almost_awakes," "s_timezone" ") VALUES (" "%Q,%Q,%d,%d,%Q,%ld" ")", hd.w3c_to_bed().c_str(), hd.w3c_alarm().c_str(), hd.window,hd.data_a, hd.w3c_almostawakes().c_str(), timezone ); try { exec(S); }catch(sqlite::exception& se) { if(se.rcode==SQLITE_CONSTRAINT) throw exception_db_already("The record seems to be already in the database"); throw exception_db("Well, some error occured"); } } void db_t::remove(const hypnodata_t& hd) { sqlite::mem_t<char*> S = sqlite3_mprintf( "DELETE FROM sleeps" " WHERE s_tobed=%Q AND s_alarm=%Q", hd.w3c_to_bed().c_str(), hd.w3c_alarm().c_str() ); exec(S); } void db_t::load(list<hypnodata_ptr_t>& rv, const string& sql) { sqlite::table_t T; int nr,nc; get_table( string( |