/*
    Copyright (C) 2013-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 "screen.h"
#include "kdm_with_metadata.h"
#include "film.h"
#include "cinema.h"
#include <libxml++/libxml++.h>
#include <boost/foreach.hpp>
#include <boost/algorithm/string.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>

using std::string;
using std::vector;
using boost::shared_ptr;
using boost::optional;
using namespace dcpomatic;

Screen::Screen (cxml::ConstNodePtr node)
	: name (node->string_child("Name"))
	, notes (node->optional_string_child("Notes").get_value_or (""))
{
	if (node->optional_string_child ("Certificate")) {
		recipient = dcp::Certificate (node->string_child ("Certificate"));
	} else if (node->optional_string_child ("Recipient")) {
		recipient = dcp::Certificate (node->string_child ("Recipient"));
	}

	BOOST_FOREACH (cxml::ConstNodePtr i, node->node_children ("TrustedDevice")) {
		if (boost::algorithm::starts_with(i->content(), "-----BEGIN CERTIFICATE-----")) {
			trusted_devices.push_back (TrustedDevice(dcp::Certificate(i->content())));
		} else {
			trusted_devices.push_back (TrustedDevice(i->content()));
		}
	}
}

void
Screen::as_xml (xmlpp::Element* parent) const
{
	parent->add_child("Name")->add_child_text (name);
	if (recipient) {
		parent->add_child("Recipient")->add_child_text (recipient->certificate (true));
	}

	parent->add_child("Notes")->add_child_text (notes);

	BOOST_FOREACH (TrustedDevice i, trusted_devices) {
		parent->add_child("TrustedDevice")->add_child_text(i.as_string());
	}
}

vector<string>
Screen::trusted_device_thumbprints () const
{
	vector<string> t;
	BOOST_FOREACH (TrustedDevice i, trusted_devices) {
		t.push_back (i.thumbprint());
	}
	return t;
}

TrustedDevice::TrustedDevice (string thumbprint)
	: _thumbprint (thumbprint)
{

}

TrustedDevice::TrustedDevice (dcp::Certificate certificate)
	: _certificate (certificate)
{

}

string
TrustedDevice::as_string () const
{
	if (_certificate) {
		return _certificate->certificate(true);
	}

	return *_thumbprint;
}

string
TrustedDevice::thumbprint () const
{
	if (_certificate) {
		return _certificate->thumbprint ();
	}

	return *_thumbprint;
}


KDMWithMetadataPtr
kdm_for_screen (
	shared_ptr<const Film> film,
	boost::filesystem::path cpl,
	shared_ptr<const dcpomatic::Screen> screen,
	boost::posix_time::ptime valid_from,
	boost::posix_time::ptime valid_to,
	dcp::Formulation formulation,
	bool disable_forensic_marking_picture,
	optional<int> disable_forensic_marking_audio
	)
{
	if (!screen->recipient) {
		return KDMWithMetadataPtr();
	}

	shared_ptr<const Cinema> cinema = screen->cinema;
	dcp::LocalTime const begin(valid_from, cinema ? cinema->utc_offset_hour() : 0, cinema ? cinema->utc_offset_minute() : 0);
	dcp::LocalTime const end  (valid_to,   cinema ? cinema->utc_offset_hour() : 0, cinema ? cinema->utc_offset_minute() : 0);

	dcp::EncryptedKDM const kdm = film->make_kdm (
			screen->recipient.get(),
			screen->trusted_device_thumbprints(),
			cpl,
			begin,
			end,
			formulation,
			disable_forensic_marking_picture,
			disable_forensic_marking_audio
			);

	dcp::NameFormat::Map name_values;
	if (cinema) {
		name_values['c'] = cinema->name;
	}
	name_values['s'] = screen->name;
	name_values['f'] = film->name();
	name_values['b'] = begin.date() + " " + begin.time_of_day(true, false);
	name_values['e'] = end.date() + " " + end.time_of_day(true, false);
	name_values['i'] = kdm.cpl_id();

	return KDMWithMetadataPtr(new DCPKDMWithMetadata(name_values, cinema, kdm));
}