#ifndef __SPL_AST_H__
#define __SPL_AST_H__

#include "../stage0/include/utils.h"
#include "spl_lexer.h"
#include "spl_tok.h"
#include <math.h>

typedef enum {
    SPL_AST_CONTAINER_MEMBER,
    SPL_AST_FN_DECL,
    SPL_AST_TYPE_DECL,
    SPL_AST_VAR_DECL,
    SPL_AST_CONST_DECL,
    SPL_AST_MEMBER_DECL,
    SPL_AST__COMPTIME_STMT,   /*不实现*/
    SPL_AST__DIRECTIVE_BLOCK, /*不实现*/

    SPL_AST_BLOCK,
    SPL_AST_BLOCK_ITEM,
    SPL_AST_EXPR,
    SPL_AST_TYPE_EXPR,
    SPL_AST_ATTR_LIST,
} spl_ast_node_kind_t;

typedef struct {
    const char *fname;
    int line;
    int col;
} spl_ast_loc_t;

struct spl_ast_node;
typedef struct spl_ast_node spl_ast_node_t;
typedef usize spl_ast_node_ref_t;
typedef VEC(spl_ast_node_ref_t) spl_ast_node_ref_vec_t;
struct spl_ast_node {
    spl_ast_node_kind_t kind;
    spl_ast_loc_t loc;
    union {
        spl_ast_node_ref_vec_t container_member;

        spl_ast_node_ref_vec_t attr_list;

        struct {
            spl_ast_node_ref_t attr_list;
            const char *name;
            spl_ast_node_ref_t param_list;
            spl_ast_node_ref_t type_expr;
            spl_ast_node_ref_t block;
        } fn_decl;
        spl_ast_node_ref_vec_t param_list;
        struct {
            spl_ast_node_ref_t attr_list;
            const char *name;
            spl_ast_node_ref_t type_expr;
        } param_decl;

        struct {
            spl_ast_node_ref_t attr_list;
            const char *name;
            enum {
                SPL_AST_TYPE_STRUCT,
                SPL_AST_TYPE_UNION,
                SPL_AST_TYPE_ENUM,
                SPL_AST_TYPE_TYPE_EXPR,
            };
            union {
                spl_ast_node_ref_t type_expr;
                spl_ast_node_ref_t aggregate_list;
            };
        } type_decl;
        struct {
            spl_ast_node_ref_t attr_list;
            const char *name;
            spl_ast_node_ref_t type_expr;
        } member_decl;

        struct {
            spl_ast_node_ref_t attr_list;
            const char *name;
            spl_ast_node_ref_t type_expr;
            spl_ast_node_ref_t expr;
        } var_decl;
        struct {
            spl_ast_node_ref_t attr_list;
            spl_ast_node_ref_t type_expr;
            const char *name;
            spl_ast_node_ref_t expr;
        } const_decl;

        spl_ast_node_ref_vec_t block;
        struct {
            enum {
                SPL_AST_IF_STATEMENT,
                SPL_AST_IFVAR_STATEMENT,
                SPL_AST_WHILE_STATEMENT,
                SPL_AST_LOOP_STATEMENT,
                SPL_AST_FOR_STATEMENT,
                SPL_AST_MATCH_STATEMENT,
                SPL_AST_RET_STATEMENT,
                SPL_AST_BREAK_STATEMENT,
                SPL_AST_CONTINUE_STATEMENT,
                SPL_AST_DEFER_STATEMENT,
                SPL_AST_VARDECL,
                SPL_AST_TYPEDECL,
                SPL_AST_EXPR_STATEMENT,
                SPL_PACED_EXPR,
            } kind;
            union {
                struct {
                    spl_ast_node_ref_t expr;
                    spl_ast_node_ref_t if_block;
                    spl_ast_node_ref_t else_block;
                } if_statement;
                struct {
                    spl_ast_node_ref_t packed_expr;
                    spl_ast_node_ref_t if_block;
                    spl_ast_node_ref_t else_block;
                } ifvar_statement;
                struct {
                    spl_ast_node_ref_t expr;
                    spl_ast_node_ref_t while_block;
                } while_statement;
                struct {
                    spl_ast_node_ref_t loop_block;
                } loop_statement;
                struct {
                    spl_ast_node_ref_vec_t expr_vec;
                    VEC(char *) ident_vec;
                    spl_ast_node_ref_t block;
                } for_statement;
                struct {
                    spl_ast_node_ref_t expr;
                    spl_ast_node_ref_vec_t paced_exprs;
                    spl_ast_node_ref_vec_t statements;
                } match_statement;
                struct {
                    spl_ast_node_ref_t expr;
                } ret_statement;
                struct {
                } break_statement;
                struct {
                } continue_statement;
                struct {
                    spl_ast_node_ref_t block_or_statement;
                } defer_statement;
                spl_ast_node_ref_t var_decl;
                spl_ast_node_ref_t type_decl;
                spl_ast_node_ref_t expr_statement;

                struct {
                    const char *ident;
                    const char *bind_ident;
                    spl_ast_node_ref_t expr;
                } packed_expr;
            };
        } block_item;

        struct {
            enum {
                SPL_AST_ASSIGN_EXPR,
                SPL_AST_BOOLOR_EXPR,
                SPL_AST_BOOLAND_EXPR,
                SPL_AST_BITOR_EXPR,
                SPL_AST_BITXOR_EXPR,
                SPL_AST_BITAND_EXPR,
                SPL_AST_CMPEQ_EXPR,
                SPL_AST_CMP_EXPR,
                SPL_AST_RANGE_EXPR,
                SPL_AST_SHIFT_EXPR,
                SPL_AST_ADD_EXPR,
                SPL_AST_MUL_EXPR,
                SPL_AST_PREFIX_EXPR,
                SPL_AST_POSTFIX_EXPR, /*left ref node*/
                SPL_AST_PRIMARY_EXPR, /*left ref node*/
            } op;
            struct {
                spl_ast_node_ref_t left;
                spl_ast_node_ref_t right;
            } op_expr;
        } expr;
        struct {
            enum {
                SPL_AST_CALL_EXPR,
                SPL_AST_IDENT_EXPR,
                SPL_AST_DEREF_EXPR,
                SPL_AST_INDEX_EXPR,
                SPL_AST_SLICE_EXPR,
                SPL_AST_AS_EXPR,
            } kind;
            union {
                spl_ast_node_ref_vec_t call_expr;
                const char *indet_expr;
                spl_ast_node_ref_t index_expr;
                struct {
                    spl_ast_node_ref_t begin;
                    spl_ast_node_ref_t end;
                } slice_expr;
                spl_ast_node_ref_t type_expr;
            };
        } postfix_expr;
        struct {
            enum {
                SPL_AST_INTEGER,
                SPL_AST_FLOAT,
                SPL_AST_CHAR_LIT,
                SPL_AST_STRING_LIT,
                SPL_AST_TRUE,
                SPL_AST_FALSE,
                SPL_AST_NULL,
                SPL_AST_IDENT,
                SPL_AST_ARGGREGATE_INIT,
                SPL_AST_EXPR_EXPR,
                SPL_AST_ARRAY_LIT,
                SPL_AST_BUILTIN_EXPR,
                SPL_AST_BLOCK_EXPR,
            };
            union {
                isize integer_expr;
                double float_expr;
                char char_lit_expr;
                const char *string_lit_expr;
                const char *ident;

                /*aggregate_init_item*/
                spl_ast_node_ref_vec_t aggregate_init_expr;

                spl_ast_node_ref_t expr;
                struct {
                    isize integer;
                    spl_ast_node_ref_t type_expr;
                    spl_ast_node_ref_vec_t expr_list;
                } array_lit_expr;
                struct {
                    const char *ident;
                    spl_ast_node_ref_vec_t expr_list;
                } builtin_expr;
                spl_ast_node_ref_t block_expr;
            };
        } primary_expr;
        struct {
            const char *ident;
            spl_ast_node_ref_t expr;
        } aggregate_init_item;

        struct {
            /* ASTERISK = 1, [] = 2, else = 0*/
            int pointer;
            /* if array_size != 0 then pointer == 2 */
            int array_size;
            enum {
                SPL_AST_BASE_TYPE_FN,
                SPL_AST_BASE_TYPE_PATH,
            } kind;

            const char *spl_base_type;
            union {
                spl_ast_node_ref_vec_t type_path;
                struct {
                    spl_ast_node_ref_t param_list;
                    spl_ast_node_ref_t type_expr;
                } fn_type;
            };
        } type_expr;
        struct {
            enum {
                SPL_AST_TYPE_VOID,
                SPL_AST_TYPE_BOOL,
                SPL_AST_TYPE_I8,
                SPL_AST_TYPE_U8,
                SPL_AST_TYPE_I16,
                SPL_AST_TYPE_U16,
                SPL_AST_TYPE_I32,
                SPL_AST_TYPE_U32,
                SPL_AST_TYPE_I64,
                SPL_AST_TYPE_U64,
                SPL_AST_TYPE_ISIZE,
                SPL_AST_TYPE_USIZE,
                SPL_AST_TYPE__F32, /*不实现*/
                SPL_AST_TYPE__F64, /*不实现*/
                SPL_AST_TYPE_PTR,
                SPL_AST_TYPE_ANY,
                SPL_AST_TYPE_IDENT,
            } kind;
            const char *ident;
        } type_atom;
    };
};

typedef VEC(spl_ast_node_t) spl_ast_node_vec_t;
typedef struct {
    int parsed;
    spl_tok_vec_t input;
    spl_ast_node_vec_t buckets;
    spl_ast_node_ref_t root;
} spl_ast_t;

void spl_ast_init(spl_ast_t *ast, const spl_tok_vec_t *tok_vec /*move*/);
void spl_ast_drop(spl_ast_t *ast);

void spl_ast_prase(spl_ast_t *ast);
void spl_ast_valid(spl_ast_t *ast);
void spl_ast_dump(spl_ast_t *ast, spl_ast_node_ref_t node);

#endif /* __SPL_AST_H__ */