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

#include <lexy/dsl/option.hpp>
#include <lexy/encoding.hpp>
#include <lexy/input_location.hpp>

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

#include "ParseState.hpp"
#include <range/v3/view/drop.hpp>

using namespace ovdl::v2script::ast;

ListValue::ListValue(dryad::node_ctor ctor, StatementList statements)
	: node_base(ctor) {
	insert_child_list_after(nullptr, statements);
	if (statements.empty()) {
		_last_statement = nullptr;
	} else {
		_last_statement = statements.back();
	}
}

ListValue::ListValue(dryad::node_ctor ctor, AssignStatementList statements) : node_base(ctor) {
	insert_child_list_after(nullptr, statements);
	if (statements.empty()) {
		_last_statement = nullptr;
	} else {
		_last_statement = statements.back();
	}
}

ListValue::ListValue(dryad::node_ctor ctor) : node_base(ctor) {
	_last_statement = nullptr;
}

FileTree::FileTree(dryad::node_ctor ctor, StatementList statements) : node_base(ctor) {
	insert_child_list_after(nullptr, statements);
	if (statements.empty()) {
		_last_node = nullptr;
	} else {
		_last_node = statements.back();
	}
}

FileTree::FileTree(dryad::node_ctor ctor, AssignStatementList statements) : node_base(ctor) {
	insert_child_list_after(nullptr, statements);
	if (statements.empty()) {
		_last_node = nullptr;
	} else {
		_last_node = statements.back();
	}
}

FileTree::FileTree(dryad::node_ctor ctor) : node_base(ctor) {
	_last_node = nullptr;
}

std::string FileAbstractSyntaxTree::make_list_visualizer() const {
	const int INDENT_SIZE = 2;

	std::string result;
	unsigned int level = 0;

	for (auto [event, node] : dryad::traverse(this->_tree)) {
		if (event == dryad::traverse_event::exit) {
			--level;
			continue;
		}

		result.append(INDENT_SIZE * level, ' ');
		result.append(fmt::format("- {}: ", get_kind_name(node->kind())));

		dryad::visit_node(
			node,
			[&](const FlatValue* value) {
				result.append(value->value().c_str());
			},
			[&](const ListValue* value) {
			},
			[&](const NullValue* value) {
			},
			[&](const EventStatement* statement) {
				result.append(statement->is_province_event() ? "province_event" : "country_event");
			},
			[&](const AssignStatement* statement) {
			},
			[&](const FileTree* tree) {
			});

		result.append(1, '\n');

		if (event == dryad::traverse_event::enter)
			++level;
	}

	return result;
}

std::string FileAbstractSyntaxTree::make_native_visualizer() const {
	constexpr int INDENT_SIZE = 2;

	std::string result;
	unsigned int level = 0;

	dryad::visit_tree(
		this->_tree,
		[&](const IdentifierValue* value) {
			result.append(value->value().c_str());
		},
		[&](const StringValue* value) {
			result.append(1, '"').append(value->value().c_str()).append(1, '"');
		},
		[&](dryad::child_visitor<NodeKind> visitor, const ValueStatement* statement) {
			visitor(statement->value());
		},
		[&](dryad::child_visitor<NodeKind> visitor, const AssignStatement* statement) {
			visitor(statement->left());
			if (statement->right()->kind() != NodeKind::NullValue) {
				result.append(" = ");
				visitor(statement->right());
			}
		},
		[&](dryad::child_visitor<NodeKind> visitor, const ListValue* value) {
			result.append(1, '{');
			level++;
			for (const auto& statement : value->statements()) {
				result
					.append(1, '\n')
					.append(INDENT_SIZE * level, ' ');
				visitor(statement);
			}
			level--;
			result
				.append(1, '\n')
				.append(INDENT_SIZE * level, ' ')
				.append(1, '}');
		},
		[&](dryad::child_visitor<NodeKind> visitor, const EventStatement* statement) {
			result.append(statement->is_province_event() ? "province_event" : "country_event");
			if (statement->right()->kind() != NodeKind::NullValue) {
				result.append(" = ");
				visitor(statement->right());
			}
		},
		[&](dryad::child_visitor<NodeKind> visitor, const FileTree* value) {
			auto statements = value->statements();
			visitor(*statements.begin());

			for (const auto& statement : statements | ranges::views::drop(1)) {
				result
					.append(1, '\n')
					.append(INDENT_SIZE * level, ' ');
				visitor(statement);
			}
		});

	return result;
}