/*

Miranda NG: the free IM client for Microsoft* Windows*

Copyright (c) 2012-18 Miranda NG team (https://miranda-ng.org)
all portions of this codebase are copyrighted to the people
listed in contributors.txt.

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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
*/

#include "stdafx.h"

int hLangpack;

static PLUGININFOEX pluginInfo =
{
	sizeof(PLUGININFOEX),
	__PLUGIN_NAME,
	PLUGIN_MAKE_VERSION(__MAJOR_VERSION, __MINOR_VERSION, __RELEASE_NUM, __BUILD_NUM),
	__DESCRIPTION,
	__AUTHOR,
	__COPYRIGHT,
	__AUTHORWEB,
	UNICODE_AWARE | STATIC_PLUGIN,
	//{F7A6B27C-9D9C-4A42-BE86-A448AE109161}
	{ 0xf7a6b27c, 0x9d9c, 0x4a42, { 0xbe, 0x86, 0xa4, 0x48, 0xae, 0x10, 0x91, 0x61 } }
};

HINSTANCE g_hInst = nullptr;

LIST<CDb3Mmap> g_Dbs(1, HandleKeySortT);

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

// returns 0 if the profile is created, EMKPRF*
static int makeDatabase(const wchar_t*)
{
	return EMKPRF_CREATEFAILED;
}

// returns 0 if the given profile has a valid header
static int grokHeader(const wchar_t *profile)
{
	std::auto_ptr<CDb3Mmap> db(new CDb3Mmap(profile, DBMODE_SHARED | DBMODE_READONLY));
	if (db->Load(true) != ERROR_SUCCESS)
		return EGROKPRF_CANTREAD;

	return db->CheckDbHeaders(false);
}

// returns 0 if all the APIs are injected otherwise, 1
#define CONVERT_MSG LPGEN("This database is in old format that isn't supported anymore. Press Yes to convert it to the new format or No to return back")
#define MISSING_DB_MSG LPGEN("To open this database you need to install the dbx_mdbx plugin. Click Yes to download it from Miranda NG's site or No to return back")
#define MISSING_PLUG_MSG LPGEN("To open this database you need to install the Import plugin. Click Yes to download it from Miranda NG's site or No to return back")

static MDatabaseCommon* LoadDatabase(const wchar_t *profile, BOOL bReadOnly)
{
	// set the memory, lists & UTF8 manager
	mir_getLP(&pluginInfo);

	////////////////////////////////////////////////////////////////////////////////////////
	// if not read only, convert the old profile to libmdbx
	if (!bReadOnly) {
		DATABASELINK *pLink = GetDatabasePlugin("dbx_mdbx");
		if (pLink == nullptr) {
			if (IDYES == MessageBoxW(nullptr, TranslateT(MISSING_DB_MSG), L"Miranda NG", MB_YESNO))
				Utils_OpenUrl("https://miranda-ng.org/p/Dbx_mdbx");
			return nullptr;
		}

		if (IDYES != MessageBoxW(nullptr, TranslateT(CONVERT_MSG), L"Miranda NG", MB_YESNO))
			return nullptr;

		int errorCode;
		CMStringW wszBackupName(profile);
		wszBackupName.Append(L".bak");
		if (!MoveFileW(profile, wszBackupName)) {
			Netlib_LogfW(nullptr, L"Cannot move old profile '%s' to '%s': error %d", profile, wszBackupName.c_str(), GetLastError());
			return nullptr;
		}

		if ((errorCode = pLink->makeDatabase(profile)) != 0) {
			Netlib_LogfW(nullptr, L"Database creation '%s' failed with error code %d", profile, errorCode);
LBL_Error: 
			DeleteFileW(profile);
			MoveFileW(wszBackupName, profile);
			return nullptr;
		}

		if (SetServiceModePlugin(L"import") != ERROR_SUCCESS) {
			if (IDYES == MessageBoxW(nullptr, TranslateT(MISSING_PLUG_MSG), L"Miranda NG", MB_YESNO))
				Utils_OpenUrl("https://miranda-ng.org/p/Import");
			goto LBL_Error;
		}

		return pLink->Load(profile, false);
	}

	std::auto_ptr<CDb3Mmap> db(new CDb3Mmap(profile, DBMODE_READONLY));
	if (db->Load(false) != ERROR_SUCCESS)
		return nullptr;

	g_Dbs.insert(db.get());
	return db.release();
}

static DATABASELINK dblink =
{
	0,
	"dbx_mmap",
	L"dbx mmap driver",
	makeDatabase,
	grokHeader,
	LoadDatabase
};

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

extern "C" __declspec(dllexport) PLUGININFOEX* MirandaPluginInfoEx(DWORD)
{
	return &pluginInfo;
}

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

extern "C" __declspec(dllexport) int Load(void)
{
	RegisterDatabasePlugin(&dblink);
	return 0;
}

extern "C" __declspec(dllexport) int Unload(void)
{
	return 0;
}

BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD, LPVOID)
{
	g_hInst = hInstDLL;
	return TRUE;
}