/*
Copyright 2010 Grégory Soutadé

This file is part of KissCount.

KissCount is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

KissCount is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with KissCount.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "Database.h"

// if (!_db.CheckSyntax(req))			
// 	{
// wxString s = req;
// 	  std::cout << s.mb_str() << " is invalid !\n" ;	
// code_if_syntax_fail;					
// return return_value;
// 	}					

#define EXECUTE_SQL_UPDATE_WITH_CODE(req, return_value, code_if_fail, code_if_syntax_fail) \
  do{									\
    try									\
      {									\
	_db.ExecuteUpdate(req);						\
      }									\
    catch (wxSQLite3Exception e)					\
      {									\
	std::cerr << req.mb_str() << "\n" ;				\
	std::cerr << e.GetMessage().mb_str() << "\n" ;			\
	code_if_fail;							\
	return return_value;						\
      }									\
  } while(0);

#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 << req.mb_str() << "\n" ;				\
	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, {}, {})

#define EXECUTE_SQL_UPDATE(req, return_value) EXECUTE_SQL_UPDATE_WITH_CODE(req, return_value, {}, {})

static inline wxString DoubleToString(double d)
{
  wxString res;

  res = wxString::Format(wxT("%.2lf"), d);
  res.Replace(wxT(","), wxT("."));

  return res;
}

Database::Database()
{
  std::ifstream bdd_file;

  bdd_file.open(BDD_FILE);

  if (!bdd_file)
    {
      CreateDatabase();
    }
  else
    _db.Open(wxT(BDD_FILE));

  bdd_file.close();
}

void Database::CreateDatabase()
{
  std::ifstream init_script;
  std::string line;
  wxString wxline;

  wxMessageBox(_("No database found, creating a new one"), wxT("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(wxT(BDD_FILE));

  do
    {
      getline(init_script, line);
      wxline = wxString(line.c_str(), wxConvUTF8);
      wxline.Trim(false);
      if (!wxline.Length() || wxline.StartsWith(wxT("--"))) continue;
      if (!_db.CheckSyntax(wxline))
	{
	  std::cout << line << " is invalid !\n" ;
	  continue;
	}
      try
	{
	  _db.ExecuteUpdate(wxline);
	}
      catch (...)
	{
	  wxMessageBox(_("Error creating original database"), _("Error"), wxICON_ERROR | wxOK );
	  remove(BDD_FILE);
	  throw line;
	}
    } while (init_script);

  init_script.close();
}


wxString Database::HashPassword(const wxString& password)
{
  blk_SHA_CTX sha_ctx;
  unsigned char sha[20];
  wxString wxSHA;

  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++)
    wxSHA += wxString::Format(wxT("%02x"), (int)sha[i]);

  return wxSHA;
}

std::list<wxString> Database::GetUsers()
{
  std::list<wxString> res;
  wxString req;
  // Check whether value exists in table
  wxSQLite3ResultSet set ;

  req = wxT("SELECT name FROM user ORDER BY name");
  EXECUTE_SQL_QUERY(req, set, res);
	
  while (set.NextRow())
    {
      res.push_front(set.GetAsString(0));
    }
  set.Finalize();

  return res;
}

bool Database::IsValidUser(const wxString& user, const wxString& password)
{
  bool res;
  wxString req;
  wxSQLite3ResultSet set;

  req = wxT("SELECT name FROM user WHERE name='") + user + wxT("' AND password='") + HashPassword(password) + wxT("'"); 

  EXECUTE_SQL_QUERY(req, set, false);
	
  res = set.NextRow() ;
  set.Finalize();

  return res;
}

User* Database::LoadUser(const wxString& name)
{
  wxSQLite3ResultSet set;
  wxString req;
  User* user;
  Account account;
  Category category;

  std::vector<Account>::iterator it;

  req = wxT("SELECT * FROM user WHERE name='") + name + wxT("'"); 

  EXECUTE_SQL_QUERY(req, set, NULL);

  if (!set.NextRow())
    return NULL;

  user = new User();
  
  user->_id = set.GetAsString(wxT("id"));
  user->_name = set.GetAsString(wxT("name"));
  user->_password = wxT("") ; // Security reasons set.GetAsString("password"); 

  user->_preferences[wxT("operation_order")] = wxT("ASC") ;

  set.Finalize();

  req = wxT("SELECT * FROM account WHERE user='") + user->_id + wxT("' ORDER BY default_account DESC, name ASC"); 

  EXECUTE_SQL_QUERY_WITH_CODE(req, set, NULL, {delete user;}, {delete user;});

  while (set.NextRow())
    {
      account.id = set.GetAsString(wxT("id"));
      account.name = set.GetAsString(wxT("name"));
      account.number = set.GetAsString(wxT("number"));
      account.shared = set.GetBool(wxT("shared"));
      account._default = set.GetBool(wxT("default_account"));
      user->_accounts.push_back(account);
    }
  set.Finalize();

  if (!user->_accounts.empty())
    {
      it = user->_accounts.begin();
      req = wxT("SELECT DISTINCT year FROM operation WHERE account IN('") + it->id;
      it++;
      for (;it != user->_accounts.end(); it++)
	{
	  req += wxT("', '") + it->id ;
	}
      req += wxT("')");
      req += wxT(" OR user='") + user->_id + wxT("'");
      req += wxT(" ORDER BY year ASC");

      EXECUTE_SQL_QUERY_WITH_CODE(req, set, NULL, {delete user;}, {delete user;});

      while (set.NextRow())
	{
	  user->_operations[set.GetInt(wxT("year"))] = NULL;
	}
      set.Finalize();
    }

  req = wxT("SELECT * FROM category WHERE user='") + user->_id + wxT("' ORDER BY name ASC");
  EXECUTE_SQL_QUERY_WITH_CODE(req, set, NULL, {delete user;}, {delete user;});

  while (set.NextRow())
    {
      category.id = set.GetAsString(wxT("id"));
      category.parent = set.GetAsString(wxT("parent"));
      category.name = set.GetAsString(wxT("name"));
      category.color = wxColour(set.GetAsString(wxT("color")));
      category.font = set.GetAsString(wxT("font"));
      if (category.name != _("Fix"))
	user->_categories.push_back(category);
      else
	user->_categories.insert(user->_categories.begin(), category);
    }

  set.Finalize();

  req = wxT("SELECT name, value FROM preference WHERE user='") + user->_id + wxT("' ORDER BY value ASC");
  EXECUTE_SQL_QUERY_WITH_CODE(req, set, NULL, {delete user;}, {delete user;});

  while (set.NextRow())
    user->_preferences[set.GetAsString(wxT("name"))] = set.GetAsString(wxT("value"));

  set.Finalize();

  return user;
}

void Database::LoadYear(User* user, int year)
{
  wxSQLite3ResultSet set;
  wxString req;
  std::vector<Account>::iterator it;

  if (user->_operations[year] == NULL)
    user->_operations[year] = new std::map<unsigned int, std::vector<Operation> >();

  if (!user->_accounts.size()) return;

  it = user->_accounts.begin();
  req = wxT("SELECT * FROM operation WHERE (account IN('") + it->id;
  it++;
  for (;it != user->_accounts.end(); it++)
    {
      req += wxT("', '") + it->id ;
    }
  req += wxT("')");
  req += wxT(" OR user='") + user->_id + wxT("')");
  req += wxT(" AND year='") + wxString::Format(wxT("%d"), year) + wxT("'");
  req += wxT(" ORDER BY fix_cost DESC, year, month ASC, day ");
  req += user->_preferences[wxT("operation_order")];

  EXECUTE_SQL_QUERY(req, set, );

  while (set.NextRow())
    {
      Operation op;
      op.id = set.GetAsString(wxT("id"));
      op.parent = set.GetAsString(wxT("parent"));
      op.account = set.GetAsString(wxT("account"));
      op.day = set.GetInt(wxT("day"));
      op.month = set.GetInt(wxT("month"));
      op.year = set.GetInt(wxT("year"));
      op.amount = set.GetDouble(wxT("amount"));
      op.description = set.GetAsString(wxT("description"));
      op.category = set.GetAsString(wxT("category"));
      op.fix_cost = set.GetBool(wxT("fix_cost"));
      op.checked = set.GetBool(wxT("checked"));
      op.formula = set.GetAsString(wxT("formula"));      
      (*user->_operations[op.year])[op.month].push_back(op);
    }

  set.Finalize();
}

double Database::GetAccountAmount(const wxString& id, int month, int year)
{
  wxSQLite3ResultSet set;
  wxString req;
  double res;

  req = wxT("SELECT amount FROM account_amount WHERE account='") + id ;
  req += wxT("' AND month='") + wxString::Format(wxT("%d"), month);
  req += wxT("' AND year='") + wxString::Format(wxT("%d"), year);
  req += wxT("'");

  EXECUTE_SQL_QUERY(req , set, 0.0);

  if (set.NextRow())
    res = set.GetDouble(wxT("amount"));
  else
    {
      SetAccountAmount(month, year, id, 0.0);
      res = 0.0;
    }

  set.Finalize();

  return res;
}

void Database::UpdateOperation(Operation& op)
{
  wxString req;
  req = wxT("UPDATE operation SET ") ;
  req += wxT("parent='") + op.parent + wxT("'");
  req += wxT(", account='") + op.account + wxT("'");
  req += wxT(", year='") + wxString::Format(wxT("%d"), op.year) + wxT("'");
  req += wxT(", month='") + wxString::Format(wxT("%d"), op.month) + wxT("'");
  req += wxT(", day='") + wxString::Format(wxT("%d"), op.day) + wxT("'");
  req += wxT(", amount='") + DoubleToString(op.amount) + wxT("'");
  req += wxT(", description=\"") + op.description + wxT("\"");
  req += wxT(", category='") + op.category + wxT("'");
  if (op.checked)
    req += wxT(", checked='1'");
  else
    req += wxT(", checked='0'");
  req += wxT(", forumla='") + op.formula + wxT("'");
  req += wxT(" WHERE id='") + op.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

wxString Database::AddOperation(User* user, Operation& op)
{
  wxString req, res;
  wxSQLite3ResultSet set;

  req = wxT("INSERT INTO operation ('user', 'parent', 'account', 'year', 'month', 'day', 'amount', 'description', 'category', 'fix_cost', 'formula') VALUES ('") ;
  req += user->_id + wxT("'");
  req += wxT(", '") + op.parent + wxT("'");
  req += wxT(", '") + op.account + wxT("'");
  req += wxT(", '") + wxString::Format(wxT("%d"), op.year) + wxT("'");
  req += wxT(", '") + wxString::Format(wxT("%d"), op.month) + wxT("'");
  req += wxT(", '") + wxString::Format(wxT("%d"), op.day) + wxT("'");
  req += wxT(", '") + DoubleToString(op.amount) + wxT("'");
  req += wxT(", \"") + op.description + wxT("\"");
  req += wxT(", '") + op.category + wxT("'");
  if (op.fix_cost)
    req += wxT(", '1'") ;
  else
    req += wxT(", '0'") ;
  req += wxT(", '") + op.formula + wxT("'");
  req += wxT(")");
	    
  EXECUTE_SQL_UPDATE(req, wxT("0"));

  req = wxT("SELECT id FROM operation WHERE ");
  req += wxT("user='") + user->_id + wxT("'");
  req += wxT(" AND parent='") + op.parent + wxT("'");
  req += wxT(" AND account='") + op.account + wxT("'");
  req += wxT(" AND year='") + wxString::Format(wxT("%d"), op.year) + wxT("'");
  req += wxT(" AND month='") + wxString::Format(wxT("%d"), op.month) + wxT("'");
  req += wxT(" AND day='") + wxString::Format(wxT("%d"), op.day) + wxT("'");
  req += wxT(" AND amount='") + DoubleToString(op.amount) + wxT("'");
  req += wxT(" AND description=\"") + op.description + wxT("\"");
  req += wxT(" AND category='") + op.category + wxT("'");
  if (op.fix_cost)
    req += wxT(" AND fix_cost='1'") ;
  else
    req += wxT(" AND fix_cost='0'") ;
  req += wxT(" AND formula='") + op.formula + wxT("'");
  req += wxT("ORDER BY id DESC") ;

  EXECUTE_SQL_QUERY(req , set, wxT("0"));

  if (set.NextRow())
    res = set.GetAsString(wxT("id"));
  else
    res = wxT("0");

  set.Finalize();

  return res;
}

void Database::DeleteOperation(Operation& op)
{
  wxString req;
  req = wxT("DELETE FROM operation WHERE id='") + op.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

void Database::DeleteOperations(User* user, int month, int year)
{
  wxString req;
  std::vector<Account>::iterator it;

  it = user->_accounts.begin();
  req = wxT("DELETE FROM account_amount WHERE account IN('") + it->id;
  it++;
  for (;it != user->_accounts.end(); it++)
    {
      req += wxT("', '") + it->id ;
    }
  req += wxT("')");
  req += wxT(" AND year='") + wxString::Format(wxT("%d"), year) + wxT("'");
  if (month != -1)
    req += wxT(" AND month='") + wxString::Format(wxT("%d"), month) + wxT("'");

  EXECUTE_SQL_UPDATE(req, );

  it = user->_accounts.begin();
  req = wxT("DELETE FROM operation WHERE (account IN('") + it->id;
  it++;
  for (;it != user->_accounts.end(); it++)
    {
      req += wxT("', '") + it->id ;
    }
  req += wxT("')");
  req += wxT(" OR user='") + user->_id + wxT("'"); 
  req += wxT(" AND year='") + wxString::Format(wxT("%d"), year) + wxT("'");
  if (month != -1)
    req += wxT(" AND month='") + wxString::Format(wxT("%d"), month) + wxT("'");

  EXECUTE_SQL_UPDATE(req, );

}

void Database::SetAccountAmount(int month, int year, const wxString& accountId, double amount)
{
  wxString req;
  req = wxT("UPDATE account_amount SET ") ;
  req += wxT("amount='") + DoubleToString(amount) + wxT("'");
  req += wxT(" WHERE account='") + accountId + wxT("'");
  req += wxT(" AND year='") + wxString::Format(wxT("%d"), year) + wxT("'");
  req += wxT(" AND month='") + wxString::Format(wxT("%d"), month) + wxT("'");

  try
    {
      if (!_db.ExecuteUpdate(req))
	{
	  req = wxT("INSERT INTO account_amount ('account', 'year', 'month', 'amount') VALUES ('") ;
	  req += accountId + wxT("'");
	  req += wxT(" ,'") + wxString::Format(wxT("%d"), year) + wxT("'");
	  req += wxT(" ,'") + wxString::Format(wxT("%d"), month) + wxT("'");
	  req += wxT(" ,'") + DoubleToString(amount) + wxT("'");
	  req += wxT(")");
	  EXECUTE_SQL_UPDATE(req, );
	}
    }
  catch (wxSQLite3Exception e)
    {
      std::cerr << req.mb_str() << "\n" ;
      std::cerr << e.GetMessage().mb_str() << "\n" ;
      return ;
    }
}

wxString Database::AddAccount(User* user, Account& ac)
{
  wxString req, res;
  wxSQLite3ResultSet set;

  req = wxT("INSERT INTO account ('user', 'name', 'number', 'shared', 'default_account') VALUES ('") ;
  req += user->_id + wxT("'");
  req += wxT(", '") + ac.name + wxT("'");
  req += wxT(", '") + ac.number + wxT("'");
  if (ac.shared)
    req += wxT(", '1'") ;
  else
    req += wxT(", '0'") ;
  if (ac._default)
    req += wxT(", '1'") ;
  else
    req += wxT(", '0'") ;
  req += wxT(")");
	    
  EXECUTE_SQL_UPDATE(req, wxT("0"));

  req = wxT("SELECT id FROM account WHERE name='") + ac.name + wxT("'") ;
  req += wxT("AND user='") + user->_id + wxT("'");

  EXECUTE_SQL_QUERY(req , set, wxT("0"));

  if (set.NextRow())
    res = set.GetAsString(wxT("id"));
  else
    res = wxT("0");

  set.Finalize();

  return res;
}

void Database::UpdateAccount(Account& ac)
{
  wxString req;
  req = wxT("UPDATE account SET ") ;
  req += wxT("name='") + ac.name + wxT("'");
  req += wxT(", number='") + ac.number + wxT("'");
  if (ac.shared)
    req += wxT(", shared='1'");
  else
    req += wxT(", shared='0'");
  if (ac._default)
    req += wxT(", default_account='1'");
  else
    req += wxT(", default_account='0'");
  req += wxT(" WHERE id='") + ac.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

void Database::DeleteAccount(Account& ac)
{
  wxString req;
  req = wxT("DELETE FROM account WHERE id='") + ac.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

wxString Database::AddCategory(User* user, Category& category)
{
  wxString req, res;
  wxSQLite3ResultSet set;
  wxString color;

  color = wxT("#") ;
  color += wxString::Format(wxT("%02X"), category.color.Red());
  color += wxString::Format(wxT("%02X"), category.color.Green());
  color += wxString::Format(wxT("%02X"), category.color.Blue());

  req = wxT("INSERT INTO category ('user', 'parent', 'name', 'color', font) VALUES ('") ;
  req += user->_id + wxT("'");
  req += wxT(", '") + category.parent + wxT("'");
  req += wxT(", '") + category.name + wxT("'");
  req += wxT(", '") + color + wxT("'");
  req += wxT(", '") + category.font + wxT("'");
  req += wxT(")");
	    
  EXECUTE_SQL_UPDATE(req, wxT("0"));

  req = wxT("SELECT id FROM preference WHERE user='") + user->_id + wxT("'") ;
  req += wxT(" AND name='") + category.name + wxT("'");

  EXECUTE_SQL_QUERY(req , set, wxT("0"));

  if (set.NextRow())
    res = set.GetAsString(wxT("id"));
  else
    res = wxT("0");

  set.Finalize();

  return res;
}

void Database::UpdateCategory(Category& category)
{
  wxString req;
  wxString color;

  color = wxT("#") ;
  color += wxString::Format(wxT("%02X"), category.color.Red());
  color += wxString::Format(wxT("%02X"), category.color.Green());
  color += wxString::Format(wxT("%02X"), category.color.Blue());

  req = wxT("UPDATE category SET") ;
  req += wxT(" parent='") + category.parent + wxT("'");
  req += wxT(", name='") + category.name + wxT("'");
  req += wxT(", color='") + color + wxT("'");
  req += wxT(", font='") + category.font + wxT("'");
  req += wxT(" WHERE id='") + category.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

void Database::DeleteCategory(User* user, Category& category)
{
  wxString req;

  req = wxT("DELETE FROM category WHERE id='") + category.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );

  req = wxT("UPDATE category SET") ;
  req += wxT(" parent='0'");
  req += wxT(" WHERE parent='") + category.id + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

std::map<int, std::vector<int> > Database::GetAllOperations(User* user)
{
  wxString req, req2, reqUnion;
  wxSQLite3ResultSet set, set2;
  std::vector<Account>::iterator it;
  std::map<int, std::vector<int> > res;
  int year;

  if (!user->_accounts.empty())
    {
      it = user->_accounts.begin();
      req = wxT("SELECT DISTINCT year FROM account_amount WHERE account IN('") + it->id;
      it++;
      for (;it != user->_accounts.end(); it++)
	{
	  req += wxT("', '") + it->id ;
	}
      req += wxT("')"); 

      it = user->_accounts.begin();
      req2 = wxT("SELECT DISTINCT year FROM operation WHERE account IN('") + it->id;
      it++;
      for (;it != user->_accounts.end(); it++)
	{
	  req2 += wxT("', '") + it->id ;
	}
      req2 += wxT("')"); 
      req2 += wxT(" OR user='") + user->_id + wxT("'"); 
      req2 += wxT(" ORDER BY year ASC");

      reqUnion = req + wxT(" UNION ") + req2;
      EXECUTE_SQL_QUERY(reqUnion, set, res);

      while (set.NextRow())
	{
	  year = set.GetInt(wxT("year"));

	  it = user->_accounts.begin();
	  req = wxT("SELECT DISTINCT month FROM account_amount WHERE account IN('") + it->id;
	  it++;
	  for (;it != user->_accounts.end(); it++)
	    {
	      req += wxT("', '") + it->id ;
	    }
	  req += wxT("')");
	  req += wxT(" AND year='") + set.GetAsString(wxT("year")) + wxT("'");

	  it = user->_accounts.begin();
	  req2 = wxT("SELECT DISTINCT month FROM operation WHERE (account IN('") + it->id;
	  it++;
	  for (;it != user->_accounts.end(); it++)
	    {
	      req2 += wxT("', '") + it->id ;
	    }
	  req2 += wxT("')");
	  req2 += wxT(" OR user='") + user->_id + wxT("')"); 
	  req2 += wxT(" AND year='") + set.GetAsString(wxT("year")) + wxT("'");
	  req2 += wxT(" ORDER BY month ASC");

	  reqUnion = req + wxT(" UNION ") + req2;
	  EXECUTE_SQL_QUERY(reqUnion, set2, res);

	  while (set2.NextRow())
	    {
	      res[year].push_back(set2.GetInt(wxT("month")));
	    }
	  set2.Finalize();
	}
      set.Finalize();
    }

  return res;
}

void Database::GenerateMonth(User* user, int monthFrom, int yearFrom, int monthTo, int yearTo)
{
  std::vector<Account>::iterator it;
  wxString req;
  wxSQLite3ResultSet set;
  double amount;

  if (monthFrom == -1 || yearFrom == -1)
    {
      for (it = user->_accounts.begin(); it != user->_accounts.end(); it++)
	{
	  req = wxT("INSERT INTO account_amount ('account', 'year', 'month', 'amount') VALUES ('") ;
	  req += it->id + wxT("'");
	  req += wxT(" ,'") + wxString::Format(wxT("%d"), yearTo) + wxT("'");
	  req += wxT(" ,'") + wxString::Format(wxT("%d"), monthTo) + wxT("'");
	  req += wxT(" ,'0.0'");
	  req += wxT(")");
	  EXECUTE_SQL_UPDATE(req, );
	}

      return;
    }

  for (it = user->_accounts.begin(); it != user->_accounts.end(); it++)
    {
      amount = 0.0;
      req = wxT("SELECT SUM(amount) AS total FROM operation WHERE") ;
      req += wxT(" account='") + it->id + wxT("'");
      req += wxT(" AND year='") + wxString::Format(wxT("%d"), yearFrom) + wxT("'");
      req += wxT(" AND month='") + wxString::Format(wxT("%d"), monthFrom) + wxT("'");

      EXECUTE_SQL_QUERY(req, set, );

      if (set.NextRow())
	amount += set.GetDouble(wxT("total"));

      req = wxT("SELECT amount FROM account_amount WHERE") ;
      req += wxT(" account='") + it->id + wxT("'");
      req += wxT(" AND year='") + wxString::Format(wxT("%d"), yearFrom) + wxT("'");
      req += wxT(" AND month='") + wxString::Format(wxT("%d"), monthFrom) + wxT("'");

      EXECUTE_SQL_QUERY(req, set, );

      if (set.NextRow())
	amount += set.GetDouble(wxT("amount"));

      req = wxT("INSERT INTO account_amount ('account', 'year', 'month', 'amount') VALUES ('") ;
      req += it->id + wxT("'");
      req += wxT(" ,'") + wxString::Format(wxT("%d"), yearTo) + wxT("'");
      req += wxT(" ,'") + wxString::Format(wxT("%d"), monthTo) + wxT("'");
      req += wxT(" ,'") + DoubleToString(amount) + wxT("'");
      req += wxT(")");

      EXECUTE_SQL_UPDATE(req, );
    }
}

void Database::ChangePassword(User* user, const wxString& password)
{
  wxString req;

  req = wxT("UPDATE user SET ") ;
  req += wxT("password='") + HashPassword(password) + wxT("'");
  req += wxT(" WHERE name='") + user->_name + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

bool Database::UserExists(const wxString& name)
{
  wxSQLite3ResultSet set;
  wxString req;
  bool res=false;

  req = wxT("SELECT name FROM user WHERE name='") + name + wxT("'") ;

  EXECUTE_SQL_QUERY(req , set, false);

  if (set.NextRow())
    res = true;
  else
    res = false;

  set.Finalize();

  return res;
}

void Database::ChangeName(User* user, const wxString& name)
{
  wxString req;

  req = wxT("UPDATE user SET ") ;
  req += wxT("name='") + name + wxT("'");
  req += wxT(" WHERE name='") + user->_name + wxT("'");

  EXECUTE_SQL_UPDATE(req, );
}

void Database::NewUser(const wxString& name)
{
  wxString req, id;
  wxSQLite3ResultSet set;

  req = wxT("INSERT INTO user ('name', 'password') VALUES ('") ;
  req += name + wxT("'");
  req += wxT(", '") + HashPassword(wxT("")) + wxT("'");
  req += wxT(")");
	    
  EXECUTE_SQL_UPDATE(req, );

  req = wxT("SELECT id FROM user WHERE ");
  req += wxT("name='") + name + wxT("'");

  EXECUTE_SQL_QUERY(req , set, );

  set.NextRow();
  id = set.GetAsString(wxT("id"));

  set.Finalize();

  req = wxT("SELECT * FROM default_preference");

  EXECUTE_SQL_QUERY(req, set,);

  while (set.NextRow())
    {
      req = wxT("INSERT INTO preference ('user', 'name', 'value') VALUES ('") ;
      req += id + wxT("'");
      req += wxT(", '") + set.GetAsString(wxT("type")) + wxT("'");
      req += wxT(", '") + set.GetAsString(wxT("value")) + wxT("'");
      req += wxT(")");

      EXECUTE_SQL_UPDATE(req, );
    }

  set.Finalize();

  return ;
}

/*
  Shared accounts not currently supported
*/
void Database::KillMe(User* user)
{
  wxString req;
  std::vector<Account>::iterator it;

  req = wxT("DELETE FROM preference WHERE user='") + user->_id + wxT("'");
  EXECUTE_SQL_UPDATE(req, );

  if (!user->_accounts.empty())
    {
      it = user->_accounts.begin();
      req = wxT("DELETE FROM account_amount WHERE account IN('") + it->id;
      it++;
      for (;it != user->_accounts.end(); it++)
	{
	  req += wxT("', '") + it->id ;
	}
      req += wxT("')");

      EXECUTE_SQL_UPDATE(req, );

      it = user->_accounts.begin();
      req = wxT("DELETE FROM operation WHERE account IN('") + it->id;
      it++;
      for (;it != user->_accounts.end(); it++)
	{
	  req += wxT("', '") + it->id ;
	}
      req += wxT("')");
      req += wxT(" OR user='") + user->_id + wxT("'"); 

      EXECUTE_SQL_UPDATE(req, );
    }

  req = wxT("DELETE FROM account WHERE user='") + user->_id + wxT("'");
  EXECUTE_SQL_UPDATE(req, );

  req = wxT("DELETE FROM category WHERE user='") + user->_id + wxT("'");
  EXECUTE_SQL_UPDATE(req, );

  req = wxT("DELETE FROM user WHERE id='") + user->_id + wxT("'");
  EXECUTE_SQL_UPDATE(req, );
}

void Database::UpdatePreference(User* user, const wxString& preference)
{
  wxString req;
  wxString value = user->_preferences[preference];

  req = wxT("UPDATE preference SET ") ;
  req += wxT("name='") + preference + wxT("'");
  req += wxT(", value='") + value + wxT("'");
  req += wxT(" WHERE user='") + user->_id + wxT("'");

  try
    {
      if (!_db.ExecuteUpdate(req))
	{
	  req = wxT("INSERT INTO preference ('user', 'name', 'value') VALUES ('") ;
	  req += user->_id + wxT("'");
	  req += wxT(" ,'") + preference + wxT("'");
	  req += wxT(" ,'") + value + wxT("'");
	  req += wxT(")");
	  EXECUTE_SQL_UPDATE(req, );
	}
    }
  catch (wxSQLite3Exception e)
    {
      std::cerr << req.mb_str() << "\n" ;
      std::cerr << e.GetMessage().mb_str() << "\n" ;
      return ;
    }
}

std::vector<Operation>* Database::Search(User* user, wxString* description, wxDateTime* dateFrom, wxDateTime* dateTo,
					 wxString* amountFrom, wxString* amountTo,
					 std::vector<wxString> categories, std::vector<wxString> accounts)
{
  wxSQLite3ResultSet set;
  wxString req;
  bool firstCond = false;
  std::vector<wxString>::iterator it;
  std::vector<Account>::iterator accountIt;
  std::vector<Operation>* res = new std::vector<Operation>;

  req = wxT("SELECT * from operation WHERE ");
  
  if (description)
    {
      req += wxT("description LIKE '%") + *description + wxT("%'");
      firstCond = true;
    }

  if (dateFrom)
    {
      if (firstCond) req += wxT(" AND ") ; else firstCond = true;
      req += wxT("(");
      req += wxT(" year >= '") + wxString::Format(wxT("%d"), dateFrom->GetYear()) + wxT("'");
      //req += wxT(" AND month >= '") + wxString::Format(wxT("%d"), dateFrom->GetMonth()) + wxT("'");
      // req += wxT(" AND day >= '") + wxString::Format(wxT("%d"), dateFrom->GetDay()) + wxT("'");
      req += wxT(")");
    }

  if (dateTo)
    {
      if (firstCond) req += wxT(" AND ") ; else firstCond = true;
      req += wxT("(");
      req += wxT(" year <= '") + wxString::Format(wxT("%d"), dateTo->GetYear()) + wxT("'");
      //req += wxT(" AND month <= '") + wxString::Format(wxT("%d"), dateTo->GetMonth()) + wxT("'");
      // req += wxT(" AND day <= '") + wxString::Format(wxT("%d"), dateTo->GetDay()) + wxT("'");
      req += wxT(")");
    }

  if (amountFrom)
    {
      if (firstCond) req += wxT(" AND ") ; else firstCond = true;
      req += wxT("ABS(amount) >= ") + *amountFrom;
    }

  if (amountTo)
    {
      if (firstCond) req += wxT(" AND ") ; else firstCond = true;
      req += wxT("ABS(amount) <= ") + *amountTo;
    }

  if (categories.size())
    {
      if (firstCond) req += wxT(" AND ") ; else firstCond = true;
      req += wxT("category IN ('");
      it = categories.begin();
      req += *it;
      it++;
      
      for (; it != categories.end(); it++)
	req += wxT("', '") + *it ;

      req += wxT("')");
    }

  if (firstCond) req += wxT(" AND ") ; else firstCond = true;

  if (accounts.size())
    {
      req += wxT("account IN ('");
      it = accounts.begin();
      req += *it;
      it++;
      
      for (; it != categories.end(); it++)
	req += wxT("', '") + *it ;

      req += wxT("')");
    }
  else
    {
      req += wxT("account IN ('");
      accountIt = user->_accounts.begin();
      req += accountIt->id;
      accountIt++;
      for (;accountIt != user->_accounts.end(); accountIt++)
	{
	  req += wxT("', '") + accountIt->id ;
	}
      req += wxT("')");
    }

  req += wxT(" ORDER BY year ") ;
  req += user->_preferences[wxT("operation_order")] ;
  req += wxT(", month ") + user->_preferences[wxT("operation_order")] ;
  req += wxT(", day ") + user->_preferences[wxT("operation_order")] ;

  // std::cout << req.mb_str() << "\n";

  EXECUTE_SQL_QUERY(req, set, res);

  while (set.NextRow())
    {
      Operation op;
      op.id = set.GetAsString(wxT("id"));
      op.account = set.GetAsString(wxT("account"));
      op.day = set.GetInt(wxT("day"));
      op.month = set.GetInt(wxT("month"));
      op.year = set.GetInt(wxT("year"));
      op.amount = set.GetDouble(wxT("amount"));
      op.description = set.GetAsString(wxT("description"));
      op.category = set.GetAsString(wxT("category"));
      op.fix_cost = set.GetBool(wxT("fix_cost"));
      op.checked = set.GetBool(wxT("checked"));

      if (dateFrom &&
	  (int)op.year == dateFrom->GetYear() &&
	  (int)op.month <= dateFrom->GetMonth() &&
	  (int)op.day < dateFrom->GetDay()-1)
	continue;

      if (dateTo &&
	  (int)op.year == dateTo->GetYear() &&
	  (int)op.month >= dateTo->GetMonth() &&
	  (int)op.day > dateTo->GetDay()-1)
	continue;

      res->push_back(op);
    }

  return res;
}