diff options
Diffstat (limited to 'plugins/DbChecker/src/selectdb.cpp')
| -rw-r--r-- | plugins/DbChecker/src/selectdb.cpp | 332 |
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; +} |