#include "token.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <inttypes.h>

static const char* NAMES[] = {
    [TOK_FN] = "fn", [TOK_LET] = "let", [TOK_VAR] = "var", [TOK_IF] = "if",
    [TOK_ELSE] = "else", [TOK_WHILE] = "while", [TOK_FOR] = "for", [TOK_IN] = "in", [TOK_RETURN] = "return",
    [TOK_STRUCT] = "struct", [TOK_TYPE] = "type", [TOK_ENUM] = "enum", [TOK_IMPL] = "impl",
    [TOK_MATCH] = "match",
    [TOK_I64] = "i64", [TOK_F64] = "f64", [TOK_BOOL] = "bool", [TOK_STR] = "str", [TOK_VOID] = "void",
    [TOK_INT_LIT] = "整数", [TOK_FLOAT_LIT] = "浮点数", [TOK_STR_LIT] = "字符串",
    [TOK_TRUE] = "true", [TOK_FALSE] = "false",
    [TOK_IDENT] = "标识符", [TOK_UNDERSCORE] = "_",
    [TOK_PLUS] = "+", [TOK_MINUS] = "-", [TOK_STAR] = "*",
    [TOK_SLASH] = "/", [TOK_PERCENT] = "%",
    [TOK_EQ_EQ] = "==", [TOK_BANG_EQ] = "!=",
    [TOK_LT] = "<", [TOK_GT] = ">", [TOK_LT_EQ] = "<=", [TOK_GT_EQ] = ">=",
    [TOK_AND_AND] = "&&", [TOK_PIPE_PIPE] = "||", [TOK_BANG] = "!",
    [TOK_ARROW] = "->", [TOK_TO] = "to", [TOK_MATCH_ARROW] = "=>",
    [TOK_PLUS_EQ] = "+=", [TOK_MINUS_EQ] = "-=", [TOK_STAR_EQ] = "*=", [TOK_SLASH_EQ] = "/=",
    [TOK_LPAREN] = "(", [TOK_RPAREN] = ")",
    [TOK_LBRACE] = "{", [TOK_RBRACE] = "}",
    [TOK_LBRACKET] = "[", [TOK_RBRACKET] = "]",
    [TOK_COMMA] = ",", [TOK_COLON] = ":", [TOK_SEMICOLON] = ";",
    [TOK_ASSIGN] = "=",
    [TOK_DOT] = ".", [TOK_COLON_COLON] = "::",
    [TOK_EOF] = "EOF", [TOK_ERROR] = "错误",
};

const char* tok_name(TokenKind kind) {
    return NAMES[kind];
}

bool tok_is_type(TokenKind kind) {
    return kind == TOK_I64 || kind == TOK_F64 || kind == TOK_BOOL || kind == TOK_STR || kind == TOK_VOID;
}

int64_t tok_int_value(const Token* tok) {
    if (tok->length <= 0 || tok->length >= 32) return 0;
    char buf[32];
    memcpy(buf, tok->start, tok->length);
    buf[tok->length] = '\0';
    return strtoll(buf, NULL, 10);
}

double tok_float_value(const Token* tok) {
    if (tok->length <= 0 || tok->length >= 64) return 0.0;
    char buf[64];
    memcpy(buf, tok->start, tok->length);
    buf[tok->length] = '\0';
    return strtod(buf, NULL);
}