more types working

natecraddock · natecraddock · Sep 27, 2024 · Sep 21, 2024 · Sep 21, 2024 · Sep 21, 2024
commit 1a04340d7de2d6ede3697ecb5e82ec0863ea1ad3
diff --git a/src/luals-docgen.zig b/src/luals-docgen.zig
@@ -1,6 +1,6 @@
 const std = @import("std");
 
-pub const Doc = struct {
+pub const Definitions = struct {
     allocator: std.mem.Allocator,
     defined: std.StringHashMap(std.ArrayList(u8)),
 
@@ -16,15 +16,15 @@ pub const Doc = struct {
         self.defined.deinit();
     }
 
-    pub fn addFileHeader(_: *@This(), result: *std.ArrayList(u8)) !void {
-        try result.appendSlice(
-            \\---@meta
-            \\
-            \\--- This is an autogenerated file,
-            \\--- Do not modify
-            \\
-            \\
-        );
+    pub fn getFileHeader(_: *@This()) []const u8 {
+        return 
+        \\---@meta
+        \\
+        \\--- This is an autogenerated file,
+        \\--- Do not modify
+        \\
+        \\
+        ;
     }
 
     pub fn addEnum(self: *@This(), name: []const u8, comptime T: type) !void {
@@ -135,6 +135,20 @@ pub const Doc = struct {
         }
     }
 
+    /// Warning: file contents will be lost
+    pub fn overwriteFile(self: *@This(), abs_path: []const u8) !void {
+        var file = try std.fs.openFileAbsolute(abs_path, .{ .mode = .write_only });
+        try file.seekTo(0);
+        try file.writeAll(self.getFileHeader());
+
+        var iter = self.defined.valueIterator();
+        while (iter.next()) |val| {
+            try file.writeAll(val.items);
+            try file.writeAll("\n");
+        }
+        try file.setEndPos(try file.getPos());
+    }
+
     pub fn printDefined(self: *@This()) !void {
         var iter = self.defined.valueIterator();
         while (iter.next()) |val| {
@@ -144,7 +158,7 @@ pub const Doc = struct {
 };
 
 test "docgen" {
-    var docs = Doc.init(std.testing.allocator);
+    var docs = Definitions.init(std.testing.allocator);
     defer docs.deinit();
 
     const MyEnum = enum { asdf, fdsa, qwer, rewq };