#include <csignal>
#include <memory>
#include <filesystem>
#include <pugixml.hpp>
#include <map>

#include "backend/Location.h"
#include "backend/LocationFactory.h"
#include "backend/Dungeon.h"

#include "load_dungeon.h"
#include "Exception.h"
#include "GameData.h"
#include "backend/Object.h"
#include "strings.h"

using namespace std;
using namespace pugi;

unique_ptr<Dungeon> load_dungeon(const string & filename) {
	unique_ptr<Dungeon> dungeon = make_unique<Dungeon>();

	xml_document doc;

	string canonical = filename;
	if (canonical.starts_with("~/"))
		canonical = getenv("HOME") + canonical.substr(1);
	try {
		canonical = filesystem::canonical(canonical);
	} catch (...) {
		throw Exception("Kon bestand niet vinden");
	}

	xml_parse_result result = doc.load_file(canonical.c_str());
	if (!result)
		throw Exception("Kon XML-bestand niet lezen");

	xml_node locations = doc.child("locaties");
	if (!locations)
		throw Exception("XML-bestand mist een <locaties> tag");

	GameData & gamedata = GameData::get_instance();

	struct TempData {
		Location * location;
		unsigned edges[4];
	};
	map<unsigned, TempData> temp_map;

	for (xml_node & tag : locations) {
		const char * name = tag.text().as_string();
		const char * description = tag.child("beschrijving").text().as_string();

		Location * location = LocationFactory::create_location(name, description);

		vector<string> objects_hidden = str_split(tag.attribute("objectenverborgen").as_string(), ";");
		for (string & name : objects_hidden) {
			Object * object = gamedata.create_object(name);
			object->set_hidden(true);
			location->add_object(object);
		}
		vector<string> objects_visible = str_split(tag.attribute("objectenzichtbaar").as_string(), ";");
		for (string & name : objects_visible) {
			Object * object = gamedata.create_object(name);
			object->set_hidden(false);
			location->add_object(object);
		}
	
		vector<string> enemies = str_split(tag.attribute("vijand").as_string(), ";");
		for (string & name : enemies) {
			Enemy * enemy = gamedata.create_enemy(name);
			location->add_enemy(enemy);
		}
	
		temp_map[tag.attribute("id").as_uint()] = {
			.location = location,
			.edges = {
				[Direction::NORTH] = tag.attribute("noord").as_uint(0),
				[Direction::EAST]  = tag.attribute("oost").as_uint(0),
				[Direction::SOUTH] = tag.attribute("zuid").as_uint(0),
				[Direction::WEST]  = tag.attribute("west").as_uint(0),
			},
		};
		dungeon->add_location(location);
	}

	// connect edges after creating all locations
	for (auto & [_, temp] : temp_map) {
		for (Direction direction : DIRECTIONS) {
			unsigned id = temp.edges[direction];
			if (temp.edges[direction] == 0) continue;
			if (!temp_map.contains(id)) continue;
			temp.location->set_exit(direction, temp_map[id].location);
		}
	}

	return dungeon;
}