/****************************** Module Header ******************************\
Module Name:  FileContextMenuExt.cpp
Project:      CppShellExtContextMenuHandler
Copyright (c) Microsoft Corporation.

The code sample demonstrates creating a Shell context menu handler with C++. 

A context menu handler is a shell extension handler that adds commands to an 
existing context menu. Context menu handlers are associated with a particular 
file class and are called any time a context menu is displayed for a member 
of the class. While you can add items to a file class context menu with the 
registry, the items will be the same for all members of the class. By 
implementing and registering such a handler, you can dynamically add items to 
an object's context menu, customized for the particular object.

The example context menu handler adds the menu item "Display File Name (C++)"
to the context menu when you right-click a .cpp file in the Windows Explorer. 
Clicking the menu item brings up a message box that displays the full path 
of the .cpp file.

This source is subject to the Microsoft Public License.
See http://www.microsoft.com/opensource/licenses.mspx#Ms-PL.
All other rights reserved.

THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, 
EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED 
WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
\***************************************************************************/

#include "ContextMenuExt.h"
#include "resource.h"
#include <strsafe.h>
#include <Shlwapi.h>

#include "RegUtils.h"
#include "MEGAinterface.h"

#pragma comment(lib, "shlwapi.lib")

extern HINSTANCE g_hInst;
extern long g_cDllRef;

#define IDM_UPLOAD             0  // The command's identifier offset
#define IDM_GETLINK            1
#define IDM_REMOVEFROMLEFTPANE 2
#define IDM_VIEWONMEGA         3
#define IDM_VIEWVERSIONS       4

ContextMenuExt::ContextMenuExt(void) : m_cRef(1),
    m_pszUploadMenuText(L"&Upload to MEGA"),
    m_pszUploadVerb("UploadToMEGA"),
    m_pwszUploadVerb(L"UploadToMEGA"),
    m_pszUploadVerbCanonicalName("UploadToMEGA"),
    m_pwszUploadVerbCanonicalName(L"UploadToMEGA"),
    m_pszUploadVerbHelpText("Upload to MEGA"),
    m_pwszUploadVerbHelpText(L"Upload to MEGA"),
    m_pszGetLinkMenuText(L"&Get MEGA link"),
    m_pszGetLinkVerb("GetMEGALink"),
    m_pwszGetLinkVerb(L"GetMEGALink"),
    m_pszGetLinkVerbCanonicalName("GetMEGALink"),
    m_pwszGetLinkVerbCanonicalName(L"GetMEGALink"),
    m_pszGetLinkVerbHelpText("Get MEGA link"),
    m_pwszGetLinkVerbHelpText(L"Get MEGA link"),
    m_pszRemoveFromLeftPaneMenuText(L"&Remove from left pane"),
    m_pszRemoveFromLeftPaneVerb("RemoveFromLeftPane"),
    m_pwszRemoveFromLeftPaneVerb(L"RemoveFromLeftPane"),
    m_pszRemoveFromLeftPaneVerbCanonicalName("RemoveFromLeftPane"),
    m_pwszRemoveFromLeftPaneVerbCanonicalName(L"RemoveFromLeftPane"),
    m_pszRemoveFromLeftPaneVerbHelpText("Remove from left pane"),
    m_pwszRemoveFromLeftPaneVerbHelpText(L"Remove from left pane"),
    m_pszViewOnMEGAVerb("ViewOnMEGA"),
    m_pwszViewOnMEGAVerb(L"ViewOnMEGA"),
    m_pszViewOnMEGAVerbCanonicalName("ViewOnMEGA"),
    m_pwszViewOnMEGAVerbCanonicalName(L"ViewOnMEGA"),
    m_pszViewOnMEGAVerbHelpText("View on MEGA"),
    m_pwszViewOnMEGAVerbHelpText(L"View on MEGA"),
    m_pszViewVersionsVerb("ViewVersions"),
    m_pwszViewVersionsVerb(L"ViewVersions"),
    m_pszViewVersionsVerbCanonicalName("ViewVersions"),
    m_pwszViewVersionsVerbCanonicalName(L"ViewVersions"),
    m_pszViewVersionsVerbHelpText("View previous versions"),
    m_pwszViewVersionsVerbHelpText(L"View previous versions")
{
    hIcon = NULL;
    m_hMenuBmp = NULL;
    legacyIcon = true;

    InterlockedIncrement(&g_cDllRef);

    HINSTANCE UxThemeDLL = NULL;
    WCHAR systemPath[MAX_PATH];
    UINT len = GetSystemDirectory(systemPath, MAX_PATH);
    if (len + 20 >= MAX_PATH)
    {
        UxThemeDLL = LoadLibrary(L"UXTHEME.DLL");
    }
    else
    {
        StringCchPrintfW(systemPath + len, MAX_PATH - len, L"\\UXTHEME.DLL");
        UxThemeDLL = LoadLibrary(systemPath);
    }

    if (UxThemeDLL)
    {
        GetBufferedPaintBits = (pGetBufferedPaintBits)::GetProcAddress(UxThemeDLL, "GetBufferedPaintBits");
        BeginBufferedPaint = (pBeginBufferedPaint)::GetProcAddress(UxThemeDLL, "BeginBufferedPaint");
        EndBufferedPaint = (pEndBufferedPaint)::GetProcAddress(UxThemeDLL, "EndBufferedPaint");
        if (GetBufferedPaintBits && BeginBufferedPaint && EndBufferedPaint)
        {
            legacyIcon = false;
        }
        else
        {
            FreeLibrary(UxThemeDLL);
        }
    }

    if (g_hInst)
    {
        hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_ICON4), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
        if (hIcon)
        {
            m_hMenuBmp = legacyIcon ? getBitmapLegacy(hIcon) : getBitmap(hIcon);
        }
    }
}

HBITMAP ContextMenuExt::getBitmapLegacy(HICON hIcon)
{
    __try
    {
        RECT rect;
        rect.right = ::GetSystemMetrics(SM_CXMENUCHECK);
        rect.bottom = ::GetSystemMetrics(SM_CYMENUCHECK);
        rect.left = rect.top  = 0;

        // Create a compatible DC
        HDC dst_hdc = ::CreateCompatibleDC(NULL);
        if (dst_hdc == NULL) return NULL;

        // Create a new bitmap of icon size
        HBITMAP bmp = ::CreateCompatibleBitmap(dst_hdc, rect.right, rect.bottom);
        if (bmp == NULL)
        {
            DeleteDC(dst_hdc);
            return NULL;
        }

        SelectObject(dst_hdc, bmp);
        SetBkColor(dst_hdc, RGB(255, 255, 255));
        ExtTextOut(dst_hdc, 0, 0, ETO_OPAQUE, &rect, NULL, 0, NULL);
        DrawIconEx(dst_hdc, 0, 0, hIcon, rect.right, rect.bottom, 0, NULL, DI_NORMAL);
        DeleteDC(dst_hdc);
        return bmp;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {
    }
    return NULL;
}

HBITMAP ContextMenuExt::getBitmap(HICON icon)
{
    __try
    {
        HBITMAP bitmap = NULL;
        HDC hdc = CreateCompatibleDC(NULL);
        if (!hdc) return NULL;

        int width   = GetSystemMetrics(SM_CXSMICON);
        int height  = GetSystemMetrics(SM_CYSMICON);
        if (!width || !height)
        {
            DeleteDC(hdc);
            return NULL;
        }

        RECT rect   = {0, 0, width, height};
        BITMAPINFO bmInfo = {0};
        bmInfo.bmiHeader.biSize        = sizeof(BITMAPINFOHEADER);
        bmInfo.bmiHeader.biPlanes      = 1;
        bmInfo.bmiHeader.biCompression = BI_RGB;
        bmInfo.bmiHeader.biWidth       = width;
        bmInfo.bmiHeader.biHeight      = height;
        bmInfo.bmiHeader.biBitCount    = 32;

        bitmap = CreateDIBSection(hdc, &bmInfo, DIB_RGB_COLORS, NULL, NULL, 0);
        if (!bitmap)
        {
            DeleteDC(hdc);
            return NULL;
        }

        HGDIOBJ h = SelectObject(hdc, bitmap);
        if (!h || (h == HGDI_ERROR))
        {
            DeleteDC(hdc);
            return NULL;
        }

        BLENDFUNCTION blendFunction = { AC_SRC_OVER, 0, 255, AC_SRC_ALPHA };
        BP_PAINTPARAMS paintParams = {0};
        paintParams.cbSize = sizeof(BP_PAINTPARAMS);
        paintParams.dwFlags = BPPF_ERASE;
        paintParams.pBlendFunction = &blendFunction;

        HDC newHdc;
        HPAINTBUFFER hPaintBuffer = BeginBufferedPaint(hdc, &rect, BPBF_DIB, &paintParams, &newHdc);
        if (hPaintBuffer)
        {
            DrawIconEx(newHdc, 0, 0, icon, width, height, 0, NULL, DI_NORMAL);
            EndBufferedPaint(hPaintBuffer, TRUE);
        }

        DeleteDC(hdc);
        return bitmap;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {
    }
    return NULL;
}

ContextMenuExt::~ContextMenuExt(void)
{
    if (m_hMenuBmp)
    {
        DeleteObject(m_hMenuBmp);
        m_hMenuBmp = NULL;
    }

    InterlockedDecrement(&g_cDllRef);
}

bool ContextMenuExt::isSynced(int type, int state)
{
    return (state == MegaInterface::FILE_SYNCED) ||
        ((type == MegaInterface::TYPE_FOLDER) && state == MegaInterface::FILE_SYNCING);
}

bool ContextMenuExt::isUnsynced(int state)
{
    return (state == MegaInterface::FILE_NOTFOUND);
}

void ContextMenuExt::processFile(HDROP hDrop, int i)
{
    WIN32_FILE_ATTRIBUTE_DATA fad;
    int characters = DragQueryFileW(hDrop, i, NULL, 0);
    int type = MegaInterface::TYPE_UNKNOWN;
    if (characters)
    {
        characters+=1; //NUL character
        std::string buffer;
        buffer.resize(characters*sizeof(wchar_t));
        int ok = DragQueryFileW(hDrop, i, (LPWSTR)buffer.data(), characters);
        ((LPWSTR)buffer.data())[characters-1]=L'\0'; //Ensure a trailing NUL character
        if (ok)
        {
            if (GetFileAttributesExW((LPCWSTR)buffer.data(),GetFileExInfoStandard,(LPVOID)&fad))
                type = (fad.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ?
                            MegaInterface::TYPE_FOLDER : MegaInterface::TYPE_FILE;

            int state = MegaInterface::getPathState((PCWSTR)buffer.data(), false);
            selectedFiles.push_back(buffer);
            pathStates.push_back(state);
            pathTypes.push_back(type);
            if (isSynced(type, state))
            {
                if (type == MegaInterface::TYPE_FOLDER)
                {
                    syncedFolders++;
                }
                else if (type == MegaInterface::TYPE_FILE)
                {
                    syncedFiles++;
                }
                else
                {
                    syncedUnknowns++;
                }
            }
            else if (isUnsynced(state))
            {
                if (type == MegaInterface::TYPE_FOLDER)
                {
                    unsyncedFolders++;
                }
                else if (type == MegaInterface::TYPE_FILE)
                {
                    unsyncedFiles++;
                }
                else
                {
                    unsyncedUnknowns++;
                }
            }

            if (type == MegaInterface::TYPE_FOLDER && !inLeftPane.size())
            {
                if (CheckLeftPaneIcon((wchar_t *)buffer.data(), false))
                {
                    inLeftPane = buffer;
                }
            }
        }
    }
}

void ContextMenuExt::requestUpload()
{
    for (unsigned int i = 0; i < selectedFiles.size(); i++)
    {
        if (isUnsynced(pathStates[i]))
        {
            MegaInterface::upload((PCWSTR)selectedFiles[i].data());
        }
    }
}

void ContextMenuExt::requestGetLinks()
{
    for (unsigned int i = 0; i < selectedFiles.size(); i++)
    {
        if (isSynced(pathTypes[i], pathStates[i]))
        {
            MegaInterface::pasteLink((PCWSTR)selectedFiles[i].data());
        }
    }
}

void ContextMenuExt::removeFromLeftPane()
{
    if (!inLeftPane.size())
    {
        return;
    }

    CheckLeftPaneIcon((wchar_t *)inLeftPane.data(), true);
}

void ContextMenuExt::viewOnMEGA()
{
    if (selectedFiles.size())
    {
        MegaInterface::viewOnMEGA((PCWSTR)selectedFiles[0].data());
    }
}

void ContextMenuExt::viewVersions()
{
    if (selectedFiles.size())
    {
        MegaInterface::viewVersions((PCWSTR)selectedFiles[0].data());
    }
}

#pragma region IUnknown

// Query to the interface the component supported.
IFACEMETHODIMP ContextMenuExt::QueryInterface(REFIID riid, void **ppv)
{
    __try
    {
        if (ppv == NULL)
        {
            return E_POINTER;
        }

        *ppv = NULL;
        if (riid == __uuidof (IContextMenu))
        {
            *ppv = (IContextMenu *) this;
        }
        else if (riid == __uuidof (IContextMenu2))
        {
            *ppv = (IContextMenu2 *) this;
        }
        else if (riid == __uuidof (IContextMenu3))
        {
            *ppv = (IContextMenu3 *) this;
        }
        else if (riid == __uuidof (IShellExtInit))
        {
            *ppv = (IShellExtInit *) this;
        }
        else if (riid == IID_IUnknown)
        {
            *ppv = this;
        }
        else
        {
            return E_NOINTERFACE;
        }

        AddRef();
        return S_OK;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {
    }
    return E_NOINTERFACE;
}

// Increase the reference count for an interface on an object.
IFACEMETHODIMP_(ULONG) ContextMenuExt::AddRef()
{
    return InterlockedIncrement(&m_cRef);
}

// Decrease the reference count for an interface on an object.
IFACEMETHODIMP_(ULONG) ContextMenuExt::Release()
{
    __try
    {
        ULONG cRef = InterlockedDecrement(&m_cRef);
        if (0 == cRef)
        {
            delete this;
        }

        return cRef;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {
    }
    return 0;
}

#pragma endregion


#pragma region IShellExtInit

// Initialize the context menu handler.
IFACEMETHODIMP ContextMenuExt::Initialize(
    LPCITEMIDLIST , LPDATAOBJECT pDataObj, HKEY )
{
    __try
    {
        if (NULL == pDataObj)
        {
            return E_INVALIDARG;
        }

        HRESULT hr = E_FAIL;

        FORMATETC fe = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
        STGMEDIUM stm;

        // The pDataObj pointer contains the objects being acted upon. In this
        // example, we get an HDROP handle for enumerating the selected files and
        // folders.
        selectedFiles.clear();
        pathStates.clear();
        pathTypes.clear();
        inLeftPane.clear();
        syncedFolders = syncedFiles = syncedUnknowns = 0;
        unsyncedFolders = unsyncedFiles = unsyncedUnknowns = 0;
        if (SUCCEEDED(pDataObj->GetData(&fe, &stm)))
        {
            // Get an HDROP handle.
            HDROP hDrop = static_cast<HDROP>(GlobalLock(stm.hGlobal));
            if (hDrop != NULL)
            {
                UINT nFiles = DragQueryFile(hDrop, 0xFFFFFFFF, NULL, 0);
                if (nFiles != 0)
                {
                    if (MegaInterface::startRequest())
                    {
                        for (unsigned int i = 0; i < nFiles; i++)
                        {
                            processFile(hDrop, i);
                        }

                        if (selectedFiles.size())
                        {
                            hr = S_OK;
                        }
                    }
                }

                GlobalUnlock(stm.hGlobal);
            }

            ReleaseStgMedium(&stm);
        }
        return hr;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {

    }

    // If any value other than S_OK is returned from the method, the context 
    // menu item is not displayed.
    return E_FAIL;
}

#pragma endregion


#pragma region IContextMenu

//
//   FUNCTION: FileContextMenuExt::QueryContextMenu
//
//   PURPOSE: The Shell calls IContextMenu::QueryContextMenu to allow the 
//            context menu handler to add its menu items to the menu. It 
//            passes in the HMENU handle in the hmenu parameter. The 
//            indexMenu parameter is set to the index to be used for the 
//            first menu item that is to be added.
//
IFACEMETHODIMP ContextMenuExt::QueryContextMenu(
    HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT , UINT uFlags)
{
    __try
    {
        // If uFlags include CMF_DEFAULTONLY then we should not do anything.
        if (CMF_DEFAULTONLY & uFlags)
        {
            return MAKE_HRESULT(SEVERITY_SUCCESS, 0, USHORT(0));
        }

        // Use either InsertMenu or InsertMenuItem to add menu items.
        // Learn how to add sub-menu from:
        // http://www.codeproject.com/KB/shell/ctxextsubmenu.aspx

        int lastItem = 0;
        if (unsyncedFolders || unsyncedFiles)
        {
            LPWSTR menuText = MegaInterface::getString(MegaInterface::STRING_UPLOAD, unsyncedFiles, unsyncedFolders);
            if (menuText)
            {
                MENUITEMINFO mii = { sizeof(mii) };
                mii.fMask = MIIM_BITMAP | MIIM_STRING | MIIM_FTYPE | MIIM_ID | MIIM_STATE;
                mii.wID = idCmdFirst + IDM_UPLOAD;
                mii.fType = MFT_STRING;
                mii.dwTypeData = menuText;
                mii.fState = MFS_ENABLED;
                mii.hbmpItem = (legacyIcon || !m_hMenuBmp) ? HBMMENU_CALLBACK : m_hMenuBmp;
                if (!InsertMenuItem(hMenu, indexMenu++, TRUE, &mii))
                {
                    delete menuText;
                    return HRESULT_FROM_WIN32(GetLastError());
                }
                delete menuText;
                lastItem = IDM_UPLOAD;
            }
        }

        if (syncedFolders || syncedFiles)
            {
                LPWSTR menuText = MegaInterface::getString(MegaInterface::STRING_GETLINK, syncedFiles, syncedFolders);
                if (menuText)
                {
                    MENUITEMINFO mii = { sizeof(mii) };
                    mii.fMask = MIIM_BITMAP | MIIM_STRING | MIIM_FTYPE | MIIM_ID | MIIM_STATE;
                    mii.wID = idCmdFirst + IDM_GETLINK;
                    mii.fType = MFT_STRING;
                    mii.dwTypeData = menuText;
                    mii.fState = MFS_ENABLED;
                    mii.hbmpItem = (legacyIcon || !m_hMenuBmp) ? HBMMENU_CALLBACK : m_hMenuBmp;
                    if (!InsertMenuItem(hMenu, indexMenu++, TRUE, &mii))
                    {
                        delete menuText;
                        return HRESULT_FROM_WIN32(GetLastError());
                    }
                    delete menuText;
                    lastItem = IDM_GETLINK;
                }
            }

        if (inLeftPane.size())
        {
            LPWSTR menuText = MegaInterface::getString(MegaInterface::STRING_REMOVE_FROM_LEFT_PANE, 0, 0);
            if (menuText)
            {
                MENUITEMINFO mii = { sizeof(mii) };
                mii.fMask = MIIM_BITMAP | MIIM_STRING | MIIM_FTYPE | MIIM_ID | MIIM_STATE;
                mii.wID = idCmdFirst + IDM_REMOVEFROMLEFTPANE;
                mii.fType = MFT_STRING;
                mii.dwTypeData = menuText;
                mii.fState = MFS_ENABLED;
                mii.hbmpItem = (legacyIcon || !m_hMenuBmp) ? HBMMENU_CALLBACK : m_hMenuBmp;
                if (!InsertMenuItem(hMenu, indexMenu++, TRUE, &mii))
                {
                    delete menuText;
                    return HRESULT_FROM_WIN32(GetLastError());
                }
                delete menuText;
                lastItem = IDM_REMOVEFROMLEFTPANE;
            }
        }

        if (!unsyncedFiles && !unsyncedFolders && !unsyncedUnknowns && !syncedUnknowns
                && selectedFiles.size() == 1
                && (syncedFiles + syncedFolders) == 1)
        {
            // One synced file or folder selected
            if (syncedFolders)
            {
                LPWSTR menuText = MegaInterface::getString(MegaInterface::STRING_VIEW_ON_MEGA, syncedFiles, syncedFolders);
                if (menuText)
                {
                    MENUITEMINFO mii = { sizeof(mii) };
                    mii.fMask = MIIM_BITMAP | MIIM_STRING | MIIM_FTYPE | MIIM_ID | MIIM_STATE;
                    mii.wID = idCmdFirst + IDM_VIEWONMEGA;
                    mii.fType = MFT_STRING;
                    mii.dwTypeData = menuText;
                    mii.fState = MFS_ENABLED;
                    mii.hbmpItem = (legacyIcon || !m_hMenuBmp) ? HBMMENU_CALLBACK : m_hMenuBmp;
                    if (!InsertMenuItem(hMenu, indexMenu++, TRUE, &mii))
                    {
                        delete menuText;
                        return HRESULT_FROM_WIN32(GetLastError());
                    }
                    delete menuText;
                    lastItem = IDM_VIEWONMEGA;
                }
            }
            else
            {
                LPWSTR menuText = MegaInterface::getString(MegaInterface::STRING_VIEW_VERSIONS, syncedFiles, syncedFolders);
                if (menuText)
                {
                    MENUITEMINFO mii = { sizeof(mii) };
                    mii.fMask = MIIM_BITMAP | MIIM_STRING | MIIM_FTYPE | MIIM_ID | MIIM_STATE;
                    mii.wID = idCmdFirst + IDM_VIEWVERSIONS;
                    mii.fType = MFT_STRING;
                    mii.dwTypeData = menuText;
                    mii.fState = MFS_ENABLED;
                    mii.hbmpItem = (legacyIcon || !m_hMenuBmp) ? HBMMENU_CALLBACK : m_hMenuBmp;
                    if (!InsertMenuItem(hMenu, indexMenu++, TRUE, &mii))
                    {
                        delete menuText;
                        return HRESULT_FROM_WIN32(GetLastError());
                    }
                    delete menuText;
                    lastItem = IDM_VIEWVERSIONS;
                }
            }
        }

        // Add a separator.
        MENUITEMINFO sep = { sizeof(sep) };
        sep.fMask = MIIM_TYPE;
        sep.fType = MFT_SEPARATOR;
        if (!InsertMenuItem(hMenu, indexMenu, TRUE, &sep))
        {
            return HRESULT_FROM_WIN32(GetLastError());
        }

        // Return an HRESULT value with the severity set to SEVERITY_SUCCESS.
        // Set the code value to the offset of the largest command identifier
        // that was assigned, plus one (1).
        return MAKE_HRESULT(SEVERITY_SUCCESS, 0, USHORT(lastItem + 1));
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {

    }
    return MAKE_HRESULT(SEVERITY_SUCCESS, 0, USHORT(0));
}


//
//   FUNCTION: FileContextMenuExt::InvokeCommand
//
//   PURPOSE: This method is called when a user clicks a menu item to tell 
//            the handler to run the associated command. The lpcmi parameter 
//            points to a structure that contains the needed information.
//
IFACEMETHODIMP ContextMenuExt::InvokeCommand(LPCMINVOKECOMMANDINFO pici)
{
    __try
    {
        BOOL fUnicode = FALSE;
        if (pici == NULL)
        {
            return E_FAIL;
        }

        // Determine which structure is being passed in, CMINVOKECOMMANDINFO or
        // CMINVOKECOMMANDINFOEX based on the cbSize member of lpcmi. Although
        // the lpcmi parameter is declared in Shlobj.h as a CMINVOKECOMMANDINFO
        // structure, in practice it often points to a CMINVOKECOMMANDINFOEX
        // structure. This struct is an extended version of CMINVOKECOMMANDINFO
        // and has additional members that allow Unicode strings to be passed.
        if (pici->cbSize == sizeof(CMINVOKECOMMANDINFOEX))
        {
            if (pici->fMask & CMIC_MASK_UNICODE)
            {
                fUnicode = TRUE;
            }
        }

        // Determines whether the command is identified by its offset or verb.
        // There are two ways to identify commands:
        //
        //   1) The command's verb string
        //   2) The command's identifier offset
        //
        // If the high-order word of lpcmi->lpVerb (for the ANSI case) or
        // lpcmi->lpVerbW (for the Unicode case) is nonzero, lpVerb or lpVerbW
        // holds a verb string. If the high-order word is zero, the command
        // offset is in the low-order word of lpcmi->lpVerb.

        // For the ANSI case, if the high-order word is not zero, the command's
        // verb string is in lpcmi->lpVerb.
        if (!fUnicode && HIWORD(pici->lpVerb))
        {
            // Is the verb supported by this context menu extension?
            if (!StrCmpIA(pici->lpVerb, m_pszUploadVerb))
            {
                requestUpload();
            }
            else if (!StrCmpIA(pici->lpVerb, m_pszGetLinkVerb))
            {
                requestGetLinks();
            }
            else if (!StrCmpIA(pici->lpVerb, m_pszRemoveFromLeftPaneVerb))
            {
                removeFromLeftPane();
            }
            else if (!StrCmpIA(pici->lpVerb, m_pszViewOnMEGAVerb))
            {
                viewOnMEGA();
            }
            else if (!StrCmpIA(pici->lpVerb, m_pszViewVersionsVerb))
            {
                viewVersions();
            }
            else
            {
                // If the verb is not recognized by the context menu handler, it
                // must return E_FAIL to allow it to be passed on to the other
                // context menu handlers that might implement that verb.
                return E_FAIL;
            }
        }

        // For the Unicode case, if the high-order word is not zero, the
        // command's verb string is in lpcmi->lpVerbW.
        else if (fUnicode && HIWORD(((CMINVOKECOMMANDINFOEX*)pici)->lpVerbW))
        {
            // Is the verb supported by this context menu extension?
            if (!StrCmpIW(((CMINVOKECOMMANDINFOEX*)pici)->lpVerbW, m_pwszUploadVerb))
            {
                requestUpload();
            }
            else if (!StrCmpIW(((CMINVOKECOMMANDINFOEX*)pici)->lpVerbW, m_pwszGetLinkVerb))
            {
                requestGetLinks();
            }
            else if (!StrCmpIW(((CMINVOKECOMMANDINFOEX*)pici)->lpVerbW, m_pwszRemoveFromLeftPaneVerb))
            {
                removeFromLeftPane();
            }
            else if (!StrCmpIW(((CMINVOKECOMMANDINFOEX*)pici)->lpVerbW, m_pwszViewOnMEGAVerb))
            {
                viewOnMEGA();
            }
            else if (!StrCmpIW(((CMINVOKECOMMANDINFOEX*)pici)->lpVerbW, m_pwszViewVersionsVerb))
            {
                viewVersions();
            }
            else
            {
                // If the verb is not recognized by the context menu handler, it
                // must return E_FAIL to allow it to be passed on to the other
                // context menu handlers that might implement that verb.
                return E_FAIL;
            }
        }

        // If the command cannot be identified through the verb string, then
        // check the identifier offset.
        else
        {
            // Is the command identifier offset supported by this context menu
            // extension?
            if (LOWORD(pici->lpVerb) == IDM_UPLOAD)
            {
                requestUpload();
            }
            else if (LOWORD(pici->lpVerb) == IDM_GETLINK)
            {
                requestGetLinks();
            }
            else if (LOWORD(pici->lpVerb) == IDM_REMOVEFROMLEFTPANE)
            {
                removeFromLeftPane();
            }
            else if (LOWORD(pici->lpVerb) == IDM_VIEWONMEGA)
            {
                viewOnMEGA();
            }
            else if (LOWORD(pici->lpVerb) == IDM_VIEWVERSIONS)
            {
                viewVersions();
            }
            else
            {
                // If the verb is not recognized by the context menu handler, it
                // must return E_FAIL to allow it to be passed on to the other
                // context menu handlers that might implement that verb.
                return E_FAIL;
            }
        }

        MegaInterface::endRequest();
        return S_OK;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {

    }

    return E_FAIL;
}


//
//   FUNCTION: CFileContextMenuExt::GetCommandString
//
//   PURPOSE: If a user highlights one of the items added by a context menu 
//            handler, the handler's IContextMenu::GetCommandString method is 
//            called to request a Help text string that will be displayed on 
//            the Windows Explorer status bar. This method can also be called 
//            to request the verb string that is assigned to a command. 
//            Either ANSI or Unicode verb strings can be requested. This 
//            example only implements support for the Unicode values of 
//            uFlags, because only those have been used in Windows Explorer 
//            since Windows 2000.
//
IFACEMETHODIMP ContextMenuExt::GetCommandString(UINT_PTR idCommand, 
    UINT uFlags, UINT *, LPSTR pszName, UINT cchMax)
{
    __try
    {
        HRESULT hr = E_INVALIDARG;

        if (idCommand == IDM_UPLOAD)
        {
            switch (uFlags)
            {
            case GCS_HELPTEXTW:
                // Only useful for pre-Vista versions of Windows that have a
                // Status bar.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszUploadVerbHelpText);
                break;

            case GCS_VERBW:
                // GCS_VERBW is an optional feature that enables a caller to
                // discover the canonical name for the verb passed in through
                // idCommand.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszUploadVerbCanonicalName);
                break;

            default:
                hr = S_OK;
            }
        }
        else if (idCommand == IDM_GETLINK)
        {
            switch (uFlags)
            {
            case GCS_HELPTEXTW:
                // Only useful for pre-Vista versions of Windows that have a
                // Status bar.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszGetLinkVerbHelpText);
                break;

            case GCS_VERBW:
                // GCS_VERBW is an optional feature that enables a caller to
                // discover the canonical name for the verb passed in through
                // idCommand.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszGetLinkVerbCanonicalName);
                break;

            default:
                hr = S_OK;
            }
        }
        else if (idCommand == IDM_REMOVEFROMLEFTPANE)
        {
            switch (uFlags)
            {
            case GCS_HELPTEXTW:
                // Only useful for pre-Vista versions of Windows that have a
                // Status bar.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszRemoveFromLeftPaneVerbHelpText);
                break;

            case GCS_VERBW:
                // GCS_VERBW is an optional feature that enables a caller to
                // discover the canonical name for the verb passed in through
                // idCommand.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszRemoveFromLeftPaneVerbCanonicalName);
                break;

            default:
                hr = S_OK;
            }
        }
        else if (idCommand == IDM_VIEWONMEGA)
        {
            switch (uFlags)
            {
            case GCS_HELPTEXTW:
                // Only useful for pre-Vista versions of Windows that have a
                // Status bar.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszViewOnMEGAVerbHelpText);
                break;

            case GCS_VERBW:
                // GCS_VERBW is an optional feature that enables a caller to
                // discover the canonical name for the verb passed in through
                // idCommand.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszViewOnMEGAVerbCanonicalName);
                break;

            default:
                hr = S_OK;
            }
        }
        else if (idCommand == IDM_VIEWVERSIONS)
        {
            switch (uFlags)
            {
            case GCS_HELPTEXTW:
                // Only useful for pre-Vista versions of Windows that have a
                // Status bar.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszViewVersionsVerbHelpText);
                break;

            case GCS_VERBW:
                // GCS_VERBW is an optional feature that enables a caller to
                // discover the canonical name for the verb passed in through
                // idCommand.
                hr = StringCchCopy(reinterpret_cast<PWSTR>(pszName), cchMax,
                    m_pwszViewVersionsVerbCanonicalName);
                break;

            default:
                hr = S_OK;
            }
        }

        // If the command (idCommand) is not supported by this context menu
        // extension handler, return E_INVALIDARG.
        return hr;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {
    }

    return E_FAIL;
}

IFACEMETHODIMP ContextMenuExt::HandleMenuMsg2(UINT uMsg, WPARAM, LPARAM lParam, LRESULT *plResult)
{
    __try
    {
        LRESULT res;
        if (plResult == NULL)
            plResult = &res;
        *plResult = FALSE;

        if (!g_hInst)
        {
            return E_FAIL;
        }

        if (!hIcon)
        {
            hIcon = (HICON)LoadImage(g_hInst, MAKEINTRESOURCE(IDI_ICON4), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);
            if (!hIcon)
            {
                return E_FAIL;
            }
        }

        switch (uMsg)
        {
            case WM_MEASUREITEM:
            {
                MEASUREITEMSTRUCT* lpmis = (MEASUREITEMSTRUCT*)lParam;
                if (lpmis==NULL)
                    break;
                lpmis->itemWidth = 16;
                lpmis->itemHeight = 16;
                *plResult = TRUE;
            }
            break;

            case WM_DRAWITEM:
            {
                DRAWITEMSTRUCT* lpdis = (DRAWITEMSTRUCT*)lParam;
                if ((lpdis==NULL)||(lpdis->CtlType != ODT_MENU))
                    return S_OK;        //not for a menu

                DrawIconEx(lpdis->hDC,
                    lpdis->rcItem.left-16,
                    lpdis->rcItem.top + (lpdis->rcItem.bottom - lpdis->rcItem.top - 16) / 2,
                    hIcon, 16, 16,
                    0, NULL, DI_NORMAL);
                *plResult = TRUE;
            }
            break;
        }
        return S_OK;
    }
    __except(EXCEPTION_EXECUTE_HANDLER)
    {

    }
    return E_FAIL;
}

IFACEMETHODIMP ContextMenuExt::HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    return HandleMenuMsg2(uMsg, wParam, lParam, NULL);
}

#pragma endregion