KissCount/src/model/Database.hpp

/*
  Copyright 2010-2011 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/>.
*/

#ifndef DATABASE_H
#define DATABASE_H

#include <list>
#include <wx/wxsqlite3.h>
#include <wx/wx.h>

#include <controller/KissCount.hpp>
#include "model.hpp"

#define BDD_FILE     "/.kisscount/kc.bdd"
#define INIT_SCRIPT  RESSOURCES_ROOT "init.sql"


// 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)					\
        {                                                               \
            wxMessageBox(_("Update failed !\n") + req, _("Error"), wxICON_ERROR | wxOK); \
            std::cerr << __FUNCTION__ << "\n" ;                         \
            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)					\
        {                                                               \
            wxMessageBox(_("Query failed !\n") + req, _("Error"), wxICON_ERROR | wxOK); \
            std::cerr << __FUNCTION__ << "\n" ;                         \
            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, {}, {})

class KissCount;
class User;

class Database
{
public:
    static const int FIX_OP         = 1;//(1 << 0);
    static const int NON_FIX_OP     = 2;//(1 << 1);
    static const int CHECKED_OP     = (1 << 2);
    static const int NOT_CHECKED_OP = (1 << 3);
    static const int ALL_OP         = (~0);

    static const int VERSION        = 2;

    Database(const char* filename, KissCount* kiss);

    std::list<wxString> GetUsers();
    bool IsValidUser(const wxString& user, const wxString& password);

    User* LoadUser(const wxString& name);
    void LoadYear(User* user, int year);

    void UpdateOperation(User* user, Operation& op, bool checkTransfert=true);
    wxString AddOperation(User* user, Operation& op, bool checkTransfert=true);
    void DeleteOperation(User* user, Operation& op);
    void DeleteOperations(User* user, int month, int year);
    bool LoadOperation(User* user, const wxString& id);
    double MetaAmount(const wxString& id);
    double MetaPositiveAmount(const wxString& id);

    double GetAccountAmount(const wxString& id, int month, int year);
    void SetAccountAmount(const wxString& accountId, int month, int year, double amount);
    double CalcAccountAmount(const wxString& id, int month, int year, bool* had_values);

    wxString AddAccount(User* user, Account& ac);
    void UpdateAccount(Account& ac);
    void DeleteAccount(User* user, Account& ac, const wxString& replacement);
    void AddSharedAccount(Account& ac, const wxString& granted);
    void RemoveSharedAccount(Account& ac, const wxString& granted);

    wxString AddCategory(User* user, Category& category);
    void UpdateCategory(Category& category);
    void DeleteCategory(User* user, Category& category, const wxString& replacement);
    bool LoadCategory(const wxString& id, const wxString& name, Category& category);

    std::map<int, std::vector<int> > GetAllOperations(User* user);
    void GenerateMonth(User* user, int monthFrom, int yearFrom, int monthTo, int yearTo);

    void ChangePassword(User* user, const wxString& password);
    bool UserExists(const wxString& name);
    void ChangeName(User* user, const wxString& name);
    void NewUser(const wxString& name);

    void UpdatePreference(User* user, const wxString& preference);

    std::vector<Operation>* Search(User* user, wxString* description, wxDateTime* dateFrom, wxDateTime* dateTo,
                                   wxString* amountFrom, wxString* amountTo,
                                   std::vector<wxString> categories, int types, std::vector<wxString> accounts, bool wildcards);
  
    void GetStats(User* user, const wxString& monthFrom, const wxString& yearFrom, const wxString& monthTo, 
                  const wxString& yearTo, std::map<wxString, std::map<int, std::map<int, double> > >* accountAmounts, 
                  std::map<wxString, double>* categories);

    void GetMonthStats(User* user, const wxString& month, const wxString& year, int nbDays,
                       std::map<wxString, std::vector<double> >* operations,
                       std::map<wxString, double>* categories);

    void KillMe(User* user);
    bool GetOperation(const wxString& id, Operation* op);
    std::map<wxString, wxString> getSharedAccountOwners(const wxString& account);
    wxString getSharedAccountOwner(const wxString& account);

    std::map<wxString, double>* GetNotChecked(User* user, int month, int year);
    std::map<wxString, double>* GetVirtualAmount(User* user, int month, int year);

    void UpdateImportPattern(User* user);

/* Database Update */
    
    void CheckDatabaseVersion();

private:
    KissCount* _kiss;
    wxSQLite3Database _db;
    
    void CreateDatabase();
    wxString HashPassword(const wxString& password);
    void LinkOrUnlinkOperation(User* user, Operation& op);
};

#endif