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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
use bat::{
self,
assets::HighlightingAssets,
inputfile::InputFile,
line_range::{LineRange, LineRanges, RangeCheckResult},
output::OutputType,
printer::{InteractivePrinter, Printer},
style::{OutputComponent, OutputComponents},
Config,
};
use std::{collections::HashSet, io};
fn main() -> bat::errors::Result<()> {
let assets = HighlightingAssets::new();
let mut config = Config {
term_width: 14, // must be greater than 13 to enable style=numbers
colored_output: true,
true_color: true,
line_ranges: LineRanges::from(vec![
LineRange::from("5:7")?,
LineRange::from("92:97")?,
LineRange::from("15:17")?,
]),
output_components: OutputComponents(with_full_decorations()),
..Default::default()
};
let mut add_file = |file: &'static str| config.files.push(InputFile::Ordinary(file));
// add_file("Cargo.toml");
add_file("build.rs");
let mut output_type = OutputType::from_mode(config.paging_mode, config.pager)?;
let writer = output_type.handle()?;
let stdin = io::stdin();
for input_file in &config.files {
let mut reader = input_file.get_reader(&stdin)?;
let mut printer = InteractivePrinter::new(&config, &assets, *input_file, &mut reader);
printer.print_header(writer, *input_file)?;
if !reader.first_line.is_empty() {
let mut line_buffer = Vec::new();
let mut line_number = 1;
let mut has_snip = true; // skip first line snip
while reader.read_line(&mut line_buffer)? {
match config.line_ranges.check(line_number) {
RangeCheckResult::OutsideRange => {
if !has_snip {
printer.print_snip(writer)?;
has_snip = true;
}
}
RangeCheckResult::InRange => {
printer.print_line(
/*out_of_range = */ false,
writer,
line_number,
&line_buffer,
)?;
has_snip = false;
}
RangeCheckResult::AfterLastRange => break,
}
line_number += 1;
line_buffer.clear();
}
}
printer.print_footer(writer)?;
}
Ok(())
}
fn with_full_decorations() -> HashSet<OutputComponent> {
OutputComponent::Full
.components(Default::default())
.iter()
.cloned()
.collect()
}
|
