temporary return of DbChecker

1 files changed, 332 insertions, 0 deletions

diff --git a/plugins/DbChecker/src/selectdb.cpp b/plugins/DbChecker/src/selectdb.cpp
new file mode 100644
index 0000000000..d395c6b4c4
--- /dev/null
+++ b/plugins/DbChecker/src/selectdb.cpp
@@ -0,0 +1,332 @@
+/*
+Miranda Database Tool
+Copyright (C) 2001-2005  Richard Hughes
+
+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"
+
+static bool CheckBroken(const wchar_t *ptszFullPath)
+{
+	DATABASELINK *dblink = FindDatabasePlugin(ptszFullPath);
+	if (dblink == nullptr)
+		return true;
+
+	return dblink->grokHeader(ptszFullPath) != EGROKPRF_NOERROR;
+}
+
+int OpenDatabase(HWND hdlg, INT iNextPage)
+{
+	wchar_t tszMsg[1024];
+	int error = 0;
+
+	if (opts.dbChecker == nullptr) {
+		DATABASELINK* dblink = FindDatabasePlugin(opts.filename);
+		if (dblink == nullptr) {
+			mir_snwprintf(tszMsg,
+				TranslateT("Database Checker cannot find a suitable database plugin to open '%s'."),
+				opts.filename);
+		LBL_Error:
+			MessageBox(hdlg, tszMsg, TranslateT("Error"), MB_OK | MB_ICONERROR);
+			return false;
+		}
+
+		auto *pDb = dblink->Load(opts.filename, false);
+		if (pDb == nullptr) {
+			PostMessage(GetParent(hdlg), WZM_GOTOPAGE, IDD_OPENERROR, (LPARAM)OpenErrorDlgProc);
+			return true;
+		}
+
+		opts.dbChecker = pDb->GetChecker();
+		if (opts.dbChecker == nullptr) {
+			mir_snwprintf(tszMsg, TranslateT("Database driver '%s' doesn't support checking."), TranslateW(dblink->szFullName));
+			goto LBL_Error;
+		}
+
+		opts.db = pDb;
+	}
+
+	// force check
+	if (error == EGROKPRF_OBSOLETE) {
+		opts.bAggressive = opts.bBackup = true;
+		PostMessage(GetParent(hdlg), WZM_GOTOPAGE, IDD_PROGRESS, (LPARAM)ProgressDlgProc);
+	}
+	else if (iNextPage == IDD_FILEACCESS)
+		PostMessage(GetParent(hdlg), WZM_GOTOPAGE, IDD_FILEACCESS, (LPARAM)FileAccessDlgProc);
+	else
+		PostMessage(GetParent(hdlg), WZM_GOTOPAGE, IDD_PROGRESS, (LPARAM)ProgressDlgProc);
+	return true;
+}
+
+void GetProfileDirectory(wchar_t* szMirandaDir, wchar_t* szPath, int cbPath)
+{
+	wchar_t szProfileDir[MAX_PATH], szExpandedProfileDir[MAX_PATH], szMirandaBootIni[MAX_PATH];
+
+	mir_wstrcpy(szMirandaBootIni, szMirandaDir);
+	mir_wstrcat(szMirandaBootIni, L"\\mirandaboot.ini");
+	GetPrivateProfileString(L"Database", L"ProfileDir", L"./Profiles", szProfileDir, _countof(szProfileDir), szMirandaBootIni);
+	ExpandEnvironmentStrings(szProfileDir, szExpandedProfileDir, _countof(szExpandedProfileDir));
+	_wchdir(szMirandaDir);
+	if (!_wfullpath(szPath, szExpandedProfileDir, cbPath))
+		mir_wstrncpy(szPath, szMirandaDir, cbPath);
+	if (szPath[mir_wstrlen(szPath) - 1] == '\\')
+		szPath[mir_wstrlen(szPath) - 1] = 0;
+}
+
+static int AddDatabaseToList(HWND hwndList, const wchar_t* filename, wchar_t* dir)
+{
+	LV_ITEM lvi;
+	lvi.mask = LVIF_PARAM;
+	lvi.iSubItem = 0;
+	for (lvi.iItem = ListView_GetItemCount(hwndList) - 1; lvi.iItem >= 0; lvi.iItem--) {
+		ListView_GetItem(hwndList, &lvi);
+		if (!mir_wstrcmpi((wchar_t*)lvi.lParam, filename))
+			return lvi.iItem;
+	}
+
+	struct _stat st;
+	if (_wstat(filename, &st) == -1)
+		return -1;
+
+	DWORD totalSize = st.st_size;
+
+	bool isBroken = CheckBroken(filename);
+
+	const wchar_t *pName = wcsrchr(filename, '\\');
+	if (pName == nullptr)
+		pName = (LPTSTR)filename;
+	else
+		pName++;
+
+	wchar_t szName[MAX_PATH];
+	mir_snwprintf(szName, L"%s%s", dir, pName);
+
+	wchar_t *pDot = wcsrchr(szName, '.');
+	if (pDot != nullptr && !mir_wstrcmpi(pDot, L".dat"))
+		*pDot = 0;
+
+	lvi.iItem = 0;
+	lvi.mask = LVIF_TEXT | LVIF_PARAM | LVIF_IMAGE;
+	lvi.iSubItem = 0;
+	lvi.lParam = (LPARAM)wcsdup(filename);
+	lvi.pszText = szName;
+	lvi.iImage = (isBroken) ? 1 : 0;
+
+	int iNewItem = ListView_InsertItem(hwndList, &lvi);
+	wchar_t szSize[20];
+	mir_snwprintf(szSize, L"%.2lf MB", totalSize / 1048576.0);
+	ListView_SetItemText(hwndList, iNewItem, 1, szSize);
+	return iNewItem;
+}
+
+void FindAdd(HWND hdlg, wchar_t *szProfileDir, wchar_t *szPrefix)
+{
+	HANDLE hFind;
+	WIN32_FIND_DATA fd;
+	wchar_t szSearchPath[MAX_PATH], szFilename[MAX_PATH];
+
+	mir_wstrcpy(szSearchPath, szProfileDir);
+	mir_wstrcat(szSearchPath, L"\\*.*");
+
+	hFind = FindFirstFile(szSearchPath, &fd);
+	if (hFind != INVALID_HANDLE_VALUE) {
+		do {
+			if (!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) || !mir_wstrcmp(fd.cFileName, L".") || !mir_wstrcmp(fd.cFileName, L".."))
+				continue;
+
+			mir_snwprintf(szFilename, L"%s\\%s\\%s.dat", szProfileDir, fd.cFileName, fd.cFileName);
+			if (_waccess(szFilename, 0) == 0)
+				AddDatabaseToList(GetDlgItem(hdlg, IDC_DBLIST), szFilename, szPrefix);
+		} while (FindNextFile(hFind, &fd));
+		FindClose(hFind);
+	}
+}
+
+wchar_t *addstring(wchar_t *str, wchar_t *add)
+{
+	mir_wstrcpy(str, add);
+	return str + mir_wstrlen(add) + 1;
+}
+
+INT_PTR CALLBACK SelectDbDlgProc(HWND hdlg, UINT message, WPARAM wParam, LPARAM lParam)
+{
+	INT_PTR bReturn;
+	if (DoMyControlProcessing(hdlg, message, wParam, lParam, &bReturn))
+		return bReturn;
+
+	switch (message) {
+	case WM_INITDIALOG:
+		TranslateDialogDefault(hdlg);
+		{
+			HIMAGELIST hIml = ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), ILC_COLOR32 | ILC_MASK, 3, 3);
+			ImageList_AddIcon(hIml, LoadIcon(g_plugin.getInst(), MAKEINTRESOURCE(IDI_PROFILEGREEN)));
+			ImageList_AddIcon(hIml, LoadIcon(g_plugin.getInst(), MAKEINTRESOURCE(IDI_BAD)));
+			ListView_SetImageList(GetDlgItem(hdlg, IDC_DBLIST), hIml, LVSIL_SMALL);
+		}
+		ListView_SetExtendedListViewStyleEx(GetDlgItem(hdlg, IDC_DBLIST), LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);
+		{
+			LV_COLUMN lvc;
+			lvc.mask = LVCF_WIDTH | LVCF_FMT | LVCF_TEXT;
+			lvc.cx = 290;
+			lvc.fmt = LVCFMT_LEFT;
+			lvc.pszText = TranslateT("Database");
+			ListView_InsertColumn(GetDlgItem(hdlg, IDC_DBLIST), 0, &lvc);
+			lvc.cx = 68;
+			lvc.fmt = LVCFMT_RIGHT;
+			lvc.pszText = TranslateT("Total size");
+			ListView_InsertColumn(GetDlgItem(hdlg, IDC_DBLIST), 1, &lvc);
+
+			wchar_t szMirandaPath[MAX_PATH];
+			GetModuleFileName(nullptr, szMirandaPath, _countof(szMirandaPath));
+			wchar_t *str2 = wcsrchr(szMirandaPath, '\\');
+			if (str2 != nullptr)
+				*str2 = 0;
+
+			int i = 0;
+			HKEY hKey;
+			wchar_t szProfileDir[MAX_PATH];
+			wchar_t szMirandaProfiles[MAX_PATH];
+			DWORD cbData = _countof(szMirandaPath);
+
+			mir_wstrcpy(szMirandaProfiles, szMirandaPath);
+			mir_wstrcat(szMirandaProfiles, L"\\Profiles");
+			GetProfileDirectory(szMirandaPath, szProfileDir, _countof(szProfileDir));
+
+			// search in profile dir (using ini file)
+			if (mir_wstrcmpi(szProfileDir, szMirandaProfiles))
+				FindAdd(hdlg, szProfileDir, L"[ini]\\");
+
+			FindAdd(hdlg, szMirandaProfiles, L"[prf]\\");
+			// search in current dir (as DBTOOL)
+			FindAdd(hdlg, szMirandaPath, L"[.]\\");
+
+			// search in profile dir (using registry path + ini file)
+			if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\miranda32.exe", 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS) {
+				if (RegQueryValueEx(hKey, L"Path", nullptr, nullptr, (PBYTE)szMirandaPath, &cbData) == ERROR_SUCCESS) {
+					if (mir_wstrcmp(szProfileDir, szMirandaPath)) {
+						GetProfileDirectory(szMirandaPath, szProfileDir, _countof(szProfileDir));
+						FindAdd(hdlg, szProfileDir, L"[reg]\\");
+					}
+				}
+				RegCloseKey(hKey);
+			}
+			// select
+			if (opts.filename[0])
+				i = AddDatabaseToList(GetDlgItem(hdlg, IDC_DBLIST), opts.filename, L"");
+			if (i == -1)
+				i = 0;
+			ListView_SetItemState(GetDlgItem(hdlg, IDC_DBLIST), i, LVIS_SELECTED, LVIS_SELECTED);
+		}
+
+		if (opts.dbChecker != nullptr) {
+			opts.dbChecker->Destroy();
+			opts.dbChecker = nullptr;
+		}
+
+		if (bShortMode)
+			if (!OpenDatabase(hdlg, IDD_FILEACCESS)) {
+				EndDialog(GetParent(hdlg), 1);
+				return FALSE;
+			}
+		return TRUE;
+
+	case WZN_PAGECHANGING:
+		GetDlgItemText(hdlg, IDC_FILE, opts.filename, _countof(opts.filename));
+		break;
+
+	case WM_COMMAND:
+		switch (LOWORD(wParam)) {
+		case IDC_FILE:
+			if (HIWORD(wParam) == EN_CHANGE)
+				EnableWindow(GetDlgItem(GetParent(hdlg), IDOK), GetWindowTextLength(GetDlgItem(hdlg, IDC_FILE)));
+			break;
+
+		case IDC_OTHER:
+		{
+			OPENFILENAME ofn = { 0 };
+			wchar_t str[MAX_PATH];
+
+			// L"Miranda Databases (*.dat)\0*.DAT\0All Files (*)\0*\0";
+			wchar_t *filter, *tmp, *tmp1, *tmp2;
+			tmp1 = TranslateT("Miranda Databases (*.dat)");
+			tmp2 = TranslateT("All Files");
+			filter = tmp = (wchar_t*)_alloca((mir_wstrlen(tmp1) + mir_wstrlen(tmp2) + 11)*sizeof(wchar_t));
+			tmp = addstring(tmp, tmp1);
+			tmp = addstring(tmp, L"*.DAT");
+			tmp = addstring(tmp, tmp2);
+			tmp = addstring(tmp, L"*");
+			*tmp = 0;
+
+			GetDlgItemText(hdlg, IDC_FILE, str, _countof(str));
+			ofn.lStructSize = sizeof(ofn);
+			ofn.hwndOwner = hdlg;
+			ofn.hInstance = nullptr;
+			ofn.lpstrFilter = filter;
+			ofn.lpstrDefExt = L"dat";
+			ofn.lpstrFile = str;
+			ofn.Flags = OFN_FILEMUSTEXIST | OFN_HIDEREADONLY;
+			ofn.nMaxFile = _countof(str);
+			ofn.nMaxFileTitle = MAX_PATH;
+			if (GetOpenFileName(&ofn)) {
+				int i = AddDatabaseToList(GetDlgItem(hdlg, IDC_DBLIST), str, L"");
+				if (i == -1)
+					i = 0;
+				ListView_SetItemState(GetDlgItem(hdlg, IDC_DBLIST), i, LVIS_SELECTED, LVIS_SELECTED);
+			}
+		}
+		break;
+
+		case IDC_BACK:
+			if (!bShortMode)
+				PostMessage(GetParent(hdlg), WZM_GOTOPAGE, IDD_WELCOME, (LPARAM)WelcomeDlgProc);
+			break;
+
+		case IDOK:
+			OpenDatabase(hdlg, IDD_FILEACCESS);
+			break;
+		}
+		break;
+
+	case WM_NOTIFY:
+		switch (((LPNMHDR)lParam)->idFrom) {
+		case IDC_DBLIST:
+			switch (((LPNMLISTVIEW)lParam)->hdr.code) {
+			case LVN_ITEMCHANGED:
+				LV_ITEM lvi;
+				lvi.iItem = ListView_GetNextItem(GetDlgItem(hdlg, IDC_DBLIST), -1, LVNI_SELECTED);
+				if (lvi.iItem == -1) break;
+				lvi.mask = LVIF_PARAM;
+				ListView_GetItem(GetDlgItem(hdlg, IDC_DBLIST), &lvi);
+				SetDlgItemText(hdlg, IDC_FILE, (wchar_t*)lvi.lParam);
+				SendMessage(hdlg, WM_COMMAND, MAKEWPARAM(IDC_FILE, EN_CHANGE), (LPARAM)GetDlgItem(hdlg, IDC_FILE));
+			}
+			break;
+		}
+		break;
+
+	case WM_DESTROY:
+	{
+		LV_ITEM lvi;
+		lvi.mask = LVIF_PARAM;
+		for (lvi.iItem = ListView_GetItemCount(GetDlgItem(hdlg, IDC_DBLIST)) - 1; lvi.iItem >= 0; lvi.iItem--) {
+			ListView_GetItem(GetDlgItem(hdlg, IDC_DBLIST), &lvi);
+			free((char*)lvi.lParam);
+		}
+	}
+	break;
+	}
+	return FALSE;
+}