/*
  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 "AccountPanel.h"

enum {SEARCH_ID, GRID_ID, CALENDAR_FROM_ID, CALENDAR_TO_ID};

BEGIN_EVENT_TABLE(SearchPanel, wxPanel)
EVT_BUTTON(SEARCH_ID, SearchPanel::OnButtonSearch)
EVT_CALENDAR_SEL_CHANGED(CALENDAR_FROM_ID, SearchPanel::OnCalendarFromChange)
EVT_CALENDAR_SEL_CHANGED(CALENDAR_TO_ID, SearchPanel::OnCalendarToChange)
EVT_GRID_CMD_CELL_CHANGE(GRID_ID, SearchPanel::OnOperationModified)
EVT_SHOW(SearchPanel::OnShow)
END_EVENT_TABLE()

#define SET_ROW_COLOR(row, color)  for(int i=0; i<NUMBER_COLS_OPS; i++)	\
    {									\
        _grid->SetCellBackgroundColour(row, i, color);			\
    }

SearchPanel::SearchPanel(KissCount* kiss, wxUI *parent) : wxScrolledWindow(&(*parent)), _kiss(kiss), _wxUI(parent), _operations(NULL)
{
    DEFAULT_FONT(font);
    User* user = _kiss->GetUser();
    std::vector<Account>::iterator accountIt;
    std::vector<Category>::iterator categoryIt;
    wxDateTime firstOfMonth;

    wxBoxSizer *vbox = new wxBoxSizer(wxVERTICAL);
    SetSizer(vbox);

    _checkDateFrom = new wxCheckBox(this, wxID_ANY, _("Date from"));
    _checkDateTo = new wxCheckBox(this, wxID_ANY, _("Date to"));

    _checkDateFrom->SetValue(wxT("1"));
    _checkDateTo->SetValue(wxT("1"));

    wxGridBagSizer *gridBagSizer = new wxGridBagSizer(3, 9);

    firstOfMonth.SetToCurrent();
    firstOfMonth.SetDay(1);
    _calendarFrom = new wxCalendarCtrl(this, CALENDAR_FROM_ID, firstOfMonth, wxDefaultPosition, wxDefaultSize, 
                                       wxCAL_MONDAY_FIRST);
    _calendarTo = new wxCalendarCtrl(this, CALENDAR_TO_ID, wxDefaultDateTime, wxDefaultPosition, wxDefaultSize, 
                                     wxCAL_MONDAY_FIRST);

  
    _description = new wxTextCtrl(this, wxID_ANY);
    wxSize size = _description->GetSize();
    size.SetWidth(size.GetWidth()*2);
    _description->SetMinSize(size);
    _amountFrom = new wxTextCtrl(this, wxID_ANY);
    _amountTo = new wxTextCtrl(this, wxID_ANY);

    _category = new wxCheckListBox(this, wxID_ANY);
    for(categoryIt = user->_categories.begin(); categoryIt != user->_categories.end(); categoryIt++)
        _category->Append(categoryIt->name);
    _account = new wxCheckListBox(this, wxID_ANY);
    for(accountIt = user->_accounts.begin(); accountIt != user->_accounts.end(); accountIt++)
        _account->Append(accountIt->name);

    _searchButton = new wxButton(this, SEARCH_ID, _("Search"));

    wxStaticText* labelDescription = new wxStaticText(this, wxID_ANY, _("Description"));
    wxStaticText* labelAmountFrom = new wxStaticText(this, wxID_ANY, _("Amount from"));
    wxStaticText* labelAmountTo = new wxStaticText(this, wxID_ANY, _("Amount to"));
    wxStaticText* labelCategory = new wxStaticText(this, wxID_ANY, _("Category"));
    wxStaticText* labelAccount = new wxStaticText(this, wxID_ANY, _("Account"));

    gridBagSizer->Add(labelDescription, wxGBPosition(0, 0));
    gridBagSizer->Add(_description, wxGBPosition(1, 0));
    gridBagSizer->Add(_checkDateFrom, wxGBPosition(0, 1));
    gridBagSizer->Add(_calendarFrom, wxGBPosition(1, 1));
    gridBagSizer->Add(_checkDateTo, wxGBPosition(0, 2));
    gridBagSizer->Add(_calendarTo, wxGBPosition(1, 2));
    gridBagSizer->Add(labelAmountFrom, wxGBPosition(0, 3));
    gridBagSizer->Add(_amountFrom, wxGBPosition(1, 3));
    gridBagSizer->Add(labelAmountTo, wxGBPosition(0, 4));
    gridBagSizer->Add(_amountTo, wxGBPosition(1, 4));
    gridBagSizer->Add(labelCategory, wxGBPosition(0, 5));
    gridBagSizer->Add(_category, wxGBPosition(1, 5));
    gridBagSizer->Add(labelAccount, wxGBPosition(0, 6));
    gridBagSizer->Add(_account, wxGBPosition(1, 6));
    gridBagSizer->Add(_searchButton, wxGBPosition(2, 0));

    vbox->Add(gridBagSizer);
    vbox->Add(-1, 20);

    _grid = new GridAccount(_kiss, this, GRID_ID);

    vbox->Add(_grid);

    Fit();

    SetMinSize(wxSize(1024, 640));
    SetScrollbars(10, 10, 100/10, 100/10);

    Hide();
}

SearchPanel::~SearchPanel()
{
    if (_operations) delete _operations;
}

void SearchPanel::OnButtonSearch(wxCommandEvent& event)
{
    wxString *description=NULL, *amountFrom=NULL, *amountTo=NULL;
    std::vector<wxString> categories, accounts;
    wxDateTime *dateFrom=NULL, *dateTo=NULL;
    User* user= _kiss->GetUser();
    int i;
    std::vector<Operation>::iterator it;
    double af, at;

    if (_calendarFrom->GetDate() > _calendarTo->GetDate())
    {
        wxMessageBox(_("Invalid date range"), _("Error"), wxICON_ERROR | wxOK);      
        return;
    }

    if (_amountFrom->GetLineText(0).Length())
    {
        amountFrom = new wxString;
        *amountFrom = _amountFrom->GetLineText(0);
        if (!amountFrom->ToDouble(&af))
	{
            wxMessageBox(_("Invalid amount from"), _("Error"), wxICON_ERROR | wxOK);      
            delete amountFrom;
            return;
	}

        if (af < 0) af *= -1;
    }

    if (_amountTo->GetLineText(0).Length())
    {
        amountTo = new wxString;
        *amountTo = _amountTo->GetLineText(0);
        if (!amountTo->ToDouble(&at))
	{
            wxMessageBox(_("Invalid amount to"), _("Error"), wxICON_ERROR | wxOK);
            delete amountFrom;
            delete amountTo;
            return;
	}

        if (at < 0) at *= -1;
    }

    if (amountFrom && amountTo && af > at)
    {
        wxMessageBox(_("Invalid amount range"), _("Error"), wxICON_ERROR | wxOK);
        delete amountFrom;
        delete amountTo;
        return;
    }

    _grid->DeleteRows(1, _grid->GetNumberRows()-1);

    if (_description->GetLineText(0).Length())
    {
        description = new wxString;
        *description = _description->GetLineText(0);
    }

    if (_checkDateFrom->IsChecked())
    {
        dateFrom = new wxDateTime;
        *dateFrom = _calendarFrom->GetDate();
    }

    if (_checkDateTo->IsChecked())
    {
        dateTo = new wxDateTime;
        *dateTo = _calendarTo->GetDate();
    }

    if (dateFrom && dateTo && *dateFrom > *dateTo)
    {
        ;
    }

    for(i=0; i<user->GetCategoriesNumber(); i++)
        if (_category->IsChecked(i))
            categories.push_back(user->_categories[i].id);

    for(i=0; i<user->GetAccountsNumber(); i++)
        if (_account->IsChecked(i))
            accounts.push_back(user->_accounts[i].id);

    if (_operations) 
        delete _operations;

    _operations = _kiss->Search(description, dateFrom, dateTo, amountFrom, amountTo, categories, accounts);

    if (_operations->size() > 1)
        wxMessageBox(wxString::Format(wxT("%d"), _operations->size()) + _(" entries found"), wxT("KissCount"), wxICON_INFORMATION | wxOK);
    else if (_operations->size() == 1)
        wxMessageBox(_("1 entry found"), wxT("KissCount"), wxICON_INFORMATION | wxOK);
    else
    {
        wxMessageBox(_("No entry found"), wxT("KissCount"), wxICON_INFORMATION | wxOK);
        return;
    }

    for(i=1, it = _operations->begin(); it != _operations->end(); it++, i++)
    {
        _grid->InsertOperation(user, &(*it), i, false, 0, 0);
    }

    _grid->AutoSizeColumn(CATEGORY, false);
    _grid->AutoSizeColumn(DATE, false);
    _grid->AutoSizeColumn(ACCOUNT, false);
    _grid->AutoSizeColumn(DELETE, false);
    _grid->AutoSizeColumn(CHECKED, false);

    _wxUI->Layout();
}

void SearchPanel::OnShow(wxShowEvent& event)
{
    _wxUI->SetTitle(_kiss->GetUser()->_name + wxT(" - ") + _("Search"));
}

void SearchPanel::OnCalendarFromChange(wxCalendarEvent& event)
{
    _checkDateFrom->SetValue(true);
}

void SearchPanel::OnCalendarToChange(wxCalendarEvent& event)
{
    _checkDateTo->SetValue(true);
}

void SearchPanel::OnOperationModified(wxGridEvent& event)
{
    User* user = _kiss->GetUser();
    int row = event.GetRow()-1;
    int col = event.GetCol();
    Operation new_op, cur_op;
    int op_complete = 6;
    wxString value ;
    wxDateTime date;
    static bool inModification = false ;
    wxColour color ;
    unsigned char r, g, b;

    // Avoid recursives calls
    if (inModification) return;

    inModification = true ;
    
    if (event.GetCol() == DEBIT)
        _grid->SetCellValue(event.GetRow(), CREDIT, wxT(""));
    else if (event.GetCol() == CREDIT)
        _grid->SetCellValue(event.GetRow(), DEBIT, wxT(""));

    value = _grid->GetCellValue(event.GetRow(), DESCRIPTION);
    if (value.Length())
    {
        new_op.description = value;
        op_complete--;
    }

    value = _grid->GetCellValue(event.GetRow(), DATE);
    if (value.Length())
    {
        date.ParseFormat(value, wxT("%d/%m/%Y"));
        new_op.day = date.GetDay()-1;
        new_op.month = date.GetMonth();
        new_op.year = date.GetYear();
        op_complete--;
    }
  
    value = _grid->GetCellValue(event.GetRow(), DEBIT);
    if (value.Length())
    {
        value.ToDouble(&new_op.amount);
        new_op.amount *= -1.0;
        op_complete--;
    }

    value = _grid->GetCellValue(event.GetRow(), CREDIT);
    if (value.Length())
    {
        value.ToDouble(&new_op.amount);
        op_complete--;
    }

    value = _grid->GetCellValue(event.GetRow(), CATEGORY);
    if (value.Length())
    {
        new_op.category = user->GetCategoryId(value);
        op_complete--;
    }

    value = _grid->GetCellValue(event.GetRow(), ACCOUNT);
    if (value.Length())
    {
        new_op.account = user->GetAccountId(value);
        op_complete--;
    }

    value = _grid->GetCellValue(event.GetRow(), CHECKED);
    if (value.Length() && value != wxT("0"))
        new_op.checked = true;
    else
        new_op.checked = false;
    op_complete--;

    if (col == CHECKED || col == CATEGORY)
    {
        color = user->GetCategory(new_op.category).color;

        if (new_op.checked)
	{
            r = ((color.Red()*1.5) >= 0xFF) ? 0xFF : color.Red()*1.5 ; 
            g = ((color.Green()*1.5) >= 0xFF) ? 0xFF : color.Green()*1.5 ; 
            b = ((color.Blue()*1.5) >= 0xFF) ? 0xFF : color.Blue()*1.5 ; 
            color.Set(r, g, b, color.Alpha());
	}

        SET_ROW_COLOR(event.GetRow(), color);
    }

    if (col == DELETE)
    {
        wxMessageDialog dialog(_wxUI, _("Are you sure want to delete : \n")+new_op.description, wxT("KissCount"), wxYES_NO);
        if (dialog.ShowModal() == wxID_NO)
	{
            _grid->SetCellValue(event.GetRow(), event.GetCol(), wxT("0"));
            _wxUI->NeedReload();
            inModification = false;
            return;
	}
    }

    // Modify an operation
    cur_op = (*_operations)[row] ;
    new_op.id = cur_op.id;
    new_op.fix_cost = false;
    new_op.transfert = cur_op.transfert;

    if (col == DELETE)
    {
        _grid->DeleteRows(event.GetRow(), 1);
        _kiss->DeleteOperation(cur_op);
        _operations->erase(_operations->begin()+row);
        _wxUI->NeedReload();
        inModification = false ;
        return ;
    }

    (*_operations)[row] = new_op;
    _kiss->UpdateOperation(new_op);

    _wxUI->NeedReload();

    inModification = false ;
}