#include <stdio.h>
#include <stddef.h>

#include "io/file.h"
#include "cli/args.h"
#include "lexer/lexer.h"
#include "parser/parser.h"

int main(int argc, char** argv)
{
	struct Args args;

	enum ParseResult result = parse_args(argc, argv, &args);
	if (result == PARSE_EXIT)
		return 0;
	if (result == PARSE_ERROR)
		return 1;

	struct File source;
	if (!read_file(args.input_path, &source))
		return 1;

	struct Lexer lexer = create_lexer(source.data);

	struct Program program;
	if (!parse_program(&lexer, &program))
	{
		free_program(&program);
		free_file(&source);
		return 1;
	}

	for (size_t i = 0; i < program.const_count; i += 1)
	{
		struct ConstDecl decl = program.consts[i];
		printf("const %.*s = %.*s\n",
			(int)decl.name.length, decl.name.start,
			(int)decl.value.length, decl.value.start);
	}

	for (size_t i = 0; i < program.data_count; i += 1)
	{
		struct DataDecl decl = program.data_decls[i];
		printf("data %.*s = %.*s\n",
			(int)decl.name.length, decl.name.start,
			(int)decl.value.length, decl.value.start);
	}

	for (size_t i = 0; i < program.proc_count; i += 1)
	{
		struct ProcDecl proc = program.procs[i];
		printf("proc %.*s(", (int)proc.name.length, proc.name.start);
		for (size_t p = 0; p < proc.param_count; p += 1)
		{
			struct Param param = proc.params[p];
			printf("%s%.*s: %.*s", p == 0 ? "" : ", ",
				(int)param.name.length, param.name.start,
				(int)param.reg.length, param.reg.start);
		}
		printf(")\n");

		for (size_t s = 0; s < proc.body_count; s += 1)
		{
			struct Statement statement = proc.body[s];
			switch (statement.kind)
			{
				case STATEMENT_ASSIGN:
				{
					struct AssignStatement assign = statement.assign;
					printf("  %s%.*s %.*s %.*s\n",
						assign.target_deref ? "^" : "",
						(int)assign.target.length, assign.target.start,
						(int)assign.op.length, assign.op.start,
						(int)assign.value.length, assign.value.start);
					break;
				}
				case STATEMENT_LABEL:
					printf("  %.*s:\n", (int)statement.label.name.length, statement.label.name.start);
					break;
				case STATEMENT_GOTO:
					printf("  goto %.*s\n", (int)statement.jump.label.length, statement.jump.label.start);
					break;
				case STATEMENT_SYSCALL:
					printf("  syscall\n");
					break;
			}
		}
	}

	free_program(&program);
	free_file(&source);
}