934 lines
30 KiB
C++
934 lines
30 KiB
C++
#include "shellprv.h"
|
|
#include "ids.h"
|
|
#include "util.h"
|
|
#include "datautil.h"
|
|
#include "foldertypes.h"
|
|
#include "basefvcb.h"
|
|
|
|
#define PROPSTR_LOGO L"Logo"
|
|
|
|
typedef struct
|
|
{
|
|
UINT uIDFriendly;
|
|
LPCTSTR pszFolderType;
|
|
DWORD dwFlags;
|
|
} WEBVIEWTEMPLATEINFO;
|
|
|
|
#define WVTI_SHOWIFOLDTEMPLATE 0x00000001
|
|
|
|
// documents must be first.
|
|
const WEBVIEWTEMPLATEINFO c_wvtiList[] =
|
|
{
|
|
{ IDS_CUSTOMIZE_USELEGACYHTT, STR_TYPE_USELEGACYHTT, WVTI_SHOWIFOLDTEMPLATE },
|
|
{ IDS_CUSTOMIZE_DOCUMENTS, STR_TYPE_DOCUMENTS, 0 },
|
|
{ IDS_CUSTOMIZE_PICTURES, STR_TYPE_PICTURES, 0 },
|
|
{ IDS_CUSTOMIZE_PHOTOALBUM, STR_TYPE_PHOTOALBUM, 0 },
|
|
{ IDS_CUSTOMIZE_MUSIC, STR_TYPE_MUSIC, 0 },
|
|
{ IDS_CUSTOMIZE_MUSICARTIST, STR_TYPE_MUSICARTIST, 0 },
|
|
{ IDS_CUSTOMIZE_MUSICALBUM, STR_TYPE_MUSICALBUM, 0 },
|
|
{ IDS_CUSTOMIZE_VIDEOS, STR_TYPE_VIDEOS, 0 },
|
|
// note: are these gonna happen?
|
|
// { IDS_CUSTOMIZE_VIDEOALBUM, STR_TYPE_VIDEOALBUM, 0 },
|
|
// { IDS_CUSTOMIZE_BOOKS, STR_TYPE_BOOKS, 0 }
|
|
};
|
|
|
|
typedef enum
|
|
{
|
|
FOLDERCUST_MODE_GENERATING,
|
|
FOLDERCUST_MODE_ICON,
|
|
FOLDERCUST_MODE_BITMAP
|
|
} FOLDERCUSTMODE;
|
|
|
|
class CFolderCustomize : public IShellExtInit,
|
|
public IShellPropSheetExt
|
|
{
|
|
public:
|
|
// IUnknown
|
|
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj);
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
|
STDMETHODIMP_(ULONG) Release(void);
|
|
|
|
// IShellExtInit
|
|
STDMETHODIMP Initialize(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID);
|
|
|
|
// IShellPropSheetExt
|
|
STDMETHODIMP AddPages(LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam);
|
|
STDMETHODIMP ReplacePage(UINT uPageID, LPFNADDPROPSHEETPAGE lpfnReplaceWith, LPARAM lParam)
|
|
{ return S_OK; };
|
|
|
|
CFolderCustomize();
|
|
|
|
private:
|
|
~CFolderCustomize();
|
|
static UINT CALLBACK _PrshtCallback(HWND hwnd, UINT uMsg, PROPSHEETPAGE *ppsp);
|
|
void _SetRecurseBox(HWND hwnd);
|
|
void _HideIconSection(HWND hwnd);
|
|
void _InitDialog(HWND hwnd);
|
|
BOOL _HandleWMCommand(HWND hwndDlg, WORD wNotify, WORD wID, HWND hwndCtrl);
|
|
void _EnableApply(HWND hwnd);
|
|
|
|
static DWORD WINAPI _ExtractThreadProc(void *pv);
|
|
HRESULT _ExtractOnSeparateThread(IPropertyBag *ppb, HWND hwndDlg);
|
|
HRESULT _CreateThumbnailBitmap(HWND hwndDlg);
|
|
HRESULT _CreateFolderIcon(HWND hwndDlg);
|
|
void _SetThumbnail(HWND hwnd);
|
|
void _FreeDlgItems(HWND hwndDlg);
|
|
void _SetPreviewToNewState(HWND hwndDlg, FOLDERCUSTMODE fcMode, HBITMAP hbitmap, HICON hicon);
|
|
|
|
BOOL _ShouldEnableChangeOfIcon();
|
|
void _ChangeFolderIcon(HWND hwndDlg);
|
|
HRESULT _ProcessIconChange(LPCTSTR pszPickIconDialogCaption, HWND hwndDlg);
|
|
|
|
void _DirTouch(LPITEMIDLIST pidl);
|
|
void _DeleteCustomizationInBag(IPropertyBag *ppb);
|
|
BOOL _NotifyAboutWebView(HWND hwnd);
|
|
static BOOL CALLBACK _RefreshView(HWND hwnd, LPCITEMIDLIST pidl, LPARAM lParam);
|
|
void _RefreshWindows(BOOL fTurnOnWebView, BOOL fApplyToChildren);
|
|
HRESULT _ApplyChangesToBag(HWND hwndDlg, IPropertyBag *ppb);
|
|
HRESULT _ApplyChanges(HWND hwndDlg);
|
|
|
|
void _UpdateViewState(HWND hwndDlg, IPropertyBag *ppb, int iIndex);
|
|
void _FillTemplateComboBox(HWND hwndTemplates);
|
|
int _GetTemplateIndexFromType(LPCTSTR pszType);
|
|
|
|
static BOOL_PTR CALLBACK _DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
|
|
|
LONG _cRef;
|
|
|
|
LPITEMIDLIST _pidl;
|
|
IPropertyBag *_ppb;
|
|
|
|
// used for background thread extraction
|
|
HWND _hwnd;
|
|
IPropertyBag *_ppbBackground;
|
|
|
|
// cached info
|
|
HBITMAP _hbmDefault;
|
|
HBITMAP _hbmLogo;
|
|
TCHAR _szCachedLogoFile[MAX_PATH];
|
|
|
|
BOOL _fUsingThumb;
|
|
|
|
ICustomIconManager *_pIconManager;
|
|
TCHAR _szLogoFile[MAX_PATH];
|
|
TCHAR _szIconPath[MAX_PATH];
|
|
int _iIconIndex;
|
|
HRESULT _hrFromIconChange;
|
|
|
|
};
|
|
|
|
CFolderCustomize::CFolderCustomize() : _cRef(1), _hrFromIconChange(E_FAIL)
|
|
{
|
|
}
|
|
|
|
CFolderCustomize::~CFolderCustomize()
|
|
{
|
|
ILFree(_pidl);
|
|
if (_ppb)
|
|
_ppb->Release();
|
|
if (_pIconManager)
|
|
_pIconManager->Release();
|
|
if (_ppbBackground)
|
|
_ppbBackground->Release();
|
|
if (_hbmDefault)
|
|
DeleteObject(_hbmDefault);
|
|
if (_hbmLogo)
|
|
DeleteObject(_hbmLogo);
|
|
}
|
|
|
|
STDAPI CFolderCustomize_CreateInstance(IUnknown *punkOuter, REFIID riid, void **ppvOut)
|
|
{
|
|
// aggregation checking is handled in class factory
|
|
|
|
HRESULT hr = E_OUTOFMEMORY;
|
|
CFolderCustomize* pfc = new CFolderCustomize();
|
|
if (pfc)
|
|
{
|
|
hr = pfc->QueryInterface(riid, ppvOut);
|
|
pfc->Release();
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CFolderCustomize::QueryInterface(REFIID riid, void **ppvObj)
|
|
{
|
|
static const QITAB qit[] = {
|
|
QITABENT(CFolderCustomize, IShellExtInit),
|
|
QITABENT(CFolderCustomize, IShellPropSheetExt),
|
|
{ 0 }
|
|
};
|
|
return QISearch(this, qit, riid, ppvObj);
|
|
}
|
|
|
|
ULONG CFolderCustomize::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
ULONG CFolderCustomize::Release()
|
|
{
|
|
if (InterlockedDecrement(&_cRef))
|
|
return _cRef;
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
STDMETHODIMP CFolderCustomize::Initialize(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID)
|
|
{
|
|
HRESULT hr;
|
|
if (!pidlFolder)
|
|
{
|
|
hr = PidlFromDataObject(pdtobj, &_pidl);
|
|
}
|
|
else
|
|
{
|
|
hr = Pidl_Set(&_pidl, pidlFolder) ? S_OK : E_OUTOFMEMORY;
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = SHGetViewStatePropertyBag(_pidl, VS_BAGSTR_EXPLORER, SHGVSPB_PERUSER | SHGVSPB_PERFOLDER, IID_PPV_ARG(IPropertyBag, &_ppb));
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
// from defview.cpp
|
|
BOOL IsCustomizable(LPCITEMIDLIST pidlFolder);
|
|
|
|
UINT CALLBACK CFolderCustomize::_PrshtCallback(HWND hwnd, UINT uMsg, PROPSHEETPAGE *ppsp)
|
|
{
|
|
if (uMsg == PSPCB_RELEASE)
|
|
{
|
|
((CFolderCustomize *)ppsp->lParam)->Release();
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
STDMETHODIMP CFolderCustomize::AddPages(LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
if (IsCustomizable(_pidl))
|
|
{
|
|
PROPSHEETPAGE psp = {0};
|
|
psp.dwSize = sizeof(psp);
|
|
psp.dwFlags = PSP_USECALLBACK;
|
|
psp.hInstance = HINST_THISDLL;
|
|
psp.pszTemplate = MAKEINTRESOURCE(DLG_FOLDER_CUSTOMIZE);
|
|
psp.pfnDlgProc = _DlgProc;
|
|
psp.pfnCallback = _PrshtCallback;
|
|
psp.lParam = (LPARAM)this;
|
|
|
|
HPROPSHEETPAGE hpsp = CreatePropertySheetPage(&psp);
|
|
if (hpsp)
|
|
{
|
|
AddRef(); // HPROPSHEETPAGE holds ref, released on _PrshtCallback
|
|
if (!pfnAddPage(hpsp, lParam))
|
|
{
|
|
DestroyPropertySheetPage(hpsp);
|
|
}
|
|
else
|
|
{
|
|
hr = S_OK;
|
|
}
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
#define IDH_FOLDER_TEMPLATES 10005
|
|
#define IDH_FOLDER_RECURSE 10006
|
|
#define IDH_FOLDER_PICKBROWSE 10007
|
|
#define IDH_FOLDER_DEFAULT 10008
|
|
#define IDH_FOLDER_CHANGEICON 10009
|
|
|
|
const static DWORD aPrshtHelpIDs[] =
|
|
{
|
|
IDC_FOLDER_TEMPLATES, IDH_FOLDER_TEMPLATES,
|
|
IDC_FOLDER_RECURSE, IDH_FOLDER_RECURSE,
|
|
IDC_FOLDER_PICKBROWSE, IDH_FOLDER_PICKBROWSE,
|
|
IDC_FOLDER_DEFAULT, IDH_FOLDER_DEFAULT,
|
|
IDC_FOLDER_CHANGEICON, IDH_FOLDER_CHANGEICON,
|
|
IDC_FOLDER_PREVIEW_ICON, NO_HELP,
|
|
IDC_FOLDER_PREVIEW_BITMAP, NO_HELP,
|
|
IDC_FOLDER_ICON, NO_HELP,
|
|
IDC_FOLDER_CHANGEICONTEXT, NO_HELP,
|
|
IDC_FOLDER_CHANGEICONGROUP, NO_HELP,
|
|
IDC_NO_HELP_1, NO_HELP,
|
|
IDC_NO_HELP_2, NO_HELP,
|
|
0, 0
|
|
};
|
|
|
|
BOOL_PTR CALLBACK CFolderCustomize::_DlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
BOOL fRet = FALSE;
|
|
CFolderCustomize *pfc = (CFolderCustomize*)GetWindowLongPtr(hwnd, DWLP_USER);
|
|
|
|
switch (uMsg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
pfc = (CFolderCustomize*)((PROPSHEETPAGE *)lParam)->lParam;
|
|
SetWindowLongPtr(hwnd, DWLP_USER, (LONG_PTR)pfc);
|
|
pfc->_InitDialog(hwnd);
|
|
break;
|
|
|
|
case WM_HELP:
|
|
WinHelp((HWND)((LPHELPINFO)lParam)->hItemHandle, L"filefold.hlp", HELP_WM_HELP, (ULONG_PTR)(LPTSTR) aPrshtHelpIDs);
|
|
break;
|
|
|
|
case WM_CONTEXTMENU:
|
|
WinHelp((HWND)wParam, L"filefold.hlp", HELP_CONTEXTMENU, (ULONG_PTR)(void *)aPrshtHelpIDs);
|
|
break;
|
|
|
|
case WM_COMMAND:
|
|
fRet = pfc->_HandleWMCommand(hwnd, HIWORD(wParam), LOWORD(wParam), (HWND)lParam);
|
|
break;
|
|
|
|
case WM_NOTIFY:
|
|
if (((LPNMHDR)lParam)->code == PSN_APPLY)
|
|
{
|
|
pfc->_ApplyChanges(hwnd);
|
|
}
|
|
fRet = TRUE;
|
|
break;
|
|
|
|
case WM_DESTROY:
|
|
pfc->_FreeDlgItems(hwnd);
|
|
break;
|
|
}
|
|
return fRet;
|
|
}
|
|
|
|
void CFolderCustomize::_FreeDlgItems(HWND hwndDlg)
|
|
{
|
|
HICON hicon = (HICON)SendDlgItemMessage(hwndDlg, IDC_FOLDER_PREVIEW_ICON, STM_SETICON, NULL, NULL);
|
|
if (hicon)
|
|
DestroyIcon(hicon);
|
|
HBITMAP hbitmap = (HBITMAP)SendDlgItemMessage(hwndDlg, IDC_FOLDER_PREVIEW_BITMAP, STM_SETIMAGE, (WPARAM)IMAGE_BITMAP, NULL);
|
|
if (hbitmap)
|
|
DeleteObject(hbitmap);
|
|
ReplaceDlgIcon(hwndDlg, IDC_FOLDER_ICON, NULL);
|
|
}
|
|
|
|
void CFolderCustomize::_SetPreviewToNewState(HWND hwndDlg, FOLDERCUSTMODE fcMode, HBITMAP hbitmap, HICON hicon)
|
|
{
|
|
// if fcMode == FOLDERCUST_MODE_ICON, we need hicon and not hbitmap
|
|
// if fcMode == FOLDERCUST_MODE_BITMAP, we need hbitmap and not hicon.
|
|
// otherwise we dont want either.
|
|
ASSERT((fcMode != FOLDERCUST_MODE_ICON) || (hicon && !hbitmap));
|
|
ASSERT((fcMode != FOLDERCUST_MODE_BITMAP) || (!hicon && hbitmap));
|
|
ASSERT((fcMode != FOLDERCUST_MODE_GENERATING) || (!hicon && !hbitmap));
|
|
|
|
switch (fcMode)
|
|
{
|
|
case FOLDERCUST_MODE_GENERATING:
|
|
{
|
|
TCHAR szText[100];
|
|
LoadString(HINST_THISDLL, IDS_CUSTOMIZE_GENERATING, szText, ARRAYSIZE(szText));
|
|
SetWindowText(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_TEXT), szText);
|
|
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_TEXT), SW_SHOW);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_ICON), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_BITMAP), SW_HIDE);
|
|
}
|
|
break;
|
|
|
|
case FOLDERCUST_MODE_ICON:
|
|
{
|
|
HICON hiconOld = (HICON)SendDlgItemMessage(hwndDlg, IDC_FOLDER_PREVIEW_ICON, STM_SETICON, NULL, NULL);
|
|
if (hiconOld)
|
|
DestroyIcon(hiconOld);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_TEXT), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_ICON), SW_SHOW);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_BITMAP), SW_HIDE);
|
|
|
|
SendDlgItemMessage(hwndDlg, IDC_FOLDER_PREVIEW_ICON, STM_SETIMAGE, IMAGE_ICON, (LPARAM)hicon);
|
|
}
|
|
break;
|
|
|
|
case FOLDERCUST_MODE_BITMAP:
|
|
{
|
|
HBITMAP hbitmapOld = (HBITMAP)SendDlgItemMessage(hwndDlg, IDC_FOLDER_PREVIEW_BITMAP, STM_SETIMAGE, (WPARAM)IMAGE_BITMAP, NULL);
|
|
if (hbitmapOld)
|
|
DeleteObject(hbitmapOld);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_TEXT), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_ICON), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_PREVIEW_BITMAP), SW_SHOW);
|
|
|
|
SendDlgItemMessage(hwndDlg, IDC_FOLDER_PREVIEW_BITMAP, STM_SETIMAGE, IMAGE_BITMAP, (LPARAM)hbitmap);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
HRESULT CFolderCustomize::_CreateFolderIcon(HWND hwndDlg)
|
|
{
|
|
IExtractIcon *peic;
|
|
HRESULT hr = SHGetUIObjectFromFullPIDL(_pidl, NULL, IID_PPV_ARG(IExtractIcon, &peic));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
TCHAR szPath[MAX_PATH];
|
|
INT iIndex;
|
|
UINT wFlags;
|
|
hr = peic->GetIconLocation(0, szPath, ARRAYSIZE(szPath), &iIndex, &wFlags);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
UINT nIconSize = MAKELONG(32, 32); // 32 for both large and small
|
|
HICON hiconLarge;
|
|
hr = peic->Extract(szPath, iIndex, NULL, &hiconLarge, nIconSize);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ReplaceDlgIcon(hwndDlg, IDC_FOLDER_ICON, hiconLarge);
|
|
}
|
|
}
|
|
peic->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
DWORD WINAPI CFolderCustomize::_ExtractThreadProc(void *pv)
|
|
{
|
|
CFolderCustomize *pfc = (CFolderCustomize*)pv;
|
|
|
|
pfc->_SetPreviewToNewState(pfc->_hwnd, FOLDERCUST_MODE_GENERATING, NULL, NULL);
|
|
|
|
IExtractImage *pei;
|
|
HRESULT hr = SHGetUIObjectFromFullPIDL(pfc->_pidl, NULL, IID_PPV_ARG(IExtractImage, &pei));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = SHLoadFromPropertyBag(pei, pfc->_ppbBackground);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
TCHAR szPath[MAX_PATH];
|
|
SIZE sz = {96, 96};
|
|
DWORD dwFlags = IEIFLAG_QUALITY;
|
|
hr = pei->GetLocation(szPath, ARRAYSIZE(szPath), NULL, &sz, 24, &dwFlags);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
HBITMAP hbitmap;
|
|
hr = pei->Extract(&hbitmap);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
pfc->_SetPreviewToNewState(pfc->_hwnd, FOLDERCUST_MODE_BITMAP, hbitmap, NULL);
|
|
|
|
TCHAR szLogo[MAX_PATH];
|
|
if (SUCCEEDED(SHPropertyBag_ReadStr(pfc->_ppbBackground, PROPSTR_LOGO, szLogo, ARRAYSIZE(szLogo))))
|
|
{
|
|
HBITMAP *phbm = szLogo[0] ? &pfc->_hbmLogo : &pfc->_hbmDefault;
|
|
if (*phbm)
|
|
DeleteObject(*phbm);
|
|
*phbm = (HBITMAP)CopyImage(hbitmap, IMAGE_BITMAP, 0, 0, 0);
|
|
|
|
if (szLogo[0])
|
|
{
|
|
lstrcpyn(pfc->_szCachedLogoFile, szLogo, ARRAYSIZE(pfc->_szCachedLogoFile));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
pei->Release();
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
// IExtractImage on a folder without any jpegs inside will fail.
|
|
// in that case we need IExtractIcon.
|
|
IExtractIcon *peic;
|
|
hr = SHGetUIObjectFromFullPIDL(pfc->_pidl, NULL, IID_PPV_ARG(IExtractIcon, &peic));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
TCHAR szPath[MAX_PATH];
|
|
INT iIndex;
|
|
UINT wFlags;
|
|
hr = peic->GetIconLocation(0, szPath, ARRAYSIZE(szPath), &iIndex, &wFlags);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
UINT nIconSize = MAKELONG(96, 96); // 96 for both large and small
|
|
HICON hiconLarge;
|
|
hr = peic->Extract(szPath, iIndex, NULL, &hiconLarge, nIconSize);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
pfc->_SetPreviewToNewState(pfc->_hwnd, FOLDERCUST_MODE_ICON, NULL, hiconLarge);
|
|
}
|
|
}
|
|
peic->Release();
|
|
}
|
|
}
|
|
|
|
pfc->Release(); // this thread holds a ref
|
|
return 0;
|
|
}
|
|
|
|
HRESULT CFolderCustomize::_ExtractOnSeparateThread(IPropertyBag *ppb, HWND hwndDlg)
|
|
{
|
|
HRESULT hr = E_OUTOFMEMORY;
|
|
|
|
IUnknown_Set((IUnknown**)&_ppbBackground, ppb);
|
|
_hwnd = hwndDlg;
|
|
|
|
AddRef();
|
|
if (SHCreateThread(_ExtractThreadProc, this, CTF_COINIT, NULL))
|
|
{
|
|
hr = S_OK;
|
|
}
|
|
else
|
|
{
|
|
Release(); // thread failed to take ref
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CFolderCustomize::_CreateThumbnailBitmap(HWND hwndDlg)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
// see if the bitmap is one we've already extracted.
|
|
// can't use the thumbs.db cache for this kind of stuff, since the changes
|
|
// havent been committed yet we really shouldnt be poking around in data.
|
|
if (!_fUsingThumb && _hbmDefault)
|
|
{
|
|
_SetPreviewToNewState(hwndDlg, FOLDERCUST_MODE_BITMAP, (HBITMAP)CopyImage(_hbmDefault, IMAGE_BITMAP, 0, 0, 0), NULL);
|
|
}
|
|
else if (_fUsingThumb && _hbmLogo && (lstrcmpi(_szLogoFile, _szCachedLogoFile) == 0))
|
|
{
|
|
_SetPreviewToNewState(hwndDlg, FOLDERCUST_MODE_BITMAP, (HBITMAP)CopyImage(_hbmLogo, IMAGE_BITMAP, 0, 0, 0), NULL);
|
|
}
|
|
else
|
|
{
|
|
// cache miss, figure it out again.
|
|
IPropertyBag *ppb;
|
|
hr = SHCreatePropertyBagOnMemory(STGM_READWRITE, IID_PPV_ARG(IPropertyBag, &ppb));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = SHPropertyBag_WriteStr(ppb, PROPSTR_LOGO, _fUsingThumb ? _szLogoFile : TEXT(""));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = _ExtractOnSeparateThread(ppb, hwndDlg);
|
|
}
|
|
ppb->Release();
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
// dont want OFN_NODEREFERENCELINKS so use the rundlg.cpp helper directly
|
|
STDAPI_(BOOL) _GetFileNameFromBrowse(HWND hwnd, LPTSTR szFilePath, UINT cbFilePath, LPCTSTR szWorkingDir, LPCTSTR szDefExt, LPCTSTR szFilters, LPCTSTR szTitle, DWORD dwFlags);
|
|
BOOL CFolderCustomize::_HandleWMCommand(HWND hwndDlg, WORD wNotify, WORD wID, HWND hwndCtrl)
|
|
{
|
|
switch(wID)
|
|
{
|
|
case IDC_FOLDER_TEMPLATES:
|
|
if (wNotify == LBN_SELCHANGE)
|
|
{
|
|
_EnableApply(hwndDlg);
|
|
}
|
|
break;
|
|
|
|
case IDC_FOLDER_DEFAULT:
|
|
_EnableApply(hwndDlg);
|
|
_fUsingThumb = FALSE;
|
|
_CreateThumbnailBitmap(hwndDlg);
|
|
break;
|
|
|
|
case IDC_FOLDER_CHANGEICON:
|
|
_ChangeFolderIcon(hwndDlg);
|
|
break;
|
|
|
|
case IDC_FOLDER_PICKBROWSE:
|
|
TCHAR szFilePath[MAX_PATH] = {0};
|
|
TCHAR szInitialDir[MAX_PATH] = {0};
|
|
|
|
// initial directory is current folder
|
|
// todo: load supported file types at runtime
|
|
if (SHGetPathFromIDList(_pidl, szInitialDir) &&
|
|
_GetFileNameFromBrowse(hwndDlg, szFilePath, ARRAYSIZE(szFilePath), szInitialDir,
|
|
MAKEINTRESOURCE(IDS_IMAGES), MAKEINTRESOURCE(IDS_IMAGESFILTER), MAKEINTRESOURCE(IDS_BROWSE),
|
|
OFN_HIDEREADONLY | OFN_FILEMUSTEXIST | OFN_NOCHANGEDIR))
|
|
{
|
|
_EnableApply(hwndDlg);
|
|
_fUsingThumb = TRUE;
|
|
lstrcpyn(_szLogoFile, szFilePath, ARRAYSIZE(_szLogoFile));
|
|
_CreateThumbnailBitmap(hwndDlg);
|
|
}
|
|
break;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL CFolderCustomize::_NotifyAboutWebView(HWND hwnd)
|
|
{
|
|
BOOL fRet = FALSE;
|
|
|
|
SHELLSTATE ss;
|
|
SHGetSetSettings(&ss, SSF_WEBVIEW, FALSE);
|
|
if (!ss.fWebView &&
|
|
(IDYES == ShellMessageBox(HINST_THISDLL, hwnd, MAKEINTRESOURCE(IDS_CUSTOMIZE_TURNONWEBVIEW),
|
|
MAKEINTRESOURCE(IDS_CUSTOMIZE), MB_YESNO | MB_ICONQUESTION)))
|
|
{
|
|
ss.fWebView = TRUE;
|
|
SHGetSetSettings(&ss, SSF_WEBVIEW, TRUE);
|
|
fRet = TRUE;
|
|
}
|
|
return fRet;
|
|
}
|
|
|
|
typedef struct
|
|
{
|
|
LPCITEMIDLIST pidlChanged;
|
|
BOOL fTurnOnWebView;
|
|
BOOL fApplyToChildren;
|
|
} CUSTENUMSTRUCT;
|
|
|
|
BOOL CALLBACK CFolderCustomize::_RefreshView(HWND hwnd, LPCITEMIDLIST pidl, LPARAM lParam)
|
|
{
|
|
CUSTENUMSTRUCT *pes = (CUSTENUMSTRUCT *)lParam;
|
|
|
|
if (pes->fTurnOnWebView)
|
|
{
|
|
PostMessage(hwnd, WM_COMMAND, SFVIDM_MISC_SETWEBVIEW, TRUE);
|
|
}
|
|
if (ILIsEqual(pes->pidlChanged, pidl) || (pes->fApplyToChildren && ILIsParent(pes->pidlChanged, pidl, FALSE)))
|
|
{
|
|
PostMessage(hwnd, WM_COMMAND, SFVIDM_MISC_HARDREFRESH, 0L);
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
void CFolderCustomize::_RefreshWindows(BOOL fTurnOnWebView, BOOL fApplyToChildren)
|
|
{
|
|
CUSTENUMSTRUCT es = { _pidl, fTurnOnWebView, fApplyToChildren };
|
|
EnumShellWindows(_RefreshView, (LPARAM)&es);
|
|
}
|
|
|
|
void CFolderCustomize::_UpdateViewState(HWND hwndDlg, IPropertyBag *ppb, int iIndex)
|
|
{
|
|
TCHAR szOriginalType[25];
|
|
szOriginalType[0] = 0;
|
|
SHPropertyBag_ReadStr(ppb, PROPSTR_FOLDERTYPE, szOriginalType, ARRAYSIZE(szOriginalType));
|
|
// only apply view state change if the folder type is changing.
|
|
// also special case so that we dont apply a view state change if the folder has no
|
|
// current folder type and the user didnt change the selection from "documents"
|
|
// (i.e. they changed folder thumbnail but nothing else).
|
|
if ((lstrcmpi(c_wvtiList[iIndex].pszFolderType, szOriginalType) != 0) &&
|
|
(szOriginalType[0] || iIndex))
|
|
{
|
|
// knock out existing state, they don't want it any more.
|
|
SHPropertyBag_Delete(ppb, VS_PROPSTR_MODE);
|
|
SHPropertyBag_Delete(ppb, VS_PROPSTR_VID);
|
|
|
|
SHPropertyBag_WriteStr(ppb, PROPSTR_FOLDERTYPE, c_wvtiList[iIndex].pszFolderType);
|
|
|
|
_RefreshWindows(_NotifyAboutWebView(hwndDlg), Button_GetCheck(GetDlgItem(hwndDlg, IDC_FOLDER_RECURSE)) == BST_CHECKED);
|
|
}
|
|
}
|
|
|
|
void CFolderCustomize::_DirTouch(LPITEMIDLIST pidl)
|
|
{
|
|
FILETIME ftCurrent;
|
|
GetSystemTimeAsFileTime(&ftCurrent);
|
|
|
|
TCHAR szPath[MAX_PATH];
|
|
if (SHGetPathFromIDList(pidl, szPath))
|
|
{
|
|
// woohoo yay for private flags
|
|
// 0x100 lets us open a directory in write access
|
|
HANDLE h = CreateFile(szPath, GENERIC_READ | 0x100,
|
|
FILE_SHARE_READ | FILE_SHARE_DELETE, NULL,
|
|
OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
|
|
if (h != INVALID_HANDLE_VALUE)
|
|
{
|
|
SetFileTime(h, NULL, NULL, &ftCurrent);
|
|
CloseHandle(h);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CFolderCustomize::_DeleteCustomizationInBag(IPropertyBag *ppb)
|
|
{
|
|
// this is only called when the inherit bag is getting written out.
|
|
// so we need to scorch the existing non-inherit bag so it doesn't
|
|
// override the inherit bag.
|
|
SHPropertyBag_Delete(ppb, PROPSTR_FOLDERTYPE);
|
|
SHPropertyBag_Delete(ppb, PROPSTR_LOGO);
|
|
SHPropertyBag_Delete(ppb, VS_PROPSTR_MODE);
|
|
SHPropertyBag_Delete(ppb, VS_PROPSTR_VID);
|
|
}
|
|
|
|
HRESULT CFolderCustomize::_ApplyChangesToBag(HWND hwndDlg, IPropertyBag *ppb)
|
|
{
|
|
// handle webview template
|
|
HWND hwndTemplates = GetDlgItem(hwndDlg, IDC_FOLDER_TEMPLATES);
|
|
if (hwndTemplates)
|
|
{
|
|
int iIndex = ComboBox_GetCurSel(hwndTemplates);
|
|
if (iIndex != CB_ERR)
|
|
{
|
|
int iViewIndex = (int)ComboBox_GetItemData(hwndTemplates, iIndex);
|
|
_UpdateViewState(hwndDlg, ppb, iViewIndex);
|
|
}
|
|
}
|
|
|
|
TCHAR szThumb[MAX_PATH];
|
|
szThumb[0] = 0;
|
|
if (_fUsingThumb)
|
|
{
|
|
lstrcpyn(szThumb, _szLogoFile, ARRAYSIZE(szThumb));
|
|
}
|
|
|
|
TCHAR szOriginalLogo[MAX_PATH];
|
|
szOriginalLogo[0] = 0;
|
|
SHPropertyBag_ReadStr(ppb, PROPSTR_LOGO, szOriginalLogo, ARRAYSIZE(szOriginalLogo));
|
|
if (lstrcmpi(szThumb, szOriginalLogo) != 0)
|
|
{
|
|
SHPropertyBag_WriteStr(ppb, PROPSTR_LOGO, szThumb);
|
|
_DirTouch(_pidl);
|
|
SHChangeNotify(SHCNE_UPDATEITEM, SHCNF_IDLIST, _pidl, NULL);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CFolderCustomize::_ApplyChanges(HWND hwndDlg)
|
|
{
|
|
// handle icon change
|
|
switch (_hrFromIconChange)
|
|
{
|
|
case S_OK:
|
|
_pIconManager->SetIcon(_szIconPath, _iIconIndex);
|
|
break;
|
|
|
|
case S_FALSE:
|
|
_pIconManager->SetDefaultIcon();
|
|
break;
|
|
}
|
|
|
|
if (Button_GetCheck(GetDlgItem(hwndDlg, IDC_FOLDER_RECURSE)) == BST_CHECKED)
|
|
{
|
|
IPropertyBag *ppbInherit;
|
|
if (SUCCEEDED(SHGetViewStatePropertyBag(_pidl, VS_BAGSTR_EXPLORER, SHGVSPB_INHERIT, IID_PPV_ARG(IPropertyBag, &ppbInherit))))
|
|
{
|
|
_DeleteCustomizationInBag(_ppb);
|
|
_ApplyChangesToBag(hwndDlg, ppbInherit);
|
|
ppbInherit->Release();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_ApplyChangesToBag(hwndDlg, _ppb);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
int CFolderCustomize::_GetTemplateIndexFromType(LPCTSTR pszType)
|
|
{
|
|
// default to "documents"
|
|
int iIndexFound = 0;
|
|
for (int iIndex = 0; iIndex < ARRAYSIZE(c_wvtiList); iIndex++)
|
|
{
|
|
if (lstrcmpi(c_wvtiList[iIndex].pszFolderType, pszType) == 0)
|
|
{
|
|
iIndexFound = iIndex;
|
|
break;
|
|
}
|
|
}
|
|
return iIndexFound;
|
|
}
|
|
|
|
// Fill the combobox with templates' friendly names.
|
|
void CFolderCustomize::_FillTemplateComboBox(HWND hwndTemplates)
|
|
{
|
|
// Disable redraws while we mess repeatedly with the contents.
|
|
SendMessage(hwndTemplates, WM_SETREDRAW, FALSE, 0);
|
|
|
|
TCHAR szType[25];
|
|
szType[0] = 0;
|
|
SHPropertyBag_ReadStr(_ppb, PROPSTR_FOLDERTYPE, szType, ARRAYSIZE(szType));
|
|
|
|
int nFolderTypeIndex = _GetTemplateIndexFromType(szType); // store index into c_wvtiList
|
|
int iIndex = 0; // index into combobox
|
|
// Add each template to the listview.
|
|
for (int nTemplate = 0; nTemplate < ARRAYSIZE(c_wvtiList); nTemplate++)
|
|
{
|
|
TCHAR szPath[MAX_PATH];
|
|
SFVM_WEBVIEW_TEMPLATE_DATA wvData;
|
|
if (!(c_wvtiList[nTemplate].dwFlags & WVTI_SHOWIFOLDTEMPLATE) ||
|
|
(SHGetPathFromIDList(_pidl, szPath) && SUCCEEDED(DefaultGetWebViewTemplateFromPath(szPath, &wvData))))
|
|
{
|
|
TCHAR szFriendlyName[100];
|
|
LoadString(HINST_THISDLL, c_wvtiList[nTemplate].uIDFriendly, szFriendlyName, ARRAYSIZE(szFriendlyName));
|
|
|
|
int iIndexAdd = ComboBox_AddString(hwndTemplates, szFriendlyName);
|
|
if (iIndexAdd != -1)
|
|
{
|
|
if (nTemplate == nFolderTypeIndex)
|
|
{
|
|
iIndex = iIndexAdd;
|
|
}
|
|
ComboBox_SetItemData(hwndTemplates, iIndexAdd, nTemplate);
|
|
}
|
|
}
|
|
}
|
|
|
|
// pick default
|
|
ComboBox_SetCurSel(hwndTemplates, iIndex);
|
|
|
|
// Reenable redraws.
|
|
SendMessage(hwndTemplates, WM_SETREDRAW, TRUE, 0);
|
|
InvalidateRect(hwndTemplates, NULL, TRUE);
|
|
}
|
|
|
|
void CFolderCustomize::_SetThumbnail(HWND hwnd)
|
|
{
|
|
_szLogoFile[0] = 0;
|
|
SHPropertyBag_ReadStr(_ppb, PROPSTR_LOGO, _szLogoFile, ARRAYSIZE(_szLogoFile));
|
|
|
|
_fUsingThumb = _szLogoFile[0];
|
|
|
|
_CreateThumbnailBitmap(hwnd);
|
|
}
|
|
|
|
void CFolderCustomize::_SetRecurseBox(HWND hwnd)
|
|
{
|
|
IPropertyBag *ppbInherit;
|
|
if (SUCCEEDED(SHGetViewStatePropertyBag(_pidl, VS_BAGSTR_EXPLORER, SHGVSPB_INHERIT, IID_PPV_ARG(IPropertyBag, &ppbInherit))))
|
|
{
|
|
TCHAR szTypeInherit[MAX_PATH];
|
|
if (SUCCEEDED(SHPropertyBag_ReadStr(ppbInherit, PROPSTR_FOLDERTYPE, szTypeInherit, ARRAYSIZE(szTypeInherit))) && szTypeInherit[0])
|
|
{
|
|
TCHAR szType[MAX_PATH];
|
|
if (SUCCEEDED(SHPropertyBag_ReadStr(_ppb, PROPSTR_FOLDERTYPE, szType, ARRAYSIZE(szType))) &&
|
|
(lstrcmpi(szTypeInherit, szType) == 0))
|
|
{
|
|
Button_SetCheck(GetDlgItem(hwnd, IDC_FOLDER_RECURSE), TRUE);
|
|
}
|
|
}
|
|
ppbInherit->Release();
|
|
}
|
|
}
|
|
|
|
// since changing the icon isn't in the peruser property bag (yet [it was punted from whistler])
|
|
// we need to disable this section if we know it can't be modified.
|
|
void CFolderCustomize::_HideIconSection(HWND hwndDlg)
|
|
{
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_CHANGEICONGROUP), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_CHANGEICON), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_CHANGEICONTEXT), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hwndDlg, IDC_FOLDER_ICON), SW_HIDE);
|
|
}
|
|
|
|
void CFolderCustomize::_InitDialog(HWND hwndDlg)
|
|
{
|
|
HWND hwndTemplates = GetDlgItem(hwndDlg, IDC_FOLDER_TEMPLATES);
|
|
if (hwndTemplates)
|
|
{
|
|
_FillTemplateComboBox(GetDlgItem(hwndDlg, IDC_FOLDER_TEMPLATES));
|
|
EnableWindow(hwndTemplates, TRUE);
|
|
|
|
_SetThumbnail(hwndDlg);
|
|
|
|
// Disable the Icon Change button if we the IShellFolder doesn't support ICustomIconManager interface.
|
|
if (_ShouldEnableChangeOfIcon())
|
|
{
|
|
_CreateFolderIcon(hwndDlg);
|
|
}
|
|
else
|
|
{
|
|
_HideIconSection(hwndDlg);
|
|
}
|
|
|
|
_SetRecurseBox(hwndDlg);
|
|
}
|
|
}
|
|
|
|
// helpers moved from mulprsht
|
|
|
|
// How do we selectively disable for .exe
|
|
BOOL CFolderCustomize::_ShouldEnableChangeOfIcon()
|
|
{
|
|
if (!_pIconManager)
|
|
{
|
|
SHGetUIObjectFromFullPIDL(_pidl, NULL, IID_PPV_ARG(ICustomIconManager, &_pIconManager));
|
|
}
|
|
|
|
return BOOLIFY(_pIconManager);
|
|
}
|
|
|
|
void CFolderCustomize::_EnableApply(HWND hwnd)
|
|
{
|
|
PropSheet_Changed(GetParent(hwnd), hwnd);
|
|
}
|
|
|
|
void CFolderCustomize::_ChangeFolderIcon(HWND hwndDlg)
|
|
{
|
|
ASSERT(_pIconManager);
|
|
|
|
TCHAR szDialogCaptionFmt[MAX_PATH];
|
|
LoadString(HINST_THISDLL, IDS_FOLDER_PICKICONDLG_CAPTION, szDialogCaptionFmt, ARRAYSIZE(szDialogCaptionFmt));
|
|
|
|
TCHAR szFileName[MAX_PATH], szDialogCaption[MAX_PATH];
|
|
if (SUCCEEDED(SHGetNameAndFlags(_pidl, SHGDN_NORMAL, szFileName, ARRAYSIZE(szFileName), NULL)))
|
|
{
|
|
wnsprintf(szDialogCaption, ARRAYSIZE(szDialogCaption), szDialogCaptionFmt, PathFindFileName(szFileName));
|
|
}
|
|
|
|
if (SUCCEEDED(_ProcessIconChange(szDialogCaption, hwndDlg)))
|
|
{
|
|
_EnableApply(hwndDlg);
|
|
}
|
|
}
|
|
|
|
HRESULT CFolderCustomize::_ProcessIconChange(LPCTSTR pszPickIconDialogCaption, HWND hwndDlg)
|
|
{
|
|
int nIconIndex = -1;
|
|
|
|
TCHAR szIconPath[MAX_PATH];
|
|
szIconPath[0] = 0;
|
|
|
|
HRESULT hr = PickIconDlgWithTitle(hwndDlg, pszPickIconDialogCaption, TRUE, szIconPath, ARRAYSIZE(szIconPath), &nIconIndex);
|
|
_hrFromIconChange = hr;
|
|
switch (hr)
|
|
{
|
|
case S_OK:
|
|
{
|
|
HICON hIcon = ExtractIcon(HINST_THISDLL, szIconPath, nIconIndex);
|
|
if (hIcon != NULL)
|
|
{
|
|
ReplaceDlgIcon(hwndDlg, IDC_FOLDER_ICON, hIcon);
|
|
StrCpyN(_szIconPath, szIconPath, ARRAYSIZE(_szIconPath));
|
|
_iIconIndex = nIconIndex;
|
|
}
|
|
else
|
|
{
|
|
_hrFromIconChange = HRESULT_FROM_WIN32(ERROR_CANCELLED);
|
|
}
|
|
break;
|
|
}
|
|
|
|
case S_FALSE:
|
|
{
|
|
HICON hIcon;
|
|
if (SUCCEEDED(_pIconManager->GetDefaultIconHandle(&hIcon)))
|
|
{
|
|
ReplaceDlgIcon(hwndDlg, IDC_FOLDER_ICON, hIcon);
|
|
}
|
|
else
|
|
{
|
|
_hrFromIconChange = HRESULT_FROM_WIN32(ERROR_CANCELLED);
|
|
}
|
|
break;
|
|
}
|
|
|
|
case HRESULT_FROM_WIN32(ERROR_CANCELLED):
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
|
|
|