#include <blue/object/string.h>
#include <ivy/lang/lex.h>
#include <stdio.h>

#include "ctx.h"
#include "node.h"

struct unit_package_parser_state {
	struct parser_state s_base;
	int s_prev_token;
};

static enum ivy_status parse_dot(struct ivy_parser *ctx, struct ivy_token *tok)
{
	struct unit_package_parser_state *state
		= parser_get_state(ctx, struct unit_package_parser_state);

	if (state->s_prev_token != IVY_TOK_IDENT) {
		return IVY_ERR_BAD_SYNTAX;
	}

	state->s_prev_token = IVY_SYM_DOT;
	return IVY_OK;
}

static enum ivy_status parse_ident(struct ivy_parser *ctx, struct ivy_token *tok)
{
	struct unit_package_parser_state *state
		= parser_get_state(ctx, struct unit_package_parser_state);

	struct ivy_ast_unit_package_node *node = (struct ivy_ast_unit_package_node *)(state->s_base.s_node);

	if (state->s_prev_token == IVY_TOK_IDENT) {
		return IVY_ERR_BAD_SYNTAX;
	}

	b_queue_push_back(&node->n_ident, &tok->t_entry);

	state->s_prev_token = IVY_TOK_IDENT;
	return IVY_OK;
}

static enum ivy_status parse_linefeed(struct ivy_parser *ctx, struct ivy_token *tok)
{
	struct unit_package_parser_state *state
		= parser_get_state(ctx, struct unit_package_parser_state);

	if (state->s_prev_token != IVY_TOK_IDENT) {
		return IVY_ERR_BAD_SYNTAX;
	}

	parser_pop_state(ctx, STATE_ADD_NODE_TO_PARENT);
	return IVY_OK;
}

static enum ivy_status add_child(
	struct ivy_ast_node *parent, struct ivy_ast_node *child)
{
	return IVY_ERR_NOT_SUPPORTED;
}

static void print(struct ivy_ast_node *node)
{
	struct ivy_ast_unit_package_node *unit_package = (struct ivy_ast_unit_package_node *)node;
	b_string *str = b_string_create();

	b_queue_iterator it = {0};
	b_queue_foreach (&it, &unit_package->n_ident) {
		struct ivy_token *tok = b_unbox(struct ivy_token, it.entry, t_entry);

		if (b_string_get_size(str, B_STRLEN_NORMAL) > 0) {
			b_string_append_cstr(str, ".");
		}

		b_string_append_cstr(str, tok->t_str);
	}

	printf("%s(%s)\n", ivy_ast_node_type_to_string(node->n_type), b_string_ptr(str));
	b_string_release(str);
}

static void init_state(struct parser_state *sp)
{
	struct unit_package_parser_state *state
		= (struct unit_package_parser_state *)sp;
	state->s_prev_token = IVY_KW_PACKAGE;
}

struct ast_node_type unit_package_node_ops = {
	.n_add_child = add_child,
        .n_print = print,
	.n_init_state = init_state,
	.n_state_size = sizeof(struct unit_package_parser_state),
	.n_node_size = sizeof(struct ivy_ast_unit_package_node),
	.n_symbol_parsers = {
		[IVY_SYM_DOT] = parse_dot,
	},
	.n_token_parsers = {
		[IVY_TOK_IDENT] = parse_ident,
		[IVY_TOK_LINEFEED] = parse_linefeed,
	}

};