base/shell/rshell/CStartMenu.cpp

   1 /*
   2  * ReactOS Explorer
   3  *
   4  * Copyright 2014 Giannis Adamopoulos
   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 Street, Fifth Floor, Boston, MA  02110-1301  USA
  19  */
  20
  21 #define TEST_TRACKPOPUPMENU_SUBMENUS
  22
  23 #include "precomp.h"
  24
  25 /* NOTE: The following constants *MUST NOT* be changed because
  26          they're hardcoded and need to be the exact values
  27          in order to get the start menu to work! */
  28 #define IDM_PROGRAMS                504
  29 #define IDM_FAVORITES               507
  30 #define IDM_DOCUMENTS               501
  31 #define IDM_SETTINGS                508
  32 #define IDM_CONTROLPANEL            505
  33 #define IDM_SECURITY                5001
  34 #define IDM_NETWORKCONNECTIONS      557
  35 #define IDM_PRINTERSANDFAXES        510
  36 #define IDM_TASKBARANDSTARTMENU     413
  37 #define IDM_SEARCH                  520
  38 #define IDM_HELPANDSUPPORT          503
  39 #define IDM_RUN                     401
  40 #define IDM_SYNCHRONIZE             553
  41 #define IDM_LOGOFF                  402
  42 #define IDM_DISCONNECT              5000
  43 #define IDM_UNDOCKCOMPUTER          410
  44 #define IDM_SHUTDOWN                506
  45 #define IDM_LASTSTARTMENU_SEPARATOR 450
  46
  47 /*
  48  * TODO:
  49  * 1. append the start menu contents from all users
  50  * 2. implement the context menu for start menu entries (programs, control panel, network connetions, printers)
  51  * 3. filter out programs folder from the shell folder part of the start menu
  52  * 4. showing the programs start menu is SLOW compared to windows. this needs some investigation
  53  */
  54
  55 class CShellMenuCallback :
  56     public CComObjectRootEx<CComMultiThreadModelNoCS>,
  57     public IShellMenuCallback
  58 {
  59 private:
  60
  61     HWND m_hwndTray;
  62     CComPtr<IShellMenu> m_pShellMenu;
  63     CComPtr<IBandSite> m_pBandSite;
  64     CComPtr<IDeskBar> m_pDeskBar;
  65     CComPtr<ITrayPriv> m_pTrayPriv;
  66
  67     HRESULT OnInitMenu()
  68     {
  69         HMENU hmenu;
  70         HRESULT hr;
  71
  72         if (m_pTrayPriv.p)
  73             return S_OK;
  74
  75         hr = IUnknown_GetSite(m_pDeskBar, IID_PPV_ARG(ITrayPriv, &m_pTrayPriv));
  76         hr = IUnknown_GetWindow(m_pTrayPriv, &m_hwndTray);
  77         hr = m_pTrayPriv->AppendMenuW(&hmenu);
  78 #ifndef TEST_TRACKPOPUPMENU_SUBMENUS
  79         hr = m_pShellMenu->SetMenu(hmenu, NULL, SMSET_BOTTOM);
  80 #else
  81         hr = m_pShellMenu->SetMenu(hmenu, m_hwndTray, SMSET_BOTTOM);
  82 #endif
  83
  84         return hr;
  85     }
  86
  87     HRESULT OnGetInfo(LPSMDATA psmd, SMINFO *psminfo)
  88     {
  89         int iconIndex = 0;
  90
  91         switch (psmd->uId)
  92         {
  93         case IDM_PROGRAMS:  iconIndex = -20; break;
  94         case IDM_FAVORITES: iconIndex = -173; break;
  95         case IDM_DOCUMENTS: iconIndex = -21; break;
  96         case IDM_SETTINGS: iconIndex = -22; break;
  97         case IDM_CONTROLPANEL: iconIndex = -22; break;
  98         //case IDM_SECURITY: iconIndex = -21; break;
  99         case IDM_NETWORKCONNECTIONS: iconIndex = -257; break;
 100         case IDM_PRINTERSANDFAXES: iconIndex = -138; break;
 101         case IDM_TASKBARANDSTARTMENU: iconIndex = -40; break;
 102         case IDM_SEARCH: iconIndex = -23; break;
 103         case IDM_HELPANDSUPPORT: iconIndex = -24; break;
 104         case IDM_RUN: iconIndex = -25; break;
 105         //case IDM_SYNCHRONIZE: iconIndex = -21; break;
 106         case IDM_LOGOFF: iconIndex = -45; break;
 107         //case IDM_DISCONNECT: iconIndex = -21; break;
 108         //case IDM_UNDOCKCOMPUTER: iconIndex = -21; break;
 109         case IDM_SHUTDOWN: iconIndex = -28; break;
 110         default:
 111             return S_FALSE;
 112         }
 113
 114         if (iconIndex)
 115         {
 116             if ((psminfo->dwMask & SMIM_ICON) != 0)
 117                 psminfo->iIcon = Shell_GetCachedImageIndex(L"shell32.dll", iconIndex, FALSE);
 118 #ifdef TEST_TRACKPOPUPMENU_SUBMENUS
 119             if ((psminfo->dwMask & SMIM_FLAGS) != 0)
 120                 psminfo->dwFlags |= SMIF_TRACKPOPUP;
 121 #endif
 122         }
 123         return S_OK;
 124     }
 125
 126     HRESULT OnGetSubMenu(LPSMDATA psmd, REFIID iid, void ** pv)
 127     {
 128         HRESULT hr;
 129         int csidl = 0;
 130         IShellMenu *pShellMenu;
 131
 132         switch (psmd->uId)
 133         {
 134         case IDM_PROGRAMS:  csidl = CSIDL_PROGRAMS; break;
 135         case IDM_FAVORITES: csidl = CSIDL_FAVORITES; break;
 136         case IDM_DOCUMENTS: csidl = CSIDL_RECENT; break;
 137         }
 138
 139 #ifndef USE_BUILTIN_MENUBAND
 140         hr = CoCreateInstance(CLSID_MenuBand,
 141             NULL,
 142             CLSCTX_INPROC_SERVER,
 143             IID_PPV_ARG(IShellMenu, &pShellMenu));
 144 #else
 145         hr = CMenuBand_Constructor(IID_PPV_ARG(IShellMenu, &pShellMenu));
 146 #endif
 147
 148         hr = pShellMenu->Initialize(this, 0, ANCESTORDEFAULT, SMINIT_VERTICAL);
 149
 150         if (csidl)
 151         {
 152             LPITEMIDLIST pidlStartMenu;
 153             IShellFolder *psfDestop, *psfStartMenu;
 154
 155             hr = SHGetFolderLocation(NULL, csidl, 0, 0, &pidlStartMenu);
 156             hr = SHGetDesktopFolder(&psfDestop);
 157             hr = psfDestop->BindToObject(pidlStartMenu, NULL, IID_PPV_ARG(IShellFolder, &psfStartMenu));
 158
 159             hr = pShellMenu->SetShellFolder(psfStartMenu, NULL, NULL, 0);
 160         }
 161         else
 162         {
 163             MENUITEMINFO mii;
 164             mii.cbSize = sizeof(mii);
 165             mii.fMask = MIIM_SUBMENU;
 166             if (GetMenuItemInfoW( psmd->hmenu, psmd->uId, FALSE, &mii))
 167             {
 168                 hr = pShellMenu->SetMenu(mii.hSubMenu, NULL, SMSET_BOTTOM);
 169             }
 170         }
 171         return pShellMenu->QueryInterface(iid, pv);
 172     }
 173
 174     HRESULT OnGetContextMenu(LPSMDATA psmd, REFIID iid, void ** pv)
 175     {
 176         if (psmd->uId == IDM_PROGRAMS ||
 177             psmd->uId == IDM_CONTROLPANEL ||
 178             psmd->uId == IDM_NETWORKCONNECTIONS ||
 179             psmd->uId == IDM_PRINTERSANDFAXES)
 180         {
 181             //UNIMPLEMENTED
 182         }
 183
 184         return S_FALSE;
 185     }
 186
 187     HRESULT OnGetObject(LPSMDATA psmd, REFIID iid, void ** pv)
 188     {
 189         if (IsEqualIID(iid , IID_IShellMenu))
 190             return OnGetSubMenu(psmd, iid, pv);
 191         else if (IsEqualIID( iid, IID_IContextMenu))
 192             return OnGetContextMenu(psmd, iid, pv);
 193
 194         return S_FALSE;
 195     }
 196
 197     HRESULT OnExec(LPSMDATA psmd)
 198     {
 199         if(psmd->uId == IDM_CONTROLPANEL)
 200             ShellExecuteW(NULL, L"open", L"::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}", NULL,NULL, 1);
 201         else if(psmd->uId == IDM_NETWORKCONNECTIONS)
 202             ShellExecuteW(NULL, L"open", L"::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{7007ACC7-3202-11D1-AAD2-00805FC1270E}", NULL,NULL, 1);
 203         else if(psmd->uId == IDM_PRINTERSANDFAXES)
 204             ShellExecuteW(NULL, L"open", L"::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{2227A280-3AEA-1069-A2DE-08002B30309D}", NULL,NULL, 1);
 205         else
 206             PostMessageW( m_hwndTray, WM_COMMAND, psmd->uId, 0);
 207
 208         return S_OK;
 209     }
 210
 211 public:
 212
 213     DECLARE_NOT_AGGREGATABLE(CShellMenuCallback)
 214     DECLARE_PROTECT_FINAL_CONSTRUCT()
 215     BEGIN_COM_MAP(CShellMenuCallback)
 216         COM_INTERFACE_ENTRY_IID(IID_IShellMenuCallback, IShellMenuCallback)
 217     END_COM_MAP()
 218
 219     void Initialize(
 220         IShellMenu* pShellMenu,
 221         IBandSite* pBandSite,
 222         IDeskBar* pDeskBar)
 223     {
 224         m_pShellMenu.Attach(pShellMenu);
 225         m_pBandSite.Attach(pBandSite);
 226         m_pDeskBar.Attach(pDeskBar);
 227     }
 228
 229     ~CShellMenuCallback()
 230     {
 231         m_pShellMenu.Release();
 232         m_pBandSite.Release();
 233         m_pDeskBar.Release();
 234     }
 235
 236     HRESULT STDMETHODCALLTYPE CallbackSM(
 237         LPSMDATA psmd,
 238         UINT uMsg,
 239         WPARAM wParam,
 240         LPARAM lParam)
 241     {
 242         switch (uMsg)
 243         {
 244         case SMC_INITMENU:
 245                 return OnInitMenu();
 246         case SMC_GETINFO:
 247                 return OnGetInfo(psmd, reinterpret_cast<SMINFO*>(lParam));
 248         case SMC_GETOBJECT:
 249                 return OnGetObject(psmd, *reinterpret_cast<IID *>(wParam), reinterpret_cast<void **>(lParam));
 250         case SMC_EXEC:
 251                 return OnExec(psmd);
 252         case SMC_SFEXEC:
 253                 m_pTrayPriv->Execute(psmd->psf, psmd->pidlItem);
 254                 break;
 255         }
 256
 257     return S_FALSE;
 258     }
 259 };
 260
 261 extern "C"
 262 HRESULT
 263 CStartMenu_Constructor(
 264     REFIID riid,
 265     void **ppv)
 266 {
 267     IShellMenu* pShellMenu;
 268     IBandSite* pBandSite;
 269     IDeskBar* pDeskBar;
 270     LPITEMIDLIST pidlStartMenu;
 271
 272     HRESULT hr;
 273     IShellFolder *shellFolder;
 274     IShellFolder *psfStartMenu;
 275
 276 #ifndef USE_BUILTIN_MENUBAND
 277     hr = CoCreateInstance(CLSID_MenuBand,
 278                           NULL,
 279                           CLSCTX_INPROC_SERVER,
 280                           IID_PPV_ARG(IShellMenu, &pShellMenu));
 281 #else
 282     hr = CMenuBand_Constructor(IID_PPV_ARG(IShellMenu, &pShellMenu));
 283 #endif
 284     if (FAILED(hr))
 285         return hr;
 286
 287 #ifndef USE_BUILTIN_MENUSITE
 288     hr = CoCreateInstance(CLSID_MenuBandSite,
 289                           NULL,
 290                           CLSCTX_INPROC_SERVER,
 291                           IID_PPV_ARG(IBandSite, &pBandSite));
 292 #else
 293     hr = CMenuSite_Constructor(IID_PPV_ARG(IBandSite, &pBandSite));
 294 #endif
 295     if (FAILED(hr))
 296         return hr;
 297
 298 #ifndef USE_BUILTIN_MENUDESKBAR
 299     hr = CoCreateInstance(CLSID_MenuDeskBar,
 300                           NULL,
 301                           CLSCTX_INPROC_SERVER,
 302                           IID_PPV_ARG(IDeskBar, &pDeskBar));
 303 #else
 304     hr = CMenuDeskBar_Constructor(IID_PPV_ARG(IDeskBar, &pDeskBar));
 305 #endif
 306     if (FAILED(hr))
 307         return hr;
 308
 309     CComObject<CShellMenuCallback> *pCallback;
 310     hr = CComObject<CShellMenuCallback>::CreateInstance(&pCallback);
 311     if (FAILED(hr))
 312         return hr;
 313     pCallback->AddRef(); // CreateInstance returns object with 0 ref count */
 314     pCallback->Initialize(pShellMenu, pBandSite, pDeskBar);
 315
 316     pShellMenu->Initialize(pCallback, -1, 0, SMINIT_TOPLEVEL | SMINIT_VERTICAL);
 317     if (FAILED(hr))
 318         return hr;
 319
 320     /* FIXME: Use CLSID_MergedFolder class and IID_IAugmentedShellFolder2 interface here */
 321     /* CLSID_MergedFolder 26fdc864-be88-46e7-9235-032d8ea5162e */
 322     /* IID_IAugmentedShellFolder2 8db3b3f4-6cfe-11d1-8ae9-00c04fd918d0 */
 323     hr = SHGetFolderLocation(NULL, CSIDL_STARTMENU, 0, 0, &pidlStartMenu);
 324     hr = SHGetDesktopFolder(&shellFolder);
 325     hr = shellFolder->BindToObject(pidlStartMenu, NULL, IID_IShellFolder, (void**)&psfStartMenu);
 326
 327     hr = pShellMenu->SetShellFolder(psfStartMenu, NULL, NULL, 0);
 328
 329     hr =  pDeskBar->SetClient(pBandSite);
 330     if (FAILED(hr))
 331         return hr;
 332
 333     hr =  pBandSite->AddBand(pShellMenu);
 334     if (FAILED(hr))
 335         return hr;
 336
 337     return pDeskBar->QueryInterface(riid, ppv);
 338 }