diff options
author | Sebastian Thiel <sebastian.thiel@icloud.com> | 2020-11-15 11:42:45 +0800 |
---|---|---|
committer | Sebastian Thiel <sebastian.thiel@icloud.com> | 2020-11-15 11:42:45 +0800 |
commit | 5782c4ff99b70ea101ed2f36711a456fd4e4e37b (patch) | |
tree | 04a5c4d2f28e42b313d606b366a86c974be104ae /src | |
parent | 88753aa6d6a7d23a7d4334b7913655009adfc079 (diff) |
switch from structup to clap 3 beta.2
…to see if the order of arguments is corrected.
It's not the case.
Related to #71
Diffstat (limited to 'src')
-rw-r--r-- | src/main.rs | 4 | ||||
-rw-r--r-- | src/options.rs | 79 |
2 files changed, 52 insertions, 31 deletions
diff --git a/src/main.rs b/src/main.rs index 09d5704..9313c26 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,8 +1,8 @@ #![forbid(unsafe_code)] use anyhow::{anyhow, Result}; +use clap::Clap; use dua::{ByteFormat, TraversalSorting}; use std::{fs, io, io::Write, path::PathBuf, process}; -use structopt::StructOpt; #[cfg(any(feature = "tui-unix", feature = "tui-crossplatform"))] mod interactive; @@ -11,7 +11,7 @@ mod options; fn main() -> Result<()> { use options::Command::*; - let opt: options::Args = options::Args::from_iter(wild::args_os()); + let opt: options::Args = options::Args::parse_from(wild::args_os()); let walk_options = dua::WalkOptions { threads: opt.threads.unwrap_or(0), byte_format: opt.format.map(Into::into).unwrap_or(ByteFormat::Metric), diff --git a/src/options.rs b/src/options.rs index a7d7618..d62c199 100644 --- a/src/options.rs +++ b/src/options.rs @@ -1,20 +1,41 @@ +use clap::Clap; use dua::ByteFormat as LibraryByteFormat; use std::path::PathBuf; -use structopt::{clap::arg_enum, StructOpt}; +use std::str::FromStr; -arg_enum! { - #[derive(PartialEq, Debug)] - pub enum ByteFormat { - Metric, - Binary, - Bytes, - GB, - GiB, - MB, - MiB +#[derive(PartialEq, Debug)] +pub enum ByteFormat { + Metric, + Binary, + Bytes, + GB, + GiB, + MB, + MiB, +} + +impl FromStr for ByteFormat { + type Err = String; + + fn from_str(s: &str) -> Result<Self, Self::Err> { + Ok(match s { + "metric" | "Metric" => ByteFormat::Metric, + "binary" | "Binary" => ByteFormat::Binary, + "bytes" | "Bytes" => ByteFormat::Bytes, + "GB" | "Gb" | "gb" => ByteFormat::GB, + "GiB" | "gib" => ByteFormat::GiB, + "MB" | "Mb" | "mb" => ByteFormat::MB, + "MiB" | "mib" => ByteFormat::MiB, + _ => return Err(format!("Invalid byte format: {:?}", s)), + }) } } +impl ByteFormat { + const VARIANTS: &'static [&'static str] = + &["metric", "binary", "bytes", "MB", "MiB", "GB", "GiB"]; +} + impl From<ByteFormat> for LibraryByteFormat { fn from(input: ByteFormat) -> Self { match input { @@ -29,16 +50,16 @@ impl From<ByteFormat> for LibraryByteFormat { } } -#[derive(Debug, StructOpt)] -#[structopt(name = "dua", about = "A tool to learn about disk usage, fast!")] -#[structopt(setting = structopt::clap::AppSettings::ColoredHelp)] +#[derive(Debug, Clap)] +#[clap(name = "dua", about = "A tool to learn about disk usage, fast!")] +#[clap(setting = clap::AppSettings::ColoredHelp)] pub struct Args { - #[structopt(subcommand)] + #[clap(subcommand)] pub command: Option<Command>, /// The amount of threads to use. Defaults to the amount of logical processors. /// Set to 1 to use only a single thread. - #[structopt(short = "t", long = "threads")] + #[clap(short = 't', long = "threads")] pub threads: Option<usize>, /// The format with which to print byte counts. @@ -49,51 +70,51 @@ pub struct Args { /// GiB - only gibibytes /// MB - only megabytes /// MiB - only mebibytes - #[structopt(short = "f", long, case_insensitive = true, possible_values(&ByteFormat::variants()))] + #[clap(short = 'f', long, case_insensitive = true, possible_values(&ByteFormat::VARIANTS))] pub format: Option<ByteFormat>, /// Display apparent size instead of disk usage. - #[structopt(short = "A", long)] + #[clap(short = 'A', long)] pub apparent_size: bool, /// Count hard-linked files each time they are seen - #[structopt(short = "l", long)] + #[clap(short = 'l', long)] pub count_hard_links: bool, /// If set, we will not cross filesystems or traverse mount points - #[structopt(short = "x", long)] + #[clap(short = 'x', long)] pub stay_on_filesystem: bool, /// One or more input files or directories. If unset, we will use all entries in the current working directory. - #[structopt(parse(from_os_str))] + #[clap(parse(from_os_str))] pub input: Vec<PathBuf>, } -#[derive(Debug, StructOpt)] +#[derive(Debug, Clap)] pub enum Command { /// Launch the terminal user interface #[cfg(any(feature = "tui-unix", feature = "tui-crossplatform"))] - #[structopt(name = "interactive", visible_alias = "i")] + #[clap(name = "interactive", visible_alias = "i")] Interactive { /// One or more input files or directories. If unset, we will use all entries in the current working directory. - #[structopt(parse(from_os_str))] + #[clap(parse(from_os_str))] input: Vec<PathBuf>, }, /// Aggregrate the consumed space of one or more directories or files - #[structopt(name = "aggregate", visible_alias = "a")] + #[clap(name = "aggregate", visible_alias = "a")] Aggregate { /// If set, print additional statistics about the file traversal to stderr - #[structopt(long = "stats")] + #[clap(long = "stats")] statistics: bool, /// If set, paths will be printed in their order of occurrence on the command-line. /// Otherwise they are sorted by their size in bytes, ascending. - #[structopt(long)] + #[clap(long)] no_sort: bool, /// If set, no total column will be computed for multiple inputs - #[structopt(long)] + #[clap(long)] no_total: bool, /// One or more input files or directories. If unset, we will use all entries in the current working directory. - #[structopt(parse(from_os_str))] + #[clap(parse(from_os_str))] input: Vec<PathBuf>, }, } |