include/ast.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

#ifndef AMALGAM_AST_H
#define AMALGAM_AST_H

#include "std/buffer_view.h"
#include "std/buffer.h"
#include "std/misc.h"
#include "std/scoped_allocator.h"

typedef struct FunctionDecl FunctionDecl;
typedef struct FunctionCall FunctionCall;
typedef struct LhsExpr LhsExpr;
typedef struct Import Import;

typedef union {
    FunctionDecl *func_decl;
    FunctionCall *func_call;
    LhsExpr *lhs_expr;
    Import *import;
} AstValue;

typedef enum {
    AST_NONE,
    AST_FUNCTION_DECL,
    AST_FUNCTION_CALL,
    AST_LHS,
    AST_IMPORT
} AstType;

typedef struct {
    AstValue value;
    AstType type;
} Ast;

struct FunctionDecl {
    BufferView name;
    Buffer body;
};

struct FunctionCall {
    BufferView name;
};

struct LhsExpr {
    int isConst;
    BufferView var_name;
    Ast rhs_expr;
};

struct Import {
    BufferView path;
};

Ast ast_none();

CHECK_RESULT int funcdecl_init(FunctionDecl *self, ScopedAllocator *allocator);
CHECK_RESULT int funcdecl_add_to_body(FunctionDecl *self, Ast ast);

void funccall_init(FunctionCall *self, BufferView name);

void lhsexpr_init(LhsExpr *self, int isConst, BufferView var_name);

void import_init(Import *self, BufferView path);

#endif