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

#include <iostream>
#include <filesystem>
#include <algorithm>
#include <source_location>

#include "configure.h"
#include "settings.h"
#include "libctor.h"
#include "tasks.h"
#include "build.h"
#include "execute.h"

std::array<BuildConfigurationEntry, 1024> configFiles;
std::size_t numConfigFiles{0};

int reg(BuildConfigurations (*cb)(),
        const std::source_location location)
{
	// NOTE: std::cout cannot be used here
	if(numConfigFiles >= configFiles.size())
	{
		fprintf(stderr, "Max %d build configurations currently supported.\n",
		        (int)configFiles.size());
		exit(1);
	}

	configFiles[numConfigFiles].file = location.file_name();
	configFiles[numConfigFiles].cb = cb;
	++numConfigFiles;

	return 0;
}

int reg(const char* location)
{
	// NOTE: std::cout cannot be used here
	if(numConfigFiles >= configFiles.size())
	{
		fprintf(stderr, "Max %d build configurations currently supported.\n",
		        (int)configFiles.size());
		exit(1);
	}

	configFiles[numConfigFiles].file = location;
	configFiles[numConfigFiles].cb =
		[](){ return std::vector<BuildConfiguration>{}; };
	++numConfigFiles;

	return 0;
}

int unreg(const char* location)
{
	std::size_t found{0};
	for(std::size_t i = 0; i < numConfigFiles;)
	{
		if(std::string(location) == configFiles[i].file)
		{
			++found;
			for(std::size_t j = i; j < numConfigFiles; ++j)
			{
				configFiles[j] = configFiles[j + 1];
			}
			--numConfigFiles;
		}
		else
		{
			++i;
		}
	}

	for(std::size_t i = 0; i < numExternalConfigFiles;)
	{
		if(std::string(location) == externalConfigFiles[i].file)
		{
			++found;
			for(std::size_t j = i; j < numExternalConfigFiles; ++j)
			{
				externalConfigFiles[j] = externalConfigFiles[j + 1];
			}
			--numExternalConfigFiles;
		}
		else
		{
			++i;
		}
	}

	return found;
}

std::array<ExternalConfigurationEntry, 1024> externalConfigFiles;
std::size_t numExternalConfigFiles{0};

int reg(ExternalConfigurations (*cb)(),
        const std::source_location location)
{
	// NOTE: std::cout cannot be used here
	if(numExternalConfigFiles >= externalConfigFiles.size())
	{
		fprintf(stderr, "Max %d external configurations currently supported.\n",
		        (int)externalConfigFiles.size());
		exit(1);
	}

	externalConfigFiles[numExternalConfigFiles].file = location.file_name();
	externalConfigFiles[numExternalConfigFiles].cb = cb;
	++numExternalConfigFiles;

	return 0;
}

namespace
{
bool contains(const std::vector<Source>& sources, const std::string& file)
{
	for(const auto& source : sources)
	{
		if(source.file == file)
		{
			return true;
		}
	}

	return false;
}
}

bool recompileCheck(const Settings& global_settings, int argc, char* argv[],
                    bool relaunch_allowed)
{
	using namespace std::string_literals;

	if(global_settings.verbose > 1)
	{
		std::cout << "Recompile check (" << numConfigFiles << "):\n";
	}

	BuildConfiguration config;

	config.name = "ctor";
	config.flags.cxxflags =
		std::vector<std::string>({ "-s", "-O3", "-std=c++20" });
	if(hasConfiguration(cfg::ctor_includedir))
	{
		config.flags.cxxflags.push_back("-I"s + getConfiguration(cfg::ctor_includedir));
	}
	if(hasConfiguration(cfg::ctor_libdir))
	{
		config.flags.ldflags.push_back("-L"s + getConfiguration(cfg::ctor_libdir));
	}
	config.flags.ldflags.push_back("-lctor");
	config.flags.ldflags.push_back("-pthread");


	Settings settings{global_settings};
	settings.verbose = -1; // Make check completely silent.
	settings.builddir += "/ctor"; // override builddir to use ctor subdir

	{
		std::filesystem::path buildfile  = settings.builddir;
		std::filesystem::path currentfile = argv[0];
		config.target = std::filesystem::relative(currentfile, buildfile).string();
	}

	if(std::filesystem::exists(configurationFile))
	{
		config.sources.push_back(configurationFile.string());
	}

	for(std::size_t i = 0; i < numConfigFiles; ++i)
	{
		std::string location = configFiles[i].file;
		if(global_settings.verbose > 1)
		{
			std::cout << " - " << location << "\n";
		}

		// Ensure that files containing multiple configurations are only added once.
		if(!contains(config.sources, location))
		{
			config.sources.push_back(location);
		}
	}

	for(std::size_t i = 0; i < numExternalConfigFiles; ++i)
	{
		std::string location = externalConfigFiles[i].file;
		if(global_settings.verbose > 1)
		{
			std::cout << " - " << location << "\n";
		}

		// Ensure that files containing multiple configurations are only added once.
		if(!contains(config.sources, location))
		{
			config.sources.push_back(location);
		}
	}

	auto tasks = taskFactory({config}, settings, {});

	for(auto task : tasks)
	{
		if(task->registerDepTasks(tasks))
		{
			return false;
		}
	}

	// Find out if reconfigure is needed
	bool reconfigure{false};
	for(auto task : tasks)
	{
		if(task->dirty() &&
		   task->source() != "" && // only look at source files
		   task->source() != "configuration.cc") // don't reconfigure if only configuration.cc is changed.
		{
			reconfigure |= true;
		}
	}

	auto dirty_tasks = build(settings, "ctor", tasks, true); // dryrun
	if(dirty_tasks)
	{
		std::cout << "Rebuilding config.\n";
		build(settings, "ctor", tasks); // run for real
	}

	if(reconfigure)
	{
		std::vector<std::string> args;
		args.push_back("reconfigure");
		if(!relaunch_allowed)
		{
			args.push_back("--no-rerun");
		}
		for(int i = 1; i < argc; ++i)
		{
			args.push_back(argv[i]);
		}
		auto ret = execute(argv[0], args);
		//if(ret != 0)
		{
			exit(ret);
		}

	}

	return dirty_tasks;
}