#include "../debug.h"
#include "../line-ed/line-ed.h"
#include "cmd.h"

#include <blue/cmd.h>
#include <blue/term.h>
#include <ivy/lang/lex.h>
#include <ivy/lang/ast.h>
#include <stdbool.h>

enum {
	OPT_LEX_ONLY = 0x1000u,
};

static int repl_lex_only()
{
	struct line_ed *ed = line_ed_create();
	line_ed_set_flags(ed, LINE_ED_REMOVE_CONTINUATIONS);
	
	struct ivy_lexer *lex;
	enum ivy_status status = ivy_lexer_create(&lex);

	if (status != IVY_OK) {
		line_ed_destroy(ed);
		return -1;
	}

	ivy_lexer_set_source(lex, &ed->l_line_source);

	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("lex error (%s)",
			      ivy_status_to_string(ivy_lexer_get_status(lex)));
			break;
		}

		print_lex_token(tok);
	}

	ivy_lexer_destroy(lex);
	line_ed_destroy(ed);
	return 0;
}

static void skip_line(struct ivy_lexer *lex)
{
	while (ivy_lexer_tokens_available(lex)) {
		struct ivy_token *tok = ivy_lexer_read(lex);
		bool line_end = (tok->t_type == IVY_TOK_LINEFEED);
		
		ivy_token_destroy(tok);
		
		if (line_end) {
			break;
		}
	}
}

static int repl_full()
{
	struct line_ed *ed = line_ed_create();
	line_ed_set_flags(ed, LINE_ED_REMOVE_CONTINUATIONS);
	
	struct ivy_lexer *lex;
	enum ivy_status status = ivy_lexer_create(&lex);

	if (status != IVY_OK) {
		line_ed_destroy(ed);
		return -1;
	}

	ivy_lexer_set_source(lex, &ed->l_line_source);

	struct ivy_parser *parser;
	status = ivy_parser_create(&parser);

	if (status != IVY_OK) {
		ivy_lexer_destroy(lex);
		line_ed_destroy(ed);
		return -1;
	}

	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("lex error (%s)",
			      ivy_status_to_string(ivy_lexer_get_status(lex)));
			continue;
		}

		status = ivy_parser_push_token(parser, tok);

		if (status != IVY_OK) {
			b_err("parse error (%s)",
			      ivy_status_to_string(status));
			ivy_token_destroy(tok);
			skip_line(lex);
			continue;
		}

		if (!ivy_parser_is_node_complete(parser)) {
			continue;
		}

		struct ivy_ast_node *root = ivy_parser_root_node(parser);
		ivy_ast_node_print(root);
#if 0
		struct ivy_ast_node *node = ivy_parser_dequeue_node(parser);
		if (!node) {
			continue;
		}

		ivy_ast_node_print(node);
		ivy_ast_node_destroy(node);
#endif
	}

	ivy_parser_destroy(parser);
	ivy_lexer_destroy(lex);
	line_ed_destroy(ed);
	return 0;
}

int repl(const b_command *cmd, const b_arglist *args, const b_array *_)
{
	if (b_arglist_get_count(args, OPT_LEX_ONLY, B_COMMAND_INVALID_ID) > 0) {
		return repl_lex_only();
	}

	return repl_full();
}

B_COMMAND(CMD_REPL, CMD_ROOT)
{
	B_COMMAND_NAME("shell");
	B_COMMAND_SHORT_NAME('S');
	B_COMMAND_DESC("start an interactive Ivy shell.");
	B_COMMAND_HELP_OPTION();
	B_COMMAND_FUNCTION(repl);

	B_COMMAND_OPTION(OPT_LEX_ONLY) {
		B_OPTION_LONG_NAME("lex-only");
		B_OPTION_SHORT_NAME('L');
		B_OPTION_DESC("print the lexical tokens generated by the user input.");
	}
}