2

问题

我有一个带有不同选项的命令，这些选项的相对顺序对命令的语义很重要。例如，在 中command --config A --some-option --config-file B --random-option --config C --another-option --more-options --config-file D，的相对顺序A, B, C, D很重要，因为它会影响命令的含义。

如果我只是定义选项如下：

#[derive(Debug, StructOpt)]
pub struct Command {
    #[structopt(long = "config")]
    configs: Vec<String>,

    #[structopt(long = "config-file")]
    config_files: Vec<String>,
}

然后我会得到两个向量，configs = [A, C]但是和config_files = [B, D]中的元素之间的相对顺序已经丢失了。configsconfig_files

想法

自定义解析函数

这个想法是提供一个自定义解析函数，并在解析每个选项时使用计数器记录索引。不幸的是，解析函数没有按照命令定义的原始顺序调用。

fn get_next_atomic_int() -> usize {
    static ATOMIC_COUNTER: Lazy<AtomicUsize> = Lazy::new(|| AtomicUsize::new(0));
    ATOMIC_COUNTER.fetch_add(1, Ordering::Relaxed)
}

fn parse_passthrough_string_ordered(arg: &str) -> (String, usize) {
    (arg.to_owned(), get_next_atomic_int())
}

#[derive(Debug, StructOpt)]
#[structopt(name = "command"]
pub struct Command {
    #[structopt(long = "config-file", parse(from_str = parse_passthrough_string_ordered))]
    config_files: Vec<(String, usize)>,
    
    #[structopt(short = "c", long = "config", parse(from_str = parse_passthrough_string_ordered))]
    configs: Vec<(String, usize)>,
}

别名

我可以为该选项添加别名，如下所示：

#[derive(Debug, StructOpt)]
pub struct Command {
    #[structopt(long = "config", visible_alias = "config-file")]
    configs: Vec<String>,
}

这种方法有两个问题：

  • 我需要一种方法来区分选项是否通过--configor--config-file传递（仅通过检查值并不总是可以弄清楚值是如何传递的）。
  • 我无法为可见别名提供简短选项。

相同的向量，多个选项

另一个想法是附加多个structopt指令，以便两个选项使用相同的底层向量。不幸的是，它不起作用——structopt只使用最后一个指令。就像是：

#[derive(Debug)]
enum Config {
    File(String),
    Literal(String),
}

fn parse_config_literal(arg: &str) -> Config {
    Config::Literal(arg.to_owned())
}

fn parse_config_file(arg: &str) -> Config {
    Config::File(arg.to_owned())
}

#[derive(Debug, StructOpt)]
#[structopt(name = "example")]
struct Opt {
    #[structopt(long = "--config-file", parse(from_str = parse_config_file))]
    #[structopt(short = "-c", long = "--config", parse(from_str = parse_config_literal))]
    options: Vec<Config>,
}

恢复订单

我可以尝试通过搜索解析值来恢复原始顺序。但这意味着我将不得不复制相当多的解析逻辑（例如，需要支持传递--config=X, --config X，需要处理X作为另一个选项的输入出现等）。

我宁愿有办法可靠地获得原件，而不是丢失订单并尝试以可能脆弱的方式恢复它。

4

1 回答 1

2

正如@TeXitoi所述，我错过了ArgMatches::indices_of()为我们提供所需信息的功能。

use structopt::StructOpt;

#[derive(Debug)]
enum Config {
    File(String),
    Literal(String),
}

fn parse_config_literal(arg: &str) -> Config {
    Config::Literal(arg.to_owned())
}

fn parse_config_file(arg: &str) -> Config {
    Config::File(arg.to_owned())
}

#[derive(Debug, StructOpt)]
#[structopt(name = "example")]
struct Opt {
    #[structopt(short = "c", long = "config", parse(from_str = parse_config_literal))]
    configs: Vec<Config>,

    #[structopt(long = "config-file", parse(from_str = parse_config_file))]
    config_files: Vec<Config>,
}

fn with_indices<'a, I: IntoIterator + 'a>(
    collection: I,
    name: &str,
    matches: &'a structopt::clap::ArgMatches,
) -> impl Iterator<Item = (usize, I::Item)> + 'a {
    matches
        .indices_of(name)
        .into_iter()
        .flatten()
        .zip(collection)
}

fn main() {
    let args = vec!["example", "--config", "A", "--config-file", "B", "--config", "C", "--config-file", "D"];
    
    let clap = Opt::clap();
    let matches = clap.get_matches_from(args);
    let opt = Opt::from_clap(&matches);

    println!("configs:");
    for (i, c) in with_indices(&opt.configs, "configs", &matches) {
        println!("{}: {:#?}", i, c);
    }

    println!("\nconfig-files:");
    for (i, c) in with_indices(&opt.config_files, "config-files", &matches) {
        println!("{}: {:#?}", i, c);
    }
}
于 2021-04-29T12:29:55.997 回答