tiny.smg.command.query

1const std = @import("std");23const smg = @import("../root.zig");4const exports = smg.exports;5const model = smg.model;6const query_mod = smg.query;7const text = smg.text;8const view = smg.view;9const cli_graph = smg.cli.graph;10const cli_json = smg.cli.json;11const cli_options = smg.cli.options;12const cli_output = smg.cli.output;13const cli_resolve = smg.cli.resolve;14const cli_session = smg.command.session;15const cli_subgraph = smg.cli.subgraph;16const cli_table = smg.cli.table;17const cli_validation = smg.cli.validation;1819pub fn run(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: cli_session.PhaseBoundary) !u8 {20    const kind = args[0];21    const rest = args[1..];22    if (!text.list.containsString(&.{ "deps", "callers", "path", "subgraph", "incoming", "outgoing" }, kind)) {23        try cli_output.writeClickUsageError(allocator, "query", "[OPTIONS] COMMAND [ARGS]...", try std.fmt.allocPrint(allocator, "No such command '{s}'.", .{kind}));24        return 2;25    }26    if (std.mem.eql(u8, kind, "deps")) {27        if (try cli_validation.validateQueryNameCommand(allocator, "query deps", "[OPTIONS] NAME", rest, &.{ "--depth", "--format", "--limit" }, &.{ "text", "json", "mermaid", "dot" })) return 2;28        const pos = try cli_options.positional(allocator, rest);29        const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);30        defer boundary.beginTeardown();31        const graph = loaded[1];32        const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);33        const names = try query_mod.transitiveDeps(allocator, graph, name, cli_options.depthValue(rest));34        return try cli_graph.writeNames(allocator, graph, name, names, try std.fmt.allocPrint(allocator, "Dependencies of {s}", .{name}), cli_options.flagValue(rest, "--format") orelse "text", cli_options.signedLimitValue(rest));35    }36    if (std.mem.eql(u8, kind, "callers")) {37        if (try cli_validation.validateQueryNameCommand(allocator, "query callers", "[OPTIONS] NAME", rest, &.{ "--depth", "--format", "--limit" }, &.{ "text", "json", "mermaid", "dot" })) return 2;38        const pos = try cli_options.positional(allocator, rest);39        const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);40        defer boundary.beginTeardown();41        const graph = loaded[1];42        const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);43        const names = try query_mod.transitiveCallers(allocator, graph, name, cli_options.depthValue(rest));44        return try cli_graph.writeNames(allocator, graph, name, names, try std.fmt.allocPrint(allocator, "Callers of {s}", .{name}), cli_options.flagValue(rest, "--format") orelse "text", cli_options.signedLimitValue(rest));45    }46    if (std.mem.eql(u8, kind, "path")) return try path(allocator, phases, rest, limits, boundary);47    if (std.mem.eql(u8, kind, "subgraph")) return try subgraph(allocator, phases, rest, limits, boundary);48    if (std.mem.eql(u8, kind, "incoming")) {49        if (try cli_validation.validateQueryNameCommand(allocator, "query incoming", "[OPTIONS] NAME", rest, &.{ "--rel", "--format", "--limit" }, &.{ "text", "json" })) return 2;50        const pos = try cli_options.positional(allocator, rest);51        const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);52        defer boundary.beginTeardown();53        const graph = loaded[1];54        const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);55        return try outputEdges(allocator, try view.incoming(graph, allocator, name, cli_options.flagValue(rest, "--rel")), cli_options.flagValue(rest, "--format") orelse "text", cli_options.signedLimitValue(rest));56    }57    if (std.mem.eql(u8, kind, "outgoing")) {58        if (try cli_validation.validateQueryNameCommand(allocator, "query outgoing", "[OPTIONS] NAME", rest, &.{ "--rel", "--format", "--limit" }, &.{ "text", "json" })) return 2;59        const pos = try cli_options.positional(allocator, rest);60        const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);61        defer boundary.beginTeardown();62        const graph = loaded[1];63        const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);64        return try outputEdges(allocator, try view.outgoing(graph, allocator, name, cli_options.flagValue(rest, "--rel")), cli_options.flagValue(rest, "--format") orelse "text", cli_options.signedLimitValue(rest));65    }66    unreachable;67}6869fn path(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: cli_session.PhaseBoundary) !u8 {70    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{"--format"})) return 2;71    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, "query path", "[OPTIONS] SOURCE TARGET", args, &.{"--format"})) return 2;72    if (try cli_validation.rejectInvalidChoiceOption(allocator, "query path", "[OPTIONS] SOURCE TARGET", args, "--format", &.{ "text", "json" })) return 2;73    const pos = try cli_options.positional(allocator, args);74    if (pos.len == 0) {75        try cli_output.writeClickUsageError(allocator, "query path", "[OPTIONS] SOURCE TARGET", "Missing argument 'SOURCE'.");76        return 2;77    }78    if (pos.len == 1) {79        try cli_output.writeClickUsageError(allocator, "query path", "[OPTIONS] SOURCE TARGET", "Missing argument 'TARGET'.");80        return 2;81    }82    if (pos.len > 2) {83        try cli_output.writeClickUsageError(allocator, "query path", "[OPTIONS] SOURCE TARGET", try cli_validation.unexpectedArgumentMessage(allocator, pos[2..]));84        return 2;85    }86    const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);87    defer boundary.beginTeardown();88    const graph = loaded[1];89    const source = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);90    const target = try cli_resolve.graphNode(allocator, graph, pos[1], limits.suggestions);91    const found_path = (try query_mod.shortestPath(allocator, graph, source, target)) orelse {92        try cli_output.writeErrFmt(allocator, "Error: no path from {s} to {s}\n", .{ source, target });93        return 1;94    };95    if (std.mem.eql(u8, cli_options.flagValue(args, "--format") orelse "", "json")) return try cli_json.stringArray(allocator, found_path);96    var out: std.Io.Writer.Allocating = .init(allocator);97    try writePathText(&out.writer, found_path);98    return try cli_output.writeOut(out.written());99}100101fn subgraph(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: cli_session.PhaseBoundary) !u8 {102    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{ "--depth", "--format", "--limit" })) return 2;103    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, "query subgraph", "[OPTIONS] NAME", args, &.{ "--depth", "--format", "--full", "--limit" })) return 2;104    if (try cli_validation.rejectInvalidChoiceOption(allocator, "query subgraph", "[OPTIONS] NAME", args, "--format", &.{ "text", "json", "mermaid", "dot" })) return 2;105    if (try cli_validation.rejectInvalidIntegerOption(allocator, "query subgraph", "[OPTIONS] NAME", args, "--depth")) return 2;106    if (try cli_validation.rejectInvalidIntegerOption(allocator, "query subgraph", "[OPTIONS] NAME", args, "--limit")) return 2;107    const pos = try cli_options.positional(allocator, args);108    if (pos.len == 0) {109        try cli_output.writeClickUsageError(allocator, "query subgraph", "[OPTIONS] NAME", "Missing argument 'NAME'.");110        return 2;111    }112    if (pos.len > 1) {113        try cli_output.writeClickUsageError(allocator, "query subgraph", "[OPTIONS] NAME", try cli_validation.unexpectedArgumentMessage(allocator, pos[1..]));114        return 2;115    }116    const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);117    defer boundary.beginTeardown();118    const graph = loaded[1];119    const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);120    const depth = cli_options.signedOptionValue(args, "--depth", 2);121    const sub = try query_mod.subgraph(allocator, graph, name, cli_options.unsignedTraversalDepth(depth), false);122    const fmt = cli_options.flagValue(args, "--format") orelse "text";123    if (!std.mem.eql(u8, fmt, "text")) return try cli_graph.write(allocator, sub, fmt);124    if (cli_options.hasFlag(args, "--full")) {125        const cap = if (cli_options.flagValue(args, "--limit")) |_| cli_options.signedLimitValue(args) else @as(i64, 50);126        if (cap > 0 and sub.nodes.items.len > @as(usize, @intCast(cap))) {127            const pruned = try cli_subgraph.prune(allocator, sub, @intCast(cap));128            var out: std.Io.Writer.Allocating = .init(allocator);129            try out.writer.writeAll(try exports.toText(allocator, pruned));130            try out.writer.print("(showing {d} of {d} nodes -- use --limit 0 for all)\n", .{ cap, sub.nodes.items.len });131            return try cli_output.writeOut(out.written());132        }133        return try cli_graph.write(allocator, sub, "text");134    }135    const limit = if (cli_options.flagValue(args, "--limit")) |_| cli_options.signedLimitValue(args) else @as(i64, 10);136    var out: std.Io.Writer.Allocating = .init(allocator);137    try cli_subgraph.writeSummary(allocator, &out.writer, sub, name, depth, limit);138    return try cli_output.writeOut(out.written());139}140141fn outputEdges(allocator: std.mem.Allocator, edges: []const model.Edge, fmt: []const u8, limit: i64) !u8 {142    if (std.mem.eql(u8, fmt, "json")) return try cli_json.edgeArray(allocator, edges);143    var out: std.Io.Writer.Allocating = .init(allocator);144    try writeEdgesText(allocator, &out.writer, edges, limit);145    return try cli_output.writeOut(out.written());146}147148fn writePathText(writer: *std.Io.Writer, path_items: []const []const u8) !void {149    for (path_items, 0..) |item, index| {150        if (index != 0) try writer.writeAll(" -> ");151        try writer.writeAll(item);152    }153    try writer.writeByte('\n');154}155156fn writeEdgesText(allocator: std.mem.Allocator, writer: *std.Io.Writer, edges: []const model.Edge, limit: i64) !void {157    if (edges.len == 0) {158        try writer.writeAll("(none)\n");159        return;160    }161    const displayed = cli_options.slicedLen(edges.len, limit);162    const columns = [_]cli_table.Column{163        .{ .header = "source", .max_width = 48 },164        .{ .header = "rel", .max_width = 16 },165        .{ .header = "target", .max_width = 48 },166    };167    var scratch_state = std.heap.ArenaAllocator.init(allocator);168    defer scratch_state.deinit();169    const scratch = scratch_state.allocator();170    var rows: std.ArrayList([]const []const u8) = .empty;171    for (edges[0..displayed]) |edge| try cli_table.appendRow(scratch, &rows, &.{ edge.source, edge.rel, edge.target });172    try cli_table.writeJsonOption(allocator, writer, &columns, rows.items, edges.len, "--format json");173}174175test "query path text joins nodes with arrows" {176    var out: std.Io.Writer.Allocating = .init(std.testing.allocator);177    defer out.deinit();178    try writePathText(&out.writer, &.{ "app.main", "app.helper", "lib.util" });179    try std.testing.expectEqualStrings("app.main -> app.helper -> lib.util\n", out.written());180}181182test "query edge text renders empty and limited tables" {183    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);184    defer arena.deinit();185    const allocator = arena.allocator();186    var empty: std.Io.Writer.Allocating = .init(allocator);187    try writeEdgesText(allocator, &empty.writer, &.{}, 10);188    try std.testing.expectEqualStrings("(none)\n", empty.written());189    var out: std.Io.Writer.Allocating = .init(allocator);190    try writeEdgesText(allocator, &out.writer, &.{191        .{ .source = "app.main", .rel = model.RelType.calls, .target = "app.helper" },192        .{ .source = "app.main", .rel = model.RelType.imports, .target = "lib.util" },193    }, 1);194    try std.testing.expect(std.mem.indexOf(u8, out.written(), "source") != null);195    try std.testing.expect(std.mem.indexOf(u8, out.written(), "app.main") != null);196    try std.testing.expect(std.mem.indexOf(u8, out.written(), "showing 1 of 2") != null);197    try std.testing.expect(std.mem.indexOf(u8, out.written(), "--format json") != null);198}

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