/*

Miranda IM: the free IM client for Microsoft* Windows*

Copyright 2000-2003 Miranda ICQ/IM project, 
all portions of this codebase are copyrighted to the people 
listed in contributors.txt.

This program 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 2
of the License, or (at your option) any later version.

This program 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 this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
*/

#include "commonheaders.h"
#include <m_utils.h>
#include <m_trigger.h>

#include <shlobj.h>

#include "database.h"
#include "options.h"

extern HINSTANCE g_hInst;

int GetProfileDirectory(char *szPath,int cbPath) {
	return CallService(MS_DB_GETPROFILEPATH, (WPARAM)cbPath, (LPARAM)szPath);
}
void InitAutoBackupsFolder(void)
{
  if ( (hAutoBackupsFolder = FoldersRegisterCustomPath(Translate("Database"), Translate("Backups"), 
	  PROFILE_PATH SUB_DIR ) ) == 0 ) 
  {
	  GetProfileDirectory(options.folder, MAX_PATH);
	  strcat(options.folder, SUB_DIR);	   
  }
}

int Backup(void);
int BackupService(WPARAM wParam, LPARAM lParam);
int SetBackupTimer(void);

int LoadOptions(void) {
//	DBVARIANT dbv;

	options.backup_types = (BackupType)DBGetContactSettingByte(0, "db3x", "BackupType", (BYTE)(BT_EXIT | BT_PERIODIC));
	options.period       = (unsigned int)DBGetContactSettingWord(0, "db3x", "Period", 1);
	options.period_type  = (PeriodType)DBGetContactSettingByte(0, "db3x", "PeriodType", (BYTE)PT_DAYS);

	//if(!DBGetContactSetting(0, "db3x", "Folder", &dbv) && dbv.type == DBVT_ASCIIZ) {
	//	if(strlen(dbv.pszVal) >= 2 && dbv.pszVal[1] == ':') {
	//		strncpy(options.folder, dbv.pszVal, MAX_PATH);
	//	} else {
	//		GetProfileDirectory(options.folder, MAX_PATH);
	//		strcat(options.folder, "\\");
	//		strcat(options.folder, dbv.pszVal);
	//	}
	//	DBFreeVariant(&dbv);
	//} else {
	//	int len;
	//	GetProfileDirectory(options.folder, MAX_PATH);
	//	len = strlen(options.folder);
	//	strncat(options.folder, SUB_DIR, MAX_PATH - len);
	//}

	options.num_backups      = (unsigned int)DBGetContactSettingWord(0, "db3x", "NumBackups", 3);
	options.disable_progress = (BOOL)DBGetContactSettingByte(0, "db3x", "NoProgress", 0);
	options.enable_restarter = (BOOL)DBGetContactSettingByte(0, "db3x", "EnableRestarter", 0);
	options.launch_type      = (BOOL)DBGetContactSettingByte(0, "db3x", "LaunchType", 0);

	SetBackupTimer();
	return 0;
}

int SaveOptions(void) {
//	char prof_dir[MAX_PATH];
//	int prof_len, opt_len;

	DBWriteContactSettingByte(0, "db3x", "BackupType", (BYTE)options.backup_types);
	if (options.period < 1) options.period = 1;
	DBWriteContactSettingWord(0, "db3x", "Period", (WORD)options.period);
	DBWriteContactSettingByte(0, "db3x", "PeriodType", (BYTE)options.period_type);

	//GetProfileDirectory(prof_dir, MAX_PATH);
	//strcat(prof_dir, "\\");
	//prof_len = strlen(prof_dir);
	//opt_len = strlen(options.folder);

	//if(opt_len > prof_len && strncmp(options.folder, prof_dir, prof_len) == 0) {
	//	DBWriteContactSettingString(0, "db3x", "Folder", (options.folder + prof_len));
	//} else 
	//	DBWriteContactSettingString(0, "db3x", "Folder", options.folder);

	if(options.num_backups < 1) options.num_backups = 1;
	DBWriteContactSettingWord(0, "db3x", "NumBackups", (WORD)options.num_backups);
	DBWriteContactSettingByte(0, "db3x", "NoProgress", (BYTE)options.disable_progress);
	//DBWriteContactSettingByte(0, "db3x", "LaunchType", (BYTE)options.launch_type);

	SetBackupTimer();
	return 0;
}

Options new_options;

int SetDlgState(HWND hwndDlg) {
	HWND hwndCtrl;
	char buff[10];

	if(new_options.backup_types == BT_DISABLED) {
		CheckDlgButton(hwndDlg, IDC_RAD_DISABLED, BST_CHECKED);
		hwndCtrl = GetDlgItem(hwndDlg, IDC_RAD_DISABLED);
		EnableWindow(hwndCtrl, FALSE);

		CheckDlgButton(hwndDlg, IDC_RAD_START, BST_UNCHECKED);
		CheckDlgButton(hwndDlg, IDC_RAD_EXIT, BST_UNCHECKED);
		CheckDlgButton(hwndDlg, IDC_RAD_PERIODIC, BST_UNCHECKED);
		hwndCtrl = GetDlgItem(hwndDlg, IDC_ED_PERIOD);
		EnableWindow(hwndCtrl, FALSE);
		hwndCtrl = GetDlgItem(hwndDlg, IDC_PT);
		EnableWindow(hwndCtrl, FALSE);
	} else {
		hwndCtrl = GetDlgItem(hwndDlg, IDC_RAD_DISABLED);
		EnableWindow(hwndCtrl, TRUE);
		CheckDlgButton(hwndDlg, IDC_RAD_DISABLED, BST_UNCHECKED);
		CheckDlgButton(hwndDlg, IDC_RAD_START, new_options.backup_types & BT_START ? BST_CHECKED : BST_UNCHECKED);
		CheckDlgButton(hwndDlg, IDC_RAD_EXIT, new_options.backup_types & BT_EXIT ? BST_CHECKED : BST_UNCHECKED);
		CheckDlgButton(hwndDlg, IDC_RAD_PERIODIC, new_options.backup_types & BT_PERIODIC ? BST_CHECKED : BST_UNCHECKED);
		hwndCtrl = GetDlgItem(hwndDlg, IDC_ED_PERIOD);
		EnableWindow(hwndCtrl, new_options.backup_types & BT_PERIODIC);
		hwndCtrl = GetDlgItem(hwndDlg, IDC_PT);
		EnableWindow(hwndCtrl, new_options.backup_types & BT_PERIODIC);
	}

	SendDlgItemMessage(hwndDlg, SPIN_PERIOD, UDM_SETRANGE32, (WPARAM)1, (LPARAM)30);
	SetDlgItemText(hwndDlg, IDC_ED_PERIOD, itoa(new_options.period, buff, 10));

	SendDlgItemMessage(hwndDlg, SPIN_NUMBACKUPS, UDM_SETRANGE32, (WPARAM)1, (LPARAM)20);
	SetDlgItemText(hwndDlg, IDC_ED_NUMBACKUPS, itoa(new_options.num_backups, buff, 10));

	//SetDlgItemText(hwndDlg, IDC_ED_FOLDER, new_options.folder);

	CheckDlgButton(hwndDlg, IDC_CHK_NOPROG, new_options.disable_progress ? BST_CHECKED : BST_UNCHECKED);

	//CheckDlgButton(hwndDlg, IDC_CHK_LAUNCHTYPE, new_options.launch_type ? BST_CHECKED : BST_UNCHECKED);

	return 0;
}

BOOL CALLBACK DlgProcOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
	char buff[10];
//	char folder_buff[MAX_PATH];
//	BROWSEINFO bi;
//	LPCITEMIDLIST pidl;

	switch ( msg ) {
	case WM_INITDIALOG:
		TranslateDialogDefault( hwndDlg );
		memcpy(&new_options, &options, sizeof(Options));
		SetDlgState(hwndDlg);

		SendMessage(GetDlgItem(hwndDlg, IDC_PT), CB_ADDSTRING, 0, (LPARAM) Translate("Days"));
		SendMessage(GetDlgItem(hwndDlg, IDC_PT), CB_ADDSTRING, 0, (LPARAM) Translate("Hours"));
		switch(new_options.period_type){
			case PT_DAYS: SendDlgItemMessage(hwndDlg, IDC_PT, CB_SETCURSEL, 0, 0); break;
			case PT_HOURS: SendDlgItemMessage(hwndDlg, IDC_PT, CB_SETCURSEL, 1, 0); break;
	}
		return TRUE;
	case WM_COMMAND:
		if ( HIWORD( wParam ) == EN_CHANGE && ( HWND )lParam == GetFocus()) {
			switch( LOWORD( wParam )) {
			case IDC_ED_PERIOD:
			//case IDC_ED_FOLDER:
			case IDC_ED_NUMBACKUPS:
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
			}	
		}
		if ( HIWORD( wParam ) == CBN_SELCHANGE) {
			SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
		}
		if ( HIWORD( wParam ) == BN_CLICKED ) {
			switch( LOWORD( wParam )) {
			case IDC_RAD_DISABLED:
				if(IsDlgButtonChecked(hwndDlg, IDC_RAD_DISABLED)) {
					new_options.backup_types = BT_DISABLED;
				}
				SetDlgState(hwndDlg);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;
			case IDC_RAD_START:
				if(IsDlgButtonChecked(hwndDlg, IDC_RAD_START))
					new_options.backup_types |= BT_START;
				else
					new_options.backup_types &= ~BT_START;
				SetDlgState(hwndDlg);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;
			case IDC_RAD_EXIT:
				if(IsDlgButtonChecked(hwndDlg, IDC_RAD_EXIT))
					new_options.backup_types |= BT_EXIT;
				else
					new_options.backup_types &= ~BT_EXIT;
				SetDlgState(hwndDlg);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;
			case IDC_RAD_PERIODIC:
				if(IsDlgButtonChecked(hwndDlg, IDC_RAD_PERIODIC))
					new_options.backup_types |= BT_PERIODIC;
				else
					new_options.backup_types &= ~BT_PERIODIC;
				SetDlgState(hwndDlg);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;
			
			/*case IDC_BUT_BROWSE:
				bi.hwndOwner = hwndDlg;
				bi.pidlRoot = 0;
				bi.pszDisplayName = folder_buff;
				bi.lpszTitle = Translate("Select Backup Folder");
				bi.ulFlags = 0;
				bi.lpfn = 0;
				bi.lParam = 0;
				bi.iImage = 0;

				if((pidl = SHBrowseForFolder(&bi)) != 0) {
					SHGetPathFromIDList(pidl, folder_buff);
					SetDlgItemText(hwndDlg, IDC_ED_FOLDER, folder_buff);
					SendMessage( GetParent( hwndDlg ), PSM_CHANGED, 0, 0 );

					CoTaskMemFree((void *)pidl);
				}
				break;*/
			case IDC_BUT_NOW:
				Backup();
				break;
			case IDC_CHK_NOPROG:
				new_options.disable_progress = IsDlgButtonChecked(hwndDlg, IDC_CHK_NOPROG);
				SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				break;
			//case IDC_CHK_LAUNCHTYPE:
			//	new_options.launch_type = IsDlgButtonChecked(hwndDlg, IDC_CHK_LAUNCHTYPE);
			//	SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
			//	break;
			}
		}

		break;
	case WM_NOTIFY:
		if (((LPNMHDR)lParam)->code == PSN_APPLY ) {
			GetDlgItemText(hwndDlg, IDC_ED_PERIOD, buff, sizeof(buff));
			new_options.period = atoi(buff);
			GetDlgItemText(hwndDlg, IDC_ED_NUMBACKUPS, buff, sizeof(buff));
			new_options.num_backups = atoi(buff);
							
			switch(SendDlgItemMessage(hwndDlg, IDC_PT, CB_GETCURSEL, 0, 0)) {
				case 0: new_options.period_type = PT_DAYS; break;
				case 1: new_options.period_type = PT_HOURS; break;
			}

			/*GetDlgItemText(hwndDlg, IDC_ED_FOLDER, folder_buff, MAX_PATH);

			{
				BOOL folder_ok = TRUE;
				if(!CreateDirectory(folder_buff, 0)) {
					DWORD err = GetLastError();
					if(err != ERROR_ALREADY_EXISTS) {
						char msg_buff[512];
						FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, 0, err, 0, msg_buff, 512, 0);
						MessageBox(0, msg_buff, Translate("Error Creating Backup Folder"), MB_OK | MB_ICONERROR);
						folder_ok = FALSE;
					}
				}
				
				if(folder_ok) {
					strncpy(new_options.folder, folder_buff, MAX_PATH);*/
					memcpy(&options, &new_options, sizeof(Options));
					SaveOptions();
				/*} else {
					memcpy(&new_options, &options, sizeof(Options));
					SetDlgState(hwndDlg);
				}
			}*/
			return TRUE;

		}
		break;
	}
	
	return FALSE;
}

int OptInit(WPARAM wParam, LPARAM lParam) {
	OPTIONSDIALOGPAGE odp = {0};
#define OPTIONPAGE_OLD_SIZE2 60
	//odp.cbSize						= sizeof(odp);
	odp.cbSize						= OPTIONPAGE_OLD_SIZE2;
	odp.position					= -790000000;
	odp.hInstance					= g_hInst;
	odp.pszTemplate					= MAKEINTRESOURCE(IDD_OPTIONS);
	odp.pszTitle					= Translate("Database");
	odp.pszGroup					= Translate("Services");
	odp.flags						= ODPF_BOLDGROUPS;
	odp.nIDBottomSimpleControl		= 0;
	odp.pfnDlgProc					= DlgProcOpts;
	CallService( MS_OPT_ADDPAGE, wParam,( LPARAM )&odp );

	return 0;
}

int ModulesLoaded(WPARAM wParam, LPARAM lParam) {

	HookEvent(ME_OPT_INITIALISE, OptInit);

	// register folder
	InitAutoBackupsFolder();
	// register trigger action for triggerplugin
	if(ServiceExists(MS_TRIGGER_REGISTERACTION)) {
		// create update action for triggerplugin
		ACTIONREGISTER ar = {0};
		ar.cbSize = sizeof(ACTIONREGISTER);
		ar.pszName = Translate("Backup Database");
		ar.pszService = MS_DB_BACKUPTRGR;

		CallService(MS_TRIGGER_REGISTERACTION, 0, (LPARAM)&ar);
	}


	return 0;
}

// can't do this on unload, since other plugins will be have already been unloaded, but their hooks 
// for setting changed event not cleared. the backup on exit function will write to the db, calling those hooks.
int PreShutdown(WPARAM wParam, LPARAM lParam) {
	extern BOOL command_restart;
	if(options.backup_types & BT_EXIT && !command_restart)
		Backup();
    	return 0;
}

int BackupServiceTrgr(WPARAM wParam, LPARAM lParam) {
	if(wParam & ACT_PERFORM) {
		return CallService(MS_DB_BACKUP, 0, 0);
	}
	return 0;
}

int InitOptions(void) {

	OleInitialize(0);

	CreateServiceFunction(MS_DB_BACKUP, BackupService);
	CreateServiceFunction(MS_DB_BACKUPTRGR, BackupServiceTrgr);

	// we have to wait for the options module to load!
	HookEvent(ME_SYSTEM_MODULESLOADED, ModulesLoaded);	

	HookEvent(ME_SYSTEM_PRESHUTDOWN, PreShutdown);	

	// but, our stuff has loaded so...
	LoadOptions();

	if(options.backup_types & BT_START)
		Backup();

	return 0;
}

int UninitOptions(void) {

	OleUninitialize();

	return 0;
}