#include "cmd.h"

#include <blue/cmd.h>
#include <blue/term.h>
#include <errno.h>
#include <ivy/file.h>
#include <ivy/lang/lex.h>
#include <stdio.h>

enum {
	ARG_SOURCE_FILE,
};

static int compile_file(const char *path)
{

	FILE *fp = fopen(path, "r");
	if (!fp) {
		b_err("cannot open source file '%s'", path);
		b_i("reason: %s", strerror(errno));
		return -1;
	}

	struct ivy_file *src = ivy_file_from_fp(fp);
	struct ivy_lexer *lex;
	if (ivy_lexer_create(&lex) != IVY_OK) {
		b_err("failed to initialise Ivy lexer");
		ivy_file_close(src);
		return -1;
	}

	ivy_lexer_set_source(lex, &src->f_base);
	enum ivy_status status = IVY_OK;

	while (true) {
		struct ivy_token *tok = ivy_lexer_read(lex);
		status = ivy_lexer_get_status(lex);
		if (status == IVY_ERR_EOF) {
			break;
		}

		if (status != IVY_OK) {
			b_err("failed to parse '%s'", path);
			b_i("reason: lex error (%s)",
			    ivy_status_to_string(ivy_lexer_get_status(lex)));
			break;
		}

		switch (tok->t_type) {
		case IVY_TOK_KEYWORD:
			b_fputs("[magenta]", stdout);
			break;
		case IVY_TOK_SYMBOL:
			b_fputs("[blue]", stdout);
			break;
		case IVY_TOK_ATOM:
			b_fputs("[yellow]", stdout);
			break;
		case IVY_TOK_NUMBER:
			b_fputs("[yellow]", stdout);
			break;
		case IVY_TOK_LABEL:
			b_fputs("[red]", stdout);
			break;
		case IVY_TOK_IDENT:
			b_fputs("[cyan]", stdout);
			break;
		case IVY_TOK_STRING:
			b_fputs("[green]", stdout);
			break;
		case IVY_TOK_STR_START:
			b_fputs("[green]", stdout);
			break;
		case IVY_TOK_STR_END:
			b_fputs("[green]", stdout);
			break;
		case IVY_TOK_LINEFEED:
			b_fputs("[bright,black]", stdout);
			break;
		default:
			break;
		}

		printf("%s", ivy_lex_token_type_to_string(tok->t_type));

		switch (tok->t_type) {
		case IVY_TOK_IDENT:
		case IVY_TOK_LABEL:
		case IVY_TOK_STRING:
			printf("(%s)", tok->t_str);
			break;
		case IVY_TOK_SYMBOL:
			printf("(%s)", ivy_symbol_to_string(tok->t_symbol));
			break;
		case IVY_TOK_KEYWORD:
			printf("(%s)", ivy_keyword_to_string(tok->t_keyword));
			break;
		default:
			break;
		}

		b_fputs("[reset]\n", stdout);
	}

	int r = (status == IVY_OK || status == IVY_ERR_EOF) ? 0 : -1;
	ivy_file_close(src);
	ivy_lexer_destroy(lex);
	return r;
}

static int compile(const b_command *cmd, const b_arglist *args, const b_array *_)
{
	b_arglist_iterator it;
	b_arglist_foreach_filtered(&it, args, B_COMMAND_INVALID_ID, ARG_SOURCE_FILE)
	{
		b_arglist_value *path = it.value;
		if (path->val_type != B_COMMAND_ARG_STRING) {
			continue;
		}

		printf("%s\n", path->val_str);
		int r = compile_file(path->val_str);
		if (r != 0) {
			return r;
		}
	}

	return 0;
}

B_COMMAND(CMD_COMPILE, CMD_ROOT)
{
	B_COMMAND_NAME("compile");
	B_COMMAND_SHORT_NAME('C');
	B_COMMAND_DESC(
		"compile one or more Ivy source files into Ivy object files.");
	B_COMMAND_FLAGS(B_COMMAND_SHOW_HELP_BY_DEFAULT);
	B_COMMAND_FUNCTION(compile);

	B_COMMAND_ARG(ARG_SOURCE_FILE)
	{
		B_ARG_NAME("source file");
		B_ARG_DESC("the .im source files to compile.");
		B_ARG_NR_VALUES(B_ARG_1_OR_MORE_VALUES);
	}

	B_COMMAND_HELP_OPTION();
}