// frmSearch.cpp - Search form.
//
// (c) 2012-2015 Xestia Software Development.
//
// This file is part of Xestia Address Book.
//
// Xestia Address Book 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, version 3 of the license.
//
// Xestia Address Book 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 Xestia Address Book. If not, see <http://www.gnu.org/licenses/>

#include "frmSearch.h"
#include "../frmMain.h"
#include "../frmContact.h"
#include "frmSearchAccounts.h"
#include "../common/preferences.h"
#include "../common/dirs.h"
#include "../vcard/vcard.h"

#include <thread>

DEFINE_EVENT_TYPE(SE_ADDSEARCHSETTING);
DEFINE_EVENT_TYPE(SE_REMOVESEARCHSETTING);
DEFINE_EVENT_TYPE(SE_RELOADACCOUNTS);
DEFINE_EVENT_TYPE(SE_ADDRESULT);
DEFINE_EVENT_TYPE(SE_SBUPDATE);
DEFINE_EVENT_TYPE(SE_SEARCHFINISHED);
DEFINE_EVENT_TYPE(SE_UPDATERESULT);
DEFINE_EVENT_TYPE(SE_DELETERESULT);
DEFINE_EVENT_TYPE(SE_OPENCONTACT);
DEFINE_EVENT_TYPE(SE_EDITCONTACT);
DEFINE_EVENT_TYPE(SE_REVEALCONTACT);

BEGIN_EVENT_TABLE(frmSearch, wxFrame)
  EVT_COMMAND(wxID_ANY, SE_ADDSEARCHSETTING, frmSearch::AddSearchSetting)
  EVT_COMMAND(wxID_ANY, SE_REMOVESEARCHSETTING, frmSearch::RemoveSearchSetting)
  EVT_COMMAND(wxID_ANY, SE_RELOADACCOUNTS, frmSearch::ReloadAccountList)
  EVT_COMMAND(wxID_ANY, SE_ADDRESULT, frmSearch::AddResult)
  EVT_COMMAND(wxID_ANY, SE_SBUPDATE, frmSearch::SearchBarUpdate)
  EVT_COMMAND(wxID_ANY, SE_SEARCHFINISHED, frmSearch::SearchFinished)
  EVT_COMMAND(wxID_ANY, SE_UPDATERESULT, frmSearch::UpdateResult)
  EVT_COMMAND(wxID_ANY, SE_DELETERESULT, frmSearch::DeleteResult)
  EVT_COMMAND(wxID_ANY, SE_OPENCONTACT, frmSearch::OpenContact)
  EVT_COMMAND(wxID_ANY, SE_EDITCONTACT, frmSearch::EditContact)
  EVT_COMMAND(wxID_ANY, SE_REVEALCONTACT, frmSearch::RevealContact)
END_EVENT_TABLE()

frmSearch::frmSearch( wxWindow* parent )
:
frmSearchADT( parent )
{
	
	XABSearchPanel *InitPanel = new XABSearchPanel( tabSearch );
	InitPanel->EnableButtons(TRUE, FALSE);
	InitPanel->SetupPointers(this);
	InitPanel->SetupInteger(ScrollGen);
	szrSearch->Add(InitPanel, 1, wxEXPAND, 0);
	szrSearch->Fit(tabSearch);
	SearchFrames.insert(std::make_pair(ScrollGen, InitPanel));
	ScrollGen++;
	
	// Get the list of accounts and mark them as true.

	wxString preffilename = GetUserPrefDir();

	XABPreferences preferences(preffilename);	
	
	wxString AccDir;
	wxString AccDirFull;
	wxString AccDirFullSfx;
	
	for (int i = 0; i < preferences.accounts.GetCount(); i++){
		AccDir = preferences.accounts.GetAccountDirectory(i);
		AccDirFull = preferences.accounts.GetAccountDirectory(i);
		AccDirFull.Trim();
		AccDirFull.Append(wxT("."));
		AccDirFullSfx.Append(preferences.accounts.GetAccountType(i));
		AccDirFullSfx.LowerCase();
		AccDirFullSfx.Trim();
		AccDirFull.Append(AccDirFullSfx);
		SearchAccountsPaths.insert(std::make_pair(AccDir, AccDirFull));
		SearchAccounts.insert(std::make_pair(AccDir, TRUE));
		SearchAccountsNames.insert(std::make_pair(AccDir,
			preferences.accounts.GetAccountName(i)));
		AccDirFull.clear();
		AccDirFullSfx.clear();
		AccDir.clear();
	}
	
	// Setup the columns.
	
	wxListItem item;
	item.SetId(0);
	item.SetText(_("Name"));
	item.SetWidth(228);
	
	lstResults->InsertColumn(0, item);
	
	item.SetId(1);
	item.SetText(_("Nickname"));
	item.SetWidth(128);
	
	lstResults->InsertColumn(1, item);
	
	item.SetId(2);
	item.SetText(_("Account"));
	item.SetWidth(128);
	
	lstResults->InsertColumn(2, item);
	
}

void frmSearch::SelectAccounts( wxCommandEvent& event )
{

	// Create the form and display as modal.

	frmSearchAccounts *frameSAccount = new frmSearchAccounts ( this );
	frameSAccount->LoadSearchAccounts(&SearchAccounts, &SearchAccountsNames);
	frameSAccount->ShowModal();
	delete frameSAccount;
	frameSAccount = NULL;

}

void frmSearch::SearchContacts( wxCommandEvent& event )
{

	// Change the button to stop.
	
	if (StopMode == FALSE){
	
		// Clear the list of search results.
		
		lstResults->DeleteAllItems();
		SearchResultAccount.clear();
		SearchResultFilename.clear();
	
		// Button clicked on as 'Search'.
	
		DisableAllSearchSettings(TRUE);
		StopMode = TRUE;
		btnSearch->SetLabel(_("Stop"));	
		
		// Spawn a thread so that searching can proceed
		// without blocking the GUI (and allow the
		// search to stop when needed).
		
		std::thread SearchThread(&frmSearch::SearchContactsThread, this);
		SearchThread.detach();
	
	} else {
	
		// Button clicked on as 'Stop'.
	
		StopMode = FALSE;
		btnSearch->SetLabel(_("Search"));
		DisableAllSearchSettings(FALSE);
	
	}

}

void frmSearch::ResetContacts( wxCommandEvent& event )
{

	// Clear all the search settings.

	for (std::map<int,void*>::iterator iter = SearchFrames.begin(); 
	 iter != SearchFrames.end(); ++iter){
	 
		XABSearchPanel *XABSPPtr;
	 
		XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
		szrSearch->Detach(XABSPPtr);

		// Remove the frame from the memory and the list.
	
		delete XABSPPtr;
		XABSPPtr = NULL;
		
		SearchFrames.erase(iter->first);
		 
	}
	
	// Clear the list of search results.
	
	lstResults->DeleteAllItems();
	
	// Clear the account settings.

	for (std::map<wxString, bool>::iterator iter = SearchAccounts.begin();
		iter != SearchAccounts.end(); iter++){
	
		iter->second = TRUE;
		
	}

	// Add a search settings with the default things.
	
	ScrollGen = 0;

	XABSearchPanel *NewPanel = new XABSearchPanel( tabSearch );
	NewPanel->EnableButtons(TRUE, FALSE);
	NewPanel->SetupPointers(this);
	NewPanel->SetupInteger(ScrollGen);
	szrSearch->Add(NewPanel, 1, wxEXPAND, 0);
	szrSearch->Fit(tabSearch);
	SearchFrames.insert(std::make_pair(ScrollGen, NewPanel));
	ScrollGen++;

}

void frmSearch::AddSearchSetting( wxCommandEvent& event )
{

	// Add a search setting frame to the list.

	XABSearchPanel *NewPanel = new XABSearchPanel( tabSearch );
	NewPanel->EnableButtons(TRUE, TRUE);
	NewPanel->SetupPointers(this);
	NewPanel->SetupInteger(ScrollGen);
	szrSearch->Add(NewPanel, 1, wxEXPAND|wxGROW, 5);
	//szrSearch->Fit(tabSearch);
	szrSearch->FitInside(tabSearch);
	szrSearch->Layout();
	szrSearch->RecalcSizes();
	tabSearch->Layout();
	SearchFrames.insert(std::make_pair(ScrollGen, NewPanel));
	ScrollGen++;
	
	// Check if number of search settings is 15 (or over).
	// If it is, disable the option of adding extra settings
	// for all frames until one is removed.

	XABSearchPanel *XABSPPtr;
	
	if (SearchFrames.size() >= 15){
	
		for (std::map<int,void*>::iterator iter = SearchFrames.begin(); 
		iter != SearchFrames.end(); ++iter){
	 
			XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
	 
			XABSPPtr->EnableButtons(FALSE, TRUE);
	 
		}
	
	} else if (SearchFrames.size() >= 2){

		for (std::map<int,void*>::iterator iter = SearchFrames.begin(); 
		iter != SearchFrames.end(); ++iter){
	 
			XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
	 
			XABSPPtr->EnableButtons(TRUE, TRUE);
	 
		}
	
	}

	//tabSearch->SetScrollbar(wxVERTICAL, 32768, 25, 32768, false);
	tabSearch->Update();
	//tabSearch->ScrollToLine(tabSearch->GetLineCount());
	
}

void frmSearch::RemoveSearchSetting( wxCommandEvent& event )
{

	// Get the integer from the event.
	
	std::map<int,void*>::iterator iter;
	iter = SearchFrames.find(event.GetInt());

	// Remove a search setting frame from the list.

	XABSearchPanel *XABSPPtr;
	XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
	
	szrSearch->Detach(XABSPPtr);

	// Remove the frame from the memory and the list.

	SearchFrames.erase(event.GetInt());
	
	delete XABSPPtr;
	XABSPPtr = NULL;
	
	//szrSearch->Fit(tabSearch);
	
	if (SearchFrames.size() < 15 && SearchFrames.size() > 1){
	
		for (std::map<int,void*>::iterator iter = SearchFrames.begin(); 
		iter != SearchFrames.end(); ++iter){
	 
			XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
	 
			XABSPPtr->EnableButtons(TRUE, TRUE);
	 
		}
	
	} else if (SearchFrames.size() == 1){

		for (std::map<int,void*>::iterator iter = SearchFrames.begin(); 
		iter != SearchFrames.end(); ++iter){
	 
			XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
	 
			XABSPPtr->EnableButtons(TRUE, FALSE);
	 
		}
	
	}

	szrSearch->FitInside(tabSearch);
	szrSearch->Layout();
	szrSearch->RecalcSizes();
	tabSearch->Layout();

}

void frmSearch::ReloadAccountList( wxCommandEvent& event ){

	// Currently unimplemented.

}

void frmSearch::SearchBarUpdate( wxCommandEvent& event ){

	wxString *SBData = (wxString*)event.GetClientData();

	stbBottom->SetStatusText(*SBData, 0);
	
	delete SBData;
	SBData = NULL;

}

void frmSearch::SearchFinished( wxCommandEvent& event ){

	StopMode = FALSE;
	btnSearch->SetLabel(_("Search"));
	DisableAllSearchSettings(FALSE);

}

void frmSearch::DisableAllSearchSettings(bool Enable){

	for (std::map<int, void*>::iterator siter = SearchFrames.begin();
		siter != SearchFrames.end(); siter++){
	
		XABSearchPanel *XABSPPtr = static_cast<XABSearchPanel*>(siter->second);
		
		if (Enable == FALSE){
		
			wxCommandEvent enboxes(XABSP_ENABLECONTROLS);
			wxPostEvent(XABSPPtr, enboxes);
		
		} else if (Enable == TRUE){
		
			wxCommandEvent disboxes(XABSP_DISABLECONTROLS);
			wxPostEvent(XABSPPtr, disboxes);
		
		}
		
	}

}

void frmSearch::CloseWindow( wxCloseEvent& event ){

	// Hide the window so users don't panic
	// whilst clearing up.

	this->Hide();
	
	// Clear up.
	
	for (std::map<int,void*>::iterator iter = SearchFrames.begin(); 
	 iter != SearchFrames.end(); ++iter){
	 
		XABSearchPanel *XABSPPtr;
	 
		XABSPPtr = static_cast<XABSearchPanel*>(iter->second);
		szrSearch->Detach(XABSPPtr);

		// Remove the frame from the memory and the list.
	
		delete XABSPPtr;
		XABSPPtr = NULL;
		 
	}

	SearchFrames.clear();
	
	// Close window.
	
	WindowData *WData = new WindowData;

	WData->DataType = 2;
	WData->WindowPointer = this;
	WData->WindowID = SearchUID;

	wxCommandEvent delevent(WINDOW_CLOSE);
	delevent.SetClientData(WData);
	wxPostEvent(GetParent(), delevent);
		
	wxCommandEvent rs(CE_REMOVESEARCH);
	wxPostEvent(this, rs);
	
	WData = NULL;
	
	this->Destroy();

}

void frmSearch::CloseWindow( wxCommandEvent& event ){

	this->Close();

}

void frmSearch::SetUID(int UID){

	SearchUID = UID;

}

void frmSearch::SetSearchMode(bool SearchModeIn){

	SearchMode = SearchModeIn;

	if (SearchMode == TRUE){
	
		mnuContactEdit->Enable(FALSE);
		mnuContactReveal->Enable(FALSE);
		
		wxFileSystem::AddHandler(new wxMemoryFSHandler);
		wxImage ciicon_png;

		wxMemoryInputStream ciptostream(icons_cipto_png, sizeof(icons_cipto_png));
		ciicon_png.LoadFile(ciptostream, wxBITMAP_TYPE_PNG);
		wxMemoryFSHandler::AddFile(wxT("cipto.png"), ciicon_png, wxBITMAP_TYPE_PNG);

		wxMemoryInputStream cilogstream(icons_cilog_png, sizeof(icons_cilog_png));
		ciicon_png.LoadFile(cilogstream, wxBITMAP_TYPE_PNG);
		wxMemoryFSHandler::AddFile(wxT("cilog.png"), ciicon_png, wxBITMAP_TYPE_PNG);

		wxMemoryInputStream cisndstream(icons_cisnd_png, sizeof(icons_cisnd_png));
		ciicon_png.LoadFile(cisndstream, wxBITMAP_TYPE_PNG);
		wxMemoryFSHandler::AddFile(wxT("cisnd.png"), ciicon_png, wxBITMAP_TYPE_PNG);

		wxMemoryInputStream cikeystream(icons_cikey_png, sizeof(icons_cikey_png));
		ciicon_png.LoadFile(cikeystream, wxBITMAP_TYPE_PNG);
		wxMemoryFSHandler::AddFile(wxT("cikey.png"), ciicon_png, wxBITMAP_TYPE_PNG);

		wxMemoryInputStream civenstream(icons_civen_png, sizeof(icons_civen_png));
		ciicon_png.LoadFile(civenstream, wxBITMAP_TYPE_PNG);
		wxMemoryFSHandler::AddFile(wxT("civen.png"), ciicon_png, wxBITMAP_TYPE_PNG);

		wxMemoryInputStream ciextstream(icons_ciext_png, sizeof(icons_ciext_png));
		ciicon_png.LoadFile(ciextstream, wxBITMAP_TYPE_PNG);
		wxMemoryFSHandler::AddFile(wxT("ciext.png"), ciicon_png, wxBITMAP_TYPE_PNG);
	
	} else {
	
		mnuContactEdit->Enable(TRUE);
		mnuContactReveal->Enable(TRUE);
	
	}
	
}