/* 
Copyright (C) 2010 Unsane

This is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.

This 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
Library General Public License for more details.

You should have received a copy of the GNU Library General Public
License along with this file; see the file license.txt.  If
not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
Boston, MA 02111-1307, USA.  
*/

#include "stdafx.h"

using namespace std;

BYTE iNumber;

HANDLE hOnOptInitialized;
HANDLE hOnButtonPressed;
HANDLE hQuickRepliesService;

INT_PTR QuickRepliesService(WPARAM, LPARAM)
{
	return TRUE;
}

static IconItem icon = { LPGEN("Button"), "qr_button", IDI_QICON };

int OnModulesLoaded(WPARAM, LPARAM)
{
	UnhookEvent(hOnModulesLoaded);

	if ( !ServiceExists(MS_QUICKREPLIES_SERVICE)) {
		iNumber = 0;
		hQuickRepliesService = CreateServiceFunction(MS_QUICKREPLIES_SERVICE, QuickRepliesService);
	}
	else iNumber = db_get_b(NULL, MODULE, "InstancesCount", 0);
	db_set_b(NULL, MODULE, "InstancesCount", iNumber + 1);

	hOnOptInitialized = HookEvent(ME_OPT_INITIALISE, OnOptInitialized);
	hOnButtonPressed = HookEvent(ME_MSG_BUTTONPRESSED, OnButtonPressed);

	Icon_Register(hInstance, "TabSRMM/Quick Replies", &icon, 1);

	char buttonNameTranslated[32], buttonName[32];
	mir_snprintf(buttonNameTranslated, "%s %x", Translate("Button"), iNumber + 1);
	mir_snprintf(buttonName, MODULE" %x", iNumber + 1);

	BBButton bbd = {};
	bbd.bbbFlags = BBBF_ISIMBUTTON | BBBF_ISCHATBUTTON;
	bbd.pszModuleName = buttonName;
	bbd.pwszTooltip = LPGENW("Quick Replies\r\nLeft button - open menu\r\nRight button - options page");
	bbd.hIcon = icon.hIcolib;
	bbd.dwButtonID = iNumber;
	bbd.dwDefPos = 220;

	Srmm_AddButton(&bbd);
	return 0;
}

int OnButtonPressed(WPARAM wParam, LPARAM lParam)
{
	CustomButtonClickData *cbcd = (CustomButtonClickData *)lParam;

	char buttonName[32];
	mir_snprintf(buttonName, MODULE" %x", iNumber + 1);
	if (mir_strcmp(cbcd->pszModule, buttonName))
		return 0;

	if (cbcd->dwButtonId != iNumber)
		return 1;

	char key[64];
	mir_snprintf(key, "RepliesCount_%x", iNumber);
	int count = db_get_w(NULL, MODULE, key, 0);

	if (count == 0 || cbcd->flags & BBCF_RIGHTBUTTON)
	{
		mir_snprintf(buttonName, "%s %x", Translate("Button"), iNumber + 1);
		Options_Open(L"Message Sessions", L"Quick Replies", _A2T(buttonName));
		return 0;
	}

	HMENU hMenu = CreatePopupMenu();

	LIST<wchar_t> replyList(1);
	for (int i = 0; i < count; i++)
	{
		mir_snprintf(key, "Reply_%x_%x", iNumber, i);
		wchar_t *value = db_get_wsa(NULL, MODULE, key);

		if (!value)
			replyList.insert(mir_wstrdup(L""));
		else
			replyList.insert(variables_parsedup(value, 0, wParam));

		if (!mir_wstrcmp(value, L"---"))
			AppendMenu((HMENU)hMenu, MF_SEPARATOR, i + 1, NULL);
		else
			AppendMenu((HMENU)hMenu, MF_STRING, i + 1, replyList[i]);

		mir_free(value);
	}

	int index = TrackPopupMenu(hMenu, TPM_RETURNCMD, cbcd->pt.x, cbcd->pt.y, 0, cbcd->hwndFrom, NULL);
	if (index > 0)
	{
		if (mir_wstrcmp(replyList[index - 1], L""))
		{
			HWND hEdit = GetDlgItem(cbcd->hwndFrom, IDC_MESSAGE);
			if (!hEdit) hEdit = GetDlgItem(cbcd->hwndFrom, IDC_CHATMESSAGE);

			SendMessage(hEdit, EM_REPLACESEL, TRUE, (LPARAM)replyList[index - 1]);

			mir_snprintf(key, "ImmediatelySend_%x", iNumber);
			if ((BYTE)db_get_b(NULL, MODULE, key, 1) || cbcd->flags & BBCF_CONTROLPRESSED)
				SendMessage(cbcd->hwndFrom, WM_COMMAND, IDOK, 0);
		}
	}
	
	for (int i = 0; i < replyList.getCount(); i++)
		mir_free(replyList[i]);
	replyList.destroy();

	return 1;
}

int OnPreShutdown(WPARAM, LPARAM)
{
	UnhookEvent(hOnButtonPressed);
	UnhookEvent(hOnOptInitialized);
	UnhookEvent(hOnPreShutdown);
	return 0;
}