ast.h

#ifndef AST_H
#define AST_H
 
#include "types.h"
 
 
// AST creation functions
ASTNode *ast_create_program(void);
ASTNode *ast_create_function_decl(DataType return_type, const char *name, ASTNode **params, ASTNode *body);
ASTNode *ast_create_var_decl(DataType var_type, const char *name, ASTNode *initializer);
ASTNode *ast_create_parameter(DataType param_type, const char *name);
ASTNode *ast_create_compound_stmt(void);
ASTNode *ast_create_expression_stmt(ASTNode *expression);
ASTNode *ast_create_return_stmt(ASTNode *expression);
ASTNode *ast_create_if_stmt(ASTNode *condition, ASTNode *then_stmt, ASTNode *else_stmt);
ASTNode *ast_create_while_stmt(ASTNode *condition, ASTNode *body);
ASTNode *ast_create_for_stmt(ASTNode *init, ASTNode *condition, ASTNode *update, ASTNode *body);
ASTNode *ast_create_break_stmt(void);
ASTNode *ast_create_continue_stmt(void);
ASTNode *ast_create_binary_expr(TokenType operator, ASTNode *left, ASTNode *right);
ASTNode *ast_create_unary_expr(TokenType operator, ASTNode *operand);
ASTNode *ast_create_call_expr(const char *function_name);
ASTNode *ast_create_identifier(const char *name);
ASTNode *ast_create_number(int value);
ASTNode *ast_create_string(const char *value);
ASTNode *ast_create_assignment(const char *variable, ASTNode *value);
 
// AST manipulation functions
void ast_add_declaration(ASTNode *program, ASTNode *declaration);
void ast_add_parameter(ASTNode *function, ASTNode *parameter);
void ast_add_statement(ASTNode *compound_stmt, ASTNode *statement);
void ast_add_argument(ASTNode *call_expr, ASTNode *argument);
 
// AST utility functions
void ast_destroy(ASTNode *node);
// In the ast.c header file
void ast_print(ASTNode *node, int indent);
const char *ast_node_type_to_string(ASTNodeType type);
const char *data_type_to_string(DataType type);
DataType token_to_data_type(TokenType token);
 
#endif // AST_H