sift-core 0.4.0

Indexed regex search over codebases (library + grep-like CLI)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ColumnOverflow {
    Omit,
    Preview,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct ColumnLimit {
    pub max: u64,
    pub overflow: ColumnOverflow,
}

pub enum ColumnAction {
    Normal,
    Omit,
    Preview,
}

impl ColumnLimit {
    #[must_use]
    pub fn classify(&self, line: &[u8]) -> ColumnAction {
        let trimmed = line.strip_suffix(b"\n").unwrap_or(line);
        let trimmed = trimmed.strip_suffix(b"\r").unwrap_or(trimmed);
        if trimmed.len() as u64 > self.max {
            match self.overflow {
                ColumnOverflow::Preview => ColumnAction::Preview,
                ColumnOverflow::Omit => ColumnAction::Omit,
            }
        } else {
            ColumnAction::Normal
        }
    }

    #[must_use]
    pub fn truncate(&self, line: &[u8]) -> Vec<u8> {
        let trimmed = line.strip_suffix(b"\n").unwrap_or(line);
        let trimmed = trimmed.strip_suffix(b"\r").unwrap_or(trimmed);
        let limit = usize::try_from(self.max).unwrap_or(usize::MAX);
        let mut out = Vec::with_capacity(limit.saturating_add(30));
        out.extend_from_slice(&trimmed[..limit.min(trimmed.len())]);
        out.extend_from_slice(b" [... omitted end ...]");
        out.push(b'\n');
        out
    }
}