git.iko.soy/bat
on 47a3721 help sign in
bat/src/line_range.rs
220 lines 6.1 kB history raw
1use crate::errors::*;
3#[derive(Debug, Clone)]
4pub struct LineRange {
5 pub lower: usize,
6 pub upper: usize,
7}
9impl Default for LineRange {
10 fn default() -> LineRange {
11 LineRange {
12 lower: usize::min_value(),
13 upper: usize::max_value(),
14 }
15 }
18impl LineRange {
19 pub fn from(range_raw: &str) -> Result<LineRange> {
20 LineRange::parse_range(range_raw)
21 }
23 fn parse_range(range_raw: &str) -> Result<LineRange> {
24 let mut new_range = LineRange::default();
26 if range_raw.bytes().nth(0).ok_or("Empty line range")? == b':' {
27 new_range.upper = range_raw[1..].parse()?;
28 return Ok(new_range);
29 } else if range_raw.bytes().last().ok_or("Empty line range")? == b':' {
30 new_range.lower = range_raw[..range_raw.len() - 1].parse()?;
31 return Ok(new_range);
32 }
34 let line_numbers: Vec<&str> = range_raw.split(':').collect();
35 match line_numbers.len() {
36 1 => {
37 new_range.lower = line_numbers[0].parse()?;
38 new_range.upper = new_range.lower;
39 Ok(new_range)
40 }
41 2 => {
42 new_range.lower = line_numbers[0].parse()?;
43 new_range.upper = line_numbers[1].parse()?;
44 Ok(new_range)
45 }
46 _ => Err(
47 "Line range contained more than one ':' character. Expected format: 'N' or 'N:M'"
48 .into(),
49 ),
50 }
51 }
53 pub(crate) fn is_inside(&self, line: usize) -> bool {
54 line >= self.lower && line <= self.upper
55 }
58#[test]
59fn test_parse_full() {
60 let range = LineRange::from("40:50").expect("Shouldn't fail on test!");
61 assert_eq!(40, range.lower);
62 assert_eq!(50, range.upper);
65#[test]
66fn test_parse_partial_min() {
67 let range = LineRange::from(":50").expect("Shouldn't fail on test!");
68 assert_eq!(usize::min_value(), range.lower);
69 assert_eq!(50, range.upper);
72#[test]
73fn test_parse_partial_max() {
74 let range = LineRange::from("40:").expect("Shouldn't fail on test!");
75 assert_eq!(40, range.lower);
76 assert_eq!(usize::max_value(), range.upper);
79#[test]
80fn test_parse_single() {
81 let range = LineRange::from("40").expect("Shouldn't fail on test!");
82 assert_eq!(40, range.lower);
83 assert_eq!(40, range.upper);
86#[test]
87fn test_parse_fail() {
88 let range = LineRange::from("40:50:80");
89 assert!(range.is_err());
90 let range = LineRange::from("40::80");
91 assert!(range.is_err());
92 let range = LineRange::from(":40:");
93 assert!(range.is_err());
96#[derive(Copy, Clone, Debug, PartialEq)]
97pub enum RangeCheckResult {
98 // Within one of the given ranges
99 InRange,
101 // Before the first range or within two ranges
102 BeforeOrBetweenRanges,
104 // Line number is outside of all ranges and larger than the last range.
105 AfterLastRange,
108#[derive(Debug, Clone)]
109pub struct LineRanges {
110 ranges: Vec<LineRange>,
111 largest_upper_bound: usize,
114impl LineRanges {
115 pub fn none() -> LineRanges {
116 LineRanges::from(vec![])
117 }
119 pub fn all() -> LineRanges {
120 LineRanges::from(vec![LineRange::default()])
121 }
123 pub fn from(ranges: Vec<LineRange>) -> LineRanges {
124 let largest_upper_bound = ranges
125 .iter()
126 .map(|r| r.upper)
127 .max()
128 .unwrap_or(usize::max_value());
129 LineRanges {
130 ranges,
131 largest_upper_bound,
132 }
133 }
135 pub(crate) fn check(&self, line: usize) -> RangeCheckResult {
136 if self.ranges.iter().any(|r| r.is_inside(line)) {
137 RangeCheckResult::InRange
138 } else if line < self.largest_upper_bound {
139 RangeCheckResult::BeforeOrBetweenRanges
140 } else {
141 RangeCheckResult::AfterLastRange
142 }
143 }
146impl Default for LineRanges {
147 fn default() -> Self {
148 Self::all()
149 }
152#[derive(Debug, Clone)]
153pub struct HighlightedLineRanges(pub LineRanges);
155impl Default for HighlightedLineRanges {
156 fn default() -> Self {
157 HighlightedLineRanges(LineRanges::none())
158 }
161#[cfg(test)]
162fn ranges(rs: &[&str]) -> LineRanges {
163 LineRanges::from(rs.iter().map(|r| LineRange::from(r).unwrap()).collect())
166#[test]
167fn test_ranges_simple() {
168 let ranges = ranges(&["3:8"]);
170 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(2));
171 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
172 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
175#[test]
176fn test_ranges_advanced() {
177 let ranges = ranges(&["3:8", "11:20", "25:30"]);
179 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(2));
180 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
181 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(9));
182 assert_eq!(RangeCheckResult::InRange, ranges.check(11));
183 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(22));
184 assert_eq!(RangeCheckResult::InRange, ranges.check(28));
185 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(31));
188#[test]
189fn test_ranges_open_low() {
190 let ranges = ranges(&["3:8", ":5"]);
192 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
193 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
194 assert_eq!(RangeCheckResult::InRange, ranges.check(7));
195 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
198#[test]
199fn test_ranges_open_high() {
200 let ranges = ranges(&["3:", "2:5"]);
202 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(1));
203 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
204 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
205 assert_eq!(RangeCheckResult::InRange, ranges.check(9));
208#[test]
209fn test_ranges_all() {
210 let ranges = LineRanges::all();
212 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
215#[test]
216fn test_ranges_none() {
217 let ranges = LineRanges::none();
219 assert_ne!(RangeCheckResult::InRange, ranges.check(1));