tiny.smg.cli.graph

1const std = @import("std");23const smg = @import("../root.zig");4const exports = smg.exports;5const graph_mod = smg.graph;6const cli_json = smg.cli.json;7const cli_options = smg.cli.options;8const cli_output = smg.cli.output;9const cli_table = smg.cli.table;1011pub fn writeNames(allocator: std.mem.Allocator, graph: graph_mod.Graph, center: []const u8, names: []const []const u8, title: []const u8, fmt: []const u8, limit: i64) !u8 {12    if (std.mem.eql(u8, fmt, "json")) return try cli_json.stringArray(allocator, names);13    if (std.mem.eql(u8, fmt, "mermaid") or std.mem.eql(u8, fmt, "dot")) {14        var sub = graph_mod.init(allocator);15        if (graph_mod.getNode(&graph, center)) |node| try graph_mod.addNode(&sub, node);16        for (names) |name| if (graph_mod.getNode(&graph, name)) |node| try graph_mod.addNode(&sub, node);17        for (graph.edges.items) |edge| if (sub.node_index.contains(edge.source) and sub.node_index.contains(edge.target)) try graph_mod.addEdge(&sub, edge);18        return try write(allocator, sub, fmt);19    }20    var out: std.Io.Writer.Allocating = .init(allocator);21    if (names.len == 0) return try cli_output.writeOutFmt(allocator, "{s}: (none)\n", .{title});22    try out.writer.print("{s}:\n", .{title});23    const displayed = cli_options.slicedLen(names.len, limit);24    const columns = [_]cli_table.Column{25        .{ .header = "name", .max_width = 72 },26    };27    var scratch_state = std.heap.ArenaAllocator.init(allocator);28    defer scratch_state.deinit();29    const scratch = scratch_state.allocator();30    var rows: std.ArrayList([]const []const u8) = .empty;31    for (names[0..displayed]) |name| try cli_table.appendRow(scratch, &rows, &.{name});32    try cli_table.writeJsonOption(allocator, &out.writer, &columns, rows.items, names.len, "--format json");33    return try cli_output.writeOut(out.written());34}3536pub fn write(allocator: std.mem.Allocator, graph: graph_mod.Graph, fmt: []const u8) !u8 {37    if (std.mem.eql(u8, fmt, "json")) return try cli_output.writeOut(try exports.toJson(allocator, graph, false));38    if (std.mem.eql(u8, fmt, "mermaid")) return try cli_output.writeOut(try exports.toMermaid(allocator, graph));39    if (std.mem.eql(u8, fmt, "dot")) return try cli_output.writeOut(try exports.toDot(allocator, graph));40    return try cli_output.writeOut(try exports.toText(allocator, graph));41}

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