From a5c24ace80a8629fbf177d966e93f82d911b3389 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Motiejus=20Jak=C5=A1tys?= <motiejus@jakstys.lt>
Date: Tue, 7 Jun 2022 05:59:00 +0300
Subject: [PATCH] writing to the db

---
 src/Group.zig        |  1 -
 src/unix2db/main.zig | 41 ++++++++++++++++++++++++++++++++++-------
 2 files changed, 34 insertions(+), 8 deletions(-)

diff --git a/src/Group.zig b/src/Group.zig
index 674818d..891d08a 100644
--- a/src/Group.zig
+++ b/src/Group.zig
@@ -81,7 +81,6 @@ pub fn fromReader(allocator: Allocator, reader: anytype) FromReaderError![]Group
         _ = it.next() orelse return error.InvalidRecord; // password
         const gids = it.next() orelse return error.InvalidRecord;
         const members_commas = it.next() orelse return error.InvalidRecord;
-        // the line must be exhaustive.
         if (it.next() != null) return error.InvalidRecord;
 
         const gid = fmt.parseInt(u32, gids, 10) catch return error.InvalidRecord;
diff --git a/src/unix2db/main.zig b/src/unix2db/main.zig
index 0970446..05ed2af 100644
--- a/src/unix2db/main.zig
+++ b/src/unix2db/main.zig
@@ -10,6 +10,8 @@ const GeneralPurposeAllocator = std.heap.GeneralPurposeAllocator;
 const flags = @import("../flags.zig");
 const User = @import("../User.zig");
 const Group = @import("../Group.zig");
+const Corpus = @import("../Corpus.zig");
+const DB = @import("../DB.zig");
 
 const usage =
     \\usage: turbonss-unix2db [options]
@@ -42,6 +44,7 @@ fn execute(
         .{ .name = "-h", .kind = .boolean },
         .{ .name = "--passwd", .kind = .arg },
         .{ .name = "--group", .kind = .arg },
+        .{ .name = "--output", .kind = .arg },
     }) catch {
         try stderr.writeAll(usage);
         return 1;
@@ -58,19 +61,37 @@ fn execute(
         return 1;
     }
 
-    const passwd = result.argFlag("--passwd") orelse "./passwd";
-    const group = result.argFlag("--group") orelse "./group";
+    const passwdFname = result.argFlag("--passwd") orelse "./passwd";
+    const groupFname = result.argFlag("--group") orelse "./group";
+    const outFile = result.argFlag("--output") orelse "./db.turbo";
 
-    var passwdFile = try fs.cwd().openFile(passwd, .{ .mode = .read_only });
+    var passwdFile = try fs.cwd().openFile(passwdFname, .{ .mode = .read_only });
     defer passwdFile.close();
-    var groupFile = try fs.cwd().openFile(group, .{ .mode = .read_only });
+    var groupFile = try fs.cwd().openFile(groupFname, .{ .mode = .read_only });
     defer groupFile.close();
 
     var users = try User.fromReader(allocator, passwdFile.reader());
+    defer for (users) |*user| user.deinit(allocator);
     var groups = try Group.fromReader(allocator, groupFile.reader());
+    defer for (groups) |*group| group.deinit(allocator);
 
-    try stderr.print("read {d} users\n", .{users.len});
-    try stderr.print("read {d} groups\n", .{groups.len});
+    var corpus = try Corpus.init(allocator, users, groups);
+    defer corpus.deinit();
+    var db = try DB.fromCorpus(allocator, &corpus);
+    defer db.deinit(allocator);
+
+    const fd = try os.open(outFile, os.O.WRONLY | os.O.TRUNC, 0644);
+    errdefer os.close(fd);
+
+    const len = try os.writev(fd, db.iov().constSlice());
+    try os.fsync(fd);
+    os.close(fd);
+
+    try stderr.print("total {d} bytes. groups={d} users={d}\n", .{
+        len,
+        users.len,
+        groups.len,
+    });
     return 0;
 }
 
@@ -84,5 +105,11 @@ test "invalid argument" {
 
     const exit_code = try execute(allocator, stderr.writer(), args[0..]);
     try testing.expectEqual(@as(u8, 1), exit_code);
-    try testing.expect(mem.startsWith(u8, stderr.items, "ERROR: unknown "));
+    try testing.expect(mem.startsWith(
+        u8,
+        stderr.items,
+        "ERROR: unknown option '--invalid-argument'",
+    ));
 }
+
+test "smoke test" {}