git.iko.soy/bat
on v0.9.0 help sign in
bat/src/inputfile.rs
131 lines 3.5 kB history raw
1use std::fs::File;
2use std::io::{self, BufRead, BufReader};
4use content_inspector::{self, ContentType};
6use errors::*;
8const THEME_PREVIEW_FILE: &[u8] = include_bytes!("../assets/theme_preview.rs");
10pub struct InputFileReader<'a> {
11 inner: Box<dyn BufRead + 'a>,
12 pub first_line: Vec<u8>,
13 pub content_type: ContentType,
16impl<'a> InputFileReader<'a> {
17 fn new<R: BufRead + 'a>(mut reader: R) -> InputFileReader<'a> {
18 let mut first_line = vec![];
19 reader.read_until(b'\n', &mut first_line).ok();
21 let content_type = content_inspector::inspect(&first_line[..]);
23 if content_type == ContentType::UTF_16LE {
24 reader.read_until(0x00, &mut first_line).ok();
25 }
27 InputFileReader {
28 inner: Box::new(reader),
29 first_line,
30 content_type,
31 }
32 }
34 pub fn read_line(&mut self, buf: &mut Vec<u8>) -> io::Result<bool> {
35 if self.first_line.is_empty() {
36 let res = self.inner.read_until(b'\n', buf).map(|size| size > 0)?;
38 if self.content_type == ContentType::UTF_16LE {
39 self.inner.read_until(0x00, buf).ok();
40 }
42 Ok(res)
43 } else {
44 buf.append(&mut self.first_line);
45 Ok(true)
46 }
47 }
50#[derive(Debug, Clone, Copy, PartialEq)]
51pub enum InputFile<'a> {
52 StdIn,
53 Ordinary(&'a str),
54 ThemePreviewFile,
57impl<'a> InputFile<'a> {
58 pub fn get_reader(&self, stdin: &'a io::Stdin) -> Result<InputFileReader> {
59 match self {
60 InputFile::StdIn => Ok(InputFileReader::new(stdin.lock())),
61 InputFile::Ordinary(filename) => {
62 let file = File::open(filename)?;
64 if file.metadata()?.is_dir() {
65 return Err(format!("'{}' is a directory.", filename).into());
66 }
68 Ok(InputFileReader::new(BufReader::new(file)))
69 }
70 InputFile::ThemePreviewFile => Ok(InputFileReader::new(THEME_PREVIEW_FILE)),
71 }
72 }
75#[test]
76fn basic() {
77 let content = b"#!/bin/bash\necho hello";
78 let mut reader = InputFileReader::new(&content[..]);
80 assert_eq!(b"#!/bin/bash\n", &reader.first_line[..]);
82 let mut buffer = vec![];
84 let res = reader.read_line(&mut buffer);
85 assert!(res.is_ok());
86 assert_eq!(true, res.unwrap());
87 assert_eq!(b"#!/bin/bash\n", &buffer[..]);
89 buffer.clear();
91 let res = reader.read_line(&mut buffer);
92 assert!(res.is_ok());
93 assert_eq!(true, res.unwrap());
94 assert_eq!(b"echo hello", &buffer[..]);
96 buffer.clear();
98 let res = reader.read_line(&mut buffer);
99 assert!(res.is_ok());
100 assert_eq!(false, res.unwrap());
101 assert!(buffer.is_empty());
104#[test]
105fn utf16le() {
106 let content = b"\xFF\xFE\x73\x00\x0A\x00\x64\x00";
107 let mut reader = InputFileReader::new(&content[..]);
109 assert_eq!(b"\xFF\xFE\x73\x00\x0A\x00", &reader.first_line[..]);
111 let mut buffer = vec![];
113 let res = reader.read_line(&mut buffer);
114 assert!(res.is_ok());
115 assert_eq!(true, res.unwrap());
116 assert_eq!(b"\xFF\xFE\x73\x00\x0A\x00", &buffer[..]);
118 buffer.clear();
120 let res = reader.read_line(&mut buffer);
121 assert!(res.is_ok());
122 assert_eq!(true, res.unwrap());
123 assert_eq!(b"\x64\x00", &buffer[..]);
125 buffer.clear();
127 let res = reader.read_line(&mut buffer);
128 assert!(res.is_ok());
129 assert_eq!(false, res.unwrap());
130 assert!(buffer.is_empty());