add arena allocator to tests

[ziggit] / git.zig

diff --git a/git.zig b/git.zig
index 0d42c2afe50e6cc448b412a30caa4559ebdde242..80e9aace87a0e8b0b28a05593101251170723ded 100644 (file)
--- a/git.zig
+++ b/git.zig
@@ -23,6 +23,11 @@ const Tree = std.ArrayList(TreeEntry);
 const Blob = struct {
     data: []u8,
 };
 const Blob = struct {
     data: []u8,
 };

+const ParsedObject = union(enum) {
+    c: Commit,
+    t: Tree,
+    b: Blob,
+};

 const Object = struct {
     kind: u3,
     data: []u8,
 const Object = struct {
     kind: u3,
     data: []u8,


@@ -33,7 +38,7 @@ const Object = struct {
             .data = data,
         };
     }
             .data = data,
         };
     }

-    pub fn parse(self: Object, alloc: Alloc) !union(enum) { c: Commit, t: Tree, b: Blob } {
+    pub fn parse(self: Object, alloc: Alloc) !ParsedObject {

         switch (self.kind) {
             1 => {
                 const authorOffset = std.mem.indexOf(u8, self.data, "author ") orelse return error.InvalidCommitFormat;
         switch (self.kind) {
             1 => {
                 const authorOffset = std.mem.indexOf(u8, self.data, "author ") orelse return error.InvalidCommitFormat;


@@ -279,6 +284,9 @@ const PackFile = struct {
                 }
                 counter += bitsSet;
 
                 }
                 counter += bitsSet;
 

+                if (dataSize == 0)
+                    dataSize = 0x10000;
+

                 std.mem.copyForwards(
                     u8,
                     result[resultCounter..result.len],
                 std.mem.copyForwards(
                     u8,
                     result[resultCounter..result.len],


@@ -294,6 +302,7 @@ const PackFile = struct {
                     result[resultCounter..result.len],
                     deltData[deltaDataOffset + counter + 1 .. deltaDataOffset + counter + 1 + dataSize],
                 );
                     result[resultCounter..result.len],
                     deltData[deltaDataOffset + counter + 1 .. deltaDataOffset + counter + 1 + dataSize],
                 );

+

                 resultCounter += dataSize;
                 counter += dataSize;
             }
                 resultCounter += dataSize;
                 counter += dataSize;
             }


@@ -331,6 +340,7 @@ const PackFile = struct {
 
         if (objectKind == 6) {
             const offset = try getOffset(reader);
 
         if (objectKind == 6) {
             const offset = try getOffset(reader);

+

             return try self.ofsDelta(
                 @intCast(offset.offset + objectSize.bytelen + offset.bytelen),
             );
             return try self.ofsDelta(
                 @intCast(offset.offset + objectSize.bytelen + offset.bytelen),
             );


@@ -345,7 +355,9 @@ const PackFile = struct {
             const pckReader = self.pckFile.reader().any();
             try self.pckFile.seekTo(offset);
 
             const pckReader = self.pckFile.reader().any();
             try self.pckFile.seekTo(offset);
 

-            return try self.readObject(pckReader);
+            const o = try self.readObject(pckReader);
+
+            return o;

         }
         return null;
     }
         }
         return null;
     }


@@ -398,7 +410,11 @@ const Repo = struct {
 };
 
 test "print HEAD" {
 };
 
 test "print HEAD" {

-    var repo = try Repo.open(std.testing.allocator, "../imgui/.git");
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");

     defer repo.close();
 
     const head = try repo.getHead();
     defer repo.close();
 
     const head = try repo.getHead();


@@ -407,7 +423,11 @@ test "print HEAD" {
 }
 
 test "parse idx" {
 }
 
 test "parse idx" {

-    var repo = try Repo.open(std.testing.allocator, "../imgui/.git");
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");

     defer repo.close();
 
     if (repo.packfile) |packfile| {
     defer repo.close();
 
     if (repo.packfile) |packfile| {


@@ -417,63 +437,163 @@ test "parse idx" {
 }
 
 test "get object" {
 }
 
 test "get object" {

-    var repo = try Repo.open(std.testing.allocator, "../imgui/.git");
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");

     defer repo.close();
 
     const head = try repo.getHead();
 
     if (try repo.getObject(head)) |o| {
     defer repo.close();
 
     const head = try repo.getHead();
 
     if (try repo.getObject(head)) |o| {

-        defer std.testing.allocator.free(o.data);
+        defer alloc.free(o.data);

 
 

-        std.debug.print("object: {s}\n", .{o.data});
+        std.debug.print("object({}): {s}\n", .{ o.kind, o.data });
+    }
+}
+
+test "parse commit" {
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");
+    defer repo.close();
+
+    const head = try repo.getHead();
+
+    if (try repo.getObject(head)) |o| {
+        defer alloc.free(o.data);
+
+        switch (try o.parse(alloc)) {
+            .c => |c| {
+                std.debug.print("commit:\n  tree: {x}\n  parent: {x}\n  author: {s}\n  committer: {s}\n  message: {s}\n", .{ c.tree, c.parent, c.author, c.committer, c.message });
+            },
+            else => {},
+        }

     }
 }
 
 test "get tree" {
     }
 }
 
 test "get tree" {

-    var repo = try Repo.open(std.testing.allocator, "../imgui/.git");
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");

     defer repo.close();
 
     if (try repo.getObject(0xceb2b2c62d6f8f3686dcacecd5be931839b02c77)) |o| {
     defer repo.close();
 
     if (try repo.getObject(0xceb2b2c62d6f8f3686dcacecd5be931839b02c77)) |o| {

-        defer std.testing.allocator.free(o.data);
+        defer alloc.free(o.data);

 
 

-        std.debug.print("tree: {s}\n", .{o.data});
+        // std.debug.print("tree({}): {any}\n", .{ o.kind, o.data });

     }
 }
     }
 }

-// test "list commits" {
-//     var repo = Repo.open(std.testing.allocator, "../imgui/.git");
-//     defer repo.close();

 
 

-//     const head = repo.getObject(repo.head);
-//     defer head.deinit();
+test "parse tree" {
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();

 
 

-//     var c = head.getCommit();
-//     for (0..3) |_| {
-//         std.debug.print("{}\n", .{c});
-//         c = c.parent;
-//     }
-// }
+    var repo = try Repo.open(alloc, "../imgui/.git");
+    defer repo.close();

 
 

-// test "tree" {
-//     var repo = Repo.open(std.testing.allocator, "../imgui/.git");
-//     defer repo.close();
+    if (try repo.getObject(0xceb2b2c62d6f8f3686dcacecd5be931839b02c77)) |o| {
+        defer alloc.free(o.data);

 
 

-//     const head = repo.getObject(repo.head);
-//     defer head.deinit();
+        switch (try o.parse(alloc)) {
+            .t => |t| {
+                defer t.deinit();
+                for (t.items) |treeEntry| {
+                    std.debug.print("{s} {s} {x}\n", .{ treeEntry.permissions, treeEntry.name, treeEntry.id });
+                }
+            },
+            else => {},
+        }
+    }
+}

 
 

-//     const commit = head.getCommit();
+test "list commits" {
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();

 
 

-//     std.debug.print("{}\n", .{commit.tree});
-// }
+    var repo = try Repo.open(alloc, "../imgui/.git");
+    defer repo.close();

 
 

-// test "blob" {
-//     var repo = Repo.open(std.testing.allocator, "../imgui/.git");
-//     defer repo.close();
+    const head = try repo.getHead();
+
+    var id = head;
+
+    for (0..3) |_| {
+        if (try repo.getObject(id)) |o| {
+            defer alloc.free(o.data);
+
+            switch (try o.parse(alloc)) {
+                .c => |c| {
+                    std.debug.print("commit {x}:\n  tree: {x}\n  parent: {x}\n  author: {s}\n  committer: {s}\n  message: {s}\n", .{ id, c.tree, c.parent, c.author, c.committer, c.message });
+                    id = c.parent;
+                },
+                else => {},
+            }
+        }
+    }
+}
+
+test "list blobs" {
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");
+    defer repo.close();
+
+    if (try repo.getObject(0xceb2b2c62d6f8f3686dcacecd5be931839b02c77)) |o| {
+        defer alloc.free(o.data);
+
+        switch (try o.parse(alloc)) {
+            .t => |t| {
+                defer t.deinit();
+                for (t.items) |treeEntry| {
+                    if (try repo.getObject(treeEntry.id)) |bo| {
+                        defer alloc.free(bo.data);
+
+                        if (treeEntry.permissions.len == 6) {
+                            std.debug.print("{s}: [{x} {}]{s}\n", .{ treeEntry.name, treeEntry.id, bo.data.len, bo.data[0..50] });
+                        } else {
+                            std.debug.print("[{s}]\n", .{treeEntry.name});
+                        }
+                    }
+                }
+            },
+            else => {},
+        }
+    }
+}
+
+test "basic frontend" {
+    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);
+    defer arena.deinit();
+    const alloc = arena.allocator();
+
+    var repo = try Repo.open(alloc, "../imgui/.git");
+    defer repo.close();
+
+    const head = try repo.getHead();

 
 

-//     const head = repo.getObject(repo.head);
-//     defer head.deinit();
+    var id = head;

 
 

-//     const commit = head.getCommit();
-//     const blob = repo.getBlob(commit.files[0].id);
+    for (0..3) |_| {
+        if (try repo.getObject(id)) |o| {
+            defer alloc.free(o.data);

 
 

-//     std.debug.print("{}\n", .{blob});
-// }
+            switch (try o.parse(alloc)) {
+                .c => |c| {
+                    std.debug.print("commit {x}:\n  tree: {x}\n  parent: {x}\n  author: {s}\n  committer: {s}\n  message: {s}\n", .{ id, c.tree, c.parent, c.author, c.committer, c.message });
+                    id = c.parent;
+                },
+                else => {},
+            }
+        }
+    }
+}