/*

Copyright (C) 2011-2014 Michael Goffioul

This file is part of Octave.

Octave 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 3 of the License, or (at your
option) any later version.

Octave 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 Octave; see the file COPYING.  If not, see
<http://www.gnu.org/licenses/>.

*/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <QAction>
#include <QMainWindow>
#include <QMenu>
#include <QMenuBar>

#include "Figure.h"
#include "Menu.h"
#include "QtHandlesUtils.h"

namespace QtHandles
{

static QKeySequence accelSequence (const uimenu::properties& up)
{
  std::string s (up.get_accelerator ());

  if (! s.empty ())
    {
      char c = s[0];
      int keyMod = Qt::CTRL;

      if (c >= 'A' && c <= 'Z')
        keyMod |= Qt::SHIFT;
      if (c >= 'a' && c <= 'z')
        c -= ('a' - 'A');
      if (c >= 'A' && c <= 'Z')
        return QKeySequence (keyMod | static_cast<int> (c));
    }

  return QKeySequence ();
}

Menu* Menu::create (const graphics_object& go)
{
  Object* parent = Object::parentObject (go);

  if (parent)
    {
      QObject* qObj = parent->qObject ();

      if (qObj)
        return new Menu (go, new QAction (qObj), parent);
    }

  return 0;
}

Menu::Menu (const graphics_object& go, QAction* action, Object* xparent)
    : Object (go, action), m_parent (0), m_separator (0)
{
  uimenu::properties& up = properties<uimenu> ();

  action->setText (Utils::fromStdString (up.get_label ()));
  if (up.is_checked ())
    {
      action->setCheckable (true);
      action->setChecked (up.is_checked ());
    }
  action->setEnabled (up.is_enable ());
  action->setShortcut (accelSequence (up));
  action->setVisible (up.is_visible ());
  if (up.is_separator ())
    {
      m_separator = new QAction (action);
      m_separator->setSeparator (true);
      m_separator->setVisible (up.is_visible ());
    }

  MenuContainer* menuContainer = dynamic_cast<MenuContainer*> (xparent);

  if (menuContainer)
    m_parent = menuContainer->menu ();

  if (m_parent)
    {
      int pos = static_cast<int> (up.get_position ());

      if (pos <= 0)
        {
          if (m_separator)
            m_parent->insertAction (0, m_separator);
          m_parent->insertAction (0, action);

          int count = 0;

          foreach (QAction* a, m_parent->actions ())
            if (! a->isSeparator () && a->objectName () != "builtinMenu")
              count++;
          up.get_property ("position").set
            (octave_value (static_cast<double> (count)), true, false);
        }
      else
        {

          int count = 0;
          QAction* before = 0;

          foreach (QAction* a, m_parent->actions ())
            if (! a->isSeparator () && a->objectName () != "builtinMenu")
              {
                count++;
                if (pos <= count)
                  {
                    before = a;
                    break;
                  }
              }

          if (m_separator)
            m_parent->insertAction (before, m_separator);
          m_parent->insertAction (before, action);

          if (before)
            updateSiblingPositions ();
          else
            up.get_property ("position").set
              (octave_value (static_cast<double> (count+1)), true, false);
        }
    }

  connect (action, SIGNAL (triggered (bool)), SLOT (actionTriggered (void)));
}

Menu::~Menu (void)
{
}

void Menu::update (int pId)
{
  uimenu::properties& up = properties<uimenu> ();
  QAction* action = qWidget<QAction> ();

  switch (pId)
    {
    case uimenu::properties::ID_LABEL:
      action->setText (Utils::fromStdString (up.get_label ()));
      break;
    case uimenu::properties::ID_CHECKED:
      if (up.is_checked ())
        {
          action->setCheckable (true);
          action->setChecked (up.is_checked ());
        }
      else
        {
          action->setChecked (false);
          action->setCheckable (false);
        }
      break;
    case uimenu::properties::ID_ENABLE:
      action->setEnabled (up.is_enable ());
      break;
    case uimenu::properties::ID_ACCELERATOR:
      if (! action->menu ())
        action->setShortcut (accelSequence (up));
      break;
    case uimenu::properties::ID_SEPARATOR:
      if (up.is_separator ())
        {
          if (! m_separator)
            {
              m_separator = new QAction (action);
              m_separator->setSeparator (true);
              m_separator->setVisible (up.is_visible ());
              if (m_parent)
                m_parent->insertAction (action, m_separator);
            }
        }
      else
        {
          if (m_separator)
            delete m_separator;
          m_separator = 0;
        }
      break;
    case uimenu::properties::ID_VISIBLE:
      action->setVisible (up.is_visible ());
      if (m_separator)
        m_separator->setVisible (up.is_visible ());
      break;
    case uimenu::properties::ID_POSITION:
      if (m_separator)
        m_parent->removeAction (m_separator);
      m_parent->removeAction (action);
        {
          int pos = static_cast<int> (up.get_position ());
          QAction* before = 0;

          if (pos > 0)
            {
              int count = 0;

              foreach (QAction* a, m_parent->actions ())
                if (! a->isSeparator () && a->objectName () != "builtinMenu")
                  {
                    count++;
                    if (pos <= count)
                      {
                        before = a;
                        break;
                      }
                  }
            }

          if (m_separator)
            m_parent->insertAction (before, m_separator);
          m_parent->insertAction (before, action);
          updateSiblingPositions ();
        }
      break;
    default:
      Object::update (pId);
      break;
    }
}

QWidget* Menu::menu (void)
{
  QAction* action = qWidget<QAction> ();
  QMenu* _menu = action->menu ();

  if (! _menu)
    {
      _menu = new QMenu (action->parentWidget ());
      action->setMenu (_menu);
      action->setShortcut (QKeySequence ());
      connect (_menu, SIGNAL (aboutToShow (void)),
               this, SLOT (actionHovered (void)));
    }

  return _menu;
}

void Menu::actionTriggered (void)
{
  QAction* action = qWidget<QAction> ();

  if (action->isCheckable ())
    action->setChecked (! action->isChecked ());
  gh_manager::post_callback (m_handle, "callback");
}

void Menu::actionHovered (void)
{
  gh_manager::post_callback (m_handle, "callback");
}

void Menu::updateSiblingPositions (void)
{
  if (m_parent)
    {
      double count = 1.0;

      foreach (QAction* a, m_parent->actions ())
        {
          if (! a->isSeparator () && a->objectName () != "builtinMenu")
            {
              Object* aObj = Object::fromQObject (a);

              if (aObj)
                {
                  graphics_object go = aObj->object ();

                  // Probably overkill as a uimenu child can only be another
                  // uimenu object.
                  if (go.isa ("uimenu"))
                    {
                      uimenu::properties& up = Utils::properties<uimenu> (go);

                      up.get_property ("position").set
                        (octave_value (count), true, false);
                    }
                }

              count++;
            }
        }
    }
}

}; // namespace QtHandles