tiny.chant.ast.stmt

1pub const stmt = @import("stmt.zig");

1const std = @import("std");2const expr = @import("expr.zig");3const variable = @import("variable.zig");45pub const Stmt = union(enum) {6    expression: *expr.Expr,7    declaration: []variable.Variable,8    compound: []*Stmt,9    if_stmt: If,10    for_stmt: For,11    while_stmt: While,12    do_stmt: While,13    label: Label,14    return_stmt: ?*expr.Expr,15    break_stmt,16    continue_stmt,17    empty,1819    pub const If = struct {20        condition: *expr.Expr,21        then_body: *Stmt,22        else_body: ?*Stmt,23    };2425    pub const For = struct {26        init: ?*Stmt,27        condition: ?*expr.Expr,28        step: ?*expr.Expr,29        body: *Stmt,30    };3132    pub const While = struct {33        condition: *expr.Expr,34        body: *Stmt,35    };3637    pub const Label = struct {38        name: []const u8,39        body: *Stmt,40    };41};4243test "statements nest" {44    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);45    defer arena.deinit();46    const allocator = arena.allocator();4748    const body = try allocator.create(Stmt);49    body.* = .empty;50    const loop = try allocator.create(Stmt);51    loop.* = .{ .for_stmt = .{ .init = null, .condition = null, .step = null, .body = body } };52    try std.testing.expect(loop.for_stmt.body.* == .empty);5354    const label = try allocator.create(Stmt);55    label.* = .{ .label = .{ .name = "again", .body = loop } };56    try std.testing.expectEqualStrings("again", label.label.name);57    try std.testing.expect(label.label.body.* == .for_stmt);58}