/*
Miranda Crash Dumper Plugin
Copyright (C) 2008 - 2012 Boris Krasnovskiy All Rights Reserved

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 version 2
of the License.

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, see <http://www.gnu.org/licenses/>.
*/

#include "utils.h"
#include <m_options.h>
#include "m_folders.h"
#include "m_toolbar.h"
#include "version.h"

MM_INTERFACE  mmi;
MD5_INTERFACE md5i;
UTF8_INTERFACE utfi;

int hLangpack;

HINSTANCE hInst;
DWORD mirandaVersion;
LCID packlcid;
HANDLE hCrashLogFolder, hVerInfoFolder;
HMODULE hRichModule;

TCHAR* vertxt;
TCHAR* profname;
TCHAR* profpath;
TCHAR* mirpath;

TCHAR CrashLogFolder[MAX_PATH];
TCHAR VersionInfoFolder[MAX_PATH];

bool servicemode;
bool clsdates;
bool dtsubfldr;

static const PLUGININFOEX pluginInfoEx = 
{
	sizeof(PLUGININFOEX),
	"Crash Dumper",
	__VERSION_DWORD,
	"Crash Dumper for Miranda IM.",
	"borkra",
	"borkra@miranda-im.org",
	"Copyright© 2008 - 2012 Boris Krasnovskiy All Rights Reserved",
	"http://code.google.com/p/mirandaimplugins/downloads/list",
	UNICODE_AWARE,		//not transient
	0,		//doesn't replace anything built-in
	// {F62C1D7A-FFA4-4065-A251-4C9DD9101CC8}
	{ 0xf62c1d7a, 0xffa4, 0x4065, { 0xa2, 0x51, 0x4c, 0x9d, 0xd9, 0x10, 0x1c, 0xc8 } }
};

const PLUGININFOEX* GetPluginInfoEx(void) { return &pluginInfoEx; }


extern "C" __declspec(dllexport) PLUGININFOEX* MirandaPluginInfoEx(DWORD mirVersion) 
{
	::mirandaVersion = mirVersion;
	return (PLUGININFOEX*)&pluginInfoEx;
}

#define MIID_CRASHDUMP  { 0x36a31a50, 0xcb55, 0x46d0, { 0xab, 0x9c, 0x1e, 0xac, 0xfb, 0x24, 0x0, 0x2a } }

// MirandaPluginInterfaces - returns the protocol interface to the core
static const MUUID interfaces[] = { MIID_SERVICEMODE, MIID_CRASHDUMP, MIID_LAST };
extern "C" __declspec(dllexport) const MUUID* MirandaPluginInterfaces(void)
{
	return interfaces;
}

PLUGINLINK *pluginLink;

HANDLE hHooks[5];
HANDLE hServices[6];


INT_PTR StoreVersionInfoToFile(WPARAM, LPARAM lParam)
{
	CreateDirectoryTree(VersionInfoFolder);

	TCHAR path[MAX_PATH];
	crs_sntprintf(path, MAX_PATH, TEXT("%s\\VersionInfo.txt"), VersionInfoFolder);

	HANDLE hDumpFile = CreateFile(path, GENERIC_WRITE, 0, NULL,
		CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

	if (hDumpFile != INVALID_HANDLE_VALUE) 
	{
		bkstring buffer;

		buffer.reserve(0x1800);
		PrintVersionInfo(buffer, (unsigned int)lParam | VI_FLAG_PRNVAR);

		char* bufu = mir_utf8encodeT(buffer.c_str());
		WriteUtfFile(hDumpFile, bufu);
		mir_free(bufu);

		CloseHandle(hDumpFile);

		ShowMessage(3, TranslateT("VersionInfo stored into file %s"), path);
	}
	else
		ShowMessage(2, TranslateT("VersionInfo file %s is inaccessible"), path);

	return 0;
}

INT_PTR StoreVersionInfoToClipboard(WPARAM, LPARAM lParam)
{
	bkstring buffer;
	buffer.reserve(0x1800);

	WriteBBFile(buffer, true);
	PrintVersionInfo(buffer, (unsigned int)lParam | VI_FLAG_PRNVAR | VI_FLAG_FORMAT);
	WriteBBFile(buffer, false);

	StoreStringToClip(buffer);

	return 0;
}

INT_PTR UploadVersionInfo(WPARAM, LPARAM lParam)
{
	bkstring buffer;
	buffer.reserve(0x1800);
	PrintVersionInfo(buffer);

	VerTrnsfr *trn = (VerTrnsfr*)mir_alloc(sizeof(VerTrnsfr));
	trn->buf = mir_utf8encodeT(buffer.c_str());
	trn->autot = lParam == 0xa1;

	mir_forkthread(VersionInfoUploadThread, trn);

	return 0;
}


INT_PTR ViewVersionInfo(WPARAM wParam, LPARAM)
{
	//	unsigned *p = (unsigned*)0x15;
	//	*p = 324;

	if (hRichModule == NULL && GetModuleHandle(TEXT("Riched20.dll")) == NULL)
		hRichModule = LoadLibrary(TEXT("Riched20.dll"));

	CreateDialogParam(hInst, MAKEINTRESOURCE(IDD_VIEWVERSION), NULL, 
		DlgProcView, wParam ? (VI_FLAG_PRNVAR | VI_FLAG_PRNDLL) : VI_FLAG_PRNVAR);

	return 0;
}

INT_PTR OpenUrl(WPARAM wParam, LPARAM)
{
	switch (wParam)
	{
	case 0:
		ShellExecute(NULL, TEXT("explore"), CrashLogFolder, NULL, NULL, SW_SHOW);
		break;

	case 1:
		OpenAuthUrl("http://%s.miranda-vi.org");
		break;
	}
	return 0;
}

INT_PTR ServiceModeLaunch(WPARAM, LPARAM)
{
	servicemode = true;
	return 0;
}


static int FoldersPathChanged(WPARAM, LPARAM)
{
	FOLDERSGETDATA fgd = {0};
	fgd.cbSize = sizeof(FOLDERSGETDATA);
	fgd.nMaxPathSize = MAX_PATH;
	fgd.szPathT = CrashLogFolder;
	CallService(MS_FOLDERS_GET_PATH, (WPARAM) hCrashLogFolder, (LPARAM) &fgd);

	fgd.szPathT = VersionInfoFolder;
	CallService(MS_FOLDERS_GET_PATH, (WPARAM) hVerInfoFolder, (LPARAM) &fgd);
	return 0;
}

int OptionsInit(WPARAM wParam, LPARAM)
{
	OPTIONSDIALOGPAGE odp = { 0 };

	odp.cbSize      = sizeof(odp);
	odp.position    = -790000000;
	odp.hInstance   = hInst;
	odp.pszTemplate = MAKEINTRESOURCEA(IDD_OPTIONS);
	odp.pszTitle    = (char*)PluginName;
	odp.pszGroup    = LPGEN("Services");
	odp.flags       = ODPF_BOLDGROUPS;
	odp.pfnDlgProc  = DlgProcOptions;
	CallService( MS_OPT_ADDPAGE, wParam,( LPARAM )&odp );

	return 0;
}


static int ToolbarModulesLoaded(WPARAM, LPARAM)
{
	TBButton tbb = {0};
	tbb.cbSize = sizeof(TBButton);

	tbb.pszButtonID = "clipvi_btn";
	tbb.pszButtonName = LPGEN("Version Information To Clipboard");
	tbb.pszServiceName = MS_CRASHDUMPER_STORETOCLIP;
	tbb.pszTooltipUp = LPGEN("Version Information To Clipboard");
	tbb.hPrimaryIconHandle = GetIconHandle("storeToClip");
	tbb.tbbFlags = TBBF_VISIBLE;
	tbb.defPos = 10000;
	CallService(MS_TB_ADDBUTTON,0, (LPARAM)&tbb);

	tbb.pszButtonID = "filevi_btn";
	tbb.pszButtonName = LPGEN("Version Information To File");
	tbb.pszServiceName = MS_CRASHDUMPER_STORETOFILE;
	tbb.pszTooltipUp = LPGEN("Version Information To File");
	tbb.hPrimaryIconHandle = GetIconHandle("storeToFile");
	tbb.tbbFlags = 0;
	tbb.defPos = 10001;
	CallService(MS_TB_ADDBUTTON,0, (LPARAM)&tbb);

	tbb.pszButtonID = "showvi_btn";
	tbb.pszButtonName = LPGEN("Show Version Information");
	tbb.pszServiceName = MS_CRASHDUMPER_VIEWINFO;
	tbb.pszTooltipUp = LPGEN("Show Version Information");
	tbb.hPrimaryIconHandle = GetIconHandle("showInfo");
	tbb.tbbFlags = 0;
	tbb.defPos = 10002;
	CallService(MS_TB_ADDBUTTON,0, (LPARAM)&tbb);

	tbb.pszButtonID = "upldvi_btn";
	tbb.pszButtonName = LPGEN("Upload Version Information");
	tbb.pszServiceName = MS_CRASHDUMPER_UPLOAD;
	tbb.pszTooltipUp = LPGEN("Upload Version Information");
	tbb.hPrimaryIconHandle = GetIconHandle("uploadInfo");
	tbb.tbbFlags = 0;
	tbb.defPos = 10003;
	CallService(MS_TB_ADDBUTTON,0, (LPARAM)&tbb);

	return 0;
}


static int ModulesLoaded(WPARAM, LPARAM)
{
	char temp[MAX_PATH];
	CallService(MS_SYSTEM_GETVERSIONTEXT, (WPARAM)SIZEOF(temp), (LPARAM)temp);
	crs_a2t(vertxt, temp);

	if (CallService(MS_SYSTEM_GETVERSION, 0, 0) >= PLUGIN_MAKE_VERSION(0,9,0,12))
	{
		profname = Utils_ReplaceVarsT(_T("%miranda_profilename%.dat"));
		if (ServiceExists(MS_FOLDERS_REGISTER_PATH))
		{
			profpath = _T("%miranda_userdata%");
			mirpath = _T("%miranda_path%");
		}
		else
		{
			profpath = Utils_ReplaceVarsT(_T("%miranda_userdata%"));
			mirpath = Utils_ReplaceVarsT(_T("%miranda_path%"));
		}
	}
	else
	{
		CallService(MS_DB_GETPROFILENAME, SIZEOF(temp), (LPARAM)temp);
		crs_a2t(profname, temp);

		CallService(MS_DB_GETPROFILEPATH, SIZEOF(temp), (LPARAM)temp);
		crs_a2t(profpath, temp);

		CallService(MS_DB_GETPROFILEPATH, SIZEOF(temp), (LPARAM)temp);
		crs_a2t(mirpath, temp);
	}

	crs_sntprintf(CrashLogFolder, MAX_PATH, TEXT("%s\\CrashLog"), profpath);
	crs_sntprintf(VersionInfoFolder, MAX_PATH, TEXT("%s"), mirpath);

	SetExceptionHandler();

	hCrashLogFolder = FoldersRegisterCustomPathT(PluginName, "Crash Reports", CrashLogFolder);
	hVerInfoFolder  = FoldersRegisterCustomPathT(PluginName, "Version Information", VersionInfoFolder);

	FoldersPathChanged(0, 0);



	hHooks[2] = HookEvent(ME_FOLDERS_PATH_CHANGED, FoldersPathChanged);
	if (hHooks[3] == NULL) hHooks[3] = HookEvent(ME_TB_MODULELOADED, ToolbarModulesLoaded);

	UploadInit();

	CLISTMENUITEM mi = { 0 };

	mi.cbSize = sizeof(mi);

	mi.popupPosition = 2000089999;
	mi.position = 2000089999;
	mi.flags = CMIF_ROOTPOPUP | CMIF_ICONFROMICOLIB | CMIF_TCHAR;
	mi.icolibItem = GetIconHandle("versionInfo");
	mi.ptszName = LPGENT("Version Information");
	mi.pszPopupName = (char *)-1;
	HANDLE hMenuRoot = Menu_AddMainMenuItem(&mi);

	mi.flags = CMIF_CHILDPOPUP | CMIF_ICONFROMICOLIB | CMIF_TCHAR;
	mi.pszPopupName = (char *)hMenuRoot;
	mi.popupPosition = 0;

	mi.position = 2000089995;
	mi.ptszName = LPGENT("Copy to clipboard");
	mi.icolibItem = GetIconHandle("storeToClip");
	mi.pszService = MS_CRASHDUMPER_STORETOCLIP;
	Menu_AddMainMenuItem(&mi);

	mi.position = 2000089996;
	mi.ptszName = LPGENT("Store to file");
	mi.icolibItem = GetIconHandle("storeToFile");
	mi.pszService = MS_CRASHDUMPER_STORETOFILE;
	Menu_AddMainMenuItem(&mi);

	mi.position  = 2000089997;
	mi.ptszName = LPGENT("Show");
	mi.icolibItem = GetIconHandle("showInfo");
	mi.pszService = MS_CRASHDUMPER_VIEWINFO;
	Menu_AddMainMenuItem(&mi);

	mi.popupPosition = 1;
	mi.position  = 2000089998;
	mi.ptszName = LPGENT("Show with DLLs");
	mi.icolibItem = GetIconHandle("showInfo");
	mi.pszService = MS_CRASHDUMPER_VIEWINFO;
	Menu_AddMainMenuItem(&mi);

	mi.popupPosition = 0;
	mi.position  = 2000089999;
	mi.ptszName = LPGENT("Upload");
	mi.icolibItem = GetIconHandle("uploadInfo");
	mi.pszService = MS_CRASHDUMPER_UPLOAD;
	Menu_AddMainMenuItem(&mi);

	mi.position  = 2000099990;
	mi.ptszName = LPGENT("Open crash report directory");
	mi.icolibItem = LoadSkinnedIconHandle(SKINICON_EVENT_FILE);
	mi.pszService = MS_CRASHDUMPER_URL;
	Menu_AddMainMenuItem(&mi);

	mi.popupPosition = 1;
	mi.position  = 2000099991;
	mi.ptszName = LPGENT("Open miranda-vi.org");
	mi.icolibItem = LoadSkinnedIconHandle(SKINICON_EVENT_URL);
	mi.pszService = MS_CRASHDUMPER_URL;
	Menu_AddMainMenuItem(&mi);

	HOTKEYDESC hk = {0};
	hk.cbSize = sizeof(hk);
	hk.pszSection = PluginName;

	hk.pszDescription = LPGEN("Copy Version Info to clipboard");
	hk.pszName = "CopyVerInfo";
	hk.pszService = MS_CRASHDUMPER_STORETOCLIP;
	CallService(MS_HOTKEY_REGISTER, 0, (LPARAM)&hk);

	hk.pszDescription = LPGEN("Show Version Info");
	hk.pszName = "ShowVerInfo";
	hk.pszService = MS_CRASHDUMPER_VIEWINFO;
	CallService(MS_HOTKEY_REGISTER, 0, (LPARAM)&hk);

	if (servicemode) ViewVersionInfo(0, 0);
	else
	{
		if (DBGetContactSettingByte(NULL, PluginName, "UploadChanged", 0) && !ProcessVIHash(false)) 
			UploadVersionInfo(0, 0xa1);
	}

	CheckForOtherCrashReportingPlugins();
	return 0;
}

static int PreShutdown(WPARAM, LPARAM)
{
	unsigned i;

	DestroyAllWindows();
	UploadClose();

	for (i=0; i<SIZEOF(hHooks); ++i)
		UnhookEvent(hHooks[i]);

	for (i=0; i<SIZEOF(hServices); ++i)
		DestroyServiceFunction(hServices[i]);

	return 0;
}

extern "C" int __declspec(dllexport) Load(PLUGINLINK *link)
{
	pluginLink = link;

	clsdates = DBGetContactSettingByte(NULL, PluginName, "ClassicDates", 1) != 0;

	dtsubfldr = DBGetContactSettingByte(NULL, PluginName, "SubFolders", 1) != 0;

	mir_getMMI(&mmi);
	mir_getMD5I(&md5i);
	mir_getUTFI(&utfi);
	mir_getLP(&pluginInfoEx);

	hHooks[0] = HookEvent(ME_SYSTEM_MODULESLOADED, ModulesLoaded);
	hHooks[1] = HookEvent(ME_OPT_INITIALISE, OptionsInit);
	hHooks[3] = HookEvent(ME_TB_MODULELOADED, ToolbarModulesLoaded);
	hHooks[4] = HookEvent(ME_SYSTEM_PRESHUTDOWN, PreShutdown);

	packlcid = (LCID)CallService(MS_LANGPACK_GETLOCALE, 0, 0);

	InitIcons();

	InitExceptionHandler();

	hServices[0] = CreateServiceFunction(MS_CRASHDUMPER_STORETOFILE, StoreVersionInfoToFile);
	hServices[1] = CreateServiceFunction(MS_CRASHDUMPER_STORETOCLIP, StoreVersionInfoToClipboard);
	hServices[2] = CreateServiceFunction(MS_CRASHDUMPER_VIEWINFO, ViewVersionInfo);
	hServices[3] = CreateServiceFunction(MS_CRASHDUMPER_UPLOAD, UploadVersionInfo);
	hServices[4] = CreateServiceFunction(MS_CRASHDUMPER_URL, OpenUrl);
	hServices[5] = CreateServiceFunction(MS_SERVICEMODE_LAUNCH, ServiceModeLaunch);

	//	unsigned *p = (unsigned*)0x15;
	//	*p = 324;

	return 0;
}


extern "C" int __declspec(dllexport) Unload(void)
{
	DestroyAllWindows();
	FreeLibrary(hRichModule);

	DestroyExceptionHandler();

	if (!_tcsstr(profpath, _T("%miranda")))
	{
		mir_free(profpath);
		mir_free(mirpath);
	}
	mir_free(profname);
	mir_free(vertxt);

	return 0;
}


extern "C" BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID /*lpvReserved*/)
{
	switch(fdwReason)
	{
	case DLL_PROCESS_ATTACH:
		DisableThreadLibraryCalls(hinstDLL);
		hInst = hinstDLL;
		break;

	case DLL_PROCESS_DETACH:
		break;
	}

	return TRUE;
}