#include "yamn.h"

static INT_PTR Service_GetCaps(WPARAM wParam, LPARAM lParam)
{
	if (wParam == PFLAGNUM_4)
		return PF4_NOCUSTOMAUTH;
	if (wParam == PFLAG_UNIQUEIDTEXT)
        return (INT_PTR) Translate("Nick");
	if (wParam == PFLAG_MAXLENOFMESSAGE)
        return 400;
	if (wParam == PFLAG_UNIQUEIDSETTING)
        return (INT_PTR) "Id";
	if (wParam == PFLAGNUM_2)
		return PF2_ONLINE | PF2_SHORTAWAY | PF2_LONGAWAY | PF2_LIGHTDND;
	if (wParam == PFLAGNUM_5) {
		if (db_get_b(NULL, YAMN_DBMODULE, YAMN_SHOWASPROTO, 1))
			return PF2_SHORTAWAY | PF2_LONGAWAY | PF2_LIGHTDND;
		return PF2_ONLINE | PF2_SHORTAWAY | PF2_LONGAWAY | PF2_LIGHTDND;
	}
	return 0;
}

static INT_PTR Service_GetStatus(WPARAM wParam, LPARAM lParam)
{
	return YAMN_STATUS;	
}

static INT_PTR Service_SetStatus(WPARAM wParam, LPARAM lParam)
{	
	int newstatus = (wParam != ID_STATUS_OFFLINE)?ID_STATUS_ONLINE:ID_STATUS_OFFLINE;
	if (newstatus != YAMN_STATUS) {
		int oldstatus = YAMN_STATUS;
		YAMN_STATUS = newstatus;
		ProtoBroadcastAck(YAMN_DBMODULE, NULL, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)oldstatus, newstatus);
	}
	return 0;

}

static INT_PTR Service_GetName(WPARAM wParam, LPARAM lParam)
{
	lstrcpynA((char *) lParam, YAMN_DBMODULE, wParam);
	return 0;
}

static INT_PTR Service_LoadIcon(WPARAM wParam, LPARAM lParam)
{
	if ( LOWORD( wParam ) == PLI_PROTOCOL )
		return (INT_PTR)CopyIcon(g_LoadIconEx(0)); // noone cares about other than PLI_PROTOCOL

	return (INT_PTR)(HICON)NULL;
}
 
INT_PTR ClistContactDoubleclicked(WPARAM wParam, LPARAM lParam)
{
	ContactDoubleclicked(((CLISTEVENT*)lParam)->lParam, lParam);
	return 0;
}

static int Service_ContactDoubleclicked(WPARAM wParam, LPARAM lParam)
{
	ContactDoubleclicked(wParam, lParam);
	return 0;
}

static INT_PTR ContactApplication(WPARAM wParam, LPARAM lParam)
{
	char *szProto = GetContactProto(wParam);
	if ( lstrcmpA(szProto, YAMN_DBMODULE))
		return 0;

	DBVARIANT dbv;
	if ( db_get(wParam, YAMN_DBMODULE, "Id", &dbv))
		return 0;

	HACCOUNT ActualAccount = (HACCOUNT) CallService(MS_YAMN_FINDACCOUNTBYNAME, (WPARAM)POP3Plugin, (LPARAM)dbv.pszVal);
	if (ActualAccount != NULL) {
		STARTUPINFOW si = { 0 };
		si.cb = sizeof(si);
				
		#ifdef DEBUG_SYNCHRO
			DebugLog(SynchroFile, "ContactApplication:ActualAccountSO-read wait\n");
		#endif
		if (WAIT_OBJECT_0 == WaitToReadFcn(ActualAccount->AccountAccessSO)) {
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "ContactApplication:ualAccountSO-read enter\n");
			#endif
			if (ActualAccount->NewMailN.App != NULL) {
				WCHAR *Command;
				if (ActualAccount->NewMailN.AppParam != NULL)
					Command = new WCHAR[wcslen(ActualAccount->NewMailN.App)+wcslen(ActualAccount->NewMailN.AppParam)+6];
				else
					Command = new WCHAR[wcslen(ActualAccount->NewMailN.App)+6];
					
				if (Command != NULL) {
					lstrcpyW(Command, L"\"");
					lstrcatW(Command, ActualAccount->NewMailN.App);
					lstrcatW(Command, L"\" ");
					if (ActualAccount->NewMailN.AppParam != NULL)
						lstrcatW(Command, ActualAccount->NewMailN.AppParam);

					PROCESS_INFORMATION pi;
					CreateProcessW(NULL, Command, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi);
					delete[] Command;
				}
			}

			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "ContactApplication:ActualAccountSO-read done\n");
			#endif
			ReadDoneFcn(ActualAccount->AccountAccessSO);
		}
		#ifdef DEBUG_SYNCHRO
		else
			DebugLog(SynchroFile, "ContactApplication:ActualAccountSO-read enter failed\n");
		#endif
	}
	db_free(&dbv);
	return 0;
}

DWORD WINAPI SWMRGWaitToRead(PSWMRG pSWMRG, DWORD dwTimeout);
static INT_PTR AccountMailCheck(WPARAM wParam, LPARAM lParam)
{
	//This service will check/sincronize the account pointed by wParam
	HACCOUNT ActualAccount = (HACCOUNT)wParam;
	HANDLE ThreadRunningEV;
	DWORD tid;
	// copy/paste make mistakes
	if (ActualAccount != NULL) {
		//we use event to signal, that running thread has all needed stack parameters copied
		if (NULL == (ThreadRunningEV = CreateEvent(NULL, FALSE, FALSE, NULL)))
			return 0;
		//if we want to close miranda, we get event and do not run pop3 checking anymore
		if (WAIT_OBJECT_0 == WaitForSingleObject(ExitEV, 0))
			return 0;

		EnterCriticalSection(&PluginRegCS);
		#ifdef DEBUG_SYNCHRO
			DebugLog(SynchroFile, "AccountCheck:ActualAccountSO-read wait\n");
		#endif
		if (WAIT_OBJECT_0 != SWMRGWaitToRead(ActualAccount->AccountAccessSO, 0)) {
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "ForceCheck:ActualAccountSO-read wait failed\n");
			#endif
		}
		else {
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "ForceCheck:ActualAccountSO-read enter\n");
			#endif
			if ((ActualAccount->Flags & YAMN_ACC_ENA) && ActualAccount->Plugin->Fcn->SynchroFcnPtr) {
				struct CheckParam ParamToPlugin = {YAMN_CHECKVERSION, ThreadRunningEV, ActualAccount, lParam?YAMN_FORCECHECK:YAMN_NORMALCHECK, (void *)0, NULL};
				HANDLE NewThread;

				ActualAccount->TimeLeft = ActualAccount->Interval;
				if (NewThread = CreateThread(NULL, 0, (YAMN_STANDARDFCN)ActualAccount->Plugin->Fcn->SynchroFcnPtr, &ParamToPlugin, 0, &tid)) {
					WaitForSingleObject(ThreadRunningEV, INFINITE);
					CloseHandle(NewThread);
				}
			}
			ReadDoneFcn(ActualAccount->AccountAccessSO);
		}
		LeaveCriticalSection(&PluginRegCS);
		CloseHandle(ThreadRunningEV);
	}
	return 0;
}

static INT_PTR ContactMailCheck(WPARAM hContact, LPARAM lParam)
{
	char *szProto = GetContactProto(hContact);
	if ( lstrcmpA(szProto, YAMN_DBMODULE))
		return 0;

	DBVARIANT dbv;
	if ( db_get(hContact, YAMN_DBMODULE, "Id", &dbv))
		return 0;

	HACCOUNT ActualAccount = (HACCOUNT) CallService(MS_YAMN_FINDACCOUNTBYNAME, (WPARAM)POP3Plugin, (LPARAM)dbv.pszVal);
	if (ActualAccount != NULL) {
		//we use event to signal, that running thread has all needed stack parameters copied
		HANDLE ThreadRunningEV;
		if (NULL == (ThreadRunningEV = CreateEvent(NULL, FALSE, FALSE, NULL)))
			return 0;
		//if we want to close miranda, we get event and do not run pop3 checking anymore
		if (WAIT_OBJECT_0 == WaitForSingleObject(ExitEV, 0))
			return 0;
		EnterCriticalSection(&PluginRegCS);
		#ifdef DEBUG_SYNCHRO
			DebugLog(SynchroFile, "ForceCheck:ActualAccountSO-read wait\n");
		#endif
		if (WAIT_OBJECT_0 != WaitToReadFcn(ActualAccount->AccountAccessSO))
		{
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "ForceCheck:ActualAccountSO-read wait failed\n");
			#endif
		}
		else
		{
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "ForceCheck:ActualAccountSO-read enter\n");
			#endif
			if ((ActualAccount->Flags & YAMN_ACC_ENA) && (ActualAccount->StatusFlags & YAMN_ACC_FORCE))			//account cannot be forced to check
			{
				if (ActualAccount->Plugin->Fcn->ForceCheckFcnPtr == NULL)
					ReadDoneFcn(ActualAccount->AccountAccessSO);

				DWORD tid;
				struct CheckParam ParamToPlugin = {YAMN_CHECKVERSION, ThreadRunningEV, ActualAccount, YAMN_FORCECHECK, (void *)0, NULL};
				if (NULL == CreateThread(NULL, 0, (YAMN_STANDARDFCN)ActualAccount->Plugin->Fcn->ForceCheckFcnPtr, &ParamToPlugin, 0, &tid))
					ReadDoneFcn(ActualAccount->AccountAccessSO);
				else
					WaitForSingleObject(ThreadRunningEV, INFINITE);
			}
			ReadDoneFcn(ActualAccount->AccountAccessSO);
		}
		LeaveCriticalSection(&PluginRegCS);
		CloseHandle(ThreadRunningEV);
	}
	db_free(&dbv);
	return 0;
}

void MainMenuAccountClicked(WPARAM wParam, LPARAM lParam) 
{
}

/*static*/ void ContactDoubleclicked(WPARAM wParam, LPARAM lParam)
{
	char *szProto = GetContactProto(wParam);
	if ( lstrcmpA(szProto, YAMN_DBMODULE))
		return;

	DBVARIANT dbv;
	if ( db_get(wParam, YAMN_DBMODULE, "Id", &dbv))
		return;

	HACCOUNT ActualAccount = (HACCOUNT)CallService(MS_YAMN_FINDACCOUNTBYNAME, (WPARAM)POP3Plugin, (LPARAM)dbv.pszVal);
	if (ActualAccount != NULL) {
		#ifdef DEBUG_SYNCHRO
			DebugLog(SynchroFile, "Service_ContactDoubleclicked:ActualAccountSO-read wait\n");
		#endif
		if (WAIT_OBJECT_0 == WaitToReadFcn(ActualAccount->AccountAccessSO)) {
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "Service_ContactDoubleclicked:ActualAccountSO-read enter\n");
			#endif
			YAMN_MAILBROWSERPARAM Param = { 0, ActualAccount, ActualAccount->NewMailN.Flags, ActualAccount->NoNewMailN.Flags, 0 };

			Param.nnflags = Param.nnflags | YAMN_ACC_MSG;			//show mails in account even no new mail in account
			Param.nnflags = Param.nnflags & ~YAMN_ACC_POP;

			Param.nflags = Param.nflags | YAMN_ACC_MSG;			//show mails in account even no new mail in account
			Param.nflags = Param.nflags & ~YAMN_ACC_POP;

			RunMailBrowserSvc((WPARAM)&Param, YAMN_MAILBROWSERVERSION);
					
			#ifdef DEBUG_SYNCHRO
				DebugLog(SynchroFile, "Service_ContactDoubleclicked:ActualAccountSO-read done\n");
			#endif
			ReadDoneFcn(ActualAccount->AccountAccessSO);
		}
		#ifdef DEBUG_SYNCHRO
		else
			DebugLog(SynchroFile, "Service_ContactDoubleclicked:ActualAccountSO-read enter failed\n");
		#endif
				
	}
	db_free(&dbv);
}

/////////////////////////////////////////////////////////////////////////////////////////

HBITMAP LoadBmpFromIcon(HICON hIcon)
{
	int IconSizeX = 16;
	int IconSizeY = 16;

	HBRUSH hBkgBrush = CreateSolidBrush(GetSysColor(COLOR_3DFACE));

	BITMAPINFOHEADER bih = {0};
	bih.biSize = sizeof(bih);
	bih.biBitCount = 24;
	bih.biPlanes = 1;
	bih.biCompression = BI_RGB;
	bih.biHeight = IconSizeY;
	bih.biWidth = IconSizeX; 
	
	int widthBytes = ((bih.biWidth*bih.biBitCount + 31) >> 5) * 4;

	RECT rc;
	rc.top = rc.left = 0;
	rc.right = bih.biWidth;
	rc.bottom = bih.biHeight;

	HDC hdc = GetDC(NULL);
	HBITMAP hBmp = CreateCompatibleBitmap(hdc, bih.biWidth, bih.biHeight);
	HDC hdcMem = CreateCompatibleDC(hdc);
	HBITMAP hoBmp = (HBITMAP)SelectObject(hdcMem, hBmp);
	FillRect(hdcMem, &rc, hBkgBrush);
	DrawIconEx(hdcMem, 0, 0, hIcon, bih.biWidth, bih.biHeight, 0, NULL, DI_NORMAL);
	SelectObject(hdcMem, hoBmp);
	return hBmp;
}

int AddTopToolbarIcon(WPARAM,LPARAM)
{
	if ( db_get_b(NULL, YAMN_DBMODULE, YAMN_TTBFCHECK, 1)) {
		if ( ServiceExists(MS_TTB_REMOVEBUTTON) && hTTButton == NULL) {
			TTBButton btn = {0};
			btn.cbSize = sizeof(btn);
			btn.pszService = MS_YAMN_FORCECHECK;
			btn.dwFlags = TTBBF_VISIBLE | TTBBF_SHOWTOOLTIP;
			btn.hIconHandleUp = btn.hIconHandleDn = g_GetIconHandle(0);
			btn.name = btn.pszTooltipUp = LPGEN("Check mail");
			hTTButton = TopToolbar_AddButton(&btn);
		}
	}
	else {
		if (hTTButton != NULL) {
			CallService(MS_TTB_REMOVEBUTTON, (WPARAM)hTTButton, 0);
			hTTButton = NULL;
		}
	}

	return 0;
}

/////////////////////////////////////////////////////////////////////////////////////////

int Shutdown(WPARAM, LPARAM)
{
	CallService(MS_TTB_REMOVEBUTTON, (WPARAM)hTTButton, 0);

	db_set_dw(NULL, YAMN_DBMODULE, YAMN_DBMSGPOSX, HeadPosX);
	db_set_dw(NULL, YAMN_DBMODULE, YAMN_DBMSGPOSY, HeadPosY);
	db_set_dw(NULL, YAMN_DBMODULE, YAMN_DBMSGSIZEX, HeadSizeX);
	db_set_dw(NULL, YAMN_DBMODULE, YAMN_DBMSGSIZEY, HeadSizeY);
	db_set_w(NULL, YAMN_DBMODULE, YAMN_DBMSGPOSSPLIT, HeadSplitPos);
	YAMNVar.Shutdown = TRUE;
	KillTimer(NULL, SecTimer);

	UnregisterProtoPlugins();
	UnregisterFilterPlugins();
	return 0;
}

int SystemModulesLoaded(WPARAM, LPARAM); //in main.cpp
typedef struct { HANDLE hookHandle;	const char *hookName; MIRANDAHOOK mirandaFunction;} HookDataType;
static HookDataType hookData[] = {
	{0, ME_SYSTEM_MODULESLOADED, SystemModulesLoaded}, //pop3 plugin must be included after all miranda modules are loaded
	{0, ME_TTB_MODULELOADED,     AddTopToolbarIcon},
	{0, ME_OPT_INITIALISE,       YAMNOptInitSvc}, 
	{0, ME_SYSTEM_PRESHUTDOWN,   Shutdown}, 
	{0, ME_CLIST_DOUBLECLICKED,  Service_ContactDoubleclicked}, 
	{0, 0, 0}//end marker
};

void HookEvents(void)
{
	//We set function which registers needed POP3 accounts. This is a part of internal POP3 plugin.
	//Your plugin should do the same task in your Load fcn. Why we call it in MODULESLOADED? Because netlib
	//user can be registered after all modules are loaded (see m_netlib.h in Miranda)
	for (int i = 0;hookData[i].hookName;i++) {
		hookData[i].hookHandle = HookEvent(hookData[i].hookName, hookData[i].mirandaFunction);
	}
}
void UnhookEvents(void)
{
	for (int i = 0;i<(sizeof(hookData)/sizeof(hookData[0]));i++)
		if (hookData[i].hookHandle)
			UnhookEvent(hookData[i].hookHandle);
}

typedef struct { HANDLE serviceHandle;	const char *serviceName; MIRANDASERVICE serviceFunction;} ServiceDataType;
static ServiceDataType serviceData[] = {
	{0, YAMN_DBMODULE PS_GETCAPS, Service_GetCaps}, 
	{0, YAMN_DBMODULE PS_GETSTATUS, Service_GetStatus}, 
	{0, YAMN_DBMODULE PS_SETSTATUS, Service_SetStatus}, 
	{0, YAMN_DBMODULE PS_GETNAME, Service_GetName}, 
	{0, YAMN_DBMODULE PS_LOADICON, Service_LoadIcon}, 

	//Function with which protocol plugin can register
	{0, MS_YAMN_GETFCNPTR, GetFcnPtrSvc}, 
	
	//Function returns pointer to YAMN variables
	{0, MS_YAMN_GETVARIABLES, GetVariablesSvc}, 
	
	//Function with which protocol plugin can register
	{0, MS_YAMN_REGISTERPROTOPLUGIN, RegisterProtocolPluginSvc}, 
	
	//Function with which protocol plugin can unregister
	{0, MS_YAMN_UNREGISTERPROTOPLUGIN, UnregisterProtocolPluginSvc}, 
	
	//Function creates an account for plugin
	{0, MS_YAMN_CREATEPLUGINACCOUNT, CreatePluginAccountSvc}, 
	
	//Function deletes plugin account 
	{0, MS_YAMN_DELETEPLUGINACCOUNT, DeletePluginAccountSvc}, 
	
	//Finds account for plugin by name
	{0, MS_YAMN_FINDACCOUNTBYNAME, FindAccountByNameSvc}, 
	
	//Creates next account for plugin
	{0, MS_YAMN_GETNEXTFREEACCOUNT, GetNextFreeAccountSvc}, 
	
	//Function removes account from YAMN queue. Does not delete it from memory
	{0, MS_YAMN_DELETEACCOUNT, DeleteAccountSvc}, 
	
	//Function finds accounts for specified plugin
	{0, MS_YAMN_READACCOUNTS, AddAccountsFromFileSvc}, 
	
	//Function that stores all plugin mails to one file 
	{0, MS_YAMN_WRITEACCOUNTS, WriteAccountsToFileSvc}, 
	
	//Function that returns user's filename
	{0, MS_YAMN_GETFILENAME, GetFileNameSvc}, 
	
	//Releases unicode string from memory
	{0, MS_YAMN_DELETEFILENAME, DeleteFileNameSvc}, 
	
	//Checks mail
	{0, MS_YAMN_FORCECHECK, ForceCheckSvc}, 
	
	//Runs YAMN's mail browser
	{0, MS_YAMN_MAILBROWSER, RunMailBrowserSvc}, 
	
	//Runs YAMN's bad conenction window
	{0, MS_YAMN_BADCONNECTION, RunBadConnectionSvc}, 
	
	//Function creates new mail for plugin
	{0, MS_YAMN_CREATEACCOUNTMAIL, CreateAccountMailSvc}, 
	
	//Function deletes plugin account 
	{0, MS_YAMN_DELETEACCOUNTMAIL, DeleteAccountMailSvc}, 
	
	//Function with which filter plugin can register
	{0, MS_YAMN_REGISTERFILTERPLUGIN, RegisterFilterPluginSvc}, 
	
	//Function with which filter plugin can unregister
	{0, MS_YAMN_UNREGISTERFILTERPLUGIN, UnregisterFilterPluginSvc}, 
	
	//Function filters mail
	{0, MS_YAMN_FILTERMAIL, FilterMailSvc}, 

	//Function contact list double click
	{0, MS_YAMN_CLISTDBLCLICK, ClistContactDoubleclicked}, 

	//Function to check individual account
	{0, MS_YAMN_ACCOUNTCHECK, AccountMailCheck}, 

	//Function contact list context menu click
	{0, MS_YAMN_CLISTCONTEXT, ContactMailCheck}, 

	//Function contact list context menu click
	{0, MS_YAMN_CLISTCONTEXTAPP, ContactApplication}, 

	{0, 0, 0}//end marker
};

void CreateServiceFunctions(void)
{
	for (int i = 0;serviceData[i].serviceName;i++) {
		serviceData[i].serviceHandle = CreateServiceFunction(serviceData[i].serviceName, serviceData[i].serviceFunction);
	}
};

void DestroyServiceFunctions(void)
{
	for (int i = 0;serviceData[i].serviceName;i++) {
		if (serviceData[i].serviceHandle) DestroyServiceFunction(serviceData[i].serviceHandle);
	}
};

//Function to put all enabled contact to the Online status
void RefreshContact(void)
{
	HACCOUNT Finder;
	for (Finder = POP3Plugin->FirstAccount;Finder != NULL;Finder = Finder->Next) {
		if (Finder->hContact != NULL) {
			if ((Finder->Flags & YAMN_ACC_ENA) && (Finder->NewMailN.Flags & YAMN_ACC_CONT))
				db_unset(Finder->hContact, "CList", "Hidden");
			else
				db_set_b(Finder->hContact, "CList", "Hidden", 1);
		}
		else if ((Finder->Flags & YAMN_ACC_ENA) && (Finder->NewMailN.Flags & YAMN_ACC_CONT)) {
			Finder->hContact = (MCONTACT)CallService(MS_DB_CONTACT_ADD, 0, 0);
			CallService(MS_PROTO_ADDTOCONTACT, (WPARAM)Finder->hContact, (LPARAM)YAMN_DBMODULE);
			db_set_s(Finder->hContact, YAMN_DBMODULE, "Id", Finder->Name);
			db_set_s(Finder->hContact, YAMN_DBMODULE, "Nick", Finder->Name);
			db_set_s(Finder->hContact, "Protocol", "p", YAMN_DBMODULE);
			db_set_w(Finder->hContact, YAMN_DBMODULE, "Status", ID_STATUS_ONLINE);
			db_set_s(Finder->hContact, "CList", "StatusMsg", Translate("No new mail message"));
}	}	}