Skip to content

Commit

Permalink
ls: use the gnu_legacy feature from lscolors
Browse files Browse the repository at this point in the history
  • Loading branch information
sylvestre committed Dec 6, 2023
1 parent 80b1ccd commit d35e183
Show file tree
Hide file tree
Showing 4 changed files with 46 additions and 32 deletions.
1 change: 1 addition & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -289,6 +289,7 @@ itertools = "0.12.0"
libc = "0.2.150"
lscolors = { version = "0.16.0", default-features = false, features = [
"nu-ansi-term",
"gnu_legacy",
] }
memchr = "2"
memmap2 = "0.9"
Expand Down
2 changes: 1 addition & 1 deletion src/uu/ls/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ number_prefix = { workspace = true }
uutils_term_grid = { workspace = true }
terminal_size = { workspace = true }
glob = { workspace = true }
lscolors = { workspace = true }
lscolors = { workspace = true, features = ["gnu_legacy"] }
uucore = { workspace = true, features = [
"entries",
"fs",
Expand Down
6 changes: 5 additions & 1 deletion src/uu/ls/src/ls.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3094,7 +3094,11 @@ fn display_file_name(
fn color_name(name: String, path: &Path, md: Option<&Metadata>, ls_colors: &LsColors) -> String {
match ls_colors.style_for_path_with_metadata(path, md) {
Some(style) => {
return style.to_nu_ansi_term_style().paint(name).to_string();
return style
.to_nu_ansi_term_style()
.reset_before_style()
.paint(name)
.to_string();
}
None => name,
}
Expand Down
69 changes: 39 additions & 30 deletions tests/by-util/test_ls.rs
Original file line number Diff line number Diff line change
Expand Up @@ -864,11 +864,11 @@ fn test_ls_zero() {
.succeeds()
.stdout_only("\"0-test-zero\"\x00\"2-test-zero\"\x00\"3-test-zero\"\x00");

scene
.ucmd()
.args(&["--zero", "--color=always"])
.succeeds()
.stdout_only("\x1b[1;34m0-test-zero\x1b[0m\x002-test-zero\x003-test-zero\x00");
let result = scene.ucmd().args(&["--zero", "--color=always"]).succeeds();
assert_eq!(
result.stdout_str(),
"\u{1b}[0m\u{1b}[01;34m0-test-zero\x1b[0m\x002-test-zero\x003-test-zero\x00"
);

scene
.ucmd()
Expand Down Expand Up @@ -921,12 +921,9 @@ fn test_ls_zero() {
"\"0-test-zero\"\x00\"1\\ntest-zero\"\x00\"2-test-zero\"\x00\"3-test-zero\"\x00",
);

scene
.ucmd()
.args(&["--zero", "--color=always"])
.succeeds()
.stdout_only(
"\x1b[1;34m0-test-zero\x1b[0m\x001\ntest-zero\x002-test-zero\x003-test-zero\x00",
let result = scene.ucmd().args(&["--zero", "--color=always"]).succeeds();
assert_eq!(result.stdout_str(),
"\u{1b}[0m\u{1b}[01;34m0-test-zero\x1b[0m\x001\ntest-zero\x002-test-zero\x003-test-zero\x00",
);

scene
Expand Down Expand Up @@ -1202,7 +1199,7 @@ fn test_ls_long_symlink_color() {
}

fn capture_colored_string(input: &str) -> (Color, Name) {
let colored_name = Regex::new(r"\x1b\[([0-9;]+)m(.+)\x1b\[0m").unwrap();
let colored_name = Regex::new(r"\x1b\[0m\x1b\[([0-9;]+)m(.+)\x1b\[0m").unwrap();
match colored_name.captures(input) {
Some(captures) => (
captures.get(1).unwrap().as_str().to_string(),
Expand Down Expand Up @@ -1977,6 +1974,20 @@ fn test_ls_recursive_1() {
.stdout_is(out);
}

// Function to convert a string to its ASCII representation
fn to_ascii_representation(input: &str) -> String {
input
.chars()
.map(|c| {
if c.is_ascii_control() || !c.is_ascii() {
format!("\\x{:02x}", c as u32)
} else {
c.to_string()
}
})
.collect::<String>()
}

#[test]
fn test_ls_color() {
let scene = TestScenario::new(util_name!());
Expand All @@ -1995,9 +2006,9 @@ fn test_ls_color() {
at.touch(nested_file);
at.touch("test-color");

let a_with_colors = "\x1b[1;34ma\x1b[0m";
let z_with_colors = "\x1b[1;34mz\x1b[0m";
let nested_dir_with_colors = "\x1b[1;34mnested_dir\x1b[0m"; // spell-checker:disable-line
let a_with_colors = "\x1b[0m\x1b[01;34ma\x1b[0m";
let z_with_colors = "\x1b[0m\x1b[01;34mz\x1b[0m";
let nested_dir_with_colors = "\x1b[0m\x1b[01;34mnested_dir\x1b[0m\x0anested_file"; // spell-checker:disable-line

// Color is disabled by default
let result = scene.ucmd().succeeds();
Expand All @@ -2006,12 +2017,9 @@ fn test_ls_color() {

// Color should be enabled
for param in ["--color", "--col", "--color=always", "--col=always"] {
scene
.ucmd()
.arg(param)
.succeeds()
.stdout_contains(a_with_colors)
.stdout_contains(z_with_colors);
let result = scene.ucmd().arg(param).succeeds();
assert!(result.stdout_str().contains(a_with_colors));
assert!(result.stdout_str().contains(z_with_colors));
}

// Color should be disabled
Expand All @@ -2020,12 +2028,8 @@ fn test_ls_color() {
assert!(!result.stdout_str().contains(z_with_colors));

// Nested dir should be shown and colored
scene
.ucmd()
.arg("--color")
.arg("a")
.succeeds()
.stdout_contains(nested_dir_with_colors);
let result = scene.ucmd().arg("--color").arg("a").succeeds();
assert!(result.stdout_str().contains(nested_dir_with_colors));

// No output
scene
Expand All @@ -2037,13 +2041,18 @@ fn test_ls_color() {

// The colors must not mess up the grid layout
at.touch("b");
scene
let result = scene
.ucmd()
.arg("--color")
.arg("-w=15")
.arg("-C")
.succeeds()
.stdout_only(format!("{a_with_colors} test-color\nb {z_with_colors}\n"));
.succeeds();
let expected = format!("{} test-color\x0ab {}\n", a_with_colors, z_with_colors);
assert_eq!(
to_ascii_representation(result.stdout_str()),
to_ascii_representation(&expected)
);
assert_eq!(result.stdout_str(), expected);
}

#[cfg(unix)]
Expand Down

0 comments on commit d35e183

Please sign in to comment.