git.iko.soy/bat
on 7c41bd7 help sign in
bat/src/line_range.rs
253 lines 7.0 kB history raw
1use crate::error::*;
3#[derive(Debug, Clone)]
4pub struct LineRange {
5 lower: usize,
6 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 new(from: usize, to: usize) -> Self {
20 LineRange {
21 lower: from,
22 upper: to,
23 }
24 }
26 pub fn from(range_raw: &str) -> Result<LineRange> {
27 LineRange::parse_range(range_raw)
28 }
30 fn parse_range(range_raw: &str) -> Result<LineRange> {
31 let mut new_range = LineRange::default();
33 if range_raw.bytes().next().ok_or("Empty line range")? == b':' {
34 new_range.upper = range_raw[1..].parse()?;
35 return Ok(new_range);
36 } else if range_raw.bytes().last().ok_or("Empty line range")? == b':' {
37 new_range.lower = range_raw[..range_raw.len() - 1].parse()?;
38 return Ok(new_range);
39 }
41 let line_numbers: Vec<&str> = range_raw.split(':').collect();
42 match line_numbers.len() {
43 1 => {
44 new_range.lower = line_numbers[0].parse()?;
45 new_range.upper = new_range.lower;
46 Ok(new_range)
47 }
48 2 => {
49 new_range.lower = line_numbers[0].parse()?;
51 new_range.upper = if line_numbers[1].bytes().next() == Some(b'+') {
52 let more_lines = &line_numbers[1][1..]
53 .parse()
54 .map_err(|_| "Invalid character after +")?;
55 new_range.lower + more_lines
56 } else {
57 line_numbers[1].parse()?
58 };
60 Ok(new_range)
61 }
62 _ => Err(
63 "Line range contained more than one ':' character. Expected format: 'N' or 'N:M'"
64 .into(),
65 ),
66 }
67 }
69 pub(crate) fn is_inside(&self, line: usize) -> bool {
70 line >= self.lower && line <= self.upper
71 }
74#[test]
75fn test_parse_full() {
76 let range = LineRange::from("40:50").expect("Shouldn't fail on test!");
77 assert_eq!(40, range.lower);
78 assert_eq!(50, range.upper);
81#[test]
82fn test_parse_partial_min() {
83 let range = LineRange::from(":50").expect("Shouldn't fail on test!");
84 assert_eq!(usize::min_value(), range.lower);
85 assert_eq!(50, range.upper);
88#[test]
89fn test_parse_partial_max() {
90 let range = LineRange::from("40:").expect("Shouldn't fail on test!");
91 assert_eq!(40, range.lower);
92 assert_eq!(usize::max_value(), range.upper);
95#[test]
96fn test_parse_single() {
97 let range = LineRange::from("40").expect("Shouldn't fail on test!");
98 assert_eq!(40, range.lower);
99 assert_eq!(40, range.upper);
102#[test]
103fn test_parse_fail() {
104 let range = LineRange::from("40:50:80");
105 assert!(range.is_err());
106 let range = LineRange::from("40::80");
107 assert!(range.is_err());
108 let range = LineRange::from(":40:");
109 assert!(range.is_err());
112#[test]
113fn test_parse_plus() {
114 let range = LineRange::from("40:+10").expect("Shouldn't fail on test!");
115 assert_eq!(40, range.lower);
116 assert_eq!(50, range.upper);
119#[test]
120fn test_parse_plus_fail() {
121 let range = LineRange::from("40:+z");
122 assert!(range.is_err());
123 let range = LineRange::from("40:+-10");
124 assert!(range.is_err());
125 let range = LineRange::from("40:+");
126 assert!(range.is_err());
129#[derive(Copy, Clone, Debug, PartialEq)]
130pub enum RangeCheckResult {
131 // Within one of the given ranges
132 InRange,
134 // Before the first range or within two ranges
135 BeforeOrBetweenRanges,
137 // Line number is outside of all ranges and larger than the last range.
138 AfterLastRange,
141#[derive(Debug, Clone)]
142pub struct LineRanges {
143 ranges: Vec<LineRange>,
144 largest_upper_bound: usize,
147impl LineRanges {
148 pub fn none() -> LineRanges {
149 LineRanges::from(vec![])
150 }
152 pub fn all() -> LineRanges {
153 LineRanges::from(vec![LineRange::default()])
154 }
156 pub fn from(ranges: Vec<LineRange>) -> LineRanges {
157 let largest_upper_bound = ranges
158 .iter()
159 .map(|r| r.upper)
160 .max()
161 .unwrap_or(usize::max_value());
162 LineRanges {
163 ranges,
164 largest_upper_bound,
165 }
166 }
168 pub(crate) fn check(&self, line: usize) -> RangeCheckResult {
169 if self.ranges.iter().any(|r| r.is_inside(line)) {
170 RangeCheckResult::InRange
171 } else if line < self.largest_upper_bound {
172 RangeCheckResult::BeforeOrBetweenRanges
173 } else {
174 RangeCheckResult::AfterLastRange
175 }
176 }
179impl Default for LineRanges {
180 fn default() -> Self {
181 Self::all()
182 }
185#[derive(Debug, Clone)]
186pub struct HighlightedLineRanges(pub LineRanges);
188impl Default for HighlightedLineRanges {
189 fn default() -> Self {
190 HighlightedLineRanges(LineRanges::none())
191 }
194#[cfg(test)]
195fn ranges(rs: &[&str]) -> LineRanges {
196 LineRanges::from(rs.iter().map(|r| LineRange::from(r).unwrap()).collect())
199#[test]
200fn test_ranges_simple() {
201 let ranges = ranges(&["3:8"]);
203 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(2));
204 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
205 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
208#[test]
209fn test_ranges_advanced() {
210 let ranges = ranges(&["3:8", "11:20", "25:30"]);
212 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(2));
213 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
214 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(9));
215 assert_eq!(RangeCheckResult::InRange, ranges.check(11));
216 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(22));
217 assert_eq!(RangeCheckResult::InRange, ranges.check(28));
218 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(31));
221#[test]
222fn test_ranges_open_low() {
223 let ranges = ranges(&["3:8", ":5"]);
225 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
226 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
227 assert_eq!(RangeCheckResult::InRange, ranges.check(7));
228 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
231#[test]
232fn test_ranges_open_high() {
233 let ranges = ranges(&["3:", "2:5"]);
235 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(1));
236 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
237 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
238 assert_eq!(RangeCheckResult::InRange, ranges.check(9));
241#[test]
242fn test_ranges_all() {
243 let ranges = LineRanges::all();
245 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
248#[test]
249fn test_ranges_none() {
250 let ranges = LineRanges::none();
252 assert_ne!(RangeCheckResult::InRange, ranges.check(1));