#pragma once

#ifndef TNZ_OPTIONMENU_INCLUDED
#define TNZ_OPTIONMENU_INCLUDED

#include "tw/tw.h"

#undef DVAPI
#undef DVVAR
#ifdef TWIN_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

// forward declaration

class TPopupMenu;

#ifdef WIN32
#pragma warning(disable : 4251)
#endif

//-------------------------------------------------------------------

class DVAPI TGenericOptionMenuAction {
public:
  virtual ~TGenericOptionMenuAction() {}
  virtual void sendCommand(string item) = 0;
};

class DVAPI TGenericOptionMenuWAction {
public:
  virtual ~TGenericOptionMenuWAction() {}
  virtual void sendCommand(wstring item) = 0;
};

//-------------------------------------------------------------------

template <class T>
class TOptionMenuAction : public TGenericOptionMenuAction {
public:
  typedef void (T::*Method)(string item);
  TOptionMenuAction(T *target, Method method)
      : m_target(target), m_method(method) {}
  void sendCommand(string item) { (m_target->*m_method)(item); }

private:
  T *m_target;
  Method m_method;
};

//-------------------------------------------------------------------
template <class T>
class TOptionMenuActionW : public TGenericOptionMenuWAction {
public:
  typedef void (T::*Method)(wstring item);
  TOptionMenuActionW(T *target, Method method)
      : m_target(target), m_method(method) {}
  void sendCommand(wstring item) { (m_target->*m_method)(item); }

private:
  T *m_target;
  Method m_method;
};

//-------------------------------------------------------------------

class DVAPI TOptionMenu : public TWidget {
public:
  TOptionMenu(TWidget *parent, string name = "optionMenu");
  ~TOptionMenu();

  void setAction(TGenericOptionMenuAction *action) { m_action = action; }

  void repaint();

  void configureNotify(const TDimension &size);

  void leftButtonDown(const TMouseEvent &);

  string getText() const;
  void setText(string s);

  void setLabel(wstring title);
  void setLabelWidth(int lx);
  wstring getLabel() const { return m_label; }
  int getLabelWidth() const { return m_labelWidth; }

  bool isOption(string s) const;
  void addOption(string cmdName);
  void addOption(string cmdName, wstring title);
  void deleteOption(string s);
  void deleteAllOptions();

private:
  TPopupMenu *m_menu;
  string m_currentOption;
  wstring m_currentOptionTitle;
  wstring m_label;
  int m_labelWidth;

  typedef std::vector<std::pair<string, wstring>> OptionList;
  OptionList m_options;

  TGenericOptionMenuAction *m_action;
};

//-------------------------------------------------------------------

template <class T>
class TOptionMenuWAction : public TGenericOptionMenuWAction {
public:
  typedef void (T::*Method)(wstring item);
  TOptionMenuWAction(T *target, Method method)
      : m_target(target), m_method(method) {}
  void sendCommand(wstring item) { (m_target->*m_method)(item); }

private:
  T *m_target;
  Method m_method;
};

//-------------------------------------------------------------------

class DVAPI TOptionMenuW : public TWidget {
public:
  TOptionMenuW(TWidget *parent, string name = "optionMenu");
  ~TOptionMenuW();

  void setAction(TGenericOptionMenuWAction *action) { m_action = action; }

  void repaint();

  void configureNotify(const TDimension &size);

  void leftButtonDown(const TMouseEvent &);

  wstring getText() const;
  void setText(wstring s);

  void setLabel(wstring label);
  void setLabelWidth(int width);

  bool isOption(wstring s) const;
  void addOption(wstring item);
  void deleteOption(wstring s);
  void deleteAllOptions();

  typedef std::vector<wstring> OptionList;
  const OptionList getOptions() const { return m_options; };

private:
  TPopupMenu *m_menu;
  wstring m_currentOption;
  wstring m_label;
  int m_labelWidth;

  OptionList m_options;

  TGenericOptionMenuWAction *m_action;
};

//-------------------------------------------------------------------

/*
class DVAPI TComboBoxActionInterface {
public:
  TComboBoxActionInterface() {}
  virtual ~TComboBoxActionInterface() {}
  virtual void triggerAction(TComboBox*cb, string text) = 0;
};

//-------------------------------------------------------------------

template <class T>
class TComboBoxAction : public TComboBoxActionInterface {
  typedef void (T::*Method)(TComboBox *vf, string text);
  T *m_target;
  Method m_method;
public:
  TComboBoxAction(T*target, Method method) : m_target(target), m_method(method)
{}
  void triggerAction(TComboBox*vf, string text)
    {(m_target->*m_method)(vf, text); }
};

//-------------------------------------------------------------------

template <class T>
inline void tconnect(TComboBox&src, T *target, void (T::*method)(TComboBox *vf,
string text))
{
  src.addAction(new TComboBoxAction<T>(target, method));
}

*/

//-------------------------------------------------------------------

template <class T>
void tconnect(TOptionMenu &menu, T *target, void (T::*method)(string)) {
  menu.setAction(new TOptionMenuAction<T>(target, method));
}

//-------------------------------------------------------------------

template <class T>
void tconnect(TOptionMenuW &menu, T *target, void (T::*method)(wstring)) {
  menu.setAction(new TOptionMenuActionW<T>(target, method));
}

#endif