#include <stdio.h>

#include "PluginSys.h"

#include "ShareSys.h"

#include "LibrarySys.h"

#include "HandleSys.h"

#include "sourcemm_api.h"

#include "sourcemod.h"

#include "CTextParsers.h"

#include "CLogger.h"

#include "ExtensionSys.h"

#include "sm_srvcmds.h"

#include "sm_stringutil.h"

CPluginManager g_PluginSys;

HandleType_t g_PluginType = 0;

IdentityType_t g_PluginIdent = 0;

CPlugin::CPlugin(const char *file)

{

	static int MySerial = 0;

	m_type = PluginType_Private;

	m_status = Plugin_Uncompiled;

	m_serial = ++MySerial;

	m_plugin = NULL;

	m_errormsg[256] = '\0';

	snprintf(m_filename, sizeof(m_filename), "%s", file);

	m_handle = 0;

	m_ident = NULL;

}

CPlugin::~CPlugin()

{

	if (m_handle)

	{

		HandleSecurity sec;

		sec.pOwner = g_PluginSys.GetIdentity();

		sec.pIdentity = sec.pOwner;

		g_HandleSys.FreeHandle(m_handle, &sec);

		g_ShareSys.DestroyIdentity(m_ident);

	}

	if (m_ctx.base)

	{

		delete m_ctx.base;

		m_ctx.base = NULL;

	}

	if (m_ctx.ctx)

	{

		m_ctx.vm->FreeContext(m_ctx.ctx);

		m_ctx.ctx = NULL;

	}

	if (m_ctx.co)

	{

		m_ctx.vm->AbortCompilation(m_ctx.co);

		m_ctx.co = NULL;

	}

	if (m_plugin)

	{

		g_pSourcePawn->FreeFromMemory(m_plugin);

		m_plugin = NULL;

	}

}

void CPlugin::InitIdentity()

{

	if (!m_handle)

	{

		m_ident = g_ShareSys.CreateIdentity(g_PluginIdent);

		m_handle = g_HandleSys.CreateHandle(g_PluginType, this, g_PluginSys.GetIdentity(), g_PluginSys.GetIdentity(), NULL);

		m_ctx.base->SetIdentity(m_ident);

	}

}

CPlugin *CPlugin::CreatePlugin(const char *file, char *error, size_t maxlength)

{

	char fullpath[PLATFORM_MAX_PATH+1];

	g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", file);

	FILE *fp = fopen(fullpath, "rb");

	CPlugin *pPlugin = new CPlugin(file);

	if (!fp)

	{

		if (error)

		{

			snprintf(error, maxlength, "Unable to open file");

		}

		pPlugin->m_status = Plugin_BadLoad;

		return pPlugin;

	}

	int err;

	sp_plugin_t *pl = g_pSourcePawn->LoadFromFilePointer(fp, &err);

	if (pl == NULL)

	{

		fclose(fp);

		if (error)

		{

			snprintf(error, maxlength, "Error %d while parsing plugin", err);

		}

		pPlugin->m_status = Plugin_BadLoad;

		return pPlugin;

	}

	fclose(fp);

	pPlugin->m_plugin = pl;

	return pPlugin;

}

ICompilation *CPlugin::StartMyCompile(IVirtualMachine *vm)

{

	if (!m_plugin)

	{

		return NULL;

	}

	/* :NOTICE: We will eventually need to change these natives

	 * for swapping in new contexts

	 */

	if (m_ctx.co || m_ctx.ctx)

	{

		return NULL;

	}

	m_status = Plugin_Uncompiled;

	m_ctx.vm = vm;

	m_ctx.co = vm->StartCompilation(m_plugin);

	return m_ctx.co;

}

void CPlugin::CancelMyCompile()

{

	if (!m_ctx.co)

	{

		return;

	}

	m_ctx.vm->AbortCompilation(m_ctx.co);

	m_ctx.co = NULL;

	m_ctx.vm = NULL;

}

bool CPlugin::FinishMyCompile(char *error, size_t maxlength)

{

	if (!m_ctx.co)

	{

		return false;

	}

	int err;

	m_ctx.ctx = m_ctx.vm->CompileToContext(m_ctx.co, &err);

	if (!m_ctx.ctx)

	{

		memset(&m_ctx, 0, sizeof(m_ctx));

		if (error)

		{

			snprintf(error, maxlength, "Failed to compile (error %d)", err);

		}

		return false;

	}

	m_ctx.base = new BaseContext(m_ctx.ctx);

	m_ctx.base->SetRunnable(false);

	m_ctx.ctx->user[SM_CONTEXTVAR_MYSELF] = (void *)this;

	m_status = Plugin_Created;

	m_ctx.co = NULL;

	UpdateInfo();

	return true;

}

void CPlugin::SetErrorState(PluginStatus status, const char *error_fmt, ...)

{

	m_status = status;

	va_list ap;

	va_start(ap, error_fmt);

	vsnprintf(m_errormsg, sizeof(m_errormsg), error_fmt, ap);

	va_end(ap);

	if (m_ctx.base)

	{

		m_ctx.base->SetRunnable(false);

	}

}

void CPlugin::UpdateInfo()

{

	/* Now grab the info */

	uint32_t idx;

	IPluginContext *base = GetBaseContext();

	int err = base->FindPubvarByName("myinfo", &idx);

	if (err == SP_ERROR_NONE)

	{

		struct sm_plugininfo_c_t

		{

			cell_t name;

			cell_t description;

			cell_t author;

			cell_t version;

			cell_t url;

		};

		sm_plugininfo_c_t *cinfo;

		cell_t local_addr;

		base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&cinfo);

		base->LocalToString(cinfo->name, (char **)&m_info.name);

		base->LocalToString(cinfo->description, (char **)&m_info.description);

		base->LocalToString(cinfo->author, (char **)&m_info.author);

		base->LocalToString(cinfo->url, (char **)&m_info.url);

		base->LocalToString(cinfo->version, (char **)&m_info.version);

	}

	m_info.author = m_info.author ? m_info.author : "";

	m_info.description = m_info.description ? m_info.description : "";

	m_info.name = m_info.name ? m_info.name : "";

	m_info.url = m_info.url ? m_info.url : "";

	m_info.version = m_info.version ? m_info.version : "";

}

void CPlugin::Call_OnPluginInit()

{

	if (m_status != Plugin_Loaded)

	{

		return;

	}

	m_status = Plugin_Running;

	cell_t result;

	IPluginFunction *pFunction = m_ctx.base->GetFunctionByName("OnPluginInit");

	if (!pFunction)

	{

		return;

	}

	pFunction->PushCell(m_handle);

	pFunction->Execute(&result);

}

void CPlugin::Call_OnPluginUnload()

{

	if (m_status < Plugin_Paused)

	{

		return;

	}

	cell_t result;

	IPluginFunction *pFunction = m_ctx.base->GetFunctionByName("OnPluginUnload");

	if (!pFunction)

	{

		return;

	}

	pFunction->Execute(&result);

}

bool CPlugin::Call_AskPluginLoad(char *error, size_t maxlength)

{

	if (m_status != Plugin_Created)

	{

		return false;

	}

	m_status = Plugin_Loaded;

	m_ctx.base->SetRunnable(true);

	int err;

	cell_t result;

	IPluginFunction *pFunction = m_ctx.base->GetFunctionByName("AskPluginLoad");

	if (!pFunction)

	{

		return true;

	}

	pFunction->PushCell(m_handle);

	pFunction->PushCell(g_PluginSys.IsLateLoadTime() ? 1 : 0);

	pFunction->PushStringEx(error, maxlength, 0, SM_PARAM_COPYBACK);

	pFunction->PushCell(maxlength);

	if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)

	{

		return false;

	}

	if (!result || m_status != Plugin_Loaded)

	{

		return false;

	}

	return true;

}

const sp_plugin_t *CPlugin::GetPluginStructure() const

{

	return m_plugin;

}

IPluginContext *CPlugin::GetBaseContext() const

{

	return m_ctx.base;

}

sp_context_t *CPlugin::GetContext() const

{

	return m_ctx.ctx;

}

const char *CPlugin::GetFilename() const

{

	return m_filename;

}

PluginType CPlugin::GetType() const

{

	return m_type;

}

const sm_plugininfo_t *CPlugin::GetPublicInfo() const

{

	return &m_info;

}

unsigned int CPlugin::GetSerial() const

{

	return m_serial;

}

PluginStatus CPlugin::GetStatus() const

{

	return m_status;

}

bool CPlugin::IsDebugging() const

{

	if (!m_ctx.ctx)

	{

		return false;

	}

	return ((m_ctx.ctx->flags & SP_FLAG_DEBUG) == SP_FLAG_DEBUG);

}

bool CPlugin::SetPauseState(bool paused)

{

	if (paused && GetStatus() != Plugin_Paused)

	{

		return false;

	} else if (!paused && GetStatus() != Plugin_Running) {

		return false;

	}

	m_status = (paused) ? Plugin_Paused : Plugin_Running;

	IPluginFunction *pFunction = m_ctx.base->GetFunctionByName("OnPluginPauseChange");

	if (pFunction)

	{

		cell_t result;

		pFunction->PushCell(paused ? 1 : 0);

		pFunction->Execute(&result);

	}

	return true;

}

IdentityToken_t *CPlugin::GetIdentity() const

{

	return m_ident;

}

bool CPlugin::ToggleDebugMode(bool debug, char *error, size_t maxlength)

{

	int err;

	if (!IsRunnable())

	{

		if (error)

		{

			snprintf(error, maxlength, "Plugin is not runnable.");

		}

		return false;

	}

	if (debug && IsDebugging())

	{

		if (error)

		{

			snprintf(error, maxlength, "Plugin is already in debug mode.");

		}

		return false;

	} else if (!debug && !IsDebugging()) {

		if (error)

		{

			snprintf(error, maxlength, "Plugins is already in production mode.");

		}

		return false;

	}

	ICompilation *co = g_pVM->StartCompilation(m_ctx.ctx->plugin);

	if (!g_pVM->SetCompilationOption(co, "debug", (debug) ? "1" : "0"))

	{

		if (error)

		{

			snprintf(error, maxlength, "Failed to change plugin mode (JIT failure).");

		}

		return false;

	}

	sp_context_t *new_ctx = g_pVM->CompileToContext(co, &err);

	if (new_ctx)

	{

		memcpy(new_ctx->memory, m_ctx.ctx->memory, m_ctx.ctx->mem_size);

		new_ctx->hp = m_ctx.ctx->hp;

		new_ctx->sp = m_ctx.ctx->sp;

		new_ctx->frm = m_ctx.ctx->frm;

		new_ctx->dbreak = m_ctx.ctx->dbreak;

		new_ctx->context = m_ctx.ctx->context;

		memcpy(new_ctx->user, m_ctx.ctx->user, sizeof(m_ctx.ctx->user));

		g_pVM->FreeContext(m_ctx.ctx);

		m_ctx.ctx = new_ctx;

		m_ctx.base->SetContext(new_ctx);

		UpdateInfo();

	} else {

		if (error)

		{

			snprintf(error, maxlength, "Failed to recompile plugin (JIT error %d).", err);

		}

		return false;

	}

	return true;

}

bool CPlugin::IsRunnable() const

{

	return (m_status <= Plugin_Paused) ? true : false;

}

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

 * PLUGIN ITERATOR *

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

CPluginManager::CPluginIterator::CPluginIterator(List<CPlugin *> *_mylist)

{

	mylist = _mylist;

	Reset();

}

IPlugin *CPluginManager::CPluginIterator::GetPlugin()

{

	return (*current);

}

bool CPluginManager::CPluginIterator::MorePlugins()

{

	return (current != mylist->end());

}

void CPluginManager::CPluginIterator::NextPlugin()

{

	current++;

}

void CPluginManager::CPluginIterator::Release()

{

	g_PluginSys.ReleaseIterator(this);

}

CPluginManager::CPluginIterator::~CPluginIterator()

{

}

void CPluginManager::CPluginIterator::Reset()

{

	current = mylist->begin();

}

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

 * PLUGIN MANAGER *

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

CPluginManager::CPluginManager()

{

	m_LoadLookup = sm_trie_create();

	m_AllPluginsLoaded = false;

	m_MyIdent = NULL;

}

CPluginManager::~CPluginManager()

{

	/* :NOTICE: 

	 * Ignore the fact that there might be plugins in the cache.

	 * This usually means that Core is not being unloaded properly, and everything

	 * will crash anyway.  YAY

	 */

	sm_trie_destroy(m_LoadLookup);

}

void CPluginManager::LoadAll_FirstPass(const char *config, const char *basedir)

{

	/* First read in the database of plugin settings */

	SMCParseError err;

	unsigned int line, col;

	m_AllPluginsLoaded = false;

	if ((err=g_TextParser.ParseFile_SMC(config, &m_PluginInfo, &line, &col)) != SMCParse_Okay)

	{

		g_Logger.LogError("[SM] Encountered fatal error parsing file \"%s\"", config);

		const char *err_msg = g_TextParser.GetSMCErrorString(err);

		if (err_msg)

		{

			g_Logger.LogError("[SM] Parse error encountered: \"%s\"", err_msg);

		}

	}

	LoadPluginsFromDir(basedir, NULL);

}

void CPluginManager::LoadPluginsFromDir(const char *basedir, const char *localpath)

{

	char base_path[PLATFORM_MAX_PATH+1];

	/* Form the current path to start reading from */

	if (localpath == NULL)

	{

		g_LibSys.PathFormat(base_path, sizeof(base_path), "%s", basedir);

	} else {

		g_LibSys.PathFormat(base_path, sizeof(base_path), "%s/%s", basedir, localpath);

	}

	IDirectory *dir = g_LibSys.OpenDirectory(base_path);

	if (!dir)

	{

		char error[256];

		g_LibSys.GetPlatformError(error, sizeof(error));

		g_Logger.LogError("[SM] Failure reading from plugins path: %s", localpath);

		g_Logger.LogError("[SM] Platform returned error: %s", error);

		return;

	}

	while (dir->MoreFiles())

	{

		if (dir->IsEntryDirectory() 

			&& (strcmp(dir->GetEntryName(), ".") != 0)

			&& (strcmp(dir->GetEntryName(), "..") != 0)

			&& (strcmp(dir->GetEntryName(), "disabled") != 0)

			&& (strcmp(dir->GetEntryName(), "optional") != 0))

		{

			char new_local[PLATFORM_MAX_PATH+1];

			if (localpath == NULL)

			{

				/* If no path yet, don't add a former slash */

				snprintf(new_local, sizeof(new_local), "%s", dir->GetEntryName());

			} else {

				g_LibSys.PathFormat(new_local, sizeof(new_local), "%s/%s", localpath, dir->GetEntryName());

			}

			LoadPluginsFromDir(basedir, new_local);

		} else if (dir->IsEntryFile()) {

			const char *name = dir->GetEntryName();

			size_t len = strlen(name);

			if (len >= 4

				&& strcmp(&name[len-4], ".smx") == 0)

			{

				/* If the filename matches, load the plugin */

				char plugin[PLATFORM_MAX_PATH+1];

				if (localpath == NULL)

				{

					snprintf(plugin, sizeof(plugin), "%s", name);

				} else {

					g_LibSys.PathFormat(plugin, sizeof(plugin), "%s/%s", localpath, name);