1 files changed, 42 insertions, 0 deletions
diff --git a/src/openvic/dataloader/NodeTools.hpp b/src/openvic/dataloader/NodeTools.hpp
new file mode 100644
index 0000000..ca7130c
--- /dev/null
+++ b/src/openvic/dataloader/NodeTools.hpp
@@ -0,0 +1,42 @@
+#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);
+	}
+}