git.iko.soy/bat
on 90e7d2f help sign in
bat/src/inputfile.rs
168 lines 4.5 kB history raw
1use std::ffi::OsStr;
2use std::fs::File;
3use std::io::{self, BufRead, BufReader};
5use content_inspector::{self, ContentType};
7use crate::errors::*;
9const THEME_PREVIEW_FILE: &[u8] = include_bytes!("../assets/theme_preview.rs");
11pub struct InputFileReader<'a> {
12 inner: Box<dyn BufRead + 'a>,
13 pub(crate) first_line: Vec<u8>,
14 pub(crate) content_type: Option<ContentType>,
17impl<'a> InputFileReader<'a> {
18 fn new<R: BufRead + 'a>(mut reader: R) -> InputFileReader<'a> {
19 let mut first_line = vec![];
20 reader.read_until(b'\n', &mut first_line).ok();
22 let content_type = if first_line.is_empty() {
23 None
24 } else {
25 Some(content_inspector::inspect(&first_line[..]))
26 };
28 if content_type == Some(ContentType::UTF_16LE) {
29 reader.read_until(0x00, &mut first_line).ok();
30 }
32 InputFileReader {
33 inner: Box::new(reader),
34 first_line,
35 content_type,
36 }
37 }
39 pub(crate) fn read_line(&mut self, buf: &mut Vec<u8>) -> io::Result<bool> {
40 if self.first_line.is_empty() {
41 let res = self.inner.read_until(b'\n', buf).map(|size| size > 0)?;
43 if self.content_type == Some(ContentType::UTF_16LE) {
44 self.inner.read_until(0x00, buf).ok();
45 }
47 Ok(res)
48 } else {
49 buf.append(&mut self.first_line);
50 Ok(true)
51 }
52 }
55#[derive(Debug, Clone, Copy, PartialEq)]
56pub struct OrdinaryFile<'a> {
57 path: &'a OsStr,
58 user_provided_name: Option<&'a OsStr>,
61impl<'a> OrdinaryFile<'a> {
62 pub fn from_path(path: &'a OsStr) -> OrdinaryFile<'a> {
63 OrdinaryFile {
64 path,
65 user_provided_name: None,
66 }
67 }
69 pub fn from_path_with_name(
70 path: &'a OsStr,
71 user_provided_name: Option<&'a OsStr>,
72 ) -> OrdinaryFile<'a> {
73 OrdinaryFile {
74 path,
75 user_provided_name,
76 }
77 }
79 pub fn filename(&self) -> &'a OsStr {
80 self.user_provided_name.unwrap_or_else(|| self.path)
81 }
84#[derive(Debug, Clone, Copy, PartialEq)]
85pub enum InputFile<'a> {
86 StdIn(Option<&'a OsStr>),
87 Ordinary(OrdinaryFile<'a>),
88 ThemePreviewFile,
91impl<'a> InputFile<'a> {
92 pub(crate) fn get_reader(&self, stdin: &'a io::Stdin) -> Result<InputFileReader> {
93 match self {
94 InputFile::StdIn(_) => Ok(InputFileReader::new(stdin.lock())),
95 InputFile::Ordinary(ofile) => {
96 let file = File::open(ofile.path)
97 .map_err(|e| format!("'{}': {}", ofile.path.to_string_lossy(), e))?;
99 if file.metadata()?.is_dir() {
100 return Err(
101 format!("'{}' is a directory.", ofile.path.to_string_lossy()).into(),
102 );
103 }
105 Ok(InputFileReader::new(BufReader::new(file)))
106 }
107 InputFile::ThemePreviewFile => Ok(InputFileReader::new(THEME_PREVIEW_FILE)),
108 }
109 }
112#[test]
113fn basic() {
114 let content = b"#!/bin/bash\necho hello";
115 let mut reader = InputFileReader::new(&content[..]);
117 assert_eq!(b"#!/bin/bash\n", &reader.first_line[..]);
119 let mut buffer = vec![];
121 let res = reader.read_line(&mut buffer);
122 assert!(res.is_ok());
123 assert_eq!(true, res.unwrap());
124 assert_eq!(b"#!/bin/bash\n", &buffer[..]);
126 buffer.clear();
128 let res = reader.read_line(&mut buffer);
129 assert!(res.is_ok());
130 assert_eq!(true, res.unwrap());
131 assert_eq!(b"echo hello", &buffer[..]);
133 buffer.clear();
135 let res = reader.read_line(&mut buffer);
136 assert!(res.is_ok());
137 assert_eq!(false, res.unwrap());
138 assert!(buffer.is_empty());
141#[test]
142fn utf16le() {
143 let content = b"\xFF\xFE\x73\x00\x0A\x00\x64\x00";
144 let mut reader = InputFileReader::new(&content[..]);
146 assert_eq!(b"\xFF\xFE\x73\x00\x0A\x00", &reader.first_line[..]);
148 let mut buffer = vec![];
150 let res = reader.read_line(&mut buffer);
151 assert!(res.is_ok());
152 assert_eq!(true, res.unwrap());
153 assert_eq!(b"\xFF\xFE\x73\x00\x0A\x00", &buffer[..]);
155 buffer.clear();
157 let res = reader.read_line(&mut buffer);
158 assert!(res.is_ok());
159 assert_eq!(true, res.unwrap());
160 assert_eq!(b"\x64\x00", &buffer[..]);
162 buffer.clear();
164 let res = reader.read_line(&mut buffer);
165 assert!(res.is_ok());
166 assert_eq!(false, res.unwrap());
167 assert!(buffer.is_empty());