#pragma once

#include "CodegenConfig.hpp"
#include "CodegenDecl.hpp"

#include "CodegenUtils.inl"

#include <Utils.hpp>

#include <robin_hood.h>
#include <cinttypes>
#include <string>
#include <string_view>
#include <vector>

using namespace std::literals;

class CodegenInput {
private:
	std::vector<DeclEnum> mEnums;
	robin_hood::unordered_flat_map<std::string, size_t, StringHash, StringEqual> mDeclByName;
	robin_hood::unordered_node_map<std::string, DeclNamespace, StringHash, StringEqual> mNamespaces;

public:
	void AddEnum(std::string fullname, DeclEnum decl) {
#if CODEGEN_DEBUG_PRINT
		printf("Committed enum '%s'\n", decl.name.c_str());
		for (auto& elm : decl.elements) {
			printf("  - element %s = %" PRId64 "\n", elm.name.c_str(), elm.value);
		}
#endif

		mDeclByName.try_emplace(std::move(fullname), mEnums.size());
		mEnums.push_back(std::move(decl));
	}

	DeclNamespace* AddNamespace(DeclNamespace ns) {
		auto path = Utils::MakeFullName(""sv, &ns);
		auto [iter, success] = mNamespaces.try_emplace(std::move(path), std::move(ns));
		auto& nsRef = iter->second;
		if (success) {
			nsRef.fullname = iter->first;
		}
		return &nsRef;
	}

	const DeclEnum* FindEnumByName(std::string_view name) const {
		// TODO handle multiple kinds of decl
		auto iter = mDeclByName.find(name);
		if (iter != mDeclByName.end()) {
			return &mEnums[iter->second];
		} else {
			return nullptr;
		}
	}

	const DeclNamespace* FindNamespace(std::string_view fullname) const {
		auto iter = mNamespaces.find(fullname);
		if (iter != mNamespaces.end()) {
			return &iter->second;
		} else {
			return nullptr;
		}
	}

	DeclNamespace* FindNamespace(std::string_view name) {
		return const_cast<DeclNamespace*>(const_cast<const CodegenInput*>(this)->FindNamespace(name));
	}
};