#pragma once

#include <map>

#include <openvic-dataloader/v2script/AbstractSyntaxTree.hpp>

#include "openvic/types/Colour.hpp"
#include "openvic/types/Return.hpp"
#include "openvic/types/Date.hpp"
#include "openvic/types/fixed_point/FP.hpp"

namespace OpenVic {
	namespace ast = ovdl::v2script::ast;

	namespace NodeTools {

		template<typename T>
		return_t expect_type(ast::NodeCPtr node, std::function<return_t(T const&)> callback);

		return_t expect_identifier(ast::NodeCPtr node, std::function<return_t(std::string_view)> callback);
		return_t expect_string(ast::NodeCPtr node, std::function<return_t(std::string_view)> callback);
		return_t expect_identifier_or_string(ast::NodeCPtr node, std::function<return_t(std::string_view)> callback);
		return_t expect_bool(ast::NodeCPtr node, std::function<return_t(bool)> callback);
		return_t expect_int(ast::NodeCPtr node, std::function<return_t(int64_t)> callback);
		return_t expect_uint(ast::NodeCPtr node, std::function<return_t(uint64_t)> callback);
		return_t expect_fixed_point(ast::NodeCPtr node, std::function<return_t(FP)> callback);
		return_t expect_colour(ast::NodeCPtr node, std::function<return_t(colour_t)> callback);
		return_t expect_date(ast::NodeCPtr node, std::function<return_t(Date)> callback);
		return_t expect_assign(ast::NodeCPtr node, std::function<return_t(std::string_view, ast::NodeCPtr)> callback);
		return_t expect_list(ast::NodeCPtr node, std::function<return_t(ast::NodeCPtr)> callback, size_t length = 0, bool file_node = false);
		return_t expect_dictionary(ast::NodeCPtr node, std::function<return_t(std::string_view, ast::NodeCPtr)> callback, bool file_node = false);

		static const std::function<return_t(ast::NodeCPtr)> success_callback = [](ast::NodeCPtr) -> return_t { return SUCCESS; };

		struct dictionary_entry_t {
			bool must_appear, can_repeat;
			std::function<return_t(ast::NodeCPtr)> callback;
		};
		using dictionary_key_map_t = std::map<std::string, dictionary_entry_t, std::less<void>>;
		return_t expect_dictionary_keys(ast::NodeCPtr node, dictionary_key_map_t const& keys, bool allow_other_keys = false, bool file_node = false);
	}
}