Improve readability

Using `Path`s for paths expresses intent more clearly.
This commit is contained in:
Aleksey Kladov 2021-03-02 11:15:49 +03:00 committed by David Peter
parent b489fc75c9
commit 35347c2310
8 changed files with 42 additions and 33 deletions

View File

@ -6,6 +6,8 @@
## Other ## Other
- `Input::ordinary_file` and `Input::with_name` now accept `Path` rather than `OsStr` see #1571 (@matklad)
## Syntaxes ## Syntaxes
## New themes ## New themes

View File

@ -327,7 +327,7 @@ mod tests {
writeln!(temp_file, "{}", first_line).unwrap(); writeln!(temp_file, "{}", first_line).unwrap();
} }
let input = Input::ordinary_file(file_path.as_os_str()); let input = Input::ordinary_file(&file_path);
let dummy_stdin: &[u8] = &[]; let dummy_stdin: &[u8] = &[];
let mut opened_input = input.open(dummy_stdin, None).unwrap(); let mut opened_input = input.open(dummy_stdin, None).unwrap();
@ -341,7 +341,7 @@ mod tests {
fn syntax_for_file_with_content_os(&self, file_name: &OsStr, first_line: &str) -> String { fn syntax_for_file_with_content_os(&self, file_name: &OsStr, first_line: &str) -> String {
let file_path = self.temp_dir.path().join(file_name); let file_path = self.temp_dir.path().join(file_name);
let input = Input::from_reader(Box::new(BufReader::new(first_line.as_bytes()))) let input = Input::from_reader(Box::new(BufReader::new(first_line.as_bytes())))
.with_name(Some(file_path.as_os_str())); .with_name(Some(&file_path));
let dummy_stdin: &[u8] = &[]; let dummy_stdin: &[u8] = &[];
let mut opened_input = input.open(dummy_stdin, None).unwrap(); let mut opened_input = input.open(dummy_stdin, None).unwrap();
@ -366,7 +366,7 @@ mod tests {
} }
fn syntax_for_stdin_with_content(&self, file_name: &str, content: &[u8]) -> String { fn syntax_for_stdin_with_content(&self, file_name: &str, content: &[u8]) -> String {
let input = Input::stdin().with_name(Some(OsStr::new(file_name))); let input = Input::stdin().with_name(Some(file_name));
let mut opened_input = input.open(content, None).unwrap(); let mut opened_input = input.open(content, None).unwrap();
self.assets self.assets
@ -521,7 +521,7 @@ mod tests {
.expect("creation of directory succeeds"); .expect("creation of directory succeeds");
symlink(&file_path, &file_path_symlink).expect("creation of symbolic link succeeds"); symlink(&file_path, &file_path_symlink).expect("creation of symbolic link succeeds");
let input = Input::ordinary_file(file_path_symlink.as_os_str()); let input = Input::ordinary_file(&file_path_symlink);
let dummy_stdin: &[u8] = &[]; let dummy_stdin: &[u8] = &[];
let mut opened_input = input.open(dummy_stdin, None).unwrap(); let mut opened_input = input.open(dummy_stdin, None).unwrap();

View File

@ -1,6 +1,6 @@
use std::collections::HashSet; use std::collections::HashSet;
use std::env; use std::env;
use std::ffi::OsStr; use std::path::Path;
use std::str::FromStr; use std::str::FromStr;
use atty::{self, Stream}; use atty::{self, Stream};
@ -248,16 +248,19 @@ impl App {
} }
_ => {} _ => {}
} }
let filenames: Option<Vec<&str>> = self let filenames: Option<Vec<&Path>> = self
.matches .matches
.values_of("file-name") .values_of_os("file-name")
.map(|values| values.collect()); .map(|values| values.map(Path::new).collect());
let mut filenames_or_none: Box<dyn Iterator<Item = _>> = match filenames { let mut filenames_or_none: Box<dyn Iterator<Item = Option<&Path>>> = match filenames {
Some(ref filenames) => Box::new(filenames.iter().map(|name| Some(OsStr::new(*name)))), Some(filenames) => Box::new(filenames.into_iter().map(Some)),
None => Box::new(std::iter::repeat(None)), None => Box::new(std::iter::repeat(None)),
}; };
let files: Option<Vec<&OsStr>> = self.matches.values_of_os("FILE").map(|vs| vs.collect()); let files: Option<Vec<&Path>> = self
.matches
.values_of_os("FILE")
.map(|vs| vs.map(Path::new).collect());
if files.is_none() { if files.is_none() {
return Ok(vec![new_stdin_input( return Ok(vec![new_stdin_input(

View File

@ -1,15 +1,15 @@
use bat::input::Input; use bat::input::Input;
use std::ffi::OsStr; use std::path::Path;
pub fn new_file_input<'a>(file: &'a OsStr, name: Option<&'a OsStr>) -> Input<'a> { pub fn new_file_input<'a>(file: &'a Path, name: Option<&'a Path>) -> Input<'a> {
named(Input::ordinary_file(file), name.or(Some(file))) named(Input::ordinary_file(file), name.or(Some(file)))
} }
pub fn new_stdin_input(name: Option<&OsStr>) -> Input { pub fn new_stdin_input(name: Option<&Path>) -> Input {
named(Input::stdin(), name) named(Input::stdin(), name)
} }
fn named<'a>(input: Input<'a>, name: Option<&OsStr>) -> Input<'a> { fn named<'a>(input: Input<'a>, name: Option<&Path>) -> Input<'a> {
if let Some(provided_name) = name { if let Some(provided_name) = name {
let mut input = input.with_name(Some(provided_name)); let mut input = input.with_name(Some(provided_name));
input.description_mut().set_kind(Some("File".to_owned())); input.description_mut().set_kind(Some("File".to_owned()));

View File

@ -9,7 +9,6 @@ mod directories;
mod input; mod input;
use std::collections::{HashMap, HashSet}; use std::collections::{HashMap, HashSet};
use std::ffi::OsStr;
use std::io; use std::io;
use std::io::{BufReader, Write}; use std::io::{BufReader, Write};
use std::path::Path; use std::path::Path;
@ -269,7 +268,7 @@ fn run() -> Result<bool> {
run_cache_subcommand(cache_matches)?; run_cache_subcommand(cache_matches)?;
Ok(true) Ok(true)
} else { } else {
let inputs = vec![Input::ordinary_file(OsStr::new("cache"))]; let inputs = vec![Input::ordinary_file("cache")];
let config = app.config(&inputs)?; let config = app.config(&inputs)?;
run_controller(inputs, &config) run_controller(inputs, &config)

View File

@ -1,7 +1,6 @@
#![cfg(feature = "git")] #![cfg(feature = "git")]
use std::collections::HashMap; use std::collections::HashMap;
use std::ffi::OsStr;
use std::fs; use std::fs;
use std::path::Path; use std::path::Path;
@ -17,7 +16,7 @@ pub enum LineChange {
pub type LineChanges = HashMap<u32, LineChange>; pub type LineChanges = HashMap<u32, LineChange>;
pub fn get_git_diff(filename: &OsStr) -> Option<LineChanges> { pub fn get_git_diff(filename: &Path) -> Option<LineChanges> {
let repo = Repository::discover(&filename).ok()?; let repo = Repository::discover(&filename).ok()?;
let repo_path_absolute = fs::canonicalize(repo.workdir()?).ok()?; let repo_path_absolute = fs::canonicalize(repo.workdir()?).ok()?;

View File

@ -1,7 +1,7 @@
use std::convert::TryFrom; use std::convert::TryFrom;
use std::ffi::{OsStr, OsString};
use std::fs::File; use std::fs::File;
use std::io::{self, BufRead, BufReader, Read}; use std::io::{self, BufRead, BufReader, Read};
use std::path::{Path, PathBuf};
use clircle::{Clircle, Identifier}; use clircle::{Clircle, Identifier};
use content_inspector::{self, ContentType}; use content_inspector::{self, ContentType};
@ -69,7 +69,7 @@ impl InputDescription {
} }
pub(crate) enum InputKind<'a> { pub(crate) enum InputKind<'a> {
OrdinaryFile(OsString), OrdinaryFile(PathBuf),
StdIn, StdIn,
CustomReader(Box<dyn Read + 'a>), CustomReader(Box<dyn Read + 'a>),
} }
@ -86,7 +86,7 @@ impl<'a> InputKind<'a> {
#[derive(Clone, Default)] #[derive(Clone, Default)]
pub(crate) struct InputMetadata { pub(crate) struct InputMetadata {
pub(crate) user_provided_name: Option<OsString>, pub(crate) user_provided_name: Option<PathBuf>,
} }
pub struct Input<'a> { pub struct Input<'a> {
@ -96,7 +96,7 @@ pub struct Input<'a> {
} }
pub(crate) enum OpenedInputKind { pub(crate) enum OpenedInputKind {
OrdinaryFile(OsString), OrdinaryFile(PathBuf),
StdIn, StdIn,
CustomReader, CustomReader,
} }
@ -109,8 +109,11 @@ pub(crate) struct OpenedInput<'a> {
} }
impl<'a> Input<'a> { impl<'a> Input<'a> {
pub fn ordinary_file(path: &OsStr) -> Self { pub fn ordinary_file(path: impl AsRef<Path>) -> Self {
let kind = InputKind::OrdinaryFile(path.to_os_string()); Self::_ordinary_file(path.as_ref())
}
fn _ordinary_file(path: &Path) -> Self {
let kind = InputKind::OrdinaryFile(path.to_path_buf());
Input { Input {
description: kind.description(), description: kind.description(),
metadata: InputMetadata::default(), metadata: InputMetadata::default(),
@ -140,7 +143,10 @@ impl<'a> Input<'a> {
matches!(self.kind, InputKind::StdIn) matches!(self.kind, InputKind::StdIn)
} }
pub fn with_name(mut self, provided_name: Option<&OsStr>) -> Self { pub fn with_name(mut self, provided_name: Option<impl AsRef<Path>>) -> Self {
self._with_name(provided_name.as_ref().map(|it| it.as_ref()))
}
pub fn _with_name(mut self, provided_name: Option<&Path>) -> Self {
if let Some(name) = provided_name { if let Some(name) = provided_name {
self.description.name = name.to_string_lossy().to_string() self.description.name = name.to_string_lossy().to_string()
} }

View File

@ -1,5 +1,5 @@
use std::ffi::OsStr;
use std::io::Read; use std::io::Read;
use std::path::Path;
use console::Term; use console::Term;
use syntect::parsing::SyntaxReference; use syntect::parsing::SyntaxReference;
@ -71,7 +71,7 @@ impl<'a> PrettyPrinter<'a> {
} }
/// Add a file which should be pretty-printed /// Add a file which should be pretty-printed
pub fn input_file(&mut self, path: impl AsRef<OsStr>) -> &mut Self { pub fn input_file(&mut self, path: impl AsRef<Path>) -> &mut Self {
self.input(Input::from_file(path).kind("File")) self.input(Input::from_file(path).kind("File"))
} }
@ -79,7 +79,7 @@ impl<'a> PrettyPrinter<'a> {
pub fn input_files<I, P>(&mut self, paths: I) -> &mut Self pub fn input_files<I, P>(&mut self, paths: I) -> &mut Self
where where
I: IntoIterator<Item = P>, I: IntoIterator<Item = P>,
P: AsRef<OsStr>, P: AsRef<Path>,
{ {
self.inputs(paths.into_iter().map(Input::from_file)) self.inputs(paths.into_iter().map(Input::from_file))
} }
@ -297,8 +297,8 @@ impl<'a> Input<'a> {
} }
/// A new input from a file. /// A new input from a file.
pub fn from_file(path: impl AsRef<OsStr>) -> Self { pub fn from_file(path: impl AsRef<Path>) -> Self {
input::Input::ordinary_file(path.as_ref()).into() input::Input::ordinary_file(path).into()
} }
/// A new input from bytes. /// A new input from bytes.
@ -313,8 +313,8 @@ impl<'a> Input<'a> {
/// The filename of the input. /// The filename of the input.
/// This affects syntax detection and changes the default header title. /// This affects syntax detection and changes the default header title.
pub fn name(mut self, name: impl AsRef<OsStr>) -> Self { pub fn name(mut self, name: impl AsRef<Path>) -> Self {
self.input = self.input.with_name(Some(name.as_ref())); self.input = self.input.with_name(Some(name));
self self
} }