581 lines
21 KiB
Zig
581 lines
21 KiB
Zig
//! From TigerBeetle, under Apache 2.0 attribution license.
|
|
//! https://github.com/tigerbeetle/tigerbeetle/blob/main/src/flags.zig TigerBeetle/
|
|
//!
|
|
//! The purpose of `flags` is to define standard behavior for parsing CLI arguments and provide
|
|
//! a specific parsing library, implementing this behavior.
|
|
//!
|
|
//! These are TigerBeetle CLI guidelines:
|
|
//!
|
|
//! - The main principle is robustness --- make operator errors harder to make.
|
|
//! - For production usage, avoid defaults.
|
|
//! - Thoroughly validate options.
|
|
//! - In particular, check that no options are repeated.
|
|
//! - Use only long options (`--addresses`).
|
|
//! - Exception: `-h/--help` is allowed.
|
|
//! - Use `--key=value` syntax for an option with an argument.
|
|
//! Don't use `--key value`, as that can be ambiguous (e.g., `--key --verbose`).
|
|
//! - Use subcommand syntax when appropriate.
|
|
//! - Use positional arguments when appropriate.
|
|
//!
|
|
//! Design choices for this particular `flags` library:
|
|
//!
|
|
//! - Be a 80% solution. Parsing arguments is a surprisingly vast topic: auto-generated help,
|
|
//! bash completions, typo correction. Rather than providing a definitive solution, `flags`
|
|
//! is just one possible option. It is ok to re-implement arg parsing in a different way, as long
|
|
//! as the CLI guidelines are observed.
|
|
//!
|
|
//! - No auto-generated help. Zig doesn't expose doc comments through `@typeInfo`, so its hard to
|
|
//! implement auto-help nicely. Additionally, fully hand-crafted `--help` message can be of
|
|
//! higher quality.
|
|
//!
|
|
//! - Fatal errors. It might be "cleaner" to use `try` to propagate the error to the caller, but
|
|
//! during early CLI parsing, it is much simpler to terminate the process directly and save the
|
|
//! caller the hassle of propagating errors. The `fatal` function is public, to allow the caller
|
|
//! to run additional validation or parsing using the same error reporting mechanism.
|
|
//!
|
|
//! - Concise DSL. Most cli parsing is done for ad-hoc tools like benchmarking, where the ability to
|
|
//! quickly add a new argument is valuable. As this is a 80% solution, production code may use
|
|
//! more verbose approach if it gives better UX.
|
|
//!
|
|
//! - Caller manages ArgsIterator. ArgsIterator owns the backing memory of the args, so we let the
|
|
//! caller to manage the lifetime. The caller should be skipping program name.
|
|
|
|
const std = @import("std");
|
|
const builtin = @import("builtin");
|
|
const assert = std.debug.assert;
|
|
const debug = @import("debug.zig");
|
|
|
|
/// Format and print an error message to stderr, then exit with an exit code of 1.
|
|
pub fn fatal(comptime fmt_string: []const u8, args: anytype) noreturn {
|
|
const stderr = std.io.getStdErr().writer();
|
|
stderr.print("error: " ++ fmt_string ++ "\n", args) catch {};
|
|
std.posix.exit(1);
|
|
}
|
|
|
|
/// Parse CLI arguments for subcommands specified as Zig `struct` or `union(enum)`:
|
|
///
|
|
/// ```
|
|
/// const CliArgs = union(enum) {
|
|
/// start: struct { addresses: []const u8, replica: u32 },
|
|
/// format: struct {
|
|
/// verbose: bool = false,
|
|
/// positional: struct {
|
|
/// path: []const u8,
|
|
/// }
|
|
/// },
|
|
///
|
|
/// pub const help =
|
|
/// \\ tigerbeetle start --addresses=<addresses> --replica=<replica>
|
|
/// \\ tigerbeetle format [--verbose] <path>
|
|
/// }
|
|
///
|
|
/// const cli_args = parse_commands(&args, CliArgs);
|
|
/// ```
|
|
///
|
|
/// `positional` field is treated specially, it designates positional arguments.
|
|
///
|
|
/// If `pub const help` declaration is present, it is used to implement `-h/--help` argument.
|
|
pub fn parse(args: *std.process.ArgIterator, comptime CliArgs: type) CliArgs {
|
|
assert(args.skip()); // Discard executable name.
|
|
|
|
return switch (@typeInfo(CliArgs)) {
|
|
.@"union" => parse_commands(args, CliArgs),
|
|
.@"struct" => parse_flags(args, CliArgs),
|
|
else => unreachable,
|
|
};
|
|
}
|
|
|
|
/// Parse CLI arguments for current process.
|
|
/// See `stdx.flags.parse` documentation for more.
|
|
pub fn parseProcessArgs(comptime CliArgs: type) CliArgs {
|
|
var args = std.process.args();
|
|
return parse(&args, CliArgs);
|
|
}
|
|
|
|
fn parse_commands(args: *std.process.ArgIterator, comptime Commands: type) Commands {
|
|
comptime assert(@typeInfo(Commands) == .Union);
|
|
comptime assert(std.meta.fields(Commands).len >= 2);
|
|
|
|
const first_arg = args.next() orelse fatal(
|
|
"subcommand required, expected {s}",
|
|
.{comptime fields_to_comma_list(Commands)},
|
|
);
|
|
|
|
// NB: help must be declared as *pub* const to be visible here.
|
|
if (@hasDecl(Commands, "help")) {
|
|
if (std.mem.eql(u8, first_arg, "-h") or std.mem.eql(u8, first_arg, "--help")) {
|
|
std.io.getStdOut().writeAll(Commands.help) catch std.posix.exit(1);
|
|
std.posix.exit(0);
|
|
}
|
|
}
|
|
|
|
inline for (comptime std.meta.fields(Commands)) |field| {
|
|
comptime assert(std.mem.indexOf(u8, field.name, "_") == null);
|
|
if (std.mem.eql(u8, first_arg, field.name)) {
|
|
return @unionInit(Commands, field.name, parse_flags(args, field.type));
|
|
}
|
|
}
|
|
fatal("unknown subcommand: '{s}'", .{first_arg});
|
|
}
|
|
|
|
fn parse_flags(args: *std.process.ArgIterator, comptime Flags: type) Flags {
|
|
@setEvalBranchQuota(5_000);
|
|
|
|
if (Flags == void) {
|
|
if (args.next()) |arg| {
|
|
fatal("unexpected argument: '{s}'", .{arg});
|
|
}
|
|
return {};
|
|
}
|
|
|
|
assert(@typeInfo(Flags) == .@"struct");
|
|
|
|
comptime var fields: [std.meta.fields(Flags).len]std.builtin.Type.StructField = undefined;
|
|
comptime var field_count = 0;
|
|
|
|
comptime var positional_fields: []const std.builtin.Type.StructField = &.{};
|
|
|
|
comptime for (std.meta.fields(Flags)) |field| {
|
|
if (std.mem.eql(u8, field.name, "positional")) {
|
|
assert(@typeInfo(field.type) == .@"struct");
|
|
positional_fields = std.meta.fields(field.type);
|
|
var optional_tail = false;
|
|
for (positional_fields) |positional_field| {
|
|
if (default_value(positional_field) == null) {
|
|
if (optional_tail) @panic("optional positional arguments must be last");
|
|
} else {
|
|
optional_tail = true;
|
|
}
|
|
switch (@typeInfo(positional_field.type)) {
|
|
.optional => |optional| {
|
|
// optional flags should have a default
|
|
assert(default_value(positional_field) != null);
|
|
assert(default_value(positional_field).? == null);
|
|
assert_valid_value_type(optional.child);
|
|
},
|
|
else => {
|
|
assert_valid_value_type(positional_field.type);
|
|
},
|
|
}
|
|
}
|
|
} else {
|
|
fields[field_count] = field;
|
|
field_count += 1;
|
|
|
|
switch (@typeInfo(field.type)) {
|
|
.bool => {
|
|
// boolean flags should have a default
|
|
debug.assertComptime(default_value(field) != null and default_value(field).? == false, "boolean flag --{s} should default to false", .{field.name});
|
|
},
|
|
.optional => |optional| {
|
|
// optional flags should have a default
|
|
debug.assertComptime(default_value(field) != null and default_value(field).? == null, "optional flag --{s} should have a null default value", .{field.name});
|
|
assert_valid_value_type(optional.child);
|
|
},
|
|
else => {
|
|
assert_valid_value_type(field.type);
|
|
},
|
|
}
|
|
}
|
|
};
|
|
|
|
var result: Flags = undefined;
|
|
// Would use std.enums.EnumFieldStruct(Flags, u32, 0) here but Flags is a struct not an Enum.
|
|
var counts = comptime blk: {
|
|
var count_fields = std.meta.fields(Flags)[0..std.meta.fields(Flags).len].*;
|
|
for (&count_fields) |*field| {
|
|
field.type = u32;
|
|
field.alignment = @alignOf(u32);
|
|
field.default_value_ptr = @ptrCast(&@as(u32, 0));
|
|
}
|
|
break :blk @Type(.{ .@"struct" = .{
|
|
.layout = .auto,
|
|
.fields = &count_fields,
|
|
.decls = &.{},
|
|
.is_tuple = false,
|
|
} }){};
|
|
};
|
|
|
|
// When parsing arguments, we must consider longer arguments first, such that `--foo-bar=92` is
|
|
// not confused for a misspelled `--foo=92`. Using `std.sort` for comptime-only values does not
|
|
// work, so open-code insertion sort, and comptime assert order during the actual parsing.
|
|
comptime {
|
|
for (fields[0..field_count], 0..) |*field_right, i| {
|
|
for (fields[0..i]) |*field_left| {
|
|
if (field_left.name.len < field_right.name.len) {
|
|
std.mem.swap(std.builtin.Type.StructField, field_left, field_right);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var parsed_positional = false;
|
|
next_arg: while (args.next()) |arg| {
|
|
comptime var field_len_prev = std.math.maxInt(usize);
|
|
inline for (fields[0..field_count]) |field| {
|
|
const flag = comptime flag_name(field);
|
|
|
|
comptime assert(field_len_prev >= field.name.len);
|
|
field_len_prev = field.name.len;
|
|
if (std.mem.startsWith(u8, arg, flag)) {
|
|
if (parsed_positional) {
|
|
fatal("unexpected trailing option: '{s}'", .{arg});
|
|
}
|
|
|
|
@field(counts, field.name) += 1;
|
|
const flag_value = parse_flag(field.type, flag, arg);
|
|
@field(result, field.name) = flag_value;
|
|
continue :next_arg;
|
|
}
|
|
}
|
|
|
|
if (@hasField(Flags, "positional")) {
|
|
counts.positional += 1;
|
|
switch (counts.positional - 1) {
|
|
inline 0...positional_fields.len - 1 => |positional_index| {
|
|
const positional_field = positional_fields[positional_index];
|
|
const flag = comptime flag_name_positional(positional_field);
|
|
|
|
if (arg.len == 0) fatal("{s}: empty argument", .{flag});
|
|
// Prevent ambiguity between a flag and positional argument value. We could add
|
|
// support for bare ` -- ` as a disambiguation mechanism once we have a real
|
|
// use-case.
|
|
if (arg[0] == '-') fatal("unexpected argument: '{s}'", .{arg});
|
|
parsed_positional = true;
|
|
|
|
@field(result.positional, positional_field.name) =
|
|
parse_value(positional_field.type, flag, arg);
|
|
continue :next_arg;
|
|
},
|
|
else => {}, // Fall-through to the unexpected argument error.
|
|
}
|
|
}
|
|
|
|
fatal("unexpected argument: '{s}'", .{arg});
|
|
}
|
|
|
|
inline for (fields[0..field_count]) |field| {
|
|
const flag = flag_name(field);
|
|
switch (@field(counts, field.name)) {
|
|
0 => if (default_value(field)) |default| {
|
|
@field(result, field.name) = default;
|
|
} else {
|
|
fatal("{s}: argument is required", .{flag});
|
|
},
|
|
1 => {},
|
|
else => fatal("{s}: duplicate argument", .{flag}),
|
|
}
|
|
}
|
|
|
|
if (@hasField(Flags, "positional")) {
|
|
assert(counts.positional <= positional_fields.len);
|
|
inline for (positional_fields, 0..) |positional_field, positional_index| {
|
|
if (positional_index >= counts.positional) {
|
|
const flag = comptime flag_name_positional(positional_field);
|
|
if (default_value(positional_field)) |default| {
|
|
@field(result.positional, positional_field.name) = default;
|
|
} else {
|
|
fatal("{s}: argument is required", .{flag});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
fn assert_valid_value_type(comptime T: type) void {
|
|
comptime {
|
|
if (T == []const u8 or T == [:0]const u8 or T == ByteSize or @typeInfo(T) == .int) return;
|
|
|
|
if (@typeInfo(T) == .@"enum") {
|
|
const info = @typeInfo(T).@"enum";
|
|
assert(info.is_exhaustive);
|
|
assert(info.fields.len >= 2);
|
|
return;
|
|
}
|
|
|
|
@compileLog("unsupported type", T);
|
|
unreachable;
|
|
}
|
|
}
|
|
|
|
/// Parse, e.g., `--cluster=123` into `123` integer
|
|
fn parse_flag(comptime T: type, flag: []const u8, arg: [:0]const u8) T {
|
|
assert(flag[0] == '-' and flag[1] == '-');
|
|
|
|
if (T == bool) {
|
|
if (!std.mem.eql(u8, arg, flag)) {
|
|
fatal("{s}: argument does not require a value in '{s}'", .{ flag, arg });
|
|
}
|
|
return true;
|
|
}
|
|
|
|
const value = parse_flag_split_value(flag, arg);
|
|
assert(value.len > 0);
|
|
return parse_value(T, flag, value);
|
|
}
|
|
|
|
/// Splits the value part from a `--arg=value` syntax.
|
|
fn parse_flag_split_value(flag: []const u8, arg: [:0]const u8) [:0]const u8 {
|
|
assert(flag[0] == '-' and flag[1] == '-');
|
|
assert(std.mem.startsWith(u8, arg, flag));
|
|
|
|
const value = arg[flag.len..];
|
|
if (value.len == 0) {
|
|
fatal("{s}: expected value separator '='", .{flag});
|
|
}
|
|
if (value[0] != '=') {
|
|
fatal(
|
|
"{s}: expected value separator '=', but found '{c}' in '{s}'",
|
|
.{ flag, value[0], arg },
|
|
);
|
|
}
|
|
if (value.len == 1) fatal("{s}: argument requires a value", .{flag});
|
|
return value[1..];
|
|
}
|
|
|
|
fn parse_value(comptime T: type, flag: []const u8, value: [:0]const u8) T {
|
|
comptime assert(T != bool);
|
|
assert((flag[0] == '-' and flag[1] == '-') or flag[0] == '<');
|
|
assert(value.len > 0);
|
|
|
|
const V = switch (@typeInfo(T)) {
|
|
.optional => |optional| optional.child,
|
|
else => T,
|
|
};
|
|
|
|
if (V == []const u8 or V == [:0]const u8) return value;
|
|
if (V == ByteSize) return parse_value_size(flag, value);
|
|
if (@typeInfo(V) == .int) return parse_value_int(V, flag, value);
|
|
if (@typeInfo(V) == .@"enum") return parse_value_enum(V, flag, value);
|
|
comptime unreachable;
|
|
}
|
|
|
|
fn parse_value_size(flag: []const u8, value: []const u8) ByteSize {
|
|
assert((flag[0] == '-' and flag[1] == '-') or flag[0] == '<');
|
|
|
|
return ByteSize.parse(value) catch |err| {
|
|
switch (err) {
|
|
error.ParseOverflow => fatal(
|
|
"{s}: value exceeds 64-bit unsigned integer: '{s}'",
|
|
.{ flag, value },
|
|
),
|
|
error.InvalidSize => fatal(
|
|
"{s}: expected a size, but found '{s}'",
|
|
.{ flag, value },
|
|
),
|
|
error.InvalidUnit => fatal(
|
|
"{s}: invalid unit in size '{s}', (needed KiB, MiB, GiB or TiB)",
|
|
.{ flag, value },
|
|
),
|
|
error.BytesOverflow => fatal(
|
|
"{s}: size in bytes exceeds 64-bit unsigned integer: '{s}'",
|
|
.{ flag, value },
|
|
),
|
|
}
|
|
};
|
|
}
|
|
|
|
pub const ByteUnit = enum(u64) {
|
|
bytes = 1,
|
|
kib = 1024,
|
|
mib = 1024 * 1024,
|
|
gib = 1024 * 1024 * 1024,
|
|
tib = 1024 * 1024 * 1024 * 1024,
|
|
};
|
|
|
|
const ByteSizeParseError = error{
|
|
ParseOverflow,
|
|
InvalidSize,
|
|
InvalidUnit,
|
|
BytesOverflow,
|
|
};
|
|
|
|
pub const ByteSize = struct {
|
|
value: u64,
|
|
unit: ByteUnit = .bytes,
|
|
|
|
fn parse(value: []const u8) ByteSizeParseError!ByteSize {
|
|
assert(value.len != 0);
|
|
|
|
const split: struct {
|
|
value_input: []const u8,
|
|
unit_input: []const u8,
|
|
} = split: for (0..value.len) |i| {
|
|
if (!std.ascii.isDigit(value[i])) {
|
|
break :split .{
|
|
.value_input = value[0..i],
|
|
.unit_input = value[i..],
|
|
};
|
|
}
|
|
} else {
|
|
break :split .{
|
|
.value_input = value,
|
|
.unit_input = "",
|
|
};
|
|
};
|
|
|
|
const amount = std.fmt.parseUnsigned(u64, split.value_input, 10) catch |err| {
|
|
switch (err) {
|
|
error.Overflow => {
|
|
return ByteSizeParseError.ParseOverflow;
|
|
},
|
|
error.InvalidCharacter => {
|
|
// The only case this can happen is for the empty string
|
|
return ByteSizeParseError.InvalidSize;
|
|
},
|
|
}
|
|
};
|
|
|
|
const unit = if (split.unit_input.len > 0)
|
|
unit: inline for (comptime std.enums.values(ByteUnit)) |tag| {
|
|
if (std.ascii.eqlIgnoreCase(split.unit_input, @tagName(tag))) {
|
|
break :unit tag;
|
|
}
|
|
} else {
|
|
return ByteSizeParseError.InvalidUnit;
|
|
}
|
|
else
|
|
ByteUnit.bytes;
|
|
|
|
_ = std.math.mul(u64, amount, @intFromEnum(unit)) catch {
|
|
return ByteSizeParseError.BytesOverflow;
|
|
};
|
|
|
|
return ByteSize{ .value = amount, .unit = unit };
|
|
}
|
|
|
|
pub fn bytes(size: *const ByteSize) u64 {
|
|
return std.math.mul(
|
|
u64,
|
|
size.value,
|
|
@intFromEnum(size.unit),
|
|
) catch unreachable;
|
|
}
|
|
|
|
pub fn suffix(size: *const ByteSize) []const u8 {
|
|
return switch (size.unit) {
|
|
.bytes => "",
|
|
.kib => "KiB",
|
|
.mib => "MiB",
|
|
.gib => "GiB",
|
|
.tib => "TiB",
|
|
};
|
|
}
|
|
};
|
|
|
|
test parse_value_size {
|
|
const kib = 1024;
|
|
const mib = kib * 1024;
|
|
const gib = mib * 1024;
|
|
const tib = gib * 1024;
|
|
|
|
const cases = .{
|
|
.{ 0, "0", 0, ByteUnit.bytes },
|
|
.{ 1, "1", 1, ByteUnit.bytes },
|
|
.{ 140737488355328, "140737488355328", 140737488355328, ByteUnit.bytes },
|
|
.{ 140737488355328, "128TiB", 128, ByteUnit.tib },
|
|
.{ 1 * tib, "1TiB", 1, ByteUnit.tib },
|
|
.{ 10 * tib, "10tib", 10, ByteUnit.tib },
|
|
.{ 1 * gib, "1GiB", 1, ByteUnit.gib },
|
|
.{ 10 * gib, "10gib", 10, ByteUnit.gib },
|
|
.{ 1 * mib, "1MiB", 1, ByteUnit.mib },
|
|
.{ 10 * mib, "10mib", 10, ByteUnit.mib },
|
|
.{ 1 * kib, "1KiB", 1, ByteUnit.kib },
|
|
.{ 10 * kib, "10kib", 10, ByteUnit.kib },
|
|
};
|
|
|
|
inline for (cases) |case| {
|
|
const bytes = case[0];
|
|
const input = case[1];
|
|
const unit_val = case[2];
|
|
const unit = case[3];
|
|
const got = parse_value_size("--size", input);
|
|
assert(bytes == got.bytes());
|
|
assert(unit_val == got.value);
|
|
assert(unit == got.unit);
|
|
}
|
|
}
|
|
|
|
/// Parse string value into an integer, providing a nice error message for the user.
|
|
fn parse_value_int(comptime T: type, flag: []const u8, value: [:0]const u8) T {
|
|
assert((flag[0] == '-' and flag[1] == '-') or flag[0] == '<');
|
|
|
|
return std.fmt.parseInt(T, value, 10) catch |err| {
|
|
switch (err) {
|
|
error.Overflow => fatal(
|
|
"{s}: value exceeds {d}-bit {s} integer: '{s}'",
|
|
.{ flag, @typeInfo(T).int.bits, @tagName(@typeInfo(T).int.signedness), value },
|
|
),
|
|
error.InvalidCharacter => fatal(
|
|
"{s}: expected an integer value, but found '{s}' (invalid digit)",
|
|
.{ flag, value },
|
|
),
|
|
}
|
|
};
|
|
}
|
|
|
|
fn parse_value_enum(comptime E: type, flag: []const u8, value: [:0]const u8) E {
|
|
assert((flag[0] == '-' and flag[1] == '-') or flag[0] == '<');
|
|
comptime assert(@typeInfo(E).@"enum".is_exhaustive);
|
|
|
|
return std.meta.stringToEnum(E, value) orelse fatal(
|
|
"{s}: expected one of {s}, but found '{s}'",
|
|
.{ flag, comptime fields_to_comma_list(E), value },
|
|
);
|
|
}
|
|
|
|
fn fields_to_comma_list(comptime E: type) []const u8 {
|
|
comptime {
|
|
const field_count = std.meta.fields(E).len;
|
|
assert(field_count >= 2);
|
|
|
|
var result: []const u8 = "";
|
|
for (std.meta.fields(E), 0..) |field, field_index| {
|
|
const separator = switch (field_index) {
|
|
0 => "",
|
|
else => ", ",
|
|
field_count - 1 => if (field_count == 2) " or " else ", or ",
|
|
};
|
|
result = result ++ separator ++ "'" ++ field.name ++ "'";
|
|
}
|
|
return result;
|
|
}
|
|
}
|
|
|
|
pub fn flag_name(comptime field: std.builtin.Type.StructField) []const u8 {
|
|
// TODO(Zig): Cleanup when this is fixed after Zig 0.11.
|
|
// Without comptime blk, the compiler thinks the result is a runtime slice returning a UAF.
|
|
return comptime blk: {
|
|
assert(!std.mem.eql(u8, field.name, "positional"));
|
|
|
|
var result: []const u8 = "--";
|
|
var index = 0;
|
|
while (std.mem.indexOf(u8, field.name[index..], "_")) |i| {
|
|
result = result ++ field.name[index..][0..i] ++ "-";
|
|
index = index + i + 1;
|
|
}
|
|
result = result ++ field.name[index..];
|
|
break :blk result;
|
|
};
|
|
}
|
|
|
|
test flag_name {
|
|
const field = @typeInfo(struct { statsd: bool }).@"struct".fields[0];
|
|
try std.testing.expectEqualStrings(flag_name(field), "--statsd");
|
|
}
|
|
|
|
fn flag_name_positional(comptime field: std.builtin.Type.StructField) []const u8 {
|
|
comptime assert(std.mem.indexOf(u8, field.name, "_") == null);
|
|
return "<" ++ field.name ++ ">";
|
|
}
|
|
|
|
/// This is essentially `field.default_value`, but with a useful type instead of `?*anyopaque`.
|
|
pub fn default_value(comptime field: std.builtin.Type.StructField) ?field.type {
|
|
return if (field.default_value_ptr) |default_opaque|
|
|
@as(*const field.type, @ptrCast(@alignCast(default_opaque))).*
|
|
else
|
|
null;
|
|
}
|