#include <Core/Core.h>

using namespace Upp;

#include "Tcc.h"

#if __unix__
#define T_tcc_new				tcc_new
#define T_tcc_delete			tcc_delete
#define T_tcc_set_output_type	tcc_set_output_type
#define T_tcc_set_error_func	tcc_set_error_func
#define T_tcc_get_symbol		tcc_get_symbol
#define T_tcc_add_symbol		tcc_add_symbol
#define T_tcc_relocate			tcc_relocate
#define T_tcc_compile_string	tcc_compile_string
#define T_tcc_add_include_path	tcc_add_include_path
#define T_tcc_add_library_path	tcc_add_library_path
#endif

#if defined(PLATFORM_WIN32)
Tcc::Tcc(const char *dllName)
#else
Tcc::Tcc()
#endif
{
#if defined(PLATFORM_WIN32)
	hinstLib = LoadLibrary(TEXT(dllName));
	if (hinstLib == NULL) 
		throw Exc("Tcc library not found");

	T_tcc_new = (TCCState *(*)(void))GetProcAddress(hinstLib, "tcc_new");
	if (T_tcc_new == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}
	T_tcc_delete = (void (*)(TCCState *))GetProcAddress(hinstLib, "tcc_delete");
	if (T_tcc_delete == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}
	T_tcc_set_output_type = (int (*)(TCCState *s, int output_type))GetProcAddress(hinstLib, "tcc_set_output_type");
	if (T_tcc_set_output_type == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}
	T_tcc_set_error_func = (void (*)(TCCState *s, void *error_opaque, void (*error_func)(void *opaque, const char *msg)))GetProcAddress(hinstLib, "tcc_set_error_func");
	if (T_tcc_set_error_func == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}		
	T_tcc_get_symbol = (int (*)(TCCState *s, unsigned long *pval, const char *name))GetProcAddress(hinstLib, "tcc_get_symbol");
	if (T_tcc_get_symbol == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}
	T_tcc_add_symbol = (int (*)(TCCState *s, const char *name, unsigned long val))GetProcAddress(hinstLib, "tcc_add_symbol");
	if (T_tcc_add_symbol == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}
	T_tcc_relocate = (int (*)(TCCState *s))GetProcAddress(hinstLib, "tcc_relocate");
	if (T_tcc_relocate == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}
	T_tcc_compile_string = (int (*)(TCCState *s, const char *buf))GetProcAddress(hinstLib, "tcc_compile_string");
	if (T_tcc_compile_string == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}	
	T_tcc_add_include_path = (int (*)(TCCState *s, const char *buf))GetProcAddress(hinstLib, "tcc_add_include_path");
	if (T_tcc_compile_string == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}	
	T_tcc_add_library_path = (int (*)(TCCState *s, const char *buf))GetProcAddress(hinstLib, "tcc_add_library_path");
	if (T_tcc_compile_string == NULL) {
		FreeLibrary(hinstLib);
		throw Exc("Tcc function not found");
	}	
#endif
    stateTcc = T_tcc_new();
    if (!stateTcc) 
        throw Exc("Tcc can not initialize");
    
	T_tcc_set_error_func(stateTcc, NULL, Tcc::DefaultErrorHandler);
    errorMsg = "";
    
    T_tcc_set_output_type(stateTcc, TCC_OUTPUT_MEMORY);
}

String Tcc::errorMsg = "";
int Tcc::initialProgramLines;

void Tcc::DefaultErrorHandler(void* opaque, const char* msg)
{
	if (!errorMsg.IsEmpty())
		errorMsg.Cat('\n');		// When calling Relocate this handle can get more than one error
	String message = msg;
	int linePos = sizeof("<string>:")-1;
	if (message.Left(linePos) == "<string>:") {		// Fix the line number in the error message
		int endLinePos = message.Find(':', linePos+1);
		int line = atoi(message.Mid(linePos, endLinePos-linePos)) - initialProgramLines;
		message = message.Left(linePos) + FormatInt(line) + message.Mid(endLinePos);
	}
	errorMsg.Cat(message);
}

Tcc::~Tcc()
{
	T_tcc_delete(stateTcc);
#if defined(PLATFORM_WIN32)	
	FreeLibrary(hinstLib);
#endif
}

void tcc_throw(char *str)
{
	 throw Exc(str);
}

void Tcc::Compile(const char *my_program)
{
	program << 	"// Basic declarations\n"	
				"typedef int bool;\n"
				"#define true  1\n"
				"#define false 0\n"	
				"#define M_PI	3.1415926535897932\n"		// It does not seem to be in math.h
				"\n"
				"void throw(char *str);\n"
				"\n";
	initialProgramLines = 0;
	int pos = 0;
	while((pos = program.Find('\n', pos)) >= 0) {
		initialProgramLines++;
		pos++;
	}
	program << my_program;
		
    if (T_tcc_compile_string(stateTcc, program) != 0) 
        throw Exc(errorMsg);
    AddSymbol("throw", (void *)&tcc_throw);
}

void Tcc::AddSymbol(const char *funName, void *fun)
{
	T_tcc_add_symbol(stateTcc, funName, (unsigned long)fun);
	if (!errorMsg.IsEmpty())
    	throw Exc(errorMsg);
}
void Tcc::Link()
{
	if(T_tcc_relocate(stateTcc) != 0)
		throw Exc(errorMsg);
}

void *Tcc::GetSymbol(const char *funName)
{
	unsigned long val;
	T_tcc_get_symbol(stateTcc, (unsigned long *)&val, funName);
	if (!errorMsg.IsEmpty())
    	throw Exc(errorMsg);
    		
	return (void *)val;
}

bool Tcc::AddIncludePath(const char *path)
{
	bool ret = T_tcc_add_include_path(stateTcc, path) == 0? true: false;
	if (!errorMsg.IsEmpty())
    	throw Exc(errorMsg);
	return ret;
}
bool Tcc::AddLibraryPath(const char *path)
{
	bool ret = T_tcc_add_library_path(stateTcc, path) == 0? true: false;
	if (!errorMsg.IsEmpty())
    	throw Exc(errorMsg);
    return ret;	
}