#include "token.h"

#include <stdio.h>
#include <stddef.h>

#include "hmap.h"

static struct hmap *keywords = NULL;
static const char  *keys[]   = {
	   "and",
	   "or",
	   "not",
	   "for",
	   "in",
	   "break",
	   "endfor",
	   "true",
	   "false",
	   "if",
	   "elif",
	   "else",
	   "endif",
	   "extends",
	   "block",
	   "endblock",
	   NULL,
};
enum token_type values[] = {
	TOKEN_AND,
	TOKEN_OR,
	TOKEN_NOT,
	TOKEN_FOR,
	TOKEN_IN,
	TOKEN_BREAK,
	TOKEN_ENDFOR,
	TOKEN_TRUE,
	TOKEN_FALSE,
	TOKEN_IF,
	TOKEN_ELIF,
	TOKEN_ELSE,
	TOKEN_ENDIF,
	TOKEN_EXTENDS,
	TOKEN_BLOCK,
	TOKEN_ENDBLOCK,
	TOKEN_EOF,
};

const char *token_types[] = {
	[TOKEN_ILLEGAL] = "ILLEGAL",
	[TOKEN_EOF]     = "EOF",
	/* Identifiers/Literals */
	[TOKEN_IDENT]  = "IDENTIFIER",
	[TOKEN_INT]    = "INTEGER",
	[TOKEN_STRING] = "STRING",
	/* Operators */
	[TOKEN_ASSIGN]   = "=",
	[TOKEN_PLUS]     = "+",
	[TOKEN_MINUS]    = "-",
	[TOKEN_BANG]     = "!",
	[TOKEN_ASTERISK] = "*",
	[TOKEN_SLASH]    = "/",
	[TOKEN_LT]       = "<",
	[TOKEN_GT]       = ">",
	[TOKEN_LTE]      = "<=",
	[TOKEN_GTE]      = ">=",
	[TOKEN_EQ]       = "==",
	[TOKEN_NOTEQ]    = "!=",
	/* Keyword-like operators */
	[TOKEN_AND] = "and",
	[TOKEN_OR]  = "or",
	[TOKEN_NOT] = "not",
	/* Delimiters */
	[TOKEN_DOT]      = ".",
	[TOKEN_COMMA]    = ",",
	[TOKEN_LPAREN]   = "(",
	[TOKEN_RPAREN]   = ")",
	[TOKEN_LBRACE]   = "{",
	[TOKEN_RBRACE]   = "}",
	[TOKEN_LBRACKET] = "[",
	[TOKEN_RBRACKET] = "]",
	[TOKEN_POUND]    = "#",
	[TOKEN_PERCENT]  = "%",
	/* Keywords */
	[TOKEN_FOR]      = "for",
	[TOKEN_IN]       = "in",
	[TOKEN_BREAK]    = "break",
	[TOKEN_ENDFOR]   = "endfor",
	[TOKEN_TRUE]     = "true",
	[TOKEN_FALSE]    = "false",
	[TOKEN_IF]       = "if",
	[TOKEN_ELIF]     = "elif",
	[TOKEN_ELSE]     = "else",
	[TOKEN_ENDIF]    = "endif",
	[TOKEN_EXTENDS]  = "extends",
	[TOKEN_BLOCK]    = "block",
	[TOKEN_ENDBLOCK] = "endblock",
	/* The document content */
	[TOKEN_CONTENT] = "CONTENT",
};

void
token_init_keywords(void)
{
	if (keywords == NULL) {
		keywords = hmap_new();
		for (size_t i = 0; keys[i] != NULL; i++) {
			hmap_set(keywords, keys[i], values + i);
		}
	}
}

enum token_type
token_lookup_ident(const struct slice *ident)
{
	enum token_type *t = hmap_gets(keywords, ident);
	if (t) {
		return *t;
	}

	return TOKEN_IDENT;
}

extern inline const char *
token_type_print(enum token_type t)
{
	return token_types[t];
}

sds
token_string(struct token *token, sds str)
{
	const char *type     = token_type_print(token->type);
	sds         slicebuf = sdsempty();
	sdscatfmt(str, "TOKEN: type: %s, literal: %s", type,
	          slice_string(&token->literal, slicebuf));
	sdsfree(slicebuf);
	return str;
}

void
token_free_keywords(void)
{
	hmap_free(keywords);
}