/*

'AutoShutdown'-Plugin for Miranda IM

Copyright 2004-2007 H. Herkenrath

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

#include "common.h"

/* Option Page */
static HANDLE hHookOptInit;
extern HINSTANCE hInst;
/* Trigger */
static HANDLE hHookModulesLoaded;

/************************* Option Page ********************************/

#define M_ENABLE_SUBCTLS  (WM_APP+111)

static int CALLBACK ShutdownOptDlgProc(HWND hwndDlg,UINT msg,WPARAM wParam,LPARAM lParam)
{
	switch(msg) {
		case WM_INITDIALOG:
			TranslateDialogDefault(hwndDlg);
			{	WORD setting=DBGetContactSettingWord(NULL,"AutoShutdown","ConfirmDlgCountdown",SETTING_CONFIRMDLGCOUNTDOWN_DEFAULT);
				if(setting<3) setting=SETTING_CONFIRMDLGCOUNTDOWN_DEFAULT;
				SendDlgItemMessage(hwndDlg,IDC_SPIN_CONFIRMDLGCOUNTDOWN,UDM_SETRANGE,0,MAKELPARAM(999,3));
				SendDlgItemMessage(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN,EM_SETLIMITTEXT,3,0);
				SendDlgItemMessage(hwndDlg,IDC_SPIN_CONFIRMDLGCOUNTDOWN,UDM_SETPOS,0,MAKELPARAM(setting,0));
				SetDlgItemInt(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN,setting,FALSE);
			}
			CheckDlgButton(hwndDlg,IDC_CHECK_SMARTOFFLINECHECK,DBGetContactSettingByte(NULL,"AutoShutdown","SmartOfflineCheck",SETTING_SMARTOFFLINECHECK_DEFAULT)!=0);
			CheckDlgButton(hwndDlg,IDC_CHECK_REMEMBERONRESTART,DBGetContactSettingByte(NULL,"AutoShutdown","RememberOnRestart",SETTING_REMEMBERONRESTART_DEFAULT)!=0);
			CheckDlgButton(hwndDlg,IDC_CHECK_SHOWCONFIRMDLG,DBGetContactSettingByte(NULL,"AutoShutdown","ShowConfirmDlg",SETTING_SHOWCONFIRMDLG_DEFAULT)!=0);
			{	BOOL enabled=ServiceIsTypeEnabled(SDSDT_SHUTDOWN,0);
				if(enabled) {
					if(ServiceExists(MS_SYSINFO_HDDTEMP)) {
						EnableWindow(GetDlgItem(hwndDlg,IDC_CHECK_HDDOVERHEAT),TRUE);
						CheckDlgButton(hwndDlg,IDC_CHECK_HDDOVERHEAT,DBGetContactSettingByte(NULL,"AutoShutdown","HddOverheatShutdown",SETTING_HDDOVERHEATSHUTDOWN_DEFAULT)!=0);
					}
					if(ServiceExists(MS_WEATHER_UPDATE)) {
						EnableWindow(GetDlgItem(hwndDlg,IDC_CHECK_WEATHER),TRUE);
						CheckDlgButton(hwndDlg,IDC_CHECK_WEATHER,DBGetContactSettingByte(NULL,"AutoShutdown","WeatherShutdown",SETTING_WEATHERSHUTDOWN_DEFAULT)!=0);
					}
				}
			}
			SendMessage(hwndDlg,M_ENABLE_SUBCTLS,0,0);
			return TRUE; /* default focus */
		case M_ENABLE_SUBCTLS:
		{	BOOL checked=IsDlgButtonChecked(hwndDlg,IDC_CHECK_SHOWCONFIRMDLG)!=0;
			if(checked!=IsWindowEnabled(GetDlgItem(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN))) {
				EnableWindow(GetDlgItem(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN),checked);
				EnableWindow(GetDlgItem(hwndDlg,IDC_SPIN_CONFIRMDLGCOUNTDOWN),checked);
				EnableWindow(GetDlgItem(hwndDlg,IDC_TEXT_COUNTDOWNSTARTS),checked);
				EnableWindow(GetDlgItem(hwndDlg,IDC_TEXT_SECONDS),checked);
			}
			return TRUE;
		}
 		case WM_COMMAND:
			switch(LOWORD(wParam)) {
				case IDC_EDIT_CONFIRMDLGCOUNTDOWN:
					if(HIWORD(wParam)==EN_KILLFOCUS)
						if((int)GetDlgItemInt(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN,NULL,TRUE)<3) {
							SendDlgItemMessage(hwndDlg,IDC_SPIN_CONFIRMDLGCOUNTDOWN,UDM_SETPOS,0,MAKELPARAM(3,0));
							SetDlgItemInt(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN,3,FALSE);
						}
					if(HIWORD(wParam)!=EN_CHANGE || (HWND)lParam!=GetFocus())
						return TRUE; /* no apply */
					break;
			}
			PostMessage(hwndDlg,M_ENABLE_SUBCTLS,0,0);
			PostMessage(GetParent(hwndDlg),PSM_CHANGED,0,0); /* enable apply */
			return FALSE;
		case WM_NOTIFY:
			switch(((NMHDR*)lParam)->code) {
				case PSN_APPLY:
					DBWriteContactSettingByte(NULL,"AutoShutdown","ShowConfirmDlg",(BYTE)(IsDlgButtonChecked(hwndDlg,IDC_CHECK_SHOWCONFIRMDLG)!=0));
					DBWriteContactSettingWord(NULL,"AutoShutdown","ConfirmDlgCountdown",(WORD)GetDlgItemInt(hwndDlg,IDC_EDIT_CONFIRMDLGCOUNTDOWN,NULL,FALSE));
					DBWriteContactSettingByte(NULL,"AutoShutdown","RememberOnRestart",(BYTE)(IsDlgButtonChecked(hwndDlg,IDC_CHECK_REMEMBERONRESTART)!=0));
					DBWriteContactSettingByte(NULL,"AutoShutdown","SmartOfflineCheck",(BYTE)(IsDlgButtonChecked(hwndDlg,IDC_CHECK_SMARTOFFLINECHECK)!=0));
					if(IsWindowEnabled(GetDlgItem(hwndDlg,IDC_CHECK_WEATHER)))
						DBWriteContactSettingByte(NULL,"AutoShutdown","WeatherShutdown",(BYTE)(IsDlgButtonChecked(hwndDlg,IDC_CHECK_WEATHER)!=0));
					if(IsWindowEnabled(GetDlgItem(hwndDlg,IDC_CHECK_HDDOVERHEAT)))
						DBWriteContactSettingByte(NULL,"AutoShutdown","HddOverheatShutdown",(BYTE)(IsDlgButtonChecked(hwndDlg,IDC_CHECK_HDDOVERHEAT)!=0));
					return TRUE;
			}
			break;
	}
	return FALSE;
}

static int ShutdownOptInit(WPARAM wParam,LPARAM lParam)
{
	OPTIONSDIALOGPAGE odp;
	UNREFERENCED_PARAMETER(lParam);
	ZeroMemory(&odp,sizeof(odp));
	odp.cbSize=sizeof(odp);
	odp.hInstance=hInst;
	odp.pszTemplate=MAKEINTRESOURCEA(IDD_OPT_SHUTDOWN);
	odp.position=900000002;
	odp.ptszGroup=_T("Events"); /* autotranslated */
	odp.ptszTitle=_T("Automatic Shutdown"); /* autotranslated */
	odp.ptszTab=_T("Automatic Shutdown");  /* autotranslated, can be made a tab */
	odp.flags=ODPF_BOLDGROUPS|ODPF_EXPERTONLY|ODPF_TCHAR;
	odp.pfnDlgProc=ShutdownOptDlgProc;
	CallService(MS_OPT_ADDPAGE,wParam,(LPARAM)&odp);
	return 0;
}

/************************* Trigger ************************************/

static int __stdcall ActionProc(DWORD idAction,REPORTINFO *ri,int shutdownType)
{
	UNREFERENCED_PARAMETER(ri);
	if(idAction&ACT_PERFORM) ServiceShutdown(shutdownType,TRUE);
	return 0;
}
#define ActionProcJmp(shutdownType) { return ActionProc(id,ri,shutdownType); }

static int CloseActionProc(DWORD id,REPORTINFO *ri)      ActionProcJmp(SDSDT_CLOSEMIRANDA)
static int SetOfflineActionProc(DWORD id,REPORTINFO *ri) ActionProcJmp(SDSDT_SETMIRANDAOFFLINE)
static int LogoffActionProc(DWORD id,REPORTINFO *ri)     ActionProcJmp(SDSDT_LOGOFF)
static int RebootActionProc(DWORD id,REPORTINFO *ri)     ActionProcJmp(SDSDT_REBOOT)
static int ShutdownActionProc(DWORD id,REPORTINFO *ri)   ActionProcJmp(SDSDT_SHUTDOWN)
static int StandbyActionProc(DWORD id,REPORTINFO *ri)    ActionProcJmp(SDSDT_STANDBY)
static int HibernateActionProc(DWORD id,REPORTINFO *ri)  ActionProcJmp(SDSDT_HIBERNATE)
static int LockActionProc(DWORD id,REPORTINFO *ri)       ActionProcJmp(SDSDT_LOCKWORKSTATION)
static int HangupActionProc(DWORD id,REPORTINFO *ri)     ActionProcJmp(SDSDT_CLOSERASCONNECTIONS)

static int TriggerRegisterActions(WPARAM wParam,LPARAM lParam)
{
	UNREFERENCED_PARAMETER(wParam);
	UNREFERENCED_PARAMETER(lParam);
	/* new trigger API (0.2.0.69+) */
	if(ServiceExists(MS_TRIGGER_REGISTERCONDITION)) {
		ACTIONREGISTER ar;
		BYTE shutdownType;
		char *pszDesc,szName[138];
		ACTIONFUNCTION actionFunctions[SDSDT_MAX]={CloseActionProc,SetOfflineActionProc,LogoffActionProc,
												   RebootActionProc,ShutdownActionProc,StandbyActionProc,
												   HibernateActionProc,LockActionProc,HangupActionProc};
		ZeroMemory(&ar,sizeof(ar));
		ar.cbSize=sizeof(ar);
		ar.flags=ARF_TCHAR|ARF_FUNCTION;
		ar.pszName=szName;
		for(shutdownType=1;shutdownType<=SDSDT_MAX;++shutdownType)
			if(ServiceIsTypeEnabled(shutdownType,0)) {
				pszDesc=(char*)ServiceGetTypeDescription(shutdownType,GSTDF_UNTRANSLATED);
				if(pszDesc!=NULL) {
					mir_snprintf(szName,sizeof(szName),"Shutdown: %s",pszDesc); /* autotranslated */
					ar.actionFunction=actionFunctions[shutdownType-1];
					CallService(MS_TRIGGER_REGISTERACTION,0,(LPARAM)&ar);
				}
			}
	}
	return 0;
}

/************************* Misc ***************************************/

void InitOptions(void)
{
	/* Option Page */
	hHookOptInit=HookEvent(ME_OPT_INITIALISE,ShutdownOptInit);
	/* Trigger */
	hHookModulesLoaded=HookEvent(ME_SYSTEM_MODULESLOADED,TriggerRegisterActions);
}

void UninitOptions(void)
{
	/* Option Page */
	UnhookEvent(hHookOptInit);
	/* Trigger */
	UnhookEvent(hHookModulesLoaded);
}