/*
    Copyright (C) 2012-2016 Carl Hetherington <cth@carlh.net>

    This file is part of DCP-o-matic.

    DCP-o-matic 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.

    DCP-o-matic 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 DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.

*/


#ifndef DCPOMATIC_EDITABLE_LIST_H
#define DCPOMATIC_EDITABLE_LIST_H


#include "dcpomatic_button.h"
#include "wx_util.h"
#include <dcp/warnings.h>
LIBDCP_DISABLE_WARNINGS
#include <wx/listctrl.h>
#include <wx/wx.h>
LIBDCP_ENABLE_WARNINGS
#include <vector>


class EditableListColumn
{
public:
	EditableListColumn(wxString name_)
		: name(name_)
		, growable(false)
	{}

	EditableListColumn(wxString name_, boost::optional<int> width_, bool growable_)
		: name(name_)
		, width(width_)
		, growable(growable_)
	{}

	wxString name;
	boost::optional<int> width;
	bool growable;
};


namespace EditableListButton
{
	static int constexpr NEW = 0x1;
	static int constexpr EDIT = 0x2;
	static int constexpr REMOVE = 0x4;
};


enum class EditableListTitle
{
	VISIBLE,
	INVISIBLE
};


/** @param T type of things being edited.
 *  @param get Function to get a std::vector of the things being edited.
 *  @param set Function set the things from a a std::vector.
 *  @param column Function to get the display string for a given column in a given item.
 */
template<class T>
class EditableList : public wxPanel
{
public:
	EditableList(
		wxWindow* parent,
		std::vector<EditableListColumn> columns,
		std::function<std::vector<T> ()> get,
		std::function<void (std::vector<T>)> set,
		std::function<std::vector<T> (wxWindow*)> add,
		std::function<void (wxWindow*, T&)> edit,
		std::function<std::string (T, int)> column,
		EditableListTitle title,
		int buttons,
		boost::optional<wxString> custom_button = {}
		)
		: wxPanel(parent)
		, _get(get)
		, _set(set)
		, _add(add)
		, _edit(edit)
		, _columns(columns)
		, _column(column)
		, _default_width(200)
	{
		_sizer = new wxBoxSizer(wxHORIZONTAL);
		SetSizer(_sizer);

		long style = wxLC_REPORT | wxLC_SINGLE_SEL;
		if (title == EditableListTitle::INVISIBLE) {
			style |= wxLC_NO_HEADER;
		}

		int total_width = 0;
		for (auto i: _columns) {
			total_width += i.width.get_value_or(_default_width);
		}

#ifdef __WXGTK3__
		/* With the GTK3 backend wxListCtrls are hard to pick out from the background of the
		 * window, so put a border in to help.
		 */
		auto border = new wxPanel(this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL | wxBORDER_THEME);
		_list = new wxListCtrl(border, wxID_ANY, wxDefaultPosition, wxSize(total_width, -1), style);
		auto border_sizer = new wxBoxSizer(wxHORIZONTAL);
		border_sizer->Add(_list, 1, wxALL | wxEXPAND, 2);
		border->SetSizer(border_sizer);
#else
		_list = new wxListCtrl(this, wxID_ANY, wxDefaultPosition, wxSize(total_width, -1), style);
#endif

		int j = 0;
		for (auto i: _columns) {
			wxListItem ip;
			ip.SetId(j);
			ip.SetText(i.name);
			_list->InsertColumn(j, ip);
			++j;
		}

#ifdef __WXGTK3__
		_sizer->Add(border, 1, wxEXPAND);
#else
		_sizer->Add(_list, 1, wxEXPAND);
#endif

		{
			auto s = new wxBoxSizer(wxVERTICAL);
			if (buttons & EditableListButton::NEW) {
				_add_button = new Button(this, _("Add..."));
				s->Add(_add_button, 1, wxEXPAND | wxTOP | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
			}
			if (buttons & EditableListButton::EDIT) {
				_edit_button = new Button(this, _("Edit..."));
				s->Add(_edit_button, 1, wxEXPAND | wxTOP | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
			}
			if (custom_button) {
				_custom_button = new Button(this, *custom_button);
				s->Add(_custom_button, 1, wxEXPAND | wxTOP | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
			}
			if (buttons & EditableListButton::REMOVE) {
				_remove_button = new Button(this, _("Remove"));
				s->Add(_remove_button, 1, wxEXPAND | wxTOP | wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
			}
			_sizer->Add(s, 0, wxLEFT, DCPOMATIC_SIZER_X_GAP);
		}

		if (_add_button) {
			_add_button->Bind(wxEVT_COMMAND_BUTTON_CLICKED, boost::bind(&EditableList::add_clicked, this));
		}
		if (_edit_button) {
			_edit_button->Bind(wxEVT_COMMAND_BUTTON_CLICKED, boost::bind(&EditableList::edit_clicked, this));
		}
		if (_remove_button) {
			_remove_button->Bind(wxEVT_COMMAND_BUTTON_CLICKED, boost::bind(&EditableList::remove_clicked, this));
		}

		_list->Bind(wxEVT_COMMAND_LIST_ITEM_SELECTED, boost::bind(&EditableList::selection_changed, this));
		_list->Bind(wxEVT_COMMAND_LIST_ITEM_DESELECTED, boost::bind(&EditableList::selection_changed, this));
#if BOOST_VERSION >= 106100
		_list->Bind(wxEVT_SIZE, boost::bind(&EditableList::resized, this, boost::placeholders::_1));
#else
		_list->Bind(wxEVT_SIZE, boost::bind(&EditableList::resized, this, _1));
#endif

		refresh();
		selection_changed();
	}

	void refresh()
	{
		_list->DeleteAllItems();

		auto current = _get();
		for (auto const& i: current) {
			add_to_control(i);
		}
	}

	boost::optional<T> selection() const
	{
		int item = _list->GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
		if (item == -1) {
			return {};
		}

		auto all = _get();
		DCPOMATIC_ASSERT(item >= 0 && item < int(all.size()));
		return all[item];
	}

	void layout()
	{
		_sizer->Layout();
	}

	Button* custom_button()
	{
		return _custom_button;
	}

	boost::signals2::signal<void ()> SelectionChanged;

	template <class S>
	static std::vector<T> add_with_dialog(wxWindow* parent)
	{
		S dialog(parent);

		if (dialog.ShowModal() == wxID_OK) {
			return dialog.get();
		} else {
			return {};
		}
	}

	template <class S>
	static void edit_with_dialog(wxWindow* parent, T& item)
	{
		S dialog(parent);
		dialog.set(item);
		if (dialog.ShowModal() == wxID_OK) {
			auto const value = dialog.get();
			if (!value.empty()) {
				DCPOMATIC_ASSERT(value.size() == 1);
				item = value[0];
			}
		}
	}

private:

	void add_to_control(T item)
	{
		wxListItem list_item;
		int const n = _list->GetItemCount();
		list_item.SetId(n);
		_list->InsertItem(list_item);

		for (size_t i = 0; i < _columns.size(); ++i) {
			_list->SetItem(n, i, std_to_wx(_column(item, i)));
		}
	}

	void selection_changed()
	{
		int const i = _list->GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
		if (_edit_button) {
			_edit_button->Enable(i >= 0);
		}
		if (_remove_button) {
			_remove_button->Enable(i >= 0);
		}

		SelectionChanged();
	}

	void add_clicked()
	{
		auto all = _get();
		for (auto item: _add(this)) {
			add_to_control(item);
			all.push_back(item);
		}
		_set(all);
	}

	void edit_clicked()
	{
		int item = _list->GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
		if (item == -1) {
			return;
		}

		std::vector<T> all = _get();
		DCPOMATIC_ASSERT(item >= 0 && item < int(all.size()));

		_edit(this, all[item]);

		for (size_t i = 0; i < _columns.size(); ++i) {
			_list->SetItem(item, i, std_to_wx(_column(all[item], i)));
		}

		_set(all);
	}

	void remove_clicked()
	{
		int i = _list->GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
		if (i == -1) {
			return;
		}

		_list->DeleteItem(i);
		auto all = _get();
		all.erase(all.begin() + i);
		_set(all);

		selection_changed();
	}

	void resized(wxSizeEvent& ev)
	{
		int const w = _list->GetSize().GetWidth() - 2;

		int fixed_width = 0;
		int growable = 0;
		int j = 0;
		for (auto i: _columns) {
			fixed_width += i.width.get_value_or(_default_width);
			if (!i.growable) {
				_list->SetColumnWidth(j, i.width.get_value_or(_default_width));
			} else {
				++growable;
			}
			++j;
		}

		j = 0;
		for (auto i: _columns) {
			if (i.growable) {
				_list->SetColumnWidth(j, i.width.get_value_or(_default_width) + (w - fixed_width) / growable);
			}
			++j;
		}

		ev.Skip();
	}

	std::function <std::vector<T> ()> _get;
	std::function <void (std::vector<T>)> _set;
	std::function<std::vector<T> (wxWindow*)> _add;
	std::function<void (wxWindow*, T&)> _edit;
	std::vector<EditableListColumn> _columns;
	std::function<std::string (T, int)> _column;

	wxButton* _add_button = nullptr;
	wxButton* _edit_button = nullptr;
	Button* _custom_button = nullptr;
	wxButton* _remove_button = nullptr;
	wxListCtrl* _list;
	wxBoxSizer* _sizer;
	int _default_width;
};

#endif