// -*- c++ -*-
// Distributed under the BSD 2-Clause License.
// See accompanying file LICENSE for details.
#include "tasks.h"

#include <filesystem>
#include <deque>
#include <list>
#include <iostream>
#include <algorithm>
#include <span>

#include "ctor.h"
#include "task.h"
#include "task_cc.h"
#include "task_ld.h"
#include "task_ar.h"
#include "task_so.h"
#include "task_fn.h"
#include "rebuild.h"
#include "configure.h"
#include "util.h"
#include "tools.h"

const std::deque<Target>& getTargets(const ctor::settings& settings,
                                     bool resolve_externals)
{
	auto config_files = std::span(configFiles).subspan(0, numConfigFiles);

	static bool initialised{false};
	static std::deque<Target> targets;
	if(!initialised)
	{
		const auto& externals = ctor::get_configuration().externals;
		for(const auto& config_file : config_files)
		{
			std::string path =
				std::filesystem::path(config_file.file).parent_path().string();
			if(settings.verbose > 1)
			{
				std::cout << config_file.file << " in path " << path << "\n";
			}
			auto configs = config_file.cb(settings);
			for(auto& config : configs)
			{
				if(resolve_externals)
				{
					// Resolv config externals
					for(const auto& external : config.externals)
					{
						if(externals.find(external) == externals.end())
						{
							std::cout << "External '" << external <<
								"' not found in cache - run configure.\n";
							exit(1);
						}
						const auto& flags = externals.at(external);
						append(config.flags.cflags, flags.cflags);
						append(config.flags.cxxflags, flags.cxxflags);
						append(config.flags.ldflags, flags.ldflags);
						append(config.flags.arflags, flags.arflags);
						append(config.flags.asmflags, flags.asmflags);
						//append(config.libs.insert(config.libs libs);
					}
				}

				targets.push_back({config, path});
			}
		}
		initialised = true;
	}

	return targets;
}

std::vector<std::shared_ptr<Task>> taskFactory(const ctor::build_configuration& config,
                                               const ctor::settings& settings,
                                               const std::string& sourceDir,
                                               bool is_self)
{
	std::vector<std::shared_ptr<Task>> tasks;

	std::filesystem::path targetFile(config.target);

	ctor::target_type target_type{config.type};
	if(target_type == ctor::target_type::automatic)
	{
		if(config.function != nullptr)
		{
			target_type = ctor::target_type::function;
		}
		else
		{
			target_type = target_type_from_extension(ctor::toolchain::any, targetFile);
		}
	}

	const auto& c = ctor::get_configuration();
	std::vector<std::string> objects;
	if(target_type != ctor::target_type::function)
	{
		for(const auto& source : config.sources)
		{
			if(source.toolchain == ctor::toolchain::any ||
			   (config.system == ctor::output_system::build && source.toolchain == c.build_toolchain) ||
			   (config.system == ctor::output_system::host && source.toolchain == c.host_toolchain))
			{
				auto task = std::make_shared<TaskCC>(config, settings, sourceDir, source);
				tasks.push_back(task);
				objects.push_back(task->targetFile().string());
			}
		}
	}
#ifndef BOOTSTRAP
	else
	{
		for(const auto& file : config.sources)
		{
			auto task = std::make_shared<TaskFn>(config, settings, sourceDir, file);
			tasks.push_back(task);
			objects.push_back(task->target());
		}
	}
#endif

	switch(target_type)
	{
	case ctor::target_type::automatic:
		// The target_type cannot be Auto
		break;

	case ctor::target_type::unknown:
		std::cerr << "Could not deduce target type from target " <<
			targetFile.string() << " please specify.\n";
		exit(1);
		break;

	case ctor::target_type::static_library:
	case ctor::target_type::unit_test_library:
		tasks.push_back(std::make_shared<TaskAR>(config, settings, config.target,
		                                         objects, sourceDir));
		break;
#ifndef BOOTSTRAP
	case ctor::target_type::dynamic_library:
		// TODO: Use C++20 starts_with
		if(targetFile.stem().string().substr(0, 3) != "lib")
		{
			std::cerr << "Dynamic library target must have 'lib' prefix\n";
			exit(1);
		}
		tasks.push_back(std::make_shared<TaskSO>(config, settings, config.target,
		                                         objects, sourceDir));
		break;

	case ctor::target_type::executable:
	case ctor::target_type::unit_test:
		tasks.push_back(std::make_shared<TaskLD>(config, settings, config.target,
		                                         objects, sourceDir, is_self));
		break;

	case ctor::target_type::object:
	case ctor::target_type::function:
		break;
#else
	default:
		break;
#endif
	}

	return tasks;
}

std::shared_ptr<Task> getNextTask([[maybe_unused]]const std::vector<std::shared_ptr<Task>>& allTasks,
                                  std::vector<std::shared_ptr<Task>>& dirtyTasks)
{
	for(auto dirtyTask = dirtyTasks.begin();
	    dirtyTask != dirtyTasks.end();
	    ++dirtyTask)
	{
		auto task = *dirtyTask;
		//std::cout << "Examining target " << (*dirtyTask)->target() << "\n";
		if(task->ready())
		{
			dirtyTasks.erase(dirtyTask);
			return task;
		}
	}

	//std::cout << "No task ready ... \n";
	return nullptr;
}

std::vector<std::shared_ptr<Task>> getTasks(const ctor::settings& settings,
                                            const std::vector<std::string> names,
                                            bool resolve_externals)
{
	auto& targets = getTargets(settings, resolve_externals);
	std::vector<std::shared_ptr<Task>> tasks;
	for(const auto& target : targets)
	{
		if(names.empty() ||
		   std::find(std::begin(names), std::end(names), target.config.target) != std::end(names))
		{
			std::vector<std::string> objects;
			auto t = taskFactory(target.config, settings, target.path, false);
			tasks.insert(tasks.end(), t.begin(), t.end());
		}
	}

	return tasks;
}