tiny.smg.command.rules.render

1const std = @import("std");2const pretty_json = @import("pretty").json;34const smg = @import("../../root.zig");5const output = smg.cli.output;6const rich_table = smg.cli.table.rich;7const rules_mod = smg.rules;89pub fn writeList(allocator: std.mem.Allocator, rule_list: []const rules_mod.Rule, json: bool) !u8 {10    if (json) {11        var out: std.Io.Writer.Allocating = .init(allocator);12        var writer = pretty_json.Writer.init(&out.writer, .minified);13        try writer.beginArray();14        for (rule_list) |rule| try writer.raw(try rules_mod.render(allocator, rule));15        try writer.endArray();16        try out.writer.writeByte('\n');17        return try output.writeOut(out.written());18    }19    if (rule_list.len == 0) return try output.writeOut("No rules defined. Add one with smg rule add.\n");20    var out: std.Io.Writer.Allocating = .init(allocator);21    try writeTable(allocator, &out.writer, rule_list);22    return try output.writeOut(out.written());23}2425fn writeTable(allocator: std.mem.Allocator, writer: *std.Io.Writer, rule_list: []const rules_mod.Rule) !void {26    var rows: std.ArrayList(rich_table.Row3) = .empty;27    for (rule_list) |rule| {28        try rows.append(allocator, .{29            .a = rule.name,30            .b = rule.type,31            .c = try constraint(allocator, rule),32        });33    }34    try rich_table.write3(writer, .{ .a = "Name", .b = "Type", .c = "Constraint" }, rows.items);35}3637fn constraint(allocator: std.mem.Allocator, rule: rules_mod.Rule) ![]const u8 {38    var value = rule.pattern orelse rule.invariant orelse "";39    if (std.mem.eql(u8, rule.type, "quantified")) value = try std.fmt.allocPrint(allocator, "forall {s}: {s}", .{ rule.selector orelse "", rule.assertion orelse "" });40    var out: std.Io.Writer.Allocating = .init(allocator);41    try out.writer.writeAll(try stripMarkup(allocator, value));42    if (rule.params.len != 0) {43        try out.writer.writeAll(" (");44        for (rule.params, 0..) |param, index| {45            if (index != 0) try out.writer.writeAll(", ");46            try out.writer.print("{s}={s}", .{ param.key, param.value });47        }48        try out.writer.writeByte(')');49    }50    if (rule.allow_empty_target) try out.writer.writeAll(" allow-empty-target");51    if (rule.scope) |scope| try out.writer.print(" scope={s}", .{scope});52    return try out.toOwnedSlice();53}5455fn stripMarkup(allocator: std.mem.Allocator, value: []const u8) ![]const u8 {56    var out: std.ArrayList(u8) = .empty;57    var i: usize = 0;58    while (i < value.len) : (i += 1) {59        if (value[i] == '[') {60            while (i < value.len and value[i] != ']') i += 1;61            continue;62        }63        try out.append(allocator, value[i]);64    }65    return try out.toOwnedSlice(allocator);66}6768test "rule table strips rich markup from constraints" {69    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);70    defer arena.deinit();71    const allocator = arena.allocator();72    const rule_list = [_]rules_mod.Rule{.{73        .name = "no-db",74        .type = "deny",75        .pattern = "app.* -[calls]-> db.*",76    }};77    var out: std.Io.Writer.Allocating = .init(allocator);78    try writeTable(allocator, &out.writer, &rule_list);79    try std.testing.expect(std.mem.indexOf(u8, out.written(), "app.* --> db.*") != null);80    try std.testing.expect(std.mem.indexOf(u8, out.written(), "Constraint") != null);81}8283test "rule table exposes empty-target policy" {84    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);85    defer arena.deinit();86    const allocator = arena.allocator();87    const rule_list = [_]rules_mod.Rule{.{88        .name = "no-integration",89        .type = "deny",90        .pattern = "pkg -[contains]-> *.integration",91        .allow_empty_target = true,92    }};93    var out: std.Io.Writer.Allocating = .init(allocator);94    try writeTable(allocator, &out.writer, &rule_list);95    try std.testing.expect(std.mem.indexOf(u8, out.written(), "allow-empty-target") != null);96}

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