2 * provides new shell item service
4 * Copyright 2007 Johannes Anderwald (johannes.anderwald@reactos.org)
5 * Copyright 2009 Andrew Hill
6 * Copyright 2012 Rafal Harabien
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
25 WINE_DEFAULT_DEBUG_CHANNEL(shell
);
27 CNewMenu::CNewMenu() :
47 void CNewMenu::UnloadItem(SHELLNEW_ITEM
*pItem
)
49 /* Note: free allows NULL as argument */
51 free(pItem
->pwszDesc
);
55 DestroyIcon(pItem
->hIcon
);
57 HeapFree(GetProcessHeap(), 0, pItem
);
60 void CNewMenu::UnloadAllItems()
62 SHELLNEW_ITEM
*pCurItem
;
64 /* Unload normal items */
68 m_pItems
= m_pItems
->pNext
;
73 /* Unload link item */
75 UnloadItem(m_pLinkItem
);
79 CNewMenu::SHELLNEW_ITEM
*CNewMenu::LoadItem(LPCWSTR pwszExt
)
82 WCHAR wszBuf
[MAX_PATH
];
86 StringCbPrintfW(wszBuf
, sizeof(wszBuf
), L
"%s\\ShellNew", pwszExt
);
88 TRACE("LoadItem Keyname %s Name %s\n", debugstr_w(pwszExt
), debugstr_w(wszBuf
));
90 if (RegOpenKeyExW(HKEY_CLASSES_ROOT
, wszBuf
, 0, KEY_READ
, &hKey
) != ERROR_SUCCESS
)
92 TRACE("Failed to open key\n");
96 /* Find first valid value */
104 {L
"FileName", SHELLNEW_TYPE_FILENAME
, TRUE
, TRUE
},
105 {L
"Command", SHELLNEW_TYPE_COMMAND
, TRUE
, TRUE
},
106 {L
"Data", SHELLNEW_TYPE_DATA
, TRUE
, FALSE
},
107 {L
"NullFile", SHELLNEW_TYPE_NULLFILE
, FALSE
},
112 for (i
= 0; Types
[i
].pszName
; ++i
)
114 /* Note: We are using ANSI function because strings can be treated as data */
116 DWORD dwFlags
= Types
[i
].bStr
? RRF_RT_REG_SZ
: RRF_RT_ANY
;
118 if (RegGetValueW(hKey
, NULL
, Types
[i
].pszName
, dwFlags
, NULL
, NULL
, &cbData
) == ERROR_SUCCESS
)
120 if (Types
[i
].bNeedData
&& cbData
> 0)
122 pData
= (BYTE
*)malloc(cbData
);
123 RegGetValueW(hKey
, NULL
, Types
[i
].pszName
, dwFlags
, &dwType
, pData
, &cbData
);
124 if (!Types
[i
].bStr
&& (dwType
== REG_SZ
|| dwType
== REG_EXPAND_SZ
))
126 PBYTE pData2
= (PBYTE
)malloc(cbData
);
127 cbData
= WideCharToMultiByte(CP_ACP
, 0, (LPWSTR
)pData
, -1, (LPSTR
)pData2
, cbData
, NULL
, NULL
);
137 /* Was any key found? */
138 if (!Types
[i
].pszName
)
142 if (!SHGetFileInfoW(pwszExt
, FILE_ATTRIBUTE_NORMAL
, &fi
, sizeof(fi
), SHGFI_USEFILEATTRIBUTES
|SHGFI_TYPENAME
|SHGFI_ICON
|SHGFI_SMALLICON
))
145 /* Create new item */
146 SHELLNEW_ITEM
*pNewItem
= static_cast<SHELLNEW_ITEM
*>(HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(SHELLNEW_ITEM
)));
153 TRACE("new item %ls\n", fi
.szTypeName
);
154 pNewItem
->Type
= Types
[i
].Type
;
155 pNewItem
->pData
= pData
;
156 pNewItem
->cbData
= pData
? cbData
: 0;
157 pNewItem
->pwszExt
= _wcsdup(pwszExt
);
158 pNewItem
->pwszDesc
= _wcsdup(fi
.szTypeName
);
160 pNewItem
->hIcon
= fi
.hIcon
;
166 CNewMenu::CacheItems()
173 WCHAR wszName
[MAX_PATH
];
174 SHELLNEW_ITEM
*pNewItem
;
175 SHELLNEW_ITEM
*pCurItem
= NULL
;
177 /* Enumerate all extensions */
178 while (RegEnumKeyW(HKEY_CLASSES_ROOT
, dwIndex
++, wszName
, _countof(wszName
)) == ERROR_SUCCESS
)
180 if (wszName
[0] != L
'.')
183 pNewItem
= LoadItem(wszName
);
186 dwSize
+= wcslen(wszName
) + 1;
187 if (wcsicmp(pNewItem
->pwszExt
, L
".lnk") == 0)
190 m_pLinkItem
= pNewItem
;
194 /* Add at the end of list */
197 pCurItem
->pNext
= pNewItem
;
201 pCurItem
= m_pItems
= pNewItem
;
208 lpValues
= (LPWSTR
) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, dwSize
* sizeof(WCHAR
));
216 memcpy(lpValue
, pCurItem
->pwszExt
, (wcslen(pCurItem
->pwszExt
) + 1) * sizeof(WCHAR
));
217 lpValue
+= wcslen(pCurItem
->pwszExt
) + 1;
218 pCurItem
= pCurItem
->pNext
;
221 if (RegCreateKeyEx(HKEY_CURRENT_USER
, ShellNewKey
, 0, NULL
, REG_OPTION_NON_VOLATILE
, KEY_WRITE
, NULL
, &hKey
, NULL
) != ERROR_SUCCESS
)
223 HeapFree(GetProcessHeap(), 0, lpValues
);
227 if (RegSetValueExW(hKey
, L
"Classes", NULL
, REG_MULTI_SZ
, (LPBYTE
)lpValues
, dwSize
* sizeof(WCHAR
)) != ERROR_SUCCESS
)
229 HeapFree(GetProcessHeap(), 0, lpValues
);
234 HeapFree(GetProcessHeap(), 0, lpValues
);
241 CNewMenu::LoadCachedItems()
247 SHELLNEW_ITEM
*pNewItem
;
248 SHELLNEW_ITEM
*pCurItem
= NULL
;
250 if (RegOpenKeyExW(HKEY_CURRENT_USER
, ShellNewKey
, 0, KEY_READ
, &hKey
) != ERROR_SUCCESS
)
253 if (RegQueryValueExW(hKey
, L
"Classes", NULL
, NULL
, NULL
, &dwSize
) != ERROR_SUCCESS
)
256 lpValues
= (LPWSTR
) HeapAlloc(GetProcessHeap(), 0, dwSize
);
260 if (RegQueryValueExW(hKey
, L
"Classes", NULL
, NULL
, (LPBYTE
)lpValues
, &dwSize
) != ERROR_SUCCESS
)
262 HeapFree(GetProcessHeap(), 0, lpValues
);
268 for (; '\0' != *wszName
; wszName
+= wcslen(wszName
) + 1)
270 pNewItem
= LoadItem(wszName
);
273 if (wcsicmp(pNewItem
->pwszExt
, L
".lnk") == 0)
276 m_pLinkItem
= pNewItem
;
280 /* Add at the end of list */
283 pCurItem
->pNext
= pNewItem
;
287 pCurItem
= m_pItems
= pNewItem
;
292 HeapFree(GetProcessHeap(), 0, lpValues
);
299 CNewMenu::LoadAllItems()
301 /* If there are any unload them */
304 if (!LoadCachedItems())
311 m_pLinkItem
= static_cast<SHELLNEW_ITEM
*>(HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(SHELLNEW_ITEM
)));
314 m_pLinkItem
->Type
= SHELLNEW_TYPE_NULLFILE
;
315 m_pLinkItem
->pwszDesc
= _wcsdup(L
"Link");
316 m_pLinkItem
->pwszExt
= _wcsdup(L
".lnk");
320 if (m_pItems
== NULL
)
327 CNewMenu::InsertShellNewItems(HMENU hMenu
, UINT idCmdFirst
, UINT Pos
)
331 UINT idCmd
= idCmdFirst
;
333 if (m_pItems
== NULL
)
339 ZeroMemory(&mii
, sizeof(mii
));
340 mii
.cbSize
= sizeof(mii
);
342 /* Insert new folder action */
343 if (!LoadStringW(shell32_hInstance
, FCIDM_SHVIEW_NEWFOLDER
, wszBuf
, _countof(wszBuf
)))
345 mii
.fMask
= MIIM_ID
| MIIM_BITMAP
| MIIM_STRING
;
346 mii
.dwTypeData
= wszBuf
;
347 mii
.cch
= wcslen(mii
.dwTypeData
);
349 mii
.hbmpItem
= HBMMENU_CALLBACK
;
350 if (InsertMenuItemW(hMenu
, Pos
++, TRUE
, &mii
))
353 /* Insert new shortcut action */
354 if (!LoadStringW(shell32_hInstance
, FCIDM_SHVIEW_NEWLINK
, wszBuf
, _countof(wszBuf
)))
356 mii
.dwTypeData
= wszBuf
;
357 mii
.cch
= wcslen(mii
.dwTypeData
);
359 if (InsertMenuItemW(hMenu
, Pos
++, TRUE
, &mii
))
362 /* Insert seperator for custom new action */
363 mii
.fMask
= MIIM_TYPE
| MIIM_ID
;
364 mii
.fType
= MFT_SEPARATOR
;
366 InsertMenuItemW(hMenu
, Pos
++, TRUE
, &mii
);
368 /* Insert rest of items */
369 mii
.fMask
= MIIM_ID
| MIIM_BITMAP
| MIIM_STRING
;
372 SHELLNEW_ITEM
*pCurItem
= m_pItems
;
375 TRACE("szDesc %s\n", debugstr_w(pCurItem
->pwszDesc
));
376 mii
.dwTypeData
= pCurItem
->pwszDesc
;
377 mii
.cch
= wcslen(mii
.dwTypeData
);
379 if (InsertMenuItemW(hMenu
, Pos
++, TRUE
, &mii
))
381 pCurItem
= pCurItem
->pNext
;
384 return idCmd
- idCmdFirst
;
387 CNewMenu::SHELLNEW_ITEM
*CNewMenu::FindItemFromIdOffset(UINT IdOffset
)
390 return NULL
; /* Folder */
393 return m_pLinkItem
; /* shortcut */
395 /* Find shell new item */
396 SHELLNEW_ITEM
*pItem
= m_pItems
;
397 for (UINT i
= 2; pItem
; ++i
)
402 pItem
= pItem
->pNext
;
408 HRESULT
CNewMenu::SelectNewItem(LPCMINVOKECOMMANDINFO lpici
, LONG wEventId
, UINT uFlags
, LPWSTR pszName
)
410 CComPtr
<IShellBrowser
> lpSB
;
411 CComPtr
<IShellView
> lpSV
;
414 PITEMID_CHILD pidlNewItem
;
416 /* Notify the view object about the new item */
417 SHChangeNotify(wEventId
, uFlags
, (LPCVOID
) pszName
, NULL
);
419 /* FIXME: I think that this can be implemented using callbacks to the shell folder */
421 /* Note: CWM_GETISHELLBROWSER returns shell browser without adding reference */
422 lpSB
= (LPSHELLBROWSER
)SendMessageA(lpici
->hwnd
, CWM_GETISHELLBROWSER
, 0, 0);
426 hr
= lpSB
->QueryActiveShellView(&lpSV
);
430 /* Attempt to get the pidl of the new item */
431 hr
= SHILCreateFromPathW(pszName
, &pidl
, NULL
);
432 if (FAILED_UNEXPECTEDLY(hr
))
435 pidlNewItem
= ILFindLastID(pidl
);
437 hr
= lpSV
->SelectItem(pidlNewItem
, SVSI_DESELECTOTHERS
| SVSI_EDIT
| SVSI_ENSUREVISIBLE
|
438 SVSI_FOCUSED
| SVSI_SELECT
);
445 // Code is duplicated in CDefaultContextMenu
446 HRESULT
CNewMenu::CreateNewFolder(LPCMINVOKECOMMANDINFO lpici
)
448 WCHAR wszPath
[MAX_PATH
];
449 WCHAR wszName
[MAX_PATH
];
450 WCHAR wszNewFolder
[25];
453 /* Get folder path */
454 hr
= SHGetPathFromIDListW(m_pidlFolder
, wszPath
);
455 if (FAILED_UNEXPECTEDLY(hr
))
458 if (!LoadStringW(shell32_hInstance
, IDS_NEWFOLDER
, wszNewFolder
, _countof(wszNewFolder
)))
461 /* Create the name of the new directory */
462 if (!PathYetAnotherMakeUniqueName(wszName
, wszPath
, NULL
, wszNewFolder
))
465 /* Create the new directory and show the appropriate dialog in case of error */
466 if (SHCreateDirectory (lpici
->hwnd
, wszName
) != ERROR_SUCCESS
)
469 /* Show and select the new item in the def view */
470 SelectNewItem(lpici
, SHCNE_MKDIR
, SHCNF_PATHW
, wszName
);
475 HRESULT
CNewMenu::CreateNewItem(SHELLNEW_ITEM
*pItem
, LPCMINVOKECOMMANDINFO lpcmi
)
477 WCHAR wszBuf
[MAX_PATH
];
478 WCHAR wszPath
[MAX_PATH
];
481 /* Get folder path */
482 hr
= SHGetPathFromIDListW(m_pidlFolder
, wszPath
);
483 if (FAILED_UNEXPECTEDLY(hr
))
488 case SHELLNEW_TYPE_COMMAND
:
491 WCHAR wszTemp
[MAX_PATH
];
493 PROCESS_INFORMATION pi
;
495 if (!ExpandEnvironmentStringsW((LPWSTR
)pItem
->pData
, wszBuf
, MAX_PATH
))
497 TRACE("ExpandEnvironmentStrings failed\n");
501 /* Expand command parameter, FIXME: there can be more modifiers */
502 Ptr
= wcsstr(wszBuf
, L
"%1");
506 StringCbPrintfW(wszTemp
, sizeof(wszTemp
), wszBuf
, wszPath
);
513 ZeroMemory(&si
, sizeof(si
));
515 if (CreateProcessW(NULL
, pwszCmd
, NULL
, NULL
, FALSE
, 0, NULL
, NULL
, &si
, &pi
))
517 CloseHandle(pi
.hProcess
);
518 CloseHandle(pi
.hThread
);
520 ERR("Failed to create process\n");
523 case SHELLNEW_TYPE_DATA
:
524 case SHELLNEW_TYPE_FILENAME
:
525 case SHELLNEW_TYPE_NULLFILE
:
527 BOOL bSuccess
= TRUE
;
528 LPWSTR pwszFilename
= NULL
;
529 size_t cchFilenameMax
= 0;
531 /* Build new file name */
532 LoadStringW(shell32_hInstance
, FCIDM_SHVIEW_NEW
, wszBuf
, _countof(wszBuf
));
533 StringCchCatExW(wszPath
, _countof(wszPath
), L
"\\", &pwszFilename
, &cchFilenameMax
, 0);
534 StringCchPrintfW(pwszFilename
, cchFilenameMax
, L
"%s %s%s", wszBuf
, pItem
->pwszDesc
, pItem
->pwszExt
);
536 /* Find unique name */
537 for (UINT i
= 2; PathFileExistsW(wszPath
); ++i
)
539 StringCchPrintfW(pwszFilename
, cchFilenameMax
, L
"%s %s (%u)%s", wszBuf
, pItem
->pwszDesc
, i
, pItem
->pwszExt
);
540 TRACE("New Filename %ls\n", pwszFilename
);
543 /* Create new file */
544 HANDLE hFile
= CreateFileW(wszPath
, GENERIC_WRITE
, 0, NULL
, CREATE_NEW
, FILE_ATTRIBUTE_NORMAL
, NULL
);
545 if (hFile
!= INVALID_HANDLE_VALUE
)
547 if (pItem
->Type
== SHELLNEW_TYPE_DATA
)
549 /* Write a content */
551 WriteFile(hFile
, pItem
->pData
, pItem
->cbData
, &cbWritten
, NULL
);
559 if (pItem
->Type
== SHELLNEW_TYPE_FILENAME
)
562 if (!CopyFileW((LPWSTR
)pItem
->pData
, wszPath
, FALSE
))
563 ERR("Copy file failed: %ls\n", (LPWSTR
)pItem
->pData
);
566 /* Show message if we failed */
569 TRACE("Notifying fs %s\n", debugstr_w(wszPath
));
570 SelectNewItem(lpcmi
, SHCNE_CREATE
, SHCNF_PATHW
, wszPath
);
574 StringCbPrintfW(wszBuf
, sizeof(wszBuf
), L
"Cannot create file: %s", pwszFilename
);
575 MessageBoxW(NULL
, wszBuf
, L
"Cannot create file", MB_OK
|MB_ICONERROR
); // FIXME
579 case SHELLNEW_TYPE_INVALID
:
580 ERR("Invalid type\n");
587 HRESULT STDMETHODCALLTYPE
CNewMenu::SetSite(IUnknown
*pUnkSite
)
593 HRESULT STDMETHODCALLTYPE
CNewMenu::GetSite(REFIID riid
, void **ppvSite
)
595 return m_pSite
->QueryInterface(riid
, ppvSite
);
600 CNewMenu::QueryContextMenu(HMENU hMenu
,
610 m_idCmdFirst
= idCmdFirst
;
612 TRACE("%p %p %u %u %u %u\n", this,
613 hMenu
, indexMenu
, idCmdFirst
, idCmdLast
, uFlags
);
615 if (!LoadStringW(shell32_hInstance
, FCIDM_SHVIEW_NEW
, wszNew
, _countof(wszNew
)))
618 m_hSubMenu
= CreateMenu();
622 cItems
= InsertShellNewItems(m_hSubMenu
, idCmdFirst
, 0);
624 memset(&mii
, 0, sizeof(mii
));
625 mii
.cbSize
= sizeof(mii
);
626 mii
.fMask
= MIIM_TYPE
| MIIM_ID
| MIIM_STATE
| MIIM_SUBMENU
;
627 mii
.fType
= MFT_STRING
;
629 mii
.dwTypeData
= wszNew
;
630 mii
.cch
= wcslen(mii
.dwTypeData
);
631 mii
.fState
= MFS_ENABLED
;
632 mii
.hSubMenu
= m_hSubMenu
;
634 if (!InsertMenuItemW(hMenu
, indexMenu
, TRUE
, &mii
))
637 return MAKE_HRESULT(SEVERITY_SUCCESS
, 0, cItems
);
642 CNewMenu::InvokeCommand(LPCMINVOKECOMMANDINFO lpici
)
646 if (LOWORD(lpici
->lpVerb
) == 0)
647 hr
= CreateNewFolder(lpici
);
650 SHELLNEW_ITEM
*pItem
= FindItemFromIdOffset(LOWORD(lpici
->lpVerb
));
652 hr
= CreateNewItem(pItem
, lpici
);
655 TRACE("CNewMenu::InvokeCommand %x\n", hr
);
661 CNewMenu::GetCommandString(UINT_PTR idCmd
,
667 FIXME("%p %lu %u %p %p %u\n", this,
668 idCmd
, uType
, pwReserved
, pszName
, cchMax
);
675 CNewMenu::HandleMenuMsg(UINT uMsg
, WPARAM wParam
, LPARAM lParam
)
682 CNewMenu::HandleMenuMsg2(UINT uMsg
, WPARAM wParam
, LPARAM lParam
, LRESULT
*plResult
)
688 MEASUREITEMSTRUCT
* lpmis
= reinterpret_cast<MEASUREITEMSTRUCT
*>(lParam
);
689 if (!lpmis
|| lpmis
->CtlType
!= ODT_MENU
)
692 if (lpmis
->itemWidth
< (UINT
)GetSystemMetrics(SM_CXMENUCHECK
))
693 lpmis
->itemWidth
= GetSystemMetrics(SM_CXMENUCHECK
);
694 if (lpmis
->itemHeight
< 16)
695 lpmis
->itemHeight
= 16;
703 DRAWITEMSTRUCT
* lpdis
= reinterpret_cast<DRAWITEMSTRUCT
*>(lParam
);
704 if (!lpdis
|| lpdis
->CtlType
!= ODT_MENU
)
707 DWORD id
= LOWORD(lpdis
->itemID
) - m_idCmdFirst
;
710 hIcon
= m_hiconFolder
;
715 SHELLNEW_ITEM
*pItem
= FindItemFromIdOffset(id
);
717 hIcon
= pItem
->hIcon
;
723 DrawIconEx(lpdis
->hDC
,
725 lpdis
->rcItem
.top
+ (lpdis
->rcItem
.bottom
- lpdis
->rcItem
.top
- 16) / 2,
740 CNewMenu::Initialize(LPCITEMIDLIST pidlFolder
,
741 IDataObject
*pdtobj
, HKEY hkeyProgID
)
743 m_pidlFolder
= ILClone(pidlFolder
);
745 /* Load folder and shortcut icons */
746 m_hiconFolder
= (HICON
)LoadImage(shell32_hInstance
, MAKEINTRESOURCE(IDI_SHELL_FOLDER
), IMAGE_ICON
, 16, 16, LR_SHARED
);
747 m_hiconLink
= (HICON
)LoadImage(shell32_hInstance
, MAKEINTRESOURCE(IDI_SHELL_SHORTCUT
), IMAGE_ICON
, 16, 16, LR_SHARED
);