/*

Simple Status Message plugin for Miranda IM
Copyright (C) 2006-2011 Bartosz 'Dezeath' Białek, (C) 2005 Harven

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.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/

#include "stdafx.h"

CMPlugin g_plugin;

PROTOACCOUNTS *accounts;
static int g_iIdleTime = -1;
UINT_PTR g_uUpdateMsgTimer = 0, *g_uSetStatusTimer;
static wchar_t *g_ptszWinampSong;
HANDLE hTTBButton = nullptr, h_statusmodechange;
HWND hwndSAMsgDialog;
static HANDLE *hProtoStatusMenuItem;

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

PLUGININFOEX pluginInfoEx =
{
	sizeof(PLUGININFOEX),
	__PLUGIN_NAME,
	PLUGIN_MAKE_VERSION(__MAJOR_VERSION, __MINOR_VERSION, __RELEASE_NUM, __BUILD_NUM),
	__DESCRIPTION,
	__AUTHOR,
	__COPYRIGHT,
	__AUTHORWEB,
	UNICODE_AWARE,
	// {768CE156-34AC-45A3-B53B-0083C47615C4}
	{ 0x768ce156, 0x34ac, 0x45a3, { 0xb5, 0x3b, 0x0, 0x83, 0xc4, 0x76, 0x15, 0xc4 }}
};

CMPlugin::CMPlugin() :
	PLUGIN<CMPlugin>(MODULENAME, pluginInfoEx)
{}

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

extern "C" __declspec(dllexport) const MUUID MirandaInterfaces[] = { MIID_SRAWAY, MIID_LAST };

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

static wchar_t* GetWinampSong(void)
{
	HWND hwndWinamp = FindWindow(L"STUDIO", nullptr);
	if (hwndWinamp == nullptr)
		hwndWinamp = FindWindow(L"Winamp v1.x", nullptr);

	if (hwndWinamp == nullptr)
		return nullptr;

	int iTitleLen = GetWindowTextLength(hwndWinamp);
	wchar_t *szTitle = (wchar_t *)mir_alloc((iTitleLen + 1) * sizeof(wchar_t));
	if (szTitle == nullptr)
		return nullptr;

	if (GetWindowText(hwndWinamp, szTitle, iTitleLen + 1) == 0) {
		mir_free(szTitle);
		return nullptr;
	}

	wchar_t *pstr = wcsstr(szTitle, L" - Winamp");
	if (pstr == nullptr) {
		mir_free(szTitle);
		return nullptr;
	}

	if (pstr < szTitle + (iTitleLen / 2)) {
		memmove(szTitle, pstr + 9, mir_wstrlen(pstr + 9) * sizeof(wchar_t));
		pstr = wcsstr(pstr + 1, L" - Winamp");
		if (pstr == nullptr) {
			mir_free(szTitle);
			return nullptr;
		}
	}
	*pstr = 0;

	pstr = wcschr(szTitle, '.');
	if (pstr == nullptr) {
		mir_free(szTitle);
		return nullptr;
	}

	pstr += 2;
	wchar_t *res = mir_wstrdup(pstr);
	mir_free(szTitle);

	return res;
}

wchar_t* InsertBuiltinVarsIntoMsg(wchar_t *in, const char *szProto, int)
{
	if (in == nullptr)
		return nullptr;

	int count = 0;
	wchar_t substituteStr[1024], *msg = mir_wstrdup(in);

	for (int i = 0; msg[i]; i++) {
		if (msg[i] == 0x0D && g_plugin.getByte("RemoveCR", 0)) {
			wchar_t *p = msg + i;
			if (i + 1 <= 1024 && msg[i + 1]) {
				if (msg[i + 1] == 0x0A) {
					if (i + 2 <= 1024 && msg[i + 2]) {
						count++;
						memmove(p, p + 1, (mir_wstrlen(p) - 1) * sizeof(wchar_t));
					}
					else {
						msg[i + 1] = 0;
						msg[i] = 0x0A;
					}
				}
			}
		}

		if (msg[i] != '%')
			continue;

		if (!wcsnicmp(msg + i, L"%winampsong%", 12)) {
			wchar_t *ptszWinampTitle = GetWinampSong();

			if (ptszWinampTitle != nullptr) {
				mir_free(g_ptszWinampSong);
				g_ptszWinampSong = mir_wstrdup(ptszWinampTitle);
			}
			else if (g_ptszWinampSong && mir_wstrcmp(g_ptszWinampSong, TEXT(MODULENAME))
				&& g_plugin.getByte("AmpLeaveTitle", 1)) {
				ptszWinampTitle = mir_wstrdup(g_ptszWinampSong);
			}
			else
				continue;

			if (mir_wstrlen(ptszWinampTitle) > 12)
				msg = (wchar_t *)mir_realloc(msg, (mir_wstrlen(msg) + 1 + mir_wstrlen(ptszWinampTitle) - 12) * sizeof(wchar_t));

			memmove(msg + i + mir_wstrlen(ptszWinampTitle), msg + i + 12, (mir_wstrlen(msg) - i - 11) * sizeof(wchar_t));
			memcpy(msg + i, ptszWinampTitle, mir_wstrlen(ptszWinampTitle) * sizeof(wchar_t));

			mir_free(ptszWinampTitle);
		}
		else if (!wcsnicmp(msg + i, L"%time%", 6)) {
			MIRANDA_IDLE_INFO mii;
			Idle_GetInfo(mii);

			if (mii.idleType) {
				int mm;
				SYSTEMTIME t;
				GetLocalTime(&t);
				if ((mm = g_iIdleTime) == -1) {
					mm = t.wMinute + t.wHour * 60;
					if (mii.idleType == 1) {
						mm -= mii.idleTime;
						if (mm < 0) mm += 60 * 24;
					}
					g_iIdleTime = mm;
				}
				t.wMinute = mm % 60;
				t.wHour = mm / 60;
				GetTimeFormat(LOCALE_USER_DEFAULT, TIME_NOSECONDS, &t, nullptr, substituteStr, _countof(substituteStr));
			}
			else GetTimeFormat(LOCALE_USER_DEFAULT, TIME_NOSECONDS, nullptr, nullptr, substituteStr, _countof(substituteStr));

			if (mir_wstrlen(substituteStr) > 6)
				msg = (wchar_t *)mir_realloc(msg, (mir_wstrlen(msg) + 1 + mir_wstrlen(substituteStr) - 6) * sizeof(wchar_t));

			memmove(msg + i + mir_wstrlen(substituteStr), msg + i + 6, (mir_wstrlen(msg) - i - 5) * sizeof(wchar_t));
			memcpy(msg + i, substituteStr, mir_wstrlen(substituteStr) * sizeof(wchar_t));
		}
		else if (!wcsnicmp(msg + i, L"%date%", 6)) {
			GetDateFormat(LOCALE_USER_DEFAULT, DATE_SHORTDATE, nullptr, nullptr, substituteStr, _countof(substituteStr));

			if (mir_wstrlen(substituteStr) > 6)
				msg = (wchar_t *)mir_realloc(msg, (mir_wstrlen(msg) + 1 + mir_wstrlen(substituteStr) - 6) * sizeof(wchar_t));

			memmove(msg + i + mir_wstrlen(substituteStr), msg + i + 6, (mir_wstrlen(msg) - i - 5) * sizeof(wchar_t));
			memcpy(msg + i, substituteStr, mir_wstrlen(substituteStr) * sizeof(wchar_t));
		}
		else if (!wcsnicmp(msg + i, L"%rand(", 6)) {
			wchar_t *temp, *token;
			int ran_from, ran_to, k;

			temp = mir_wstrdup(msg + i + 6);
			token = wcstok(temp, L",)");
			ran_from = _wtoi(token);
			token = wcstok(nullptr, L",)%%");
			ran_to = _wtoi(token);

			if (ran_to > ran_from) {
				mir_snwprintf(substituteStr, L"%d", GetRandom(ran_from, ran_to));
				for (k = i + 1; msg[k]; k++) if (msg[k] == '%') { k++; break; }

				if ((int)mir_wstrlen(substituteStr) > k - i)
					msg = (wchar_t *)mir_realloc(msg, (mir_wstrlen(msg) + 1 + mir_wstrlen(substituteStr) - (k - i)) * sizeof(wchar_t));

				memmove(msg + i + mir_wstrlen(substituteStr), msg + i + (k - i), (mir_wstrlen(msg) - i - (k - i - 1)) * sizeof(wchar_t));
				memcpy(msg + i, substituteStr, mir_wstrlen(substituteStr) * sizeof(wchar_t));
			}
			mir_free(temp);
		}
		else if (!wcsnicmp(msg + i, L"%randmsg%", 9)) {
			char buff[16];
			int k, k2 = 0;
			BOOL rmark[25];

			for (k = 0; k < _countof(rmark); k++) rmark[k] = FALSE;
			int maxk = g_plugin.getByte("MaxHist", 10);
			if (maxk == 0) rmark[0] = TRUE;

			while (!rmark[0]) {
				k = GetRandom(1, maxk);
				if (rmark[k]) continue;
				rmark[k] = TRUE;
				k2++;
				if (k2 == maxk || k2 > maxk) rmark[0] = TRUE;

				mir_snprintf(buff, "SMsg%d", k);

				wchar_t *tszStatusMsg = g_plugin.getWStringA(buff);
				if (tszStatusMsg == nullptr)
					continue;

				mir_wstrcpy(substituteStr, tszStatusMsg);
				mir_free(tszStatusMsg);

				if (!mir_wstrlen(substituteStr))
					continue;

				if (wcsstr(substituteStr, L"%randmsg%") != nullptr || wcsstr(substituteStr, L"%randdefmsg%") != nullptr) {
					if (k == maxk) maxk--;
				}
				else rmark[0] = TRUE;
			}

			if (k2 == maxk || k2 > maxk) mir_wstrcpy(substituteStr, L"");

			if (mir_wstrlen(substituteStr) > 9)
				msg = (wchar_t *)mir_realloc(msg, (mir_wstrlen(msg) + 1 + mir_wstrlen(substituteStr) - 9) * sizeof(wchar_t));

			memmove(msg + i + mir_wstrlen(substituteStr), msg + i + 9, (mir_wstrlen(msg) - i - 8) * sizeof(wchar_t));
			memcpy(msg + i, substituteStr, mir_wstrlen(substituteStr) * sizeof(wchar_t));
		}
		else if (!wcsnicmp(msg + i, L"%randdefmsg%", 12)) {
			char buff[16];
			int k, k2 = 0;
			BOOL rmark[25];

			for (k = 0; k < _countof(rmark); k++) rmark[k] = FALSE;
			int maxk = g_plugin.getWord("DefMsgCount", 0);
			if (maxk == 0) rmark[0] = TRUE;

			while (!rmark[0]) {
				k = GetRandom(1, maxk);
				if (rmark[k]) continue;
				rmark[k] = TRUE;
				k2++;
				if (k2 == maxk || k2 > maxk) rmark[0] = TRUE;

				mir_snprintf(buff, "DefMsg%d", k);

				wchar_t *tszStatusMsg = g_plugin.getWStringA(buff);
				if (tszStatusMsg == nullptr)
					continue;

				mir_wstrcpy(substituteStr, tszStatusMsg);
				mir_free(tszStatusMsg);

				if (!mir_wstrlen(substituteStr))
					continue;

				if (wcsstr(substituteStr, L"%randmsg%") != nullptr || wcsstr(substituteStr, L"%randdefmsg%") != nullptr) {
					if (k == maxk) maxk--;
				}
				else rmark[0] = TRUE;
			}

			if (k2 == maxk || k2 > maxk) mir_wstrcpy(substituteStr, L"");

			if (mir_wstrlen(substituteStr) > 12)
				msg = (wchar_t *)mir_realloc(msg, (mir_wstrlen(msg) + 1 + mir_wstrlen(substituteStr) - 12) * sizeof(wchar_t));

			memmove(msg + i + mir_wstrlen(substituteStr), msg + i + 12, (mir_wstrlen(msg) - i - 11) * sizeof(wchar_t));
			memcpy(msg + i, substituteStr, mir_wstrlen(substituteStr) * sizeof(wchar_t));
		}
	}

	if (count)
		msg[mir_wstrlen(msg) - count] = 0;

	if (szProto) {
		char szSetting[80];
		mir_snprintf(szSetting, "Proto%sMaxLen", szProto);
		size_t len = g_plugin.getWord(szSetting, 1024);
		if (len < mir_wstrlen(msg)) {
			msg = (wchar_t *)mir_realloc(msg, len * sizeof(wchar_t));
			msg[len] = 0;
		}
	}

	return msg;
}

wchar_t* InsertVarsIntoMsg(wchar_t *tszMsg, const char *szProto, int iStatus, MCONTACT hContact)
{
	if (ServiceExists(MS_VARS_FORMATSTRING) && g_plugin.getByte("EnableVariables", 1)) {
		wchar_t *tszVarsMsg = variables_parse(tszMsg, nullptr, hContact);
		if (tszVarsMsg != nullptr) {
			wchar_t *format = InsertBuiltinVarsIntoMsg(tszVarsMsg, szProto, iStatus);
			mir_free(tszVarsMsg);
			return format;
		}
	}

	return InsertBuiltinVarsIntoMsg(tszMsg, szProto, iStatus);
}

static wchar_t* GetAwayMessageFormat(int iStatus, const char *szProto)
{
	char szSetting[80];
	wchar_t *format;

	mir_snprintf(szSetting, "%sFlags", szProto ? szProto : "");
	int flags = g_plugin.getByte(StatusModeToDbSetting(iStatus, szSetting), STATUS_DEFAULT);

	if (flags & STATUS_EMPTY_MSG) {
		return mir_wstrdup(L"");
	}
	else if (flags & STATUS_LAST_STATUS_MSG) {
		if (szProto)
			mir_snprintf(szSetting, "%sMsg", szProto);
		else
			mir_snprintf(szSetting, "Msg");

		format = db_get_wsa(0, "SRAway", StatusModeToDbSetting(iStatus, szSetting));
	}
	else if (flags & STATUS_LAST_MSG) {
		if (szProto)
			mir_snprintf(szSetting, "Last%sMsg", szProto);
		else
			mir_snprintf(szSetting, "LastMsg");

		char *szLastMsg = g_plugin.getStringA(szSetting);
		if (szLastMsg == nullptr)
			return nullptr; //mir_wstrdup(L"");

		format = g_plugin.getWStringA(szLastMsg);
		mir_free(szLastMsg);
	}
	else if (flags & STATUS_THIS_MSG) {
		if (szProto)
			mir_snprintf(szSetting, "%sDefault", szProto);
		else
			mir_snprintf(szSetting, "Default");

		format = db_get_wsa(0, "SRAway", StatusModeToDbSetting(iStatus, szSetting));
		if (format == nullptr)
			format = mir_wstrdup(L"");
	}
	else
		format = mir_wstrdup(GetDefaultMessage(iStatus));

	return format;
}

static void DBWriteMessage(const char *szSetting, const wchar_t *tszMsg)
{
	if (tszMsg && mir_wstrlen(tszMsg))
		g_plugin.setWString(szSetting, tszMsg);
	else
		g_plugin.delSetting(szSetting);
}

static void SaveMessageToDB(const char *szProto, const wchar_t *tszMsg, BOOL bIsFormat)
{
	char szSetting[80];

	if (!szProto) {
		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
				continue;

			mir_snprintf(szSetting, bIsFormat ? "FCur%sMsg" : "Cur%sMsg", pa->szModuleName);
			DBWriteMessage(szSetting, tszMsg);
#ifdef _DEBUG
			if (bIsFormat)
				g_plugin.debugLogA("SaveMessageToDB(): Set \"%S\" status message (without inserted vars) for %s.", tszMsg, pa->szModuleName);
			else
				g_plugin.debugLogA("SaveMessageToDB(): Set \"%S\" status message for %s.", tszMsg, pa->szModuleName);
#endif
		}
	}
	else {
		if (!(CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
			return;

		mir_snprintf(szSetting, bIsFormat ? "FCur%sMsg" : "Cur%sMsg", szProto);
		DBWriteMessage(szSetting, tszMsg);
#ifdef _DEBUG
		if (bIsFormat)
			g_plugin.debugLogA("SaveMessageToDB(): Set \"%S\" status message (without inserted vars) for %s.", tszMsg, szProto);
		else
			g_plugin.debugLogA("SaveMessageToDB(): Set \"%S\" status message for %s.", tszMsg, szProto);
#endif
	}
}

void SaveStatusAsCurrent(const char *szProto, int iStatus)
{
	char szSetting[80];
	mir_snprintf(szSetting, "Cur%sStatus", szProto);
	g_plugin.setWord(szSetting, (WORD)iStatus);
}

static wchar_t *GetAwayMessage(int iStatus, const char *szProto, BOOL bInsertVars, MCONTACT hContact)
{
	wchar_t *format;
	char szSetting[80];

	if ((!iStatus || iStatus == ID_STATUS_CURRENT) && szProto) {
		mir_snprintf(szSetting, "FCur%sMsg", szProto);
		format = g_plugin.getWStringA(szSetting);
	}
	else {
		if (!iStatus || iStatus == ID_STATUS_CURRENT)
			iStatus = GetCurrentStatus(szProto);

		if (szProto && !(CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_3, 0) & Proto_Status2Flag(iStatus)))
			return nullptr;

		mir_snprintf(szSetting, "Proto%sFlags", szProto ? szProto : "");
		int flags = g_plugin.getByte(szSetting, PROTO_DEFAULT);

		if (flags & PROTO_THIS_MSG) {
			mir_snprintf(szSetting, "Proto%sDefault", szProto);
			format = g_plugin.getWStringA(szSetting);
			if (format == nullptr)
				format = mir_wstrdup(L"");
		}
		else if (flags & PROTO_NOCHANGE && szProto) {
			mir_snprintf(szSetting, "FCur%sMsg", szProto);
			format = g_plugin.getWStringA(szSetting);
		}
		else if (flags & PROTO_POPUPDLG)
			format = GetAwayMessageFormat(iStatus, szProto);
		else
			format = nullptr;
	}
#ifdef _DEBUG
	g_plugin.debugLogA("GetAwayMessage(): %s has %s status and \"%S\" status message.", szProto, StatusModeToDbSetting(iStatus, ""), format);
#endif

	if (bInsertVars && format != nullptr) {
		wchar_t *tszVarsMsg = InsertVarsIntoMsg(format, szProto, iStatus, hContact); // TODO random values not the same!
		mir_free(format);
		return tszVarsMsg;
	}

	return format;
}

int CheckProtoSettings(const char *szProto, int iInitialStatus)
{
	int iSetting = db_get_w(0, szProto, "LeaveStatus", -1); //GG settings
	if (iSetting != -1)
		return iSetting ? iSetting : iInitialStatus;
	iSetting = db_get_w(0, szProto, "OfflineMessageOption", -1); //TLEN settings
	if (iSetting != -1) {
		switch (iSetting) {
		case 1: return ID_STATUS_ONLINE;
		case 2: return ID_STATUS_AWAY;
		case 3: return ID_STATUS_NA;
		case 4: return ID_STATUS_DND;
		case 5: return ID_STATUS_FREECHAT;
		case 6: return ID_STATUS_INVISIBLE;
		default: return iInitialStatus;
		}
	}
	return iInitialStatus;
}

static void Proto_SetAwayMsgT(const char *szProto, int iStatus, wchar_t *tszMsg)
{
	if (!(CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_INDIVMODEMSG))
		CallProtoService(szProto, PS_SETAWAYMSG, (WPARAM)iStatus, (LPARAM)tszMsg);
}

static void Proto_SetStatus(const char *szProto, int iInitialStatus, int iStatus, wchar_t *tszMsg)
{
	if (iStatus == ID_STATUS_OFFLINE && iStatus != iInitialStatus) {
		// ugly hack to set offline status message
		if (!(CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_INDIVMODEMSG)) {
			int iMsgStatus = CheckProtoSettings(szProto, iInitialStatus);
			CallProtoService(szProto, PS_SETAWAYMSG, (WPARAM)iMsgStatus, (LPARAM)tszMsg);
			CallProtoService(szProto, PS_SETSTATUS, (WPARAM)iMsgStatus, 0);
		}
		if (ServiceExists(MS_KS_ANNOUNCESTATUSCHANGE))
			announce_status_change((char*)szProto, ID_STATUS_OFFLINE, nullptr);
		CallProtoService(szProto, PS_SETSTATUS, ID_STATUS_OFFLINE, 0);
		return;
	}

	Proto_SetAwayMsgT(szProto, iStatus, tszMsg /* ? tszMsg : L""*/);
	if (iStatus != iInitialStatus)
		CallProtoService(szProto, PS_SETSTATUS, iStatus, 0);
}

int HasProtoStaticStatusMsg(const char *szProto, int iInitialStatus, int iStatus)
{
	char szSetting[80];
	mir_snprintf(szSetting, "Proto%sFlags", szProto);
	int flags = g_plugin.getByte(szSetting, PROTO_DEFAULT);

	if (flags & PROTO_NO_MSG) {
		Proto_SetStatus(szProto, iInitialStatus, iStatus, nullptr);
		SaveMessageToDB(szProto, nullptr, TRUE);
		SaveMessageToDB(szProto, nullptr, FALSE);
		return 1;
	}
	else if (flags & PROTO_THIS_MSG) {
		mir_snprintf(szSetting, "Proto%sDefault", szProto);
		wchar_t *szSimpleStatusMsg = g_plugin.getWStringA(szSetting);
		if (szSimpleStatusMsg != nullptr) {
			SaveMessageToDB(szProto, szSimpleStatusMsg, TRUE);
			wchar_t *msg = InsertVarsIntoMsg(szSimpleStatusMsg, szProto, iStatus, NULL);
			mir_free(szSimpleStatusMsg);
			Proto_SetStatus(szProto, iInitialStatus, iStatus, msg);
			SaveMessageToDB(szProto, msg, FALSE);
			mir_free(msg);
		}
		else {
			Proto_SetStatus(szProto, iInitialStatus, iStatus, L"");
			SaveMessageToDB(szProto, L"", TRUE);
			SaveMessageToDB(szProto, L"", FALSE);
		}
		return 1;
	}
	return 0;
}

INT_PTR SetStatusModeFromExtern(WPARAM wParam, LPARAM lParam)
{
	if ((wParam < ID_STATUS_OFFLINE && wParam != 0) || (wParam > ID_STATUS_OUTTOLUNCH && wParam != ID_STATUS_CURRENT))
		return 0;

	int newStatus = (int)wParam;

	for (int i = 0; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0) &~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
			continue;

		if (db_get_b(0, pa->szModuleName, "LockMainStatus", 0))
			continue;

		if (wParam == ID_STATUS_CURRENT || wParam == 0)
			newStatus = GetCurrentStatus(pa->szModuleName);

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND)) {
			CallProtoService(pa->szModuleName, PS_SETSTATUS, newStatus, 0);
			continue;
		}

		int status_modes_msg = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);

		if ((Proto_Status2Flag(newStatus) & status_modes_msg) || (newStatus == ID_STATUS_OFFLINE && (Proto_Status2Flag(ID_STATUS_INVISIBLE) & status_modes_msg))) {
			wchar_t *msg = nullptr;

			if (HasProtoStaticStatusMsg(pa->szModuleName, GetCurrentStatus(pa->szModuleName), newStatus))
				continue;

			if (lParam)
				msg = InsertVarsIntoMsg((wchar_t *)lParam, pa->szModuleName, newStatus, NULL);

			SaveMessageToDB(pa->szModuleName, (wchar_t *)lParam, TRUE);
			SaveMessageToDB(pa->szModuleName, msg, FALSE);
			Proto_SetStatus(pa->szModuleName, GetCurrentStatus(pa->szModuleName), newStatus, msg /*? msg : L""*/);
			mir_free(msg);
		}
		else
			CallProtoService(pa->szModuleName, PS_SETSTATUS, newStatus, 0);
	}

	return 0;
}

int ChangeStatusMessage(WPARAM wParam, LPARAM lParam);

void SetStatusMessage(const char *szProto, int iInitialStatus, int iStatus, wchar_t *message, BOOL bOnStartup)
{
	wchar_t *msg = nullptr;
#ifdef _DEBUG
	g_plugin.debugLogA("SetStatusMessage(\"%s\", %d, %d, \"%S\", %d)", szProto, iInitialStatus, iStatus, message, bOnStartup);
#endif
	if (szProto) {
		if (bOnStartup && accounts->statusCount > 1) // TODO not only at startup?
		{
			int status;
			for (int i = 0; i < accounts->count; ++i) {
				auto *pa = accounts->pa[i];
				if (!pa->IsEnabled())
					continue;

				if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
					continue;

				status = iStatus == ID_STATUS_CURRENT ? GetStartupStatus(pa->szModuleName) : iStatus;

				if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0) ||
					!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND)) {
					if (!(bOnStartup && status == ID_STATUS_OFFLINE) && GetCurrentStatus(pa->szModuleName) != status)
						CallProtoService(pa->szModuleName, PS_SETSTATUS, (WPARAM)status, 0);
				}
			}
	}

		if (message)
			msg = InsertVarsIntoMsg(message, szProto, iStatus, NULL);

		SaveMessageToDB(szProto, message, TRUE);
		SaveMessageToDB(szProto, msg, FALSE);

		if (iInitialStatus == ID_STATUS_CURRENT)
			iInitialStatus = bOnStartup ? ID_STATUS_OFFLINE : GetCurrentStatus(szProto);

		Proto_SetStatus(szProto, iInitialStatus, iStatus, msg);
		mir_free(msg);
}
	else {
		int iProfileStatus = iStatus > ID_STATUS_CURRENT ? iStatus : 0;
		BOOL bIsStatusCurrent = iStatus == ID_STATUS_CURRENT;
		BOOL bIsInitialStatusCurrent = iInitialStatus == ID_STATUS_CURRENT;

		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
				continue;

			if (!bOnStartup && db_get_b(0, pa->szModuleName, "LockMainStatus", 0))
				continue;

			if (iProfileStatus) {
				int iProfileNumber = iStatus - 40083;
				char szSetting[128];
				mir_snprintf(szSetting, "%d_%s", iProfileNumber, pa->szModuleName);
				iStatus = db_get_w(0, "StartupStatus", szSetting, ID_STATUS_OFFLINE);
				if (iStatus == ID_STATUS_IDLE) // the same as ID_STATUS_LAST in StartupStatus
				{
					mir_snprintf(szSetting, "last_%s", pa->szModuleName);
					iStatus = db_get_w(0, "StartupStatus", szSetting, ID_STATUS_OFFLINE);
				}
				else if (iStatus == ID_STATUS_CURRENT)
					iStatus = GetCurrentStatus(pa->szModuleName);
			}

			if (bIsStatusCurrent)
				iStatus = bOnStartup ? GetStartupStatus(pa->szModuleName) : GetCurrentStatus(pa->szModuleName);

			if (bIsInitialStatusCurrent)
				iInitialStatus = bOnStartup ? ID_STATUS_OFFLINE : GetCurrentStatus(pa->szModuleName);

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0) & Proto_Status2Flag(iStatus)) ||
				!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND)) {
				if (!(bOnStartup && iStatus == ID_STATUS_OFFLINE) && GetCurrentStatus(pa->szModuleName) != iStatus && iStatus != iInitialStatus) {
					CallProtoService(pa->szModuleName, PS_SETSTATUS, (WPARAM)iStatus, 0);
#ifdef _DEBUG
					g_plugin.debugLogA("SetStatusMessage(): Set %s status for %s.", StatusModeToDbSetting(iStatus, ""), pa->szModuleName);
#endif
				}
				continue;
			}

			if (HasProtoStaticStatusMsg(pa->szModuleName, iInitialStatus, iStatus))
				continue;

			if (message)
				msg = InsertVarsIntoMsg(message, pa->szModuleName, iStatus, NULL);

			SaveMessageToDB(pa->szModuleName, message, TRUE);
			SaveMessageToDB(pa->szModuleName, msg, FALSE);

			Proto_SetStatus(pa->szModuleName, iInitialStatus, iStatus, msg);
			mir_free(msg);
			}

		if (GetCurrentStatus(nullptr) != iStatus && !bIsStatusCurrent && !iProfileStatus) {
			// not so nice...
			UnhookEvent(h_statusmodechange);
			Clist_SetStatusMode(iStatus);
			h_statusmodechange = HookEvent(ME_CLIST_STATUSMODECHANGE, ChangeStatusMessage);
		}
		}
	}

INT_PTR ShowStatusMessageDialogInternal(WPARAM, LPARAM lParam)
{
	bool idvstatusmsg = false;

	if (Miranda_IsTerminated())
		return 0;

	if (hTTBButton) {
		CallService(MS_TTB_SETBUTTONSTATE, (WPARAM)hTTBButton, 0);
		CallService(MS_TTB_SETBUTTONOPTIONS, MAKEWPARAM(TTBO_TIPNAME, (LPARAM)hTTBButton), (LPARAM)Translate("Change status message"));
	}

	MsgBoxInitData *box_data = (struct MsgBoxInitData *)mir_alloc(sizeof(struct MsgBoxInitData));

	if (accounts->statusMsgCount == 1) {
		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
				continue;

			box_data->m_szProto = pa->szModuleName;
			box_data->m_iStatusModes = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0);
			box_data->m_iStatusMsgModes = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);
			break;
		}
	}
	else {
		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
				continue;

			if (!pa->bIsVisible)
				continue;

			if (hProtoStatusMenuItem[i] == (HANDLE)lParam) {
				box_data->m_szProto = pa->szModuleName;
				box_data->m_iStatusModes = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0);
				box_data->m_iStatusMsgModes = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);

				idvstatusmsg = true;
				break;
			}
		}
		if (!idvstatusmsg) {
			box_data->m_szProto = nullptr;
			box_data->m_iStatusModes = accounts->statusFlags;
			box_data->m_iStatusMsgModes = accounts->statusMsgFlags;
		}
	}
	box_data->m_iStatus = ID_STATUS_CURRENT;
	box_data->m_bOnEvent = FALSE;
	box_data->m_bOnStartup = FALSE;

	if (hwndSAMsgDialog)
		DestroyWindow(hwndSAMsgDialog);
	hwndSAMsgDialog = CreateDialogParam(g_plugin.getInst(), MAKEINTRESOURCE(IDD_AWAYMSGBOX), nullptr, AwayMsgBoxDlgProc, (LPARAM)box_data);
	return 0;
}

INT_PTR ShowStatusMessageDialog(WPARAM, LPARAM lParam)
{
	struct MsgBoxInitData *box_data;
	BOOL idvstatusmsg = FALSE;

	if (Miranda_IsTerminated()) return 0;

	box_data = (struct MsgBoxInitData *)mir_alloc(sizeof(struct MsgBoxInitData));

	for (int i = 0; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
			continue;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
			continue;

		if (!pa->bIsVisible)
			continue;

		if (!mir_strcmp(pa->szModuleName, (char *)lParam)) {
			box_data->m_szProto = pa->szModuleName;
			box_data->m_iStatusModes = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0);
			box_data->m_iStatusMsgModes = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);

			idvstatusmsg = TRUE;
			break;
		}
	}
	if (!idvstatusmsg) {
		box_data->m_szProto = nullptr;
		box_data->m_iStatusModes = accounts->statusFlags;
		box_data->m_iStatusMsgModes = accounts->statusMsgFlags;
	}
	box_data->m_iStatus = ID_STATUS_CURRENT;
	box_data->m_bOnEvent = FALSE;
	box_data->m_bOnStartup = FALSE;

	if (hwndSAMsgDialog)
		DestroyWindow(hwndSAMsgDialog);
	hwndSAMsgDialog = CreateDialogParam(g_plugin.getInst(), MAKEINTRESOURCE(IDD_AWAYMSGBOX), nullptr, AwayMsgBoxDlgProc, (LPARAM)box_data);

	return 0;
}

int ChangeStatusMessage(WPARAM wParam, LPARAM lParam)
{
	if (Miranda_IsTerminated())
		return 0;

	int iStatus = (int)wParam;
	char *szProto = (char*)lParam;

	// TODO this could be done better
	BOOL bOnStartup = FALSE, bGlobalStartupStatus = TRUE;
	if (szProto && !mir_strcmp(szProto, "SimpleStatusMsgGlobalStartupStatus")) {
		szProto = nullptr;
		bOnStartup = TRUE;
	}

	if (accounts->statusMsgCount == 1 && !szProto) {
		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
				continue;

			szProto = pa->szModuleName;
			if (bOnStartup && iStatus == ID_STATUS_CURRENT) {
				iStatus = GetStartupStatus(pa->szModuleName);
				bGlobalStartupStatus = FALSE;
			}
			break;
		}
	}

	char szSetting[80];
	mir_snprintf(szSetting, "%sFlags", szProto ? szProto : "");
	int iDlgFlags = g_plugin.getByte((char *)StatusModeToDbSetting(iStatus, szSetting), STATUS_DEFAULT);

	BOOL bShowDlg = iDlgFlags & STATUS_SHOW_DLG || bOnStartup;
	BOOL bScreenSaverRunning = IsScreenSaverRunning();

	if (szProto) {
		struct MsgBoxInitData *box_data;
		int status_modes = 0, status_modes_msg = 0, iProtoFlags;

		status_modes = CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_5, 0);
		if (!(Proto_Status2Flag(iStatus) & status_modes) && iStatus != ID_STATUS_OFFLINE)
			return 0;

		status_modes_msg = CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_3, 0);
		if (!(Proto_Status2Flag(iStatus) & status_modes_msg) || !(CallProtoService(szProto, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND)) {
			if (bOnStartup && GetCurrentStatus(szProto) != iStatus) {
				CallProtoService(szProto, PS_SETSTATUS, iStatus, 0);
#ifdef _DEBUG
				g_plugin.debugLogA("ChangeStatusMessage(): Set %s status for %s.", StatusModeToDbSetting(iStatus, ""), szProto);
#endif
			}
			return 0;
		}

		mir_snprintf(szSetting, "Proto%sFlags", szProto);
		iProtoFlags = g_plugin.getByte(szSetting, PROTO_DEFAULT);
		if (iProtoFlags & PROTO_NO_MSG || iProtoFlags & PROTO_THIS_MSG) {
			if (HasProtoStaticStatusMsg(szProto, iStatus, iStatus))
				return 1;
		}
		else if (iProtoFlags & PROTO_NOCHANGE && !bOnStartup) {
			mir_snprintf(szSetting, "FCur%sMsg", szProto);
			wchar_t *msg = g_plugin.getWStringA(szSetting);

#ifdef _DEBUG
			g_plugin.debugLogA("ChangeStatusMessage(): Set %s status and \"%S\" status message for %s.", StatusModeToDbSetting(iStatus, ""), msg, szProto);
#endif
			SetStatusMessage(szProto, iStatus, iStatus, msg, FALSE);
			if (msg) mir_free(msg);
			return 1;
		}

		if (!bShowDlg || bScreenSaverRunning) {
			wchar_t *msg = GetAwayMessageFormat(iStatus, szProto);
#ifdef _DEBUG
			g_plugin.debugLogA("ChangeStatusMessage(): Set %s status and \"%S\" status message for %s.", StatusModeToDbSetting(iStatus, ""), msg, szProto);
#endif
			SetStatusMessage(szProto, iStatus, iStatus, msg, FALSE);
			if (msg) mir_free(msg);
			return 1;
		}

		box_data = (struct MsgBoxInitData *) mir_alloc(sizeof(struct MsgBoxInitData));
		box_data->m_szProto = szProto;

		if (!bOnStartup)
			SaveStatusAsCurrent(szProto, iStatus);

		if (GetCurrentStatus(szProto) == iStatus || (bOnStartup && !bGlobalStartupStatus))
			box_data->m_iStatus = ID_STATUS_CURRENT;
		else
			box_data->m_iStatus = iStatus;

		box_data->m_iStatusModes = status_modes;
		box_data->m_iStatusMsgModes = status_modes_msg;
		box_data->m_bOnEvent = TRUE;
		box_data->m_bOnStartup = bOnStartup;

		if (hwndSAMsgDialog)
			DestroyWindow(hwndSAMsgDialog);
		hwndSAMsgDialog = CreateDialogParam(g_plugin.getInst(), MAKEINTRESOURCE(IDD_AWAYMSGBOX), nullptr, AwayMsgBoxDlgProc, (LPARAM)box_data);
		}
	else {
		// iStatus == ID_STATUS_CURRENT only when bOnStartup == TRUE
		if (iStatus == ID_STATUS_OFFLINE || (!(accounts->statusMsgFlags & Proto_Status2Flag(iStatus)) && iStatus != ID_STATUS_CURRENT))
			return 0;

		int iProtoFlags = g_plugin.getByte("ProtoFlags", PROTO_DEFAULT);
		if (!bShowDlg || bScreenSaverRunning || (iProtoFlags & PROTO_NOCHANGE && !bOnStartup)) {
			for (int i = 0; i < accounts->count; ++i) {
				auto *pa = accounts->pa[i];
				if (!pa->IsEnabled())
					continue;

				if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
					continue;

				if (db_get_b(0, pa->szModuleName, "LockMainStatus", 0))
					continue;

				if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0) & Proto_Status2Flag(iStatus)) ||
					!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
					continue;

				wchar_t *msg;
				if (iProtoFlags & PROTO_NOCHANGE) {
					mir_snprintf(szSetting, "FCur%sMsg", pa->szModuleName);
					msg = g_plugin.getWStringA(szSetting);
				}
				else
					msg = GetAwayMessageFormat(iStatus, nullptr);
#ifdef _DEBUG
				g_plugin.debugLogA("ChangeStatusMessage(): Set %s status and \"%S\" status message for %s.", StatusModeToDbSetting(iStatus, ""), msg, pa->szModuleName);
#endif
				SetStatusMessage(pa->szModuleName, iStatus, iStatus, msg, FALSE);
				if (msg)
					mir_free(msg);
			}
			return 1;
			}

		MsgBoxInitData *box_data = (MsgBoxInitData*)mir_alloc(sizeof(MsgBoxInitData));
		box_data->m_szProto = nullptr;
		box_data->m_iStatus = iStatus;
		box_data->m_iStatusModes = accounts->statusFlags;
		box_data->m_iStatusMsgModes = accounts->statusMsgFlags;
		box_data->m_bOnEvent = TRUE;
		box_data->m_bOnStartup = bOnStartup;

		if (hwndSAMsgDialog)
			DestroyWindow(hwndSAMsgDialog);
		hwndSAMsgDialog = CreateDialogParam(g_plugin.getInst(), MAKEINTRESOURCE(IDD_AWAYMSGBOX), nullptr, AwayMsgBoxDlgProc, (LPARAM)box_data);
		}
	return 0;
	}

static INT_PTR ChangeStatusMsg(WPARAM wParam, LPARAM lParam)
{
	ChangeStatusMessage(wParam, lParam);
	return 0;
}

static int ProcessProtoAck(WPARAM, LPARAM lParam)
{
	ACKDATA *ack = (ACKDATA *)lParam;
	if (!ack || !ack->szModule)
		return 0;

	if (ack->type != ACKTYPE_STATUS || ack->result != ACKRESULT_SUCCESS || ack->hContact != NULL)
		return 0;

	if (IsStatusConnecting(ack->lParam))
		ack->lParam = ID_STATUS_OFFLINE;

	SaveStatusAsCurrent(ack->szModule, (int)ack->lParam);
#ifdef _DEBUG
	g_plugin.debugLogA("ProcessProtoAck(): Set %s (%d) status for %s.", StatusModeToDbSetting((int)ack->lParam, ""), (int)ack->lParam, (char *)ack->szModule);
#endif

	return 0;
}

int SetStartupStatus(int i)
{
	char szSetting[80];
	wchar_t *fmsg, *msg = nullptr;

	auto *pa = accounts->pa[i];
	int iStatus = GetStartupStatus(pa->szModuleName);

	if (iStatus == ID_STATUS_OFFLINE)
		return -1;

	if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0) ||
		!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND)) {
		CallProtoService(pa->szModuleName, PS_SETSTATUS, (WPARAM)iStatus, 0);
		return -1;
	}

	mir_snprintf(szSetting, "Proto%sFlags", pa->szModuleName);
	int flags = g_plugin.getByte(szSetting, PROTO_DEFAULT);
	if (flags & PROTO_NO_MSG || flags & PROTO_THIS_MSG) {
		if (HasProtoStaticStatusMsg(pa->szModuleName, ID_STATUS_OFFLINE, iStatus))
			return 0;
		else
			fmsg = nullptr;
	}
	else if (flags & PROTO_NOCHANGE) {
		mir_snprintf(szSetting, "FCur%sMsg", pa->szModuleName);
		fmsg = g_plugin.getWStringA(szSetting);
	}
	else
		fmsg = GetAwayMessageFormat(iStatus, pa->szModuleName);

#ifdef _DEBUG
	g_plugin.debugLogA("SetStartupStatus(): Set %s status and \"%S\" status message for %s.", StatusModeToDbSetting(iStatus, ""), fmsg, pa->szModuleName);
#endif

	if (fmsg)
		msg = InsertVarsIntoMsg(fmsg, pa->szModuleName, iStatus, NULL);

	SaveMessageToDB(pa->szModuleName, fmsg, TRUE);
	SaveMessageToDB(pa->szModuleName, msg, FALSE);

	if (fmsg)
		mir_free(fmsg);

	Proto_SetStatus(pa->szModuleName, ID_STATUS_OFFLINE, iStatus, msg /*? msg : L""*/);
	mir_free(msg);

	return 0;
}

VOID CALLBACK SetStartupStatusGlobal(HWND hwnd, UINT, UINT_PTR idEvent, DWORD)
{
	int prev_status_mode = -1, status_mode = 0, temp_status_mode = ID_STATUS_OFFLINE;
	bool globalstatus = true;

	KillTimer(hwnd, idEvent);

	// is global status mode going to be set?
	for (int i = 0; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
			continue;

		status_mode = GetStartupStatus(pa->szModuleName);

		if (status_mode != ID_STATUS_OFFLINE)
			temp_status_mode = status_mode;

		if (status_mode != prev_status_mode && prev_status_mode != -1) {
			globalstatus = false;
			break;
		}

		prev_status_mode = status_mode;
	}

	// popup status msg dialog at startup?
	if (g_plugin.getByte("StartupPopupDlg", 1) && accounts->statusMsgFlags) {
		if (globalstatus)
			ChangeStatusMessage(status_mode, (LPARAM)"SimpleStatusMsgGlobalStartupStatus");
		else {
			// pseudo-currentDesiredStatusMode ;-)
			g_plugin.setWord("StartupStatus", (WORD)temp_status_mode);
			ChangeStatusMessage(ID_STATUS_CURRENT, (LPARAM)"SimpleStatusMsgGlobalStartupStatus");
		}
		return;
	}

	for (int i = 0; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
			continue;

		SetStartupStatus(i);
	}
}

VOID CALLBACK SetStartupStatusProc(HWND hwnd, UINT, UINT_PTR idEvent, DWORD)
{
	bool found = false;
	int i = 0;

	for (; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0)&~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
			continue;

		if (g_uSetStatusTimer[i] == idEvent) {
			KillTimer(nullptr, g_uSetStatusTimer[i]);
			found = true;
			break;
		}
	}

	if (found)
		SetStartupStatus(i);
	else
		KillTimer(hwnd, idEvent);

}

VOID CALLBACK UpdateMsgTimerProc(HWND, UINT, UINT_PTR, DWORD)
{
	MIRANDA_IDLE_INFO mii;
	Idle_GetInfo(mii);
	if (g_plugin.getByte("NoUpdateOnIdle", 1) && mii.idleType)
		return;

	if (!hwndSAMsgDialog) {
		char szBuffer[64];
		wchar_t *tszMsg;
		int iCurrentStatus;

		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
				continue;

			iCurrentStatus = Proto_GetStatus(pa->szModuleName);
			if (iCurrentStatus < ID_STATUS_ONLINE)
				continue;

			mir_snprintf(szBuffer, "FCur%sMsg", pa->szModuleName);
			wchar_t *tszStatusMsg = g_plugin.getWStringA(szBuffer);
			if (tszStatusMsg == nullptr)
				continue;

			tszMsg = InsertVarsIntoMsg(tszStatusMsg, pa->szModuleName, iCurrentStatus, NULL);
			mir_free(tszStatusMsg);

			mir_snprintf(szBuffer, "Cur%sMsg", pa->szModuleName);

			tszStatusMsg = g_plugin.getWStringA(szBuffer);
			if ((tszMsg && tszStatusMsg && !mir_wstrcmp(tszMsg, tszStatusMsg)) || (!tszMsg && !tszStatusMsg)) {
				mir_free(tszStatusMsg);
				mir_free(tszMsg);
				continue;
			}
			else
				mir_free(tszStatusMsg);

			if (tszMsg && mir_wstrlen(tszMsg)) {
#ifdef _DEBUG
				g_plugin.debugLogA("UpdateMsgTimerProc(): Set %s status and \"%S\" status message for %s.", StatusModeToDbSetting(iCurrentStatus, ""), tszMsg, pa->szModuleName);
#endif
				Proto_SetStatus(pa->szModuleName, iCurrentStatus, iCurrentStatus, tszMsg);
				SaveMessageToDB(pa->szModuleName, tszMsg, FALSE);
			}
			mir_free(tszMsg);
			}
		}
	}

static int AddTopToolbarButton(WPARAM, LPARAM)
{
	TTBButton tbb = {};
	tbb.hIconHandleUp = tbb.hIconHandleDn = g_plugin.getIconHandle(IDI_CSMSG);
	tbb.pszService = MS_SIMPLESTATUSMSG_SHOWDIALOGINT;
	tbb.dwFlags = TTBBF_VISIBLE | TTBBF_SHOWTOOLTIP;
	tbb.name = tbb.pszTooltipUp = LPGEN("Change status message");
	hTTBButton = g_plugin.addTTB(&tbb);

	g_plugin.releaseIcon(IDI_CSMSG);
	return 0;
}

void RegisterHotkey(void)
{
	HOTKEYDESC hkd = {};
	hkd.dwFlags = HKD_UNICODE;
	hkd.pszName = "SimpleStatusMsg_OpenDialog";
	hkd.szDescription.w = LPGENW("Open status message dialog");
	hkd.szSection.w = LPGENW("Status message");
	hkd.pszService = MS_SIMPLESTATUSMSG_SHOWDIALOGINT;
	hkd.DefHotKey = HOTKEYCODE(HOTKEYF_CONTROL, VK_OEM_3);
	g_plugin.addHotkey(&hkd);
}

static int ChangeStatusMsgPrebuild(WPARAM, LPARAM)
{
	int iStatusMenuItemCount = 0;
	DWORD iStatusMsgFlags = 0;

	auto &accs = Accounts();
	hProtoStatusMenuItem = (HANDLE *)mir_realloc(hProtoStatusMenuItem, sizeof(HANDLE) * accs.getCount());
	for (auto &pa : accs) {
		if (!pa->IsEnabled())
			continue;

		if (CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND)
			iStatusMsgFlags |= CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);

		if (!pa->bIsVisible)
			continue;

		iStatusMenuItemCount++;
	}

	if (!iStatusMsgFlags || !iStatusMenuItemCount)
		return 0;

	CMenuItem mi(&g_plugin);
	mi.flags = CMIF_UNICODE;
	if (!g_plugin.getByte("ShowStatusMenuItem", 1))
		mi.flags |= CMIF_HIDDEN;
	mi.hIcolibItem = g_plugin.getIconHandle(IDI_CSMSG);
	mi.pszService = MS_SIMPLESTATUSMSG_SHOWDIALOGINT;
	mi.name.w = LPGENW("Status message...");
	mi.position = 2000200000;
	Menu_AddStatusMenuItem(&mi);

	// mi.popupPosition = 500084000; !!!!!!!!!!!!!!!!!!!!!!!
	mi.position = 2000040000;

	int i = 0;
	for (auto &pa : accs) {
		if (!pa->IsEnabled())
			continue;

		if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
			continue;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
			continue;

		if (!pa->bIsVisible)
			continue;

		char szSetting[80];
		mir_snprintf(szSetting, "Proto%sFlags", pa->szModuleName);
		int iProtoFlags = g_plugin.getByte(szSetting, PROTO_DEFAULT);
		if (iProtoFlags & PROTO_NO_MSG || iProtoFlags & PROTO_THIS_MSG)
			continue;

		if (pa->IsLocked()) {
			wchar_t szBuffer[256];
			mir_snwprintf(szBuffer, TranslateT("%s (locked)"), pa->tszAccountName);
			mi.root = g_plugin.addRootMenu(MO_STATUS, szBuffer, mi.position);
		}
		else mi.root = g_plugin.addRootMenu(MO_STATUS, pa->tszAccountName, mi.position);

		hProtoStatusMenuItem[i++] = Menu_AddStatusMenuItem(&mi);
	}

	return 0;
}

static int OnIdleChanged(WPARAM, LPARAM lParam)
{
#ifdef _DEBUG
	g_plugin.debugLogA("OnIdleChanged()");
#endif
	if (!(lParam & IDF_ISIDLE))
		g_iIdleTime = -1;

	MIRANDA_IDLE_INFO mii;
	Idle_GetInfo(mii);

	if (mii.aaStatus == 0) {
#ifdef _DEBUG
		g_plugin.debugLogA("OnIdleChanged(): AutoAway disabled");
#endif
		return 0;
	}

	for (int i = 0; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		if (db_get_b(0, pa->szModuleName, "LockMainStatus", 0))
			continue;

		int iStatusBits = CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);
		int iStatus = mii.aaStatus;
		if (!(iStatusBits & Proto_Status2Flag(iStatus))) {
			if (iStatusBits & Proto_Status2Flag(ID_STATUS_AWAY))
				iStatus = ID_STATUS_AWAY;
			else
				continue;
		}

		int iCurrentStatus = Proto_GetStatus(pa->szModuleName);
		if (iCurrentStatus < ID_STATUS_ONLINE || iCurrentStatus == ID_STATUS_INVISIBLE)
			continue;

		// we're entering idle
		if (lParam & IDF_ISIDLE) {
			if (!db_get_b(0, "AutoAway", pa->szModuleName, 0) && iCurrentStatus != ID_STATUS_ONLINE && iCurrentStatus != ID_STATUS_FREECHAT)
				return 0;
		}
		else {
			// if new status shall not be changed, do nothing, else switch the status to ONLINE
			if (mii.aaLock)
				return 0;
			iStatus = ID_STATUS_ONLINE;
		}
			
		ptrW tszMsg(GetAwayMessage(iStatus, pa->szModuleName, FALSE, 0));
		SaveMessageToDB(pa->szModuleName, tszMsg, TRUE);
		SaveMessageToDB(pa->szModuleName, ptrW(InsertVarsIntoMsg(tszMsg, pa->szModuleName, iStatus, 0)), FALSE);
	}

	return 0;
}

static int CSStatusChange(WPARAM wParam, LPARAM iCount)
{
	PROTOCOLSETTINGEX **ps = *(PROTOCOLSETTINGEX***)wParam;
	if (ps == nullptr)
		return -1;

	int status_mode;
	char szSetting[80];

	for (int i = 0; i < iCount; ++i) {
		if (ps[i]->m_szName == nullptr || !*ps[i]->m_szName) continue;
		if (ps[i]->m_status == ID_STATUS_IDLE)
			status_mode = ps[i]->m_lastStatus;
		else if (ps[i]->m_status == ID_STATUS_CURRENT)
			status_mode = Proto_GetStatus(ps[i]->m_szName);
		else
			status_mode = ps[i]->m_status;

		SaveStatusAsCurrent(ps[i]->m_szName, status_mode);
#ifdef _DEBUG
		g_plugin.debugLogA("CSStatusChange(): Set %s status for %s.", StatusModeToDbSetting(status_mode, ""), ps[i]->m_szName);
#endif

		// TODO SaveMessageToDB also when NULL?
		if (ps[i]->m_szMsg) {
			char buff[80];
			bool found = false;

#ifdef _DEBUG
			g_plugin.debugLogA("CSStatusChange(): Set \"%s\" status message for %s.", ps[i]->m_szMsg, ps[i]->m_szName);
#endif
			int max_hist_msgs = g_plugin.getByte("MaxHist", 10);
			for (int j = 1; j <= max_hist_msgs; j++) {
				mir_snprintf(buff, "SMsg%d", j);
				ptrW tszStatusMsg(g_plugin.getWStringA(buff));
				if (tszStatusMsg != nullptr) {
					if (!mir_wstrcmp(tszStatusMsg, ps[i]->m_szMsg)) {
						found = true;
						mir_snprintf(szSetting, "Last%sMsg", ps[i]->m_szName);
						g_plugin.setString(szSetting, buff);
						break;
					}
				}
			}

			if (!found) {
				mir_snprintf(buff, "FCur%sMsg", ps[i]->m_szName);
				mir_snprintf(szSetting, "Last%sMsg", ps[i]->m_szName);
				g_plugin.setString(szSetting, buff);
			}

			mir_snprintf(szSetting, "%sMsg", ps[i]->m_szName);

			db_set_ws(0, "SRAway", StatusModeToDbSetting(status_mode, szSetting), ps[i]->m_szMsg);
			SaveMessageToDB(ps[i]->m_szName, ps[i]->m_szMsg, TRUE);
			SaveMessageToDB(ps[i]->m_szName, ptrW(InsertVarsIntoMsg(ps[i]->m_szMsg, ps[i]->m_szName, status_mode, 0)), FALSE);
		}
	}

	return 0;
}

static wchar_t* ParseWinampSong(ARGUMENTSINFO *ai)
{
	if (ai->argc != 1)
		return nullptr;

	ai->flags |= AIF_DONTPARSE;
	wchar_t *ptszWinampTitle = GetWinampSong();

	if (ptszWinampTitle != nullptr) {
		mir_free(g_ptszWinampSong);
		g_ptszWinampSong = mir_wstrdup(ptszWinampTitle);
	}
	else if (g_ptszWinampSong && mir_wstrcmp(g_ptszWinampSong, TEXT(MODULENAME)) && g_plugin.getByte("AmpLeaveTitle", 1))
		ptszWinampTitle = mir_wstrdup(g_ptszWinampSong);

	return ptszWinampTitle;
}

static wchar_t* ParseDate(ARGUMENTSINFO *ai)
{
	if (ai->argc != 1)
		return nullptr;

	wchar_t szStr[128] = { 0 };
	ai->flags |= AIF_DONTPARSE;
	GetDateFormat(LOCALE_USER_DEFAULT, DATE_SHORTDATE, nullptr, nullptr, szStr, _countof(szStr));

	return mir_wstrdup(szStr);
}

static int OnAccListChanged(WPARAM, LPARAM)
{
#ifdef _DEBUG
	g_plugin.debugLogA("OnAccListChanged()");
#endif
	accounts->statusFlags = 0;
	accounts->statusCount = 0;
	accounts->statusMsgFlags = 0;
	accounts->statusMsgCount = 0;
	UnhookProtoEvents();

	Proto_EnumAccounts(&accounts->count, &accounts->pa);
	for (int i = 0; i < accounts->count; ++i) {
		auto *pa = accounts->pa[i];
		if (!pa->IsEnabled())
			continue;

		accounts->statusFlags |= (CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0) & ~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0));

		if (CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0) & ~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0))
			accounts->statusCount++;

		if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_1, 0) & PF1_MODEMSGSEND))
			continue;

		accounts->statusMsgFlags |= CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0);

		if (!CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_3, 0))
			continue;

		accounts->statusMsgCount++;
	}

	return 0;
}

static int OnModulesLoaded(WPARAM, LPARAM)
{
#ifdef _DEBUG
	g_plugin.debugLogA("### Session started ###");
#endif
	OnAccListChanged(0, 0);

	LoadAwayMsgModule();

	HookEvent(ME_TTB_MODULELOADED, AddTopToolbarButton);

	RegisterHotkey();

	h_statusmodechange = HookEvent(ME_CLIST_STATUSMODECHANGE, ChangeStatusMessage);
	HookEvent(ME_PROTO_ACK, ProcessProtoAck);
	HookEvent(ME_IDLE_CHANGED, OnIdleChanged);

	HookEvent(ME_CLIST_PREBUILDSTATUSMENU, ChangeStatusMsgPrebuild);
	ChangeStatusMsgPrebuild(0, 0);

	if (ServiceExists(MS_VARS_REGISTERTOKEN)) {
		TOKENREGISTER tr = { 0 };
		tr.cbSize = sizeof(TOKENREGISTER);
		tr.memType = TR_MEM_MIRANDA;
		tr.flags = TRF_FREEMEM | TRF_FIELD | TRF_TCHAR | TRF_PARSEFUNC;
		tr.szTokenString.w = L"winampsong";
		tr.parseFunctionW = ParseWinampSong;
		tr.szHelpText = LPGEN("External Applications") "\t" LPGEN("retrieves song name of the song currently playing in Winamp (Simple Status Message compatible)");
		CallService(MS_VARS_REGISTERTOKEN, 0, (LPARAM)&tr);

		if (g_plugin.getByte("ExclDateToken", 0) != 0) {
			tr.szTokenString.w = L"date";
			tr.parseFunctionW = ParseDate;
			tr.szHelpText = LPGEN("Miranda Related") "\t" LPGEN("get the date (Simple Status Message compatible)");
			CallService(MS_VARS_REGISTERTOKEN, 0, (LPARAM)&tr);
		}
	}

	g_ptszWinampSong = g_plugin.getWStringA("AmpLastTitle");
	if (g_ptszWinampSong == nullptr)
		g_ptszWinampSong = mir_wstrdup(TEXT(MODULENAME));

	if (g_plugin.getByte("UpdateMsgOn", 1))
		g_uUpdateMsgTimer = SetTimer(nullptr, 0, g_plugin.getWord("UpdateMsgInt", 10) * 1000, UpdateMsgTimerProc);

	HookEvent(ME_CS_STATUSCHANGEEX, CSStatusChange);

	if (accounts->statusCount == 0)
		return 0;

	if (!ServiceExists(MS_SS_GETPROFILECOUNT)) {
		if (g_plugin.getByte("GlobalStatusDelay", 1))
			SetTimer(nullptr, 0, g_plugin.getWord("SetStatusDelay", 300), SetStartupStatusGlobal);
		else {
			g_uSetStatusTimer = (UINT_PTR *)mir_alloc(sizeof(UINT_PTR) * accounts->count);
			for (int i = 0; i < accounts->count; ++i) {
				auto *pa = accounts->pa[i];
				if (!pa->IsEnabled())
					continue;

				if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0) & ~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
					continue;

				char szSetting[80];
				mir_snprintf(szSetting, "Set%sStatusDelay", pa->szModuleName);
				g_uSetStatusTimer[i] = SetTimer(nullptr, 0, g_plugin.getWord(szSetting, 300), SetStartupStatusProc);
			}
		}
	}

	return 0;
}

static int OnOkToExit(WPARAM, LPARAM)
{
	if (accounts->statusCount) {
		char szSetting[80];

		for (int i = 0; i < accounts->count; ++i) {
			auto *pa = accounts->pa[i];
			if (!pa->IsEnabled())
				continue;

			if (!(CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_2, 0) & ~CallProtoService(pa->szModuleName, PS_GETCAPS, PFLAGNUM_5, 0)))
				continue;

			mir_snprintf(szSetting, "Last%sStatus", pa->szModuleName);
			g_plugin.setWord(szSetting, Proto_GetStatus(pa->szModuleName));
		}

		if (g_ptszWinampSong && mir_wstrcmp(g_ptszWinampSong, TEXT(MODULENAME)) /*&& g_plugin.getByte("AmpLeaveTitle", 1)*/)
			DBWriteMessage("AmpLastTitle", g_ptszWinampSong);
	}

	return 0;
}

static int OnPreShutdown(WPARAM, LPARAM)
{
	if (!accounts->statusMsgFlags)
		return 0;

	AwayMsgPreShutdown();

	if (hwndSAMsgDialog)
		DestroyWindow(hwndSAMsgDialog);
	mir_free(hProtoStatusMenuItem);
	mir_free(g_uSetStatusTimer);
	mir_free(g_ptszWinampSong);
	if (g_uUpdateMsgTimer)
		KillTimer(nullptr, g_uUpdateMsgTimer);
	return 0;
}

// remember to mir_free() the return value
static INT_PTR sttGetAwayMessageT(WPARAM wParam, LPARAM lParam)
{
	return (INT_PTR)GetAwayMessage((int)wParam, (char *)lParam, TRUE, NULL);
}

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

int CMPlugin::Load()
{
	hwndSAMsgDialog = nullptr;
	accounts = (PROTOACCOUNTS *)mir_alloc(sizeof(PROTOACCOUNTS));

	db_set_w(0, "CList", "Status", (WORD)ID_STATUS_OFFLINE);

	HookEvent(ME_OPT_INITIALISE, InitOptions);
	HookEvent(ME_SYSTEM_MODULESLOADED, OnModulesLoaded);
	HookEvent(ME_PROTO_ACCLISTCHANGED, OnAccListChanged);

	CreateServiceFunction(MS_AWAYMSG_GETSTATUSMSGW, sttGetAwayMessageT);

	CreateServiceFunction(MS_SIMPLESTATUSMSG_SETSTATUS, SetStatusModeFromExtern);
	CreateServiceFunction(MS_SIMPLESTATUSMSG_SHOWDIALOG, ShowStatusMessageDialog);
	CreateServiceFunction(MS_SIMPLESTATUSMSG_CHANGESTATUSMSG, ChangeStatusMsg);
	CreateServiceFunction(MS_SIMPLESTATUSMSG_SHOWDIALOGINT, ShowStatusMessageDialogInternal); // internal use ONLY

	HookEvent(ME_SYSTEM_OKTOEXIT, OnOkToExit);
	HookEvent(ME_SYSTEM_PRESHUTDOWN, OnPreShutdown);

	IconsInit();
	return 0;
}

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

int CMPlugin::Unload()
{
	UnhookEvent(h_statusmodechange);
	UnhookProtoEvents();

	mir_free(accounts);

#ifdef _DEBUG
	g_plugin.debugLogA("### Session ended ###");
#endif

	return 0;
}