/*
 * FolderItemVerb(s) implementation
 *
 * Copyright 2015 Mark Jansen
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

#include "precomp.h"

WINE_DEFAULT_DEBUG_CHANNEL(shell);


CFolderItemVerb::CFolderItemVerb()
{
}

CFolderItemVerb::~CFolderItemVerb()
{
}

void CFolderItemVerb::Init(IContextMenu* menu, BSTR name)
{
    m_contextmenu = menu;
    m_name.m_str = name;
}

// *** FolderItemVerb methods ***

HRESULT STDMETHODCALLTYPE CFolderItemVerb::get_Application(IDispatch **ppid)
{
    TRACE("(%p, %p)\n", this, ppid);

    if (ppid)
        *ppid = NULL;

    return E_NOTIMPL;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerb::get_Parent(IDispatch **ppid)
{
    TRACE("(%p, %p)\n", this, ppid);

    if (ppid)
        *ppid = NULL;

    return E_NOTIMPL;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerb::get_Name(BSTR *pbs)
{
    if (!pbs)
        return E_POINTER;
    *pbs = SysAllocString(m_name);
    return S_OK;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerb::DoIt()
{
    TRACE("(%p, %p)\n", this);
    return E_NOTIMPL;
}






CFolderItemVerbs::CFolderItemVerbs()
    :m_menu(NULL)
    ,m_count(0)
{
}

CFolderItemVerbs::~CFolderItemVerbs()
{
    DestroyMenu(m_menu);
}

HRESULT CFolderItemVerbs::Init(LPITEMIDLIST idlist)
{
    CComPtr<IShellFolder> folder;
    LPCITEMIDLIST child;
    HRESULT hr = SHBindToParent(idlist, IID_PPV_ARG(IShellFolder, &folder), &child);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;

    hr = folder->GetUIObjectOf(NULL, 1, &child, IID_IContextMenu, NULL, (PVOID*)&m_contextmenu);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;

    m_menu = CreatePopupMenu();
    hr = m_contextmenu->QueryContextMenu(m_menu, 0, FCIDM_SHVIEWFIRST, FCIDM_SHVIEWLAST, CMF_NORMAL);
    if (!SUCCEEDED(hr))
        return hr;

    m_count = GetMenuItemCount(m_menu);
    return hr;
}


// *** FolderItemVerbs methods ***
HRESULT STDMETHODCALLTYPE CFolderItemVerbs::get_Count(LONG *plCount)
{
    if (!plCount)
        return E_INVALIDARG;
    *plCount = m_count;
    return S_OK;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerbs::get_Application(IDispatch **ppid)
{
    TRACE("(%p, %p)\n", this, ppid);

    if (ppid)
        *ppid = NULL;

    return E_NOTIMPL;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerbs::get_Parent(IDispatch **ppid)
{
    TRACE("(%p, %p)\n", this, ppid);

    if (ppid)
        *ppid = NULL;

    return E_NOTIMPL;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerbs::Item(VARIANT indexVar, FolderItemVerb **ppid)
{
    if (!ppid)
        return E_POINTER;

    CComVariant var;

    HRESULT hr = VariantChangeType(&var, &indexVar, 0, VT_I4);
    if (FAILED_UNEXPECTEDLY(hr))
        return E_INVALIDARG;

    int index = V_I4(&var);

    if (index > m_count)
        return S_OK;

    BSTR name = NULL;

    if(index == m_count)
    {
        name = SysAllocStringLen(NULL, 0);
    }
    else
    {
        MENUITEMINFOW info = { sizeof(info), 0 };
        info.fMask = MIIM_STRING;
        if (!GetMenuItemInfoW(m_menu, index, TRUE, &info))
            return E_FAIL;
        name = SysAllocStringLen(NULL, info.cch);
        if (name)
        {
            info.dwTypeData = name;
            info.cch++;
            GetMenuItemInfoW(m_menu, index, TRUE, &info);
        }
    }

    if (!name)
        return E_OUTOFMEMORY;

    CFolderItemVerb* verb = new CComObject<CFolderItemVerb>();
    verb->Init(m_contextmenu, name);
    verb->AddRef();
    *ppid = verb;

    return S_OK;
}

HRESULT STDMETHODCALLTYPE CFolderItemVerbs::_NewEnum(IUnknown **ppunk)
{
    TRACE("(%p, %p)\n", this, ppunk);
    return E_NOTIMPL;
}