git.iko.soy/bat
on 35a1236 help sign in
bat/src/controller.rs
104 lines 3.5 kB history raw
1use std::fs::File;
2use std::io::{self, BufRead, BufReader, Write};
4use app::{Config, InputFile};
5use assets::HighlightingAssets;
6use errors::*;
7use line_range::LineRange;
8use output::OutputType;
9use printer::{InteractivePrinter, Printer, SimplePrinter};
11const THEME_PREVIEW_FILE: &[u8] = include_bytes!("../assets/theme_preview.rs");
13pub struct Controller<'a> {
14 config: &'a Config<'a>,
15 assets: &'a HighlightingAssets,
18impl<'b> Controller<'b> {
19 pub fn new<'a>(config: &'a Config, assets: &'a HighlightingAssets) -> Controller<'a> {
20 Controller { config, assets }
21 }
23 pub fn run(&self) -> Result<bool> {
24 let mut output_type = OutputType::from_mode(self.config.paging_mode);
25 let writer = output_type.handle()?;
26 let mut no_errors: bool = true;
28 for filename in &self.config.files {
29 let result = if self.config.loop_through {
30 let mut printer = SimplePrinter::new();
31 self.print_file(&mut printer, writer, *filename)
32 } else {
33 let mut printer = InteractivePrinter::new(&self.config, &self.assets, *filename);
34 self.print_file(&mut printer, writer, *filename)
35 };
37 if let Err(error) = result {
38 handle_error(&error);
39 no_errors = false;
40 }
41 }
43 Ok(no_errors)
44 }
46 fn print_file<'a, P: Printer>(
47 &self,
48 printer: &mut P,
49 writer: &mut Write,
50 filename: InputFile<'a>,
51 ) -> Result<()> {
52 let stdin = io::stdin();
53 {
54 let reader: Box<BufRead> = match filename {
55 InputFile::StdIn => Box::new(stdin.lock()),
56 InputFile::Ordinary(filename) => Box::new(BufReader::new(File::open(filename)?)),
57 InputFile::ThemePreviewFile => Box::new(THEME_PREVIEW_FILE),
58 };
60 printer.print_header(writer, filename)?;
61 self.print_file_ranges(printer, writer, reader, &self.config.line_range)?;
62 printer.print_footer(writer)?;
63 }
64 Ok(())
65 }
67 fn print_file_ranges<'a, P: Printer>(
68 &self,
69 printer: &mut P,
70 writer: &mut Write,
71 mut reader: Box<BufRead + 'a>,
72 line_ranges: &Option<LineRange>,
73 ) -> Result<()> {
74 let mut line_buffer = Vec::new();
76 let mut line_number: usize = 1;
78 while reader.read_until(b'\n', &mut line_buffer)? > 0 {
79 {
80 match line_ranges {
81 &Some(ref range) => {
82 if line_number < range.lower {
83 // Call the printer in case we need to call the syntax highlighter
84 // for this line. However, set `out_of_range` to `true`.
85 printer.print_line(true, writer, line_number, &line_buffer)?;
86 } else if line_number > range.upper {
87 // no more lines in range, exit early
88 break;
89 } else {
90 printer.print_line(false, writer, line_number, &line_buffer)?;
91 }
92 }
93 &None => {
94 printer.print_line(false, writer, line_number, &line_buffer)?;
95 }
96 }
98 line_number += 1;
99 }
100 line_buffer.clear();
101 }
102 Ok(())
103 }