tiny.smg.storage.files

1const pretty_json = @import("pretty").json;2const std = @import("std");3const sys = @import("sys");4const smg = @import("../root.zig");56const paths = smg.storage.paths;7const text = smg.text;89pub fn rulesPath(allocator: std.mem.Allocator, root: []const u8) ![]const u8 {10    return std.fs.path.join(allocator, &.{ root, paths.rules_relative_path });11}1213pub fn loadRules(14    allocator: std.mem.Allocator,15    root: []const u8,16    limits: smg.StorageLimits,17) ![]const std.json.Value {18    const path = try rulesPath(allocator, root);19    const raw = text.readFile(allocator, path, limits.max_rules_bytes) catch |err| switch (err) {20        error.FileNotFound => return &.{},21        else => return err,22    };23    return std.json.parseFromSliceLeaky([]std.json.Value, allocator, raw, .{}) catch |err| switch (err) {24        error.OutOfMemory => error.OutOfMemory,25        else => error.InvalidRules,26    };27}2829pub fn saveRules(allocator: std.mem.Allocator, root: []const u8, documents: []const []const u8) !void {30    const path = try rulesPath(allocator, root);31    var out: std.Io.Writer.Allocating = .init(allocator);32    errdefer out.deinit();33    var writer = pretty_json.Writer.init(&out.writer, .indent_2);34    const rules = try writer.array();35    for (documents) |document| try rules.raw(document);36    try rules.endLine();37    const tmp_path = try std.fmt.allocPrint(allocator, "{s}.tmp", .{path});38    try text.writeFile(tmp_path, out.written());39    try sys.fs.rename(tmp_path, path);40}4142test "rules are one owned JSON ledger rooted in the project" {43    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);44    defer arena.deinit();45    const allocator = arena.allocator();46    const root = try testRoot(allocator);47    defer sys.fs.deleteTree(root) catch {};48    try sys.fs.createDirPath(root);4950    const missing = try loadRules(allocator, root, smg.default_limits.storage);51    try std.testing.expectEqual(@as(usize, 0), missing.len);52    const path = try rulesPath(allocator, root);53    try std.testing.expect(std.mem.endsWith(u8, path, paths.rules_relative_path));54    try saveRules(allocator, root, &.{});55    try std.testing.expectEqualStrings(56        "[]\n",57        try text.readFile(allocator, path, 4096),58    );59    try saveRules(allocator, root, &.{ "{\"name\":\"deny\"}", "{\"name\":\"allow\"}" });60    const raw = try text.readFile(allocator, path, 4096);61    try std.testing.expectEqualStrings(62        "[\n  {\"name\":\"deny\"},\n  {\"name\":\"allow\"}\n]\n",63        raw,64    );65    const values = try loadRules(allocator, root, smg.default_limits.storage);66    try std.testing.expectEqual(@as(usize, 2), values.len);67    try std.testing.expectEqualStrings("deny", values[0].object.get("name").?.string);68    try std.testing.expectEqualStrings("allow", values[1].object.get("name").?.string);69}7071fn testRoot(allocator: std.mem.Allocator) ![]const u8 {72    return try std.fmt.allocPrint(allocator, "/tmp/smg-storage-files-test-{x}", .{@as(u64, @intCast(@max(0, sys.time.realMilliTimestamp())))});73}

1pub const files = @import("files.zig");