#include "stdafx.h"

static HGENMENU hStopItem, hStartItem;

////////////////////////////////
///////// Menu Services ////////
///////////////////////////////
int StartOTR(MCONTACT hContact)
{
	const char *proto = GetContactProto(hContact);
	if (!proto) return 1; // error
	char *uname = contact_get_id(hContact);
	if (!uname) return 1; // error
	DWORD pol = db_get_dw(hContact, MODULENAME, "Policy", CONTACT_DEFAULT_POLICY);
	if (pol == CONTACT_DEFAULT_POLICY) pol = options.default_policy;

	lib_cs_lock();
#ifndef MIROTR_PROTO_HELLO_MSG
	char *msg = otrl_proto_default_query_msg(MODULENAME, pol);
	otr_gui_inject_message((void*)hContact, proto, proto, uname, msg ? msg : MIROTR_PROTO_HELLO);
	free(msg);
#else
	wchar_t* nick = ProtoGetNickname(proto);
	if (nick) {
		wchar_t msg[1024];
		wchar_t* msgend = msg + _countof(msg) - 1;
		wchar_t* msgoff = msg;
		for (const char* c = MIROTR_PROTO_HELLO; *c; *msgoff++ = *c++);
		*msgoff++ = '\n';
		for (const wchar_t* c = nick; *c && msgoff < msgend; *msgoff++ = *c++);
		for (const wchar_t* c = MIROTR_PROTO_HELLO_MSG; *c && msgoff < msgend; *msgoff++ = *c++);
		LCID langid = Langpack_GetDefaultLocale();
		if (langid != 0x0409/*US*/ && langid != 0x1009/*CA*/ && langid != 0x0809/*GB*/) { // non english
			const wchar_t* translated = TranslateW(MIROTR_PROTO_HELLO_MSG);
			if (mir_wstrcmp(MIROTR_PROTO_HELLO_MSG, translated)) {
				*msgoff++ = '\n';
				for (const wchar_t* c = nick; *c && msgoff < msgend; *msgoff++ = *c++);
				for (const wchar_t* c = translated; *c && msgoff < msgend; *msgoff++ = *c++);
			}
		}
		*msgoff = '\0';
		mir_free(nick);

		T2Utf msg_utf8(msg);
		otr_gui_inject_message((void*)hContact, proto, proto, uname, msg_utf8 ? msg_utf8 : MIROTR_PROTO_HELLO);
	}
#endif
	mir_free(uname);
	return 0;
}

INT_PTR SVC_StartOTR(WPARAM hContact, LPARAM)
{
	wchar_t buff[512];

	MCONTACT hSub = db_mc_getMostOnline(hContact);
	if (hSub != 0)
		hContact = hSub;

	if (options.bHaveSecureIM && CallService("SecureIM/IsContactSecured", hContact, 0) != 0) {
		mir_snwprintf(buff, TranslateW(LANG_OTR_SECUREIM_STARTED), contact_get_nameT(hContact));
		ShowError(buff);
		return 0;
	}

	mir_snwprintf(buff, TranslateW(LANG_SESSION_REQUEST_OTR), contact_get_nameT(hContact));
	ShowMessage(hContact, buff);

	return StartOTR(hContact);
}

INT_PTR SVC_RefreshOTR(WPARAM hContact, LPARAM)
{
	wchar_t buff[512];

	MCONTACT hSub = db_mc_getMostOnline(hContact);
	if (hSub != 0)
		hContact = hSub;

	if (options.bHaveSecureIM && CallService("SecureIM/IsContactSecured", hContact, 0) != 0) {
		mir_snwprintf(buff, 512, TranslateW(LANG_OTR_SECUREIM_STARTED), contact_get_nameT(hContact));
		ShowError(buff);
		return 0;
	}

	mir_snwprintf(buff, TranslateW(LANG_SESSION_TRY_CONTINUE_OTR), contact_get_nameT(hContact));
	ShowMessage(hContact, buff);

	int res = StartOTR(hContact);
	if (res) return res;

	return 0;
}

int otr_disconnect_contact(MCONTACT hContact)
{
	MCONTACT hSub = db_mc_getMostOnline(hContact);
	if (hSub != 0)
		hContact = hSub;

	const char *proto = GetContactProto(hContact);
	if (!proto) return 1; // error
	char *uname = contact_get_id(hContact);
	if (!uname) return 1; // error

	lib_cs_lock();
	otrl_message_disconnect_all_instances(otr_user_state, &ops, (void*)hContact, proto, proto, uname);
	mir_free(uname);
	return 0;
}

INT_PTR SVC_StopOTR(WPARAM hContact, LPARAM)
{
	// prevent this filter from acting on injeceted messages for metas, when they are passed though the subcontact's proto send chain
	if (otr_disconnect_contact(hContact)) return 0;

	SetEncryptionStatus(hContact, TRUST_NOT_PRIVATE);

	wchar_t buff[512];
	mir_snwprintf(buff, TranslateW(LANG_SESSION_TERMINATED_OTR), contact_get_nameT(hContact));
	ShowMessage(hContact, buff);
	return 0;
}

INT_PTR SVC_VerifyOTR(WPARAM hContact, LPARAM)
{
	MCONTACT hSub = db_mc_getMostOnline(hContact);
	if (hSub != 0)
		hContact = hSub;

	ConnContext *context = otrl_context_find_miranda(otr_user_state, hContact);
	if (!context)
		return 1;

	//VerifyContextDialog(context);	
	SMPInitDialog(context);
	return 0;
}

INT_PTR SVC_ToggleHTMLOTR(WPARAM hContact, LPARAM)
{
	MCONTACT hSub = db_mc_getMostOnline(hContact);
	if (hSub != 0)
		hContact = hSub;

	if (db_get_b(hContact, MODULENAME, "HTMLConv", 0))
		db_set_b(hContact, MODULENAME, "HTMLConv", 0);
	else
		db_set_b(hContact, MODULENAME, "HTMLConv", 1);

	return 0;
}

void InitMenu()
{
	CreateServiceFunction(MS_OTR_MENUSTART, SVC_StartOTR);
	CreateServiceFunction(MS_OTR_MENUSTOP, SVC_StopOTR);
	CreateServiceFunction(MS_OTR_MENUREFRESH, SVC_RefreshOTR);
	CreateServiceFunction(MS_OTR_MENUVERIFY, SVC_VerifyOTR);
	CreateServiceFunction(MS_OTR_MENUTOGGLEHTML, SVC_ToggleHTMLOTR);

	CMenuItem mi(&g_plugin);
	mi.flags = CMIF_NOTOFFLINE | CMIF_UNICODE;
	mi.position = -400000;

	SET_UID(mi, 0xAB574FAD, 0x15D8, 0x49FF, 0xB7, 0x03, 0xDA, 0x2B, 0x45, 0x46, 0xC3, 0x56);
	mi.name.w = _A2W(LANG_STOP_OTR);
	mi.pszService = MS_OTR_MENUSTOP;
	mi.hIcolibItem = IcoLib_GetIconHandle(ICON_PRIVATE);
	hStopItem = Menu_AddContactMenuItem(&mi);

	mi.uid.d[7]++;
	mi.name.w = _A2W(LANG_START_OTR);
	mi.pszService = MS_OTR_MENUSTART;
	mi.hIcolibItem = IcoLib_GetIconHandle(ICON_NOT_PRIVATE);
	hStartItem = Menu_AddContactMenuItem(&mi);

	HookEvent(ME_CLIST_PREBUILDCONTACTMENU, SVC_PrebuildContactMenu);
}

int SVC_PrebuildContactMenu(WPARAM hContact, LPARAM)
{
	Menu_ShowItem(hStartItem, false);
	Menu_ShowItem(hStopItem, false);

	const char *proto = GetContactProto(hContact);
	if (!proto || db_get_b(hContact, proto, "ChatRoom", 0) == 1)
		return 0;

	if (proto && mir_strcmp(proto, META_PROTO) == 0) {
		// make menu act as per most online subcontact
		hContact = db_mc_getMostOnline(hContact);
		if (!hContact)
			return 0;
		proto = GetContactProto(hContact);
	}

	INT_PTR caps = CallProtoService(proto, PS_GETCAPS, PFLAGNUM_1, 0);
	if ((caps & PF1_IM) == 0)
		return 0;

	DWORD pol = db_get_dw(hContact, MODULENAME, "Policy", CONTACT_DEFAULT_POLICY);
	if (pol == CONTACT_DEFAULT_POLICY)
		pol = options.default_policy;

	if (pol == OTRL_POLICY_NEVER || pol == OTRL_POLICY_ALWAYS)
		return 0;

	ConnContext *context = otrl_context_find_miranda(otr_user_state, hContact);
	TrustLevel encrypted = otr_context_get_trust(context);
	Menu_ShowItem(hStartItem, encrypted == TRUST_NOT_PRIVATE);
	Menu_ShowItem(hStopItem, encrypted != TRUST_NOT_PRIVATE);
	return 0;
}