tiny.smg.command.context

1const std = @import("std");23const smg = @import("../root.zig");4const context = smg.context;5const cli_options = smg.cli.options;6const cli_output = smg.cli.output;7const cli_resolve = smg.cli.resolve;8const cli_session = smg.command.session;9const cli_validation = smg.cli.validation;1011pub fn run(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: cli_session.PhaseBoundary) !u8 {12    if (cli_options.hasFlag(args, "--inspect")) return smg.command.inspection.run(allocator, args);13    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{ "--tokens", "--format" })) return 2;14    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, "context", "[OPTIONS] NAME", args, &.{ "--tokens", "--format", "--with-source" })) return 2;15    if (try cli_validation.rejectInvalidChoiceOption(allocator, "context", "[OPTIONS] NAME", args, "--format", &.{ "text", "json" })) return 2;16    if (try cli_validation.rejectInvalidIntegerOption(allocator, "context", "[OPTIONS] NAME", args, "--tokens")) return 2;17    const pos = try cli_options.positional(allocator, args);18    if (pos.len == 0) {19        try cli_output.writeClickUsageError(allocator, "context", "[OPTIONS] NAME", "Missing argument 'NAME'.");20        return 2;21    }22    if (pos.len > 1) {23        try cli_output.writeClickUsageError(allocator, "context", "[OPTIONS] NAME", try cli_validation.unexpectedArgumentMessage(allocator, pos[1..]));24        return 2;25    }26    _ = phases;27    const root = try smg.storage.requireRoot(allocator);28    try cli_session.noteGraphAge(allocator, root);29    var opened = try smg.storage.store.openRead(allocator, root, limits.storage);30    defer opened.close();31    const name = try cli_resolve.storedNode(32        allocator,33        root,34        &opened.reader,35        pos[0],36        limits.suggestions,37    );38    const result = try context.buildFromReader(39        allocator,40        &opened.reader,41        root,42        name,43        cli_options.signedOptionValue(args, "--tokens", 4000),44        cli_options.hasFlag(args, "--with-source"),45        limits.context,46    );47    boundary.seal();48    defer boundary.beginTeardown();49    if (std.mem.eql(u8, cli_options.flagValue(args, "--format") orelse "", "json")) return try cli_output.writeOut(try context.toJson(allocator, result));50    return try cli_output.writeOut(try context.toText(allocator, result));51}

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