tiny.smg.command.exports

1const std = @import("std");23const smg = @import("../root.zig");4const exports = smg.exports;5const text = smg.text;6const cli_options = smg.cli.options;7const cli_output = smg.cli.output;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    const fmt = args[0];13    if (!isFormat(fmt)) {14        try cli_output.writeClickUsageError(allocator, "export", "[OPTIONS] COMMAND [ARGS]...", try std.fmt.allocPrint(allocator, "No such command '{s}'.", .{fmt}));15        return 2;16    }17    const rest = args[1..];18    if (try validate(allocator, fmt, rest)) return 2;19    const loaded = try cli_session.loadRead(allocator, phases, limits, boundary);20    defer boundary.beginTeardown();21    const graph = loaded[1];22    if (std.mem.eql(u8, fmt, "json")) return try cli_output.writeOut(try exports.toJson(allocator, graph, cli_options.hasFlag(rest, "--indent")));23    if (std.mem.eql(u8, fmt, "mermaid")) return try cli_output.writeOut(try exports.toMermaid(allocator, graph));24    if (std.mem.eql(u8, fmt, "dot")) return try cli_output.writeOut(try exports.toDot(allocator, graph));25    if (std.mem.eql(u8, fmt, "text")) return try cli_output.writeOut(try exports.toText(allocator, graph));26    return try cli_output.writeOut(try exports.toDsm(allocator, graph, cli_options.flagValue(rest, "--level") orelse "module"));27}2829fn validate(allocator: std.mem.Allocator, fmt: []const u8, args: []const []const u8) !bool {30    if (std.mem.eql(u8, fmt, "json")) return try validateJson(allocator, args);31    if (std.mem.eql(u8, fmt, "dsm")) return try validateDsm(allocator, args);32    return try validatePlain(allocator, fmt, args);33}3435fn validateJson(allocator: std.mem.Allocator, args: []const []const u8) !bool {36    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, "export json", "[OPTIONS]", args, &.{ "--indent", "--no-indent" })) return true;37    return try rejectPositionals(allocator, "export json", args);38}3940fn validateDsm(allocator: std.mem.Allocator, args: []const []const u8) !bool {41    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{"--level"})) return true;42    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, "export dsm", "[OPTIONS]", args, &.{"--level"})) return true;43    if (try rejectPositionals(allocator, "export dsm", args)) return true;44    const level = cli_options.flagValue(args, "--level") orelse return false;45    if (isDsmLevel(level)) return false;46    try cli_output.writeClickUsageError(allocator, "export dsm", "[OPTIONS]", try std.fmt.allocPrint(allocator, "Invalid value for '--level': '{s}' is not one of 'module', 'class', 'all'.", .{level}));47    return true;48}4950fn validatePlain(allocator: std.mem.Allocator, fmt: []const u8, args: []const []const u8) !bool {51    const command = try std.fmt.allocPrint(allocator, "export {s}", .{fmt});52    if (try cli_validation.rejectUnexpectedOptionsPlain(allocator, command, "[OPTIONS]", args, &.{})) return true;53    return try rejectPositionals(allocator, command, args);54}5556fn rejectPositionals(allocator: std.mem.Allocator, command: []const u8, args: []const []const u8) !bool {57    const pos = try cli_options.positional(allocator, args);58    if (pos.len == 0) return false;59    try cli_output.writeClickUsageError(allocator, command, "[OPTIONS]", try cli_validation.unexpectedArgumentMessage(allocator, pos));60    return true;61}6263fn isFormat(fmt: []const u8) bool {64    return text.list.containsString(&.{ "json", "mermaid", "dot", "text", "dsm" }, fmt);65}6667fn isDsmLevel(level: []const u8) bool {68    return text.list.containsString(&.{ "module", "class", "all" }, level);69}7071test "export command choices match help surface" {72    try std.testing.expect(isFormat("json"));73    try std.testing.expect(isFormat("mermaid"));74    try std.testing.expect(isFormat("dot"));75    try std.testing.expect(isFormat("text"));76    try std.testing.expect(isFormat("dsm"));77    try std.testing.expect(!isFormat("csv"));78}7980test "export dsm levels are explicit" {81    try std.testing.expect(isDsmLevel("module"));82    try std.testing.expect(isDsmLevel("class"));83    try std.testing.expect(isDsmLevel("all"));84    try std.testing.expect(!isDsmLevel("package"));85}

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