// Copyright 2008-2018 Yolo Technologies, Inc. All Rights Reserved. https://www.comblockengine.com

#include "baseapp.h"
#include "entity_autoloader.h"
#include "initprogress_handler.h"
#include "network/bundle.h"
#include "network/channel.h"
#include "entitydef/entitydef.h"
#include "server/serverconfig.h"

#include "../../server/dbmgr/dbmgr_interface.h"

namespace KBEngine{	

#define LOAD_ENTITY_SIZE 32

//-------------------------------------------------------------------------------------
EntityAutoLoader::EntityAutoLoader(Network::NetworkInterface & networkInterface, InitProgressHandler* pInitProgressHandler):
networkInterface_(networkInterface),
pInitProgressHandler_(pInitProgressHandler),
entityTypes_(),
start_(0),
end_(0),
querying_(false)
{
	ENGINE_COMPONENT_INFO& dbcfg = g_kbeSrvConfig.getDBMgr();

	std::vector<DBInterfaceInfo>::iterator dbinfo_iter = dbcfg.dbInterfaceInfos.begin();
	for (; dbinfo_iter != dbcfg.dbInterfaceInfos.end(); ++dbinfo_iter)
	{
		entityTypes_.push_back(std::vector<ENTITY_SCRIPT_UID>());

		const EntityDef::SCRIPT_MODULES& modules = EntityDef::getScriptModules();
		EntityDef::SCRIPT_MODULES::const_iterator iter = modules.begin();
		for (; iter != modules.end(); ++iter)
		{
			entityTypes_[entityTypes_.size() - 1].push_back((*iter)->getUType());
		}
	}
}

//-------------------------------------------------------------------------------------
EntityAutoLoader::~EntityAutoLoader()
{
	DEBUG_MSG("EntityAutoLoader::~EntityAutoLoader()\n");

	if(pInitProgressHandler_)
		pInitProgressHandler_->setAutoLoadState(1);
}

//-------------------------------------------------------------------------------------
void EntityAutoLoader::onEntityAutoLoadCBFromDBMgr(Network::Channel* pChannel, MemoryStream& s)
{
	uint16 dbInterfaceIndex = 0;
	s >> dbInterfaceIndex;

	int size = 0;
	s >> size;

	if(size >= LOAD_ENTITY_SIZE)
	{
		start_ = end_;
		end_ += LOAD_ENTITY_SIZE;
	}
	else
	{
		start_ = 0;
		end_ = 0;

		(*entityTypes_.begin()).erase((*entityTypes_.begin()).begin());
	}

	querying_ = false;

	ENTITY_SCRIPT_UID entityType;
	s >> entityType;

	for(int i=0; i<size; ++i)
	{
		DBID dbid;
		s >> dbid;

		if(PyObject_HasAttrString(Baseapp::getSingleton().getEntryScript().get(), "onAutoLoadEntityCreate") > 0)
		{
			PyObject* pyResult = PyObject_CallMethod(Baseapp::getSingleton().getEntryScript().get(), 
												const_cast<char*>("onAutoLoadEntityCreate"), 
												const_cast<char*>("sK"), 
												EntityDef::findScriptModule(entityType)->getName(),
												dbid);

			if(pyResult != NULL) {
				AsyncioHelper::submitCoroutine(pyResult);
				Py_DECREF(pyResult);
			}
			else
			{
				SCRIPT_ERROR_CHECK();

				if(pInitProgressHandler_)
					pInitProgressHandler_->setError();
			}
		}
		else
		{
			Baseapp::getSingleton().createEntityAnywhereFromDBID(EntityDef::findScriptModule(entityType)->getName(), dbid, NULL, 
				g_kbeSrvConfig.dbInterfaceIndex2dbInterfaceName(dbInterfaceIndex));
		}
	}
}

//-------------------------------------------------------------------------------------
bool EntityAutoLoader::process()
{
	Network::Channel* pChannel = Components::getSingleton().getDbmgrChannel();
	if(pChannel == NULL || querying_)
		return true;

	if(entityTypes_.size() > 0)
	{
		if ((*entityTypes_.begin()).size() > 0)
		{
			Network::Bundle* pBundle = Network::Bundle::createPoolObject(OBJECTPOOL_POINT);

			if (start_ == 0 && end_ == 0)
				end_ = LOAD_ENTITY_SIZE;

			uint16 dbInterfaceIndex = (uint16)(g_kbeSrvConfig.getDBMgr().dbInterfaceInfos.size() - entityTypes_.size());

			if (!g_kbeSrvConfig.isPureDBInterfaceName(g_kbeSrvConfig.dbInterfaceIndex2dbInterfaceName(dbInterfaceIndex)))
			{
				(*pBundle).newMessage(DbmgrInterface::entityAutoLoad);
				(*pBundle) << dbInterfaceIndex << g_componentID << (*(*entityTypes_.begin()).begin()) << start_ << end_;
				pChannel->send(pBundle);
				querying_ = true;
			}
			else
			{
				start_ = 0;
				end_ = 0;
				(*entityTypes_.begin()).erase((*entityTypes_.begin()).begin());
			}
		}
		else
		{
			entityTypes_.erase(entityTypes_.begin());
		}

		return true;
	}

	delete this;
	return false;
}

//-------------------------------------------------------------------------------------

}