// Copyright (C) 2023, Jakob Wakeling
// All rights reserved.

#include "parse.h"
#include "util/log.h"
#include "util/stack.h"

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>

char *ast_ks[] = {
	"AK_VOID", "AK_COMM", "AK_PIPE",
};

/* Initialise an AST node. */
extern ast *ast_init(ast_k kind) {
	ast *a = xcalloc(1, sizeof(*a));
	a->k = kind; return a;
}

/* Uninitialise an AST node and its children. */
extern void ast_free(ast *a) {
	if (a == NULL) { return; }
	ast_free(a->lc); ast_free(a->rc); stack_free(&a->c);
	free(a->rin.s); free(a->rout.s); free(a->rerr.s); free(a);
}

#define T (l->t) /* Current Token */

static ast *parse_comm(lex *l);
static ast *parse_pipe(lex *l, ast *lhs);

/* Parse a program. */
extern ast *parse(lex *l) {
	for (; T.k == TK_END; lex_next(l));
	if (T.k == TK_EOF) { return NULL; }

	if (T.k != TK_WORD) {
		log_warn("Unexpected \"%s\", was expecting a word", tok_ks[T.k]); return NULL;
	}

	ast *lhs = parse_comm(l);

	for (;;) switch (lex_next(l).k) {
	case TK_PIPE: { lhs = parse_pipe(l, lhs); } continue;
	default: {} return lhs;
	}
}

/* Parse a command statement. */
static ast *parse_comm(lex *l) {
	assert(T.k == TK_WORD);

	ast *a = ast_init(AK_COMM);
	a->c = stack_init(sizeof (char *), &free);

	/* Push each command argument onto the child stack */
	stack_push(&a->c, (a->s = lex_next(l).s));

	for (;;) switch (lex_peek(l).k) {
	case TK_WORD: { stack_push(&a->c, lex_next(l).s); } continue;
	case TK_RIN: {
		if (lex_next(l), T.k == TK_WORD) { a->rin.s = lex_next(l).s; }
		else { log_warn("Unexpected token \"%s\"", tok_ks[T.k]); }
	} continue;
	case TK_ROUT: {
		if (lex_next(l), T.k == TK_WORD) { a->rout.s = lex_next(l).s; a->rout.app = false; }
		else { log_warn("Unexpected token \"%s\"", tok_ks[T.k]); }
	} continue;
	case TK_RAPP: {
		if (lex_next(l), T.k == TK_WORD) { a->rout.s = lex_next(l).s; a->rout.app = true; }
		else { log_warn("Unexpected token \"%s\"", tok_ks[T.k]); }
	} continue;
	default: { stack_push(&a->c, NULL); } return a;
	}
}

/* Parse a pipe statement. */
static ast *parse_pipe(lex *l, ast *lhs) {
	ast *a = ast_init(AK_PIPE);
	a->lc = lhs; a->rc = parse_comm(l); return a;
}

/* Print parser debug output. */
extern void ast_debug(ast *a, u64 indent) {
	for (u64 i = 0; i != indent; i += 1) { printf("\t"); }

	printf("%s: %s", ast_ks[a->k], a->s);

	if (a->k == AK_COMM) {
		for (u64 i = 1; i != a->c.al - 1; i += 1) {
			printf(" %s", ((char **)a->c.a)[i]);
		}
		if (a->rin.s) { printf(" < %s", a->rin.s); }
		if (a->rout.s) { printf(a->rout.app ? " >> %s" : " > %s", a->rout.s); }
	}

	printf("\n");

	if (a->lc) { ast_debug(a->lc, indent + 1); }
	if (a->rc) { ast_debug(a->rc, indent + 1); }
}