#include "nasal.h"

resource_file        resource;
nasal_lexer          lexer;
nasal_parse          parser;
abstract_syntax_tree libroot;
abstract_syntax_tree root;
abstract_syntax_tree linker;

nasal_runtime runtime;

std::string command;

int main()
{
	std::cout<<"       __                _ "<<std::endl;
	std::cout<<"    /\\ \\ \\__ _ ___  __ _| |"<<std::endl;
	std::cout<<"   /  \\/ / _` / __|/ _` | |"<<std::endl;
	std::cout<<"  / /\\  / (_| \\__ \\ (_| | |"<<std::endl;
	std::cout<<"  \\_\\ \\/ \\__,_|___/\\__,_|_|"<<std::endl;
#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 2.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")
		{
			std::cout<<">> Be careful that this program does not support unicode(unicode will be set to \'?\')"<<std::endl;
			std::cout<<">> [\'file\'] input a file."<<std::endl;
			std::cout<<">> [cls   ] 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<<">> [info  ] print lexer,parser and ast on screen."<<std::endl;
			std::cout<<">> [exit  ] quit nasal interpreter."<<std::endl;
		}
		else if(command=="cls")
		{
#ifdef _WIN32
			system("cls");
#endif
#ifdef _linux_
			system("clear");
#endif
#ifdef TARGET_OS_MAC
			system("clear");
#endif
		}
		else if(command=="del")
		{
			resource.delete_all_source();
			lexer.delete_all_tokens();
			parser.delete_all_elements();
			root.set_clear();
			linker.set_clear();
			std::cout<<">> [Delete] complete."<<std::endl;
		}
		else if(command=="lib")
		{
			libroot.set_clear();
			resource.load_lib_file();
			lexer.scanner(resource.get_source());
			lexer.generate_detail_token();
			if(!lexer.get_error())
			{
				parser.get_token_list(lexer.get_detail_token_list());
				parser.main_generate();
				if(!parser.get_error())
				{
					libroot=parser.get_root();
					std::cout<<">> [Lib] loaded."<<std::endl;
				}
				else
					std::cout<<">> [Lib] lib files have parse error(s),stop."<<std::endl;
			}
			else
				std::cout<<">> [Lib] lib files have lexer error(s),stop."<<std::endl;
			resource.delete_all_source();
			lexer.delete_all_tokens();
			parser.delete_all_elements();
		}
		else if(command=="rs")
		{
			resource.print_resource();
		}
		else if(command=="lex")
		{
			lexer.scanner(resource.get_source());
			lexer.generate_detail_token();
			if(!lexer.get_error())
				lexer.print_token_list();
			else
				std::cout<<">> [Lexer] error occurred,stop."<<std::endl;
		}
		else if(command=="par")
		{
			lexer.scanner(resource.get_source());
			lexer.generate_detail_token();
			if(!lexer.get_error())
			{
				parser.get_token_list(lexer.get_detail_token_list());
				parser.print_detail_token();
				parser.main_generate();
			}
			else
				std::cout<<">> [Lexer] error occurred,stop."<<std::endl;
		}
		else if(command=="ast")
		{
			lexer.scanner(resource.get_source());
			lexer.generate_detail_token();
			if(!lexer.get_error())
			{
				parser.get_token_list(lexer.get_detail_token_list());
				parser.main_generate();
				if(!parser.get_error())
					parser.get_root().print_tree();
				else
					std::cout<<">> [Parse] error occurred,stop."<<std::endl;
			}
			else
				std::cout<<">> [Lexer] error occurred,stop."<<std::endl;
		}
		else if(command=="run")
		{
			lexer.scanner(resource.get_source());
			lexer.generate_detail_token();
			if(!lexer.get_error())
			{
				parser.get_token_list(lexer.get_detail_token_list());
				parser.main_generate();
				if(!parser.get_error())
				{
					linker.set_clear();
					linker=libroot;
					linker.merge_children(parser.get_root());
					root=linker;
					runtime.main_proc(root);
					//root.print_tree();
				}
				else
					std::cout<<">> [Parse] error occurred,stop."<<std::endl;
			}
			else
				std::cout<<">> [Lexer] error occurred,stop."<<std::endl;
		}
		else if(command=="info")
		{
			lexer.scanner(resource.get_source());
			lexer.print_token_list();
			lexer.generate_detail_token();
			parser.get_token_list(lexer.get_detail_token_list());
			parser.print_detail_token();
			parser.main_generate();
			parser.get_root().print_tree();
		}
		else if(command=="exit")
			break;
		else
			resource.input_file(command);
	}
	return 0;
}