tiny.smg.command.between

1const std = @import("std");23const smg = @import("../root.zig");4const graph_mod = smg.graph;5const model = smg.model;6const query = smg.query;7const cli_json = smg.cli.json;8const cli_options = smg.cli.options;9const cli_output = smg.cli.output;10const cli_resolve = smg.cli.resolve;11const cli_session = smg.command.session;12const cli_validation = smg.cli.validation;13const view = smg.view;1415pub fn run(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: cli_session.PhaseBoundary) !u8 {16    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{"--format"})) return 2;17    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, "between", "[OPTIONS] A B", args, &.{"--format"})) return 2;18    if (try cli_validation.rejectInvalidChoiceOption(allocator, "between", "[OPTIONS] A B", args, "--format", &.{ "text", "json" })) return 2;19    const pos = try cli_options.positional(allocator, args);20    if (pos.len == 0) {21        try cli_output.writeClickUsageError(allocator, "between", "[OPTIONS] A B", "Missing argument 'A'.");22        return 2;23    }24    if (pos.len == 1) {25        try cli_output.writeClickUsageError(allocator, "between", "[OPTIONS] A B", "Missing argument 'B'.");26        return 2;27    }28    if (pos.len > 2) {29        try cli_output.writeClickUsageError(allocator, "between", "[OPTIONS] A B", try cli_validation.unexpectedArgumentMessage(allocator, pos[2..]));30        return 2;31    }32    const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);33    defer boundary.beginTeardown();34    const graph = loaded[1];35    const source = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);36    const target = try cli_resolve.graphNode(allocator, graph, pos[1], limits.suggestions);37    const path = try query.shortestPath(allocator, graph, source, target);38    const direct = try directEdges(allocator, graph, source, target);39    var out: std.Io.Writer.Allocating = .init(allocator);40    if (std.mem.eql(u8, cli_options.flagValue(args, "--format") orelse "", "json")) {41        try cli_json.writeBetween(&out.writer, source, target, path, direct);42        return try cli_output.writeOut(out.written());43    }44    try writeText(allocator, &out.writer, source, target, path, direct);45    return try cli_output.writeOut(out.written());46}4748pub fn directEdges(allocator: std.mem.Allocator, graph: graph_mod.Graph, source: []const u8, target: []const u8) ![]const model.Edge {49    var out: std.ArrayList(model.Edge) = .empty;50    for (try view.allEdges(graph, allocator)) |edge| {51        const forward = std.mem.eql(u8, edge.source, source) and std.mem.eql(u8, edge.target, target);52        const backward = std.mem.eql(u8, edge.source, target) and std.mem.eql(u8, edge.target, source);53        if (forward or backward) try out.append(allocator, edge);54    }55    return try out.toOwnedSlice(allocator);56}5758pub fn writeText(allocator: std.mem.Allocator, writer: *std.Io.Writer, source: []const u8, target: []const u8, path: ?[]const []const u8, direct: []const model.Edge) !void {59    if (path) |items| {60        var line: std.Io.Writer.Allocating = .init(allocator);61        defer line.deinit();62        try line.writer.writeAll("Path: ");63        for (items, 0..) |item, index| {64            if (index != 0) try line.writer.writeAll(" -> ");65            try line.writer.writeAll(item);66        }67        try writeWrappedPlainLine(writer, line.written(), 80);68    } else {69        try writer.print("No path between {s} and {s}\n", .{ source, target });70    }71    if (direct.len != 0) {72        try writer.writeAll("\nDirect edges:\n");73        for (direct) |edge| {74            var line: std.Io.Writer.Allocating = .init(allocator);75            defer line.deinit();76            try line.writer.print("  {s} --{s}--> {s}", .{ edge.source, edge.rel, edge.target });77            try writeWrappedPlainLine(writer, line.written(), 80);78        }79    }80}8182fn writeWrappedPlainLine(writer: *std.Io.Writer, line: []const u8, width: usize) !void {83    var start: usize = 0;84    while (line.len - start > width) {85        var split = start + width;86        while (split > start and line[split] != ' ') split -= 1;87        if (split == start) {88            split = start + width;89            try writer.writeAll(line[start..split]);90            try writer.writeByte('\n');91            start = split;92        } else {93            try writer.writeAll(line[start .. split + 1]);94            try writer.writeByte('\n');95            start = split + 1;96            while (start < line.len and line[start] == ' ') start += 1;97        }98    }99    try writer.writeAll(line[start..]);100    try writer.writeByte('\n');101}102103test "between renders python text contracts" {104    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);105    defer arena.deinit();106    const allocator = arena.allocator();107    var graph = graph_mod.init(allocator);108    for ([_][]const u8{109        "a",110        "b",111        "c",112        "d",113        "app.api.handler",114        "app.service.Service",115        "app.service.Service.load",116        "app.db.query",117        "really.long.source.module.component.Handler",118        "really.long.target.module.component.ServiceMethod",119    }) |name| try graph_mod.addNode(&graph, .{ .name = name, .type = model.NodeType.module });120    try graph_mod.addEdge(&graph, .{ .source = "a", .rel = model.RelType.calls, .target = "b" });121    try graph_mod.addEdge(&graph, .{ .source = "b", .rel = model.RelType.imports, .target = "c" });122    try graph_mod.addEdge(&graph, .{ .source = "app.api.handler", .rel = model.RelType.calls, .target = "app.service.Service" });123    try graph_mod.addEdge(&graph, .{ .source = "app.service.Service", .rel = model.RelType.calls, .target = "app.service.Service.load" });124    try graph_mod.addEdge(&graph, .{ .source = "app.service.Service.load", .rel = model.RelType.calls, .target = "app.db.query" });125    try graph_mod.addEdge(&graph, .{ .source = "really.long.source.module.component.Handler", .rel = model.RelType.calls, .target = "really.long.target.module.component.ServiceMethod" });126    const path = try query.shortestPath(allocator, graph, "a", "c");127    const long_path = try query.shortestPath(allocator, graph, "app.api.handler", "app.db.query");128    const long_direct = try directEdges(allocator, graph, "really.long.source.module.component.Handler", "really.long.target.module.component.ServiceMethod");129    var out: std.Io.Writer.Allocating = .init(allocator);130    try writeText(allocator, &out.writer, "a", "c", path, &.{});131    try writeText(allocator, &out.writer, "app.api.handler", "app.db.query", long_path, &.{});132    try writeText(allocator, &out.writer, "really.long.source.module.component.Handler", "really.long.target.module.component.ServiceMethod", try query.shortestPath(allocator, graph, "really.long.source.module.component.Handler", "really.long.target.module.component.ServiceMethod"), long_direct);133    try std.testing.expectEqualStrings("Path: a -> b -> c\n" ++134        "Path: app.api.handler -> app.service.Service -> app.service.Service.load -> \n" ++135        "app.db.query\n" ++136        "Path: really.long.source.module.component.Handler -> \n" ++137        "really.long.target.module.component.ServiceMethod\n" ++138        "\n" ++139        "Direct edges:\n" ++140        "  really.long.source.module.component.Handler --calls--> \n" ++141        "really.long.target.module.component.ServiceMethod\n", out.written());142}

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