fd/src/output.rs

135 lines
4.1 KiB
Rust
Raw Normal View History

2017-10-21 10:16:03 +02:00
// Copyright (c) 2017 fd developers
// Licensed under the Apache License, Version 2.0
// <LICENSE-APACHE or http://www.apache.org/licenses/LICENSE-2.0>
// or the MIT license <LICENSE-MIT or http://opensource.org/licenses/MIT>,
// at your option. All files in the project carrying such
// notice may not be copied, modified, or distributed except
// according to those terms.
2018-03-25 19:48:09 +02:00
use internal::{FdOptions, EXITCODE_ERROR, EXITCODE_SIGINT};
use fshelper::is_executable;
use lscolors::LsColors;
2017-10-10 08:01:17 +02:00
2018-03-25 17:49:32 +02:00
use std::process;
2017-10-10 08:01:17 +02:00
use std::io::{self, Write};
use std::ops::Deref;
2018-01-01 12:16:43 +01:00
use std::path::{self, Component, Path, PathBuf};
use std::sync::Arc;
use std::sync::atomic::{AtomicBool, Ordering};
2017-10-10 08:01:17 +02:00
use ansi_term;
2018-02-24 17:02:14 +01:00
/// Remove the `./` prefix from a path.
fn strip_current_dir<'a>(pathbuf: &'a PathBuf) -> &'a Path {
let mut iter = pathbuf.components();
let mut iter_next = iter.clone();
if iter_next.next() == Some(Component::CurDir) {
iter.next();
}
iter.as_path()
}
pub fn print_entry(entry: &PathBuf, config: &FdOptions, wants_to_quit: &Arc<AtomicBool>) {
2018-02-24 17:02:14 +01:00
let path = if entry.is_absolute() {
entry.as_path()
} else {
strip_current_dir(entry)
};
2017-10-14 03:44:24 +02:00
let r = if let Some(ref ls_colors) = config.ls_colors {
print_entry_colorized(path, config, ls_colors, &wants_to_quit)
} else {
print_entry_uncolorized(path, config)
};
2017-10-10 08:01:17 +02:00
if r.is_err() {
// Probably a broken pipe. Exit gracefully.
2018-01-03 10:00:22 +01:00
process::exit(EXITCODE_ERROR);
}
}
2017-10-10 08:01:17 +02:00
fn print_entry_colorized(
path: &Path,
config: &FdOptions,
ls_colors: &LsColors,
wants_to_quit: &Arc<AtomicBool>,
) -> io::Result<()> {
let default_style = ansi_term::Style::default();
2017-10-10 08:01:17 +02:00
let stdout = io::stdout();
let mut handle = stdout.lock();
2017-10-14 03:44:24 +02:00
// Separator to use before the current component.
let mut separator = String::new();
2017-10-14 03:44:24 +02:00
// Full path to the current component.
let mut component_path = PathBuf::new();
// Traverse the path and colorize each component
2017-10-14 03:44:24 +02:00
for component in path.components() {
let comp_str = component.as_os_str().to_string_lossy();
component_path.push(Path::new(comp_str.deref()));
2017-10-10 08:01:17 +02:00
2017-10-14 03:44:24 +02:00
let style = get_path_style(&component_path, ls_colors).unwrap_or(&default_style);
2017-10-10 08:01:17 +02:00
2017-10-14 03:44:24 +02:00
write!(handle, "{}{}", separator, style.paint(comp_str))?;
2017-10-10 08:01:17 +02:00
2017-10-14 03:44:24 +02:00
// Determine separator to print before next component.
separator = match component {
// Prefix needs no separator, as it is always followed by RootDir.
Component::Prefix(_) => String::new(),
// RootDir is already a separator.
Component::RootDir => String::new(),
// Everything else uses a separator that is painted the same way as the component.
_ => style.paint(path::MAIN_SEPARATOR.to_string()).to_string(),
};
if wants_to_quit.load(Ordering::Relaxed) {
write!(handle, "\n")?;
2018-01-03 10:00:22 +01:00
process::exit(EXITCODE_SIGINT);
}
}
2017-10-10 08:01:17 +02:00
if config.null_separator {
write!(handle, "\0")
2017-10-10 08:01:17 +02:00
} else {
writeln!(handle, "")
}
}
2017-10-10 08:01:17 +02:00
2017-10-14 03:44:24 +02:00
fn print_entry_uncolorized(path: &Path, config: &FdOptions) -> io::Result<()> {
let separator = if config.null_separator { "\0" } else { "\n" };
2017-10-10 08:01:17 +02:00
2017-10-14 03:44:24 +02:00
let path_str = path.to_string_lossy();
write!(&mut io::stdout(), "{}{}", path_str, separator)
}
fn get_path_style<'a>(path: &Path, ls_colors: &'a LsColors) -> Option<&'a ansi_term::Style> {
if path.symlink_metadata()
.map(|md| md.file_type().is_symlink())
.unwrap_or(false)
{
return Some(&ls_colors.symlink);
}
let metadata = path.metadata();
if metadata.as_ref().map(|md| md.is_dir()).unwrap_or(false) {
Some(&ls_colors.directory)
} else if metadata.map(|md| is_executable(&md)).unwrap_or(false) {
Some(&ls_colors.executable)
2018-01-01 12:16:43 +01:00
} else if let Some(filename_style) = path.file_name()
.and_then(|n| n.to_str())
.and_then(|n| ls_colors.filenames.get(n))
2017-10-14 03:44:24 +02:00
{
Some(filename_style)
2018-01-01 12:16:43 +01:00
} else if let Some(extension_style) = path.extension()
.and_then(|e| e.to_str())
.and_then(|e| ls_colors.extensions.get(e))
2017-10-14 03:44:24 +02:00
{
Some(extension_style)
} else {
None
}
}