tiny.smg.command.status

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

1const std = @import("std");2const pretty_json = @import("pretty").json;34const smg = @import("../root.zig");5const options = smg.cli.options;6const output = smg.cli.output;7const session = smg.command.session;8const storage = smg.storage;9const table = smg.cli.table;10const validation = smg.cli.validation;1112pub fn run(allocator: std.mem.Allocator, _: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: session.PhaseBoundary) !u8 {13    if (try validation.rejectMissingOptionValues(allocator, args, &.{"--format"})) return 2;14    if (try validation.rejectUnexpectedOptionsPlain(allocator, "status", "[OPTIONS]", args, &.{ "--format", "--json", "--full" })) return 2;15    if (try validation.rejectInvalidChoiceOption(allocator, "status", "[OPTIONS]", args, "--format", &.{ "text", "json" })) return 2;16    const pos = try options.positional(allocator, args);17    if (pos.len != 0) {18        try output.writeClickUsageError(allocator, "status", "[OPTIONS]", try validation.unexpectedArgumentMessage(allocator, pos));19        return 2;20    }21    const root = try storage.requireRoot(allocator);22    try session.noteGraphAge(allocator, root);23    const graph_summary = try storage.summary.load(allocator, root, limits.storage);24    boundary.seal();25    defer boundary.beginTeardown();26    if (options.hasJson(args)) return try output.writeOut(try renderJson(allocator, graph_summary));27    var out: std.Io.Writer.Allocating = .init(allocator);28    try writeText(allocator, &out.writer, graph_summary);29    return try output.writeOut(out.written());30}3132pub fn renderJson(allocator: std.mem.Allocator, graph_summary: storage.summary.Graph) ![]const u8 {33    var out: std.Io.Writer.Allocating = .init(allocator);34    errdefer out.deinit();35    var writer = pretty_json.Writer.init(&out.writer, .minified);36    try writer.beginObject();37    try writer.objectField("nodes");38    try writer.write(graph_summary.nodes);39    try writer.objectField("edges");40    try writer.write(graph_summary.edges);41    try writer.objectField("node_types");42    try countJson(&writer, graph_summary.node_types);43    try writer.objectField("rel_types");44    try countJson(&writer, graph_summary.rel_types);45    try writer.endObject();46    try out.writer.writeByte('\n');47    return try out.toOwnedSlice();48}4950pub fn writeText(allocator: std.mem.Allocator, writer: *std.Io.Writer, graph_summary: storage.summary.Graph) !void {51    try writer.print("Nodes ({d})\n", .{graph_summary.nodes});52    try countText(allocator, writer, graph_summary.node_types, "type");53    try writer.print("\nEdges ({d})\n", .{graph_summary.edges});54    try countText(allocator, writer, graph_summary.rel_types, "relationship");55}5657fn countJson(writer: *pretty_json.Writer, counts: []const storage.summary.Count) !void {58    try writer.beginObject();59    for (counts) |item| {60        try writer.objectField(item.name);61        try writer.write(item.count);62    }63    try writer.endObject();64}6566fn countText(allocator: std.mem.Allocator, writer: *std.Io.Writer, counts: []const storage.summary.Count, header: []const u8) !void {67    const sorted = try allocator.dupe(storage.summary.Count, counts);68    std.mem.sort(storage.summary.Count, sorted, {}, countRowLess);69    try writeStatusCountTable(allocator, writer, sorted, header);70}7172fn countRowLess(_: void, a: storage.summary.Count, b: storage.summary.Count) bool {73    return std.mem.lessThan(u8, a.name, b.name);74}7576fn writeStatusCountTable(allocator: std.mem.Allocator, writer: *std.Io.Writer, count_rows: []const storage.summary.Count, name_header: []const u8) !void {77    const columns = [_]table.Column{78        .{ .header = name_header, .max_width = 40 },79        .{ .header = "count", .alignment = .right, .max_width = 12 },80    };81    var scratch_state = std.heap.ArenaAllocator.init(allocator);82    defer scratch_state.deinit();83    const scratch = scratch_state.allocator();84    var rows: std.ArrayList([]const []const u8) = .empty;85    for (count_rows) |row| {86        try table.appendRow(scratch, &rows, &.{87            row.name,88            try std.fmt.allocPrint(scratch, "{d}", .{row.count}),89        });90    }91    try table.write(allocator, writer, &columns, rows.items, null);92}9394test "status count table matches compact oracle shape" {95    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);96    defer arena.deinit();97    const allocator = arena.allocator();98    const counts = [_]storage.summary.Count{99        .{ .name = "function", .count = 2, .first = "app.helper" },100        .{ .name = "module", .count = 1, .first = "app" },101        .{ .name = "class", .count = 1, .first = "app.Engine" },102    };103    var out: std.Io.Writer.Allocating = .init(allocator);104    try countText(allocator, &out.writer, &counts, "type");105    try std.testing.expectEqualStrings(106        \\type      count107        \\--------  -----108        \\class         1109        \\function      2110        \\module        1111        \\112    , out.written());113}114115test "status json counts preserve graph traversal order" {116    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);117    defer arena.deinit();118    const allocator = arena.allocator();119    const graph_summary = storage.summary.Graph{120        .nodes = 4,121        .edges = 3,122        .node_types = &.{123            .{ .name = "module", .count = 2, .first = "app" },124            .{ .name = "function", .count = 2, .first = "app.main" },125        },126        .rel_types = &.{127            .{ .name = "contains", .count = 2, .first = "app\x00contains\x00app.main" },128            .{ .name = "calls", .count = 1, .first = "app.main\x00calls\x00lib.util" },129        },130    };131    try std.testing.expectEqualStrings(132        \\{"nodes":4,"edges":3,"node_types":{"module":2,"function":2},"rel_types":{"contains":2,"calls":1}}133        \\134    , try renderJson(allocator, graph_summary));135}