/*
    Copyright (C) 2018-2020 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/>.

*/


#include "content_view.h"
#include "dcpomatic_button.h"
#include "film_viewer.h"
#include "playlist_controls.h"
#include "static_text.h"
#include "wx_util.h"
#include "wx_variant.h"
#include "lib/constants.h"
#include "lib/cross.h"
#include "lib/dcp_content.h"
#include "lib/ffmpeg_content.h"
#include "lib/film.h"
#include "lib/internet.h"
#include "lib/player_video.h"
#include "lib/scoped_temporary.h"
#include <dcp/exceptions.h>
#include <dcp/warnings.h>
LIBDCP_DISABLE_WARNINGS
#include <wx/listctrl.h>
#include <wx/progdlg.h>
LIBDCP_ENABLE_WARNINGS


using std::cout;
using std::dynamic_pointer_cast;
using std::exception;
using std::shared_ptr;
using std::sort;
using std::string;
using std::vector;
using boost::optional;
using namespace dcpomatic;


PlaylistControls::PlaylistControls(wxWindow* parent, FilmViewer& viewer)
	: Controls(parent, viewer, false)
	, _play_button(new Button(this, _("Play")))
	, _pause_button(new Button(this, _("Pause")))
	, _stop_button(new Button(this, _("Stop")))
	, _next_button(new Button(this, _("Next")))
	, _previous_button(new Button(this, _("Previous")))
{
	_button_sizer->Add(_previous_button, 0, wxEXPAND);
	_button_sizer->Add(_play_button, 0, wxEXPAND);
	_button_sizer->Add(_pause_button, 0, wxEXPAND);
	_button_sizer->Add(_stop_button, 0, wxEXPAND);
	_button_sizer->Add(_next_button, 0, wxEXPAND);

	_spl_view = new wxListCtrl(this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxLC_REPORT | wxLC_NO_HEADER);
	_spl_view->AppendColumn({}, wxLIST_FORMAT_LEFT, 740);

	auto left_sizer = new wxBoxSizer(wxVERTICAL);
	auto e_sizer = new wxBoxSizer(wxHORIZONTAL);

	wxFont subheading_font(*wxNORMAL_FONT);
	subheading_font.SetWeight(wxFONTWEIGHT_BOLD);

	auto spl_header = new wxBoxSizer(wxHORIZONTAL);
	{
		auto m = new StaticText(this, _("Playlists"));
		m->SetFont(subheading_font);
		spl_header->Add(m, 1, wxALIGN_CENTER_VERTICAL);
	}
	_refresh_spl_view = new Button(this, _("Refresh"));
	spl_header->Add(_refresh_spl_view, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP / 2);

	left_sizer->Add(spl_header, 0, wxLEFT | wxRIGHT | wxEXPAND, DCPOMATIC_SIZER_GAP);
	left_sizer->Add(_spl_view, 1, wxLEFT | wxRIGHT | wxBOTTOM | wxEXPAND, DCPOMATIC_SIZER_GAP);

	_content_view = new ContentView(this);

	auto content_header = new wxBoxSizer(wxHORIZONTAL);
	{
		auto m = new StaticText(this, _("Content"));
		m->SetFont(subheading_font);
		content_header->Add(m, 1, wxALIGN_CENTER_VERTICAL);
	}
	_refresh_content_view = new Button(this, _("Refresh"));
	content_header->Add(_refresh_content_view, 0, wxBOTTOM, DCPOMATIC_SIZER_GAP / 2);

	left_sizer->Add(content_header, 0, wxTOP | wxLEFT | wxRIGHT | wxEXPAND, DCPOMATIC_SIZER_GAP);
	left_sizer->Add(_content_view, 1, wxLEFT | wxRIGHT | wxBOTTOM | wxEXPAND, DCPOMATIC_SIZER_GAP);

	_current_spl_view = new wxListCtrl(this, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxLC_REPORT | wxLC_NO_HEADER);
	_current_spl_view->AppendColumn({}, wxLIST_FORMAT_LEFT, 500);
	_current_spl_view->AppendColumn({}, wxLIST_FORMAT_LEFT, 80);
	e_sizer->Add(left_sizer, 1, wxALL | wxEXPAND, DCPOMATIC_SIZER_GAP);
	e_sizer->Add(_current_spl_view, 1, wxALL | wxEXPAND, DCPOMATIC_SIZER_GAP);

	_v_sizer->Add(e_sizer, 1, wxEXPAND);

	_play_button->Bind    (wxEVT_BUTTON, boost::bind(&PlaylistControls::play_clicked,  this));
	_pause_button->Bind   (wxEVT_BUTTON, boost::bind(&PlaylistControls::pause_clicked, this));
	_stop_button->Bind    (wxEVT_BUTTON, boost::bind(&PlaylistControls::stop_clicked,  this));
	_next_button->Bind    (wxEVT_BUTTON, boost::bind(&PlaylistControls::next_clicked,  this));
	_previous_button->Bind(wxEVT_BUTTON, boost::bind(&PlaylistControls::previous_clicked,  this));
	_spl_view->Bind       (wxEVT_LIST_ITEM_SELECTED,   boost::bind(&PlaylistControls::spl_selection_changed, this));
	_spl_view->Bind       (wxEVT_LIST_ITEM_DESELECTED, boost::bind(&PlaylistControls::spl_selection_changed, this));
	_viewer.Finished.connect(boost::bind(&PlaylistControls::viewer_finished, this));
	_refresh_spl_view->Bind(wxEVT_BUTTON, boost::bind(&PlaylistControls::update_playlist_directory, this));
	_refresh_content_view->Bind(wxEVT_BUTTON, boost::bind(&ContentView::update, _content_view));

	_content_view->update();
	update_playlist_directory();
}


void
PlaylistControls::started()
{
	Controls::started();
	_play_button->Enable(false);
	_pause_button->Enable(true);
}


/** Called when the viewer finishes a single piece of content, or it is explicitly stopped */
void
PlaylistControls::stopped()
{
	Controls::stopped();
	_play_button->Enable(true);
	_pause_button->Enable(false);
}


void
PlaylistControls::deselect_playlist()
{
	long int const selected = _spl_view->GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
	if (selected != -1) {
		_selected_playlist = boost::none;
		_spl_view->SetItemState(selected, 0, wxLIST_STATE_SELECTED);
	}
	ResetFilm(std::make_shared<Film>(optional<boost::filesystem::path>()), {});
}


void
PlaylistControls::play_clicked()
{
	_viewer.start();
}


void
PlaylistControls::setup_sensitivity()
{
	Controls::setup_sensitivity();
	bool const active_job = _active_job && *_active_job != "examine_content";
	bool const c = _film && !_film->content().empty() && !active_job;
	_play_button->Enable(c && !_viewer.playing());
	_pause_button->Enable(_viewer.playing());
	_spl_view->Enable(!_viewer.playing());
	_next_button->Enable(can_do_next());
	_previous_button->Enable(can_do_previous());
}


void
PlaylistControls::pause_clicked()
{
	_viewer.stop();
}


void
PlaylistControls::stop_clicked()
{
	_viewer.stop();
	_viewer.seek(DCPTime(), true);
	if (_selected_playlist) {
		_selected_playlist_position = 0;
		update_current_content();
	}
	deselect_playlist();
}


bool
PlaylistControls::can_do_previous()
{
	return _selected_playlist && (_selected_playlist_position - 1) >= 0;
}


void
PlaylistControls::previous_clicked()
{
	if (!can_do_previous()) {
		return;
	}

	_selected_playlist_position--;
	update_current_content();
}


bool
PlaylistControls::can_do_next()
{
	return _selected_playlist && (_selected_playlist_position + 1) < int(_playlists[*_selected_playlist].get().size());
}


void
PlaylistControls::next_clicked()
{
	if (!can_do_next()) {
		return;
	}

	_selected_playlist_position++;
	update_current_content();
}


void
PlaylistControls::add_playlist_to_list(SPL spl)
{
	int const N = _spl_view->GetItemCount();

	wxListItem it;
	it.SetId(N);
	it.SetColumn(0);
	string t = spl.name();
	if (spl.missing()) {
		t += " (content missing)";
	}
	it.SetText(std_to_wx(t));
	_spl_view->InsertItem(it);
}

struct SPLComparator
{
	bool operator() (SPL const & a, SPL const & b) {
		return a.name() < b.name();
	}
};

void
PlaylistControls::update_playlist_directory()
{
	using namespace boost::filesystem;

	_spl_view->DeleteAllItems();
	optional<path> dir = Config::instance()->player_playlist_directory();
	if (!dir) {
		return;
	}

	_playlists.clear();

	for (directory_iterator i = directory_iterator(*dir); i != directory_iterator(); ++i) {
		try {
			if (is_regular_file(i->path()) && i->path().extension() == ".xml") {
				SPL spl;
				spl.read(i->path(), _content_view);
				_playlists.push_back(spl);
			}
		} catch (exception& e) {
			/* Never mind */
		}
	}

	sort(_playlists.begin(), _playlists.end(), SPLComparator());
	for (auto i: _playlists) {
		add_playlist_to_list(i);
	}

	_selected_playlist = boost::none;
}


optional<dcp::EncryptedKDM>
PlaylistControls::get_kdm_from_directory(shared_ptr<DCPContent> dcp)
{
	using namespace boost::filesystem;
	optional<path> kdm_dir = Config::instance()->player_kdm_directory();
	if (!kdm_dir) {
		return optional<dcp::EncryptedKDM>();
	}
	for (directory_iterator i = directory_iterator(*kdm_dir); i != directory_iterator(); ++i) {
		try {
			if (file_size(i->path()) < MAX_KDM_SIZE) {
				dcp::EncryptedKDM kdm(dcp::file_to_string(i->path()));
				if (kdm.cpl_id() == dcp->cpl()) {
					return kdm;
				}
			}
		} catch (std::exception& e) {
			/* Hey well */
		}
	}
	return optional<dcp::EncryptedKDM>();
}


void
PlaylistControls::spl_selection_changed()
{
	long int selected = _spl_view->GetNextItem(-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
	if (selected == -1) {
		_current_spl_view->DeleteAllItems();
		_selected_playlist = boost::none;
		return;
	}

	if (_playlists[selected].missing()) {
		error_dialog(this, _("This playlist cannot be loaded as some content is missing."));
		deselect_playlist();
		return;
	}

	if (_playlists[selected].get().empty()) {
		error_dialog(this, _("This playlist is empty."));
		return;
	}

	select_playlist(selected, 0);
}


void
PlaylistControls::select_playlist(int selected, int position)
{
	wxProgressDialog dialog(variant::wx::dcpomatic(), _("Loading playlist and KDMs"));

	for (auto const& i: _playlists[selected].get()) {
		dialog.Pulse();
		shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent>(i.content);
		if (dcp && dcp->needs_kdm()) {
			optional<dcp::EncryptedKDM> kdm;
			kdm = get_kdm_from_directory(dcp);
			if (kdm) {
				try {
					dcp->add_kdm(*kdm);
					dcp->examine(_film, shared_ptr<Job>(), true);
				} catch (KDMError& e) {
					error_dialog(this, _("Could not load KDM."));
				}
			}
			if (dcp->needs_kdm()) {
				/* We didn't get a KDM for this */
				error_dialog(this, _("This playlist cannot be loaded as a KDM is missing or incorrect."));
				deselect_playlist();
				return;
			}
		}
	}

	_current_spl_view->DeleteAllItems();

	int N = 0;
	for (auto i: _playlists[selected].get()) {
		wxListItem it;
		it.SetId(N);
		it.SetColumn(0);
		it.SetText(std_to_wx(i.name));
		_current_spl_view->InsertItem(it);
		++N;
	}

	_selected_playlist = selected;
	_selected_playlist_position = position;
 	dialog.Pulse();
	reset_film();
 	dialog.Pulse();
	update_current_content();
}


void
PlaylistControls::reset_film()
{
	DCPOMATIC_ASSERT(_selected_playlist);
	auto film = std::make_shared<Film>(optional<boost::filesystem::path>());
	auto entry = _playlists[*_selected_playlist].get(_selected_playlist_position);
	film->add_content(vector<shared_ptr<Content>>{entry.content});
	ResetFilm(film, entry.crop_to_ratio);
}


void
PlaylistControls::config_changed(int property)
{
	Controls::config_changed(property);

	if (property == Config::PLAYER_CONTENT_DIRECTORY) {
		_content_view->update();
	} else if (property == Config::PLAYER_PLAYLIST_DIRECTORY) {
		update_playlist_directory();
	}
}


void
PlaylistControls::update_current_content()
{
	DCPOMATIC_ASSERT(_selected_playlist);

	wxProgressDialog dialog(variant::wx::dcpomatic(), _("Loading content"));

	setup_sensitivity();
	dialog.Pulse();
	reset_film();
}


/** One piece of content in our SPL has finished playing */
void
PlaylistControls::viewer_finished()
{
	if (!_selected_playlist) {
		return;
	}

	_selected_playlist_position++;
	if (_selected_playlist_position < int(_playlists[*_selected_playlist].get().size())) {
		/* Next piece of content on the SPL */
		update_current_content();
		_viewer.start();
	} else {
		/* Finished the whole SPL */
		_selected_playlist_position = 0;
		ResetFilm(std::make_shared<Film>(optional<boost::filesystem::path>()), {});
		_play_button->Enable(true);
		_pause_button->Enable(false);
	}
}


void
PlaylistControls::play()
{
	play_clicked();
}


void
PlaylistControls::stop()
{
	stop_clicked();
}