Compare commits
3 Commits
1ab966b853
...
4f0c43865f
Author | SHA1 | Date |
---|---|---|
Luna | 4f0c43865f | |
Luna | 19b79514b4 | |
Luna | 460055e186 |
|
@ -6,6 +6,11 @@ a vlang parser in zig
|
||||||
|
|
||||||
because i can
|
because i can
|
||||||
|
|
||||||
|
## variations
|
||||||
|
|
||||||
|
- `for` is split between `for` and `loop` because my fucking god i cant stand
|
||||||
|
having *four* different variations of `for`.
|
||||||
|
|
||||||
## how
|
## how
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
|
@ -20,4 +20,14 @@ fn main(a int) int {
|
||||||
|
|
||||||
a && b
|
a && b
|
||||||
a || b
|
a || b
|
||||||
|
|
||||||
|
loop {}
|
||||||
|
loop {
|
||||||
|
println('Ballse')
|
||||||
|
}
|
||||||
|
|
||||||
|
loop a > 2 {}
|
||||||
|
loop a > 2 {
|
||||||
|
println('skirts')
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
59
src/ast.zig
59
src/ast.zig
|
@ -97,12 +97,17 @@ pub const Expr = union(ExprType) {
|
||||||
Grouping: *Expr,
|
Grouping: *Expr,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub const IfBranch = std.ArrayList(*Stmt);
|
pub const Block = std.ArrayList(*Stmt);
|
||||||
|
|
||||||
pub const IfStmt = struct {
|
pub const IfStmt = struct {
|
||||||
condition: *Expr,
|
condition: *Expr,
|
||||||
then_branch: IfBranch,
|
then_branch: Block,
|
||||||
else_branch: ?IfBranch,
|
else_branch: ?Block,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub const LoopStmt = struct {
|
||||||
|
condition: ?*Expr,
|
||||||
|
then_branch: Block,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub const Stmt = union(enum) {
|
pub const Stmt = union(enum) {
|
||||||
|
@ -110,28 +115,46 @@ pub const Stmt = union(enum) {
|
||||||
Println: *Expr,
|
Println: *Expr,
|
||||||
|
|
||||||
If: IfStmt,
|
If: IfStmt,
|
||||||
|
Loop: LoopStmt,
|
||||||
|
|
||||||
pub fn mkPrintln(allocator: *std.mem.Allocator, expr: *Expr) !*Stmt {
|
pub fn mkPrintln(allocator: *std.mem.Allocator, expr: *Expr) !*Stmt {
|
||||||
var println = try allocator.create(Stmt);
|
var stmt = try allocator.create(Stmt);
|
||||||
println.* = Stmt{ .Println = expr };
|
stmt.* = Stmt{ .Println = expr };
|
||||||
return println;
|
return stmt;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn mkIfStmt(
|
pub fn mkIfStmt(
|
||||||
allocator: *std.mem.Allocator,
|
allocator: *std.mem.Allocator,
|
||||||
condition: *Expr,
|
condition: *Expr,
|
||||||
then: IfBranch,
|
then: Block,
|
||||||
else_branch: ?IfBranch,
|
else_branch: ?Block,
|
||||||
) !*Stmt {
|
) !*Stmt {
|
||||||
var println = try allocator.create(Stmt);
|
var stmt = try allocator.create(Stmt);
|
||||||
println.* = Stmt{
|
stmt.* = Stmt{
|
||||||
.If = IfStmt{
|
.If = IfStmt{
|
||||||
.condition = condition,
|
.condition = condition,
|
||||||
.then_branch = then,
|
.then_branch = then,
|
||||||
.else_branch = else_branch,
|
.else_branch = else_branch,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
return println;
|
|
||||||
|
return stmt;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mkLoop(
|
||||||
|
allocator: *std.mem.Allocator,
|
||||||
|
condition: ?*Expr,
|
||||||
|
then: Block,
|
||||||
|
) !*Stmt {
|
||||||
|
var stmt = try allocator.create(Stmt);
|
||||||
|
stmt.* = Stmt{
|
||||||
|
.Loop = LoopStmt{
|
||||||
|
.condition = condition,
|
||||||
|
.then_branch = then,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
return stmt;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -304,6 +327,20 @@ pub fn printStmt(ident: usize, stmt: *Stmt) void {
|
||||||
|
|
||||||
std.debug.warn(")\n");
|
std.debug.warn(")\n");
|
||||||
},
|
},
|
||||||
|
|
||||||
|
.Loop => |loop| {
|
||||||
|
std.debug.warn("(loop ");
|
||||||
|
if (loop.condition) |cond| {
|
||||||
|
printExpr(cond);
|
||||||
|
} else {
|
||||||
|
std.debug.warn("true");
|
||||||
|
}
|
||||||
|
std.debug.warn(" ");
|
||||||
|
|
||||||
|
printBlock(ident + 1, loop.then_branch, false);
|
||||||
|
std.debug.warn(")\n");
|
||||||
|
},
|
||||||
|
|
||||||
else => std.debug.warn("UnknownStmt-{}", @tagName(stmt.*)),
|
else => std.debug.warn("UnknownStmt-{}", @tagName(stmt.*)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -449,6 +449,7 @@ pub const Parser = struct {
|
||||||
fn parseStmt(self: *@This()) anyerror!*Stmt {
|
fn parseStmt(self: *@This()) anyerror!*Stmt {
|
||||||
return switch (self.peek().ttype) {
|
return switch (self.peek().ttype) {
|
||||||
.If => try self.parseIfStmt(),
|
.If => try self.parseIfStmt(),
|
||||||
|
.Loop => try self.parseLoop(),
|
||||||
.Println => try self.parsePrintln(),
|
.Println => try self.parsePrintln(),
|
||||||
|
|
||||||
// TODO make newlines tokens and consume newline?
|
// TODO make newlines tokens and consume newline?
|
||||||
|
@ -456,35 +457,35 @@ pub const Parser = struct {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Copy of parseBlock for if branches
|
/// Copy of parseBlock for blocks in statements
|
||||||
fn parseIfBranch(self: *@This()) !ast.IfBranch {
|
fn parseStmtBlock(self: *@This()) !ast.Block {
|
||||||
var branch = ast.IfBranch.init(self.allocator);
|
var block = ast.Block.init(self.allocator);
|
||||||
errdefer branch.deinit();
|
errdefer block.deinit();
|
||||||
|
|
||||||
_ = try self.consumeSingle(.LeftBrace);
|
_ = try self.consumeSingle(.LeftBrace);
|
||||||
|
|
||||||
while (self.peek().ttype != .RightBrace) {
|
while (self.peek().ttype != .RightBrace) {
|
||||||
var stmt = try self.parseDecl();
|
var stmt = try self.parseDecl();
|
||||||
ast.printNode(try self.mkStmt(stmt), 0);
|
ast.printNode(try self.mkStmt(stmt), 0);
|
||||||
try branch.append(stmt);
|
try block.append(stmt);
|
||||||
}
|
}
|
||||||
|
|
||||||
_ = try self.consumeSingle(.RightBrace);
|
_ = try self.consumeSingle(.RightBrace);
|
||||||
|
|
||||||
return branch;
|
return block;
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parseIfStmt(self: *@This()) !*Stmt {
|
fn parseIfStmt(self: *@This()) !*Stmt {
|
||||||
_ = try self.consumeSingle(.If);
|
_ = try self.consumeSingle(.If);
|
||||||
var condition = (try self.parseExpr()).Expr;
|
var condition = (try self.parseExpr()).Expr;
|
||||||
|
|
||||||
const then_branch = try self.parseIfBranch();
|
const then_branch = try self.parseStmtBlock();
|
||||||
|
|
||||||
var else_branch: ?ast.IfBranch = null;
|
var else_branch: ?ast.Block = null;
|
||||||
|
|
||||||
if (self.check(.Else)) {
|
if (self.check(.Else)) {
|
||||||
_ = try self.consumeSingle(.Else);
|
_ = try self.consumeSingle(.Else);
|
||||||
else_branch = try self.parseIfBranch();
|
else_branch = try self.parseStmtBlock();
|
||||||
}
|
}
|
||||||
|
|
||||||
return try Stmt.mkIfStmt(
|
return try Stmt.mkIfStmt(
|
||||||
|
@ -495,6 +496,22 @@ pub const Parser = struct {
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn parseLoop(self: *@This()) !*Stmt {
|
||||||
|
_ = try self.consumeSingle(.Loop);
|
||||||
|
var expr: ?*Expr = null;
|
||||||
|
var body: ast.Block = undefined;
|
||||||
|
|
||||||
|
// infinite loop
|
||||||
|
if (self.check(.LeftBrace)) {
|
||||||
|
body = try self.parseStmtBlock();
|
||||||
|
} else {
|
||||||
|
expr = (try self.parseExpr()).Expr;
|
||||||
|
body = try self.parseStmtBlock();
|
||||||
|
}
|
||||||
|
|
||||||
|
return try Stmt.mkLoop(self.allocator, expr, body);
|
||||||
|
}
|
||||||
|
|
||||||
fn parsePrintln(self: *@This()) !*Stmt {
|
fn parsePrintln(self: *@This()) !*Stmt {
|
||||||
_ = try self.consumeSingle(.Println);
|
_ = try self.consumeSingle(.Println);
|
||||||
|
|
||||||
|
|
|
@ -50,6 +50,7 @@ const keywords = [_][]const u8{
|
||||||
"false",
|
"false",
|
||||||
"None",
|
"None",
|
||||||
"println",
|
"println",
|
||||||
|
"loop",
|
||||||
};
|
};
|
||||||
|
|
||||||
const keyword_ttypes = [_]TokenType{
|
const keyword_ttypes = [_]TokenType{
|
||||||
|
@ -78,6 +79,7 @@ const keyword_ttypes = [_]TokenType{
|
||||||
.False,
|
.False,
|
||||||
.None,
|
.None,
|
||||||
.Println,
|
.Println,
|
||||||
|
.Loop,
|
||||||
};
|
};
|
||||||
|
|
||||||
fn getKeyword(keyword: []const u8) ?TokenType {
|
fn getKeyword(keyword: []const u8) ?TokenType {
|
||||||
|
|
|
@ -55,6 +55,7 @@ pub const TokenType = enum {
|
||||||
Enum,
|
Enum,
|
||||||
Fn,
|
Fn,
|
||||||
For,
|
For,
|
||||||
|
Loop,
|
||||||
Go,
|
Go,
|
||||||
Goto,
|
Goto,
|
||||||
If,
|
If,
|
||||||
|
|
Loading…
Reference in New Issue