Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155/* * Provides default file shell extension * * Copyright 2012 Rafal Harabien * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#ifndef _FILE_DEF_EXT_H_#define _FILE_DEF_EXT_H_
class CFileVersionInfo{ private: PVOID m_pInfo; WORD m_wLang, m_wCode; WCHAR m_wszLang[64];
typedef struct _LANGANDCODEPAGE_ { WORD wLang; WORD wCode; } LANGANDCODEPAGE, *LPLANGANDCODEPAGE;
public: inline CFileVersionInfo(): m_pInfo(NULL), m_wLang(0), m_wCode(0) { m_wszLang[0] = L'\0'; }
inline ~CFileVersionInfo() { if (m_pInfo) HeapFree(GetProcessHeap(), 0, m_pInfo); }
BOOL Load(LPCWSTR pwszPath); LPCWSTR GetString(LPCWSTR pwszName); VS_FIXEDFILEINFO *GetFixedInfo(); LPCWSTR GetLangName();};
class CFileDefExt : public CComCoClass<CFileDefExt, &CLSID_ShellFileDefExt>, public CComObjectRootEx<CComMultiThreadModelNoCS>, public IShellExtInit, public IContextMenu, public IShellPropSheetExt, public CObjectWithSiteBase{private: VOID InitOpensWithField(HWND hwndDlg); BOOL InitFileType(HWND hwndDlg); BOOL InitFilePath(HWND hwndDlg); static BOOL GetFileTimeString(LPFILETIME lpFileTime, LPWSTR pwszResult, UINT cchResult); BOOL InitFileAttr(HWND hwndDlg); BOOL InitGeneralPage(HWND hwndDlg); BOOL SetVersionLabel(HWND hwndDlg, DWORD idCtrl, LPCWSTR pwszName); BOOL AddVersionString(HWND hwndDlg, LPCWSTR pwszName); BOOL InitVersionPage(HWND hwndDlg); BOOL InitFolderCustomizePage(HWND hwndDlg); void InitMultifilePage(HWND hwndDlg); void InitMultifilePageThread(); void CountFolderAndFiles(); static INT_PTR CALLBACK GeneralPageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam); static INT_PTR CALLBACK VersionPageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam); static INT_PTR CALLBACK FolderCustomizePageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam); static INT_PTR CALLBACK MultifilePageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
WCHAR m_wszPath[MAX_PATH]; CFileVersionInfo m_VerInfo; BOOL m_bDir; BOOL m_bMultifile;
LPITEMIDLIST m_pidlFolder = NULL; LPITEMIDLIST *m_pidls = NULL; UINT m_cidl = 0; DWORD m_cFiles; DWORD m_cFolders; ULARGE_INTEGER m_DirSize; ULARGE_INTEGER m_DirSizeOnDisc; enum { WM_UPDATEDIRSTATS = WM_APP }; HWND m_hWndDirStatsDlg; void InitDirStats(struct DIRTREESTATS *pStats); BOOL WalkDirTree(PCWSTR pszPath, struct DIRTREESTATS *pStats, WIN32_FIND_DATAW *pWFD); void UpdateDirStatsResults();
LONG volatile m_Destroyed = 0; BOOL IsDestroyed() const { return m_Destroyed; }
static DWORD WINAPI _CountFolderAndFilesThreadProc(LPVOID lpParameter); static DWORD WINAPI _InitializeMultifileThreadProc(LPVOID lpParameter);
// FolderCustomize WCHAR m_szFolderIconPath[MAX_PATH]; INT m_nFolderIconIndex; HICON m_hFolderIcon; BOOL m_bFolderIconIsSet;
public: CFileDefExt(); ~CFileDefExt();
// FolderCustomize BOOL OnFolderCustApply(HWND hwndDlg); void OnFolderCustChangeIcon(HWND hwndDlg); void OnFolderCustDestroy(HWND hwndDlg); void UpdateFolderIcon(HWND hwndDlg);
// IShellExtInit STDMETHOD(Initialize)(PCIDLIST_ABSOLUTE pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID) override;
// IContextMenu STDMETHOD(QueryContextMenu)(HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags) override; STDMETHOD(InvokeCommand)(LPCMINVOKECOMMANDINFO lpici) override; STDMETHOD(GetCommandString)(UINT_PTR idCmd, UINT uType, UINT *pwReserved, LPSTR pszName, UINT cchMax) override;
// IShellPropSheetExt STDMETHOD(AddPages)(LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam) override; STDMETHOD(ReplacePage)(UINT uPageID, LPFNADDPROPSHEETPAGE pfnReplacePage, LPARAM lParam) override;
DECLARE_REGISTRY_RESOURCEID(IDR_FILEDEFEXT)DECLARE_NOT_AGGREGATABLE(CFileDefExt)
DECLARE_PROTECT_FINAL_CONSTRUCT()
BEGIN_COM_MAP(CFileDefExt) COM_INTERFACE_ENTRY_IID(IID_IShellExtInit, IShellExtInit) COM_INTERFACE_ENTRY_IID(IID_IContextMenu, IContextMenu) COM_INTERFACE_ENTRY_IID(IID_IShellPropSheetExt, IShellPropSheetExt) COM_INTERFACE_ENTRY_IID(IID_IObjectWithSite, IObjectWithSite)END_COM_MAP()};
struct _CountFolderAndFilesData { CFileDefExt *This; HWND hwndDlg; LPWSTR pwszBuf;};
#endif /* _FILE_DEF_EXT_H_ */