git.iko.soy/bat
on b68c5d8 help sign in
bat/src/line_range.rs
180 lines 5.1 kB history raw
1use errors::*;
3#[derive(Clone)]
4pub struct LineRange {
5 pub lower: usize,
6 pub upper: usize,
7}
9impl LineRange {
10 pub fn from(range_raw: &str) -> Result<LineRange> {
11 LineRange::parse_range(range_raw)
12 }
14 pub fn new() -> LineRange {
15 LineRange {
16 lower: usize::min_value(),
17 upper: usize::max_value(),
18 }
19 }
21 pub fn parse_range(range_raw: &str) -> Result<LineRange> {
22 let mut new_range = LineRange::new();
24 if range_raw.bytes().nth(0).ok_or("Empty line range")? == b':' {
25 new_range.upper = range_raw[1..].parse()?;
26 return Ok(new_range);
27 } else if range_raw.bytes().last().ok_or("Empty line range")? == b':' {
28 new_range.lower = range_raw[..range_raw.len() - 1].parse()?;
29 return Ok(new_range);
30 }
32 let line_numbers: Vec<&str> = range_raw.split(':').collect();
33 if line_numbers.len() == 2 {
34 new_range.lower = line_numbers[0].parse()?;
35 new_range.upper = line_numbers[1].parse()?;
36 return Ok(new_range);
37 }
39 Err("expected single ':' character".into())
40 }
42 pub fn is_inside(&self, line: usize) -> bool {
43 line >= self.lower && line <= self.upper
44 }
47#[test]
48fn test_parse_full() {
49 let range = LineRange::from("40:50").expect("Shouldn't fail on test!");
50 assert_eq!(40, range.lower);
51 assert_eq!(50, range.upper);
54#[test]
55fn test_parse_partial_min() {
56 let range = LineRange::from(":50").expect("Shouldn't fail on test!");
57 assert_eq!(usize::min_value(), range.lower);
58 assert_eq!(50, range.upper);
61#[test]
62fn test_parse_partial_max() {
63 let range = LineRange::from("40:").expect("Shouldn't fail on test!");
64 assert_eq!(40, range.lower);
65 assert_eq!(usize::max_value(), range.upper);
68#[test]
69fn test_parse_fail() {
70 let range = LineRange::from("40:50:80");
71 assert!(range.is_err());
72 let range = LineRange::from("40::80");
73 assert!(range.is_err());
74 let range = LineRange::from(":40:");
75 assert!(range.is_err());
76 let range = LineRange::from("40");
77 assert!(range.is_err());
80#[derive(Copy, Clone, Debug, PartialEq)]
81pub enum RangeCheckResult {
82 // Within one of the given ranges
83 InRange,
85 // Before the first range or within two ranges
86 OutsideRange,
88 // Line number is outside of all ranges and larger than the last range.
89 AfterLastRange,
92#[derive(Clone)]
93pub struct LineRanges {
94 ranges: Vec<LineRange>,
95 largest_upper_bound: usize,
98impl LineRanges {
99 pub fn from(ranges: Vec<LineRange>) -> LineRanges {
100 let largest_upper_bound = ranges
101 .iter()
102 .map(|r| r.upper)
103 .max()
104 .unwrap_or(usize::max_value());
105 LineRanges {
106 ranges,
107 largest_upper_bound,
108 }
109 }
111 pub fn check(&self, line: usize) -> RangeCheckResult {
112 if self.ranges.is_empty() {
113 RangeCheckResult::InRange
114 } else {
115 if self.ranges.iter().any(|r| r.is_inside(line)) {
116 RangeCheckResult::InRange
117 } else {
118 if line < self.largest_upper_bound {
119 RangeCheckResult::OutsideRange
120 } else {
121 RangeCheckResult::AfterLastRange
122 }
123 }
124 }
125 }
128#[cfg(test)]
129fn ranges(rs: &[&str]) -> LineRanges {
130 LineRanges::from(rs.iter().map(|r| LineRange::from(r).unwrap()).collect())
133#[test]
134fn test_ranges_simple() {
135 let ranges = ranges(&["3:8"]);
137 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(2));
138 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
139 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
142#[test]
143fn test_ranges_advanced() {
144 let ranges = ranges(&["3:8", "11:20", "25:30"]);
146 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(2));
147 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
148 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(9));
149 assert_eq!(RangeCheckResult::InRange, ranges.check(11));
150 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(22));
151 assert_eq!(RangeCheckResult::InRange, ranges.check(28));
152 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(31));
155#[test]
156fn test_ranges_open_low() {
157 let ranges = ranges(&["3:8", ":5"]);
159 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
160 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
161 assert_eq!(RangeCheckResult::InRange, ranges.check(7));
162 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
165#[test]
166fn test_ranges_open_high() {
167 let ranges = ranges(&["3:", "2:5"]);
169 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(1));
170 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
171 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
172 assert_eq!(RangeCheckResult::InRange, ranges.check(9));
175#[test]
176fn test_ranges_empty() {
177 let ranges = ranges(&[]);
179 assert_eq!(RangeCheckResult::InRange, ranges.check(1));