tiny.smg.command.scan.changed

1const std = @import("std");23const smg = @import("../../root.zig");4const git = smg.git;5const text = smg.text;6const sys = @import("sys");78pub fn files(allocator: std.mem.Allocator, root: []const u8, ref: []const u8, invalid_ref: *bool) ![]const []const u8 {9    invalid_ref.* = false;10    var io_state = std.Io.Threaded.init(allocator, .{});11    defer io_state.deinit();12    if (!std.mem.eql(u8, ref, "HEAD")) {13        const commit_ref = try std.fmt.allocPrint(allocator, "{s}^{{commit}}", .{ref});14        const verify = std.process.run(allocator, io_state.io(), .{15            .argv = &.{ "git", "rev-parse", "--verify", "--quiet", commit_ref },16            .cwd = .{ .path = root },17            .expand_arg0 = .expand,18            .stdout_limit = .limited(4 * 1024 * 1024),19            .stderr_limit = .limited(1024 * 1024),20        }) catch return &.{};21        if (!processExitedZero(verify)) {22            invalid_ref.* = true;23            return &.{};24        }25    }26    var out: std.ArrayList([]const u8) = .empty;27    const diff_result = std.process.run(allocator, io_state.io(), .{28        .argv = &.{ "git", "diff", "--no-renames", "--name-only", ref },29        .cwd = .{ .path = root },30        .expand_arg0 = .expand,31        .stdout_limit = .limited(4 * 1024 * 1024),32        .stderr_limit = .limited(1024 * 1024),33    }) catch return &.{};34    if (processExitedZero(diff_result)) try append(allocator, root, &out, diff_result.stdout);35    const untracked_result = std.process.run(allocator, io_state.io(), .{36        .argv = &.{ "git", "ls-files", "--others", "--exclude-standard" },37        .cwd = .{ .path = root },38        .expand_arg0 = .expand,39        .stdout_limit = .limited(4 * 1024 * 1024),40        .stderr_limit = .limited(1024 * 1024),41    }) catch return &.{};42    if (processExitedZero(untracked_result)) try append(allocator, root, &out, untracked_result.stdout);43    std.mem.sort([]const u8, out.items, {}, stringLess);44    return try out.toOwnedSlice(allocator);45}4647fn processExitedZero(result: anytype) bool {48    return switch (result.term) {49        .exited => |code| code == 0,50        else => false,51    };52}5354fn append(allocator: std.mem.Allocator, root: []const u8, out: *std.ArrayList([]const u8), stdout: []const u8) !void {55    var lines = std.mem.splitScalar(u8, stdout, '\n');56    while (lines.next()) |line_raw| {57        const line = text.trim(line_raw);58        if (line.len == 0 or !supportedPath(line)) continue;59        const path = try std.fs.path.join(allocator, &.{ root, line });60        if (!text.list.containsString(out.items, path)) try out.append(allocator, path);61    }62}6364fn supportedPath(path: []const u8) bool {65    return std.mem.endsWith(u8, path, ".py") or66        std.mem.endsWith(u8, path, ".ts") or67        std.mem.endsWith(u8, path, ".tsx") or68        std.mem.endsWith(u8, path, ".js") or69        std.mem.endsWith(u8, path, ".jsx") or70        std.mem.endsWith(u8, path, ".mjs") or71        std.mem.endsWith(u8, path, ".cjs") or72        std.mem.endsWith(u8, path, ".zig") or73        std.mem.endsWith(u8, path, ".c") or74        std.mem.endsWith(u8, path, ".h") or75        std.mem.endsWith(u8, path, ".cpp") or76        std.mem.endsWith(u8, path, ".hpp") or77        std.mem.endsWith(u8, path, ".cc") or78        std.mem.endsWith(u8, path, ".cxx") or79        std.mem.endsWith(u8, path, ".hh") or80        std.mem.endsWith(u8, path, ".hxx") or81        std.mem.endsWith(u8, path, ".cu") or82        std.mem.endsWith(u8, path, ".cuh") or83        std.mem.endsWith(u8, path, ".metal") or84        std.mem.endsWith(u8, path, ".chic");85}8687fn stringLess(_: void, a: []const u8, b: []const u8) bool {88    return std.mem.lessThan(u8, a, b);89}9091fn replaceStoredGraph(92    allocator: std.mem.Allocator,93    value: smg.graph.Graph,94    root: []const u8,95) !void {96    var opened = try smg.storage.store.open(allocator, root, smg.default_limits.storage);97    defer opened.close();98    _ = try smg.storage.graph.replace(allocator, &opened.database, value);99}100101test "changed-file filter matches supported source extensions" {102    try std.testing.expect(supportedPath("src/app.py"));103    try std.testing.expect(supportedPath("src/app.tsx"));104    try std.testing.expect(supportedPath("src/kernel.cuh"));105    try std.testing.expect(supportedPath("src/library.chic"));106    try std.testing.expect(!supportedPath("README.md"));107    try std.testing.expect(!supportedPath("src/app.pyc"));108}109110test "changed files preserve a deleted source tombstone" {111    const fs_io = std.Options.debug_io;112    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);113    defer arena.deinit();114    const allocator = arena.allocator();115    var tmp = std.testing.tmpDir(.{});116    defer tmp.cleanup();117    const root = try tmp.dir.realPathFileAlloc(fs_io, ".", allocator);118    try tmp.dir.createDirPath(fs_io, "src");119    try tmp.dir.writeFile(fs_io, .{ .sub_path = "src/drop.py", .data = "def drop():\n    return 1\n" });120    try std.testing.expect((try git.run(allocator, root, &.{ "git", "init" }, 1024 * 1024)).ok);121    try std.testing.expect((try git.run(allocator, root, &.{ "git", "add", "src/drop.py" }, 1024 * 1024)).ok);122    try std.testing.expect((try git.run(allocator, root, &.{ "git", "-c", "user.name=smg", "-c", "user.email=smg@example.invalid", "commit", "-m", "base" }, 1024 * 1024)).ok);123    try tmp.dir.deleteFile(fs_io, "src/drop.py");124    try std.testing.expect((try git.run(allocator, root, &.{ "git", "add", "-u" }, 1024 * 1024)).ok);125    try std.testing.expect((try git.run(allocator, root, &.{ "git", "-c", "user.name=smg", "-c", "user.email=smg@example.invalid", "commit", "-m", "delete" }, 1024 * 1024)).ok);126    var invalid = false;127128    const changed = try files(allocator, root, "HEAD~1", &invalid);129130    try std.testing.expect(!invalid);131    try std.testing.expectEqual(@as(usize, 1), changed.len);132    try std.testing.expectEqualStrings(try std.fs.path.join(allocator, &.{ root, "src/drop.py" }), changed[0]);133    try std.testing.expect(!sys.fs.exists(changed[0]));134}135136test "changed files include both sides of a source rename" {137    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);138    defer arena.deinit();139    const allocator = arena.allocator();140    const root = try std.fmt.allocPrint(allocator, "/tmp/smg-changed-rename-test-{x}", .{@intFromPtr(&arena)});141    defer sys.fs.deleteTree(root) catch {};142    try sys.fs.createDirPath(try std.fs.path.join(allocator, &.{ root, "src" }));143    const before = try std.fs.path.join(allocator, &.{ root, "src/before.py" });144    const after = try std.fs.path.join(allocator, &.{ root, "src/after.py" });145    try text.writeFile(before, "def value():\n    return 1\n");146    try std.testing.expect((try git.run(allocator, root, &.{ "git", "init" }, 1024 * 1024)).ok);147    try std.testing.expect((try git.run(allocator, root, &.{ "git", "add", "src/before.py" }, 1024 * 1024)).ok);148    try std.testing.expect((try git.run(allocator, root, &.{ "git", "-c", "user.name=smg", "-c", "user.email=smg@example.invalid", "commit", "-m", "base" }, 1024 * 1024)).ok);149    var initial = smg.graph.init(allocator);150    _ = try smg.scan.scanPaths(allocator, &initial, root, &.{}, false, smg.default_limits);151    try replaceStoredGraph(allocator, initial, root);152    try std.testing.expect((try git.run(allocator, root, &.{ "git", "mv", "src/before.py", "src/after.py" }, 1024 * 1024)).ok);153    try std.testing.expect((try git.run(allocator, root, &.{ "git", "-c", "user.name=smg", "-c", "user.email=smg@example.invalid", "commit", "-m", "rename" }, 1024 * 1024)).ok);154    var invalid = false;155156    const changed = try files(allocator, root, "HEAD~1", &invalid);157    var loaded = try smg.storage.graph.load(allocator, allocator, root, smg.default_limits.storage);158    _ = try smg.scan.scanPaths(allocator, &loaded, root, changed, true, smg.default_limits);159    try replaceStoredGraph(allocator, loaded, root);160    const persisted = try smg.storage.graph.load(allocator, allocator, root, smg.default_limits.storage);161162    try std.testing.expect(!invalid);163    try std.testing.expectEqual(@as(usize, 2), changed.len);164    try std.testing.expect(text.list.containsString(changed, before));165    try std.testing.expect(text.list.containsString(changed, after));166    try std.testing.expect(!sys.fs.exists(before));167    try std.testing.expect(sys.fs.exists(after));168    try std.testing.expect(smg.graph.getNode(&persisted, "src.before") == null);169    try std.testing.expect(smg.graph.getNode(&persisted, "src.before.value") == null);170    try std.testing.expect(smg.graph.getNode(&persisted, "src.after.value") != null);171}

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