git.iko.soy/bat
on f9fd5e4 help sign in
bat/src/line_range.rs
74 lines 2.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 }
43#[test]
44fn test_parse_full() {
45 let range = LineRange::from("40:50").expect("Shouldn't fail on test!");
46 assert_eq!(40, range.lower);
47 assert_eq!(50, range.upper);
50#[test]
51fn test_parse_partial_min() {
52 let range = LineRange::from(":50").expect("Shouldn't fail on test!");
53 assert_eq!(usize::min_value(), range.lower);
54 assert_eq!(50, range.upper);
57#[test]
58fn test_parse_partial_max() {
59 let range = LineRange::from("40:").expect("Shouldn't fail on test!");
60 assert_eq!(40, range.lower);
61 assert_eq!(usize::max_value(), range.upper);
64#[test]
65fn test_parse_fail() {
66 let range = LineRange::from("40:50:80");
67 assert!(range.is_err());
68 let range = LineRange::from("40::80");
69 assert!(range.is_err());
70 let range = LineRange::from(":40:");
71 assert!(range.is_err());
72 let range = LineRange::from("40");
73 assert!(range.is_err());