Skip to content

Commit

Permalink
refac: add support for Windows in EscapedPathDisplay
Browse files Browse the repository at this point in the history
  • Loading branch information
vrmiguel committed Jan 5, 2023
1 parent 0deb182 commit 2caeb10
Show file tree
Hide file tree
Showing 6 changed files with 37 additions and 22 deletions.
4 changes: 2 additions & 2 deletions src/archive/tar.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ use crate::{
error::FinalError,
info,
list::FileInArchive,
utils::{self, EscapedUtf8Display, FileVisibilityPolicy},
utils::{self, EscapedPathDisplay, FileVisibilityPolicy},
warning,
};

Expand Down Expand Up @@ -121,7 +121,7 @@ where
// spoken text for users using screen readers, braille displays
// and so on
if !quiet {
info!(inaccessible, "Compressing '{}'.", EscapedUtf8Display::new(path));
info!(inaccessible, "Compressing '{}'.", EscapedPathDisplay::new(path));
}

if path.is_dir() {
Expand Down
4 changes: 2 additions & 2 deletions src/archive/zip.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ use crate::{
list::FileInArchive,
utils::{
self, cd_into_same_dir_as, get_invalid_utf8_paths, pretty_format_list_of_paths, strip_cur_dir,
EscapedUtf8Display, FileVisibilityPolicy,
EscapedPathDisplay, FileVisibilityPolicy,
},
warning,
};
Expand Down Expand Up @@ -191,7 +191,7 @@ where
// spoken text for users using screen readers, braille displays
// and so on
if !quiet {
info!(inaccessible, "Compressing '{}'.", EscapedUtf8Display::new(path));
info!(inaccessible, "Compressing '{}'.", EscapedPathDisplay::new(path));
}

let metadata = match path.metadata() {
Expand Down
12 changes: 6 additions & 6 deletions src/commands/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ use crate::{
info,
list::ListOptions,
utils::{
self, pretty_format_list_of_paths, to_utf, try_infer_extension, user_wants_to_continue, EscapedUtf8Display,
self, pretty_format_list_of_paths, to_utf, try_infer_extension, user_wants_to_continue, EscapedPathDisplay,
FileVisibilityPolicy,
},
warning, Opts, QuestionAction, QuestionPolicy, Subcommand,
Expand Down Expand Up @@ -116,7 +116,7 @@ pub fn run(
let formats = extension::extensions_from_path(&output_path);

let first_format = formats.first().ok_or_else(|| {
let output_path = EscapedUtf8Display::new(&output_path);
let output_path = EscapedPathDisplay::new(&output_path);
FinalError::with_title(format!("Cannot compress to '{output_path}'."))
.detail("You shall supply the compression format")
.hint("Try adding supported extensions (see --help):")
Expand All @@ -139,7 +139,7 @@ pub fn run(
// To file.tar.bz.xz
let suggested_output_path = build_archive_file_suggestion(&output_path, ".tar")
.expect("output path should contain a compression format");
let output_path = EscapedUtf8Display::new(&output_path);
let output_path = EscapedPathDisplay::new(&output_path);
let first_detail_message = if is_multiple_inputs {
"You are trying to compress multiple files."
} else {
Expand All @@ -163,7 +163,7 @@ pub fn run(
if let Some(format) = formats.iter().skip(1).find(|format| format.is_archive()) {
let error = FinalError::with_title(format!(
"Cannot compress to '{}'.",
EscapedUtf8Display::new(&output_path)
EscapedPathDisplay::new(&output_path)
))
.detail(format!("Found the format '{}' in an incorrect position.", format))
.detail(format!(
Expand All @@ -177,7 +177,7 @@ pub fn run(
.hint(format!(
"Otherwise, remove the last '{}' from '{}'.",
format,
EscapedUtf8Display::new(&output_path)
EscapedPathDisplay::new(&output_path)
));

return Err(error.into());
Expand Down Expand Up @@ -213,7 +213,7 @@ pub fn run(
eprintln!("{red}FATAL ERROR:\n", red = *colors::RED);
eprintln!(
" Ouch failed to delete the file '{}'.",
EscapedUtf8Display::new(&output_path)
EscapedPathDisplay::new(&output_path)
);
eprintln!(" Please delete it manually.");
eprintln!(" This file is corrupted if compression didn't finished.");
Expand Down
33 changes: 24 additions & 9 deletions src/utils/formatting.rs
Original file line number Diff line number Diff line change
@@ -1,27 +1,42 @@
use std::{borrow::Cow, fmt::Display, path::Path};

use bstr::BStr;

use crate::CURRENT_DIRECTORY;

/// Converts invalid UTF-8 bytes to the Unicode replacement codepoint (�) in its Display implementation.
pub struct EscapedUtf8Display<'a> {
bstr: &'a BStr,
pub struct EscapedPathDisplay<'a> {
path: &'a Path,
}

impl<'a> EscapedUtf8Display<'a> {
impl<'a> EscapedPathDisplay<'a> {
pub fn new(path: &'a Path) -> Self {
Self { path }
}
}

#[cfg(unix)]
impl Display for EscapedPathDisplay<'_> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
use std::os::unix::prelude::OsStrExt;

let bytes = path.as_os_str().as_bytes();
let bstr = bstr::BStr::new(self.path.as_os_str().as_bytes());

Self { bstr: BStr::new(bytes) }
write!(f, "{}", bstr)
}
}

impl Display for EscapedUtf8Display<'_> {
#[cfg(windows)]
impl Display for EscapedPathDisplay<'_> {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.bstr)
use std::{char, fmt::Write, os::windows::prelude::OsStrExt};

let utf16 = self.path.as_os_str().encode_wide();
let chars = char::decode_utf16(utf16).map(|decoded| decoded.unwrap_or(char::REPLACEMENT_CHARACTER));

for char in chars {
f.write_char(char)?;
}

Ok(())
}
}

Expand Down
4 changes: 2 additions & 2 deletions src/utils/fs.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ use std::{
use fs_err as fs;

use super::user_wants_to_overwrite;
use crate::{extension::Extension, info, utils::EscapedUtf8Display, QuestionPolicy};
use crate::{extension::Extension, info, utils::EscapedPathDisplay, QuestionPolicy};

/// Remove `path` asking the user to overwrite if necessary.
///
Expand Down Expand Up @@ -41,7 +41,7 @@ pub fn create_dir_if_non_existent(path: &Path) -> crate::Result<()> {
fs::create_dir_all(path)?;
// creating a directory is an important change to the file system we
// should always inform the user about
info!(accessible, "directory {} created.", EscapedUtf8Display::new(path));
info!(accessible, "directory {} created.", EscapedPathDisplay::new(path));
}
Ok(())
}
Expand Down
2 changes: 1 addition & 1 deletion src/utils/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ mod fs;
mod question;

pub use file_visibility::FileVisibilityPolicy;
pub use formatting::{nice_directory_display, pretty_format_list_of_paths, strip_cur_dir, to_utf, EscapedUtf8Display};
pub use formatting::{nice_directory_display, pretty_format_list_of_paths, strip_cur_dir, to_utf, EscapedPathDisplay};
pub use fs::{
cd_into_same_dir_as, clear_path, create_dir_if_non_existent, is_symlink, remove_file_or_dir, try_infer_extension,
};
Expand Down

0 comments on commit 2caeb10

Please sign in to comment.