ChordEase Code

// Copyleft 2008 Chris Korda
// 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 any later version.
/*
        chris korda
 
		revision history:
		rev		date	comments
        00      14feb08	initial version
        01      06jan10	W64: in DeleteDirectory, use size_t
        02      13jan12	add GetTempPath and GetAppPath
		03		21nov12	add DockControlBarLeftOf
		04		30nov12	add UpdateMenu
        05      17apr13	add temporary files folder
		06		21may13	add GetSpecialFolderPath
		07		10jul13	add GetLastErrorString
		08		19nov13	in EnableChildWindows, add Deep argument
		09		07may15	SHGetSpecialFolderPath return value is BOOL, not HRESULT

        enhanced application
 
*/

// WinAppCK.cpp : Defines the class behaviors for the application.
//

#include "stdafx.h"
#include "Resource.h"
#include "WinAppCK.h"
#include "PathStr.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

// MakeSureDirectoryPathExists doesn't support Unicode; SHCreateDirectoryEx
// is a reasonable substitute, but older versions of the SDK don't define it
#if defined(UNICODE)
#ifndef	SHCreateDirectoryEx
int WINAPI SHCreateDirectoryExW(HWND hwnd, LPCWSTR pszPath, const SECURITY_ATTRIBUTES *psa);
#define SHCreateDirectoryEx SHCreateDirectoryExW
#define SHCreateDirectoryProxy
#endif
#else
#include "imagehlp.h"	// for MakeSureDirectoryPathExists
#endif

/////////////////////////////////////////////////////////////////////////////
// CWinAppCK

BEGIN_MESSAGE_MAP(CWinAppCK, CWinApp)
	//{{AFX_MSG_MAP(CWinAppCK)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWinAppCK construction

bool CWinAppCK::GetTempPath(CString& Path)
{
	if (!m_TempFolderPath.IsEmpty()) {	// if temporary files folder specified
		if (PathFileExists(m_TempFolderPath)) {	// if folder exists
			Path = m_TempFolderPath;
			return(TRUE);
		}
		CString	msg;
		AfxFormatString1(msg, IDS_APP_TEMP_FOLDER_NOT_FOUND, m_TempFolderPath);
		AfxMessageBox(msg);	// warn user
	}
	LPTSTR	pBuf = Path.GetBuffer(MAX_PATH);
	DWORD	retc = ::GetTempPath(MAX_PATH, pBuf);
	Path.ReleaseBuffer();
	return(retc != 0);
}

bool CWinAppCK::GetTempFileName(CString& Path, LPCTSTR Prefix, UINT Unique)
{
	CString	TempPath;
	if (!GetTempPath(TempPath))
		return(FALSE);
	if (Prefix == NULL)
		Prefix = m_pszAppName;
	LPTSTR	pBuf = Path.GetBuffer(MAX_PATH);
	DWORD	retc = ::GetTempFileName(TempPath, Prefix, Unique, pBuf);
	Path.ReleaseBuffer();
	return(retc != 0);
}

void CWinAppCK::GetCurrentDirectory(CString& Path)
{
	LPTSTR	pBuf = Path.GetBuffer(MAX_PATH);
	::GetCurrentDirectory(MAX_PATH, pBuf);
	Path.ReleaseBuffer();
}

bool CWinAppCK::GetSpecialFolderPath(int FolderID, CString& Path)
{
	LPTSTR	p = Path.GetBuffer(MAX_PATH);
	bool	retc = SHGetSpecialFolderPath(NULL, p, FolderID, 0) != 0;
	Path.ReleaseBuffer();
	return(retc);
}

bool CWinAppCK::GetAppDataFolder(CString& Folder) const
{
	CPathStr	path;
	if (!GetSpecialFolderPath(CSIDL_APPDATA, path))
		return(FALSE);
	path.Append(m_pszAppName);
	Folder = path;
	return(TRUE);
}

CString CWinAppCK::GetAppPath()
{
	CString	s = GetCommandLine();
	s.TrimLeft();	// trim leading whitespace just in case
	if (s[0] == '"')	// if first char is a quote
		s = s.Mid(1).SpanExcluding(_T("\""));	// span to next quote
	else
		s = s.SpanExcluding(_T(" \t"));	// span to next whitespace
	return(s);
}

CString CWinAppCK::GetAppFolder()
{
	CPathStr	path(GetAppPath());
	path.RemoveFileSpec();
	return(path);
}

bool CWinAppCK::GetIconSize(HICON Icon, CSize& Size)
{
	ICONINFO	IconInfo;
	if (!GetIconInfo(Icon, &IconInfo))
		return(FALSE);
	BITMAP	bmp;
	bool	retc = GetObject(IconInfo.hbmColor, sizeof(bmp), &bmp) != 0;
	if (retc)
		Size = CSize(bmp.bmWidth, bmp.bmHeight);
	DeleteObject(IconInfo.hbmColor);
	DeleteObject(IconInfo.hbmMask);
	return(retc);
}

void CWinAppCK::UpdateAllViews(CView* pSender, LPARAM lHint, CObject* pHint)
{
	POSITION	TplPos = GetFirstDocTemplatePosition();
	while (TplPos != NULL) {	// for each document template
		CDocTemplate	*pTpl = GetNextDocTemplate(TplPos);
		POSITION	DocPos = pTpl->GetFirstDocPosition();
		while (DocPos != NULL) {	// for each document, update all its views
			CDocument	*pDoc = pTpl->GetNextDoc(DocPos);
			pDoc->UpdateAllViews(pSender, lHint, pHint);
		}
	}
}

bool CWinAppCK::DeleteDirectory(LPCTSTR lpszDir, bool bAllowUndo)
{
	size_t	len = _tcslen(lpszDir);
	TCHAR	*pszFrom = new TCHAR[len + 2];
	_tcscpy(pszFrom, lpszDir);
	pszFrom[len] = 0;
	pszFrom[len + 1] = 0;
	SHFILEOPSTRUCT fileop;
	fileop.hwnd   = NULL;	// no status display
	fileop.wFunc  = FO_DELETE;	// delete operation
	fileop.pFrom  = pszFrom;	// source file name as double null terminated string
	fileop.pTo    = NULL;	// no destination needed
	fileop.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;  // do not prompt the user
	if (bAllowUndo)
		fileop.fFlags |= FOF_ALLOWUNDO;
	fileop.fAnyOperationsAborted = FALSE;
	fileop.lpszProgressTitle = NULL;
	fileop.hNameMappings = NULL;
	int ret = SHFileOperation(&fileop);
	delete [] pszFrom;  
	return(ret == 0);
}

void CWinAppCK::EnableChildWindows(CWnd& Wnd, bool Enable, bool Deep)
{
	CWnd	*wp = Wnd.GetWindow(GW_CHILD);
	while (wp != NULL) {
		if (Deep)	// if recursively searching all children
			EnableChildWindows(*wp, Enable);
		wp->EnableWindow(Enable);
		wp = wp->GetNextWindow();
	}
}

CString CWinAppCK::GetTitleFromPath(LPCTSTR Path)
{
	CPathStr	s(PathFindFileName(Path));
	s.RemoveExtension();
	return(s);
}

bool CWinAppCK::GetComputerName(CString& Name)
{
	DWORD	len = MAX_COMPUTERNAME_LENGTH + 1;
	LPTSTR	p = Name.GetBuffer(len);
	bool	retc = ::GetComputerName(p, &len) != 0;
	Name.ReleaseBuffer(len);
	return(retc);
}

// MakeSureDirectoryPathExists doesn't support Unicode; SHCreateDirectoryEx
// is a reasonable substitute, but older versions of the SDK don't define it
#ifdef SHCreateDirectoryProxy
int WINAPI SHCreateDirectoryExW(HWND hwnd, LPCWSTR pszPath, const SECURITY_ATTRIBUTES *psa)
{
	int	retc = ERROR_INVALID_FUNCTION;
	typedef int (WINAPI* lpfnSHCreateDirectoryExW)(HWND hwnd, LPCWSTR pszPath, const SECURITY_ATTRIBUTES *psa);
	HMODULE hShell = LoadLibrary(_T("shell32.dll"));
	lpfnSHCreateDirectoryExW lpfn = NULL;
	if (hShell) {
		lpfn = (lpfnSHCreateDirectoryExW)GetProcAddress(hShell, "SHCreateDirectoryExW");
		if (lpfn)
			retc = lpfn(hwnd, pszPath, psa);
		FreeLibrary(hShell);
	}
	return(retc);
}
#endif

bool CWinAppCK::CreateFolder(LPCTSTR Path)
{
#ifdef UNICODE
	return(SHCreateDirectoryEx(NULL, Path, NULL) == ERROR_SUCCESS);
#else
	CString	s(Path);
	LPTSTR	p = s.GetBuffer(MAX_PATH);
	PathAddBackslash(p);	// trailing backslash is required
	s.ReleaseBuffer();
	return(MakeSureDirectoryPathExists(s) != 0);	// very slow
#endif
}

// This function fixes the dot bitmap CCmdUI::SetRadio uses in radio-style menu
// items, which is too small if the menu font is large; taken from MSJ Jan 1998
void CWinAppCK::FixMFCDotBitmap()
{
	HBITMAP	hbmDot = GetMFCDotBitmap();
	if (hbmDot) {
		// Draw a centered dot of appropriate size
		BITMAP	bm;
		::GetObject(hbmDot, sizeof(bm), &bm);
		CRect rcDot(0, 0, bm.bmWidth, bm.bmHeight);
		rcDot.DeflateRect((bm.bmWidth >> 1) - 2, (bm.bmHeight >> 1) - 2);
		CWindowDC	dcScreen(NULL);
		CDC	memdc;
		memdc.CreateCompatibleDC(&dcScreen);
		int	nSave = memdc.SaveDC();
		memdc.SelectStockObject(BLACK_PEN);
		memdc.SelectStockObject(BLACK_BRUSH);
		memdc.SelectObject((HGDIOBJ)hbmDot);
		memdc.PatBlt(0, 0, bm.bmWidth, bm.bmHeight, WHITENESS);
		memdc.Ellipse(&rcDot);
		memdc.RestoreDC(nSave);
	}
}

HBITMAP CWinAppCK::GetMFCDotBitmap()
{
	// The bitmap is stored in afxData.hbmMenuDot, but afxData is MFC-private,
	// so the only way to get it is create a menu, set the radio check,
	// and then see what bitmap MFC set in the menu item.
	CMenu	menu;
	VERIFY(menu.CreateMenu());
	VERIFY(menu.AppendMenu(MFT_STRING, 0, (LPCTSTR)NULL));
	CCmdUI	cui;
	cui.m_pMenu = &menu;
	cui.m_nIndex = 0;
	cui.m_nIndexMax = 1;
	cui.SetRadio(TRUE);
	MENUITEMINFO	info;
	ZeroMemory(&info, sizeof(info));
	info.cbSize = sizeof(info);
	info.fMask = MIIM_CHECKMARKS;
	GetMenuItemInfo(menu, 0, MF_BYPOSITION, &info);
	HBITMAP hbmDot = info.hbmpChecked;
	menu.DestroyMenu();
	return hbmDot;
}

void CWinAppCK::DockControlBarLeftOf(CFrameWnd *Frame, CControlBar *Bar, CControlBar *LeftOf)
{
	UINT n = 0;
	CRect	rect, pRect;
	if (LeftOf != NULL) {
		// Get MFC to adjust the dimensions of all docked ToolBars
		// so that GetWindowRect will be accurate.
		Frame->RecalcLayout(TRUE);
		LeftOf->GetWindowRect(&rect);
		rect.OffsetRect(1, 0);
		DWORD	dw = LeftOf->GetBarStyle();
		n = (dw & CBRS_ALIGN_TOP)				? AFX_IDW_DOCKBAR_TOP : n;
		n = ((dw & CBRS_ALIGN_BOTTOM) && !n)	? AFX_IDW_DOCKBAR_BOTTOM : n;
		n = ((dw & CBRS_ALIGN_LEFT) && !n)		? AFX_IDW_DOCKBAR_LEFT : n;
		n = ((dw & CBRS_ALIGN_RIGHT) && !n)		? AFX_IDW_DOCKBAR_RIGHT : n;
		// When we take the default parameters on rect, DockControlBar will dock
		// each Toolbar on a separate line. By calculating a rectangle, we
		// are simulating a Toolbar being dragged to that location and docked.
		pRect = ▭
	} else	// left of wasn't specified
		pRect = NULL;
	Frame->DockControlBar(Bar, n, pRect);
}

void CWinAppCK::UpdateMenu(CWnd *pWnd, CMenu *pMenu)
{
	CCmdUI	cui;
	cui.m_pMenu = pMenu;
	cui.m_nIndexMax = pMenu->GetMenuItemCount();
	for (UINT i = 0; i < cui.m_nIndexMax; i++) {
		cui.m_nID = pMenu->GetMenuItemID(i);
		if (!cui.m_nID)	// separator
			continue;
		if (cui.m_nID == -1) {	// popup submenu
			CMenu	*pSubMenu = pMenu->GetSubMenu(i);
			if (pSubMenu != NULL)
				UpdateMenu(pWnd, pSubMenu);	// recursive call
		}
		cui.m_nIndex = i;
		cui.m_pMenu = pMenu;
		cui.DoUpdate(pWnd, FALSE);
	}
}

CString CWinAppCK::FormatSystemError(DWORD ErrorCode)
{
	LPTSTR	lpszTemp;
	DWORD	bRet = FormatMessage(
		FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
		NULL, ErrorCode, 0, (LPTSTR)&lpszTemp, 0, NULL);	// default language
	CString	sError;	
	if (bRet) {	// if format succeeded
		sError = lpszTemp;	// create string from message buffer
		LocalFree(lpszTemp);	// free message buffer
	} else	// format failed
		sError.Format(IDS_APP_UNKNOWN_SYSTEM_ERROR, ErrorCode, ErrorCode);
	return(sError);
}

CString	CWinAppCK::GetLastErrorString()
{
	return(FormatSystemError(GetLastError()));
}