tiny.smg.command.inspection

1const std = @import("std");2const smg = @import("../root.zig");3const cli = @import("../root.zig").cli;45pub fn run(allocator: std.mem.Allocator, args: []const []const u8) !u8 {6    std.debug.assert(cli.options.hasFlag(args, "--inspect"));7    const options = &.{ "--root", "--bytes", "--deadline-ms" };8    if (try cli.validation.rejectMissingOptionValues(allocator, args, options)) return 2;9    if (try cli.validation.rejectUnexpectedOptionsPlain(10        allocator,11        "context",12        "--inspect --root ABS [OPTIONS] TARGET...",13        args,14        &.{ "--inspect", "--root", "--bytes", "--deadline-ms" },15    )) return 2;16    const request = parse(args) catch |err| {17        try cli.output.writeClickUsageError(18            allocator,19            "context",20            "--inspect --root ABS [OPTIONS] TARGET...",21            @errorName(err),22        );23        return 2;24    };25    var inspected = try smg.context.inspect(26        allocator,27        request.root,28        request.targets[0..request.count],29        request.options,30    );31    defer inspected.deinit();32    const bytes = try smg.context.render.bounded(33        allocator,34        inspected.result,35        request.options.bytes,36    );37    defer allocator.free(bytes);38    std.debug.assert(bytes.len <= request.options.bytes);39    return cli.output.writeOut(bytes);40}4142const Request = struct {43    root: []const u8 = "",44    targets: [smg.context.model.targets_max][]const u8 = undefined,45    count: usize = 0,46    options: smg.context.InspectOptions = .{},47};4849fn parse(args: []const []const u8) !Request {50    var result = Request{};51    var seen: u8 = 0;52    var index: usize = 0;53    while (index < args.len) : (index += 1) {54        std.debug.assert(result.count <= result.targets.len);55        const arg = args[index];56        if (std.mem.eql(u8, arg, "--inspect")) continue;57        if (std.mem.startsWith(u8, arg, "--")) {58            if (index + 1 == args.len) return error.MissingInspectionOption;59            index += 1;60            try option(&result, &seen, arg, args[index]);61            continue;62        }63        if (result.count == result.targets.len) return error.InvalidInspectionTargets;64        result.targets[result.count] = arg;65        result.count += 1;66    }67    try smg.context.model.validate(result.root, result.targets[0..result.count], result.options);68    return result;69}7071fn option(result: *Request, seen: *u8, name: []const u8, value: []const u8) !void {72    const bit: u8 = if (std.mem.eql(u8, name, "--root"))73        174    else if (std.mem.eql(u8, name, "--bytes"))75        276    else if (std.mem.eql(u8, name, "--deadline-ms"))77        478    else79        return error.InvalidInspectionOption;80    std.debug.assert(seen.* <= 7);81    if (seen.* & bit != 0) return error.DuplicateInspectionOption;82    seen.* |= bit;83    switch (bit) {84        1 => result.root = value,85        2 => result.options.bytes = try std.fmt.parseInt(u32, value, 10),86        4 => result.options.deadline_ms = try std.fmt.parseInt(u16, value, 10),87        else => unreachable,88    }89}9091test "context inspection arguments expose immediate cooperative exhaustion" {92    const parsed = try parse(&.{93        "--inspect", "--root", "/tmp", "--deadline-ms", "0", "src/a.zig",94    });95    try std.testing.expectEqual(@as(u16, 0), parsed.options.deadline_ms);96    try std.testing.expectEqual(@as(usize, 1), parsed.count);97    try std.testing.expectError(error.InvalidInspectionRoot, parse(&.{ "--inspect", "a" }));98    try std.testing.expectError(error.InvalidInspectionLimit, parse(&.{99        "--inspect", "--root", "/tmp", "--bytes", "65537", "a",100    }));101}

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