/*
    Copyright (C) 2020-2022 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 "lib/audio_content.h"
#include "lib/config.h"
#include "lib/constants.h"
#include "lib/content.h"
#include "lib/content_factory.h"
#include "lib/cross.h"
#include "lib/film.h"
#include "lib/font.h"
#include "lib/hints.h"
#include "lib/text_content.h"
#include "test.h"
#include <boost/test/unit_test.hpp>


using std::make_shared;
using std::shared_ptr;
using std::string;
using std::vector;
using boost::optional;


vector<string> current_hints;


static
void
collect_hint (string hint)
{
	current_hints.push_back (hint);
}


static
vector<string>
get_hints (shared_ptr<Film> film)
{
	current_hints.clear ();
	Hints hints (film);
	/* None of our tests need the audio analysis, and it is quite time-consuming */
	hints.disable_audio_analysis ();
	hints.Hint.connect (collect_hint);
	hints.start ();
	hints.join ();
	while (signal_manager->ui_idle()) {}
	hints.rethrow();
	return current_hints;
}


static
void
check (TextType type, string name, optional<string> expected_hint = optional<string>())
{
	auto film = new_test_film(name);
	auto content = content_factory("test/data/" + name + ".srt")[0];
	content->text.front()->set_type (type);
	content->text.front()->set_language (dcp::LanguageTag("en-US"));
	film->examine_and_add_content (content);
	BOOST_REQUIRE (!wait_for_jobs());
	auto hints = get_hints (film);

	if (expected_hint) {
		BOOST_REQUIRE_EQUAL (hints.size(), 1U);
		BOOST_CHECK_EQUAL (hints[0], *expected_hint);
	} else {
		string message;
		for (auto hint: hints) {
			message += hint + "\n";
		}
		BOOST_CHECK_MESSAGE(hints.empty(), "Found: " << message);
	}
}


BOOST_AUTO_TEST_CASE (hint_closed_caption_too_long)
{
	check (
		TextType::CLOSED_CAPTION,
		"hint_closed_caption_too_long",
		String::compose("At least one of your closed caption lines has more than %1 characters.  It is advisable to make each line %1 characters at most in length.", MAX_CLOSED_CAPTION_LENGTH, MAX_CLOSED_CAPTION_LENGTH)
	      );
}


BOOST_AUTO_TEST_CASE (hint_many_closed_caption_lines)
{
	check (
		TextType::CLOSED_CAPTION,
		"hint_many_closed_caption_lines",
		String::compose("Some of your closed captions span more than %1 lines, so they will be truncated.", MAX_CLOSED_CAPTION_LINES)
	      );
}


BOOST_AUTO_TEST_CASE (hint_subtitle_too_early)
{
	check (
		TextType::OPEN_SUBTITLE,
		"hint_subtitle_too_early",
		string("It is advisable to put your first subtitle at least 4 seconds after the start of the DCP to make sure it is seen.")
		);
}


BOOST_AUTO_TEST_CASE (hint_short_subtitles)
{
	check (
		TextType::OPEN_SUBTITLE,
		"hint_short_subtitles",
		string("At least one of your subtitles lasts less than 15 frames.  It is advisable to make each subtitle at least 15 frames long.")
		);
}


BOOST_AUTO_TEST_CASE (hint_subtitles_too_close)
{
	check (
		TextType::OPEN_SUBTITLE,
		"hint_subtitles_too_close",
		string("At least one of your subtitles starts less than 2 frames after the previous one.  It is advisable to make the gap between subtitles at least 2 frames.")
	      );
}


BOOST_AUTO_TEST_CASE (hint_many_subtitle_lines)
{
	check (
		TextType::OPEN_SUBTITLE,
		"hint_many_subtitle_lines",
		string("At least one of your subtitles has more than 3 lines.  It is advisable to use no more than 3 lines.")
	      );
}


BOOST_AUTO_TEST_CASE(hint_many_subtitle_lines2)
{
	check(TextType::OPEN_SUBTITLE, "hint_many_subtitle_lines2");
}


BOOST_AUTO_TEST_CASE (hint_subtitle_too_long)
{
	check (
		TextType::OPEN_SUBTITLE,
		"hint_subtitle_too_long",
		string("At least one of your subtitle lines has more than 52 characters.  It is recommended to make each line 52 characters at most in length.")
	      );
}


BOOST_AUTO_TEST_CASE (hint_subtitle_much_too_long)
{
	check (
		TextType::OPEN_SUBTITLE,
		"hint_subtitle_much_too_long",
		string("At least one of your subtitle lines has more than 79 characters.  You should make each line 79 characters at most in length.")
	      );
}


BOOST_AUTO_TEST_CASE (hint_subtitle_mxf_too_big)
{
	string const name = "hint_subtitle_mxf_too_big";

	auto film = new_test_film(name);

	for (int i = 0; i < 4; ++i) {
		dcp::File fake_font("build/test/hint_subtitle_mxf_too_big.ttf", "w");
		for (int i = 0; i < 512; ++i) {
			std::vector<uint8_t> rubbish(65536);
			fake_font.write(rubbish.data(), 1, rubbish.size());
		}
		fake_font.close();

		auto content = content_factory(String::compose("test/data/%1%2.xml", name, i))[0];
		content->text[0]->set_type(TextType::OPEN_SUBTITLE);
		content->text[0]->set_language(dcp::LanguageTag("en-US"));
		film->examine_and_add_content(content);
		BOOST_REQUIRE (!wait_for_jobs());
		auto const font = content->text[0]->get_font(String::compose("font_%1", i));
		BOOST_REQUIRE(font);
		font->set_file("build/test/hint_subtitle_mxf_too_big.ttf");
	}

	auto hints = get_hints (film);

	BOOST_REQUIRE_EQUAL (hints.size(), 1U);
	BOOST_CHECK_EQUAL (
		hints[0],
		"At least one of your subtitle files is larger than " MAX_TEXT_MXF_SIZE_TEXT " in total.  "
		"You should divide the DCP into shorter reels."
		);
}


BOOST_AUTO_TEST_CASE (hint_closed_caption_xml_too_big)
{
	string const name = "hint_closed_caption_xml_too_big";

	auto film = new_test_film(name);

	dcp::File ccap(String::compose("build/test/%1.srt", name), "w");
	BOOST_REQUIRE (ccap);
	for (int i = 0; i < 2048; ++i) {
		fprintf(ccap.get(), "%d\n", i + 1);
		int second = i * 2;
		int minute = second % 60;
		fprintf(ccap.get(), "00:%02d:%02d,000 --> 00:%02d:%02d,000\n", minute, second, minute, second + 1);
		fprintf(ccap.get(), "Here are some closed captions.\n\n");
	}
	ccap.close();

	auto content = content_factory("build/test/" + name + ".srt")[0];
	content->text.front()->set_type (TextType::CLOSED_CAPTION);
	content->text.front()->set_language (dcp::LanguageTag("en-US"));
	film->examine_and_add_content (content);
	BOOST_REQUIRE (!wait_for_jobs());
	auto hints = get_hints (film);

	BOOST_REQUIRE_EQUAL (hints.size(), 1U);
	BOOST_CHECK_EQUAL (
		hints[0],
		"At least one of your closed caption files' XML part is larger than " MAX_CLOSED_CAPTION_XML_SIZE_TEXT ".  "
		"You should divide the DCP into shorter reels."
		);
}


BOOST_AUTO_TEST_CASE (hints_destroyed_while_running)
{
	auto film = new_test_film("hints_destroyed_while_running");
	auto content = content_factory(TestPaths::private_data() / "boon_telly.mkv")[0];
	film->examine_and_add_content (content);
	BOOST_REQUIRE (!wait_for_jobs());

	auto hints = make_shared<Hints>(film);
	hints->start ();
	dcpomatic_sleep_seconds (1);
	hints.reset ();
	dcpomatic_sleep_seconds (1);
}


BOOST_AUTO_TEST_CASE (hints_audio_with_no_language)
{
	auto content = content_factory("test/data/sine_440.wav")[0];
	auto film = new_test_film("hints_audio_with_no_language", { content });
	content->audio->set_gain (-6);

	auto hints = get_hints (film);
	BOOST_REQUIRE_EQUAL (hints.size(), 1U);
	BOOST_CHECK_EQUAL (
		hints[0],
		"Some of your content has audio but you have not set the audio language.  It is advisable to set the audio language "
		"in the \"DCP\" tab unless your audio has no spoken parts."
		);
}


BOOST_AUTO_TEST_CASE (hints_certificate_validity)
{
	ConfigRestorer cr;

	Config::instance()->set_signer_chain(make_shared<dcp::CertificateChain>(openssl_path(), 40 * 365));

	auto film = new_test_film("hints_certificate_validity");
	auto hints = get_hints (film);
	BOOST_REQUIRE_EQUAL (hints.size(), 1U);
	BOOST_CHECK_EQUAL (
		hints[0],
		"The certificate chain that DCP-o-matic uses for signing DCPs and KDMs has a validity period "
		"that is too long.  This will cause problems playing back DCPs on some systems. "
		"It is advisable to re-create the signing certificate chain by clicking the "
		"\"Re-make certificates and key...\" button in the Keys page of Preferences."
		);
}


BOOST_AUTO_TEST_CASE(hints_mpeg2)
{
	auto film = new_test_film("hints_certificate_validity");
	film->set_video_encoding(VideoEncoding::MPEG2);
	auto hints = get_hints(film);
	BOOST_REQUIRE_EQUAL(hints.size(), 1U);
	BOOST_CHECK_EQUAL(
		hints[0],
		"The vast majority of cinemas in Europe, Australasia and North America expect DCPs "
		"encoded with JPEG2000 rather than MPEG2.  Make sure that your cinema really wants an old-style MPEG2 DCP."
		);
}