/*
    Copyright (C) 2012-2019 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 "config_dialog.h"
#include "static_text.h"
#include "check_box.h"
#include "nag_dialog.h"
#include "dcpomatic_button.h"
#include "audio_mapping_view.h"
#include <dcp/raw_convert.h>
#include <iostream>

using std::string;
using std::vector;
using std::pair;
using std::make_pair;
using std::map;
using boost::bind;
using boost::optional;
using boost::shared_ptr;
using boost::function;
#if BOOST_VERSION >= 106100
using namespace boost::placeholders;
#endif

static
bool
do_nothing ()
{
	return false;
}

Page::Page (wxSize panel_size, int border)
	: _border (border)
	, _panel (0)
	, _panel_size (panel_size)
	, _window_exists (false)
{
	_config_connection = Config::instance()->Changed.connect (bind (&Page::config_changed_wrapper, this));
}


wxWindow*
Page::CreateWindow (wxWindow* parent)
{
	return create_window (parent);
}


wxWindow*
Page::create_window (wxWindow* parent)
{
	_panel = new wxPanel (parent, wxID_ANY, wxDefaultPosition, _panel_size);
	wxBoxSizer* s = new wxBoxSizer (wxVERTICAL);
	_panel->SetSizer (s);

	setup ();
	_window_exists = true;
	config_changed ();

	_panel->Bind (wxEVT_DESTROY, bind (&Page::window_destroyed, this));

	return _panel;
}

void
Page::config_changed_wrapper ()
{
	if (_window_exists) {
		config_changed ();
	}
}

void
Page::window_destroyed ()
{
	_window_exists = false;
}


GeneralPage::GeneralPage (wxSize panel_size, int border)
	: Page (panel_size, border)
{

}


wxString
GeneralPage::GetName () const
{
	return _("General");
}


void
GeneralPage::add_language_controls (wxGridBagSizer* table, int& r)
{
	_set_language = new CheckBox (_panel, _("Set language"));
	table->Add (_set_language, wxGBPosition (r, 0), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
	_language = new wxChoice (_panel, wxID_ANY);
	vector<pair<string, string> > languages;
	languages.push_back (make_pair ("Čeština", "cs_CZ"));
	languages.push_back (make_pair ("汉语/漢語", "zh_CN"));
	languages.push_back (make_pair ("Dansk", "da_DK"));
	languages.push_back (make_pair ("Deutsch", "de_DE"));
	languages.push_back (make_pair ("English", "en_GB"));
	languages.push_back (make_pair ("Español", "es_ES"));
	languages.push_back (make_pair ("Français", "fr_FR"));
	languages.push_back (make_pair ("Italiano", "it_IT"));
	languages.push_back (make_pair ("Nederlands", "nl_NL"));
	languages.push_back (make_pair ("Русский", "ru_RU"));
	languages.push_back (make_pair ("Polski", "pl_PL"));
	languages.push_back (make_pair ("Português europeu", "pt_PT"));
	languages.push_back (make_pair ("Português do Brasil", "pt_BR"));
	languages.push_back (make_pair ("Svenska", "sv_SE"));
	languages.push_back (make_pair ("Slovenský jazyk", "sk_SK"));
	languages.push_back (make_pair ("Türkçe", "tr_TR"));
	languages.push_back (make_pair ("українська мова", "uk_UA"));
	checked_set (_language, languages);
	table->Add (_language, wxGBPosition (r, 1));
	++r;

	wxStaticText* restart = add_label_to_sizer (
		table, _panel, _("(restart DCP-o-matic to see language changes)"), false, wxGBPosition (r, 0), wxGBSpan (1, 2)
		);
	wxFont font = restart->GetFont();
	font.SetStyle (wxFONTSTYLE_ITALIC);
	font.SetPointSize (font.GetPointSize() - 1);
	restart->SetFont (font);
	++r;

	_set_language->Bind (wxEVT_CHECKBOX, bind (&GeneralPage::set_language_changed, this));
	_language->Bind     (wxEVT_CHOICE,   bind (&GeneralPage::language_changed,     this));
}

void
GeneralPage::add_update_controls (wxGridBagSizer* table, int& r)
{
	_check_for_updates = new CheckBox (_panel, _("Check for updates on startup"));
	table->Add (_check_for_updates, wxGBPosition (r, 0), wxGBSpan (1, 2));
	++r;

	_check_for_test_updates = new CheckBox (_panel, _("Check for testing updates on startup"));
	table->Add (_check_for_test_updates, wxGBPosition (r, 0), wxGBSpan (1, 2));
	++r;

	_check_for_updates->Bind (wxEVT_CHECKBOX, bind (&GeneralPage::check_for_updates_changed, this));
	_check_for_test_updates->Bind (wxEVT_CHECKBOX, bind (&GeneralPage::check_for_test_updates_changed, this));
}

void
GeneralPage::config_changed ()
{
	Config* config = Config::instance ();

	checked_set (_set_language, static_cast<bool>(config->language()));

	/* Backwards compatibility of config file */

	map<string, string> compat_map;
	compat_map["fr"] = "fr_FR";
	compat_map["it"] = "it_IT";
	compat_map["es"] = "es_ES";
	compat_map["sv"] = "sv_SE";
	compat_map["de"] = "de_DE";
	compat_map["nl"] = "nl_NL";
	compat_map["ru"] = "ru_RU";
	compat_map["pl"] = "pl_PL";
	compat_map["da"] = "da_DK";
	compat_map["pt"] = "pt_PT";
	compat_map["sk"] = "sk_SK";
	compat_map["cs"] = "cs_CZ";
	compat_map["uk"] = "uk_UA";

	string lang = config->language().get_value_or ("en_GB");
	if (compat_map.find (lang) != compat_map.end ()) {
		lang = compat_map[lang];
	}

	checked_set (_language, lang);

	checked_set (_check_for_updates, config->check_for_updates ());
	checked_set (_check_for_test_updates, config->check_for_test_updates ());

	setup_sensitivity ();
}

void
GeneralPage::setup_sensitivity ()
{
	_language->Enable (_set_language->GetValue ());
	_check_for_test_updates->Enable (_check_for_updates->GetValue ());
}

void
GeneralPage::set_language_changed ()
{
	setup_sensitivity ();
	if (_set_language->GetValue ()) {
		language_changed ();
	} else {
		Config::instance()->unset_language ();
	}
}

void
GeneralPage::language_changed ()
{
	int const sel = _language->GetSelection ();
	if (sel != -1) {
		Config::instance()->set_language (string_client_data (_language->GetClientObject (sel)));
	} else {
		Config::instance()->unset_language ();
	}
}

void
GeneralPage::check_for_updates_changed ()
{
	Config::instance()->set_check_for_updates (_check_for_updates->GetValue ());
}

void
GeneralPage::check_for_test_updates_changed ()
{
	Config::instance()->set_check_for_test_updates (_check_for_test_updates->GetValue ());
}

CertificateChainEditor::CertificateChainEditor (
	wxWindow* parent,
	wxString title,
	int border,
	function<void (shared_ptr<dcp::CertificateChain>)> set,
	function<shared_ptr<const dcp::CertificateChain> (void)> get,
	function<bool (void)> nag_alter
	)
	: wxDialog (parent, wxID_ANY, title)
	, _set (set)
	, _get (get)
	, _nag_alter (nag_alter)
{
	_sizer = new wxBoxSizer (wxVERTICAL);

	wxBoxSizer* certificates_sizer = new wxBoxSizer (wxHORIZONTAL);
	_sizer->Add (certificates_sizer, 0, wxALL, border);

	_certificates = new wxListCtrl (this, wxID_ANY, wxDefaultPosition, wxSize (440, 150), wxLC_REPORT | wxLC_SINGLE_SEL);

	{
		wxListItem ip;
		ip.SetId (0);
		ip.SetText (_("Type"));
		ip.SetWidth (100);
		_certificates->InsertColumn (0, ip);
	}

	{
		wxListItem ip;
		ip.SetId (1);
		ip.SetText (_("Thumbprint"));
		ip.SetWidth (340);

		wxFont font = ip.GetFont ();
		font.SetFamily (wxFONTFAMILY_TELETYPE);
		ip.SetFont (font);

		_certificates->InsertColumn (1, ip);
	}

	certificates_sizer->Add (_certificates, 1, wxEXPAND);

	{
		wxSizer* s = new wxBoxSizer (wxVERTICAL);
		_add_certificate = new Button (this, _("Add..."));
		s->Add (_add_certificate, 1, wxTOP | wxBOTTOM | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP);
		_remove_certificate = new Button (this, _("Remove"));
		s->Add (_remove_certificate, 1, wxTOP | wxBOTTOM | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP);
		_export_certificate = new Button (this, _("Export certificate..."));
		s->Add (_export_certificate, 1, wxTOP | wxBOTTOM | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP);
		_export_chain = new Button (this, _("Export chain..."));
		s->Add (_export_chain, 1, wxTOP | wxBOTTOM | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP);
		certificates_sizer->Add (s, 0, wxLEFT, DCPOMATIC_SIZER_X_GAP);
	}

	wxGridBagSizer* table = new wxGridBagSizer (DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
	_sizer->Add (table, 1, wxALL | wxEXPAND, border);
	int r = 0;

	add_label_to_sizer (table, this, _("Leaf private key"), true, wxGBPosition (r, 0));
	_private_key = new StaticText (this, wxT(""));
	wxFont font = _private_key->GetFont ();
	font.SetFamily (wxFONTFAMILY_TELETYPE);
	_private_key->SetFont (font);
	table->Add (_private_key, wxGBPosition (r, 1), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
	_import_private_key = new Button (this, _("Import..."));
	table->Add (_import_private_key, wxGBPosition (r, 2));
	_export_private_key = new Button (this, _("Export..."));
	table->Add (_export_private_key, wxGBPosition (r, 3));
	++r;

	_button_sizer = new wxBoxSizer (wxHORIZONTAL);
	_remake_certificates = new Button (this, _("Re-make certificates and key..."));
	_button_sizer->Add (_remake_certificates, 1, wxRIGHT, border);
	table->Add (_button_sizer, wxGBPosition (r, 0), wxGBSpan (1, 4));
	++r;

	_private_key_bad = new StaticText (this, _("Leaf private key does not match leaf certificate!"));
	font = *wxSMALL_FONT;
	font.SetWeight (wxFONTWEIGHT_BOLD);
	_private_key_bad->SetFont (font);
	table->Add (_private_key_bad, wxGBPosition (r, 0), wxGBSpan (1, 3));
	++r;

	_add_certificate->Bind     (wxEVT_BUTTON,       bind (&CertificateChainEditor::add_certificate, this));
	_remove_certificate->Bind  (wxEVT_BUTTON,       bind (&CertificateChainEditor::remove_certificate, this));
	_export_certificate->Bind  (wxEVT_BUTTON,       bind (&CertificateChainEditor::export_certificate, this));
	_certificates->Bind        (wxEVT_LIST_ITEM_SELECTED,   bind (&CertificateChainEditor::update_sensitivity, this));
	_certificates->Bind        (wxEVT_LIST_ITEM_DESELECTED, bind (&CertificateChainEditor::update_sensitivity, this));
	_remake_certificates->Bind (wxEVT_BUTTON,       bind (&CertificateChainEditor::remake_certificates, this));
	_export_chain->Bind        (wxEVT_BUTTON,       bind (&CertificateChainEditor::export_chain, this));
	_import_private_key->Bind  (wxEVT_BUTTON,       bind (&CertificateChainEditor::import_private_key, this));
	_export_private_key->Bind  (wxEVT_BUTTON,       bind (&CertificateChainEditor::export_private_key, this));

	wxSizer* buttons = CreateSeparatedButtonSizer (wxCLOSE);
	if (buttons) {
		_sizer->Add (buttons, wxSizerFlags().Expand().DoubleBorder());
	}

	SetSizerAndFit (_sizer);

	update_certificate_list ();
	update_private_key ();
	update_sensitivity ();
}

void
CertificateChainEditor::add_button (wxWindow* button)
{
	_button_sizer->Add (button, 0, wxLEFT | wxRIGHT, DCPOMATIC_SIZER_X_GAP);
	_sizer->Layout ();
}

void
CertificateChainEditor::add_certificate ()
{
	wxFileDialog* d = new wxFileDialog (this, _("Select Certificate File"));

	if (d->ShowModal() == wxID_OK) {
		try {
			dcp::Certificate c;
			string extra;
			try {
				extra = c.read_string (dcp::file_to_string (wx_to_std (d->GetPath ())));
			} catch (boost::filesystem::filesystem_error& e) {
				error_dialog (this, _("Could not import certificate (%s)"), d->GetPath());
				d->Destroy ();
				return;
			}

			if (!extra.empty ()) {
				message_dialog (
					this,
					_("This file contains other certificates (or other data) after its first certificate. "
					  "Only the first certificate will be used.")
					);
			}
			shared_ptr<dcp::CertificateChain> chain(new dcp::CertificateChain(*_get().get()));
			chain->add (c);
			if (!chain->chain_valid ()) {
				error_dialog (
					this,
					_("Adding this certificate would make the chain inconsistent, so it will not be added. "
					  "Add certificates in order from root to intermediate to leaf.")
					);
				chain->remove (c);
			} else {
				_set (chain);
				update_certificate_list ();
			}
		} catch (dcp::MiscError& e) {
			error_dialog (this, _("Could not read certificate file."), std_to_wx(e.what()));
		}
	}

	d->Destroy ();

	update_sensitivity ();
}

void
CertificateChainEditor::remove_certificate ()
{
	if (_nag_alter()) {
		/* Cancel was clicked */
		return;
	}

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

	_certificates->DeleteItem (i);
	shared_ptr<dcp::CertificateChain> chain(new dcp::CertificateChain(*_get().get()));
	chain->remove (i);
	_set (chain);

	update_sensitivity ();
	update_certificate_list ();
}

void
CertificateChainEditor::export_certificate ()
{
	int i = _certificates->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
	if (i == -1) {
		return;
	}

	wxFileDialog* d = new wxFileDialog (
		this, _("Select Certificate File"), wxEmptyString, wxEmptyString, wxT ("PEM files (*.pem)|*.pem"),
		wxFD_SAVE | wxFD_OVERWRITE_PROMPT
		);

	dcp::CertificateChain::List all = _get()->root_to_leaf ();
	dcp::CertificateChain::List::iterator j = all.begin ();
	for (int k = 0; k < i; ++k) {
		++j;
	}

	if (d->ShowModal () == wxID_OK) {
		boost::filesystem::path path (wx_to_std(d->GetPath()));
		FILE* f = fopen_boost (path, "w");
		if (!f) {
			throw OpenFileError (path, errno, OpenFileError::WRITE);
		}

		string const s = j->certificate (true);
		checked_fwrite (s.c_str(), s.length(), f, path);
		fclose (f);
	}
	d->Destroy ();
}

void
CertificateChainEditor::export_chain ()
{
	wxFileDialog* d = new wxFileDialog (
		this, _("Select Chain File"), wxEmptyString, wxEmptyString, wxT("PEM files (*.pem)|*.pem"),
		wxFD_SAVE | wxFD_OVERWRITE_PROMPT
		);

	if (d->ShowModal () == wxID_OK) {
		boost::filesystem::path path (wx_to_std(d->GetPath()));
		FILE* f = fopen_boost (path, "w");
		if (!f) {
			throw OpenFileError (path, errno, OpenFileError::WRITE);
		}

		string const s = _get()->chain();
		checked_fwrite (s.c_str(), s.length(), f, path);
		fclose (f);
	}

	d->Destroy ();
}

void
CertificateChainEditor::update_certificate_list ()
{
	_certificates->DeleteAllItems ();
	size_t n = 0;
	dcp::CertificateChain::List certs = _get()->root_to_leaf ();
	BOOST_FOREACH (dcp::Certificate const & i, certs) {
		wxListItem item;
		item.SetId (n);
		_certificates->InsertItem (item);
		_certificates->SetItem (n, 1, std_to_wx (i.thumbprint ()));

		if (n == 0) {
			_certificates->SetItem (n, 0, _("Root"));
		} else if (n == (certs.size() - 1)) {
			_certificates->SetItem (n, 0, _("Leaf"));
		} else {
			_certificates->SetItem (n, 0, _("Intermediate"));
		}

		++n;
	}

	static wxColour normal = _private_key_bad->GetForegroundColour ();

	if (_get()->private_key_valid()) {
		_private_key_bad->Hide ();
		_private_key_bad->SetForegroundColour (normal);
	} else {
		_private_key_bad->Show ();
		_private_key_bad->SetForegroundColour (wxColour (255, 0, 0));
	}
}

void
CertificateChainEditor::remake_certificates ()
{
	shared_ptr<const dcp::CertificateChain> chain = _get();

	string subject_organization_name;
	string subject_organizational_unit_name;
	string root_common_name;
	string intermediate_common_name;
	string leaf_common_name;

	dcp::CertificateChain::List all = chain->root_to_leaf ();

	if (all.size() >= 1) {
		/* Have a root */
		subject_organization_name = chain->root().subject_organization_name ();
		subject_organizational_unit_name = chain->root().subject_organizational_unit_name ();
		root_common_name = chain->root().subject_common_name ();
	}

	if (all.size() >= 2) {
		/* Have a leaf */
		leaf_common_name = chain->leaf().subject_common_name ();
	}

	if (all.size() >= 3) {
		/* Have an intermediate */
		dcp::CertificateChain::List::iterator i = all.begin ();
		++i;
		intermediate_common_name = i->subject_common_name ();
	}

	if (_nag_alter()) {
		/* Cancel was clicked */
		return;
	}

	MakeChainDialog* d = new MakeChainDialog (
		this,
		subject_organization_name,
		subject_organizational_unit_name,
		root_common_name,
		intermediate_common_name,
		leaf_common_name
		);

	if (d->ShowModal () == wxID_OK) {
		_set (
			shared_ptr<dcp::CertificateChain> (
				new dcp::CertificateChain (
					openssl_path (),
					d->organisation (),
					d->organisational_unit (),
					d->root_common_name (),
					d->intermediate_common_name (),
					d->leaf_common_name ()
					)
				)
			);

		update_certificate_list ();
		update_private_key ();
	}

	d->Destroy ();
}

void
CertificateChainEditor::update_sensitivity ()
{
	/* We can only remove the leaf certificate */
	_remove_certificate->Enable (_certificates->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED) == (_certificates->GetItemCount() - 1));
	_export_certificate->Enable (_certificates->GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED) != -1);
}

void
CertificateChainEditor::update_private_key ()
{
	checked_set (_private_key, dcp::private_key_fingerprint (_get()->key().get()));
	_sizer->Layout ();
}

void
CertificateChainEditor::import_private_key ()
{
	wxFileDialog* d = new wxFileDialog (this, _("Select Key File"));

	if (d->ShowModal() == wxID_OK) {
		try {
			boost::filesystem::path p (wx_to_std (d->GetPath ()));
			if (boost::filesystem::file_size (p) > 8192) {
				error_dialog (
					this,
					wxString::Format (_("Could not read key file; file is too long (%s)"), std_to_wx (p.string ()))
					);
				return;
			}

			shared_ptr<dcp::CertificateChain> chain(new dcp::CertificateChain(*_get().get()));
			chain->set_key (dcp::file_to_string (p));
			_set (chain);
			update_private_key ();
		} catch (dcp::MiscError& e) {
			error_dialog (this, _("Could not read certificate file."), std_to_wx(e.what()));
		}
	}

	d->Destroy ();

	update_sensitivity ();
}

void
CertificateChainEditor::export_private_key ()
{
	optional<string> key = _get()->key();
	if (!key) {
		return;
	}

	wxFileDialog* d = new wxFileDialog (
		this, _("Select Key File"), wxEmptyString, wxEmptyString, wxT ("PEM files (*.pem)|*.pem"),
		wxFD_SAVE | wxFD_OVERWRITE_PROMPT
		);

	if (d->ShowModal () == wxID_OK) {
		boost::filesystem::path path (wx_to_std(d->GetPath()));
		FILE* f = fopen_boost (path, "w");
		if (!f) {
			throw OpenFileError (path, errno, OpenFileError::WRITE);
		}

		string const s = _get()->key().get ();
		checked_fwrite (s.c_str(), s.length(), f, path);
		fclose (f);
	}
	d->Destroy ();
}

wxString
KeysPage::GetName () const
{
	return _("Keys");
}

void
KeysPage::setup ()
{
	wxFont subheading_font (*wxNORMAL_FONT);
	subheading_font.SetWeight (wxFONTWEIGHT_BOLD);

	wxSizer* sizer = _panel->GetSizer();

	{
		wxStaticText* m = new StaticText (_panel, _("Decrypting KDMs"));
		m->SetFont (subheading_font);
		sizer->Add (m, 0, wxALL, _border);
	}

	wxSizer* buttons = new wxBoxSizer (wxVERTICAL);

	wxButton* export_decryption_certificate = new Button (_panel, _("Export KDM decryption certificate..."));
	buttons->Add (export_decryption_certificate, 0, wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
	wxButton* export_settings = new Button (_panel, _("Export all KDM decryption settings..."));
	buttons->Add (export_settings, 0, wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
	wxButton* import_settings = new Button (_panel, _("Import all KDM decryption settings..."));
	buttons->Add (import_settings, 0, wxBOTTOM, DCPOMATIC_BUTTON_STACK_GAP);
	wxButton* decryption_advanced = new Button (_panel, _("Advanced..."));
	buttons->Add (decryption_advanced, 0);

	sizer->Add (buttons, 0, wxLEFT, _border);

	export_decryption_certificate->Bind (wxEVT_BUTTON, bind (&KeysPage::export_decryption_certificate, this));
	export_settings->Bind (wxEVT_BUTTON, bind (&KeysPage::export_decryption_chain_and_key, this));
	import_settings->Bind (wxEVT_BUTTON, bind (&KeysPage::import_decryption_chain_and_key, this));
	decryption_advanced->Bind (wxEVT_BUTTON, bind (&KeysPage::decryption_advanced, this));

	{
		wxStaticText* m = new StaticText (_panel, _("Signing DCPs and KDMs"));
		m->SetFont (subheading_font);
		sizer->Add (m, 0, wxALL, _border);
	}

	wxButton* signing_advanced = new Button (_panel, _("Advanced..."));
	sizer->Add (signing_advanced, 0, wxLEFT | wxBOTTOM, _border);
	signing_advanced->Bind (wxEVT_BUTTON, bind (&KeysPage::signing_advanced, this));
}

void
KeysPage::decryption_advanced ()
{
	CertificateChainEditor* c = new CertificateChainEditor (
		_panel, _("Decrypting KDMs"), _border,
		bind (&Config::set_decryption_chain, Config::instance (), _1),
		bind (&Config::decryption_chain, Config::instance ()),
		bind (&KeysPage::nag_alter_decryption_chain, this)
		);

	c->ShowModal();
}

void
KeysPage::signing_advanced ()
{
	CertificateChainEditor* c = new CertificateChainEditor (
		_panel, _("Signing DCPs and KDMs"), _border,
		bind (&Config::set_signer_chain, Config::instance (), _1),
		bind (&Config::signer_chain, Config::instance ()),
		bind (&do_nothing)
		);

	c->ShowModal();
}

void
KeysPage::export_decryption_chain_and_key ()
{
	wxFileDialog* d = new wxFileDialog (
		_panel, _("Select Export File"), wxEmptyString, wxEmptyString, wxT ("DOM files (*.dom)|*.dom"),
		wxFD_SAVE | wxFD_OVERWRITE_PROMPT
		);

	if (d->ShowModal () == wxID_OK) {
		boost::filesystem::path path (wx_to_std(d->GetPath()));
		FILE* f = fopen_boost (path, "w");
		if (!f) {
			throw OpenFileError (path, errno, OpenFileError::WRITE);
		}

		string const chain = Config::instance()->decryption_chain()->chain();
		checked_fwrite (chain.c_str(), chain.length(), f, path);
		optional<string> const key = Config::instance()->decryption_chain()->key();
		DCPOMATIC_ASSERT (key);
		checked_fwrite (key->c_str(), key->length(), f, path);
		fclose (f);
	}
	d->Destroy ();

}

void
KeysPage::import_decryption_chain_and_key ()
{
	if (NagDialog::maybe_nag (
		    _panel,
		    Config::NAG_IMPORT_DECRYPTION_CHAIN,
		    _("If you continue with this operation you will no longer be able to use any DKDMs that you have created with the current certificates and key.  Also, any KDMs that have been sent to you for those certificates will become useless.  Proceed with caution!"),
		    true
		    )) {
		return;
	}

	wxFileDialog* d = new wxFileDialog (
		_panel, _("Select File To Import"), wxEmptyString, wxEmptyString, wxT ("DOM files (*.dom)|*.dom")
		);

	if (d->ShowModal () == wxID_OK) {
		shared_ptr<dcp::CertificateChain> new_chain(new dcp::CertificateChain());

		FILE* f = fopen_boost (wx_to_std (d->GetPath ()), "r");
		if (!f) {
			throw OpenFileError (wx_to_std (d->GetPath ()), errno, OpenFileError::WRITE);
		}

		string current;
		while (!feof (f)) {
			char buffer[128];
			if (fgets (buffer, 128, f) == 0) {
				break;
			}
			current += buffer;
			if (strncmp (buffer, "-----END CERTIFICATE-----", 25) == 0) {
				new_chain->add (dcp::Certificate (current));
				current = "";
			} else if (strncmp (buffer, "-----END RSA PRIVATE KEY-----", 29) == 0) {
				new_chain->set_key (current);
				current = "";
			}
		}
		fclose (f);

		if (new_chain->chain_valid() && new_chain->private_key_valid()) {
			Config::instance()->set_decryption_chain (new_chain);
		} else {
			error_dialog (_panel, _("Invalid DCP-o-matic export file"));
		}
	}
	d->Destroy ();
}

bool
KeysPage::nag_alter_decryption_chain ()
{
	return NagDialog::maybe_nag (
		_panel,
		Config::NAG_ALTER_DECRYPTION_CHAIN,
		_("If you continue with this operation you will no longer be able to use any DKDMs that you have created.  Also, any KDMs that have been sent to you will become useless.  Proceed with caution!"),
		true
		);
}

void
KeysPage::export_decryption_certificate ()
{
	wxFileDialog* d = new wxFileDialog (
		_panel, _("Select Certificate File"), wxEmptyString, _("dcpomatic_kdm_decryption_cert.pem"), wxT ("PEM files (*.pem)|*.pem"),
		wxFD_SAVE | wxFD_OVERWRITE_PROMPT
		);

	if (d->ShowModal () == wxID_OK) {
		boost::filesystem::path path (wx_to_std(d->GetPath()));
		FILE* f = fopen_boost (path, "w");
		if (!f) {
			throw OpenFileError (path, errno, OpenFileError::WRITE);
		}

		string const s = Config::instance()->decryption_chain()->leaf().certificate (true);
		checked_fwrite (s.c_str(), s.length(), f, path);
		fclose (f);
	}

	d->Destroy ();
}

wxString
SoundPage::GetName () const
{
	return _("Sound");
}

void
SoundPage::setup ()
{
	wxGridBagSizer* table = new wxGridBagSizer (DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
	_panel->GetSizer()->Add (table, 1, wxALL | wxEXPAND, _border);

	int r = 0;

	_sound = new CheckBox (_panel, _("Play sound via"));
	table->Add (_sound, wxGBPosition (r, 0), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
	wxBoxSizer* s = new wxBoxSizer (wxHORIZONTAL);
	_sound_output = new wxChoice (_panel, wxID_ANY);
	s->Add (_sound_output, 0);
	_sound_output_details = new wxStaticText (_panel, wxID_ANY, wxT(""));
	s->Add (_sound_output_details, 1, wxALIGN_CENTER_VERTICAL | wxLEFT, DCPOMATIC_SIZER_X_GAP);
	table->Add (s, wxGBPosition(r, 1));
	++r;

	add_label_to_sizer (table, _panel, _("Mapping"), true, wxGBPosition(r, 0));
	_map = new AudioMappingView (_panel, _("DCP"), _("DCP"), _("Output"), _("output"));
	_map->SetSize (-1, 600);
	table->Add (_map, wxGBPosition(r, 1), wxDefaultSpan, wxEXPAND);
	++r;

	_reset_to_default = new Button (_panel, _("Reset to default"));
	table->Add (_reset_to_default, wxGBPosition(r, 1));
	++r;

	wxFont font = _sound_output_details->GetFont();
	font.SetStyle (wxFONTSTYLE_ITALIC);
	font.SetPointSize (font.GetPointSize() - 1);
	_sound_output_details->SetFont (font);

	RtAudio audio (DCPOMATIC_RTAUDIO_API);
	for (unsigned int i = 0; i < audio.getDeviceCount(); ++i) {
		try {
			RtAudio::DeviceInfo dev = audio.getDeviceInfo (i);
			if (dev.probed && dev.outputChannels > 0) {
				_sound_output->Append (std_to_wx (dev.name));
			}
		} catch (RtAudioError&) {
			/* Something went wrong so let's just ignore that device */
		}
	}

	_sound->Bind        (wxEVT_CHECKBOX, bind(&SoundPage::sound_changed, this));
	_sound_output->Bind (wxEVT_CHOICE,   bind(&SoundPage::sound_output_changed, this));
	_map->Changed.connect (bind(&SoundPage::map_changed, this, _1));
	_reset_to_default->Bind (wxEVT_BUTTON,   bind(&SoundPage::reset_to_default, this));
}

void
SoundPage::reset_to_default ()
{
	Config::instance()->set_audio_mapping_to_default ();
}

void
SoundPage::map_changed (AudioMapping m)
{
	Config::instance()->set_audio_mapping (m);
}

void
SoundPage::sound_changed ()
{
	Config::instance()->set_sound (_sound->GetValue ());
}

void
SoundPage::sound_output_changed ()
{
	RtAudio audio (DCPOMATIC_RTAUDIO_API);
	optional<string> const so = get_sound_output();
	string default_device;
	try {
		default_device = audio.getDeviceInfo(audio.getDefaultOutputDevice()).name;
	} catch (RtAudioError&) {
		/* Never mind */
	}
	if (!so || *so == default_device) {
		Config::instance()->unset_sound_output ();
	} else {
		Config::instance()->set_sound_output (*so);
	}
}

void
SoundPage::config_changed ()
{
	Config* config = Config::instance ();

	checked_set (_sound, config->sound ());

	optional<string> const current_so = get_sound_output ();
	optional<string> configured_so;

	if (config->sound_output()) {
		configured_so = config->sound_output().get();
	} else {
		/* No configured output means we should use the default */
		RtAudio audio (DCPOMATIC_RTAUDIO_API);
		try {
			configured_so = audio.getDeviceInfo(audio.getDefaultOutputDevice()).name;
		} catch (RtAudioError&) {
			/* Probably no audio devices at all */
		}
	}

	if (configured_so && current_so != configured_so) {
		/* Update _sound_output with the configured value */
		unsigned int i = 0;
		while (i < _sound_output->GetCount()) {
			if (_sound_output->GetString(i) == std_to_wx(*configured_so)) {
				_sound_output->SetSelection (i);
				break;
			}
			++i;
		}
	}

	RtAudio audio (DCPOMATIC_RTAUDIO_API);

	map<int, wxString> apis;
	apis[RtAudio::MACOSX_CORE]    = _("CoreAudio");
	apis[RtAudio::WINDOWS_ASIO]   = _("ASIO");
	apis[RtAudio::WINDOWS_DS]     = _("Direct Sound");
	apis[RtAudio::WINDOWS_WASAPI] = _("WASAPI");
	apis[RtAudio::UNIX_JACK]      = _("JACK");
	apis[RtAudio::LINUX_ALSA]     = _("ALSA");
	apis[RtAudio::LINUX_PULSE]    = _("PulseAudio");
	apis[RtAudio::LINUX_OSS]      = _("OSS");
	apis[RtAudio::RTAUDIO_DUMMY]  = _("Dummy");

	int channels = 0;
	if (configured_so) {
		for (unsigned int i = 0; i < audio.getDeviceCount(); ++i) {
			try {
				RtAudio::DeviceInfo info = audio.getDeviceInfo(i);
				if (info.name == *configured_so && info.outputChannels > 0) {
					channels = info.outputChannels;
				}
			} catch (RtAudioError&) {
				/* Never mind */
			}
		}
	}

	_sound_output_details->SetLabel (
		wxString::Format(_("%d channels on %s"), channels, apis[audio.getCurrentApi()])
		);

	_map->set (Config::instance()->audio_mapping(channels));

	vector<NamedChannel> input;
	for (int i = 0; i < MAX_DCP_AUDIO_CHANNELS; ++i) {
		input.push_back (NamedChannel(short_audio_channel_name(i), i));
	}
	_map->set_input_channels (input);

	vector<NamedChannel> output;
	for (int i = 0; i < channels; ++i) {
		output.push_back (NamedChannel(dcp::raw_convert<string>(i), i));
	}
	_map->set_output_channels (output);

	setup_sensitivity ();
}

void
SoundPage::setup_sensitivity ()
{
	_sound_output->Enable (_sound->GetValue());
}

/** @return Currently-selected preview sound output in the dialogue */
optional<string>
SoundPage::get_sound_output ()
{
	int const sel = _sound_output->GetSelection ();
	if (sel == wxNOT_FOUND) {
		return optional<string> ();
	}

	return wx_to_std (_sound_output->GetString (sel));
}


LocationsPage::LocationsPage (wxSize panel_size, int border)
	: Page (panel_size, border)
{

}

wxString
LocationsPage::GetName () const
{
	return _("Locations");
}

#ifdef DCPOMATIC_OSX
wxBitmap
LocationsPage::GetLargeIcon () const
{
	return wxBitmap ("locations", wxBITMAP_TYPE_PNG_RESOURCE);
}
#endif

void
LocationsPage::setup ()
{
	int r = 0;

	wxGridBagSizer* table = new wxGridBagSizer (DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
	_panel->GetSizer()->Add (table, 1, wxALL | wxEXPAND, _border);

	add_label_to_sizer (table, _panel, _("Content directory"), true, wxGBPosition (r, 0));
	_content_directory = new wxDirPickerCtrl (_panel, wxID_ANY, wxEmptyString, wxDirSelectorPromptStr, wxDefaultPosition, wxSize (300, -1));
	table->Add (_content_directory, wxGBPosition (r, 1));
	++r;

	add_label_to_sizer (table, _panel, _("Playlist directory"), true, wxGBPosition (r, 0));
	_playlist_directory = new wxDirPickerCtrl (_panel, wxID_ANY, wxEmptyString, wxDirSelectorPromptStr, wxDefaultPosition, wxSize (300, -1));
	table->Add (_playlist_directory, wxGBPosition (r, 1));
	++r;

	add_label_to_sizer (table, _panel, _("KDM directory"), true, wxGBPosition (r, 0));
	_kdm_directory = new wxDirPickerCtrl (_panel, wxID_ANY, wxEmptyString, wxDirSelectorPromptStr, wxDefaultPosition, wxSize (300, -1));
	table->Add (_kdm_directory, wxGBPosition (r, 1));
	++r;

	_content_directory->Bind (wxEVT_DIRPICKER_CHANGED, bind(&LocationsPage::content_directory_changed, this));
	_playlist_directory->Bind (wxEVT_DIRPICKER_CHANGED, bind(&LocationsPage::playlist_directory_changed, this));
	_kdm_directory->Bind (wxEVT_DIRPICKER_CHANGED, bind(&LocationsPage::kdm_directory_changed, this));
}

void
LocationsPage::config_changed ()
{
	Config* config = Config::instance ();

	if (config->player_content_directory()) {
		checked_set (_content_directory, *config->player_content_directory());
	}
	if (config->player_playlist_directory()) {
		checked_set (_playlist_directory, *config->player_playlist_directory());
	}
	if (config->player_kdm_directory()) {
		checked_set (_kdm_directory, *config->player_kdm_directory());
	}
}

void
LocationsPage::content_directory_changed ()
{
	Config::instance()->set_player_content_directory(wx_to_std(_content_directory->GetPath()));
}

void
LocationsPage::playlist_directory_changed ()
{
	Config::instance()->set_player_playlist_directory(wx_to_std(_playlist_directory->GetPath()));
}

void
LocationsPage::kdm_directory_changed ()
{
	Config::instance()->set_player_kdm_directory(wx_to_std(_kdm_directory->GetPath()));
}