git.iko.soy/bat
on 5faf7ca help sign in
bat/src/line_range.rs
185 lines 5.4 kB history raw
1use crate::errors::*;
3#[derive(Debug, 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 match line_numbers.len() {
34 1 => {
35 new_range.lower = line_numbers[0].parse()?;
36 new_range.upper = new_range.lower;
37 Ok(new_range)
38 },
39 2 => {
40 new_range.lower = line_numbers[0].parse()?;
41 new_range.upper = line_numbers[1].parse()?;
42 Ok(new_range)
43 },
44 _ => Err("Line range contained more than one ':' character. Expected format: 'N' or 'N:M'".into()),
45 }
46 }
48 pub fn is_inside(&self, line: usize) -> bool {
49 line >= self.lower && line <= self.upper
50 }
53#[test]
54fn test_parse_full() {
55 let range = LineRange::from("40:50").expect("Shouldn't fail on test!");
56 assert_eq!(40, range.lower);
57 assert_eq!(50, range.upper);
60#[test]
61fn test_parse_partial_min() {
62 let range = LineRange::from(":50").expect("Shouldn't fail on test!");
63 assert_eq!(usize::min_value(), range.lower);
64 assert_eq!(50, range.upper);
67#[test]
68fn test_parse_partial_max() {
69 let range = LineRange::from("40:").expect("Shouldn't fail on test!");
70 assert_eq!(40, range.lower);
71 assert_eq!(usize::max_value(), range.upper);
74#[test]
75fn test_parse_single() {
76 let range = LineRange::from("40").expect("Shouldn't fail on test!");
77 assert_eq!(40, range.lower);
78 assert_eq!(40, range.upper);
81#[test]
82fn test_parse_fail() {
83 let range = LineRange::from("40:50:80");
84 assert!(range.is_err());
85 let range = LineRange::from("40::80");
86 assert!(range.is_err());
87 let range = LineRange::from(":40:");
88 assert!(range.is_err());
91#[derive(Copy, Clone, Debug, PartialEq)]
92pub enum RangeCheckResult {
93 // Within one of the given ranges
94 InRange,
96 // Before the first range or within two ranges
97 OutsideRange,
99 // Line number is outside of all ranges and larger than the last range.
100 AfterLastRange,
103#[derive(Debug, Clone, Default)]
104pub struct LineRanges {
105 ranges: Vec<LineRange>,
106 largest_upper_bound: usize,
109impl LineRanges {
110 pub fn from(ranges: Vec<LineRange>) -> LineRanges {
111 let largest_upper_bound = ranges
112 .iter()
113 .map(|r| r.upper)
114 .max()
115 .unwrap_or(usize::max_value());
116 LineRanges {
117 ranges,
118 largest_upper_bound,
119 }
120 }
122 pub fn check(&self, line: usize) -> RangeCheckResult {
123 if self.ranges.is_empty() | self.ranges.iter().any(|r| r.is_inside(line)) {
124 RangeCheckResult::InRange
125 } else if line < self.largest_upper_bound {
126 RangeCheckResult::OutsideRange
127 } else {
128 RangeCheckResult::AfterLastRange
129 }
130 }
133#[cfg(test)]
134fn ranges(rs: &[&str]) -> LineRanges {
135 LineRanges::from(rs.iter().map(|r| LineRange::from(r).unwrap()).collect())
138#[test]
139fn test_ranges_simple() {
140 let ranges = ranges(&["3:8"]);
142 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(2));
143 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
144 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
147#[test]
148fn test_ranges_advanced() {
149 let ranges = ranges(&["3:8", "11:20", "25:30"]);
151 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(2));
152 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
153 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(9));
154 assert_eq!(RangeCheckResult::InRange, ranges.check(11));
155 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(22));
156 assert_eq!(RangeCheckResult::InRange, ranges.check(28));
157 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(31));
160#[test]
161fn test_ranges_open_low() {
162 let ranges = ranges(&["3:8", ":5"]);
164 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
165 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
166 assert_eq!(RangeCheckResult::InRange, ranges.check(7));
167 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
170#[test]
171fn test_ranges_open_high() {
172 let ranges = ranges(&["3:", "2:5"]);
174 assert_eq!(RangeCheckResult::OutsideRange, ranges.check(1));
175 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
176 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
177 assert_eq!(RangeCheckResult::InRange, ranges.check(9));
180#[test]
181fn test_ranges_empty() {
182 let ranges = ranges(&[]);
184 assert_eq!(RangeCheckResult::InRange, ranges.check(1));