/*
Copyright (C) 2006-2010 Ricardo Pescuma Domenecci

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"

// Prototypes ///////////////////////////////////////////////////////////////////////////

CMPlugin g_plugin;

HANDLE hDictionariesFolder = nullptr;
wchar_t *dictionariesFolder;

HANDLE hCustomDictionariesFolder = nullptr;
wchar_t *customDictionariesFolder;

HANDLE hFlagsDllFolder = nullptr;
wchar_t *flagsDllFolder;

HBITMAP hCheckedBmp;
BITMAP bmpChecked;

BOOL loaded = FALSE;

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

static int CompareDicts(const Dictionary *p1, const Dictionary *p2)
{
	return mir_wstrcmpi(p1->full_name, p2->full_name);
}

OBJLIST<Dictionary> languages(1, CompareDicts);

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

PLUGININFOEX pluginInfoEx =
{
	sizeof(PLUGININFOEX),
	__PLUGIN_NAME,
	PLUGIN_MAKE_VERSION(__MAJOR_VERSION, __MINOR_VERSION, __RELEASE_NUM, __BUILD_NUM),
	__DESCRIPTION,
	__AUTHOR,
	__COPYRIGHT,
	__AUTHORWEB,
	UNICODE_AWARE,
	// {36753AE3-840B-4797-94A5-FD9F5852B942}
	{ 0x36753ae3, 0x840b, 0x4797, { 0x94, 0xa5, 0xfd, 0x9f, 0x58, 0x52, 0xb9, 0x42 }}
};

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

// Functions ////////////////////////////////////////////////////////////////////////////

static int OnModuleLoad(WPARAM, LPARAM)
{
	g_plugin.hasVariables = ServiceExists(MS_VARS_FORMATSTRING);
	return 0;
}

static int PreShutdown(WPARAM, LPARAM)
{
	mir_free(dictionariesFolder);
	mir_free(customDictionariesFolder);
	mir_free(flagsDllFolder);
	return 0;
}

// Called when all the modules are loaded
static int ModulesLoaded(WPARAM, LPARAM)
{
	// Folders plugin support
	if (hDictionariesFolder = FoldersRegisterCustomPathW(LPGEN("Spell Checker"), LPGEN("Dictionaries"), DICTIONARIES_FOLDER)) {
		dictionariesFolder = (wchar_t *)mir_alloc(sizeof(wchar_t) * MAX_PATH);
		FoldersGetCustomPathW(hDictionariesFolder, dictionariesFolder, MAX_PATH, L".");
	}
	else dictionariesFolder = Utils_ReplaceVarsW(DICTIONARIES_FOLDER);

	if (hCustomDictionariesFolder = FoldersRegisterCustomPathW(LPGEN("Spell Checker"), LPGEN("Custom Dictionaries"), CUSTOM_DICTIONARIES_FOLDER)) {
		customDictionariesFolder = (wchar_t *)mir_alloc(sizeof(wchar_t) * MAX_PATH);
		FoldersGetCustomPathW(hCustomDictionariesFolder, customDictionariesFolder, MAX_PATH, L".");
	}
	else customDictionariesFolder = Utils_ReplaceVarsW(CUSTOM_DICTIONARIES_FOLDER);

	if (hFlagsDllFolder = FoldersRegisterCustomPathW(LPGEN("Spell Checker"), LPGEN("Flags DLL"), FLAGS_DLL_FOLDER)) {
		flagsDllFolder = (wchar_t *)mir_alloc(sizeof(wchar_t) * MAX_PATH);
		FoldersGetCustomPathW(hFlagsDllFolder, flagsDllFolder, MAX_PATH, L".");
	}
	else flagsDllFolder = Utils_ReplaceVarsW(FLAGS_DLL_FOLDER);

	GetAvaibleDictionaries(languages, dictionariesFolder, customDictionariesFolder);

	LoadOptions();

	if (opts.use_flags) {
		// Load flags dll
		wchar_t flag_file[MAX_PATH];
		mir_snwprintf(flag_file, L"%s\\flags_icons.dll", flagsDllFolder);
		HMODULE hFlagsDll = LoadLibraryEx(flag_file, nullptr, LOAD_LIBRARY_AS_DATAFILE);

		wchar_t path[MAX_PATH];
		GetModuleFileName(g_plugin.getInst(), path, MAX_PATH);

		SKINICONDESC sid = {};
		sid.flags = SIDF_ALL_UNICODE | SIDF_SORTED;
		sid.section.w = LPGENW("Spell Checker") L"/" LPGENW("Flags");

		// Get language flags
		for (auto &p : languages) {
			sid.description.w = p->full_name;

			char lang[32];
			mir_snprintf(lang, "spell_lang_%d", languages.indexOf(&p));
			sid.pszName = lang;

			HICON hFlag = nullptr, hFlagIcoLib = nullptr;
			if (hFlagsDll != nullptr)
				hFlag = (HICON)LoadImage(hFlagsDll, p->language, IMAGE_ICON, 16, 16, 0);

			if (hFlag != nullptr) {
				sid.hDefaultIcon = hFlag;
				sid.defaultFile.w = nullptr;
				sid.iDefaultIndex = 0;
			}
			else {
				hFlagIcoLib = IcoLib_GetIcon("spellchecker_unknown");
				sid.hDefaultIcon = hFlagIcoLib;
				sid.defaultFile.w = nullptr;
				sid.iDefaultIndex = 0;
			}

			// Oki, lets add to IcoLib, then
			p->hIcolib = g_plugin.addIcon(&sid);

			if (hFlag != nullptr)
				DestroyIcon(hFlag);
			else
				IcoLib_ReleaseIcon(hFlagIcoLib);
		}
		FreeLibrary(hFlagsDll);
	}

	for (auto &dict : languages) {
		wchar_t filename[MAX_PATH];
		mir_snwprintf(filename, L"%s\\%s.ar", customDictionariesFolder, dict->language);
		dict->autoReplace = new AutoReplaceMap(filename, dict);

		if (mir_wstrcmp(dict->language, opts.default_language) == 0)
			dict->load();
	}

	HookEvent(ME_MSG_WINDOWEVENT, MsgWindowEvent);
	HookEvent(ME_MSG_WINDOWPOPUP, MsgWindowPopup);
	HookEvent(ME_MSG_ICONPRESSED, IconPressed);

	HookEvent(ME_SYSTEM_MODULELOAD, OnModuleLoad);
	HookEvent(ME_SYSTEM_MODULEUNLOAD, OnModuleLoad);
	OnModuleLoad(0, 0);

	StatusIconData sid = {};
	sid.szModule = MODULENAME;
	sid.hIconDisabled = IcoLib_GetIcon("spellchecker_disabled");
	sid.flags = MBF_UNICODE | MBF_HIDDEN;

	for (int i = 0; i < languages.getCount(); i++) {
		sid.dwId = i;

		wchar_t tmp[128];
		mir_snwprintf(tmp, L"%s - %s", TranslateT("Spell Checker"), languages[i].full_name);
		sid.szTooltip.w = tmp;
		sid.hIcon = (opts.use_flags) ? IcoLib_GetIconByHandle(languages[i].hIcolib) : IcoLib_GetIcon("spellchecker_enabled");
		Srmm_AddIcon(&sid, &g_plugin);
	}

	HOTKEYDESC hkd = {};
	hkd.pszName = "Spell Checker/Toggle";
	hkd.szSection.a = LPGEN("Spell Checker");
	hkd.szDescription.a = LPGEN("Enable/disable spell checker");
	hkd.DefHotKey = HOTKEYCODE(HOTKEYF_SHIFT | HOTKEYF_ALT, 'S');
	hkd.lParam = HOTKEY_ACTION_TOGGLE;
	g_plugin.addHotkey(&hkd);

	loaded = TRUE;

	return 0;
}

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

static bool bComInited = false;

static IconItem iconList[] =
{
	{ LPGEN("Enabled"), "spellchecker_enabled", IDI_CHECK },
	{ LPGEN("Disabled"), "spellchecker_disabled", IDI_NO_CHECK },
	{ LPGEN("Unknown"), "spellchecker_unknown", IDI_UNKNOWN_FLAG }
};

static BOOL CALLBACK EnumLocalesProc(LPWSTR lpLocaleString)
{
	wchar_t *stopped = nullptr;
	USHORT langID = (USHORT)wcstol(lpLocaleString, &stopped, 16);

	wchar_t ini[32];
	wchar_t end[32];
	GetLocaleInfo(MAKELCID(langID, 0), LOCALE_SISO639LANGNAME, ini, _countof(ini));
	GetLocaleInfo(MAKELCID(langID, 0), LOCALE_SISO3166CTRYNAME, end, _countof(end));

	wchar_t name[64];
	mir_snwprintf(name, L"%s_%s", ini, end);
	g_plugin.locales[name] = langID;
	return TRUE;
}

int CMPlugin::Load()
{
	bComInited = SUCCEEDED(CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED));
	if (bComInited) {
		if (FAILED(m_spellFactory.CoCreateInstance(__uuidof(SpellCheckerFactory), nullptr, CLSCTX_INPROC_SERVER))) {
			bComInited = false;
			CoUninitialize();
		}
	}

	EnumSystemLocalesW(EnumLocalesProc, LCID_SUPPORTED);

	// icons
	g_plugin.registerIcon(LPGEN("Spell Checker"), iconList);

	// hooks
	HookEvent(ME_SYSTEM_MODULESLOADED, ModulesLoaded);
	HookEvent(ME_SYSTEM_PRESHUTDOWN, PreShutdown);
	HookEvent(ME_OPT_INITIALISE, InitOptionsCallback);

	LoadOptions();

	CreateServiceFunction(MS_SPELLCHECKER_ADD_RICHEDIT, AddContactTextBoxService);
	CreateServiceFunction(MS_SPELLCHECKER_REMOVE_RICHEDIT, RemoveContactTextBoxService);
	CreateServiceFunction(MS_SPELLCHECKER_SHOW_POPUP_MENU, ShowPopupMenuService);

	hCheckedBmp = LoadBitmap(nullptr, MAKEINTRESOURCE(OBM_CHECK));
	if (GetObject(hCheckedBmp, sizeof(bmpChecked), &bmpChecked) == 0)
		bmpChecked.bmHeight = bmpChecked.bmWidth = 10;

	return 0;
}

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

int CMPlugin::Unload()
{
	DeleteObject(hCheckedBmp);
	languages.destroy();

	if (bComInited)
		CoUninitialize();
	return 0;
}