// Cyphesis Online RPG Server and AI Engine
// Copyright (C) 2005 Alistair Riddoch
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA


#ifdef NDEBUG
#undef NDEBUG
#endif
#ifndef DEBUG
#define DEBUG
#endif

#include "../TestBase.h"
#include "../TestWorld.h"

#include "rules/simulation/CorePropertyManager.h"
#include "server/EntityBuilder.h"
#include "server/EntityFactory_impl.h"
#include "server/ArchetypeFactory.h"

#include "rules/simulation/LocatedEntity.h"
#include "rules/Script.h"
#include "rules/simulation/Task.h"

#include "common/id.h"
#include "rules/simulation/Inheritance.h"
#include "common/log.h"
#include "common/Monitors.h"
#include "common/Property_impl.h"
#include "common/TypeNode_impl.h"
#include "common/Variable.h"
#include "common/custom.h"


#include "rules/Location_impl.h"



#include <Atlas/Objects/Anonymous.h>
#include <Atlas/Objects/RootOperation.h>

#include <cstdlib>

#include <cassert>
#include "rules/simulation/AtlasProperties.h"
#include "rules/PhysicalProperties.h"

using Atlas::Message::Element;
using Atlas::Message::ListType;
using Atlas::Message::MapType;
using Atlas::Objects::Entity::Anonymous;
using Atlas::Objects::Root;
namespace {
Atlas::Objects::Factories factories;
}
class EntityBuildertest : public Cyphesis::TestBase {
private:
	Ref<LocatedEntity> e;
	BaseWorld* test_world;
	EntityBuilder* eb;
public:
	EntityBuildertest();

	void setup();

	void teardown();

	void test_sequence1();

	void test_sequence2();

	void test_sequence3();

	void test_sequence4();

	void test_sequence5();

	void test_installFactory_duplicate();

	Inheritance* inheritance;
};

EntityBuildertest::EntityBuildertest() {
	ADD_TEST(EntityBuildertest::test_sequence1);
	ADD_TEST(EntityBuildertest::test_sequence2);
	ADD_TEST(EntityBuildertest::test_sequence3);
	ADD_TEST(EntityBuildertest::test_sequence4);
	ADD_TEST(EntityBuildertest::test_sequence5);
	ADD_TEST(EntityBuildertest::test_installFactory_duplicate);
}

void EntityBuildertest::setup() {
	inheritance = new Inheritance();
	e = new LocatedEntity(1);
	test_world = new TestWorld(e);
	eb = new EntityBuilder();

	eb->installBaseFactory("thing", "game_entity", std::make_unique<EntityFactory<LocatedEntity>>());
}

void EntityBuildertest::teardown() {
	delete test_world;
	e = nullptr;
	delete eb;
	delete inheritance;
}

enum action {
	DO_NOTHING,
	SET_POS,
	SET_VELOCITY
} LocatedEntity_merge_action = DO_NOTHING;

void EntityBuildertest::test_sequence1() {
	Anonymous attributes;

	assert(eb->newEntity(1, "world", attributes) == nullptr);
	assert(eb->newEntity(1, "nonexistant", attributes) == nullptr);
	assert(eb->newEntity(1, "thing", attributes) != nullptr);
}

void EntityBuildertest::test_sequence2() {
	Anonymous attributes;

	// Create a normal LocatedEntity
	auto test_ent = eb->newEntity(1, "thing", attributes);
	assert(test_ent);

	// Create an entity specifying an attrbute
	attributes->setAttr("funky", "true");

	test_ent = eb->newEntity(1, "thing", attributes);
	assert(test_ent);

	// Create an entity causing VELOCITY to be set
	attributes = Anonymous();

	attributes->setVelocity(std::vector<double>(3, 1.5));

	LocatedEntity_merge_action = SET_VELOCITY;

	test_ent = eb->newEntity(1, "thing", attributes);
	assert(test_ent);

	LocatedEntity_merge_action = DO_NOTHING;

	// Create an entity causing VELOCITY to be set for no obvious reason
	attributes = Anonymous();

	LocatedEntity_merge_action = SET_VELOCITY;

	test_ent = eb->newEntity(1, "thing", attributes);
	assert(test_ent);

	LocatedEntity_merge_action = DO_NOTHING;

	// Create an entity specifying a LOC
	attributes = Anonymous();

	attributes->setLoc("1");

	test_ent = eb->newEntity(1, "thing", attributes);
	assert(test_ent);
}

void EntityBuildertest::test_sequence3() {
	// Attributes for test entities being created
	Anonymous attributes;

	// Create an entity which is an instance of one of the core classes
	auto test_ent = eb->newEntity(1, "thing", attributes);
	assert(test_ent);
}

void EntityBuildertest::test_sequence4() {
	TypeNode<LocatedEntity> factoryType("custom_type");
	// Attributes for test entities being created
	Anonymous attributes;

	// Check that creating an entity of a type we know we have not yet
	// installed results in a null pointer.
	assert(eb->newEntity(1, "custom_type", attributes) == nullptr);

	// Get a reference to the internal dictionary of entity factories.
	const auto& factory_dict = eb->getFactories();

	// Make sure it has some factories in it already.
	assert(!factory_dict.empty());

	// Assert the dictionary does not contain the factory we know we have
	// have not yet installed.
	assert(factory_dict.find("custom_type") == factory_dict.end());

	// Set up a type description for a new type, and install it
	EntityFactoryBase* custom_type_factory = new EntityFactory<LocatedEntity>();
	custom_type_factory->m_attributes["test_custom_type_attr"] =
			"test_value";
	{
		int ret;
		ret = eb->installFactory("custom_type",
								 atlasClass("custom_type", "thing"),
								 std::unique_ptr<EntityFactoryBase>(custom_type_factory));
		custom_type_factory->m_type = &factoryType;

		ASSERT_EQUAL(ret, 0);
	}

	PropertyBase* p = new Property<std::string, LocatedEntity>;
	p->set("test_value");
	custom_type_factory->m_type->injectProperty("test_custom_type_attr", std::unique_ptr<PropertyBase>(p));

	// Check that the factory dictionary now contains a factory for
	// the custom type we just installed.
	auto I = factory_dict.find("custom_type");
	assert(I != factory_dict.end());
	assert(custom_type_factory == I->second.get());

	MapType::const_iterator J;
	// Check the factory has the attributes we described on the custom
	// type.
	J = custom_type_factory->m_attributes.find("test_custom_type_attr");
	assert(J != custom_type_factory->m_attributes.end());
	assert(J->second.isString());
	assert(J->second.String() == "test_value");

	// Create an instance of our custom type, ensuring that it works.
	auto test_ent = eb->newEntity(1, "custom_type", attributes);
	assert(test_ent);

	assert(test_ent->getType() == custom_type_factory->m_type);

	// Check that creating an entity of a type we know we have not yet
	// installed results in a null pointer.
	assert(eb->newEntity(1, "custom_inherited_type", attributes) == nullptr);

	// Assert the dictionary does not contain the factory we know we have
	// have not yet installed.
	assert(factory_dict.find("custom_inherited_type") == factory_dict.end());
}

void EntityBuildertest::test_sequence5() {
	TypeNode<LocatedEntity> factoryType("custom_scripted_type");
	Anonymous attributes;

	// Get a reference to the internal dictionary of entity factories.
	const auto& factory_dict = eb->getFactories();

	// Make sure it has some factories in it already.
	assert(!factory_dict.empty());

	// Assert the dictionary does not contain the factory we know we have
	// have not yet installed.
	assert(factory_dict.find("custom_scripted_type") == factory_dict.end());

	// Set up a type description for a new type, and install it
	EntityFactoryBase* custom_type_factory = new EntityFactory<LocatedEntity>();
	custom_type_factory->m_attributes["test_custom_type_attr"] =
			"test_value";

	{
		eb->installFactory("custom_scripted_type",
						   atlasClass("custom_scripted_type", "thing"),
						   std::unique_ptr<EntityFactoryBase>(custom_type_factory));
		custom_type_factory->m_type = &factoryType;
	}

	// Check that the factory dictionary now contains a factory for
	// the custom type we just installed.
	auto I = factory_dict.find("custom_scripted_type");
	assert(I != factory_dict.end());
	assert(custom_type_factory == I->second.get());

	// Create an instance of our custom type, ensuring that it works.
	auto test_ent = eb->newEntity(1, "custom_scripted_type", attributes);
	assert(test_ent);

	assert(test_ent->getType() == custom_type_factory->m_type);
	test_ent->destroy();
}

void EntityBuildertest::test_installFactory_duplicate() {
	auto& factory_dict = eb->getFactories();

	EntityFactoryBase* custom_type_factory = new EntityFactory<LocatedEntity>();

	int ret = eb->installFactory("custom_type",
								 atlasClass("custom_type", "thing"),
								 std::unique_ptr<EntityFactoryBase>(custom_type_factory));

	ASSERT_EQUAL(ret, 0);
	ASSERT_TRUE(factory_dict.find("custom_type") != factory_dict.end());
	ASSERT_EQUAL(factory_dict.find("custom_type")->second.get(), custom_type_factory);

	EntityFactoryBase* custom_type_factory2 = new EntityFactory<LocatedEntity>();

	ret = eb->installFactory("custom_type",
							 atlasClass("custom_type", "thing"),
							 std::unique_ptr<EntityFactoryBase>(custom_type_factory2));

	ASSERT_EQUAL(ret, -1);
	ASSERT_TRUE(factory_dict.find("custom_type") != factory_dict.end());
	ASSERT_EQUAL(factory_dict.find("custom_type")->second.get(), custom_type_factory);
}

int main(int argc, char** argv) {
	Monitors m;
	EntityBuildertest t;

	return t.run();
}

// stubs





class World;

template<>
Ref<LocatedEntity> EntityFactory<World>::newEntity(RouterId id,
											const Atlas::Objects::Entity::RootEntity& attributes) {
	return 0;
}

class LocatedEntity;

class Character;


template
class EntityFactory<LocatedEntity>;

void LocatedEntity::makeContainer() {
	if (m_contains == 0) {
		m_contains.reset(new LocatedEntitySet);
	}
}


void LocatedEntity::merge(const MapType& ent) {
	switch (LocatedEntity_merge_action) {
		case SET_POS:
			this->requirePropertyClassFixed<PositionProperty<LocatedEntity>>().data().setValid();
			break;
		case SET_VELOCITY:
			this->requirePropertyClassFixed<VelocityProperty<LocatedEntity>>().data().setValid();
			break;
		case DO_NOTHING:
		default:
			break;
	};

}


Inheritance::Inheritance(Atlas::Objects::Factories& factories) : noClass(0), m_factories(factories) {
	Atlas::Objects::Entity::Anonymous root_desc;

	root_desc->setObjtype("meta");
	root_desc->setId("root");

	auto root = new TypeNode<LocatedEntity>("root", root_desc);

	atlasObjects["root"].reset(root);
	installStandardObjects(*this);
}


const TypeNode<LocatedEntity>* Inheritance::getType(const std::string& typeName) const {
	auto I = atlasObjects.find(typeName);
	if (I == atlasObjects.end()) {
		return 0;
	}
	return I->second.get();
}


TypeNode<LocatedEntity>* Inheritance::addChild(const Atlas::Objects::Root& obj) {
	const std::string& child = obj->getId();
	const std::string& parent = obj->getParent();
	if (atlasObjects.find(child) != atlasObjects.end()) {
		std::cerr << fmt::format("Installing type \"{}\"(\"{}\") "
								 "which was already installed",
								 child, parent) << std::endl;
		return 0;
	}
	auto I = atlasObjects.find(parent);
	if (I == atlasObjects.end()) {
		std::cerr << fmt::format("Installing type \"{}\" "
								 "which has unknown parent \"{}\".",
								 child, parent) << std::endl;
		return 0;
	}
	Element children(ListType(1, child));
	if (I->second->description(Visibility::PRIVATE)->copyAttr("children", children) == 0) {
		assert(children.isList());
		children.asList().push_back(child);
	}
	I->second->description(Visibility::PRIVATE)->setAttr("children", children);

	auto type = new TypeNode<LocatedEntity>(child, obj);
	type->setParent(I->second.get());

	atlasObjects[child].reset(type);

	return type;
}


void installStandardObjects(TypeStore<LocatedEntity>& i) {

	i.addChild(atlasClass("root_entity", "root"));
	i.addChild(atlasClass("admin_entity", "root_entity"));
	i.addChild(atlasClass("account", "admin_entity"));
	i.addChild(atlasClass("player", "account"));
	i.addChild(atlasClass("admin", "account"));
	i.addChild(atlasClass("game", "admin_entity"));
	i.addChild(atlasClass("game_entity", "root_entity"));
}


Ref<LocatedEntity> BaseWorld::getEntity(const std::string& id) const {
	return getEntity(integerId(id));
}

Ref<LocatedEntity> BaseWorld::getEntity(long id) const {
	auto I = m_eobjects.find(id);
	if (I != m_eobjects.end()) {
		assert(I->second);
		return I->second;
	} else {
		return nullptr;
	}
}


Root atlasClass(const std::string& name, const std::string& parent) {
	Atlas::Objects::Entity::Anonymous r;
	r->setParent(parent);
	r->setObjtype("class");
	r->setId(name);
	return r;
}

sigc::signal<void()> python_reload_scripts;
