#include "pch.h"
#include "modmanager.h"
#include "scriptsrson.h"
#include "filesystem.h"
#include "squirrel.h"

#include <fstream>

void ModManager::BuildScriptsRson()
{
	spdlog::info("Building custom scripts.rson");
	fs::path MOD_SCRIPTS_RSON_PATH = fs::path(GetCompiledAssetsPath() / MOD_SCRIPTS_RSON_SUFFIX);
	fs::remove(MOD_SCRIPTS_RSON_PATH);

	std::string scriptsRson = R2::ReadVPKOriginalFile(VPK_SCRIPTS_RSON_PATH);
	scriptsRson += "\n\n// START MODDED SCRIPT CONTENT\n\n"; // newline before we start custom stuff

	for (Mod& mod : m_loadedMods)
	{
		if (!mod.Enabled)
			continue;

		// this isn't needed at all, just nice to have imo
		scriptsRson += "// MOD: ";
		scriptsRson += mod.Name;
		scriptsRson += ":\n\n";

		for (ModScript& script : mod.Scripts)
		{
			/* should create something with this format for each script
			When: "CONTEXT"
			Scripts:
			[
				_coolscript.gnut
			]*/

			scriptsRson += "When: \"";
			scriptsRson += script.RsonRunOn;
			scriptsRson += "\"\n";

			scriptsRson += "Scripts:\n[\n\t";
			scriptsRson += script.Path;
			scriptsRson += "\n]\n\n";
		}
	}

	fs::create_directories(MOD_SCRIPTS_RSON_PATH.parent_path());

	std::ofstream writeStream(MOD_SCRIPTS_RSON_PATH, std::ios::binary);
	writeStream << scriptsRson;
	writeStream.close();

	ModOverrideFile overrideFile;
	overrideFile.owningMod = nullptr;
	overrideFile.path = VPK_SCRIPTS_RSON_PATH;

	if (m_modFiles.find(VPK_SCRIPTS_RSON_PATH) == m_modFiles.end())
		m_modFiles.insert(std::make_pair(VPK_SCRIPTS_RSON_PATH, overrideFile));
	else
		m_modFiles[VPK_SCRIPTS_RSON_PATH] = overrideFile;

	// todo: for preventing dupe scripts in scripts.rson, we could actually parse when conditions with the squirrel vm, just need a way to
	// get a result out of squirrelmanager.ExecuteCode this would probably be the best way to do this, imo
}