// Copyright (C) 2009,2010,2011,2012 GlavSoft LLC.
// All rights reserved.
//
//-------------------------------------------------------------------------
// This file is part of the TightVNC software.  Please visit our Web site:
//
//                       http://www.tightvnc.com/
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//-------------------------------------------------------------------------
//

#include "Menu.h"

Menu::Menu()
: m_menu(0),
  m_bCreated(false)
{
}

HMENU Menu::getMenu()
{
  return m_menu;
}

void Menu::setMenu(HMENU hmenu)
{
  _ASSERT(m_menu == 0);

  m_menu = hmenu;
}

bool Menu::getWindowMenu(HWND hwnd)
{
  _ASSERT(m_menu == 0);

  m_menu = GetMenu(hwnd);
  return (m_menu == 0 ? false : true);
}

void Menu::getSystemMenu(HWND hwnd)
{
  _ASSERT(m_menu == 0);

  m_menu = GetSystemMenu(hwnd, FALSE);
}

void Menu::create()
{
  _ASSERT(m_menu == 0);

  m_menu = CreateMenu();
  m_bCreated = true;
}

bool Menu::loadMenu(LPCTSTR lpMenuName)
{
  _ASSERT(m_menu == 0);

  m_menu = LoadMenu(GetModuleHandle(0), lpMenuName);
  m_bCreated = true;
  return (!m_menu?false:true);
}

void Menu::createPopupMenu()
{
  _ASSERT(m_menu == 0);

  m_menu = CreatePopupMenu();
  m_bCreated = true;
}

int Menu::getMenuItemCount()
{
  _ASSERT(m_menu != 0);

  return GetMenuItemCount(m_menu);
}

bool Menu::appendMenu(UINT uFlags, UINT_PTR uIDNewItem, LPCTSTR lpNewItem)
{
  _ASSERT(m_menu != 0);

  return !!AppendMenu(m_menu, uFlags, uIDNewItem, lpNewItem);
}

bool Menu::insertMenuItem(UINT uItem, BOOL fByPosition, LPCMENUITEMINFO lpmii)
{
  _ASSERT(m_menu != 0);

  return !!InsertMenuItem(m_menu, uItem, fByPosition, lpmii);
}

bool Menu::modifyMenu(UINT uPosition, UINT uFlags, LONG_PTR uIDNewItem, LPCTSTR lpNewItem)
{
  _ASSERT(m_menu != 0);

  return !!ModifyMenu(m_menu, uPosition, uFlags, uIDNewItem, lpNewItem);
}

bool Menu::deleteMenu(UINT uPosition)
{
  _ASSERT(m_menu != 0);

  return !!DeleteMenu(m_menu, uPosition, MF_BYPOSITION);
}

bool Menu::getSubMenu(int nPos, Menu *menu)
{
  HMENU tmenu;

  _ASSERT(m_menu != 0 && menu);
  tmenu = GetSubMenu(m_menu, nPos);
  if (tmenu) {
    menu->setMenu(tmenu);
  }
  return (!tmenu?false:true);
}

bool Menu::enableMenuItem(UINT uID, UINT uEnable)
{
  _ASSERT(m_menu != 0);

  return !!EnableMenuItem(m_menu, uID, uEnable);
}

bool Menu::setMenuItem(UINT uItem, BOOL fByPosition, LPMENUITEMINFO lpmii)
{
  _ASSERT(m_menu != 0);

  return !!SetMenuItemInfo(m_menu, uItem, fByPosition, lpmii);
}

void Menu::termMenu()
{
  if (m_menu && m_bCreated) {
    DestroyMenu(m_menu);
  }
  m_menu = 0;
  m_bCreated = false;
}

Menu::~Menu()
{
  if (m_menu && m_bCreated) {
    DestroyMenu(m_menu);
  }
}

int Menu::findMenuItem(UINT uID)
{
  _ASSERT(m_menu != 0);

  for (int i = 0; i < getMenuItemCount(); i++) {
    if (GetMenuItemID(m_menu, i) == uID) {
      return i;
    }
  }
  return -1;
}

bool Menu::appendMenu(StringStorage strVal, UINT uID)
{
  return appendMenu(MF_STRING, uID, strVal.getString());
}

bool Menu::appendSeparator()
{
  return appendMenu(MF_MENUBREAK, 0, 0);
}

bool Menu::appendSubMenu(StringStorage strVal, Menu *pMenu)
{
  return appendMenu(MF_POPUP, (UINT_PTR)pMenu->getMenu(), strVal.getString());
}

bool Menu::insertMenuItem(UINT uItem, StringStorage strVal, UINT uID)
{
  MENUITEMINFO mii;

  ZeroMemory(&mii, sizeof(MENUITEMINFO));
  mii.cbSize = sizeof(MENUITEMINFO);
  mii.fMask = MIIM_STRING | MIIM_ID;
  mii.dwTypeData = (LPTSTR)strVal.getString();
  mii.fState = MFS_DEFAULT; 
  mii.wID = uID;
  return insertMenuItem(uItem, TRUE, (LPMENUITEMINFO)&mii);
}

bool Menu::insertCheckMenuItem(UINT uItem, StringStorage strVal, UINT uID)
{
  MENUITEMINFO mii;

  ZeroMemory(&mii, sizeof(MENUITEMINFO));
  mii.cbSize = sizeof(MENUITEMINFO);
  mii.fMask = MIIM_STRING | MIIM_ID | MIIM_STATE;
  mii.dwTypeData = (LPTSTR)strVal.getString();
  mii.fState = MFS_UNCHECKED; 
  mii.wID = uID;
  return insertMenuItem(uItem, TRUE, (LPMENUITEMINFO)&mii);
}

bool Menu::checkedMenuItem(UINT uID, bool bEnable)
{
  _ASSERT(m_menu != 0);

  int pos = findMenuItem(uID);
  if (pos == -1) {
    return false;
  }
  UINT command = bEnable ? MF_CHECKED : MF_UNCHECKED;
  DWORD res = CheckMenuItem(m_menu, pos, MF_BYPOSITION | command);
  return (res == -1) ? false : true;
}

bool Menu::insertSeparator(UINT uItem)
{
  MENUITEMINFO mii;

  ZeroMemory(&mii, sizeof(MENUITEMINFO));
  mii.cbSize = sizeof(MENUITEMINFO);
  mii.fType = MFT_SEPARATOR;
  return insertMenuItem(uItem, TRUE, (LPMENUITEMINFO)&mii);
}

bool Menu::insertSubMenu(UINT uItem, StringStorage strVal, Menu *pMenu)
{
  MENUITEMINFO mii;

  ZeroMemory(&mii, sizeof(MENUITEMINFO));
  mii.cbSize = sizeof(MENUITEMINFO);
  mii.fMask = MIIM_SUBMENU;
  mii.hSubMenu = pMenu->getMenu();
  return insertMenuItem(uItem, TRUE, (LPMENUITEMINFO)&mii);
}

bool Menu::setDefaultItem(UINT uID)
{
  _ASSERT(m_menu != 0);

  return !!SetMenuDefaultItem(m_menu, uID, FALSE);
}