tiny.smg.storage.project

1const std = @import("std");2const sys = @import("sys");3const repo = @import("repo");4const smg = @import("../root.zig");5const storage = @import("root.zig");67const paths = storage.paths;8const store = storage.store;9const text = smg.text;1011const fs_io = std.Options.debug_io;1213pub fn initProject(14    allocator: std.mem.Allocator,15    maybe_path: ?[]const u8,16    limits: smg.StorageLimits,17) ![]const u8 {18    const root = if (maybe_path) |path| try sys.fs.realPathAlloc(allocator, path) else try sys.fs.cwdAlloc(allocator);19    var opened = try store.open(allocator, root, limits);20    opened.close();21    try ensureGitExclude(allocator, root);22    return root;23}2425pub fn findRoot(allocator: std.mem.Allocator) !?[]const u8 {26    const start = try sys.fs.cwdAlloc(allocator);27    return try findRootFrom(allocator, start);28}2930pub fn initCurrentProject(31    allocator: std.mem.Allocator,32    limits: smg.StorageLimits,33) ![]const u8 {34    if (try findRoot(allocator)) |root| return root;35    const start = try sys.fs.cwdAlloc(allocator);36    const root = (try checkoutRootFrom(allocator, start)) orelse start;37    return try initProject(allocator, root, limits);38}3940fn findRootFrom(allocator: std.mem.Allocator, start: []const u8) !?[]const u8 {41    var repo_storage: [repo.default_capacity.storage_bytes]u8 = undefined;42    var resolver = try repo.Resolver.init(&repo_storage, repo.default_limits);43    resolver.activate();44    defer _ = resolver.deinit();45    const checkout_root = try checkoutRoot(allocator, &resolver, start);46    const store_dir = (try resolver.findStoreDirFrom(start, paths.smg_dir_name)) orelse return null;47    const root = std.fs.path.dirname(store_dir) orelse return null;48    if (checkout_root) |checkout| {49        if (!pathWithinRoot(checkout, root)) return null;50    }51    return try allocator.dupe(u8, root);52}5354fn checkoutRootFrom(allocator: std.mem.Allocator, start: []const u8) !?[]const u8 {55    var repo_storage: [repo.default_capacity.storage_bytes]u8 = undefined;56    var resolver = try repo.Resolver.init(&repo_storage, repo.default_limits);57    resolver.activate();58    defer _ = resolver.deinit();59    return try checkoutRoot(allocator, &resolver, start);60}6162fn checkoutRoot(63    allocator: std.mem.Allocator,64    resolver: *repo.Resolver,65    start: []const u8,66) !?[]const u8 {67    const git_path = (try resolver.findStoreDirFrom(start, ".git")) orelse return null;68    const root = std.fs.path.dirname(git_path) orelse return null;69    return try allocator.dupe(u8, root);70}7172fn pathWithinRoot(root: []const u8, path: []const u8) bool {73    if (std.mem.eql(u8, root, path)) return true;74    return path.len > root.len and75        std.mem.startsWith(u8, path, root) and76        std.fs.path.isSep(path[root.len]);77}7879pub fn requireRoot(allocator: std.mem.Allocator) ![]const u8 {80    return (try findRoot(allocator)) orelse error.NoProject;81}8283fn ensureGitExclude(allocator: std.mem.Allocator, root: []const u8) !void {84    const exclude = try std.fs.path.join(allocator, &.{ root, ".git", "info", "exclude" });85    const raw = text.readFile(allocator, exclude, 1024 * 1024) catch return;86    var lines = std.mem.splitScalar(u8, raw, '\n');87    while (lines.next()) |line| {88        if (std.mem.eql(u8, text.trim(line), ".smg/")) return;89    }90    var out: std.Io.Writer.Allocating = .init(allocator);91    errdefer out.deinit();92    try out.writer.writeAll(raw);93    if (raw.len != 0 and raw[raw.len - 1] != '\n') try out.writer.writeByte('\n');94    try out.writer.writeAll(".smg/\n");95    try text.writeFile(exclude, out.written());96}9798test "init project creates store directory" {99    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);100    defer arena.deinit();101    const allocator = arena.allocator();102    const root = try std.fmt.allocPrint(allocator, "/tmp/smg-storage-root-test-{x}", .{@as(u64, @intCast(@max(0, sys.time.realMilliTimestamp())))});103    defer sys.fs.deleteTree(root) catch {};104    try sys.fs.createDirPath(root);105    const initialized = try initProject(allocator, root, smg.default_limits.storage);106    try std.testing.expectEqualStrings(root, initialized);107    const store_dir = try std.fs.path.join(allocator, &.{ root, paths.smg_dir_name });108    try std.testing.expect(sys.fs.exists(store_dir));109    const database_path = try std.fs.path.join(allocator, &.{ store_dir, paths.database_file_name });110    try std.testing.expect(sys.fs.exists(database_path));111    var dir = try std.Io.Dir.openDirAbsolute(fs_io, store_dir, .{});112    dir.close(fs_io);113}114115test "linked checkout never inherits the main graph store" {116    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);117    defer arena.deinit();118    const allocator = arena.allocator();119    var temporary = std.testing.tmpDir(.{});120    defer temporary.cleanup();121    try temporary.dir.createDirPath(fs_io, "main/.git/worktrees/linked");122    try temporary.dir.createDirPath(fs_io, "main/.smg");123    try temporary.dir.createDirPath(fs_io, "linked/nested");124    const main_git = try temporary.dir.realPathFileAlloc(fs_io, "main/.git", allocator);125    const linked_git = try std.fs.path.join(allocator, &.{ main_git, "worktrees", "linked" });126    try temporary.dir.writeFile(fs_io, .{127        .sub_path = "linked/.git",128        .data = try std.fmt.allocPrint(allocator, "gitdir: {s}\n", .{linked_git}),129    });130    try temporary.dir.writeFile(fs_io, .{131        .sub_path = "main/.git/worktrees/linked/commondir",132        .data = "../..\n",133    });134    const nested = try temporary.dir.realPathFileAlloc(fs_io, "linked/nested", allocator);135    try std.testing.expectEqual(@as(?[]const u8, null), try findRootFrom(allocator, nested));136    try temporary.dir.createDirPath(fs_io, "linked/.smg");137    const linked = try temporary.dir.realPathFileAlloc(fs_io, "linked", allocator);138    try std.testing.expectEqualStrings(linked, (try findRootFrom(allocator, nested)).?);139}

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