git.iko.soy/bat
on 14e6b41 help sign in
bat/src/controller.rs
103 lines 3.3 kB history raw
1use std::io::{self, Write};
3use app::Config;
4use assets::HighlightingAssets;
5use errors::*;
6use inputfile::{InputFile, InputFileReader};
7use line_range::{LineRanges, RangeCheckResult};
8use output::OutputType;
9use printer::{InteractivePrinter, Printer, SimplePrinter};
11pub struct Controller<'a> {
12 config: &'a Config<'a>,
13 assets: &'a HighlightingAssets,
16impl<'b> Controller<'b> {
17 pub fn new<'a>(config: &'a Config, assets: &'a HighlightingAssets) -> Controller<'a> {
18 Controller { config, assets }
19 }
21 pub fn run(&self) -> Result<bool> {
22 let mut output_type = OutputType::from_mode(self.config.paging_mode, self.config.pager)?;
23 let writer = output_type.handle()?;
24 let mut no_errors: bool = true;
26 let stdin = io::stdin();
28 for input_file in &self.config.files {
29 match input_file.get_reader(&stdin) {
30 Err(error) => {
31 handle_error(&error);
32 no_errors = false;
33 }
34 Ok(mut reader) => {
35 let result = if self.config.loop_through {
36 let mut printer = SimplePrinter::new();
37 self.print_file(reader, &mut printer, writer, *input_file)
38 } else {
39 let mut printer = InteractivePrinter::new(
40 &self.config,
41 &self.assets,
42 *input_file,
43 &mut reader,
44 );
45 self.print_file(reader, &mut printer, writer, *input_file)
46 };
48 if let Err(error) = result {
49 handle_error(&error);
50 no_errors = false;
51 }
52 }
53 }
54 }
56 Ok(no_errors)
57 }
59 fn print_file<'a, P: Printer>(
60 &self,
61 reader: InputFileReader,
62 printer: &mut P,
63 writer: &mut Write,
64 input_file: InputFile<'a>,
65 ) -> Result<()> {
66 printer.print_header(writer, input_file)?;
67 self.print_file_ranges(printer, writer, reader, &self.config.line_ranges)?;
68 printer.print_footer(writer)?;
70 Ok(())
71 }
73 fn print_file_ranges<'a, P: Printer>(
74 &self,
75 printer: &mut P,
76 writer: &mut Write,
77 mut reader: InputFileReader,
78 line_ranges: &LineRanges,
79 ) -> Result<()> {
80 let mut line_buffer = Vec::new();
81 let mut line_number: usize = 1;
83 while reader.read_line(&mut line_buffer)? {
84 match line_ranges.check(line_number) {
85 RangeCheckResult::OutsideRange => {
86 // Call the printer in case we need to call the syntax highlighter
87 // for this line. However, set `out_of_range` to `true`.
88 printer.print_line(true, writer, line_number, &line_buffer)?;
89 }
90 RangeCheckResult::InRange => {
91 printer.print_line(false, writer, line_number, &line_buffer)?;
92 }
93 RangeCheckResult::AfterLastRange => {
94 break;
95 }
96 }
98 line_number += 1;
99 line_buffer.clear();
100 }
101 Ok(())
102 }