summaryrefslogtreecommitdiffstats
path: root/src/options.rs
blob: 5c30573d690934e7244f7e767b66ad74fb7e5d80 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
use dua::ByteFormat as LibraryByteFormat;
use std::path::PathBuf;
use structopt::{clap::arg_enum, StructOpt};

arg_enum! {
    #[derive(PartialEq, Debug)]
    pub enum ByteFormat {
        HumanMetric,
        HumanBinary,
        Bytes
    }
}

impl From<ByteFormat> for LibraryByteFormat {
    fn from(input: ByteFormat) -> Self {
        match input {
            ByteFormat::HumanMetric => LibraryByteFormat::Metric,
            ByteFormat::HumanBinary => LibraryByteFormat::Binary,
            ByteFormat::Bytes => LibraryByteFormat::Bytes,
        }
    }
}

#[derive(Debug, StructOpt)]
#[structopt(name = "dua", about = "A tool to learn about disk usage, fast!")]
#[structopt(raw(setting = "structopt::clap::AppSettings::ColoredHelp"))]
pub struct Args {
    #[structopt(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")]
    pub threads: Option<usize>,

    /// The format with which to print byte counts.
    /// HumanMetric - uses 1000 as base (default)
    /// HumanBinary - uses 1024 as base
    /// Bytes - plain bytes without any formatting
    #[structopt(short = "f", long = "format")]
    pub format: Option<ByteFormat>,

    /// One or more input files. If unset, we will assume the current directory
    #[structopt(parse(from_os_str))]
    pub input: Vec<PathBuf>,
}

#[derive(Debug, StructOpt)]
pub enum Command {
    /// Aggregrate the consumed space of one or more directories or files
    #[structopt(name = "aggregate", alias = "a")]
    Aggregate {
        /// If set, print additional statistics about the file traversal to stderr
        #[structopt(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 = "no-sort")]
        no_sort: bool,
        /// If set, no total column will be computed for multiple inputs
        #[structopt(long = "no-total")]
        no_total: bool,
        /// One or more input files. If unset, we will assume the current directory
        #[structopt(parse(from_os_str))]
        input: Vec<PathBuf>,
    },
}