#include "block.h"
#include "ctx.h"
#include "iterate.h"
#include "ivy/status.h"
#include "node.h"

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

struct msgh_parser_state {
	struct parser_state s_base;
	bool s_oneline;

	unsigned int s_i;
	unsigned int s_prev;
};

static struct token_parse_result parse_linefeed(
	struct ivy_parser *ctx, struct ivy_token *tok)
{
	struct msgh_parser_state *state
		= parser_get_state(ctx, struct msgh_parser_state);

	struct ivy_ast_msgh_node *msgh
		= (struct ivy_ast_msgh_node *)state->s_base.s_node;

	if (state->s_oneline) {
		if (msgh->n_body) {
			parser_pop_state(ctx, STATE_ADD_NODE_TO_PARENT);
			return PARSE_RESULT(IVY_OK, 0);
		}

		return PARSE_RESULT(IVY_ERR_BAD_SYNTAX, 0);
	}

	if (!msgh->n_sel) {
		return PARSE_RESULT(IVY_ERR_BAD_SYNTAX, 0);
	}

	struct block_parser_state *block_state
		= (struct block_parser_state *)parser_push_state(ctx, IVY_AST_BLOCK);
	block_state->s_terminator = IVY_SYM_BANG;

	return PARSE_RESULT(IVY_OK, 0);
}

static struct token_parse_result parse_bang(
	struct ivy_parser *ctx, struct ivy_token *tok)
{
	struct msgh_parser_state *state
		= parser_get_state(ctx, struct msgh_parser_state);

	struct ivy_ast_msgh_node *msgh
		= (struct ivy_ast_msgh_node *)state->s_base.s_node;

	if (!msgh->n_sel || state->s_oneline) {
		return PARSE_RESULT(IVY_ERR_BAD_SYNTAX, 0);
	}

	parser_pop_state(ctx, STATE_ADD_NODE_TO_PARENT);
	return PARSE_RESULT(IVY_OK, 0);
}

static struct token_parse_result parse_pipe(
	struct ivy_parser *ctx, struct ivy_token *tok)
{
	struct msgh_parser_state *state
		= parser_get_state(ctx, struct msgh_parser_state);

	struct ivy_ast_msgh_node *msgh
		= (struct ivy_ast_msgh_node *)state->s_base.s_node;

	if (!msgh->n_sel || state->s_oneline) {
		return PARSE_RESULT(IVY_ERR_BAD_SYNTAX, 0);
	}

	state->s_oneline = true;
	parser_push_state(ctx, IVY_AST_EXPR);
	return PARSE_RESULT(IVY_OK, 0);
}

static enum ivy_status add_child(
	struct ivy_ast_node *parent, struct ivy_ast_node *child)
{
	struct ivy_ast_msgh_node *msgh = (struct ivy_ast_msgh_node *)parent;

	if (child->n_type == IVY_AST_SELECTOR && !msgh->n_sel) {
		msgh->n_sel = (struct ivy_ast_selector_node *)child;
		return IVY_OK;
	}

	if (!msgh->n_body) {
		msgh->n_body = child;
		return IVY_OK;
	}

	return IVY_OK;
}

static void init_state(struct ivy_parser *ctx, struct parser_state *sp)
{
	struct msgh_parser_state *state = (struct msgh_parser_state *)sp;
	state->s_oneline = false;
	state->s_i = 0;
	state->s_prev = IVY_SYM_HYPHEN;

	parser_push_state(ctx, IVY_AST_SELECTOR);
}

static void collect_children(
	struct ivy_ast_node *node, struct ivy_ast_node_iterator *iterator)
{
	struct ivy_ast_msgh_node *msgh = (struct ivy_ast_msgh_node *)node;
	ast_node_iterator_enqueue_node(iterator, node, &msgh->n_sel->n_base);
	ast_node_iterator_enqueue_node(iterator, node, msgh->n_body);
}

struct ast_node_type msgh_node_ops = {
	.n_add_child = add_child,
	.n_init_state = init_state,
	.n_collect_children = collect_children,
	.n_state_size = sizeof(struct msgh_parser_state),
	.n_node_size = sizeof(struct ivy_ast_msgh_node),
	.n_token_parsers = {
		TOK_PARSER(LINEFEED, parse_linefeed),
	},
	.n_symbol_parsers = {
		SYM_PARSER(BANG, parse_bang),
		SYM_PARSER(PIPE, parse_pipe),
	},
};