Changeset 576 for branches/erasershellext/cpp/CtxMenu.cpp

Timestamp:

11/15/2008 5:05:52 PM (5 years ago)

Author:

cjax

Message:

Some functions

File:

• branches/erasershellext/cpp/CtxMenu.cpp (modified) (8 diffs)

branches/erasershellext/cpp/CtxMenu.cpp

@@ -9 +9 @@
 using namespace Eraser;
 /////////////////////////////////////////////////////////////////////////////
-// COpenWithCtxMenuExt
 
 HRESULT CCtxMenu::Initialize (
-    LPCITEMIDLIST   pidlFolder,
-    LPDATAOBJECT    pDataObj,
-    HKEY            hProgID )
-{
-
+                        LPCITEMIDLIST   pidlFolder,
+                        LPDATAOBJECT    pDataObj,
+                        HKEY            hProgID )
+{
     FORMATETC fmt = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
     STGMEDIUM stg = { TYMED_HGLOBAL };
     HDROP     hDrop;
 
-    // Look for CF_HDROP data in the data object.
-    if ( FAILED( pDataObj->GetData ( &fmt, &stg )))
-        // Nope! Return an "invalid argument" error back to Explorer.
-        return E_INVALIDARG;
-
-    // Get a pointer to the actual data.
-    hDrop = (HDROP) GlobalLock ( stg.hGlobal );
-
-    // Make sure it worked.
-    if ( NULL == hDrop )
-        return E_INVALIDARG;
-
-    // Sanity check - make sure there is at least one filename.
+    // Look for CF_HDROP data in the data object.
+    if ( FAILED( pDataObj->GetData ( &fmt, &stg )))
+        // Nope! Return an "invalid argument" error back to Explorer.
+        return E_INVALIDARG;
+
+    // Get a pointer to the actual data.
+    hDrop = (HDROP) GlobalLock ( stg.hGlobal );
+
+    // Make sure it worked.
+    if ( NULL == hDrop )
+        return E_INVALIDARG;
+
+    // Sanity check - make sure there is at least one filename.
     UINT uNumFiles = DragQueryFile ( hDrop, 0xFFFFFFFF, NULL, 0 );
 
-    if (!uNumFiles)
+    if (!uNumFiles)
     {
         GlobalUnlock(stg.hGlobal);
@@ -42 +40 @@
         return E_INVALIDARG;
     }
-    
+
     HRESULT hr = S_OK;
 
@@ -55 +53 @@
     }
 
-    // TODO: load Eraser Icon
-    // m_szEraserIcon
-
-    GlobalUnlock ( stg.hGlobal );
-    ReleaseStgMedium ( &stg );
-
-    return hr;
+    // TODO: load Eraser Icon and store HBITMAP in m_szEraserIcon
+
+    GlobalUnlock ( stg.hGlobal );
+    ReleaseStgMedium ( &stg );
+
+    return hr;
 }
 /*
@@ -70 +67 @@
 |                   |
 +-------------------+    +-------------------+
-|(ICON) Eraser v6 > |    | Schedule Selected |
-+-------------------+    +-------------------+
-|                   |    |(*) Select Default | // Eraser.Manager.Wipe Algorithms popup
-|                   |    |(*) Chose algorithm| // set algorithm for this query only
+|(ICON) Eraser v6 > |    | Erase selected    | //--> erase the files immediately using defaults
++-------------------+    | Schedule Selected | //--> open the scheduler menu, with files/folders filled in
+|                   |    +-------------------+
+|                   |    | Secure move       | //--> secure move the files
+|                   |    +-------------------+ // Eraser.Manager Algorithms popup
+|                   |    |(*) Customise      | // set algorithm for this query only
 +-------------------+    +-------------------+
 */
 
+/*
+we have eraser shell scripts (.eshlls) where we use them
+to generate shell extention menus dynamically.
+the syntax is simple.
+This project should be aimed at the 6.1 release.
+
+BEGIN @MENU : POSITION($INDEX)
+    [ICON="PATH"]
+    [TEXT="DISPLAY_TEXT"]
+
+    [FLAGS=MIIM_*]          
+        //MIIM_BITMAP
+        //MIIM_CHECKMARKS
+        //MIIM_DATA
+        //MIIM_FTYPE
+        //MIIM_ID
+        //MIIM_STATE
+        //MIIM_STRING
+        //MIIM_SUBMENU
+        //MIIM_TYPE
+
+    [TYPE=MF_*]
+        //MF_APPEND,
+        //MF_BITMAP,
+        //MF_BYCOMMAND,
+        //MF_BYPOSITION
+        //MF_CHECKED
+        //MF_DEFAULT
+        //MF_DELETE
+        //MF_DISABLED
+        //MF_ENABLED
+        //MF_END
+        //MF_GRAYED
+        //MF_HELP
+        //MF_HILITE
+        //MF_INSERT
+        //MF_MENUBARBREAK
+        //MF_MENUBREAK
+        //MF_MOUSESELECT
+        //MF_OWNERDRAW
+        //MF_POPUP
+        //MF_POPUP
+        //MF_REMOVE,
+        //MF_RIGHTJUSTIFY,
+        //MF_SEPARATOR,
+        //MF_STRING,
+        //MF_SYSMENU,
+        //MF_UNCHECKED,
+        //MF_UNHILITE
+        //MF_USECHECKBITMAPS
+END
+
+// Desirable to have bitmaps cached
+@MENU 
+{
+    BITMAP = "FILE";
+    TEXT="Eraser";
+    BITMAP CHECKED="FILE";
+    BITMAP UNCHECKED="FILE";
+
+    // submenu creation
+    @MENU 
+    {
+        @CHECKBOX 
+        {
+        }
+        @SEPERATOR
+    }
+}
+
+UINT     cbSize;
+UINT     fMask;
+UINT     fType;         // used if MIIM_TYPE (4.0) or MIIM_FTYPE (>4.0)
+UINT     fState;        // used if MIIM_STATE
+UINT     wID;           // used if MIIM_ID
+HMENU    hSubMenu;      // used if MIIM_SUBMENU
+HBITMAP  hbmpChecked;   // used if MIIM_CHECKMARKS
+HBITMAP  hbmpUnchecked; // used if MIIM_CHECKMARKS
+ULONG_PTR dwItemData;   // used if MIIM_DATA
+__field_ecount_opt(cch) LPSTR    dwTypeData;    // used if MIIM_TYPE (4.0) or MIIM_STRING (>4.0)
+UINT     cch;           // used if MIIM_TYPE (4.0) or MIIM_STRING (>4.0)
+HBITMAP  hbmpItem;      // used if MIIM_BITMAP
+
+*/
+
+/**
+ * return a seperate MENUITEMINFO structure */
+static MENUITEMINFO* GetSeperator()
+{
+    MENUITEMINFO *mii = new MENUITEMINFO();     
+    mii->cbSize = sizeof(MENUITEMINFO);
+    mii->fMask = MIIM_TYPE;
+    mii->fType = MF_SEPARATOR;
+    return mii;
+}
+
 HRESULT CCtxMenu::QueryContextMenu (
-    HMENU hmenu,
-    UINT  uMenuIndex, 
-    UINT  uidFirstCmd, 
-    UINT  uidLastCmd,
-    UINT  uFlags )
-{
-    // If the flags include CMF_DEFAULTONLY then we shouldn't do anything.
-    if ( uFlags & CMF_DEFAULTONLY )
-        return MAKE_HRESULT ( SEVERITY_SUCCESS, FACILITY_NULL, 0 );
-
-    // First, create and populate a submenu.
-    HMENU hSubmenu = CreatePopupMenu();
-    UINT uID = uidFirstCmd;
+                HMENU hmenu,
+                UINT  uMenuIndex,
+                UINT  uidFirstCmd, 
+                UINT  uidLastCmd,
+                UINT  uFlags )
+{
+    // If the flags include CMF_DEFAULTONLY then we shouldn't do anything.
+    if ( uFlags & CMF_DEFAULTONLY )
+        return MAKE_HRESULT ( SEVERITY_SUCCESS, FACILITY_NULL, 0 );
+
+    // First, create and populate a submenu.
     UINT ctrPos = 0;
-
-    InsertMenu ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("&Erase"   ) );
-    InsertMenu ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("&Set Algorithm" ) );
-    InsertMenu ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("S&chedule" ) );
-    InsertMenu ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("Se&lect Default") );
-    
-    // Insert the submenu into the ctx menu provided by Explorer.
-    {// sperator
-        MENUITEMINFO mii = { sizeof(MENUITEMINFO) };        
-        mii.fMask = MIIM_TYPE;
-        mii.fType = MF_SEPARATOR;
-        InsertMenuItem(hmenu, uMenuIndex++, TRUE, &mii);
-    }
-
+    UINT uID = uidFirstCmd; 
+    HMENU hSubmenu = CreatePopupMenu(); 
+
+
+    /* Remember this order is defined _Statically_ on the CEraserLPVERB enum */
+    /* -------------------------------- Submenu -------------------------------- */
+    /* [0] */ InsertMenu     ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("&Erase"    ) );
+    /* [1] */ InsertMenu     ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("&Schedule" ) );
+    /* [2] */ InsertMenu     ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("Erase on &Restart" ) );
+    /* ------------------------------------------------------------------------- */
+    /* [3] */ InsertMenuItem ( hSubmenu, ctrPos++, TRUE, GetSeperator() );
+    /* [4] */ InsertMenu     ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("Secure &Move" ) ); 
+    /* ------------------------------------------------------------------------- */
+    /* [5] */ InsertMenuItem ( hSubmenu, ctrPos++, TRUE, GetSeperator() );
+    /* [6] */ InsertMenu     ( hSubmenu, ctrPos++, MF_BYPOSITION, uID++, _T("&Customise") );    
+    /* ------------------------------ Submenu end ------------------------------ */
+
+    // Insert the submenu into the ctx menu provided by Explorer.
+    /* ------------------------------------------------------------------------- */
+    InsertMenuItem ( hSubmenu, uMenuIndex++, TRUE, GetSeperator() );
     {
         MENUITEMINFO mii = { sizeof(MENUITEMINFO) };
@@ -114 +213 @@
         InsertMenuItem(hmenu, uMenuIndex++, TRUE, &mii);
     }
-    
-    {// sperator
-        MENUITEMINFO mii = { sizeof(MENUITEMINFO) };        
-        mii.fMask = MIIM_TYPE;
-        mii.fType = MF_SEPARATOR;
-        InsertMenuItem(hmenu, uMenuIndex++, TRUE, &mii);
-    }
+    /* ------------------------------------------------------------------------- */
+    InsertMenuItem ( hSubmenu, uMenuIndex++, TRUE, GetSeperator() );
 
     // Set the bitmap for the register item.
-    if ( m_szEraserIcon != nullpointer(HBITMAP*) )
-    {
-        DebugMessageBox(DebugHWND,
-                    _T("Loading Eraser Icon Bitmap"),
-                    _T("Eraser Shell Extention - Debug"),
-                    MB_OK);
-
-        SetMenuItemBitmaps(hmenu, uMenuIndex, MF_BYPOSITION, *m_szEraserIcon, NULL );
-    }
-
-    return MAKE_HRESULT ( SEVERITY_SUCCESS, FACILITY_NULL, uID - uidFirstCmd );
+    if ( m_szEraserIcon != ((HBITMAP)0) )
+        SetMenuItemBitmaps(hmenu, uMenuIndex++, MF_BYPOSITION, m_szEraserIcon, NULL );
+
+    return MAKE_HRESULT ( SEVERITY_SUCCESS, FACILITY_NULL, uID - uidFirstCmd );
 }
 
 HRESULT CCtxMenu::GetCommandString (
-    UINT  idCmd,   
-    UINT uFlags,
-    UINT* pwReserved,
-    LPSTR pszName,
-    UINT  cchMax )
-{
-USES_CONVERSION;
-
-    // Check idCmd, it must be 0 or 1 since we have two menu items.
-    if ( idCmd > 4 )
-        return E_INVALIDARG;
-
-    // If Explorer is asking for a help string, copy our string into the
-    // supplied buffer.
-    if (!(uFlags & GCS_HELPTEXT)) 
-        return E_INVALIDARG;
-
+                                    UINT  idCmd,   
+                                    UINT uFlags,
+                                    UINT* pwReserved,
+                                    LPSTR pszName,
+                                    UINT  cchMax )
+{
+    USES_CONVERSION;
+
+    // Check idCmd, it must be 0 or 1 since we have two menu items.
+    if ( idCmd > 4 )
+        return E_INVALIDARG;
+
+    // If Explorer is asking for a help string, copy our string into the
+    // supplied buffer.
+    if (!(uFlags & GCS_HELPTEXT)) 
+        return E_INVALIDARG;
+ 
     static LPCTSTR szNotepadText = _T("Open the selected file in Notepad");
     static LPCTSTR szIEText = _T("Open the selected file in Internet Explorer");
@@ -162 +249 @@
         // Unicode string copy API.
         lstrcpynW ( (LPWSTR) pszName, T2CW(pszText), cchMax );
-    else
-        // Use the ANSI string copy API to return the help string.
-        lstrcpynA ( pszName, T2CA(pszText), cchMax );
-    
-    return S_OK;
+    else
+        // Use the ANSI string copy API to return the help string.
+        lstrcpynA ( pszName, T2CA(pszText), cchMax );
+
+    return S_OK;
 }
 
@@ -175 +262 @@
 3) Set default algorithm
 */
+
+#define foreach(iter, container) \
+    for(string_list::iterator iter = container.begin(); \
+    iter != container.end(); iter++)
+
 HRESULT CCtxMenu::InvokeCommand ( LPCMINVOKECOMMANDINFO pCmdInfo )
 {
-    // If lpVerb really points to a string, ignore this function call and bail out.
-    if ( 0 != HIWORD( pCmdInfo->lpVerb ))
-        return E_INVALIDARG;
-
-    // Get the command index.
-    switch 
-        (
-        LOWORD(pCmdInfo->lpVerb + 1 + CERASER_ENUM(CEraserLPVERB::LPVERB_FRIST))
-        )
-    {
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_ERASE_FILE):
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_ERASE_FREE_SPACE):
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_SET_DEFAULT_FILE_ALGORITHM):
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_SET_DEFAULT_FREE_SPACE_ALGORITHM):
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_SET_FILE_ALGORITHM):
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_SET_FREE_SPACE_ALGORITHM):
-        {
-            WCHAR szMsg [MAX_PATH + 32];
-            wsprintf ( szMsg, _T("Not Implemented Querry!\n\nID = %d\n\n"), LOWORD(pCmdInfo->lpVerb) );
-            MessageBox ( pCmdInfo->hwnd, szMsg, _T("Eraser v6 - Shell Extention Query"), MB_ICONINFORMATION );
-            return E_INVALIDARG;
-        }
-        break; // unreachable code
-        
-    case CERASER_ENUM(CEraserLPVERB::LPVERB_SHOW_FILE_NAME):
-        {
-            std::basic_string<WCHAR> map = _T("The selected files were:\n\n");
-
-            for(string_list::iterator i = m_szSelectedFiles.begin();
-                i != m_szSelectedFiles.end(); i++, map += _T("\r\n"))
-                    map += i->c_str();
-
-            MessageBox ( pCmdInfo->hwnd, (LPWSTR)map.c_str(), _T("Eraser v6 - Shell Extention Query"), MB_ICONINFORMATION );
-
-            return S_OK; 
-        }
-        break; // unreachable code
+    // If lpVerb really points to a string, ignore this function call and bail out.
+    if ( HIWORD( pCmdInfo->lpVerb )  != 0)
+        return E_INVALIDARG;
+
+    // Get the command index.
+    switch( LOWORD(pCmdInfo->lpVerb + 1) )
+    {
+    case CERASER_ENUM(CEraserLPVERBS::CERASER_ERASE):
+        {
+            List<String ^> ^files = gcnew List<String ^>(m_szSelectedFiles.size());
+
+            foreach(file, m_szSelectedFiles)
+            {
+                files->Add( UnmanagedToManagedString(*file) );
+            }
+            EraseFiles(files);
+        }
+    case CERASER_ENUM(CEraserLPVERBS::CERASER_SECURE_MOVE):
+        {
+            List<String ^> ^files = gcnew List<String ^>(m_szSelectedFiles.size());
+
+            foreach(file, m_szSelectedFiles)
+            {
+                files->Add( UnmanagedToManagedString(*file) );
+            }
+            EraseFiles(files);
+        }
+        // NOT IMPLEMENTED METHODS
+    case CERASER_ENUM(CEraserLPVERBS::CERASER_ERASE_ON_RESTART):
+        {
+            List<String ^> ^files = gcnew List<String ^>(m_szSelectedFiles.size());
+
+            foreach(file, m_szSelectedFiles)
+            {
+                files->Add( UnmanagedToManagedString(*file) );
+            }
+            EraseFiles(files);
+        }
+    case CERASER_ENUM(CEraserLPVERBS::CERASER_SCHEDULE):
+        {
+            foreach(file, m_szSelectedFiles)
+            {
+            }
+        }
+    case CERASER_ENUM(CEraserLPVERBS::CERASER_CUSTOMISE):
+        {
+        }
     default:
         {
@@ -221 +323 @@
             return E_INVALIDARG;
         }
-        break; // unreachable code
-    }
-}
+        break; // unreachable code
+    }
+}