Fix docs.

fubark · Jul 10, 2024 · 54ae7b9 · 54ae7b9
1 parent 7bc9639
commit 54ae7b9
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/docs/gen-docs.cy b/docs/gen-docs.cy
@@ -451,6 +451,10 @@ func genDecl(decl Map) String:
         gen += "### `type $(decl['name'])`\n\n"
         for decl['funcs'] -> fdecl:
             gen += genFuncDecl(fdecl)
+    case 'trait_decl':
+        gen += "### `type $(decl['name']) trait`\n\n"
+        for decl['funcs'] -> fdecl:
+            gen += genFuncDecl(fdecl)
     case 'template':
         gen += genDecl(decl['child'])
     case 'specialization':

diff --git a/src/ast.zig b/src/ast.zig
@@ -1487,6 +1487,9 @@ pub const Encoder = struct {
                 try w.writeAll("*");
                 try self.write(w, node.cast(.expand_ptr).elem);
             },
+            .comptimeExpr => {
+                try self.write(w, node.cast(.comptimeExpr).child);
+            },
             else => {
                 try w.writeByte('<');
                 try w.writeAll(@tagName(node.type()));

diff --git a/src/builtins/cy.zig b/src/builtins/cy.zig
@@ -356,6 +356,15 @@ fn genDeclEntry(vm: *cy.VM, view: ast.AstView, decl: *ast.Node, state: *ParseCyb
             }
             try vm.mapSet(entry, try vm.retainOrAllocAstring("funcs"), funcs_);
         },
+        .trait_decl => {
+            const trait_decl = decl.cast(.trait_decl);
+            const funcs_ = try vm.allocEmptyListDyn();
+            for (trait_decl.funcs) |func_decl| {
+                const f = try genImplicitFuncDeclEntry(vm, view, func_decl, state);
+                try funcs_.asHeapObject().list.append(vm.alloc, f);
+            }
+            try vm.mapSet(entry, try vm.retainOrAllocAstring("funcs"), funcs_);
+        },
         else => {
             log.tracev("{}", .{decl.type()});
             return error.Unsupported;