tiny.smg.command.mutate.node

1const std = @import("std");23const smg = @import("../../root.zig");4const graph_mod = smg.graph;5const model = smg.model;6const text = smg.text;7const cli_metadata = smg.cli.metadata;8const cli_options = smg.cli.options;9const cli_output = smg.cli.output;10const cli_resolve = smg.cli.resolve;11const cli_session = smg.command.session;12const cli_validation = smg.cli.validation;1314pub fn add(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits) !u8 {15    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{ "--file", "--line", "--doc", "--meta" })) return 2;16    if (try cli_validation.rejectUnexpectedOptions(allocator, "add", "[OPTIONS] TYPE NAME", args, &.{ "--file", "--line", "--doc", "--meta" })) return 2;17    const pos = try cli_options.positional(allocator, args);18    if (pos.len == 0) {19        try cli_output.writeClickUsageError(allocator, "add", "[OPTIONS] TYPE NAME", "Missing argument 'TYPE'.");20        return 2;21    }22    if (pos.len == 1) {23        try cli_output.writeClickUsageError(allocator, "add", "[OPTIONS] TYPE NAME", "Missing argument 'NAME'.");24        return 2;25    }26    if (pos.len > 2) {27        try cli_output.writeClickUsageError(allocator, "add", "[OPTIONS] TYPE NAME", try cli_validation.unexpectedArgumentMessage(allocator, pos[2..]));28        return 2;29    }30    if (try cli_validation.rejectInvalidLine(allocator, "add", "[OPTIONS] TYPE NAME", args)) return 2;31    if (try cli_validation.rejectInvalidMeta(allocator, args)) return 2;32    const loaded = try cli_session.loadMutation(allocator, phases, limits);33    var graph = loaded.graph;34    const meta = try cli_metadata.pairs(allocator, args, "manual");35    try graph_mod.addNode(&graph, .{36        .type = try allocator.dupe(u8, pos[0]),37        .name = try allocator.dupe(u8, pos[1]),38        .file = if (cli_options.flagValue(args, "--file")) |v| try allocator.dupe(u8, v) else null,39        .line = if (cli_options.flagValue(args, "--line")) |v| text.parseInt(v, 0) else null,40        .docstring = if (cli_options.flagValue(args, "--doc")) |v| try allocator.dupe(u8, v) else null,41        .metadata = meta,42    });43    try smg.storage.graph.publish(phases, loaded.root, graph, loaded.head, limits);44    return try cli_output.writeOutFmt(allocator, "Added [{s}] {s}\n", .{ pos[0], pos[1] });45}4647pub fn remove(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits) !u8 {48    if (try cli_validation.rejectUnexpectedOptions(allocator, "rm", "[OPTIONS] NAME", args, &.{})) return 2;49    const pos = try cli_options.positional(allocator, args);50    if (pos.len == 0) {51        try cli_output.writeClickUsageError(allocator, "rm", "[OPTIONS] NAME", "Missing argument 'NAME'.");52        return 2;53    }54    if (pos.len > 1) {55        try cli_output.writeClickUsageError(allocator, "rm", "[OPTIONS] NAME", try cli_validation.unexpectedArgumentMessage(allocator, pos[1..]));56        return 2;57    }58    const loaded = try cli_session.loadMutation(allocator, phases, limits);59    var graph = loaded.graph;60    const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);61    try graph_mod.removeNodes(&graph, &.{name});62    try smg.storage.graph.publish(phases, loaded.root, graph, loaded.head, limits);63    return try cli_output.writeOutFmt(allocator, "Removed {s}\n", .{name});64}6566pub fn update(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits) !u8 {67    if (try cli_validation.rejectMissingOptionValues(allocator, args, &.{ "--type", "--file", "--line", "--doc", "--meta" })) return 2;68    if (try cli_validation.rejectUnexpectedOptions(allocator, "update", "[OPTIONS] NAME", args, &.{ "--type", "--file", "--line", "--doc", "--meta" })) return 2;69    const pos = try cli_options.positional(allocator, args);70    if (pos.len == 0) {71        try cli_output.writeClickUsageError(allocator, "update", "[OPTIONS] NAME", "Missing argument 'NAME'.");72        return 2;73    }74    if (pos.len > 1) {75        try cli_output.writeClickUsageError(allocator, "update", "[OPTIONS] NAME", try cli_validation.unexpectedArgumentMessage(allocator, pos[1..]));76        return 2;77    }78    if (try cli_validation.rejectInvalidLine(allocator, "update", "[OPTIONS] NAME", args)) return 2;79    if (try cli_validation.rejectInvalidMeta(allocator, args)) return 2;80    const loaded = try cli_session.loadMutation(allocator, phases, limits);81    var graph = loaded.graph;82    const name = try cli_resolve.graphNode(allocator, graph, pos[0], limits.suggestions);83    const existing = graph_mod.getNode(&graph, name) orelse return error.NodeNotFound;84    var node = existing;85    if (cli_options.flagValue(args, "--type")) |v| node.type = try allocator.dupe(u8, v);86    if (cli_options.flagValue(args, "--file")) |v| node.file = try allocator.dupe(u8, v);87    if (cli_options.flagValue(args, "--line")) |v| node.line = text.parseInt(v, 0);88    if (cli_options.flagValue(args, "--doc")) |v| node.docstring = try allocator.dupe(u8, v);89    const updates = try cli_metadata.pairs(allocator, args, "");90    if (updates.len != 0) node.metadata = try model.mergePairs(allocator, node.metadata, updates);91    try graph_mod.addNode(&graph, node);92    try smg.storage.graph.publish(phases, loaded.root, graph, loaded.head, limits);93    return try cli_output.writeOutFmt(allocator, "Updated {s}\n", .{name});94}

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