1 /* 2 * Provides default file shell extension 3 * 4 * Copyright 2012 Rafal Harabien 5 * 6 * This library is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * This library is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with this library; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA 19 */ 20 21 #ifndef _FILE_DEF_EXT_H_ 22 #define _FILE_DEF_EXT_H_ 23 24 class CFileVersionInfo 25 { 26 private: 27 PVOID m_pInfo; 28 WORD m_wLang, m_wCode; 29 WCHAR m_wszLang[64]; 30 31 typedef struct _LANGANDCODEPAGE_ 32 { 33 WORD wLang; 34 WORD wCode; 35 } LANGANDCODEPAGE, *LPLANGANDCODEPAGE; 36 37 public: CFileVersionInfo()38 inline CFileVersionInfo(): 39 m_pInfo(NULL), m_wLang(0), m_wCode(0) 40 { 41 m_wszLang[0] = L'\0'; 42 } 43 ~CFileVersionInfo()44 inline ~CFileVersionInfo() 45 { 46 if (m_pInfo) 47 HeapFree(GetProcessHeap(), 0, m_pInfo); 48 } 49 50 BOOL Load(LPCWSTR pwszPath); 51 LPCWSTR GetString(LPCWSTR pwszName); 52 VS_FIXEDFILEINFO *GetFixedInfo(); 53 LPCWSTR GetLangName(); 54 }; 55 56 class CFileDefExt : 57 public CComCoClass<CFileDefExt, &CLSID_ShellFileDefExt>, 58 public CComObjectRootEx<CComMultiThreadModelNoCS>, 59 public IShellExtInit, 60 public IContextMenu, 61 public IShellPropSheetExt, 62 public IObjectWithSite 63 { 64 private: 65 VOID InitOpensWithField(HWND hwndDlg); 66 BOOL InitFileType(HWND hwndDlg); 67 BOOL InitFilePath(HWND hwndDlg); 68 static BOOL GetFileTimeString(LPFILETIME lpFileTime, LPWSTR pwszResult, UINT cchResult); 69 BOOL InitFileAttr(HWND hwndDlg); 70 BOOL InitGeneralPage(HWND hwndDlg); 71 BOOL SetVersionLabel(HWND hwndDlg, DWORD idCtrl, LPCWSTR pwszName); 72 BOOL AddVersionString(HWND hwndDlg, LPCWSTR pwszName); 73 BOOL InitVersionPage(HWND hwndDlg); 74 BOOL InitFolderCustomizePage(HWND hwndDlg); 75 static INT_PTR CALLBACK GeneralPageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam); 76 static INT_PTR CALLBACK VersionPageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam); 77 static INT_PTR CALLBACK FolderCustomizePageProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam); 78 BOOL CountFolderAndFiles(HWND hwndDlg, LPCWSTR pwszBuf, LPDWORD ticks); 79 80 WCHAR m_wszPath[MAX_PATH]; 81 CFileVersionInfo m_VerInfo; 82 BOOL m_bDir; 83 84 DWORD m_cFiles; 85 DWORD m_cFolders; 86 ULARGE_INTEGER m_DirSize; 87 ULARGE_INTEGER m_DirSizeOnDisc; 88 89 static DWORD WINAPI _CountFolderAndFilesThreadProc(LPVOID lpParameter); 90 91 // FolderCustomize 92 WCHAR m_szFolderIconPath[MAX_PATH]; 93 INT m_nFolderIconIndex; 94 HICON m_hFolderIcon; 95 BOOL m_bFolderIconIsSet; 96 97 public: 98 CFileDefExt(); 99 ~CFileDefExt(); 100 101 // FolderCustomize 102 BOOL OnFolderCustApply(HWND hwndDlg); 103 void OnFolderCustChangeIcon(HWND hwndDlg); 104 void OnFolderCustDestroy(HWND hwndDlg); 105 void UpdateFolderIcon(HWND hwndDlg); 106 107 // IShellExtInit 108 STDMETHOD(Initialize)(PCIDLIST_ABSOLUTE pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID) override; 109 110 // IContextMenu 111 STDMETHOD(QueryContextMenu)(HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags) override; 112 STDMETHOD(InvokeCommand)(LPCMINVOKECOMMANDINFO lpici) override; 113 STDMETHOD(GetCommandString)(UINT_PTR idCmd, UINT uType, UINT *pwReserved, LPSTR pszName, UINT cchMax) override; 114 115 // IShellPropSheetExt 116 STDMETHOD(AddPages)(LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam) override; 117 STDMETHOD(ReplacePage)(UINT uPageID, LPFNADDPROPSHEETPAGE pfnReplacePage, LPARAM lParam) override; 118 119 // IObjectWithSite 120 STDMETHOD(SetSite)(IUnknown *punk) override; 121 STDMETHOD(GetSite)(REFIID iid, void **ppvSite) override; 122 123 DECLARE_REGISTRY_RESOURCEID(IDR_FILEDEFEXT) 124 DECLARE_NOT_AGGREGATABLE(CFileDefExt) 125 126 DECLARE_PROTECT_FINAL_CONSTRUCT() 127 128 BEGIN_COM_MAP(CFileDefExt) 129 COM_INTERFACE_ENTRY_IID(IID_IShellExtInit, IShellExtInit) 130 COM_INTERFACE_ENTRY_IID(IID_IContextMenu, IContextMenu) 131 COM_INTERFACE_ENTRY_IID(IID_IShellPropSheetExt, IShellPropSheetExt) 132 COM_INTERFACE_ENTRY_IID(IID_IObjectWithSite, IObjectWithSite) 133 END_COM_MAP() 134 }; 135 136 struct _CountFolderAndFilesData { 137 CFileDefExt *This; 138 HWND hwndDlg; 139 LPWSTR pwszBuf; 140 }; 141 142 #endif /* _FILE_DEF_EXT_H_ */ 143