tiny.tracy.cli.args.system

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

1const std = @import("std");2const tracy = @import("../../root.zig");3const scalar = @import("scalar.zig");4const types = @import("types.zig");56pub fn parse(args: []const []const u8) !types.SystemArgs {7    if (args.len == 0) return error.InvalidArguments;8    var parsed = types.SystemArgs{ .path = args[0] };9    var around_ns: ?u64 = null;10    var window_ns: ?u64 = null;11    var index: usize = 1;12    while (index < args.len) : (index += 1) {13        if (std.mem.eql(u8, args[index], "--format")) {14            index += 1;15            if (index >= args.len) return error.InvalidArguments;16            if (std.mem.eql(u8, args[index], "text")) {17                parsed.format = .text;18            } else if (std.mem.eql(u8, args[index], "jsonl")) {19                parsed.format = .jsonl;20            } else {21                return error.UnsupportedFormat;22            }23        } else if (std.mem.eql(u8, args[index], "--top")) {24            index += 1;25            if (index >= args.len) return error.InvalidArguments;26            parsed.options.top = try scalar.parsePositiveUsize(args[index]);27        } else if (std.mem.eql(u8, args[index], "--occurrences")) {28            index += 1;29            if (index >= args.len) return error.InvalidArguments;30            parsed.options.occurrences = try scalar.parseUsize(args[index]);31        } else if (std.mem.eql(u8, args[index], "--group")) {32            index += 1;33            if (index >= args.len) return error.InvalidArguments;34            parsed.options.group = tracy.system.Group.fromName(args[index]) orelse return error.UnsupportedGroup;35        } else if (std.mem.eql(u8, args[index], "--sort")) {36            index += 1;37            if (index >= args.len) return error.InvalidArguments;38            parsed.options.sort = tracy.system.Sort.fromName(args[index]) orelse return error.UnsupportedSort;39        } else if (std.mem.eql(u8, args[index], "--thread")) {40            index += 1;41            if (index >= args.len) return error.InvalidArguments;42            parsed.options.thread = try scalar.parseU64(args[index]);43        } else if (std.mem.eql(u8, args[index], "--cpu")) {44            index += 1;45            if (index >= args.len) return error.InvalidArguments;46            parsed.options.cpu = @intCast(try scalar.parseU64(args[index]));47        } else if (std.mem.eql(u8, args[index], "--address")) {48            index += 1;49            if (index >= args.len) return error.InvalidArguments;50            parsed.options.address = try scalar.parseAddress(args[index]);51        } else if (std.mem.eql(u8, args[index], "--kind")) {52            index += 1;53            if (index >= args.len) return error.InvalidArguments;54            parsed.options.kind = args[index];55        } else if (std.mem.eql(u8, args[index], "--since-ns")) {56            index += 1;57            if (index >= args.len) return error.InvalidArguments;58            parsed.options.since_ns = try scalar.parseU64(args[index]);59        } else if (std.mem.eql(u8, args[index], "--until-ns")) {60            index += 1;61            if (index >= args.len) return error.InvalidArguments;62            parsed.options.until_ns = try scalar.parseU64(args[index]);63        } else if (std.mem.eql(u8, args[index], "--around-ns")) {64            index += 1;65            if (index >= args.len) return error.InvalidArguments;66            around_ns = try scalar.parseU64(args[index]);67        } else if (std.mem.eql(u8, args[index], "--window-ns")) {68            index += 1;69            if (index >= args.len) return error.InvalidArguments;70            window_ns = try scalar.parseU64(args[index]);71        } else if (std.mem.eql(u8, args[index], "--match")) {72            index += 1;73            if (index >= args.len) return error.InvalidArguments;74            parsed.options.match = args[index];75        } else if (std.mem.eql(u8, args[index], "--ignore-case")) {76            parsed.options.ignore_case = true;77        } else {78            return error.UnknownArgument;79        }80    }81    if (around_ns) |center| {82        if (parsed.options.since_ns != null or parsed.options.until_ns != null) return error.InvalidArguments;83        const width = window_ns orelse 1_000_000;84        const half = width / 2;85        parsed.options.since_ns = if (center > half) center - half else 0;86        parsed.options.until_ns = center +| (width - half);87    } else if (window_ns != null) {88        return error.InvalidArguments;89    }90    if (parsed.options.since_ns != null and parsed.options.until_ns != null and parsed.options.since_ns.? > parsed.options.until_ns.?) {91        return error.InvalidArguments;92    }93    return parsed;94}9596test "system args parse filters window grouping and address" {97    const parsed = try parse(&.{98        "events.jsonl",99        "--format",100        "jsonl",101        "--top",102        "12",103        "--occurrences",104        "4",105        "--group",106        "address",107        "--sort",108        "value",109        "--thread",110        "7",111        "--cpu",112        "2",113        "--address",114        "0x1234",115        "--kind",116        "hw",117        "--around-ns",118        "1000",119        "--window-ns",120        "200",121        "--match",122        "cache",123        "--ignore-case",124    });125    try std.testing.expectEqualStrings("events.jsonl", parsed.path);126    try std.testing.expectEqual(types.Format.jsonl, parsed.format);127    try std.testing.expectEqual(@as(usize, 12), parsed.options.top);128    try std.testing.expectEqual(@as(usize, 4), parsed.options.occurrences);129    try std.testing.expectEqual(tracy.system.Group.address, parsed.options.group);130    try std.testing.expectEqual(tracy.system.Sort.value, parsed.options.sort);131    try std.testing.expectEqual(@as(?u64, 7), parsed.options.thread);132    try std.testing.expectEqual(@as(?u32, 2), parsed.options.cpu);133    try std.testing.expectEqual(@as(?u64, 0x1234), parsed.options.address);134    try std.testing.expectEqualStrings("hw", parsed.options.kind.?);135    try std.testing.expectEqual(@as(?u64, 900), parsed.options.since_ns);136    try std.testing.expectEqual(@as(?u64, 1100), parsed.options.until_ns);137    try std.testing.expectEqualStrings("cache", parsed.options.match.?);138    try std.testing.expect(parsed.options.ignore_case);139}