1 files changed, 2 insertions, 0 deletions

diff --git a/src/db.cc b/src/db.cc
index c350e68..2b4a7eb 100644
--- a/src/db.cc
+++ b/src/db.cc
@@ -1,120 +1,122 @@
 #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(
                     "SELECT"
                      " s_tobed, s_alarm,"
                      " s_window, s_data_a,"
                      " s_almost_awakes"
                     " FROM sleeps"
                     " "+sql).c_str(),T,&nr,&nc );
         if(nr) {
             assert(nc==5);
             for(int r=1;r<=nr;++r) {
                 hypnodata_ptr_t hd(new hypnodata_t());
                 hd->set_to_bed(T.get(r,0,nc));
                 hd->set_alarm(T.get(r,1,nc));
                 hd->set_window(T.get(r,2,nc));
                 hd->set_data_a(T.get(r,3,nc));
                 hd->set_almost_awakes(T.get(r,4,nc));
                 rv.push_back(hd);
             }
         }
     }
 
 }