tiny.smg.cli.dispatch

1const std = @import("std");2const pretty_usage = @import("pretty_usage");34const smg = @import("../root.zig");5const cli_about = smg.command.about;6const cli_analyze = smg.command.analyze;7const cli_batch = smg.command.batch;8const cli_between = smg.command.between;9const cli_blame = smg.command.blame;10const cli_calls = smg.command.calls;11const cli_check = smg.command.check;12const cli_concepts = smg.command.concepts;13const cli_context = smg.command.context;14const cli_diff = smg.command.diff;15const cli_exports = smg.command.exports;16const cli_impact = smg.command.impact;17const cli_init = smg.command.init;18const cli_mutate = smg.command.mutate;19const cli_node = smg.command.node;20const cli_output = smg.cli.output;21const cli_options = smg.cli.options;22const cli_query = smg.command.query;23const cli_rules = smg.command.rules;24const cli_scan = smg.command.scan;25const cli_search = smg.command.search;26const cli_session = smg.command.session;27const cli_status = smg.command.status;28const cli_usages = smg.command.usages;29const cli_validate = smg.command.validate;30const cli_watch = smg.command.watch;3132const help_data = @import("../help/root.zig");3334const click_help = help_data.click;35const help = help_data.help;3637const CommandRun = *const fn (std.mem.Allocator, std.mem.Allocator, []const []const u8, smg.Limits) anyerror!u8;38const PhasedCommandRun = *const fn (std.mem.Allocator, std.mem.Allocator, []const []const u8, smg.Limits, cli_session.PhaseBoundary) anyerror!u8;39const BatchCommandRun = *const fn (std.mem.Allocator, std.mem.Allocator, []const []const u8, smg.Limits) anyerror!u8;4041const CommandAction = union(enum) {42    regular: CommandRun,43    phased: PhasedCommandRun,44    batch: BatchCommandRun,45};4647pub const Build = struct {48    version: []const u8,49};5051pub fn diagnosedErrorExitCode(err: anyerror) ?u8 {52    if (err == error.NodeNotFound or err == error.AmbiguousName) return 1;53    return null;54}5556const EmptyHelp = enum {57    none,58    query,59    export_root,60    concept,61};6263const Command = struct {64    name: []const u8,65    action: CommandAction,66    empty_help: EmptyHelp = .none,67};6869const commands = [_]Command{70    .{ .name = "init", .action = .{ .regular = cli_init.run } },71    .{ .name = "scan", .action = .{ .regular = cli_scan.run } },72    .{ .name = "add", .action = .{ .regular = cli_mutate.add } },73    .{ .name = "link", .action = .{ .regular = cli_mutate.link } },74    .{ .name = "rm", .action = .{ .regular = cli_mutate.remove } },75    .{ .name = "unlink", .action = .{ .regular = cli_mutate.unlink } },76    .{ .name = "update", .action = .{ .regular = cli_mutate.update } },77    .{ .name = "status", .action = .{ .phased = cli_status.run } },78    .{ .name = "list", .action = .{ .phased = cli_node.list } },79    .{ .name = "show", .action = .{ .phased = cli_node.show } },80    .{ .name = "about", .action = .{ .phased = cli_about.run } },81    .{ .name = "calls", .action = .{ .phased = cli_calls.run } },82    .{ .name = "usages", .action = .{ .phased = cli_usages.run } },83    .{ .name = "impact", .action = .{ .phased = cli_impact.run } },84    .{ .name = "between", .action = .{ .phased = cli_between.run } },85    .{ .name = "query", .action = .{ .phased = cli_query.run }, .empty_help = .query },86    .{ .name = "export", .action = .{ .phased = cli_exports.run }, .empty_help = .export_root },87    .{ .name = "analyze", .action = .{ .phased = cli_analyze.run } },88    .{ .name = "overview", .action = .{ .phased = cli_analyze.overview } },89    .{ .name = "search", .action = .{ .phased = cli_search.run } },90    .{ .name = "index", .action = .{ .regular = cli_search.index } },91    .{ .name = "validate", .action = .{ .phased = cli_validate.run } },92    .{ .name = "rule", .action = .{ .phased = cli_rules.run } },93    .{ .name = "check", .action = .{ .phased = cli_check.command.run } },94    .{ .name = "refresh-check", .action = .{ .regular = cli_check.refresh.run } },95    .{ .name = "concept", .action = .{ .phased = cli_concepts.run }, .empty_help = .concept },96    .{ .name = "context", .action = .{ .phased = cli_context.run } },97    .{ .name = "blame", .action = .{ .phased = cli_blame.run } },98    .{ .name = "diff", .action = .{ .phased = cli_diff.run } },99    .{ .name = "watch", .action = .{ .regular = cli_watch.run } },100    .{ .name = "batch", .action = .{ .batch = cli_batch.run } },101};102103pub fn run(104    allocator: std.mem.Allocator,105    phases: std.mem.Allocator,106    args: []const []const u8,107    command_name: []const u8,108    build: Build,109    limits: smg.Limits,110    boundary: cli_session.PhaseBoundary,111) !u8 {112    if (args.len < 1) {113        try click_help.writeRoot(allocator, command_name);114        return 2;115    }116    if (std.mem.eql(u8, args[0], "--help")) {117        try click_help.writeRoot(allocator, command_name);118        return 0;119    }120    if (std.mem.eql(u8, args[0], "--version")) return try writeVersion(allocator, args, command_name, build.version);121    if (std.mem.startsWith(u8, args[0], "-")) {122        try cli_output.writeRootUsageError(allocator, command_name, try std.fmt.allocPrint(allocator, "No such option: {s}", .{cli_options.optionName(args[0])}));123        return 2;124    }125    const command = args[0];126    const rest = args[1..];127    const entry = commandEntry(command) orelse {128        try cli_output.writeRootUsageError(allocator, command_name, try std.fmt.allocPrint(allocator, "No such command '{s}'.", .{command}));129        return 2;130    };131    if (click_help.hasLongArg(rest)) {132        if (try click_help.writeForCommandPath(allocator, args)) |code| return code;133        return try writeCommandHelp(allocator, args);134    }135    return try runCommand(136        allocator,137        phases,138        entry,139        rest,140        limits,141        boundary,142    );143}144145fn commandEntry(command: []const u8) ?Command {146    for (commands) |entry| if (std.mem.eql(u8, entry.name, command)) return entry;147    return null;148}149150fn runCommand(151    allocator: std.mem.Allocator,152    phases: std.mem.Allocator,153    command: Command,154    args: []const []const u8,155    limits: smg.Limits,156    boundary: cli_session.PhaseBoundary,157) !u8 {158    if (args.len == 0) {159        if (try writeEmptyHelp(allocator, command.empty_help)) |code| return code;160    }161    return switch (command.action) {162        .regular => |run_regular| try run_regular(allocator, phases, args, limits),163        .phased => |run_phased| try run_phased(allocator, phases, args, limits, boundary),164        .batch => |run_batch| try run_batch(allocator, phases, args, limits),165    };166}167168fn writeEmptyHelp(allocator: std.mem.Allocator, empty_help: EmptyHelp) !?u8 {169    switch (empty_help) {170        .none => return null,171        .query => try click_help.writeQueryRoot(allocator),172        .export_root => try click_help.writeExportRoot(allocator),173        .concept => try click_help.writeConceptRoot(allocator),174    }175    return 2;176}177178fn writeVersion(allocator: std.mem.Allocator, args: []const []const u8, command_name: []const u8, version: []const u8) !u8 {179    for (args[1..]) |arg| {180        if (!std.mem.startsWith(u8, arg, "-")) break;181        if (std.mem.eql(u8, arg, "--help")) continue;182        try cli_output.writeRootUsageError(allocator, command_name, try std.fmt.allocPrint(allocator, "No such option: {s}", .{cli_options.optionName(arg)}));183        return 2;184    }185    return try cli_output.writeOutFmt(allocator, "smg, version {s}\n", .{version});186}187188fn writeCommandHelp(allocator: std.mem.Allocator, args: []const []const u8) !u8 {189    var stdout_buffer: [8192]u8 = undefined;190    var stdout_writer = std.Io.File.stdout().writer(std.Options.debug_io, &stdout_buffer);191    defer stdout_writer.interface.flush() catch {};192    var stderr_buffer: [8192]u8 = undefined;193    var stderr_writer = std.Io.File.stderr().writer(std.Options.debug_io, &stderr_buffer);194    defer stderr_writer.interface.flush() catch {};195    return try pretty_usage.writeCommandPathHelpOrError(196        allocator,197        &stdout_writer.interface,198        &stderr_writer.interface,199        help,200        args,201        .{202            .layout = .{ .width = 88 },203            .suggestions = .{ .enabled = true },204            .parent_fallback_help = true,205        },206    );207}208209test "top usage includes historical python command surface" {210    const rendered = try pretty_usage.renderHelpAlloc(std.testing.allocator, help, .{ .layout = .{ .width = 120 } });211    defer std.testing.allocator.free(rendered);212    for ([_][]const u8{213        "calls NAME",214        "diff [REF]",215        "context NAME",216        "blame NAME",217        "index",218        "batch",219    }) |needle| {220        try std.testing.expect(std.mem.indexOf(u8, rendered, needle) != null);221    }222}223224test "diagnosed resolver errors exit without process label" {225    try std.testing.expectEqual(226        @as(?u8, 1),227        diagnosedErrorExitCode(error.NodeNotFound),228    );229    try std.testing.expectEqual(230        @as(?u8, 1),231        diagnosedErrorExitCode(error.AmbiguousName),232    );233    try std.testing.expect(234        diagnosedErrorExitCode(error.MissingArgument) == null,235    );236}237238test "command usage resolves before project loading" {239    const diff_detail = help.commandHelp("diff").?;240    const diff_rendered = try pretty_usage.renderHelpAlloc(std.testing.allocator, diff_detail.*, .{ .layout = .{ .width = 120 } });241    defer std.testing.allocator.free(diff_rendered);242    try std.testing.expect(std.mem.indexOf(u8, diff_rendered, "usage: smg diff [REF] [--json]") != null);243    try std.testing.expect(std.mem.indexOf(u8, diff_rendered, "Compare the current graph against a git ref.") != null);244245    const batch_detail = help.commandHelp("batch").?;246    const batch_rendered = try pretty_usage.renderHelpAlloc(std.testing.allocator, batch_detail.*, .{ .layout = .{ .width = 120 } });247    defer std.testing.allocator.free(batch_rendered);248    try std.testing.expect(std.mem.indexOf(u8, batch_rendered, "usage: smg batch [--format text|json]") != null);249    try std.testing.expect(std.mem.indexOf(u8, batch_rendered, "Execute JSONL graph mutations from stdin.") != null);250    try std.testing.expect(std.mem.indexOf(u8, batch_rendered, "Read up to 64 MiB inclusive from stdin.") != null);251    try std.testing.expect(help.commandHelp("unknown") == null);252}253254test "scan usage omits removed jobs option" {255    const scan_detail = help.commandHelp("scan").?;256    const scan_rendered = try pretty_usage.renderHelpAlloc(257        std.testing.allocator,258        scan_detail.*,259        .{ .layout = .{ .width = 120 } },260    );261    defer std.testing.allocator.free(scan_rendered);262    try std.testing.expect(std.mem.indexOf(u8, scan_rendered, "--jobs") == null);263}264265test "one-shot queries and mutations have distinct phase actions" {266    for ([_][]const u8{267        "status",268        "list",269        "show",270        "about",271        "calls",272        "usages",273        "impact",274        "between",275        "query",276        "export",277        "analyze",278        "overview",279        "search",280        "validate",281        "rule",282        "check",283        "concept",284        "context",285        "blame",286        "diff",287    }) |name| {288        const entry = commandEntry(name).?;289        try std.testing.expectEqual(@as(std.meta.Tag(CommandAction), .phased), std.meta.activeTag(entry.action));290    }291    for ([_][]const u8{292        "init",293        "scan",294        "add",295        "link",296        "rm",297        "unlink",298        "update",299        "index",300        "refresh-check",301        "watch",302    }) |name| {303        const entry = commandEntry(name).?;304        try std.testing.expectEqual(@as(std.meta.Tag(CommandAction), .regular), std.meta.activeTag(entry.action));305    }306}307308test "batch is the sole SMG command action with input storage" {309    comptime {310        @stardustClaim(311            @import("alloc_phase").capacity.witness(@import("../batch/input/root.zig").Storage, "smg_batch_input_cli_composition"),312            null,313            null,314            null,315            null,316            null,317            null,318        );319    }320321    for (commands) |entry| {322        try std.testing.expectEqual(323            std.mem.eql(u8, entry.name, "batch"),324            std.meta.activeTag(entry.action) == .batch,325        );326    }327}

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