git.iko.soy/bat
on 9e11d66 help sign in
bat/src/inputfile.rs
133 lines 3.6 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) -> Result<InputFileReader<'a>> {
18 let mut first_line = vec![];
19 if reader.read_until(b'\n', &mut first_line)? == 0 {
20 return Err(ErrorKind::ImmediateEOF.into());
21 }
23 let content_type = content_inspector::inspect(&first_line[..]);
25 if content_type == ContentType::UTF_16LE {
26 reader.read_until(0x00, &mut first_line).ok();
27 }
29 Ok(InputFileReader {
30 inner: Box::new(reader),
31 first_line,
32 content_type,
33 })
34 }
36 pub fn read_line(&mut self, buf: &mut Vec<u8>) -> io::Result<bool> {
37 if self.first_line.is_empty() {
38 let res = self.inner.read_until(b'\n', buf).map(|size| size > 0)?;
40 if self.content_type == ContentType::UTF_16LE {
41 self.inner.read_until(0x00, buf).ok();
42 }
44 Ok(res)
45 } else {
46 buf.append(&mut self.first_line);
47 Ok(true)
48 }
49 }
52#[derive(Debug, Clone, Copy, PartialEq)]
53pub enum InputFile<'a> {
54 StdIn,
55 Ordinary(&'a str),
56 ThemePreviewFile,
59impl<'a> InputFile<'a> {
60 pub fn get_reader(&self, stdin: &'a io::Stdin) -> Result<InputFileReader> {
61 match self {
62 InputFile::StdIn => InputFileReader::new(stdin.lock()),
63 InputFile::Ordinary(filename) => {
64 let file = File::open(filename).map_err(|e| format!("'{}': {}", filename, e))?;
66 if file.metadata()?.is_dir() {
67 return Err(format!("'{}' is a directory.", filename).into());
68 }
70 InputFileReader::new(BufReader::new(file))
71 }
72 InputFile::ThemePreviewFile => InputFileReader::new(THEME_PREVIEW_FILE),
73 }
74 }
77#[test]
78fn basic() {
79 let content = b"#!/bin/bash\necho hello";
80 let mut reader = InputFileReader::new(&content[..]).unwrap();
82 assert_eq!(b"#!/bin/bash\n", &reader.first_line[..]);
84 let mut buffer = vec![];
86 let res = reader.read_line(&mut buffer);
87 assert!(res.is_ok());
88 assert_eq!(true, res.unwrap());
89 assert_eq!(b"#!/bin/bash\n", &buffer[..]);
91 buffer.clear();
93 let res = reader.read_line(&mut buffer);
94 assert!(res.is_ok());
95 assert_eq!(true, res.unwrap());
96 assert_eq!(b"echo hello", &buffer[..]);
98 buffer.clear();
100 let res = reader.read_line(&mut buffer);
101 assert!(res.is_ok());
102 assert_eq!(false, res.unwrap());
103 assert!(buffer.is_empty());
106#[test]
107fn utf16le() {
108 let content = b"\xFF\xFE\x73\x00\x0A\x00\x64\x00";
109 let mut reader = InputFileReader::new(&content[..]).unwrap();
111 assert_eq!(b"\xFF\xFE\x73\x00\x0A\x00", &reader.first_line[..]);
113 let mut buffer = vec![];
115 let res = reader.read_line(&mut buffer);
116 assert!(res.is_ok());
117 assert_eq!(true, res.unwrap());
118 assert_eq!(b"\xFF\xFE\x73\x00\x0A\x00", &buffer[..]);
120 buffer.clear();
122 let res = reader.read_line(&mut buffer);
123 assert!(res.is_ok());
124 assert_eq!(true, res.unwrap());
125 assert_eq!(b"\x64\x00", &buffer[..]);
127 buffer.clear();
129 let res = reader.read_line(&mut buffer);
130 assert!(res.is_ok());
131 assert_eq!(false, res.unwrap());
132 assert!(buffer.is_empty());