#pragma once

#include <stdint.h>
#include <string>
#include <vector>
#include <memory>
#include <type_traits>

#include "Tile.h"
#include "Item.h"
#include "WorldGen.h"
#include "Entity.h"
#include "Collection.h"
#include "OS.h"
#include "util.h"

namespace Swan {

class ModWrapper;

class Mod {
public:
	Mod(std::string name): name_(std::move(name)) {}
	virtual ~Mod() = default;

	void registerTile(Tile::Builder tile) { tiles_.push_back(tile); }
	void registerItem(Item::Builder item) { items_.push_back(item); }
	void registerSprite(std::string sprite) { sprites_.push_back(sprite); }

	template<typename WG>
	void registerWorldGen(std::string name) {
		worldGens_.push_back(WorldGen::Factory{
			.name = name,
			.create = [](World &world) -> std::unique_ptr<WorldGen> {
				return std::make_unique<WG>(world);
			}
		});
	}

	template<typename Ent>
	void registerEntity(const std::string &name) {
		static_assert(
			std::is_move_constructible_v<Ent>,
			"Entities must be movable");
		entities_.push_back(EntityCollection::Factory{
			.name = name,
			.create = [](std::string name) -> std::unique_ptr<EntityCollection> {
				return std::make_unique<EntityCollectionImpl<Ent>>(std::move(name));
			}
		});
	}

private:
	const std::string name_;
	std::vector<Tile::Builder> tiles_;
	std::vector<Item::Builder> items_;
	std::vector<std::string> sprites_;
	std::vector<WorldGen::Factory> worldGens_;
	std::vector<EntityCollection::Factory> entities_;

	friend ModWrapper;
};

class ModWrapper {
public:
	ModWrapper(std::unique_ptr<Mod> mod, std::string path, OS::Dynlib lib):
		mod_(std::move(mod)), path_(std::move(path)), dynlib_(std::move(lib)) {}

	ModWrapper(ModWrapper &&other) noexcept = default;

	~ModWrapper() {
		// Mod::~Mod will destroy stuff allocated by the dynlib,
		// so we must run its destructor before deleting the dynlib
		mod_.reset();
	}

	const std::string &name() { return mod_->name_; }
	const std::vector<Tile::Builder> &tiles() { return mod_->tiles_; }
	const std::vector<Item::Builder> &items() { return mod_->items_; }
	const std::vector<std::string> &sprites() { return mod_->sprites_; }
	const std::vector<WorldGen::Factory> &worldGens() { return mod_->worldGens_; }
	const std::vector<EntityCollection::Factory> &entities() { return mod_->entities_; }

	std::unique_ptr<Mod> mod_;
	std::string path_;
	OS::Dynlib dynlib_;
};

}