#pragma once

#include "CodegenDecl.hpp"
#include "CodegenMacros.hpp"

#include <Utils.hpp>

#include <robin_hood.h>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <string>
#include <vector>

// A generic "thing" (could be anything, comments, string-concated functionsm, etc.) to spit into the output file
struct CodegenOutputThing {
	std::string text;
};

class CodegenOutput {
private:
	robin_hood::unordered_set<std::string, StringHash, StringEqual> mRequestIncludes;
	std::vector<CodegenOutputThing> mOutThings;
	std::vector<DeclStruct> mOutStructs;
	std::vector<DeclEnum> mOutEnums;
	std::vector<DeclFunction> mOutFunctions;

public:
	std::string optionOutPrefix;
	// Whether to add prefixes mOutPrefix to all global names or not
	bool optionAutoAddPrefix : 1 = false;

public:
	void AddRequestInclude(std::string_view include) {
		if (!mRequestIncludes.contains(include)) {
			mRequestIncludes.insert(std::string(include));
		}
	}

	void AddOutputThing(CodegenOutputThing thing) {
		mOutThings.push_back(std::move(thing));
	}

	void MergeContents(CodegenOutput other) {
		std::move(other.mOutThings.begin(), other.mOutThings.end(), std::back_inserter(this->mOutThings));
		std::move(other.mOutStructs.begin(), other.mOutStructs.end(), std::back_inserter(this->mOutStructs));
		std::move(other.mOutEnums.begin(), other.mOutEnums.end(), std::back_inserter(this->mOutEnums));
		std::move(other.mOutFunctions.begin(), other.mOutFunctions.end(), std::back_inserter(this->mOutFunctions));
	}

	void Write(FILE* file) const {
		for (auto& include : mRequestIncludes) {
			// TODO how to resolve to the correct include paths?
			WRITE_FMT_LN(file, "#include <%s>", include.c_str());
		}

		for (auto& thing : mOutThings) {
			fwrite(thing.text.c_str(), sizeof(char), thing.text.size(), file);
			WRITE_LIT(file, "\n");
		}

		for (auto& declStruct : mOutStructs) {
			WRITE_FMT_LN(file, "struct %s {", declStruct.name.c_str());
			// TODO
			WRITE_LIT_LN(file, "};");
		}

		for (auto& declEnum : mOutEnums) {
			// TODO
		}

		for (auto& declFunc : mOutFunctions) {
			// TODO
		}
	}
};