/**

 * vim: set ts=4 :

 * ===============================================================

 * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.

 * ===============================================================

 *

 * This file is not open source and may not be copied without explicit

 * written permission of AlliedModders LLC.  This file may not be redistributed 

 * in whole or significant part.

 * For information, see LICENSE.txt or http://www.sourcemod.net/license.php

 *

 * Version: $Id$

 */

#include <stdlib.h>

#include "ExtensionSys.h"

#include "LibrarySys.h"

#include "ShareSys.h"

#include "Logger.h"

#include "sourcemm_api.h"

#include "sm_srvcmds.h"

#include "sm_stringutil.h"

CExtensionManager g_Extensions;

IdentityType_t g_ExtType;

CExtension::CExtension(const char *filename, char *error, size_t maxlength)

{

	m_File.assign(filename);

	m_pAPI = NULL;

	m_pIdentToken = NULL;

	m_PlId = 0;

	unload_code = 0;

	m_FullyLoaded = false;

	char path[PLATFORM_MAX_PATH];

	g_SourceMod.BuildPath(Path_SM, path, PLATFORM_MAX_PATH, "extensions/%s", filename);

	m_Path.assign(path);

	m_pLib = NULL;

}

bool CExtension::Load(char *error, size_t maxlength)

{

	m_pLib = g_LibSys.OpenLibrary(m_Path.c_str(), error, maxlength);

	if (m_pLib == NULL)

	{

		return false;

	}

	typedef IExtensionInterface *(*GETAPI)();

	GETAPI pfnGetAPI = NULL;

	if ((pfnGetAPI=(GETAPI)m_pLib->GetSymbolAddress("GetSMExtAPI")) == NULL)

	{

		m_pLib->CloseLibrary();

		m_pLib = NULL;

		snprintf(error, maxlength, "Unable to find extension entry point");

		return false;

	}

	m_pAPI = pfnGetAPI();

	if (!m_pAPI || m_pAPI->GetExtensionVersion() > SMINTERFACE_EXTENSIONAPI_VERSION)

	{

		m_pLib->CloseLibrary();

		m_pLib = NULL;

		snprintf(error, maxlength, "Extension version is too new to load (%d, max is %d)", m_pAPI->GetExtensionVersion(), SMINTERFACE_EXTENSIONAPI_VERSION);

		return false;

	}

	if (m_pAPI->IsMetamodExtension())

	{

		bool already;

		m_PlId = g_pMMPlugins->Load(m_Path.c_str(), g_PLID, already, error, maxlength);

	}

	m_pIdentToken = g_ShareSys.CreateIdentity(g_ExtType, this);

	if (!m_pAPI->OnExtensionLoad(this, &g_ShareSys, error, maxlength, !g_SourceMod.IsMapLoading()))

	{

		if (m_pAPI->IsMetamodExtension())

		{

			if (m_PlId)

			{

				char dummy[255];

				g_pMMPlugins->Unload(m_PlId, true, dummy, sizeof(dummy));

				m_PlId = 0;

			}

		}

		m_pAPI = NULL;

		m_pLib->CloseLibrary();

		m_pLib = NULL;

		g_ShareSys.DestroyIdentity(m_pIdentToken);

		m_pIdentToken = NULL;

		return false;

	} else {

		/* Check if we're past load time */

		if (!g_SourceMod.IsMapLoading())

		{

			m_pAPI->OnExtensionsAllLoaded();

		}

	}

	return true;

}

CExtension::~CExtension()

{

	if (m_pAPI)

	{

		m_pAPI->OnExtensionUnload();

		if (m_PlId)

		{

			g_pMMPlugins->Unload(m_PlId, true, NULL, 0);

		}

	}

	if (m_pIdentToken)

	{

		g_ShareSys.DestroyIdentity(m_pIdentToken);

	}

	if (m_pLib)

	{

		m_pLib->CloseLibrary();

	}

}

void CExtension::MarkAllLoaded()

{

	assert(m_FullyLoaded == false);

	if (!m_FullyLoaded)

	{

		m_FullyLoaded = true;

		m_pAPI->OnExtensionsAllLoaded();

	}

}

void CExtension::AddPlugin(IPlugin *pPlugin)

{

	/* Unfortunately we have to do this :( */

	if (m_Plugins.find(pPlugin) != m_Plugins.end())

	{

		m_Plugins.push_back(pPlugin);

	}

}

void CExtension::RemovePlugin(IPlugin *pPlugin)

{

	m_Plugins.remove(pPlugin);

	List<WeakNative>::iterator iter = m_WeakNatives.begin();

	while (iter != m_WeakNatives.end())

	{

		if ((*iter).pl == pPlugin)

		{

			iter = m_WeakNatives.erase(iter);

		} else {

			iter++;

		}

	}

}

void CExtension::SetError(const char *error)

{

	m_Error.assign(error);

}

IExtensionInterface *CExtension::GetAPI()

{

	return m_pAPI;

}

const char *CExtension::GetFilename()

{

	return m_File.c_str();

}

IdentityToken_t *CExtension::GetIdentity()

{

	return m_pIdentToken;

}

bool CExtension::IsLoaded()

{

	return (m_pLib != NULL);

}

void CExtension::AddDependency(IfaceInfo *pInfo)

{

	m_Deps.push_back(*pInfo);

}

bool operator ==(const IfaceInfo &i1, const IfaceInfo &i2)

{

	return (i1.iface == i2.iface) && (i1.owner == i2.owner);

}

void CExtension::AddChildDependent(CExtension *pOther, SMInterface *iface)

{

	IfaceInfo info;

	info.iface = iface;

	info.owner = pOther;

	List<IfaceInfo>::iterator iter;

	for (iter = m_ChildDeps.begin();

		 iter != m_ChildDeps.end();

		 iter++)

	{

		IfaceInfo &other = (*iter);

		if (other == info)

		{

			return;

		}

	}

	m_ChildDeps.push_back(info);

}

ITERATOR *CExtension::FindFirstDependency(IExtension **pOwner, SMInterface **pInterface)

{

	List<IfaceInfo>::iterator iter = m_Deps.begin();

	if (iter == m_Deps.end())

	{

		return NULL;

	}

	if (pOwner)

	{

		*pOwner = (*iter).owner;

	}

	if (pInterface)

	{

		*pInterface = (*iter).iface;

	}

	List<IfaceInfo>::iterator *pIter = new List<IfaceInfo>::iterator(iter);

	return (ITERATOR *)pIter;

}

bool CExtension::FindNextDependency(ITERATOR *iter, IExtension **pOwner, SMInterface **pInterface)

{

	List<IfaceInfo>::iterator *pIter = (List<IfaceInfo>::iterator *)iter;

	List<IfaceInfo>::iterator _iter;

	if (_iter == m_Deps.end())

	{

		return false;

	}

	_iter++;

	if (pOwner)

	{

		*pOwner = (*_iter).owner;

	}

	if (pInterface)

	{

		*pInterface = (*_iter).iface;

	}

	*pIter = _iter;

	if (_iter == m_Deps.end())

	{

		return false;

	}

	return true;

}

void CExtension::FreeDependencyIterator(ITERATOR *iter)

{

	List<IfaceInfo>::iterator *pIter = (List<IfaceInfo>::iterator *)iter;

	delete pIter;

}

void CExtension::AddInterface(SMInterface *pInterface)

{

	m_Interfaces.push_back(pInterface);

}

bool CExtension::IsRunning(char *error, size_t maxlength)

{

	if (!IsLoaded())

	{

		if (error)

		{

			snprintf(error, maxlength, "%s", m_Error.c_str());

		}

		return false;

	}

	return m_pAPI->QueryRunning(error, maxlength);

}

/*********************

 * EXTENSION MANAGER *

 *********************/

void CExtensionManager::OnSourceModAllInitialized()

{

	g_ExtType = g_ShareSys.CreateIdentType("EXTENSION");

	g_PluginSys.AddPluginsListener(this);

	g_RootMenu.AddRootConsoleCommand("exts", "Manage extensions", this);

	g_ShareSys.AddInterface(NULL, this);

}

void CExtensionManager::OnSourceModShutdown()

{

	g_RootMenu.RemoveRootConsoleCommand("exts", this);

	g_ShareSys.DestroyIdentType(g_ExtType);

}

void CExtensionManager::Shutdown()

{

	List<CExtension *>::iterator iter;

	while ((iter = m_Libs.begin()) != m_Libs.end())

	{

		UnloadExtension((*iter));

	}

}

void CExtensionManager::TryAutoload()

{

	char path[PLATFORM_MAX_PATH];

	g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "extensions");

	IDirectory *pDir = g_LibSys.OpenDirectory(path);

	if (!pDir)

	{

		return;

	}

	const char *lfile;

	size_t len;

	while (pDir->MoreFiles())

	{

		if (pDir->IsEntryDirectory())

		{

			pDir->NextEntry();

			continue;

		}

		lfile = pDir->GetEntryName();

		len = strlen(lfile);

		if (len <= 9) /* size of ".autoload" */

		{

			pDir->NextEntry();

			continue;

		}

		if (strcmp(&lfile[len - 9], ".autoload") != 0)

		{

			pDir->NextEntry();

			continue;

		}

		char file[PLATFORM_MAX_PATH];

		len = UTIL_Format(file, sizeof(file), "%s", lfile);

		strcpy(&file[len - 9], ".ext." PLATFORM_LIB_EXT);

		LoadAutoExtension(file);

		pDir->NextEntry();

	}

}

IExtension *CExtensionManager::LoadAutoExtension(const char *path)

{

	if (!strstr(path, "." PLATFORM_LIB_EXT))

	{

		char newpath[PLATFORM_MAX_PATH];

		snprintf(newpath, PLATFORM_MAX_PATH, "%s.%s", path, PLATFORM_LIB_EXT);

		return LoadAutoExtension(newpath);

	}

	IExtension *pAlready;

	if ((pAlready=FindExtensionByFile(path)) != NULL)

	{

		return pAlready;

	}

	char error[256];

	CExtension *p = new CExtension(path, error, sizeof(error));

	/* We put us in the list beforehand so extensions that check for each other

	 * won't recursively load each other.

	 */

	m_Libs.push_back(p);

	if (!p->Load(error, sizeof(error)) || !p->IsLoaded())

	{

		g_Logger.LogError("[SM] Unable to load extension \"%s\": %s", path, error);

		p->SetError(error);

	}

	return p;

}

IExtension *CExtensionManager::FindExtensionByFile(const char *file)

{

	List<CExtension *>::iterator iter;

	CExtension *pExt;

	if (!strstr(file, "." PLATFORM_LIB_EXT))

	{

		char path[PLATFORM_MAX_PATH];

		snprintf(path, PLATFORM_MAX_PATH, "%s.%s", file, PLATFORM_LIB_EXT);

		return FindExtensionByFile(path);

	}

	/* Make sure the file direction is right */

	char path[PLATFORM_MAX_PATH];

	g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);

	for (iter=m_Libs.begin(); iter!=m_Libs.end(); iter++)

	{

		pExt = (*iter);

		if (strcmp(pExt->GetFilename(), path) == 0)

		{

			return pExt;

		}

	}

	return NULL;

}

IExtension *CExtensionManager::FindExtensionByName(const char *ext)

{

	List<CExtension *>::iterator iter;

	CExtension *pExt;

	IExtensionInterface *pAPI;

	const char *name;

	for (iter=m_Libs.begin(); iter!=m_Libs.end(); iter++)

	{

		pExt = (*iter);

		if (!pExt->IsLoaded())

		{

			continue;

		}

		if ((pAPI = pExt->GetAPI()) == NULL)

		{

			continue;

		}

		name = pAPI->GetExtensionName();

		if (!name)

		{

			continue;

		}

		if (strcmp(name, ext) == 0)

		{

			return pExt;

		}

	}

	return NULL;

}

IExtension *CExtensionManager::LoadExtension(const char *file, ExtensionLifetime lifetime, char *error, size_t maxlength)

{

	IExtension *pAlready;

	if ((pAlready=FindExtensionByFile(file)) != NULL)

	{

		return pAlready;

	}

	CExtension *pExt = new CExtension(file, error, maxlength);

	/* :NOTE: lifetime is currently ignored */

	if (!pExt->Load(error, maxlength) || !pExt->IsLoaded())

	{

		delete pExt;

		return NULL;

	}

	/* :TODO: do QueryRunning check if the map is loaded */

	m_Libs.push_back(pExt);

	return pExt;

}

void CExtensionManager::BindDependency(IExtension *pRequester, IfaceInfo *pInfo)

{

	CExtension *pExt = (CExtension *)pRequester;

	CExtension *pOwner = (CExtension *)pInfo->owner;

	pExt->AddDependency(pInfo);

	IExtensionInterface *pAPI = pExt->GetAPI();

	if (pAPI && !pAPI->QueryInterfaceDrop(pInfo->iface))

	{

		pOwner->AddChildDependent(pExt, pInfo->iface);

	}

}

void CExtensionManager::AddInterface(IExtension *pOwner, SMInterface *pInterface)

{

	CExtension *pExt = (CExtension *)pOwner;

	pExt->AddInterface(pInterface);

}

void CExtensionManager::BindChildPlugin(IExtension *pParent, IPlugin *pPlugin)

{

	CExtension *pExt = (CExtension *)pParent;

	pExt->AddPlugin(pPlugin);

}

void CExtensionManager::OnPluginDestroyed(IPlugin *plugin)

{

	List<CExtension *>::iterator iter;

	for (iter=m_Libs.begin(); iter!=m_Libs.end(); iter++)

	{

		(*iter)->RemovePlugin(plugin);

	}

}

CExtension *CExtensionManager::FindByOrder(unsigned int num)

{

	if (num < 1 || num > m_Libs.size())

	{

		return NULL;

	}

	List<CExtension *>::iterator iter = m_Libs.begin();

	while (iter != m_Libs.end())

	{

		if (--num == 0)

		{

			return (*iter);

		}

		iter++;

	}

	return NULL;

}

void CExtensionManager::BindAllNativesToPlugin(IPlugin *pPlugin)

{

	List<CExtension *>::iterator iter;

	CExtension *pExt;

	IPluginContext *pContext = pPlugin->GetBaseContext();

	for (iter=m_Libs.begin(); iter!=m_Libs.end(); iter++)

	{

		pExt = (*iter);

		if (!pExt->IsLoaded())

		{

			continue;

		}

		if (!pExt->m_Natives.size())

		{

			continue;

		}

		bool set = false;

		List<const sp_nativeinfo_t *>::iterator n_iter;

		const sp_nativeinfo_t *natives;

		for (n_iter = pExt->m_Natives.begin();

			 n_iter != pExt->m_Natives.end();

			 n_iter++)

		{

			natives = (*n_iter);

			uint32_t idx;

			unsigned int i=0;

			while (natives[i].func != NULL && natives[i].name != NULL)

			{

				if (pContext->BindNative(&natives[i]) == SP_ERROR_NONE)

				{

					pContext->FindNativeByName(natives[i].name, &idx);

					if (!(pContext->GetContext()->natives[idx].flags & SP_NTVFLAG_OPTIONAL))

					{

						set = true;

					} else {

						WeakNative wkn = WeakNative((CPlugin *)pPlugin, idx);

						pExt->m_WeakNatives.push_back(wkn);

					}

				}

				i++;

			}

		}