git.iko.soy/bat
on 411d68e help sign in
bat/src/line_range.rs
227 lines 6.2 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()?;
50 new_range.upper = line_numbers[1].parse()?;
51 Ok(new_range)
52 }
53 _ => Err(
54 "Line range contained more than one ':' character. Expected format: 'N' or 'N:M'"
55 .into(),
56 ),
57 }
58 }
60 pub(crate) fn is_inside(&self, line: usize) -> bool {
61 line >= self.lower && line <= self.upper
62 }
65#[test]
66fn test_parse_full() {
67 let range = LineRange::from("40:50").expect("Shouldn't fail on test!");
68 assert_eq!(40, range.lower);
69 assert_eq!(50, range.upper);
72#[test]
73fn test_parse_partial_min() {
74 let range = LineRange::from(":50").expect("Shouldn't fail on test!");
75 assert_eq!(usize::min_value(), range.lower);
76 assert_eq!(50, range.upper);
79#[test]
80fn test_parse_partial_max() {
81 let range = LineRange::from("40:").expect("Shouldn't fail on test!");
82 assert_eq!(40, range.lower);
83 assert_eq!(usize::max_value(), range.upper);
86#[test]
87fn test_parse_single() {
88 let range = LineRange::from("40").expect("Shouldn't fail on test!");
89 assert_eq!(40, range.lower);
90 assert_eq!(40, range.upper);
93#[test]
94fn test_parse_fail() {
95 let range = LineRange::from("40:50:80");
96 assert!(range.is_err());
97 let range = LineRange::from("40::80");
98 assert!(range.is_err());
99 let range = LineRange::from(":40:");
100 assert!(range.is_err());
103#[derive(Copy, Clone, Debug, PartialEq)]
104pub enum RangeCheckResult {
105 // Within one of the given ranges
106 InRange,
108 // Before the first range or within two ranges
109 BeforeOrBetweenRanges,
111 // Line number is outside of all ranges and larger than the last range.
112 AfterLastRange,
115#[derive(Debug, Clone)]
116pub struct LineRanges {
117 ranges: Vec<LineRange>,
118 largest_upper_bound: usize,
121impl LineRanges {
122 pub fn none() -> LineRanges {
123 LineRanges::from(vec![])
124 }
126 pub fn all() -> LineRanges {
127 LineRanges::from(vec![LineRange::default()])
128 }
130 pub fn from(ranges: Vec<LineRange>) -> LineRanges {
131 let largest_upper_bound = ranges
132 .iter()
133 .map(|r| r.upper)
134 .max()
135 .unwrap_or(usize::max_value());
136 LineRanges {
137 ranges,
138 largest_upper_bound,
139 }
140 }
142 pub(crate) fn check(&self, line: usize) -> RangeCheckResult {
143 if self.ranges.iter().any(|r| r.is_inside(line)) {
144 RangeCheckResult::InRange
145 } else if line < self.largest_upper_bound {
146 RangeCheckResult::BeforeOrBetweenRanges
147 } else {
148 RangeCheckResult::AfterLastRange
149 }
150 }
153impl Default for LineRanges {
154 fn default() -> Self {
155 Self::all()
156 }
159#[derive(Debug, Clone)]
160pub struct HighlightedLineRanges(pub LineRanges);
162impl Default for HighlightedLineRanges {
163 fn default() -> Self {
164 HighlightedLineRanges(LineRanges::none())
165 }
168#[cfg(test)]
169fn ranges(rs: &[&str]) -> LineRanges {
170 LineRanges::from(rs.iter().map(|r| LineRange::from(r).unwrap()).collect())
173#[test]
174fn test_ranges_simple() {
175 let ranges = ranges(&["3:8"]);
177 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(2));
178 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
179 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
182#[test]
183fn test_ranges_advanced() {
184 let ranges = ranges(&["3:8", "11:20", "25:30"]);
186 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(2));
187 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
188 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(9));
189 assert_eq!(RangeCheckResult::InRange, ranges.check(11));
190 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(22));
191 assert_eq!(RangeCheckResult::InRange, ranges.check(28));
192 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(31));
195#[test]
196fn test_ranges_open_low() {
197 let ranges = ranges(&["3:8", ":5"]);
199 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
200 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
201 assert_eq!(RangeCheckResult::InRange, ranges.check(7));
202 assert_eq!(RangeCheckResult::AfterLastRange, ranges.check(9));
205#[test]
206fn test_ranges_open_high() {
207 let ranges = ranges(&["3:", "2:5"]);
209 assert_eq!(RangeCheckResult::BeforeOrBetweenRanges, ranges.check(1));
210 assert_eq!(RangeCheckResult::InRange, ranges.check(3));
211 assert_eq!(RangeCheckResult::InRange, ranges.check(5));
212 assert_eq!(RangeCheckResult::InRange, ranges.check(9));
215#[test]
216fn test_ranges_all() {
217 let ranges = LineRanges::all();
219 assert_eq!(RangeCheckResult::InRange, ranges.check(1));
222#[test]
223fn test_ranges_none() {
224 let ranges = LineRanges::none();
226 assert_ne!(RangeCheckResult::InRange, ranges.check(1));