#include <gtest/gtest.h>

#include <QDialogButtonBox>
#include <QDir>
#include <QFile>
#include <QFileInfo>
#include <QListWidget>
#include <QPushButton>
#include <QStandardPaths>
#include <QTemporaryDir>

#include "core.h"
#include "node/color/colormanager/colormanager.h"
#include "node/factory.h"
#include "node/input/time/timeinput.h"
#include "node/project.h"
#include "node/project/folder/folder.h"
#include "node/project/footage/footage.h"
#include "node/project/serializer/serializer.h"
#include "render/diskmanager.h"
#include "task/project/import/import.h"
#include "dialog/projectimport/projectimporterrordialog.h"
#include "task/project/load/load.h"
#include "undo/undocommand.h"

namespace
{

QString test_image_path()
{
	return QDir(QStringLiteral(OAK_TEST_SOURCE_DIR))
		.filePath(QStringLiteral("tests/img.png"));
}

} // namespace

class TaskProjectImportTest : public ::testing::Test {
protected:
	void SetUp() override
	{
		if (!olive::Core::instance()) {
			// Leaked intentionally: Core is process-wide (matches footage_probe_test)
			new olive::Core();
		}

		created_disk_manager_ = (olive::DiskManager::instance() == nullptr);
		if (created_disk_manager_) {
			olive::DiskManager::create_instance();
		}

		// Sandbox the footage metadata cache so real probes write into the
		// temp dir instead of the user's cache
		old_cache_home_ = qgetenv("XDG_CACHE_HOME");
		had_cache_home_ = qEnvironmentVariableIsSet("XDG_CACHE_HOME");
		qputenv("XDG_CACHE_HOME",
				QDir(temp_dir_.path()).filePath(QStringLiteral("xdg")).toUtf8());
		QDir().mkpath(
			QStandardPaths::writableLocation(QStandardPaths::CacheLocation));

		olive::ColorManager::set_up_default_config();

		project_ = std::make_unique<olive::Project>();
		project_->initialize();
	}

	void TearDown() override
	{
		project_.reset();
		if (created_disk_manager_) {
			olive::DiskManager::destroy_instance();
		}
		if (had_cache_home_) {
			qputenv("XDG_CACHE_HOME", old_cache_home_);
		} else {
			qunsetenv("XDG_CACHE_HOME");
		}
	}

	QTemporaryDir temp_dir_;
	QByteArray old_cache_home_;
	bool had_cache_home_ = false;
	bool created_disk_manager_ = false;
	std::unique_ptr<olive::Project> project_;
};

TEST_F(TaskProjectImportTest, ImportOfUnprobeableFileCollectsInvalidList)
{
	const QString path =
		QDir(temp_dir_.path()).filePath(QStringLiteral("not_media.txt"));
	{
		QFile file(path);
		ASSERT_TRUE(file.open(QFile::WriteOnly));
		file.write("this is not a media file");
	}

	olive::ProjectImportTask task(project_->root(), { path });
	EXPECT_EQ(task.get_file_count(), 1);

	double last_progress = -1.0;
	QObject::connect(&task, &olive::Task::progress_changed, &task,
					 [&last_progress](double p) { last_progress = p; });

	ASSERT_TRUE(task.start());

	EXPECT_TRUE(task.has_invalid_files());
	ASSERT_EQ(task.get_invalid_files().size(), 1);
	EXPECT_EQ(task.get_invalid_files().first(), path);
	EXPECT_TRUE(task.get_imported_footage().isEmpty());
	EXPECT_DOUBLE_EQ(last_progress, 1.0);

	// The undo command exists but contains no children since nothing was added
	olive::MultiUndoCommand *cmd = task.take_command();
	ASSERT_NE(cmd, nullptr);
	EXPECT_EQ(cmd->child_count(), 0);
	delete cmd;
}

TEST_F(TaskProjectImportTest, ImportOfImageFileAddsFootageThroughUndoCommand)
{
	const QString path = test_image_path();
	ASSERT_TRUE(QFileInfo::exists(path));

	olive::ProjectImportTask task(project_->root(), { path });
	EXPECT_EQ(task.get_file_count(), 1);

	ASSERT_TRUE(task.start());

	EXPECT_FALSE(task.has_invalid_files());
	ASSERT_EQ(task.get_imported_footage().size(), 1);

	olive::Footage *footage = task.get_imported_footage().first();
	EXPECT_EQ(footage->filename(), path);
	EXPECT_EQ(footage->get_label(), QStringLiteral("img.png"));
	EXPECT_TRUE(footage->is_valid());

	// Nothing is in the folder until the command is redone
	EXPECT_TRUE(project_->root()->children().isEmpty());

	olive::MultiUndoCommand *cmd = task.take_command();
	ASSERT_NE(cmd, nullptr);
	cmd->redo_now();

	ASSERT_EQ(project_->root()->children().size(), 1);
	EXPECT_EQ(project_->root()->children().first(),
			  static_cast<olive::Node *>(footage));
	EXPECT_TRUE(project_->nodes().contains(footage));

	cmd->undo_now();
	EXPECT_TRUE(project_->root()->children().isEmpty());

	delete cmd;
}

TEST_F(TaskProjectImportTest, ImportOfDirectoryCreatesFolderHierarchy)
{
	const QString src = test_image_path();
	ASSERT_TRUE(QFileInfo::exists(src));

	const QString dir_path =
		QDir(temp_dir_.path()).filePath(QStringLiteral("media"));
	const QString sub_path = QDir(dir_path).filePath(QStringLiteral("sub"));
	ASSERT_TRUE(QDir().mkpath(sub_path));

	const QString first = QDir(dir_path).filePath(QStringLiteral("a.png"));
	const QString second = QDir(sub_path).filePath(QStringLiteral("b.png"));
	ASSERT_TRUE(QFile::copy(src, first));
	ASSERT_TRUE(QFile::copy(src, second));

	olive::ProjectImportTask task(project_->root(), { dir_path });
	EXPECT_EQ(task.get_file_count(), 2);

	ASSERT_TRUE(task.start());
	EXPECT_FALSE(task.has_invalid_files());
	EXPECT_EQ(task.get_imported_footage().size(), 2);

	olive::MultiUndoCommand *cmd = task.take_command();
	ASSERT_NE(cmd, nullptr);
	cmd->redo_now();

	// Importing a directory creates a folder named after it under the target
	const QVector<olive::Node *> &root_children = project_->root()->children();
	ASSERT_EQ(root_children.size(), 1);
	auto *top_folder = dynamic_cast<olive::Folder *>(root_children.first());
	ASSERT_NE(top_folder, nullptr);
	EXPECT_EQ(top_folder->get_label(), QStringLiteral("media"));

	// ...which holds the first image plus a subfolder with the second image
	QVector<olive::Footage *> all_footage =
		top_folder->list_children_of_type<olive::Footage>();
	EXPECT_EQ(all_footage.size(), 2);

	QVector<olive::Folder *> sub_folders =
		top_folder->list_children_of_type<olive::Folder>();
	ASSERT_EQ(sub_folders.size(), 1);
	EXPECT_EQ(sub_folders.first()->get_label(), QStringLiteral("sub"));

	cmd->undo_now();
	EXPECT_TRUE(project_->root()->children().isEmpty());

	delete cmd;
}

TEST_F(TaskProjectImportTest, CancelledBeforeRunReturnsFalseAndDropsCommand)
{
	const QString path = test_image_path();
	ASSERT_TRUE(QFileInfo::exists(path));

	olive::ProjectImportTask task(project_->root(), { path });
	task.Cancel();

	EXPECT_FALSE(task.start());
	EXPECT_EQ(task.take_command(), nullptr);
	EXPECT_TRUE(task.get_imported_footage().isEmpty());
	EXPECT_FALSE(task.has_invalid_files());
	EXPECT_TRUE(project_->root()->children().isEmpty());
}

TEST(TaskProjectImportErrorDialog, PopulatesListWithFailedFilenames)
{
	const QStringList failed = { QStringLiteral("/tmp/first.xyz"),
								 QStringLiteral("/tmp/second.xyz"),
								 QStringLiteral("/tmp/third.xyz") };

	olive::ProjectImportErrorDialog dialog(failed);

	EXPECT_EQ(dialog.windowTitle(), QStringLiteral("Import Error"));

	auto *list = dialog.findChild<QListWidget *>();
	ASSERT_NE(list, nullptr);
	ASSERT_EQ(list->count(), failed.size());
	for (int i = 0; i < failed.size(); i++) {
		EXPECT_EQ(list->item(i)->text(), failed.at(i));
	}

	auto *buttons = dialog.findChild<QDialogButtonBox *>();
	ASSERT_NE(buttons, nullptr);
	EXPECT_NE(buttons->button(QDialogButtonBox::Ok), nullptr);
}

TEST(TaskProjectImportErrorDialog, OkButtonAcceptsDialog)
{
	olive::ProjectImportErrorDialog dialog(
		{ QStringLiteral("/tmp/first.xyz") });

	auto *buttons = dialog.findChild<QDialogButtonBox *>();
	ASSERT_NE(buttons, nullptr);

	QPushButton *ok = buttons->button(QDialogButtonBox::Ok);
	ASSERT_NE(ok, nullptr);
	ok->click();

	EXPECT_EQ(dialog.result(), QDialog::Accepted);
	EXPECT_TRUE(dialog.isHidden());
}

class TaskProjectLoadTest : public ::testing::Test {
protected:
	void SetUp() override
	{
		created_disk_manager_ = (olive::DiskManager::instance() == nullptr);
		if (created_disk_manager_) {
			olive::DiskManager::create_instance();
		}

		olive::ColorManager::set_up_default_config();
		olive::NodeFactory::initialize();
		olive::ProjectSerializer::initialize();
	}

	void TearDown() override
	{
		olive::ProjectSerializer::destroy();
		if (created_disk_manager_) {
			olive::DiskManager::destroy_instance();
		}
	}

	QString save_project_to_temp_file(olive::Project *project,
								  const QString &filename)
	{
		const QString path = QDir(temp_dir_.path()).filePath(filename);
		olive::ProjectSerializer::SaveData data(
			olive::ProjectSerializer::k_project, project, path);
		olive::ProjectSerializer::Result result =
			olive::ProjectSerializer::save(data, false);
		if (result.code() != olive::ProjectSerializer::k_success) {
			return QString();
		}
		return path;
	}

	QTemporaryDir temp_dir_;
	bool created_disk_manager_ = false;
};

TEST_F(TaskProjectLoadTest, LoadingValidProjectSucceeds)
{
	olive::Project project;
	project.initialize();

	auto *node = new olive::TimeInput();
	node->set_label(QStringLiteral("TimeInput"));
	node->setParent(&project);

	const QString path =
		save_project_to_temp_file(&project, QStringLiteral("project.ove"));
	ASSERT_FALSE(path.isEmpty());
	ASSERT_TRUE(QFileInfo::exists(path));

	olive::ProjectLoadTask task(path);
	EXPECT_EQ(task.get_filename(), path);
	EXPECT_EQ(task.get_loaded_project(), nullptr);

	ASSERT_TRUE(task.start()) << task.get_error().toStdString();

	olive::Project *loaded = task.get_loaded_project();
	ASSERT_NE(loaded, nullptr);
	// Project::set_filename() stores native separators on Windows
	EXPECT_EQ(QDir::fromNativeSeparators(loaded->filename()),
			  QDir::fromNativeSeparators(path));
	EXPECT_FALSE(loaded->nodes().isEmpty());

	delete loaded;
}

TEST_F(TaskProjectLoadTest, LoadingMissingFileFails)
{
	const QString path =
		QDir(temp_dir_.path()).filePath(QStringLiteral("missing.ove"));

	olive::ProjectLoadTask task(path);

	EXPECT_FALSE(task.start());
	EXPECT_FALSE(task.get_error().isEmpty());
	EXPECT_EQ(task.get_loaded_project(), nullptr);
}

TEST_F(TaskProjectLoadTest, LoadingCorruptFileFails)
{
	const QString path =
		QDir(temp_dir_.path()).filePath(QStringLiteral("corrupt.ove"));
	{
		QFile file(path);
		ASSERT_TRUE(file.open(QFile::WriteOnly));
		file.write("this is not a project file");
	}

	olive::ProjectLoadTask task(path);

	EXPECT_FALSE(task.start());
	EXPECT_FALSE(task.get_error().isEmpty());
	EXPECT_EQ(task.get_loaded_project(), nullptr);
}