#pragma once

#include <concepts>
#include <cstdio>
#include <string_view>
#include <type_traits>
#include <utility>

#include <openvic-dataloader/NodeLocation.hpp>
#include <openvic-dataloader/detail/SymbolIntern.hpp>
#include <openvic-dataloader/detail/Utility.hpp>

#include <lexy/lexeme.hpp>

#include <dryad/node.hpp>
#include <dryad/node_map.hpp>
#include <dryad/symbol.hpp>
#include <dryad/tree.hpp>

#include <fmt/core.h>

#include <detail/InternalConcepts.hpp>

namespace ovdl {
	struct AbstractSyntaxTree : SymbolIntern {
		AbstractSyntaxTree() = default;
		explicit AbstractSyntaxTree(std::size_t max_elements) : _symbol_interner(max_elements) {}

		symbol_type intern(const char* str, std::size_t length);
		symbol_type intern(std::string_view str);
		const char* intern_cstr(const char* str, std::size_t length);
		const char* intern_cstr(std::string_view str);
		symbol_interner_type& symbol_interner();
		const symbol_interner_type& symbol_interner() const;

		template<typename Reader>
		symbol_type intern(lexy::lexeme<Reader> lexeme) {
			return intern(lexeme.begin(), lexeme.size());
		}
		template<typename Reader>
		const char* intern_cstr(lexy::lexeme<Reader> lexeme) {
			return intern_cstr(lexeme.begin(), lexeme.size());
		}

	protected:
		symbol_interner_type _symbol_interner;
	};

	template<detail::IsFile FileT, std::derived_from<typename FileT::node_type> RootNodeT>
	struct BasicAbstractSyntaxTree : AbstractSyntaxTree {
		using file_type = FileT;
		using root_node_type = RootNodeT;
		using node_type = typename file_type::node_type;

		BasicAbstractSyntaxTree() = default;

		explicit BasicAbstractSyntaxTree(file_type&& file)
			: AbstractSyntaxTree(file.size() * file.visit_buffer([](auto&& buffer) -> size_t { return sizeof(typename std::decay_t<decltype(buffer)>::char_type); })),
			  _file { std::move(file) } {}

		template<typename Encoding, typename MemoryResource = void>
		explicit BasicAbstractSyntaxTree(lexy::buffer<Encoding, MemoryResource>&& buffer)
			: AbstractSyntaxTree(buffer.size() * sizeof(Encoding::char_type)),
			  _file { std::move(buffer) } {}

		void set_location(const node_type* n, NodeLocation loc) {
			_file.set_location(n, loc);
		}

		NodeLocation location_of(const node_type* n) const {
			return _file.location_of(n);
		}

		root_node_type* root() {
			return _tree.root();
		}

		const root_node_type* root() const {
			return _tree.root();
		}

		file_type& file() {
			return _file;
		}

		const file_type& file() const {
			return _file;
		}

		template<typename T, typename... Args>
		T* create(NodeLocation loc, Args&&... args) {
			auto node = _tree.template create<T>(DRYAD_FWD(args)...);
			set_location(node, loc);
			return node;
		}

		template<typename T, typename... Args>
		T* create(const char* begin, const char* end, Args&&... args) {
			return create<T>(NodeLocation::make_from(begin, end), DRYAD_FWD(args)...);
		}

		void set_root(root_node_type* node) {
			_tree.set_root(node);
		}

	protected:
		dryad::tree<root_node_type> _tree;
		file_type _file;
	};
}