Compiler/ast.h

#ifndef MINI_C_AST_H
#define MINI_C_AST_H

#include <string>
#include <vector>
#include <cstdint>

/*──────────────────────────────
 * 0.  语义分析辅助
 *─────────────────────────────*/

// 在 type.h 中定义具体的 Type 类
struct Type;

// 用于记录 AST 节点在源文件中的行、列位置
struct SourceLoc {
    int line = 0;
    int col  = 0;
};

/*──────────────────────────────
 * 1.  词法‑语法层枚举
 *─────────────────────────────*/
//——单目运算符——
enum UnaryOp : int {
    op_address, op_deref, op_unary_plus, op_neg,
    op_bitnot, op_not
};

//——赋值运算符（含复合）——
enum AssignOp : int {
    op_assign, op_mul_assign, op_div_assign, op_mod_assign,
    op_add_assign, op_sub_assign, op_shl_assign, op_shr_assign,
    op_and_assign, op_xor_assign, op_or_assign
};

//——额外二元运算符——
enum BinaryOp : int {
    // ASCII 运算符直接用字面字符
    SHL = 256, SHR,
    LE, GE, EQ, NE
};

//——存储类别说明符——
enum StorageClass : int {
    SC_TYPEDEF, SC_EXTERN, SC_STATIC, SC_AUTO, SC_REGISTER
};

//——基本类型说明符——
enum TypeSpecifier : int {
    TS_VOID, TS_CHAR, TS_SHORT, TS_INT, TS_LONG,
    TS_FLOAT, TS_DOUBLE, TS_SIGNED, TS_UNSIGNED,
    TS_BOOL, TS_COMPLEX, TS_IMAGINARY,
    TS_TYPE_NAME            // typedef‑alias
};

//——struct / union——
enum StructUnionKind : int { SU_STRUCT, SU_UNION };

//——限定符 / 函数特性——
enum TypeQualifier : int { TQ_CONST, TQ_RESTRICT, TQ_VOLATILE };
enum FuncSpecifier : int { FS_INLINE };

/*──────────────────────────────
 * 2.  AST 结点与标签
 *─────────────────────────────*/
enum class ASTTag : std::uint16_t {
    /*── 基本 / 运算类 ──*/
    ID, CONST, STRING,
    UNARY, BINARY, LOGIC_AND, LOGIC_OR,
    COND, ASSIGN,
    ARRAY_REF, STRUCT_REF, FUNC_CALL,
    SIZEOF_EXPR, CAST_EXPR, COMPOUND_LITERAL,
    EXPR_LIST, ARG_LIST,
    PRE_INC,  PRE_DEC,      // ← 新增
    POST_INC, POST_DEC,     // ← 新增
    /*── 声明 & 类型 ──*/
    SPECIFIER, SPEC_LIST, DECLARATION, INIT_DECL, INIT_DECL_LIST,
    DECL_SPEC_LIST, DECLARATOR, POINTER_DECL,
    ARRAY_DECL, FUNC_DECL, OLD_FUNC_DECL,
    PARAM_DECL, PARAM_LIST, PARAM_LIST_ELIPS,
    ID_LIST, TYPE_NAME_NODE,
    ABS_DECL, ABS_ARRAY, ABS_ARRAY_CHILD,
    ABS_FUNC, ABS_FUNC_CHILD,
    POINTER, POINTER_CHAIN,
    TQ_LIST,

    /*── struct / enum ──*/
    SU_SPEC, STRUCT_DECL, STRUCT_DECL_LIST,
    STRUCT_DECLARATOR, STRUCT_DECLARATOR_LIST,
    BITFIELD,
    ENUM_SPEC, ENUM_LIST, ENUM_CONST,

    SPECQ_LIST,

    /*── 初始化 / 设计化初始化 ──*/
    INIT_EXPR, INIT_LIST_NODE, INIT_ITEM_LIST,
    DESIGNATOR_LIST, DESIGNATED_INIT,
    ARRAY_DESIGNATOR, FIELD_DESIGNATOR,

    /*── 语句 ──*/
    LABELED_ID_STMT, CASE_STMT, DEFAULT_STMT,
    COMPOUND_STMT, BLOCK_ITEM_LIST, BLOCK_DECL, BLOCK_STMT,
    EXPR_STMT,
    IF_STMT, SWITCH_STMT,
    WHILE_STMT, DO_WHILE_STMT,
    FOR_STMT, FOR_DECL_STMT,
    GOTO_STMT, CONTINUE_STMT, BREAK_STMT, RETURN_STMT,

    /*── 顶层 ──*/
    TRANSL_UNIT, DECL_STMT, FUNCTION_DEF, DECL_LIST
};

struct ASTNode {
    ASTTag                    tag;
    std::string               text;     // 标识符
    std::int64_t              ival{};   // 数值或枚举值
    bool                      flag{};
    SourceLoc                 loc{};    // 源代码行列信息
    Type*                     type{};   // 语义分析后填充的类型指针
    std::vector<ASTNode*>     kids;     // 子节点（按动作顺序）
    explicit ASTNode(ASTTag t, SourceLoc loc_ = SourceLoc{})
        : tag(t), loc(loc_) {}
};

/* 方便在别处直接引用整棵语法树 */
extern ASTNode *ast_root;

/*──────────────────────────────
 * 3.  表达式构造函数
 *─────────────────────────────*/
ASTNode* new_id_node   (const char* name,   SourceLoc loc);
ASTNode* new_const_node(const char* literal, SourceLoc loc);
ASTNode* new_string_node(const char* literal, SourceLoc loc);

ASTNode* new_array_ref_node(ASTNode* array, ASTNode* index, SourceLoc loc);
ASTNode* new_func_call_node(ASTNode* callee, ASTNode* arg_list, SourceLoc loc);
ASTNode* new_struct_ref_node(ASTNode* base, const char* field, bool is_ptr_op, SourceLoc loc);

ASTNode* new_post_inc_node(ASTNode* expr, SourceLoc loc);
ASTNode* new_post_dec_node(ASTNode* expr, SourceLoc loc);
ASTNode* new_pre_inc_node(ASTNode* expr, SourceLoc loc);
ASTNode* new_pre_dec_node(ASTNode* expr, SourceLoc loc);
ASTNode* new_compound_literal_node(ASTNode* type_name, ASTNode* init_list, SourceLoc loc);

ASTNode* new_arg_list(ASTNode* first_arg, SourceLoc loc);
ASTNode* append_arg_list(ASTNode* list, ASTNode* arg, SourceLoc loc);

ASTNode* new_unary_op_node(int op, ASTNode* operand, SourceLoc loc);
ASTNode* new_sizeof_node(ASTNode* target, bool is_type_name, SourceLoc loc);
ASTNode* new_cast_node(ASTNode* type_name, ASTNode* expr, SourceLoc loc);
ASTNode* new_binop_node  (int op, ASTNode* lhs, ASTNode* rhs, SourceLoc loc);    // op 可为字符或 BinaryOp
ASTNode* new_logical_and_node(ASTNode* lhs, ASTNode* rhs, SourceLoc loc);
ASTNode* new_logical_or_node(ASTNode* lhs, ASTNode* rhs, SourceLoc loc);
ASTNode* new_conditional_node(ASTNode* cond, ASTNode* then_expr, ASTNode* else_expr, SourceLoc loc);
ASTNode* new_assign_node (ASTNode* lhs, int op, ASTNode* rhs, SourceLoc loc);

ASTNode* new_expr_list(ASTNode* left, ASTNode* right, SourceLoc loc);

/*──────────────────────────────
 * 4.  声明 & 类型系统
 *─────────────────────────────*/
ASTNode* new_declaration(ASTNode* spec_list, ASTNode* init_list, SourceLoc loc);
ASTNode* new_spec_list(int spec, SourceLoc loc);                   // 建立首节点
ASTNode* append_spec_list(ASTNode* list, int spec, SourceLoc loc); // 追加

ASTNode* new_init_list(ASTNode* init_decl, SourceLoc loc);
ASTNode* append_init_list(ASTNode* list, ASTNode* init_decl, SourceLoc loc);
ASTNode* new_init_decl(ASTNode* declarator, ASTNode* initializer, SourceLoc loc);

ASTNode* new_struct_su_node(StructUnionKind su, const char* id, ASTNode* decl_list, SourceLoc loc);
ASTNode* new_sdecl_list(ASTNode* sdecl, SourceLoc loc);
ASTNode* append_sdecl_list(ASTNode* list, ASTNode* sdecl, SourceLoc loc);
ASTNode* new_struct_decl(ASTNode* specq_list, ASTNode* sdecl_list, SourceLoc loc);

ASTNode* new_specq_list(int spec_or_qual, SourceLoc loc);
ASTNode* append_specq_list(ASTNode* list, int spec_or_qual, SourceLoc loc);

ASTNode* new_sdeclarator_list(ASTNode* sdecltor, SourceLoc loc);
ASTNode* append_sdeclarator_list(ASTNode* list, ASTNode* sdecltor, SourceLoc loc);

ASTNode* new_bitfield_node(ASTNode* declarator /*nullable*/,
                           ASTNode* width_expr, SourceLoc loc);

ASTNode* new_enum_node(const char* id /*nullable*/, ASTNode* enumerator_list, SourceLoc loc);
ASTNode* new_enum_list(ASTNode* enumerator, SourceLoc loc);
ASTNode* append_enum_list(ASTNode* list, ASTNode* enumerator, SourceLoc loc);
ASTNode* new_enum_const(const char* id, ASTNode* value_expr /*nullable*/, SourceLoc loc);

ASTNode* new_tq_list(int tq, SourceLoc loc);
ASTNode* append_tq_list(ASTNode* list, int tq, SourceLoc loc);

ASTNode* new_declarator_node(ASTNode* pointer /*nullable*/, ASTNode* direct_decl, SourceLoc loc);
ASTNode* new_decl_ident(const char* id, SourceLoc loc);
ASTNode* new_array_decl(ASTNode* decl, ASTNode* tq_list /*nullable*/,
                        ASTNode* size_expr /*nullable*/,
                        bool is_static /*nullable*/,
                        bool is_star /*nullable*/, SourceLoc loc);
ASTNode* new_func_decl(ASTNode* decl, ASTNode* param_type_list /*nullable*/, SourceLoc loc);
ASTNode* new_oldstyle_func_decl(ASTNode* decl, ASTNode* id_list, SourceLoc loc);

ASTNode* new_pointer(ASTNode* tq_list /*nullable*/, SourceLoc loc);
ASTNode* prepend_pointer(ASTNode* tq_list /*nullable*/, ASTNode* existing, SourceLoc loc);

ASTNode* new_param_list(ASTNode* param_decl, SourceLoc loc);
ASTNode* append_param_list(ASTNode* list, ASTNode* param_decl, SourceLoc loc);
ASTNode* new_param_list_ellipsis(ASTNode* list, SourceLoc loc);
ASTNode* new_param_decl(ASTNode* decl_spec, ASTNode* declarator /*nullable*/, SourceLoc loc);

ASTNode* new_id_list(const char* id, SourceLoc loc);
ASTNode* append_id_list(ASTNode* list, const char* id, SourceLoc loc);

ASTNode* new_type_name(ASTNode* specq_list, ASTNode* abs_decl /*nullable*/, SourceLoc loc);

ASTNode* new_abs_decl(ASTNode* pointer /*nullable*/, ASTNode* direct_abs_decl /*nullable*/, SourceLoc loc);
ASTNode* new_abs_array(ASTNode* size_expr /*nullable*/, bool is_star /*nullable*/, SourceLoc loc);
ASTNode* new_abs_array_child(ASTNode* parent, ASTNode* size_expr /*nullable*/, bool is_star /*nullable*/, SourceLoc loc);
ASTNode* new_abs_func(ASTNode* param_type_list /*nullable*/, SourceLoc loc);
ASTNode* new_abs_func_child(ASTNode* parent, ASTNode* param_type_list /*nullable*/, SourceLoc loc);

/*──────────────────────────────
 * 5.  初始化 & 设计化初始化
 *─────────────────────────────*/
ASTNode* new_init_expr(ASTNode* expr, SourceLoc loc);
ASTNode* new_init_list_node(ASTNode* init_item_list, SourceLoc loc);
ASTNode* new_init_item_list(ASTNode* initializer, SourceLoc loc);
ASTNode* append_init_item(ASTNode* list, ASTNode* initializer, SourceLoc loc);

ASTNode* new_designated_init(ASTNode* designator_list, ASTNode* initializer, SourceLoc loc);
ASTNode* new_designator_list(ASTNode* designator, SourceLoc loc);
ASTNode* append_designator_list(ASTNode* list, ASTNode* designator, SourceLoc loc);
ASTNode* new_array_designator(ASTNode* const_expr, SourceLoc loc);
ASTNode* new_field_designator(const char* field, SourceLoc loc);

/*──────────────────────────────
 * 6.  语句
 *─────────────────────────────*/
ASTNode* new_labeled_stmt_id(const char* id, ASTNode* stmt, SourceLoc loc);
ASTNode* new_case_stmt(ASTNode* const_expr, ASTNode* stmt, SourceLoc loc);
ASTNode* new_default_stmt(ASTNode* stmt, SourceLoc loc);

ASTNode* new_compound_stmt(ASTNode* block_item_list /*nullable*/, SourceLoc loc);
ASTNode* new_block_item_list(ASTNode* item, SourceLoc loc);
ASTNode* append_block_item_list(ASTNode* list, ASTNode* item, SourceLoc loc);
ASTNode* new_block_decl(ASTNode* decl, SourceLoc loc);
ASTNode* new_block_stmt(ASTNode* stmt, SourceLoc loc);

ASTNode* new_expr_stmt(ASTNode* expr /*nullable*/, SourceLoc loc);

ASTNode* new_if_stmt(ASTNode* cond, ASTNode* then_stmt, ASTNode* else_stmt /*nullable*/, SourceLoc loc);
ASTNode* new_switch_stmt(ASTNode* expr, ASTNode* stmt, SourceLoc loc);

ASTNode* new_while_stmt(ASTNode* cond, ASTNode* body, SourceLoc loc);
ASTNode* new_do_while_stmt(ASTNode* body, ASTNode* cond, SourceLoc loc);
ASTNode* new_for_stmt(ASTNode* init, ASTNode* cond, ASTNode* iter /*nullable*/, ASTNode* body, SourceLoc loc);
ASTNode* new_for_decl_stmt(ASTNode* decl, ASTNode* cond_expr_stmt,
                           ASTNode* iter_expr /*nullable*/, ASTNode* body, SourceLoc loc);

ASTNode* new_goto_stmt(const char* id, SourceLoc loc);
ASTNode* new_continue_stmt(SourceLoc loc);
ASTNode* new_break_stmt(SourceLoc loc);
ASTNode* new_return_stmt  (ASTNode* expr /*nullable*/, SourceLoc loc);

/*──────────────────────────────
 * 7.  翻译单元 / 顶层
 *─────────────────────────────*/
ASTNode* new_translation_unit(ASTNode* prev, ASTNode* ext_decl, SourceLoc loc);
ASTNode* new_decl_stmt    (ASTNode* declaration, SourceLoc loc);
ASTNode* new_function_def (ASTNode* decl_spec, ASTNode* declarator,
                           ASTNode* decl_list /*nullable*/, ASTNode* compound_stmt,
                           SourceLoc loc);

ASTNode* new_declaration_list(ASTNode* decl, SourceLoc loc);
ASTNode* append_declaration_list(ASTNode* list, ASTNode* decl, SourceLoc loc);

// ★ 放在初始化区域旁边
ASTNode* append_designated_init(ASTNode* list,
                                ASTNode* designator_list,
                                ASTNode* initializer, SourceLoc loc);

// ★ 新增两种 overload，让既能接 enum 也能接节点
ASTNode* new_spec_list(ASTNode* spec_node, SourceLoc loc);
ASTNode* append_spec_list(ASTNode* list, ASTNode* spec_node, SourceLoc loc);


/* 让列表函数既接受 enum 也接受 ASTNode* -------------------*/
// ASTNode* (ASTNode* spec_node);
ASTNode* append_spec_list(ASTNode* list, ASTNode* spec_node, SourceLoc loc);

ASTNode* new_specq_list(ASTNode* node, SourceLoc loc);
ASTNode* append_specq_list(ASTNode* list, ASTNode* node, SourceLoc loc);

static std::string extractNameFromDeclarator(ASTNode* dtor) {
    std::string name = "";
    if (!dtor) return name;

    ASTNode* current = dtor;
    // 循环查找，直到找到带 text 的 DECLARATOR 或 ID
    while (current) {
        if (current->tag == ASTTag::DECLARATOR && !current->text.empty()) {
            name = current->text;
            break;
        }
        if (current->tag == ASTTag::ID) {
            name = current->text;
            break;
        }
        // 如果当前节点是 DECLARATOR 或 FUNC_DECL 且有子节点，则深入第一个子节点
        // （这部分逻辑需要根据 AST 结构仔细调整）
        if ((current->tag == ASTTag::DECLARATOR || current->tag == ASTTag::FUNC_DECL) && !current->kids.empty()) {
            // 特殊处理函数名情况：FUNC_DECL -> DECLARATOR "name"
            if (current->tag == ASTTag::FUNC_DECL && current->kids[0] && current->kids[0]->tag == ASTTag::DECLARATOR && !current->kids[0]->text.empty()) {
                name = current->kids[0]->text;
                break;
            }
            current = current->kids[0]; // 默认深入第一个孩子
        } else {
            break; // 无法继续深入
        }
    }
    return name;
}


#endif /* MINI_C_AST_H */