cpulist 1.1.6

Parse and emit the Linux 'cpulist' data format used to list processors, memory regions and similar entities
Documentation 
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

use itertools::Itertools;

use crate::Item;

/// Parses a [cpulist][crate] and returns the numeric items in ascending order, removing duplicates.
///
/// An empty string is valid input and returns an empty result.
///
/// See [package-level documentation][crate] for details.
pub fn parse(cpulist: &str) -> crate::Result<Vec<Item>> {
    let parts = cpulist.split(',');

    let item_ranges: crate::Result<Vec<Vec<Item>>> = parts.map(parse_part).collect();

    item_ranges.map(|x| x.into_iter().flatten().sorted().dedup().collect())
}

fn parse_part(part: &str) -> crate::Result<Vec<Item>> {
    if part.is_empty() {
        return Ok(vec![]);
    }

    if let Some((range_start, range_end_inc)) = part.split_once('-') {
        parse_range(range_start, range_end_inc)
    } else {
        parse_single(part).map(|item| vec![item])
    }
}

fn parse_range(range_start: &str, range_end_inc: &str) -> crate::Result<Vec<Item>> {
    // This is a range, with optional stride.
    let range_start = range_start.parse::<Item>().map_err(|inner| {
        crate::Error::caused_by(
            range_start.to_string(),
            "range start could not be parsed as an integer".to_string(),
            inner,
        )
    })?;

    // If no stride is specified, we just default to 1 and pretend it was specified.
    let (range_end_inc, stride) =
        if let Some((range_end_inc, stride)) = range_end_inc.split_once(':') {
            (
                range_end_inc.parse::<Item>().map_err(|inner| {
                    crate::Error::caused_by(
                        range_end_inc.to_string(),
                        "range end could not be parsed as an integer".to_string(),
                        inner,
                    )
                })?,
                stride.parse::<Item>().map_err(|inner| {
                    crate::Error::caused_by(
                        stride.to_string(),
                        "range stride could not be parsed as an integer".to_string(),
                        inner,
                    )
                })?,
            )
        } else {
            (
                range_end_inc.parse::<Item>().map_err(|inner| {
                    crate::Error::caused_by(
                        range_end_inc.to_string(),
                        "range end could not be parsed as an integer".to_string(),
                        inner,
                    )
                })?,
                1,
            )
        };

    if stride == 0 {
        return Err(crate::Error::new(
            stride.to_string(),
            "range stride must not be zero".to_string(),
        ));
    }

    if range_start > range_end_inc {
        return Err(crate::Error::new(
            format!("{range_start}-{range_end_inc}"),
            "range start must be <= end".to_string(),
        ));
    }

    Ok((range_start..=range_end_inc)
        .step_by(stride as usize)
        .collect())
}

fn parse_single(single_item_part: &str) -> crate::Result<Item> {
    single_item_part.parse::<Item>().map_err(|inner| {
        crate::Error::caused_by(
            single_item_part.to_string(),
            "part was not a range but could not be parsed as an integer either".to_string(),
            inner,
        )
    })
}

#[cfg(test)]
#[cfg_attr(coverage_nightly, coverage(off))]
mod tests {
    use super::*;

    #[test]
    fn parse_smoke_test() {
        assert_eq!(parse("").unwrap(), vec![]);

        assert_eq!(parse("555").unwrap(), vec![555]);

        assert_eq!(parse("0,1,2,3").unwrap(), vec![0, 1, 2, 3]);

        assert_eq!(parse("2,3,1").unwrap(), vec![1, 2, 3]);

        assert_eq!(parse("0-5,1-6").unwrap(), vec![0, 1, 2, 3, 4, 5, 6]);

        assert_eq!(parse("0-0:5").unwrap(), vec![0]);

        assert_eq!(parse("0-0,1-1,3-3").unwrap(), vec![0, 1, 3]);
        assert_eq!(
            parse("0-10:3,5-15:3").unwrap(),
            vec![0, 3, 5, 6, 8, 9, 11, 14]
        );

        assert_eq!(parse("0-10:999999").unwrap(), vec![0]);

        assert_eq!(
            parse("0,1,2-4,5-9:2,6-10:2").unwrap(),
            vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
        );
    }

    #[test]
    fn zero_stride_is_error() {
        parse("1-22:0").unwrap_err();
    }

    #[test]
    fn range_direction_fail_is_error() {
        parse("2-1").unwrap_err();
    }

    #[test]
    fn garbage_is_error() {
        parse("foo").unwrap_err();
        parse("123-foo").unwrap_err();
        parse("foo-123").unwrap_err();
        parse("123-456:foo").unwrap_err();
        parse("123-foo:456").unwrap_err();
    }
}