#include "nasal.h"

nasal_resource resource;
nasal_lexer    lexer;
nasal_parse    parse;
std::string    command;
nasal_runtime  runtime;

void help()
{
	std::cout<<">> [\'file\'] input a file."<<std::endl;
	std::cout<<">> [cls   ] clear the screen."<<std::endl;
	std::cout<<">> [clear ] clear the screen."<<std::endl;
	std::cout<<">> [del   ] clear the resource code."<<std::endl;
	std::cout<<">> [lib   ] add lib file."<<std::endl;
	std::cout<<">> [rs    ] print resource code."<<std::endl;
	std::cout<<">> [lex   ] turn code into tokens."<<std::endl;
	std::cout<<">> [par   ] turn tokens into abstract syntax tree."<<std::endl;
	std::cout<<">> [ast   ] check the abstract syntax tree."<<std::endl;
	std::cout<<">> [run   ] run code."<<std::endl;
	std::cout<<">> [logo  ] print logo of nasal ."<<std::endl;
	std::cout<<">> [exit  ] quit nasal interpreter."<<std::endl;
	return;
}

void logo()
{
    std::cout<<"       __                _      "<<std::endl;
	std::cout<<"    /\\ \\ \\__ _ ___  __ _| |  "<<std::endl;
	std::cout<<"   /  \\/ / _` / __|/ _` | |    "<<std::endl;
	std::cout<<"  / /\\  / (_| \\__ \\ (_| | |  "<<std::endl;
	std::cout<<"  \\_\\ \\/ \\__,_|___/\\__,_|_|"<<std::endl;
    return;
}

void del_func()
{
	resource.clear();
	lexer.clear();
	parse.clear();
	std::cout<<">> [Delete] complete."<<std::endl;
	return;
}

void lex_func()
{
	lexer.scanner(resource.get_file());
	if(!lexer.get_error())
		lexer.print_token();
	else
		std::cout<<">> [lexer] error(s) occurred,stop.\n";
	return;
}

void par_func()
{
	lexer.scanner(resource.get_file());
	if(!lexer.get_error())
	{
		parse.set_toklist(lexer.get_token_list());
		parse.main_process();
		if(parse.get_error())
			std::cout<<">> [parse] error(s) occurred,stop.\n";
	}
	else
		std::cout<<">> [lexer] error(s) occurred,stop.\n";
	return;
}

void ast_print()
{
	lexer.scanner(resource.get_file());
	if(!lexer.get_error())
	{
		parse.set_toklist(lexer.get_token_list());
		parse.main_process();
		if(parse.get_error())
			std::cout<<">> [parse] error(s) occurred,stop.\n";
		else
			parse.get_root().print_ast(0);
	}
	else
		std::cout<<">> [lexer] error(s) occurred,stop.\n";
	return;
}
void runtime_start()
{
	lexer.scanner(resource.get_file());
	if(!lexer.get_error())
	{
		parse.set_toklist(lexer.get_token_list());
		parse.main_process();
		if(parse.get_error())
			std::cout<<">> [parse] error(s) occurred,stop.\n";
		else
		{
			runtime.set_root(parse.get_root());
			runtime.run();
		}
	}
	else
		std::cout<<">> [lexer] error(s) occurred,stop.\n";
	return;
}

int main()
{
#ifdef _WIN32
	// use chcp 65001 to use unicode io
	system("chcp 65001");
	system("cls");
#endif
	// this curve looks really cool
	logo();
#ifdef _WIN32
	std::cout<<">> [system] Windows system."<<std::endl;
#endif
#ifdef _linux_
	std::cout<<">> [system] Linux system."<<std::endl;
#endif
#ifdef TARGET_OS_MAC
	std::cout<<">> [system] MacOS system."<<std::endl;
#endif
	std::cout<<">> Nasal interpreter ver 3.0 ."<<std::endl;
	std::cout<<">> Code: https://github.com/ValKmjolnir/Nasal-Interpreter"<<std::endl;
	std::cout<<">> More info: http://wiki.flightgear.org/Nasal_scripting_language"<<std::endl;
	std::cout<<">> Input \"help\" to get help ."<<std::endl;
    while(1)
	{
		std::cout<<">> ";
		std::cin>>command;
		if(command=="help")
			help();
		else if(command=="cls" || command=="clear")
		{
#ifdef _WIN32
			system("cls");
#endif
#ifdef _linux_
			system("clear");
#endif
#ifdef TARGET_OS_MAC
			system("clear");
#endif
		}
		else if(command=="del")
			del_func();
		else if(command=="lib")
			resource.load_lib();
		else if(command=="rs")
			resource.print_file();
		else if(command=="lex")
			lex_func();
		else if(command=="par")
			par_func();
		else if(command=="ast")
			ast_print();
		else if(command=="run")
			runtime_start();
		else if(command=="logo")
			logo();
		else if(command=="exit")
			break;
		else
			resource.input_file(command);
	}
    return 0;
}