有没有办法使用 docopt 从命令行传递 u8 的向量？

Question

有没有办法让用户提示输入括号内的字节并用逗号或类似的东西分隔？

./main bytes [0, 1, 2, 3, 4, 5]

我设法让它看起来像这样：

./main bytes 0 1 2 3 4 5

这是我的代码：

extern crate docopt;
#[macro_use]
extern crate serde_derive;

use docopt::Docopt;

const USAGE: &'static str = "
    Puzzle Solver.

    Usage:
      puzzle_solver string <text>
      puzzle_solver bytes [<bin>...] 
      puzzle_solver (-h | --help)
      puzzle_solver --version

    Options:
      -h --help     Show this screen.
      --version     Show version.
    ";

#[derive(Debug, Deserialize)]
struct Args {
    cmd_string: bool,
    arg_text: Option<String>,
    cmd_bytes: bool,
    arg_bin: Option<Vec<u8>>,
}

fn main() {
    let args: Args = Docopt::new(USAGE)
        .and_then(|d| d.deserialize())
        .unwrap_or_else(|e| e.exit());

    println!("ARGS: {:?}", args);
}

Answer 1

可以，但您必须手动实施Deserialize。

Vec<u8> 已经实现了 Deserialize，并且该实现不知道包含逗号分隔的括号列表的字符串，docopt::Deserializer 也不知道，因为传递列表的正常方式在命令行上是逐个元素的。所以你必须创建一个新的类型来反序列化你想要的格式。

当然，如果你想把它当作Vec<u8>，你也可以为Bytes实现Deref<Target = Vec<u8>>和DerefMut。，但在这种情况下可能没问题。

extern crate docopt;
extern crate serde;
#[macro_use]
extern crate serde_derive;

use docopt::Docopt;
use serde::de;
use std::fmt;

const USAGE: &'static str = "
    Puzzle Solver.

    Usage:
      puzzle_solver string <text>
      puzzle_solver bytes <bin>
      puzzle_solver (-h | --help)
      puzzle_solver --version

    Options:
      -h --help     Show this screen.
      --version     Show version.
    ";

#[derive(Debug, Deserialize)]
struct Args {
    cmd_string: bool,
    arg_text: Option<String>,
    cmd_bytes: bool,
    arg_bin: Option<Bytes>,
}

#[derive(Debug)]
struct Bytes(Vec<u8>);

impl<'de> de::Deserialize<'de> for Bytes {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: de::Deserializer<'de>,
    {
        struct BytesVisitor;

        impl<'de> de::Visitor<'de> for BytesVisitor {
            type Value = Bytes;

            fn expecting(&self, formatter: &mut fmt::Formatter) -> Result<(), fmt::Error> {
                write!(formatter, "a bracketed, comma-delimited string")
            }

            fn visit_str<E: de::Error>(self, v: &str) -> Result<Self::Value, E> {
                let v = if v.starts_with('[') && v.ends_with(']') {
                    &v[1..v.len() - 1]
                } else {
                    return Err(E::custom(format!("expected a bracketed list, got {:?}", v)));
                };
                let values: Result<Vec<u8>, _> = v.split(",").map(|s| s.trim().parse()).collect();
                Ok(Bytes(values.map_err(E::custom)?))
            }
        }

        deserializer.deserialize_str(BytesVisitor)
    }
}

Here it is in the playground. 这些是我为使其正常工作所做的更改：

将 [<bin>...] 替换为 <bin>，这样 docopt 就会知道寻找单个事物而不是一系列...事物。（如果你不这样做，docopt 实际上只会抛出一个空字符串。）
在 Vec<u8> 周围引入新类型 Bytes 包装器。
Implement serde::de::Deserialize Bytes。这需要创建一个实现 serde::de::Visitor 特征的结构，将分离字符串的代码放在其 visit_str 方法中，并将访问者传递给 deserialize_str，后者告诉 Deserializer 期待一个字符串并将其传递给访问者的 visit_str.

直到快完成我才意识到，但您可以改为实现 visit_seq，并使其解析 bytes [1, 2, 3]（不引用列表）。但我不会，因为这违背了命令行约定；如果您无论如何都使用 shell 来拆分参数，那么您应该完全接受 bytes 1 2 3.

有没有办法使用 docopt 从命令行传递 u8 的向量？

Is there a way to use docopt to pass a vector of u8 from the command line?

rust

docopt