#include <assert.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include "o_staticmap.h"
#include "compiler/parse.h"
#include "compiler/serialize.h"
#include "compiler/compile.h"
#include "runtime/runtime.h"
#include "runtime/runtime_error.h"
#include "builtins/basic.h"

static const char usage[] = 
	"Usage patterns:\n"
	"    $ noja run file.noja\n"
	"    $ noja run inline \"print('some noja code');\"\n"
	"    $ noja dis file.noja\n"
	"    $ noja dis inline \"print('some noja code');\"\n";
static _Bool interpret_file(const char *file);
static _Bool interpret_code(const char *code);
static _Bool disassemble_file(const char *file);
static _Bool disassemble_code(const char *code);

static void print_error(const char *type, Error *error)
{
	if(type == NULL)
		fprintf(stderr, "Error");
	else if(error->internal)
		fprintf(stderr, "Internal Error");
	else
		fprintf(stderr, "%s Error", type);

	fprintf(stderr, ": %s. ", error->message);

	if(error->file != NULL)
		{
			if(error->line > 0 && error->func != NULL)
				fprintf(stderr, "(Reported in %s:%d in %s)", error->file, error->line, error->func);
			else if(error->line > 0 && error->func == NULL)
				fprintf(stderr, "(Reported in %s:%d)", error->file, error->line);
			else if(error->line < 1 && error->func != NULL)
				fprintf(stderr, "(Reported in %s in %s)", error->file, error->func);
		}

	fprintf(stderr, "\n");
}

_Bool noja(int argc, char **argv)
{
	assert(argc > 0);

	if(argc == 1)
		{
			// $ noja
			fprintf(stderr, "Error: Incorrect usage.\n\n");
			fprintf(stderr, usage);
			return 0;
		}

	if(!strcmp(argv[1], "run"))
		{
			Error error;
			Error_Init(&error);
			
			if(argc == 2)
				{
					Error_Report(&error, 0, "Missing source file");
					print_error(NULL, &error);
					Error_Free(&error);
					return 0;
				}

			_Bool r;

			if(!strcmp(argv[2], "inline"))
				{
					if(argc == 3)
						{
							Error_Report(&error, 0, "Missing source string");
							print_error(NULL, &error);
							Error_Free(&error);
							return 0;
						}

					r = interpret_code(argv[3]);
				}
			else
				r = interpret_file(argv[2]);
			return r;
		}
	
	if(!strcmp(argv[1], "dis"))
		{
			Error error;
			Error_Init(&error);
			
			if(argc == 2)
				{
					Error_Report(&error, 0, "Missing source file");
					print_error(NULL, &error);
					Error_Free(&error);
					return 0;
				}

			_Bool r;

			if(!strcmp(argv[2], "inline"))
				{
					if(argc == 3)
						{
							Error_Report(&error, 0, "Missing source string");
							print_error(NULL, &error);
							Error_Free(&error);
							return 0;
						}

					r = disassemble_code(argv[3]);
				}
			else
				r = disassemble_file(argv[2]);
			return r;
		}

	if(!strcmp(argv[1], "help"))
		{
			fprintf(stdout, usage);
			return 1;
		}

	fprintf(stderr, "Error: Incorrect usage.\n\n");
	fprintf(stderr, usage);
	return 0;
}

static _Bool interpret(Source *src)
{
	// Compile the code. This section transforms
	// a [Source] into an [Executable].
	Executable *exe;
	{
		// Create a bump-pointer allocator to hold the AST.
		BPAlloc *alloc = BPAlloc_Init(-1);

		if(alloc == NULL)
			{
				fprintf(stderr, "Internal Error: Couldn't allocate bump-pointer allocator to hold the AST.\n");
				return 0;
			}

		Error error;
		Error_Init(&error);

		// NOTE: The AST is stored in the BPAlloc. It's
		//       lifetime is the same as the pool.
		AST *ast = parse(src, alloc, &error);
	
		if(ast == NULL)
			{
				assert(error.occurred);
				print_error("Parsing", &error);
				Error_Free(&error);
				BPAlloc_Free(alloc);
				return 0;
			}
		
		exe = compile(ast, alloc, &error);

		// We're done with the AST, independently from
		// the compilation result.
		BPAlloc_Free(alloc);

		if(exe == NULL)
			{
				assert(error.occurred);
				print_error("Compilation", &error);
				Error_Free(&error);
				return 0;
			}
	}			

	// Now execute it.
	{
		Runtime *runt = Runtime_New(-1, -1, NULL, NULL);

		if(runt == NULL)
			{
				Error error;
				Error_Init(&error);
				Error_Report(&error, 1, "Couldn't initialize runtime");
				print_error(NULL, &error);
				Error_Free(&error);
				Executable_Free(exe);
				return 0;
			}

		// We use a [RuntimeError] instead of a simple [Error]
		// because the [RuntimeError] makes a snapshot of the
		// runtime state when an error is reported. Other than
		// this fact they are interchangable. Any function that
		// expects a pointer to [Error] can receive a [RuntimeError]
		// upcasted to [Error].
		RuntimeError error;
		RuntimeError_Init(&error, runt); // Here we specify the runtime to snapshot in case of failure.
		
		Object *bins = Object_NewStaticMap(bins_basic, runt, (Error*) &error);

		if(bins == NULL)
			{
				assert(error.base.occurred == 1);
				print_error(NULL, (Error*) &error);
				RuntimeError_Free(&error);
				Executable_Free(exe);
				Runtime_Free(runt);
				return 0;
			}

		Runtime_SetBuiltins(runt, bins);

		Object *o = run(runt, (Error*) &error, exe, 0, NULL, NULL, 0);

		// NOTE: The pointer to the builtins object is invalidated
		//       now because it may be moved by the garbage collector.

		if(o == NULL)
			{
				print_error("Runtime", (Error*) &error);

				if(error.snapshot == NULL)
					fprintf(stderr, "No snapshot available.\n");
				else
					Snapshot_Print(error.snapshot, stderr);

				RuntimeError_Free(&error);
			}

		Runtime_Free(runt);
		Executable_Free(exe);

		return o != NULL;
	}
}

static _Bool disassemble(Source *src)
{
	// Compile the code. This section transforms
	// a [Source] into an [Executable].
	Executable *exe;
	{
		// Create a bump-pointer allocator to hold the AST.
		BPAlloc *alloc = BPAlloc_Init(-1);

		if(alloc == NULL)
			{
				fprintf(stderr, "Internal Error: Couldn't allocate bump-pointer allocator to hold the AST.\n");
				return 0;
			}

		Error error;
		Error_Init(&error);

		// NOTE: The AST is stored in the BPAlloc. It's
		//       lifetime is the same as the pool.
		AST *ast = parse(src, alloc, &error);
	
		if(ast == NULL)
			{
				assert(error.occurred);
				print_error("Parsing", &error);
				Error_Free(&error);
				BPAlloc_Free(alloc);
				return 0;
			}

		exe = compile(ast, alloc, &error);

		// We're done with the AST, independently from
		// the compilation result.
		BPAlloc_Free(alloc);

		if(exe == NULL)
			{
				assert(error.occurred);
				print_error("Compilation", &error);
				Error_Free(&error);
				return 0;
			}
	}

	Executable_Dump(exe);
	return 1;
}

static _Bool interpret_file(const char *file)
{
	Error error;
	Error_Init(&error);
	
	Source *src = Source_FromFile(file, &error);

	if(src == NULL)
		{
			assert(error.occurred == 1);
			print_error(NULL, &error);
			Error_Free(&error);
			return 0;
		}

	_Bool r = interpret(src);

	Source_Free(src);
	return r;
}

static _Bool interpret_code(const char *code)
{
	Error error;
	Error_Init(&error);
	
	Source *src = Source_FromString(NULL, code, -1, &error);

	if(src == NULL)
		{
			assert(error.occurred);
			print_error(NULL, &error);
			Error_Free(&error);
			return 0;
		}

	_Bool r = interpret(src);

	Source_Free(src);
	return r;
}

static _Bool disassemble_file(const char *file)
{
	Error error;
	Error_Init(&error);
	
	Source *src = Source_FromFile(file, &error);

	if(src == NULL)
		{
			assert(error.occurred == 1);
			print_error(NULL, &error);
			Error_Free(&error);
			return 0;
		}

	_Bool r = disassemble(src);

	Source_Free(src);
	return r;
}

static _Bool disassemble_code(const char *code)
{
	Error error;
	Error_Init(&error);
	
	Source *src = Source_FromString(NULL, code, -1, &error);

	if(src == NULL)
		{
			assert(error.occurred);
			print_error(NULL, &error);
			Error_Free(&error);
			return 0;
		}

	_Bool r = disassemble(src);

	Source_Free(src);
	return r;
}