tiny.smg.command.search

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

1const std = @import("std");2const pretty_json = @import("pretty").json;34const smg = @import("../root.zig");5const options = smg.cli.options;6const output = smg.cli.output;7const search_mod = smg.search;8const session = smg.command.session;9const storage = smg.storage;10const table = smg.cli.table;11const validation = smg.cli.validation;1213pub fn run(allocator: std.mem.Allocator, _: std.mem.Allocator, args: []const []const u8, limits: smg.Limits, boundary: session.PhaseBoundary) !u8 {14    const pos = try options.positional(allocator, args);15    if (pos.len < 1) return error.MissingArgument;16    const root = try storage.requireRoot(allocator);17    try session.noteGraphAge(allocator, root);18    const kind = options.flagValue(args, "--kind");19    const limit = options.searchLimitValue(args);20    const result = search_mod.searchStored(21        allocator,22        root,23        pos[0],24        kind,25        limit,26        limits.storage,27    ) catch |err| switch (err) {28        search_mod.Error.InvalidSearchQuery => {29            try output.writeErrFmt(allocator, "Error: invalid search query: '{s}'\n", .{pos[0]});30            return 2;31        },32        else => return err,33    };34    boundary.seal();35    defer boundary.beginTeardown();36    if (options.hasJson(args)) return try writeJson(allocator, result, limit);37    if (result.hits.len == 0) return try output.writeOut("No results.\n");38    var out: std.Io.Writer.Allocating = .init(allocator);39    if (options.hasFlag(args, "--full")) {40        try writeFull(allocator, &out.writer, result);41    } else {42        try writeTable(allocator, &out.writer, result);43    }44    return try output.writeOut(out.written());45}4647pub fn index(allocator: std.mem.Allocator, phases: std.mem.Allocator, args: []const []const u8, limits: smg.Limits) !u8 {48    if (try validation.rejectUnexpectedOptionsPlain(allocator, "index", "[OPTIONS]", args, &.{})) return 2;49    const pos = try options.positional(allocator, args);50    if (pos.len != 0) {51        try output.writeClickUsageError(allocator, "index", "[OPTIONS]", try validation.unexpectedArgumentMessage(allocator, pos));52        return 2;53    }54    const root = try storage.requireRoot(allocator);55    const count = try search_mod.rebuildStored(phases, root, limits);56    return try output.writeOutFmt(allocator, "Indexed {d} nodes into search cache\n", .{count});57}5859pub fn writeJson(allocator: std.mem.Allocator, result: search_mod.Result, limit: i64) !u8 {60    var out: std.Io.Writer.Allocating = .init(allocator);61    var writer = pretty_json.Writer.init(&out.writer, .minified);62    try writer.beginObject();63    try writer.objectField("rows");64    try writer.beginArray();65    for (result.hits) |hit| {66        try writer.beginObject();67        try writer.objectField("rank");68        try writer.write(hit.rank);69        try writer.objectField("kind");70        try writer.write(hit.kind);71        try writer.objectField("name");72        try writer.write(hit.name);73        try writer.objectField("location");74        try writer.write(try search_mod.location(allocator, hit));75        try writer.objectField("snippet");76        try writer.write(search_mod.snippet(hit));77        try writer.endObject();78    }79    try writer.endArray();80    try writer.objectField("total");81    try writer.write(result.total);82    try writer.objectField("displayed");83    try writer.write(result.hits.len);84    try writer.objectField("truncated");85    try writer.write(result.total > @as(i64, @intCast(result.hits.len)));86    try writer.objectField("limit");87    try writer.write(limit);88    try writer.endObject();89    try out.writer.writeByte('\n');90    return try output.writeOut(out.written());91}9293pub fn writeTable(allocator: std.mem.Allocator, writer: *std.Io.Writer, result: search_mod.Result) !void {94    const columns = [_]table.Column{95        .{ .header = "rank", .alignment = .right, .max_width = 6 },96        .{ .header = "kind", .max_width = 12 },97        .{ .header = "name", .max_width = 40 },98        .{ .header = "location", .max_width = 30 },99        .{ .header = "snippet", .max_width = 40 },100    };101    var scratch_state = std.heap.ArenaAllocator.init(allocator);102    defer scratch_state.deinit();103    const scratch = scratch_state.allocator();104    var rows: std.ArrayList([]const []const u8) = .empty;105    for (result.hits) |hit| {106        try table.appendRow(scratch, &rows, &.{107            try std.fmt.allocPrint(scratch, "{d}", .{hit.rank}),108            hit.kind,109            hit.name,110            try search_mod.location(scratch, hit),111            search_mod.snippet(hit),112        });113    }114    const total = if (result.total > 0) @as(usize, @intCast(result.total)) else result.hits.len;115    try table.write(allocator, writer, &columns, rows.items, total);116}117118pub fn writeFull(allocator: std.mem.Allocator, writer: *std.Io.Writer, result: search_mod.Result) !void {119    for (result.hits) |hit| {120        try writer.print("{d}. [{s}] {s}\n", .{ hit.rank, hit.kind, hit.name });121        try writer.print("   {s}\n", .{try search_mod.location(allocator, hit)});122        if (hit.docstring) |doc| {123            if (doc.len != 0) {124                var lines = std.mem.splitScalar(u8, doc, '\n');125                var count: usize = 0;126                while (count < 5) : (count += 1) {127                    const line = lines.next() orelse break;128                    try writer.print("   {s}\n", .{line});129                }130            }131        }132        try writer.writeByte('\n');133    }134    if (result.total > @as(i64, @intCast(result.hits.len))) try writer.print("(showing {d} of {d} -- use --limit 0 for all, --json for exact records)\n", .{ result.hits.len, result.total });135}136137test "search renders table and full text" {138    var arena = std.heap.ArenaAllocator.init(std.testing.allocator);139    defer arena.deinit();140    const allocator = arena.allocator();141    const hits = [_]search_mod.Hit{142        .{143            .rank = 1,144            .name = "app.handler",145            .kind = "function",146            .file = "src/app.py",147            .line_start = 12,148            .line_end = 14,149            .docstring = "first line\nsecond line",150            .score = 10,151        },152        .{153            .rank = 2,154            .name = "app.module",155            .kind = "module",156            .file = null,157            .line_start = null,158            .line_end = null,159            .docstring = null,160            .score = 5,161        },162    };163    const result: search_mod.Result = .{ .hits = &hits, .total = 3 };164    var table_out: std.Io.Writer.Allocating = .init(allocator);165    try writeTable(allocator, &table_out.writer, result);166    try std.testing.expect(std.mem.indexOf(u8, table_out.written(), "rank") != null);167    try std.testing.expect(std.mem.indexOf(u8, table_out.written(), "app.handler") != null);168    try std.testing.expect(std.mem.indexOf(u8, table_out.written(), "src/app.py:12") != null);169    try std.testing.expect(std.mem.indexOf(u8, table_out.written(), "first line") != null);170    try std.testing.expect(std.mem.indexOf(u8, table_out.written(), "showing 2 of 3") != null);171    var full_out: std.Io.Writer.Allocating = .init(allocator);172    try writeFull(allocator, &full_out.writer, result);173    try std.testing.expect(std.mem.indexOf(u8, full_out.written(), "1. [function] app.handler\n") != null);174    try std.testing.expect(std.mem.indexOf(u8, full_out.written(), "   src/app.py:12\n") != null);175    try std.testing.expect(std.mem.indexOf(u8, full_out.written(), "   first line\n") != null);176    try std.testing.expect(std.mem.indexOf(u8, full_out.written(), "(showing 2 of 3 -- use --limit 0 for all, --json for exact records)\n") != null);177}