✨ Add support --color

Author: ChanTsune

Cargo.lock

@@ -11,6 +11,7 @@ rust-version = "1.85"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+anstream = "0.6.16"
 anyhow = { version = "1.0.100", features = ["backtrace"] }
 base64 = "0.22.1"
 bitflags = "2.10.0"

cli/Cargo.toml

@@ -25,6 +25,8 @@ pub struct Cli {
     pub(crate) commands: Commands,
     #[command(flatten)]
     pub(crate) verbosity: VerbosityArgs,
+    #[command(flatten)]
+    pub(crate) color: ColorArgs,
     #[arg(
         long,
         global = true,
@@ -70,6 +72,18 @@ impl VerbosityArgs {
     }
 }
 
+#[derive(Parser, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Default)]
+pub(crate) struct ColorArgs {
+    #[arg(
+        long,
+        global = true,
+        value_name = "WHEN",
+        default_value = "auto",
+        help = "Control color output"
+    )]
+    pub(crate) color: ColorChoice,
+}
+
 #[derive(Subcommand, Clone, Debug)]
 pub(crate) enum Commands {
     #[command(visible_alias = "c", about = "Create archive")]

cli/src/cli.rs

@@ -1,10 +1,12 @@
 mod argon2id_params;
+mod color_choice;
 mod compression_level;
 mod datetime;
 mod pbkdf2_sha256_params;
 mod private_chunk_type;
 
 pub(crate) use argon2id_params::Argon2idParams;
+pub(crate) use color_choice::ColorChoice;
 pub(crate) use compression_level::{DeflateLevel, XzLevel, ZstdLevel};
 pub use datetime::DateTime;
 pub(crate) use pbkdf2_sha256_params::Pbkdf2Sha256Params;

cli/src/cli/value.rs

@@ -0,0 +1,20 @@
+use clap::ValueEnum;
+
+#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Default, ValueEnum)]
+pub(crate) enum ColorChoice {
+    #[default]
+    Auto,
+    Always,
+    Never,
+}
+
+impl From<ColorChoice> for anstream::ColorChoice {
+    #[inline]
+    fn from(value: ColorChoice) -> Self {
+        match value {
+            ColorChoice::Auto => anstream::ColorChoice::Auto,
+            ColorChoice::Always => anstream::ColorChoice::Always,
+            ColorChoice::Never => anstream::ColorChoice::Never,
+        }
+    }
+}

cli/src/cli/value/color_choice.rs

@@ -66,7 +66,10 @@ impl Command for Cli {
             Commands::Create(cmd) => cmd.execute(),
             Commands::Append(cmd) => cmd.execute(),
             Commands::Extract(cmd) => cmd.execute(),
-            Commands::List(cmd) => cmd.execute(),
+            Commands::List(mut cmd) => {
+                cmd.color = self.color.color;
+                cmd.execute()
+            }
             Commands::Split(cmd) => cmd.execute(),
             Commands::Concat(cmd) => cmd.execute(),
             Commands::Strip(cmd) => cmd.execute(),

cli/src/command.rs

@@ -2,7 +2,7 @@
 use crate::command::core::run_read_entries_mem;
 use crate::{
     chunk,
-    cli::{FileArgsCompat, PasswordArgs},
+    cli::{ColorChoice, FileArgsCompat, PasswordArgs},
     command::{
         Command, ask_password,
         core::{PathFilter, collect_split_archives, read_paths, run_read_entries},
@@ -52,6 +52,8 @@ use tabled::{
     group(ArgGroup::new("null-requires").arg("null").requires("exclude_from")),
 )]
 pub(crate) struct ListCommand {
+    #[arg(skip)]
+    pub(crate) color: ColorChoice,
     #[arg(short, long, help = "Display extended file metadata as a table")]
     pub(crate) long: bool,
     #[arg(short, long, help = "Add a header row to each column")]
@@ -286,6 +288,7 @@ fn list_archive(args: ListCommand) -> anyhow::Result<()> {
         classify: args.classify,
         format: args.format,
         out_to_stderr: false,
+        color: args.color,
     };
     let archive = args.file.archive();
     let files = args.file.files();
@@ -365,6 +368,7 @@ pub(crate) struct ListOptions {
     pub(crate) classify: bool,
     pub(crate) format: Option<Format>,
     pub(crate) out_to_stderr: bool,
+    pub(crate) color: ColorChoice,
 }
 
 pub(crate) fn run_list_archive<'a>(
@@ -444,9 +448,11 @@ fn print_entries<'a>(
         .collect::<Vec<_>>();
     globs.ensure_all_matched()?;
     if options.out_to_stderr {
-        print_formatted_entries(entries, &options, io::stderr().lock())
+        let out = anstream::AutoStream::new(io::stderr().lock(), options.color.into());
+        print_formatted_entries(entries, &options, out)
     } else {
-        print_formatted_entries(entries, &options, io::stdout().lock())
+        let out = anstream::AutoStream::new(io::stdout().lock(), options.color.into());
+        print_formatted_entries(entries, &options, out)
     }
 }
 

cli/src/command/list.rs

@@ -1,6 +1,7 @@
 use crate::{
     cli::{
-        CipherAlgorithmArgs, CompressionAlgorithmArgs, DateTime, HashAlgorithmArgs, PasswordArgs,
+        CipherAlgorithmArgs, ColorChoice, CompressionAlgorithmArgs, DateTime, HashAlgorithmArgs,
+        PasswordArgs,
     },
     command::{
         Command,
@@ -681,6 +682,7 @@ fn run_list_archive(args: StdioCommand) -> anyhow::Result<()> {
             Format::Line
         }),
         out_to_stderr: args.to_stdout,
+        color: ColorChoice::Auto,
     };
     let files_globs = GlobPatterns::new(args.files.iter().map(|it| it.as_str()))?;