#include "common.h"
#include "services.h"
#include "popwin.h"
#include "message_pump.h"
#include "resource.h"
#include "popup_history.h"
#include <time.h>
#include "str_utils.h"

#define NUM_SERVICES		20
HANDLE hService[NUM_SERVICES];
HANDLE hMenuShowHistory, hMenuToggleOnOff;

void StripBBCodesInPlace(wchar_t *text) {
	if(text == 0 || DBGetContactSettingByte(0, MODULE, "StripBBCodes", 1) == 0)
		return;

	int read = 0, write = 0;
	int len = (int)wcslen(text);

	while(read <= len) { // copy terminating null too
		while(read <= len && text[read] != L'[') {
			if(text[read] != text[write]) text[write] = text[read];
			read++; write++;
		}
		if(read > len) break;

		if(len - read >= 3 && (wcsnicmp(text + read, L"[b]", 3) == 0 || wcsnicmp(text + read, L"[i]", 3) == 0))
			read += 3;
		else if(len - read >= 4 && (wcsnicmp(text + read, L"[/b]", 4) == 0 || wcsnicmp(text + read, L"[/i]", 4) == 0))
			read += 4;
		else if(len - read >= 6 && (wcsnicmp(text + read, L"[color", 6) == 0)) {
			while(read < len && text[read] != L']') read++; 
			read++;// skip the ']'
		} else if(len - read >= 8 && (wcsnicmp(text + read, L"[/color]", 8) == 0))
			read += 8;
		else if(len - read >= 5 && (wcsnicmp(text + read, L"[size", 5) == 0)) {
			while(read < len && text[read] != L']') read++; 
			read++;// skip the ']'
		} else if(len - read >= 7 && (wcsnicmp(text + read, L"[/size]", 7) == 0))
			read += 7;
		else {
			if(text[read] != text[write]) text[write] = text[read];
			read++; write++;
		}
	}
}

INT_PTR OldCreatePopupA(WPARAM wParam, LPARAM lParam) {
	POPUPDATA *pd_in = (POPUPDATA *)wParam;
	PopupData *pd_out = (PopupData *)malloc(sizeof(PopupData));
	ZeroMemory(pd_out, sizeof(PopupData));

	pd_out->cbSize = sizeof(PopupData);
	pd_out->flags = PDF_UNICODE;
	pd_out->pwzTitle = a2w(pd_in->lpzContactName);
	pd_out->pwzText = a2w(pd_in->lpzText);
	StripBBCodesInPlace(pd_out->pwzTitle);
	StripBBCodesInPlace(pd_out->pwzText);

	pd_out->hContact = pd_in->lchContact;
	pd_out->hIcon = pd_in->lchIcon;
	if(pd_in->colorBack == 0xffffffff) // that's the old #define for 'skinned bg'
		pd_out->colorBack = pd_out->colorText = 0;
	else {
		pd_out->colorBack = pd_in->colorBack & 0xFFFFFF;
		pd_out->colorText = pd_in->colorText & 0xFFFFFF;
	}
	pd_out->windowProc = pd_in->PluginWindowProc;
	pd_out->opaque = pd_in->PluginData;
	pd_out->timeout = 0;

	lstPopupHistory.Add(pd_out->pwzTitle, pd_out->pwzText, time(0));

	if(!DBGetContactSettingByte(0, MODULE, "Enabled", 1)) {
		free(pd_out->pwzTitle);
		free(pd_out->pwzText);
		free(pd_out);
		return -1;
	}

	//MessageBox(0, pd_out->lpwzContactName, _T("CreatePopupA"), MB_OK);
	PostMPMessage(MUM_CREATEPOPUP, 0, (LPARAM)pd_out);
	return 0;
}

INT_PTR OldCreatePopupExA(WPARAM wParam, LPARAM lParam) {

	POPUPDATAEX *pd_in = (POPUPDATAEX *)wParam;
	PopupData *pd_out = (PopupData *)malloc(sizeof(PopupData));
	ZeroMemory(pd_out, sizeof(PopupData));
	pd_out->cbSize = sizeof(PopupData);
	pd_out->flags = PDF_UNICODE;
	pd_out->pwzTitle = a2w(pd_in->lpzContactName);
	pd_out->pwzText = a2w(pd_in->lpzText);
	StripBBCodesInPlace(pd_out->pwzTitle);
	StripBBCodesInPlace(pd_out->pwzText);

	pd_out->hContact = pd_in->lchContact;
	pd_out->hIcon = pd_in->lchIcon;
	if(pd_in->colorBack == 0xffffffff) // that's the old #define for 'skinned bg'
		pd_out->colorBack = pd_out->colorText = 0;
	else {
		pd_out->colorBack = pd_in->colorBack & 0xFFFFFF;
		pd_out->colorText = pd_in->colorText & 0xFFFFFF;
	}
	pd_out->windowProc = pd_in->PluginWindowProc;
	pd_out->opaque= pd_in->PluginData;
	pd_out->timeout = pd_in->iSeconds;

	lstPopupHistory.Add(pd_out->pwzTitle, pd_out->pwzText, time(0));
	if(!DBGetContactSettingByte(0, MODULE, "Enabled", 1)) {
		free(pd_out->pwzTitle);
		free(pd_out->pwzText);
		free(pd_out);
		return -1;
	}

	//MessageBox(0, pd_out->lpwzContactName, _T("CreatePopupExA"), MB_OK);
	PostMPMessage(MUM_CREATEPOPUP, 0, (LPARAM)pd_out);
	return 0;
}

INT_PTR OldCreatePopupW(WPARAM wParam, LPARAM lParam) {

	POPUPDATAW *pd_in = (POPUPDATAW *)wParam;
	PopupData *pd_out = (PopupData *)malloc(sizeof(PopupData));
	ZeroMemory(pd_out, sizeof(PopupData));
	pd_out->cbSize = sizeof(PopupData);
	pd_out->flags = PDF_UNICODE;
	pd_out->pwzTitle = wcsdup(pd_in->lpwzContactName);
	pd_out->pwzText = wcsdup(pd_in->lpwzText);
	StripBBCodesInPlace(pd_out->pwzTitle);
	StripBBCodesInPlace(pd_out->pwzText);

	pd_out->hContact = pd_in->lchContact;
	pd_out->hIcon = pd_in->lchIcon;
	if(pd_in->colorBack == 0xffffffff) // that's the old #define for 'skinned bg'
		pd_out->colorBack = pd_out->colorText = 0;
	else {
		pd_out->colorBack = pd_in->colorBack & 0xFFFFFF;
		pd_out->colorText = pd_in->colorText & 0xFFFFFF;
	}
	pd_out->windowProc = pd_in->PluginWindowProc;
	pd_out->opaque= pd_in->PluginData;
	pd_out->timeout = pd_in->iSeconds;

	lstPopupHistory.Add(pd_out->pwzTitle, pd_out->pwzText, time(0));
	if(!DBGetContactSettingByte(0, MODULE, "Enabled", 1)) {
		free(pd_out->pwzTitle);
		free(pd_out->pwzText);
		free(pd_out);
		return -1;
	}

	//MessageBox(0, pd_out->lpwzContactName, _T("CreatePopupW"), MB_OK);
	PostMPMessage(MUM_CREATEPOPUP, 0, (LPARAM)pd_out);
	return 0;
}

INT_PTR ChangeTextW(WPARAM wParam, LPARAM lParam) {
	HWND hwndPop = (HWND)wParam;
	wchar_t *newText = wcsdup((wchar_t *)lParam);
	StripBBCodesInPlace(newText);

	if(IsWindow(hwndPop))
		PostMessage(hwndPop, PUM_SETTEXT, 0, (LPARAM)newText);
	return 0;
}

INT_PTR ChangeTextA(WPARAM wParam, LPARAM lParam) {
	HWND hwndPop = (HWND)wParam;
	char *newText = (char *)lParam;
	wchar_t buff[MAX_SECONDLINE];
	a2w(newText, buff, MAX_SECONDLINE);
	StripBBCodesInPlace(buff);

	if(IsWindow(hwndPop))
		PostMessage(hwndPop, PUM_SETTEXT, 0, (LPARAM)wcsdup(buff));
	return 0;
}

INT_PTR ShowPopup(WPARAM wParam, LPARAM lParam) {
	PopupData *pd_in = (PopupData *)wParam;
	PopupData *pd_out = (PopupData *)malloc(sizeof(PopupData));
	*pd_out = *pd_in;
	if(pd_in->flags & PDF_UNICODE) {
		if(pd_in->pwzTitle) pd_out->pwzTitle = wcsdup(pd_in->pwzTitle);
		if(pd_in->pwzText) pd_out->pwzText = wcsdup(pd_in->pwzText);
	} else {
		pd_out->flags |= PDF_UNICODE;
		if(pd_in->pszTitle) pd_out->pwzTitle = a2w(pd_in->pszTitle);
		if(pd_in->pszText) pd_out->pwzText = a2w(pd_in->pszText); 
	}
	StripBBCodesInPlace(pd_out->pwzTitle);
	StripBBCodesInPlace(pd_out->pwzText);

	lstPopupHistory.Add(pd_out->pwzTitle, pd_out->pwzText, time(0));
	if(!DBGetContactSettingByte(0, MODULE, "Enabled", 1)) {
		if(pd_out->pwzTitle) free(pd_out->pwzTitle);
		if(pd_out->pwzText) free(pd_out->pwzText);
		free(pd_out);
		return -1;
	}

	//MessageBox(0, pd_out->lpwzContactName, _T("CreatePopupW"), MB_OK);
	PostMPMessage(MUM_CREATEPOPUP, 0, (LPARAM)pd_out);
	return 0;
}

INT_PTR GetContact(WPARAM wParam, LPARAM lParam) {
	HWND hwndPop = (HWND)wParam;
	HANDLE hContact;
	if(GetCurrentThreadId() == message_pump_thread_id) {
		SendMessage(hwndPop, PUM_GETCONTACT, (WPARAM)&hContact, 0);
	} else {
		HANDLE hEvent = CreateEvent(0, 0, 0, 0);
		PostMessage(hwndPop, PUM_GETCONTACT, (WPARAM)&hContact, (LPARAM)hEvent);
		MsgWaitForMultipleObjectsEx(1, &hEvent, INFINITE, 0, 0);
		CloseHandle(hEvent);
	}

	return (INT_PTR)hContact;
}

INT_PTR GetOpaque(WPARAM wParam, LPARAM lParam) {
	HWND hwndPop = (HWND)wParam;
	void *data = 0;
	if(GetCurrentThreadId() == message_pump_thread_id) {
		SendMessage(hwndPop, PUM_GETOPAQUE, (WPARAM)&data, 0);
	} else {
		HANDLE hEvent = CreateEvent(0, 0, 0, 0);
		PostMessage(hwndPop, PUM_GETOPAQUE, (WPARAM)&data, (LPARAM)hEvent);
		MsgWaitForMultipleObjectsEx(1, &hEvent, INFINITE, 0, 0);
		CloseHandle(hEvent);
	}

	return (INT_PTR)data;
}

INT_PTR IsSecondLineShown(WPARAM wParam, LPARAM lParam) {
	return TRUE;
}

void UpdateMenu() {
	CLISTMENUITEM menu = {0};
	menu.cbSize = sizeof(CLISTMENUITEM);
	menu.pszName = (DBGetContactSettingByte(0, MODULE, "Enabled", 1) == 1 ? Translate("Disable Popups") : Translate("Enable Popups"));
	menu.flags = CMIM_NAME;// | CMIM_ICON;
	CallService(MS_CLIST_MODIFYMENUITEM, (WPARAM)hMenuToggleOnOff, (LPARAM)&menu);
}

INT_PTR PopupQuery(WPARAM wParam, LPARAM lParam) {
	switch(wParam) {
		case PUQS_ENABLEPOPUPS:
			{
				bool enabled = (DBGetContactSettingByte(0, MODULE, "Enabled", 1) == 1);
				if(!enabled) DBWriteContactSettingByte(0, MODULE, "Enabled", 1);
				return enabled ? 0: 1;
			}
			break;
		case PUQS_DISABLEPOPUPS:
			{
				bool enabled = (DBGetContactSettingByte(0, MODULE, "Enabled", 1) == 1);
				if(enabled) DBWriteContactSettingByte(0, MODULE, "Enabled", 0);
				DBWriteContactSettingByte(0, MODULE, "Enabled", 0);
				return enabled ? 1 : 0;
			}
			break;
		case PUQS_GETSTATUS:
			return DBGetContactSettingByte(0, MODULE, "Enabled", 1);
		default:
			return 1;
	}
	UpdateMenu();
	return 0;
}

INT_PTR TogglePopups(WPARAM wParam, LPARAM lParam) {
	if(DBGetContactSettingByte(0, MODULE, "Enabled", 1)) {
		DBWriteContactSettingByte(0, MODULE, "Enabled", 0);
	} else {
		DBWriteContactSettingByte(0, MODULE, "Enabled", 1);
	}
	UpdateMenu();
	return 0;
}

INT_PTR PopupChangeA(WPARAM wParam, LPARAM lParam) {
	HWND hwndPop = (HWND)wParam;
	POPUPDATAEX *pd_in = (POPUPDATAEX *)lParam;

	if(IsWindow(hwndPop)) {
		PopupData *pd_out = (PopupData *)malloc(sizeof(PopupData));
		pd_out->cbSize = sizeof(PopupData);
		pd_out->flags = PDF_UNICODE;

		pd_out->pwzTitle = a2w(pd_in->lpzContactName);
		pd_out->pwzText = a2w(pd_in->lpzText);
		StripBBCodesInPlace(pd_out->pwzTitle);
		StripBBCodesInPlace(pd_out->pwzText);

		pd_out->hContact = pd_in->lchContact;
		pd_out->hIcon = pd_in->lchIcon;
		if(pd_in->colorBack == 0xffffffff) // that's the old #define for 'skinned bg'
			pd_out->colorBack = pd_out->colorText = 0;
		else {
			pd_out->colorBack = pd_in->colorBack & 0xFFFFFF;
			pd_out->colorText = pd_in->colorText & 0xFFFFFF;
		}
		pd_out->colorBack = pd_in->colorBack;
		pd_out->colorText = pd_in->colorText;
		pd_out->windowProc = pd_in->PluginWindowProc;
		pd_out->opaque = pd_in->PluginData;
		pd_out->timeout = pd_in->iSeconds;

		lstPopupHistory.Add(pd_out->pwzTitle, pd_out->pwzText, time(0));
	
		PostMessage(hwndPop, PUM_CHANGE, 0, (LPARAM)pd_out);
	}
	return 0;
}

INT_PTR PopupChangeW(WPARAM wParam, LPARAM lParam) {
	HWND hwndPop = (HWND)wParam;
	POPUPDATAW *pd_in = (POPUPDATAW *)lParam;

	if(IsWindow(hwndPop)) {
		PopupData *pd_out = (PopupData *)malloc(sizeof(PopupData));
		pd_out->cbSize = sizeof(PopupData);
		pd_out->flags = PDF_UNICODE;

		pd_out->pwzTitle = wcsdup(pd_in->lpwzContactName);
		pd_out->pwzText = wcsdup(pd_in->lpwzText);
		StripBBCodesInPlace(pd_out->pwzTitle);
		StripBBCodesInPlace(pd_out->pwzText);

		pd_out->hContact = pd_in->lchContact;
		pd_out->hIcon = pd_in->lchIcon;
		if(pd_in->colorBack == 0xffffffff) // that's the old #define for 'skinned bg'
			pd_out->colorBack = pd_out->colorText = 0;
		else {
			pd_out->colorBack = pd_in->colorBack & 0xFFFFFF;
			pd_out->colorText = pd_in->colorText & 0xFFFFFF;
		}
		pd_out->colorBack = pd_in->colorBack;
		pd_out->colorText = pd_in->colorText;
		pd_out->windowProc = pd_in->PluginWindowProc;
		pd_out->opaque = pd_in->PluginData;
		pd_out->timeout = pd_in->iSeconds;

		lstPopupHistory.Add(pd_out->pwzTitle, pd_out->pwzText, time(0));
	
		PostMessage(hwndPop, PUM_CHANGE, 0, (LPARAM)pd_out);
	}
	return 0;
}

INT_PTR ShowMessage(WPARAM wParam, LPARAM lParam) {
	if(!DBGetContactSettingByte(0, MODULE, "Enabled", 1)) return 0;

	POPUPDATAT pd = {0};
	_tcscpy(pd.lptzContactName, lParam == SM_WARNING ? _T("Warning") : _T("Notification"));
	pd.lchIcon = LoadIcon(0, lParam == SM_WARNING ? IDI_WARNING : IDI_INFORMATION);
	TCHAR buff[MAX_SECONDLINE];
	a2t((char *)wParam, buff, MAX_SECONDLINE);
	_tcscpy(pd.lptzText, buff);

	CallService(MS_POPUP_ADDPOPUPT, (WPARAM)&pd, 0);

	return 0;
}

INT_PTR ShowMessageW(WPARAM wParam, LPARAM lParam) {
	if(!DBGetContactSettingByte(0, MODULE, "Enabled", 1)) return 0;

	POPUPDATAW pd = {0};
	wcscpy(pd.lpwzContactName, lParam == SM_WARNING ? L"Warning" : L"Notification");
	pd.lchIcon = LoadIcon(0, lParam == SM_WARNING ? IDI_WARNING : IDI_INFORMATION);
	wcsncpy(pd.lpwzText, (wchar_t *)wParam, MAX_SECONDLINE);

	CallService(MS_POPUP_ADDPOPUPW, (WPARAM)&pd, 0);

	return 0;
}

//=====PopUp/ShowHistory
//extern BOOL CALLBACK DlgProcHistLstOpts(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
INT_PTR PopUp_ShowHistory(WPARAM wParam, LPARAM lParam)
{
	if (!hHistoryWindow) {
		hHistoryWindow = CreateDialog(hInst, MAKEINTRESOURCE(IDD_LST_HISTORY), NULL, DlgProcHistLst);
	}
	ShowWindow(hHistoryWindow, SW_SHOW);
	return 0;
}

int num_classes = 0;
PopupClass *classes = 0;

INT_PTR RegisterPopupClass(WPARAM wParam, LPARAM lParam) {
	PopupClass *pc = (PopupClass *)lParam;
	pc->pszName = strdup(pc->pszName);
	if(pc->flags & PCF_UNICODE)
		pc->pwszDescription = wcsdup(pc->pwszDescription);
	else
		pc->pszDescription = strdup(pc->pszDescription);
	
	char setting[256];
	mir_snprintf(setting, 256, "%s/Timeout", pc->pszName);
	pc->timeout = DBGetContactSettingWord(0, MODULE, setting, pc->timeout);
	if(pc->timeout == (WORD)-1) pc->timeout = -1;
	mir_snprintf(setting, 256, "%s/TextCol", pc->pszName);
	pc->colorText = (COLORREF)DBGetContactSettingDword(0, MODULE, setting, (DWORD)pc->colorText);
	mir_snprintf(setting, 256, "%s/BgCol", pc->pszName);
	pc->colorBack = (COLORREF)DBGetContactSettingDword(0, MODULE, setting, (DWORD)pc->colorBack);

	classes = (PopupClass *)realloc(classes, sizeof(PopupClass) * (num_classes + 1));
	memcpy(classes + num_classes, pc, sizeof(PopupClass));
	num_classes++;
	return 0;
}

INT_PTR CreateClassPopup(WPARAM wParam, LPARAM lParam) {
	PopupClassInstance *pdc = (PopupClassInstance *)lParam;
	if(pdc->cbSize != sizeof(PopupClassInstance)) return 1;

	PopupClass *pc = 0;
	if(wParam) pc = (PopupClass *)wParam;
	else {
		for(int i = 0; i < num_classes; i++) {
			if(strcmp(classes[i].pszName, pdc->pszClassName) == 0) {
				pc = &classes[i];
				break;
			}
		}
	}
	if(pc) {
		PopupData pd = {sizeof(PopupData)};
		if(pc->flags & PCF_UNICODE) pd.flags |= PDF_UNICODE;
		pd.colorBack = pc->colorBack;
		pd.colorText = pc->colorText;
		pd.hIcon = pc->hIcon;
		pd.timeout = pc->timeout;
		pd.windowProc = pc->windowProc;

		pd.hContact = pdc->hContact;
		pd.opaque = pdc->opaque;
		pd.pszTitle = (char *)pdc->pszTitle;
		pd.pszText = (char *)pdc->pszText;
	
		CallService(MS_YAPP_SHOWPOPUP, (WPARAM)&pd, 0);
	}
	return 0;
}

int PrebuildMenu(WPARAM wParam, LPARAM lParam) {
	return 0;
}

HANDLE hEventBuildMenu;

void InitServices() {
	int i = 0;
	hService[i++] = CreateServiceFunction(MS_YAPP_SHOWPOPUP, ShowPopup);
	hService[i++] = CreateServiceFunction(MS_YAPP_REGISTERCLASS, RegisterPopupClass);
	hService[i++] = CreateServiceFunction(MS_YAPP_CLASSINSTANCE, CreateClassPopup);
	hService[i++] = CreateServiceFunction(MS_YAPP_GETCONTACT, GetContact);
	hService[i++] = CreateServiceFunction(MS_YAPP_GETOPAQUE, GetOpaque);

	// legacy support
	hService[i++] = CreateServiceFunction(MS_POPUP_ADDPOPUP, OldCreatePopupA);
	hService[i++] = CreateServiceFunction(MS_POPUP_ADDPOPUPEX, OldCreatePopupExA);
	hService[i++] = CreateServiceFunction(MS_POPUP_ADDPOPUPW, OldCreatePopupW);
	hService[i++] = CreateServiceFunction(MS_POPUP_CHANGETEXTW, ChangeTextW);
	hService[i++] = CreateServiceFunction(MS_POPUP_CHANGETEXT, ChangeTextA);
	hService[i++] = CreateServiceFunction(MS_POPUP_CHANGE, PopupChangeA);
	hService[i++] = CreateServiceFunction(MS_POPUP_CHANGEW, PopupChangeW);
	hService[i++] = CreateServiceFunction(MS_POPUP_GETCONTACT, GetContact);
	hService[i++] = CreateServiceFunction(MS_POPUP_GETPLUGINDATA, GetOpaque);
	hService[i++] = CreateServiceFunction(MS_POPUP_ISSECONDLINESHOWN, IsSecondLineShown);
	hService[i++] = CreateServiceFunction(MS_POPUP_QUERY, PopupQuery);

	hService[i++] = CreateServiceFunction(MS_POPUP_SHOWMESSAGE, ShowMessage);
	hService[i++] = CreateServiceFunction(MS_POPUP_SHOWMESSAGE"W", ShowMessageW);

	hService[i++] = CreateServiceFunction(MS_POPUP_SHOWHISTORY, PopUp_ShowHistory);
	hService[i++] = CreateServiceFunction("PopUp/ToggleEnabled", TogglePopups);


	CLISTMENUITEM menu = {0};

	menu.cbSize=sizeof(menu);
	menu.flags = CMIM_ALL;

	menu.hIcon=0;
	menu.position = 500010000;
	menu.pszPopupName = Translate("PopUps");

	hiPopupHistory = LoadIcon(hInst, MAKEINTRESOURCE(IDI_POPUP_HISTORY));
	menu.hIcon = hiPopupHistory;
	menu.pszService= MS_POPUP_SHOWHISTORY;
	menu.pszName = Translate("Popup History");
	hMenuShowHistory = (HANDLE)CallService(MS_CLIST_ADDMAINMENUITEM, 0, (LPARAM)&menu);
	menu.hIcon = 0;

	menu.pszService= "PopUp/ToggleEnabled";
	menu.pszName = (DBGetContactSettingByte(0, MODULE, "Enabled", 1) == 1 ? Translate("Disable Popups") : Translate("Enable Popups"));
	hMenuToggleOnOff = (HANDLE)CallService(MS_CLIST_ADDMAINMENUITEM, 0, (LPARAM)&menu);

	hEventBuildMenu = HookEvent(ME_CLIST_PREBUILDCONTACTMENU, PrebuildMenu);

	if(ServiceExists(MS_LANGPACK_GETCODEPAGE)) 
		code_page = CallService(MS_LANGPACK_GETCODEPAGE, 0, 0);
}

void DeinitServices() {
	int i;
	for(i = 0; i < num_classes; i++) {
		free(classes[i].pszName);
		free(classes[i].pszDescription);
	}
	free(classes); num_classes = 0;

	UnhookEvent(hEventBuildMenu);

	for(i = 0; i < NUM_SERVICES; i++)
		if(hService[i]) DestroyServiceFunction(hService[i]);
}