#include <stdio.h>

#include "sourcemod.h"

#include "sourcemm_api.h"

#include "systems/LibrarySys.h"

#include "vm/sp_vm_engine.h"

#include <sh_string.h>

SourcePawnEngine g_SourcePawn;

SourceModBase g_SourceMod;

ILibrary *g_pJIT = NULL;

SourceHook::String g_BaseDir;

ISourcePawnEngine *g_pSourcePawn = &g_SourcePawn;

IVirtualMachine *g_pVM;

typedef int (*GIVEENGINEPOINTER)(ISourcePawnEngine *);

typedef unsigned int (*GETEXPORTCOUNT)();

typedef IVirtualMachine *(*GETEXPORT)(unsigned int);

typedef void (*NOTIFYSHUTDOWN)();

void ShutdownJIT()

{

	NOTIFYSHUTDOWN notify = (NOTIFYSHUTDOWN)g_pJIT->GetSymbolAddress("NotifyShutdown");

	if (notify)

	{

		notify();

	}

	g_pJIT->CloseLibrary();

}

bool SourceModBase::InitializeSourceMod(char *error, size_t err_max, bool late)

{

	//:TODO: we need a localinfo system!

	g_BaseDir.assign(g_SMAPI->GetBaseDir());

	/* Attempt to load the JIT! */

	char file[PLATFORM_MAX_PATH];

	char myerror[255];

	g_SMAPI->PathFormat(file, sizeof(file), "%s/addons/sourcemod/bin/sourcepawn.jit.x86.%s",

		g_BaseDir.c_str(),

		PLATFORM_LIB_EXT

		);

	g_pJIT = g_LibSys.OpenLibrary(file, myerror, sizeof(myerror));

	if (!g_pJIT)

	{

		if (error && err_max)

		{

			snprintf(error, err_max, "%s (failed to load bin/sourcepawn.jit.x86.%s)", 

				myerror,

				PLATFORM_LIB_EXT);

		}

		return false;

	}

	int err;

	GIVEENGINEPOINTER jit_init = (GIVEENGINEPOINTER)g_pJIT->GetSymbolAddress("GiveEnginePointer");

	if (!jit_init)

	{

		ShutdownJIT();

		if (error && err_max)

		{

			snprintf(error, err_max, "Failed to find GiveEnginePointer in JIT!");

		}

		return false;

	}

	if ((err=jit_init(g_pSourcePawn)) != 0)

	{

		ShutdownJIT();

		if (error && err_max)

		{

			snprintf(error, err_max, "GiveEnginePointer returned %d in the JIT", err);

		}

		return false;

	}

	GETEXPORTCOUNT jit_getnum = (GETEXPORTCOUNT)g_pJIT->GetSymbolAddress("GetExportCount");

	GETEXPORT jit_get = (GETEXPORT)g_pJIT->GetSymbolAddress("GetExport");

	if (!jit_get)

	{

		ShutdownJIT();

		if (error && err_max)

		{

			snprintf(error, err_max, "JIT is missing a necessary export!");

		}

		return false;

	}

	unsigned int num = jit_getnum();

	if (!num || ((g_pVM=jit_get(0)) == NULL))

	{

		ShutdownJIT();

		if (error && err_max)

		{

			snprintf(error, err_max, "JIT did not export any virtual machines!");

		}

		return false;

	}

	return true;

}