summaryrefslogtreecommitdiff
path: root/clap/parsers.zig
blob: 2bf89761a312d69b1a8be3d0c74b5b5e5d43a122 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
pub const default = .{
    .string = string,
    .str = string,
    .u8 = int(u8, 0),
    .u16 = int(u16, 0),
    .u32 = int(u32, 0),
    .u64 = int(u64, 0),
    .usize = int(usize, 0),
    .i8 = int(i8, 0),
    .i16 = int(i16, 0),
    .i32 = int(i32, 0),
    .i64 = int(i64, 0),
    .isize = int(isize, 0),
    .f32 = float(f32),
    .f64 = float(f64),
};

/// A parser that does nothing.
pub fn string(in: []const u8) error{}![]const u8 {
    return in;
}

test "string" {
    try std.testing.expectEqualStrings("aa", try string("aa"));
}

/// A parser that uses `std.fmt.parseInt` to parse the string into an integer value.
/// See `std.fmt.parseInt` documentation for more information.
pub fn int(comptime T: type, comptime base: u8) fn ([]const u8) std.fmt.ParseIntError!T {
    return struct {
        fn parse(in: []const u8) std.fmt.ParseIntError!T {
            return std.fmt.parseInt(T, in, base);
        }
    }.parse;
}

test "int" {
    try std.testing.expectEqual(@as(u8, 0), try int(u8, 10)("0"));
    try std.testing.expectEqual(@as(u8, 1), try int(u8, 10)("1"));
    try std.testing.expectEqual(@as(u8, 10), try int(u8, 10)("10"));
    try std.testing.expectEqual(@as(u8, 0b10), try int(u8, 2)("10"));
    try std.testing.expectEqual(@as(u8, 0x10), try int(u8, 0)("0x10"));
    try std.testing.expectEqual(@as(u8, 0b10), try int(u8, 0)("0b10"));
    try std.testing.expectEqual(@as(u16, 0), try int(u16, 10)("0"));
    try std.testing.expectEqual(@as(u16, 1), try int(u16, 10)("1"));
    try std.testing.expectEqual(@as(u16, 10), try int(u16, 10)("10"));
    try std.testing.expectEqual(@as(u16, 0b10), try int(u16, 2)("10"));
    try std.testing.expectEqual(@as(u16, 0x10), try int(u16, 0)("0x10"));
    try std.testing.expectEqual(@as(u16, 0b10), try int(u16, 0)("0b10"));
}

/// A parser that uses `std.fmt.parseFloat` to parse the string into an float value.
/// See `std.fmt.parseFloat` documentation for more information.
pub fn float(comptime T: type) fn ([]const u8) std.fmt.ParseFloatError!T {
    return struct {
        fn parse(in: []const u8) std.fmt.ParseFloatError!T {
            return std.fmt.parseFloat(T, in);
        }
    }.parse;
}

test "float" {
    try std.testing.expectEqual(@as(f32, 0), try float(f32)("0"));
}

pub const EnumError = error{
    NameNotPartOfEnum,
};

/// A parser that uses `std.meta.stringToEnum` to parse the string into an enum value. On `null`,
/// this function returns the error `NameNotPartOfEnum`.
/// See `std.meta.stringToEnum` documentation for more information.
pub fn enumeration(comptime T: type) fn ([]const u8) EnumError!T {
    return struct {
        fn parse(in: []const u8) EnumError!T {
            return std.meta.stringToEnum(T, in) orelse error.NameNotPartOfEnum;
        }
    }.parse;
}

test "enumeration" {
    const E = enum { a, b, c };
    try std.testing.expectEqual(E.a, try enumeration(E)("a"));
    try std.testing.expectEqual(E.b, try enumeration(E)("b"));
    try std.testing.expectEqual(E.c, try enumeration(E)("c"));
    try std.testing.expectError(EnumError.NameNotPartOfEnum, enumeration(E)("d"));
}

fn ReturnType(comptime P: type) type {
    return @typeInfo(P).@"fn".return_type.?;
}

pub fn Result(comptime P: type) type {
    return @typeInfo(ReturnType(P)).error_union.payload;
}

const std = @import("std");