initial impl: basic comptime if

StunxFS · StunxFS · commit 53b0992d7d5f · 2023-12-25T10:08:00.000-04:00
diff --git a/lib/rivet/src/ast/Decl.ri b/lib/rivet/src/ast/Decl.ri
@@ -1,5 +1,5 @@
-// Copyright (C) 2023-present Jose Mendoza - All rights reserved. Use of this 
-// source code is governed by an MIT license that can be found in the LICENSE 
+// Copyright (C) 2023-present Jose Mendoza - All rights reserved. Use of this
+// source code is governed by an MIT license that can be found in the LICENSE
 // file.
 
 import ../token;
@@ -8,6 +8,7 @@ import ../token;
 pub enum Decl {
     Empty(token.Pos),
     Comment(Comment),
+    ComptimeIf(ComptimeIf),
     Import {
         attributes: Attributes;
         is_public: bool;
@@ -154,6 +155,7 @@ pub enum Decl {
         return match self {
             .Empty(empty_pos) -> empty_pos,
             .Comment(comment) -> comment.pos,
+            .ComptimeIf(comptime_if) -> comptime_if.pos,
             .Import(import_d) -> import_d.pos,
             .Extern(extern_d) -> extern_d.pos,
             .Alias(alias_d) -> alias_d.pos,
@@ -171,6 +173,9 @@ pub enum Decl {
 
     pub func decls(self) -> ?[]Self {
         return match self {
+            .ComptimeIf(comptime_if) -> nodes_to_decls(
+                comptime_if.branches[comptime_if.branch_idx?].nodes
+            ),
             .Extern(extern_decl) -> extern_decl.decls,
             .Trait(trait_decl) -> trait_decl.decls,
             .Enum(enum_decl) -> enum_decl.decls,
diff --git a/lib/rivet/src/ast/Expr.ri b/lib/rivet/src/ast/Expr.ri
@@ -265,6 +265,7 @@ pub enum Expr < traits.Stringable {
         mut type: Type;
         mut defer_stmts: []Stmt.Defer;
     },
+    ComptimeIf(ComptimeIf),
     If {
         mut branches: []mut IfBranch;
         has_else: bool;
@@ -338,6 +339,7 @@ pub enum Expr < traits.Stringable {
             .Return(return_expr) -> return_expr.pos,
             .Throw(throw_expr) -> throw_expr.pos,
             .Block(block) -> block.pos,
+            .ComptimeIf(comptime_if) -> comptime_if.pos,
             .If(if_expr) -> if_expr.pos,
             .Match(match_expr) -> match_expr.pos,
             .Guard(guard) -> guard.pos
@@ -523,6 +525,22 @@ pub enum Expr < traits.Stringable {
                 sb.write_string(" }");
                 sb.to_string()
             },
+            .ComptimeIf(comptime_if) -> {
+                mut sb := strings.Builder.new();
+                for branch in comptime_if.branches {
+                    if branch.is_else {
+                        sb.write_string("else ");
+                    } else if branch.is_else_if {
+                        sb.write_string("else if ");
+                        sb.write_string(branch.cond.to_string());
+                    } else {
+                        sb.write_string("comptime if ");
+                        sb.write_string(branch.cond.to_string());
+                    }
+                    sb.write_fmt("{{ {} nodes }}", branch.nodes.len);
+                }
+                sb.to_string()
+            },
             .If(if_expr) -> {
                 mut sb := strings.Builder.new();
                 for branch in if_expr.branches {
diff --git a/lib/rivet/src/ast/Stmt.ri b/lib/rivet/src/ast/Stmt.ri
@@ -1,5 +1,5 @@
-// Copyright (C) 2023-present Jose Mendoza - All rights reserved. Use of this 
-// source code is governed by an MIT license that can be found in the LICENSE 
+// Copyright (C) 2023-present Jose Mendoza - All rights reserved. Use of this
+// source code is governed by an MIT license that can be found in the LICENSE
 // file.
 
 import ../token;
@@ -8,6 +8,7 @@ import ../token;
 pub enum Stmt {
     Empty(token.Pos),
     Comment(Comment),
+    ComptimeIf(ComptimeIf),
     Expr(Expr),
     VarDecl {
         mut lefts: []mut ObjectData;
@@ -50,6 +51,7 @@ pub enum Stmt {
         return match self {
             .Empty(empty_pos) -> empty_pos,
             .Comment(comment) -> comment.pos,
+            .ComptimeIf(comptime_if) -> comptime_if.pos,
             .Expr(expr) -> expr.position(),
             .VarDecl(var_decl) -> var_decl.pos,
             .While(while_stmt) -> while_stmt.pos,
diff --git a/lib/rivet/src/ast/mod.ri b/lib/rivet/src/ast/mod.ri
@@ -1,23 +1,9 @@
-// Copyright (C) 2023-present Jose Mendoza - All rights reserved. Use of this 
-// source code is governed by an MIT license that can be found in the LICENSE 
+// Copyright (C) 2023-present Jose Mendoza - All rights reserved. Use of this
+// source code is governed by an MIT license that can be found in the LICENSE
 // file.
 
 import ../token;
 
-pub enum Node {
-    Decl(Decl),
-    Stmt(Stmt),
-    Expr(Expr);
-
-    pub func position(self) -> token.Pos {
-        return match self {
-            .Decl(decl) -> decl.position(),
-            .Stmt(stmt) -> stmt.position(),
-            .Expr(expr) -> expr.position()
-        };
-    }
-}
-
 pub struct Comment {
     pub mut is_doc: bool;
     pub mut text: string;
@@ -34,37 +20,6 @@ pub struct Comment {
     }
 }
 
-#[boxed]
-pub struct ImportedSymbol {
-    pub name: string;
-    pub sym: Sym;
-    pub pos: token.Pos;
-    pub mut is_used: bool;
-}
-
-pub struct ImportedSymbols {
-    pub mut syms: []ImportedSymbol;
-
-    #[inline]
-    pub func add(mut self, name: string, sym: Sym, pos: token.Pos, is_used: bool := false) {
-        self.syms.push(ImportedSymbol(name, sym, pos, is_used));
-    }
-
-    pub func find(&self, name: string) -> ?ImportedSymbol {
-        for imported_sym in self.syms {
-            if name == imported_sym.name {
-                return imported_sym;
-            }
-        }
-        return none;
-    }
-
-    #[inline]
-    pub func exists(&self, name: string) -> bool {
-        return self.find(name) != none;
-    }
-}
-
 // Used in variable declarations/statements and guard expressions.
 #[boxed]
 pub struct ObjectData {
diff --git a/lib/rivet/src/parser/decls.ri b/lib/rivet/src/parser/decls.ri
@@ -97,6 +97,13 @@ extend Parser {
         mut pos := self.tok.pos;
         match {
             self.tok.kind == .Comment -> return .Comment(self.parse_comment()),
+            self.accept(.KwComptime) -> match self.tok.kind {
+                .KwIf -> return .ComptimeIf(self.parse_comptime_if(.Decl)),
+                else -> {
+                    report.error("invalid comptime construction", self.tok.pos);
+                    return .Empty(self.tok.pos);
+                }
+            },
             self.accept(.KwImport) -> {
                 mut has_custom_alias := false;
                 mut mod_path_pos := self.tok.pos;
diff --git a/lib/rivet/src/parser/exprs.ri b/lib/rivet/src/parser/exprs.ri
@@ -214,6 +214,13 @@ extend Parser {
 
     func parse_primary_expr(mut self) -> ast.Expr {
         mut expr: ast.Expr := match {
+            self.accept(.KwComptime) -> match self.tok.kind {
+                .KwIf -> .ComptimeIf(self.parse_comptime_if(.Expr)),
+                else -> {
+                    report.error("invalid comptime construction", self.tok.pos);
+                    .Empty(self.tok.pos)
+                }
+            },
             self.tok.kind in [
                 .KwTrue, .KwFalse, .Char, .Number, .String,
                 .KwNone, .KwSelf, .KwSelfTy
diff --git a/lib/rivet/src/parser/stmts.ri b/lib/rivet/src/parser/stmts.ri
@@ -9,6 +9,13 @@ extend Parser {
     func parse_stmt(mut self) -> ast.Stmt {
         return match {
             self.tok.kind == .Comment -> .Comment(self.parse_comment()),
+            self.accept(.KwComptime) -> match self.tok.kind {
+                .KwIf -> .ComptimeIf(self.parse_comptime_if(.Stmt)),
+                else -> {
+                    report.error("invalid comptime construction", self.tok.pos);
+                    .Empty(self.tok.pos)
+                }
+            },
             self.accept(.KwWhile) -> {
                 pos := self.prev_tok.pos;
                 mut is_inf := false;
diff --git a/lib/rivet/src/resolver/Register.ri b/lib/rivet/src/resolver/Register.ri
