initial commit

git-svn-id: http://svn.miranda-ng.org/main/trunk@2 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c

1 files changed, 303 insertions, 0 deletions

diff --git a/plugins/Dbx_mmap_SA/Dbtool/selectdb.cpp b/plugins/Dbx_mmap_SA/Dbtool/selectdb.cpp
new file mode 100644
index 0000000000..d8fa36ed75
--- /dev/null
+++ b/plugins/Dbx_mmap_SA/Dbtool/selectdb.cpp
@@ -0,0 +1,303 @@
+/*

+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 "dbtool.h"

+

+void GetProfileDirectory(TCHAR* szMirandaDir, TCHAR* szPath, int cbPath)

+{

+	TCHAR szProfileDir[MAX_PATH], szExpandedProfileDir[MAX_PATH], szMirandaBootIni[MAX_PATH];

+

+	lstrcpy(szMirandaBootIni,szMirandaDir);

+	lstrcat(szMirandaBootIni,_T("\\mirandaboot.ini"));

+	GetPrivateProfileString(_T("Database"),_T("ProfileDir"),_T("./Profiles"),szProfileDir,SIZEOF(szProfileDir),szMirandaBootIni);

+	ExpandEnvironmentStrings(szProfileDir,szExpandedProfileDir,SIZEOF(szExpandedProfileDir));

+	_tchdir(szMirandaDir);

+	if(!_tfullpath(szPath,szExpandedProfileDir,cbPath))

+		lstrcpyn(szPath,szMirandaDir,cbPath);

+	if(szPath[lstrlen(szPath)-1]=='\\')

+		szPath[lstrlen(szPath)-1] = 0;

+}

+

+static int AddDatabaseToList(HWND hwndList, TCHAR* filename, TCHAR* dir)

+{

+	LV_ITEM lvi;

+	int iNewItem;

+	TCHAR szSize[20], *pName, *pDot, szName[MAX_PATH];

+	HANDLE hDbFile;

+	DBHeader dbhdr;

+	DWORD bytesRead;

+	DWORD totalSize,wasted=0;

+	int broken=0;

+

+	lvi.mask=LVIF_PARAM;

+	lvi.iSubItem=0;

+	for(lvi.iItem=ListView_GetItemCount(hwndList)-1;lvi.iItem>=0;lvi.iItem--) {

+		ListView_GetItem(hwndList,&lvi);

+		if( !_tcsicmp(( TCHAR* )lvi.lParam,filename)) return lvi.iItem;

+	}

+	hDbFile=CreateFile(filename,GENERIC_READ,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,0,NULL);

+	if ( hDbFile == INVALID_HANDLE_VALUE )

+		return -1;

+

+	ReadFile(hDbFile,&dbhdr,sizeof(dbhdr),&bytesRead,NULL);

+	totalSize=GetFileSize(hDbFile,NULL);

+

+	broken = (bytesRead<sizeof(dbhdr) || memcmp(dbhdr.signature,&dbSignature,sizeof(dbhdr.signature)));

+

+	if(!broken) {

+		wasted = dbhdr.slackSpace;

+		if (totalSize>dbhdr.ofsFileEnd)

+			wasted+=totalSize-dbhdr.ofsFileEnd;

+	}

+	CloseHandle(hDbFile);

+

+	pName = _tcsrchr(filename,'\\');

+	if ( pName == NULL )

+		pName = ( LPTSTR )filename;

+	else

+		pName++;

+	_tcscpy(szName,dir);

+	_tcscat(szName,pName);

+	pDot = _tcsrchr( szName, '.' );

+	if ( pDot != NULL && !_tcsicmp( pDot, _T(".dat")) )

+		*pDot=0;

+

+	lvi.iItem = 0;

+	lvi.mask = LVIF_TEXT | LVIF_PARAM | LVIF_IMAGE;

+	lvi.iSubItem = 0;

+	lvi.lParam = ( LPARAM )_tcsdup(filename);

+	lvi.pszText = szName;

+	if (broken)

+		lvi.iImage = 3;

+	else if ( wasted < 1024*128 )

+		lvi.iImage = 0;

+	else if ( wasted < 1024*256 + (DWORD)(totalSize > 2*1024*1024 ) ? 256 * 1024 : 0 )

+		lvi.iImage=1;

+	else

+		lvi.iImage=2;

+

+	iNewItem = ListView_InsertItem(hwndList, &lvi );

+	_stprintf(szSize,_T("%.2lf MB"),totalSize/1048576.0);

+	ListView_SetItemText(hwndList,iNewItem,1,szSize);

+	if ( !broken ) {

+		_stprintf(szSize,_T("%.2lf MB"),wasted/1048576.0);

+		ListView_SetItemText(hwndList,iNewItem,2,szSize);

+	}

+	return iNewItem;

+}

+

+void FindAdd(HWND hdlg, TCHAR *szProfileDir, TCHAR *szPrefix)

+{

+	HANDLE hFind;

+	WIN32_FIND_DATA fd;

+	TCHAR szSearchPath[MAX_PATH],szFilename[MAX_PATH];

+

+	lstrcpy(szSearchPath,szProfileDir);

+	lstrcat(szSearchPath,_T("\\*.*"));

+

+	hFind=FindFirstFile(szSearchPath,&fd);

+	if (hFind!=INVALID_HANDLE_VALUE) {

+		do {

+			if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) && _tcscmp(fd.cFileName, _T(".")) && _tcscmp(fd.cFileName, _T(".."))) {

+				wsprintf(szFilename,_T("%s\\%s\\%s.dat"),szProfileDir,fd.cFileName,fd.cFileName);

+				if (_taccess(szFilename, 0) == 0)

+					AddDatabaseToList(GetDlgItem(hdlg,IDC_DBLIST),szFilename,szPrefix);

+			}

+		} while(FindNextFile(hFind,&fd));

+		FindClose(hFind);

+	}

+}

+

+TCHAR *addstring(TCHAR *str, TCHAR *add) {

+	_tcscpy(str,add);

+	return str + _tcslen(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:

+		{

+			TCHAR szMirandaPath[MAX_PATH];

+			szMirandaPath[ 0 ] = 0;

+			{	HIMAGELIST hIml;

+				hIml=ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON),

+					(IsWinVerXPPlus() ? ILC_COLOR32 : ILC_COLOR16) | ILC_MASK, 3, 3);

+				ImageList_AddIcon(hIml,LoadIcon(hInst,MAKEINTRESOURCE(IDI_PROFILEGREEN)));

+				ImageList_AddIcon(hIml,LoadIcon(hInst,MAKEINTRESOURCE(IDI_PROFILEYELLOW)));

+				ImageList_AddIcon(hIml,LoadIcon(hInst,MAKEINTRESOURCE(IDI_PROFILERED)));

+				ImageList_AddIcon(hIml,LoadIcon(hInst,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 = 205;

+				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 );

+				lvc.pszText = TranslateT("Wasted");

+				ListView_InsertColumn(GetDlgItem(hdlg,IDC_DBLIST), 2, &lvc );

+			}

+			{

+				TCHAR *str2;

+				GetModuleFileName(NULL,szMirandaPath,SIZEOF(szMirandaPath));

+				str2 = _tcsrchr(szMirandaPath,'\\');

+				if( str2 != NULL )

+					*str2=0;

+			}

+			{

+				int i = 0;

+				HKEY hKey;

+				TCHAR szProfileDir[MAX_PATH];

+				DWORD cbData = SIZEOF(szMirandaPath);

+				TCHAR szMirandaProfiles[MAX_PATH];

+

+				_tcscpy(szMirandaProfiles, szMirandaPath);

+				_tcscat(szMirandaProfiles, _T("\\Profiles"));

+				GetProfileDirectory(szMirandaPath,szProfileDir,SIZEOF(szProfileDir));

+

+				// search in profile dir (using ini file)

+				if( lstrcmpi(szProfileDir,szMirandaProfiles) )

+					FindAdd(hdlg, szProfileDir, _T("[ini]\\"));

+

+				FindAdd(hdlg, szMirandaProfiles, _T("[prf]\\"));

+				// search in current dir (as DBTOOL)

+        FindAdd(hdlg, szMirandaPath, _T("[ . ]\\"));

+

+				// search in profile dir (using registry path + ini file)

+				if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,_T("Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\miranda32.exe"),0,KEY_QUERY_VALUE,&hKey) == ERROR_SUCCESS) {

+					if(RegQueryValueEx(hKey,_T("Path"),NULL,NULL,(PBYTE)szMirandaPath,&cbData) == ERROR_SUCCESS) {

+						if( lstrcmp(szProfileDir,szMirandaPath) ) {

+							GetProfileDirectory(szMirandaPath,szProfileDir,SIZEOF(szProfileDir));

+							FindAdd(hdlg, szProfileDir, _T("[reg]\\"));

+						}

+					}

+					RegCloseKey(hKey);

+				}

+				// select

+				if ( opts.filename[0] )

+					i = AddDatabaseToList( GetDlgItem( hdlg, IDC_DBLIST ), opts.filename, _T("") );

+				if ( i == -1 )

+					i = 0;

+				ListView_SetItemState( GetDlgItem(hdlg,IDC_DBLIST), i, LVIS_SELECTED, LVIS_SELECTED );

+			}

+			if ( opts.hFile != NULL && opts.hFile != INVALID_HANDLE_VALUE ) {

+				CloseHandle( opts.hFile );

+				opts.hFile = NULL;

+			}

+			TranslateDialog( hdlg );

+			return TRUE;

+		}

+

+		case WZN_PAGECHANGING:

+			GetDlgItemText( hdlg, IDC_FILE, opts.filename, SIZEOF(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};

+					TCHAR str[MAX_PATH];

+

+					// _T("Miranda Databases (*.dat)\0*.DAT\0All Files (*)\0*\0");

+					TCHAR *filter, *tmp, *tmp1, *tmp2;

+					tmp1 = TranslateT("Miranda Databases (*.dat)");

+					tmp2 = TranslateT("All Files");

+					filter = tmp = (TCHAR*)_alloca((_tcslen(tmp1)+_tcslen(tmp2)+11)*sizeof(TCHAR));

+					tmp = addstring(tmp, tmp1);

+					tmp = addstring(tmp, _T("*.DAT"));

+					tmp = addstring(tmp, tmp2);

+					tmp = addstring(tmp, _T("*"));

+					*tmp = 0;

+

+					GetDlgItemText( hdlg, IDC_FILE, str, SIZEOF( str ));

+					ofn.lStructSize = sizeof(ofn);

+					ofn.hwndOwner = hdlg;

+					ofn.hInstance = NULL;

+					ofn.lpstrFilter = filter;

+					ofn.lpstrDefExt = _T("dat");

+					ofn.lpstrFile = str;

+					ofn.Flags = OFN_FILEMUSTEXIST | OFN_HIDEREADONLY;

+					ofn.nMaxFile = SIZEOF(str);

+					ofn.nMaxFileTitle = MAX_PATH;

+					if ( GetOpenFileName( &ofn )) {

+						int i;

+						i = AddDatabaseToList( GetDlgItem(hdlg,IDC_DBLIST), str, _T("") );

+						if ( i == -1 )

+							i=0;

+						ListView_SetItemState( GetDlgItem(hdlg,IDC_DBLIST), i, LVIS_SELECTED, LVIS_SELECTED );

+					}

+					break;

+				}

+				case IDC_BACK:

+					SendMessage(GetParent(hdlg),WZM_GOTOPAGE,IDD_WELCOME,(LPARAM)WelcomeDlgProc);

+					break;

+				case IDOK:

+					opts.hFile = CreateFile( opts.filename, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL );

+					if ( opts.hFile == INVALID_HANDLE_VALUE ) {

+						opts.hFile = NULL;

+						opts.error = GetLastError();

+						SendMessage( GetParent(hdlg), WZM_GOTOPAGE, IDD_OPENERROR, ( LPARAM )OpenErrorDlgProc );

+					}

+					else SendMessage( GetParent(hdlg), WZM_GOTOPAGE, IDD_FILEACCESS, (LPARAM)FileAccessDlgProc );

+					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,(TCHAR*)lvi.lParam);

+							SendMessage(hdlg,WM_COMMAND,MAKEWPARAM(IDC_FILE,EN_CHANGE),(LPARAM)GetDlgItem(hdlg,IDC_FILE));

+							break;

+						}

+					}

+					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;

+}