KissCount/model/Database.cpp

229 lines
5.4 KiB
C++
Raw Normal View History

#include "Database.h"
2010-05-16 10:35:34 +02:00
// if (!_db.CheckSyntax(req))
// {
// wxString s = req;
// std::cout << s.mb_str() << " is invalid !\n" ;
// code_if_syntax_fail;
// return return_value;
// }
2010-05-15 11:21:42 +02:00
#define EXECUTE_SQL_QUERY_WITH_CODE(req, res, return_value, code_if_fail, code_if_syntax_fail) \
do{\
try\
{\
res = _db.ExecuteQuery(req);\
}\
catch (wxSQLite3Exception e)\
{\
std::cerr << e.GetMessage().mb_str() << "\n" ;\
code_if_fail; \
return return_value;\
}\
} while(0);
#define EXECUTE_SQL_QUERY(req, res, return_value) EXECUTE_SQL_QUERY_WITH_CODE(req, res, return_value, {}, {})
Database::Database()
{
std::ifstream bdd_file;
bdd_file.open(BDD_FILE);
if (!bdd_file)
{
CreateDatabase();
}
else
_db.Open(_(BDD_FILE));
bdd_file.close();
}
void Database::CreateDatabase()
{
std::ifstream init_script;
std::string line;
2010-05-15 11:21:42 +02:00
wxString wxline;
wxMessageBox(_("No database found, creating a new one"), _("KissCount"), wxICON_EXCLAMATION | wxOK );
init_script.open(INIT_SCRIPT);
if (!init_script)
{
wxMessageBox(_(INIT_SCRIPT " not found, aborting"), _("Error"), wxICON_ERROR | wxOK );
throw "init.sql not found, aborting";
}
_db.Open(_(BDD_FILE));
do
{
getline(init_script, line);
2010-05-15 11:21:42 +02:00
wxline = wxString(line.c_str(), wxConvUTF8);
wxline.Trim(false);
2010-05-16 10:35:34 +02:00
if (!wxline.Length() || wxline.StartsWith(_("--"))) continue;
2010-05-15 11:21:42 +02:00
if (!_db.CheckSyntax(wxline))
{
std::cout << line << " is invalid !\n" ;
continue;
}
try
{
2010-05-15 11:21:42 +02:00
_db.ExecuteUpdate(wxline);
}
catch (...)
{
wxMessageBox(_("Error creating original database"), _("Error"), wxICON_ERROR | wxOK );
2010-05-15 11:21:42 +02:00
remove(BDD_FILE);
throw line;
}
} while (init_script);
init_script.close();
}
std::list<wxString> Database::GetUsers()
{
std::list<wxString> res;
// Check whether value exists in table
wxSQLite3ResultSet set ;
2010-05-15 11:21:42 +02:00
EXECUTE_SQL_QUERY(_("SELECT name FROM user ORDER BY name"), set, res);
while (set.NextRow())
{
res.push_front(set.GetAsString(0));
}
set.Finalize();
return res;
}
bool Database::IsValidUser(wxString user, wxString password)
{
bool res;
blk_SHA_CTX sha_ctx;
unsigned char sha[20];
wxString req, wxSHA;
wxSQLite3ResultSet set;
blk_SHA1_Init(&sha_ctx);
blk_SHA1_Update(&sha_ctx, password.c_str(), password.Length());
blk_SHA1_Final(sha, &sha_ctx);
for(int i=0; i<20; i++)
2010-05-16 10:35:34 +02:00
wxSHA += wxString::Format(wxT("%02x"), (int)sha[i]);
req = _("SELECT name FROM user WHERE name='") + user + _("' AND password='") + wxSHA + _("'");
2010-05-15 11:21:42 +02:00
EXECUTE_SQL_QUERY(req, set, false);
res = set.NextRow() ;
set.Finalize();
return res;
}
2010-05-15 11:21:42 +02:00
User* Database::LoadUser(wxString name)
{
wxSQLite3ResultSet set;
wxString req;
User* user;
Account* account;
std::list<Account*>::iterator it;
req = _("SELECT * FROM user WHERE name='") + name + _("'");
EXECUTE_SQL_QUERY(req, set, NULL);
if (!set.NextRow())
return NULL;
user = new User();
user->_id = set.GetAsString(_("id"));
user->_name = set.GetAsString(_("name"));
user->_password = _("") ; // Security reasons set.GetAsString("password");
set.Finalize();
2010-05-16 10:35:34 +02:00
req = _("SELECT * FROM account WHERE user='") + user->_id + _("' ORDER BY default_account DESC, name ASC");
2010-05-15 11:21:42 +02:00
2010-05-15 11:45:57 +02:00
EXECUTE_SQL_QUERY_WITH_CODE(req, set, NULL, {delete user;}, {delete user;});
2010-05-15 11:21:42 +02:00
while (set.NextRow())
{
account = new Account();
account->_id = set.GetAsString(_("id"));
account->_name = set.GetAsString(_("name"));
account->_number = set.GetAsString(_("number"));
account->_amount = set.GetInt(_("amount"));
account->_shared = set.GetBool(_("shared"));
account->_default = set.GetBool(_("default_account"));
user->_accounts.push_back(account);
}
set.Finalize();
if (!user->_accounts.empty())
{
it = user->_accounts.begin();
2010-05-16 10:35:34 +02:00
req = _("SELECT DISTINCT year FROM operation WHERE account IN('") + (*it)->_id;
2010-05-15 11:21:42 +02:00
it++;
for (;it != user->_accounts.end(); it++)
{
2010-05-16 10:35:34 +02:00
req += _("', '") + (*it)->_id ;
2010-05-15 11:21:42 +02:00
}
2010-05-16 10:35:34 +02:00
req += _("') ORDER BY year ASC");
2010-05-15 11:21:42 +02:00
2010-05-15 11:45:57 +02:00
EXECUTE_SQL_QUERY_WITH_CODE(req, set, NULL, {delete user;}, {delete user;});
2010-05-15 11:21:42 +02:00
while (set.NextRow())
{
user->_operations[set.GetInt(_("year"))] = NULL;
}
set.Finalize();
}
return user;
}
2010-05-16 10:35:34 +02:00
void Database::LoadYear(User* user, int year)
{
wxSQLite3ResultSet set;
wxString req;
std::list<Account*>::iterator it;
if (user->_operations[year] == NULL)
user->_operations[year] = new std::map<unsigned int, std::list<operation> >();
it = user->_accounts.begin();
req = _("SELECT * FROM operation WHERE account IN('") + (*it)->_id;
it++;
for (;it != user->_accounts.end(); it++)
{
req += _("', '") + (*it)->_id ;
}
req += _("') ORDER BY fix_cost DESC, year,month,day ASC");
EXECUTE_SQL_QUERY(req, set, );
while (set.NextRow())
{
operation op;
op.id = set.GetAsString(_("id"));
2010-05-16 20:09:18 +02:00
op.account = set.GetAsString(_("account"));
2010-05-16 10:35:34 +02:00
op.day = set.GetInt(_("day"));
op.month = set.GetInt(_("month"));
op.year = set.GetInt(_("year"));
op.amount = set.GetInt(_("amount"));
op.description = set.GetAsString(_("description"));
op.category = set.GetAsString(_("category"));
op.fix_cost = set.GetBool(_("fix_cost"));
(*user->_operations[op.year])[op.month].push_back(op);
}
set.Finalize();
}