+++ /dev/null
-// Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
-// file at the top-level directory of this distribution and at
-// http://rust-lang.org/COPYRIGHT.
-//
-// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
-// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
-// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
-// option. This file may not be copied, modified, or distributed
-// except according to those terms.
-
-//! Simple getopt alternative.
-//!
-//! Construct a vector of options, either by using `reqopt`, `optopt`, and `optflag`
-//! or by building them from components yourself, and pass them to `getopts`,
-//! along with a vector of actual arguments (not including `argv[0]`). You'll
-//! either get a failure code back, or a match. You'll have to verify whether
-//! the amount of 'free' arguments in the match is what you expect. Use `opt_*`
-//! accessors to get argument values out of the matches object.
-//!
-//! Single-character options are expected to appear on the command line with a
-//! single preceding dash; multiple-character options are expected to be
-//! proceeded by two dashes. Options that expect an argument accept their
-//! argument following either a space or an equals sign. Single-character
-//! options don't require the space.
-//!
-//! # Example
-//!
-//! The following example shows simple command line parsing for an application
-//! that requires an input file to be specified, accepts an optional output
-//! file name following `-o`, and accepts both `-h` and `--help` as optional flags.
-//!
-//! ```{.rust}
-//! #![feature(rustc_private)]
-//!
-//! extern crate getopts;
-//! use getopts::{optopt,optflag,getopts,OptGroup,usage};
-//! use std::env;
-//!
-//! fn do_work(inp: &str, out: Option<String>) {
-//! println!("{}", inp);
-//! match out {
-//! Some(x) => println!("{}", x),
-//! None => println!("No Output"),
-//! }
-//! }
-//!
-//! fn print_usage(program: &str, opts: &[OptGroup]) {
-//! let brief = format!("Usage: {} [options]", program);
-//! print!("{}", usage(&brief, opts));
-//! }
-//!
-//! fn main() {
-//! let args: Vec<String> = env::args().collect();
-//!
-//! let program = args[0].clone();
-//!
-//! let opts = &[
-//! optopt("o", "", "set output file name", "NAME"),
-//! optflag("h", "help", "print this help menu")
-//! ];
-//! let matches = match getopts(&args[1..], opts) {
-//! Ok(m) => { m }
-//! Err(f) => { panic!(f.to_string()) }
-//! };
-//! if matches.opt_present("h") {
-//! print_usage(&program, opts);
-//! return;
-//! }
-//! let output = matches.opt_str("o");
-//! let input = if !matches.free.is_empty() {
-//! matches.free[0].clone()
-//! } else {
-//! print_usage(&program, opts);
-//! return;
-//! };
-//! do_work(&input, output);
-//! }
-//! ```
-
-#![doc(html_logo_url = "https://www.rust-lang.org/logos/rust-logo-128x128-blk-v2.png",
- html_favicon_url = "https://doc.rust-lang.org/favicon.ico",
- html_root_url = "https://doc.rust-lang.org/nightly/",
- html_playground_url = "https://play.rust-lang.org/",
- test(attr(deny(warnings))))]
-
-#![deny(missing_docs)]
-#![deny(warnings)]
-
-use self::Name::*;
-use self::HasArg::*;
-use self::Occur::*;
-use self::Fail::*;
-use self::Optval::*;
-use self::SplitWithinState::*;
-use self::Whitespace::*;
-use self::LengthLimit::*;
-
-use std::fmt;
-use std::iter::repeat;
-use std::result;
-
-/// Name of an option. Either a string or a single char.
-#[derive(Clone, PartialEq, Eq, Debug)]
-pub enum Name {
- /// A string representing the long name of an option.
- /// For example: "help"
- Long(String),
- /// A char representing the short name of an option.
- /// For example: 'h'
- Short(char),
-}
-
-/// Describes whether an option has an argument.
-#[derive(Clone, Copy, PartialEq, Eq, Debug)]
-pub enum HasArg {
- /// The option requires an argument.
- Yes,
- /// The option takes no argument.
- No,
- /// The option argument is optional.
- Maybe,
-}
-
-/// Describes how often an option may occur.
-#[derive(Clone, Copy, PartialEq, Eq, Debug)]
-pub enum Occur {
- /// The option occurs once.
- Req,
- /// The option occurs at most once.
- Optional,
- /// The option occurs zero or more times.
- Multi,
-}
-
-/// A description of a possible option.
-#[derive(Clone, PartialEq, Eq, Debug)]
-pub struct Opt {
- /// Name of the option
- pub name: Name,
- /// Whether it has an argument
- pub hasarg: HasArg,
- /// How often it can occur
- pub occur: Occur,
- /// Which options it aliases
- pub aliases: Vec<Opt>,
-}
-
-/// One group of options, e.g., both `-h` and `--help`, along with
-/// their shared description and properties.
-#[derive(Clone, PartialEq, Eq, Debug)]
-pub struct OptGroup {
- /// Short name of the option, e.g. `h` for a `-h` option
- pub short_name: String,
- /// Long name of the option, e.g. `help` for a `--help` option
- pub long_name: String,
- /// Hint for argument, e.g. `FILE` for a `-o FILE` option
- pub hint: String,
- /// Description for usage help text
- pub desc: String,
- /// Whether option has an argument
- pub hasarg: HasArg,
- /// How often it can occur
- pub occur: Occur,
-}
-
-/// Describes whether an option is given at all or has a value.
-#[derive(Clone, PartialEq, Eq, Debug)]
-enum Optval {
- Val(String),
- Given,
-}
-
-/// The result of checking command line arguments. Contains a vector
-/// of matches and a vector of free strings.
-#[derive(Clone, PartialEq, Eq, Debug)]
-pub struct Matches {
- /// Options that matched
- opts: Vec<Opt>,
- /// Values of the Options that matched
- vals: Vec<Vec<Optval>>,
- /// Free string fragments
- pub free: Vec<String>,
-}
-
-/// The type returned when the command line does not conform to the
-/// expected format. Use the `Debug` implementation to output detailed
-/// information.
-#[derive(Clone, PartialEq, Eq, Debug)]
-pub enum Fail {
- /// The option requires an argument but none was passed.
- ArgumentMissing(String),
- /// The passed option is not declared among the possible options.
- UnrecognizedOption(String),
- /// A required option is not present.
- OptionMissing(String),
- /// A single occurrence option is being used multiple times.
- OptionDuplicated(String),
- /// There's an argument being passed to a non-argument option.
- UnexpectedArgument(String),
-}
-
-/// The type of failure that occurred.
-#[derive(Copy, Clone, PartialEq, Eq, Debug)]
-#[allow(missing_docs)]
-pub enum FailType {
- ArgumentMissing_,
- UnrecognizedOption_,
- OptionMissing_,
- OptionDuplicated_,
- UnexpectedArgument_,
-}
-
-/// The result of parsing a command line with a set of options.
-pub type Result = result::Result<Matches, Fail>;
-
-impl Name {
- fn from_str(nm: &str) -> Name {
- if nm.len() == 1 {
- Short(nm.chars().next().unwrap())
- } else {
- Long(nm.to_owned())
- }
- }
-
- fn to_string(&self) -> String {
- match *self {
- Short(ch) => ch.to_string(),
- Long(ref s) => s.to_owned(),
- }
- }
-}
-
-impl OptGroup {
- /// Translate OptGroup into Opt.
- /// (Both short and long names correspond to different Opts).
- pub fn long_to_short(&self) -> Opt {
- let OptGroup {
- short_name,
- long_name,
- hasarg,
- occur,
- ..
- } = (*self).clone();
-
- match (short_name.len(), long_name.len()) {
- (0, 0) => panic!("this long-format option was given no name"),
- (0, _) => {
- Opt {
- name: Long((long_name)),
- hasarg,
- occur,
- aliases: Vec::new(),
- }
- }
- (1, 0) => {
- Opt {
- name: Short(short_name.chars().next().unwrap()),
- hasarg,
- occur,
- aliases: Vec::new(),
- }
- }
- (1, _) => {
- Opt {
- name: Long((long_name)),
- hasarg,
- occur,
- aliases: vec![Opt {
- name: Short(short_name.chars().next().unwrap()),
- hasarg,
- occur,
- aliases: Vec::new(),
- }],
- }
- }
- _ => panic!("something is wrong with the long-form opt"),
- }
- }
-}
-
-impl Matches {
- fn opt_vals(&self, nm: &str) -> Vec<Optval> {
- match find_opt(&self.opts[..], Name::from_str(nm)) {
- Some(id) => self.vals[id].clone(),
- None => panic!("No option '{}' defined", nm),
- }
- }
-
- fn opt_val(&self, nm: &str) -> Option<Optval> {
- let vals = self.opt_vals(nm);
- if vals.is_empty() {
- None
- } else {
- Some(vals[0].clone())
- }
- }
-
- /// Returns true if an option was matched.
- pub fn opt_present(&self, nm: &str) -> bool {
- !self.opt_vals(nm).is_empty()
- }
-
- /// Returns the number of times an option was matched.
- pub fn opt_count(&self, nm: &str) -> usize {
- self.opt_vals(nm).len()
- }
-
- /// Returns true if any of several options were matched.
- pub fn opts_present(&self, names: &[String]) -> bool {
- for nm in names {
- match find_opt(&self.opts, Name::from_str(&**nm)) {
- Some(id) if !self.vals[id].is_empty() => return true,
- _ => (),
- };
- }
- false
- }
-
- /// Returns the string argument supplied to one of several matching options or `None`.
- pub fn opts_str(&self, names: &[String]) -> Option<String> {
- for nm in names {
- if let Some(Val(ref s)) = self.opt_val(&nm[..]) {
- return Some(s.clone())
- }
- }
- None
- }
-
- /// Returns a vector of the arguments provided to all matches of the given
- /// option.
- ///
- /// Used when an option accepts multiple values.
- pub fn opt_strs(&self, nm: &str) -> Vec<String> {
- let mut acc: Vec<String> = Vec::new();
- let r = self.opt_vals(nm);
- for v in &r {
- match *v {
- Val(ref s) => acc.push((*s).clone()),
- _ => (),
- }
- }
- acc
- }
-
- /// Returns the string argument supplied to a matching option or `None`.
- pub fn opt_str(&self, nm: &str) -> Option<String> {
- let vals = self.opt_vals(nm);
- if vals.is_empty() {
- return None::<String>;
- }
- match vals[0] {
- Val(ref s) => Some((*s).clone()),
- _ => None,
- }
- }
-
-
- /// Returns the matching string, a default, or none.
- ///
- /// Returns none if the option was not present, `def` if the option was
- /// present but no argument was provided, and the argument if the option was
- /// present and an argument was provided.
- pub fn opt_default(&self, nm: &str, def: &str) -> Option<String> {
- let vals = self.opt_vals(nm);
- if vals.is_empty() {
- None
- } else {
- match vals[0] {
- Val(ref s) => Some((*s).clone()),
- _ => Some(def.to_owned()),
- }
- }
- }
-}
-
-fn is_arg(arg: &str) -> bool {
- arg.len() > 1 && arg.as_bytes()[0] == b'-'
-}
-
-fn find_opt(opts: &[Opt], nm: Name) -> Option<usize> {
- // Search main options.
- let pos = opts.iter().position(|opt| opt.name == nm);
- if pos.is_some() {
- return pos;
- }
-
- // Search in aliases.
- for candidate in opts {
- if candidate.aliases.iter().position(|opt| opt.name == nm).is_some() {
- return opts.iter().position(|opt| opt.name == candidate.name);
- }
- }
-
- None
-}
-
-/// Create a long option that is required and takes an argument.
-///
-/// * `short_name` - e.g. `"h"` for a `-h` option, or `""` for none
-/// * `long_name` - e.g. `"help"` for a `--help` option, or `""` for none
-/// * `desc` - Description for usage help
-/// * `hint` - Hint that is used in place of the argument in the usage help,
-/// e.g. `"FILE"` for a `-o FILE` option
-pub fn reqopt(short_name: &str, long_name: &str, desc: &str, hint: &str) -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: hint.to_owned(),
- desc: desc.to_owned(),
- hasarg: Yes,
- occur: Req,
- }
-}
-
-/// Create a long option that is optional and takes an argument.
-///
-/// * `short_name` - e.g. `"h"` for a `-h` option, or `""` for none
-/// * `long_name` - e.g. `"help"` for a `--help` option, or `""` for none
-/// * `desc` - Description for usage help
-/// * `hint` - Hint that is used in place of the argument in the usage help,
-/// e.g. `"FILE"` for a `-o FILE` option
-pub fn optopt(short_name: &str, long_name: &str, desc: &str, hint: &str) -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: hint.to_owned(),
- desc: desc.to_owned(),
- hasarg: Yes,
- occur: Optional,
- }
-}
-
-/// Create a long option that is optional and does not take an argument.
-///
-/// * `short_name` - e.g. `"h"` for a `-h` option, or `""` for none
-/// * `long_name` - e.g. `"help"` for a `--help` option, or `""` for none
-/// * `desc` - Description for usage help
-pub fn optflag(short_name: &str, long_name: &str, desc: &str) -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: "".to_owned(),
- desc: desc.to_owned(),
- hasarg: No,
- occur: Optional,
- }
-}
-
-/// Create a long option that can occur more than once and does not
-/// take an argument.
-///
-/// * `short_name` - e.g. `"h"` for a `-h` option, or `""` for none
-/// * `long_name` - e.g. `"help"` for a `--help` option, or `""` for none
-/// * `desc` - Description for usage help
-pub fn optflagmulti(short_name: &str, long_name: &str, desc: &str) -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: "".to_owned(),
- desc: desc.to_owned(),
- hasarg: No,
- occur: Multi,
- }
-}
-
-/// Create a long option that is optional and takes an optional argument.
-///
-/// * `short_name` - e.g. `"h"` for a `-h` option, or `""` for none
-/// * `long_name` - e.g. `"help"` for a `--help` option, or `""` for none
-/// * `desc` - Description for usage help
-/// * `hint` - Hint that is used in place of the argument in the usage help,
-/// e.g. `"FILE"` for a `-o FILE` option
-pub fn optflagopt(short_name: &str, long_name: &str, desc: &str, hint: &str) -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: hint.to_owned(),
- desc: desc.to_owned(),
- hasarg: Maybe,
- occur: Optional,
- }
-}
-
-/// Create a long option that is optional, takes an argument, and may occur
-/// multiple times.
-///
-/// * `short_name` - e.g. `"h"` for a `-h` option, or `""` for none
-/// * `long_name` - e.g. `"help"` for a `--help` option, or `""` for none
-/// * `desc` - Description for usage help
-/// * `hint` - Hint that is used in place of the argument in the usage help,
-/// e.g. `"FILE"` for a `-o FILE` option
-pub fn optmulti(short_name: &str, long_name: &str, desc: &str, hint: &str) -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: hint.to_owned(),
- desc: desc.to_owned(),
- hasarg: Yes,
- occur: Multi,
- }
-}
-
-/// Create a generic option group, stating all parameters explicitly
-pub fn opt(short_name: &str,
- long_name: &str,
- desc: &str,
- hint: &str,
- hasarg: HasArg,
- occur: Occur)
- -> OptGroup {
- let len = short_name.len();
- assert!(len == 1 || len == 0);
- OptGroup {
- short_name: short_name.to_owned(),
- long_name: long_name.to_owned(),
- hint: hint.to_owned(),
- desc: desc.to_owned(),
- hasarg,
- occur,
- }
-}
-
-impl fmt::Display for Fail {
- fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
- match *self {
- ArgumentMissing(ref nm) => write!(f, "Argument to option '{}' missing.", *nm),
- UnrecognizedOption(ref nm) => write!(f, "Unrecognized option: '{}'.", *nm),
- OptionMissing(ref nm) => write!(f, "Required option '{}' missing.", *nm),
- OptionDuplicated(ref nm) => write!(f, "Option '{}' given more than once.", *nm),
- UnexpectedArgument(ref nm) => write!(f, "Option '{}' does not take an argument.", *nm),
- }
- }
-}
-
-/// Parse command line arguments according to the provided options.
-///
-/// On success returns `Ok(Matches)`. Use methods such as `opt_present`
-/// `opt_str`, etc. to interrogate results.
-/// # Panics
-///
-/// Returns `Err(Fail)` on failure: use the `Debug` implementation of `Fail` to display
-/// information about it.
-pub fn getopts(args: &[String], optgrps: &[OptGroup]) -> Result {
- let opts: Vec<Opt> = optgrps.iter().map(|x| x.long_to_short()).collect();
- let n_opts = opts.len();
-
- fn f(_x: usize) -> Vec<Optval> {
- Vec::new()
- }
-
- let mut vals: Vec<_> = (0..n_opts).map(f).collect();
- let mut free: Vec<String> = Vec::new();
- let l = args.len();
- let mut i = 0;
- while i < l {
- let cur = args[i].clone();
- let curlen = cur.len();
- if !is_arg(&cur[..]) {
- free.push(cur);
- } else if cur == "--" {
- let mut j = i + 1;
- while j < l {
- free.push(args[j].clone());
- j += 1;
- }
- break;
- } else {
- let mut names;
- let mut i_arg = None;
- if cur.as_bytes()[1] == b'-' {
- let tail = &cur[2..curlen];
- let tail_eq: Vec<&str> = tail.splitn(2, '=').collect();
- if tail_eq.len() <= 1 {
- names = vec![Long(tail.to_owned())];
- } else {
- names = vec![Long(tail_eq[0].to_owned())];
- i_arg = Some(tail_eq[1].to_owned());
- }
- } else {
- let mut j = 1;
- names = Vec::new();
- while j < curlen {
- let ch = cur[j..].chars().next().unwrap();
- let opt = Short(ch);
-
- // In a series of potential options (eg. -aheJ), if we
- // see one which takes an argument, we assume all
- // subsequent characters make up the argument. This
- // allows options such as -L/usr/local/lib/foo to be
- // interpreted correctly
-
- let opt_id = match find_opt(&opts, opt.clone()) {
- Some(id) => id,
- None => return Err(UnrecognizedOption(opt.to_string())),
- };
-
- names.push(opt);
-
- let arg_follows = match opts[opt_id].hasarg {
- Yes | Maybe => true,
- No => false,
- };
-
- let next = j + ch.len_utf8();
- if arg_follows && next < curlen {
- i_arg = Some((&cur[next..curlen]).to_owned());
- break;
- }
-
- j = next;
- }
- }
- let mut name_pos = 0;
- for nm in &names {
- name_pos += 1;
- let optid = match find_opt(&opts, (*nm).clone()) {
- Some(id) => id,
- None => return Err(UnrecognizedOption(nm.to_string())),
- };
- match opts[optid].hasarg {
- No => {
- if name_pos == names.len() && !i_arg.is_none() {
- return Err(UnexpectedArgument(nm.to_string()));
- }
- let v = &mut vals[optid];
- v.push(Given);
- }
- Maybe => {
- if !i_arg.is_none() {
- let v = &mut vals[optid];
- v.push(Val((i_arg.clone()).unwrap()));
- } else if name_pos < names.len() || i + 1 == l || is_arg(&args[i + 1][..]) {
- let v = &mut vals[optid];
- v.push(Given);
- } else {
- i += 1;
- let v = &mut vals[optid];
- v.push(Val(args[i].clone()));
- }
- }
- Yes => {
- if !i_arg.is_none() {
- let v = &mut vals[optid];
- v.push(Val(i_arg.clone().unwrap()));
- } else if i + 1 == l {
- return Err(ArgumentMissing(nm.to_string()));
- } else {
- i += 1;
- let v = &mut vals[optid];
- v.push(Val(args[i].clone()));
- }
- }
- }
- }
- }
- i += 1;
- }
- for i in 0..n_opts {
- let n = vals[i].len();
- let occ = opts[i].occur;
- if occ == Req && n == 0 {
- return Err(OptionMissing(opts[i].name.to_string()));
- }
- if occ != Multi && n > 1 {
- return Err(OptionDuplicated(opts[i].name.to_string()));
- }
- }
- Ok(Matches {
- opts,
- vals,
- free,
- })
-}
-
-/// Derive a usage message from a set of long options.
-pub fn usage(brief: &str, opts: &[OptGroup]) -> String {
-
- let desc_sep = format!("\n{}", repeat(" ").take(24).collect::<String>());
-
- let rows = opts.iter().map(|optref| {
- let OptGroup{short_name,
- long_name,
- hint,
- desc,
- hasarg,
- ..} = (*optref).clone();
-
- let mut row = repeat(" ").take(4).collect::<String>();
-
- // short option
- match short_name.len() {
- 0 => {}
- 1 => {
- row.push('-');
- row.push_str(&short_name[..]);
- row.push(' ');
- }
- _ => panic!("the short name should only be 1 ascii char long"),
- }
-
- // long option
- match long_name.len() {
- 0 => {}
- _ => {
- row.push_str("--");
- row.push_str(&long_name[..]);
- row.push(' ');
- }
- }
-
- // arg
- match hasarg {
- No => {}
- Yes => row.push_str(&hint[..]),
- Maybe => {
- row.push('[');
- row.push_str(&hint[..]);
- row.push(']');
- }
- }
-
- // FIXME(https://github.com/rust-lang-nursery/getopts/issues/7)
- // should be graphemes not codepoints
- //
- // here we just need to indent the start of the description
- let rowlen = row.chars().count();
- if rowlen < 24 {
- for _ in 0..24 - rowlen {
- row.push(' ');
- }
- } else {
- row.push_str(&desc_sep[..]);
- }
-
- // Normalize desc to contain words separated by one space character
- let mut desc_normalized_whitespace = String::new();
- for word in desc.split_whitespace() {
- desc_normalized_whitespace.push_str(word);
- desc_normalized_whitespace.push(' ');
- }
-
- // FIXME(https://github.com/rust-lang-nursery/getopts/issues/7)
- // should be graphemes not codepoints
- let mut desc_rows = Vec::new();
- each_split_within(&desc_normalized_whitespace[..], 54, |substr| {
- desc_rows.push(substr.to_owned());
- true
- });
-
- // FIXME(https://github.com/rust-lang-nursery/getopts/issues/7)
- // should be graphemes not codepoints
- //
- // wrapped description
- row.push_str(&desc_rows.join(&desc_sep[..]));
-
- row
- });
-
- format!("{}\n\nOptions:\n{}\n",
- brief,
- rows.collect::<Vec<String>>().join("\n"))
-}
-
-fn format_option(opt: &OptGroup) -> String {
- let mut line = String::new();
-
- if opt.occur != Req {
- line.push('[');
- }
-
- // Use short_name is possible, but fallback to long_name.
- if !opt.short_name.is_empty() {
- line.push('-');
- line.push_str(&opt.short_name[..]);
- } else {
- line.push_str("--");
- line.push_str(&opt.long_name[..]);
- }
-
- if opt.hasarg != No {
- line.push(' ');
- if opt.hasarg == Maybe {
- line.push('[');
- }
- line.push_str(&opt.hint[..]);
- if opt.hasarg == Maybe {
- line.push(']');
- }
- }
-
- if opt.occur != Req {
- line.push(']');
- }
- if opt.occur == Multi {
- line.push_str("..");
- }
-
- line
-}
-
-/// Derive a short one-line usage summary from a set of long options.
-pub fn short_usage(program_name: &str, opts: &[OptGroup]) -> String {
- let mut line = format!("Usage: {} ", program_name);
- line.push_str(&opts.iter()
- .map(format_option)
- .collect::<Vec<String>>()
- .join(" ")[..]);
- line
-}
-
-#[derive(Copy, Clone)]
-enum SplitWithinState {
- A, // leading whitespace, initial state
- B, // words
- C, // internal and trailing whitespace
-}
-#[derive(Copy, Clone)]
-enum Whitespace {
- Ws, // current char is whitespace
- Cr, // current char is not whitespace
-}
-#[derive(Copy, Clone)]
-enum LengthLimit {
- UnderLim, // current char makes current substring still fit in limit
- OverLim, // current char makes current substring no longer fit in limit
-}
-
-
-/// Splits a string into substrings with possibly internal whitespace,
-/// each of them at most `lim` bytes long. The substrings have leading and trailing
-/// whitespace removed, and are only cut at whitespace boundaries.
-///
-/// Note: Function was moved here from `std::str` because this module is the only place that
-/// uses it, and because it was too specific for a general string function.
-///
-/// # Panics
-///
-/// Panics during iteration if the string contains a non-whitespace
-/// sequence longer than the limit.
-fn each_split_within<F>(ss: &str, lim: usize, mut it: F) -> bool
- where F: FnMut(&str) -> bool
-{
- // Just for fun, let's write this as a state machine:
-
- let mut slice_start = 0;
- let mut last_start = 0;
- let mut last_end = 0;
- let mut state = A;
- let mut fake_i = ss.len();
- let mut lim = lim;
-
- let mut cont = true;
-
- // if the limit is larger than the string, lower it to save cycles
- if lim >= fake_i {
- lim = fake_i;
- }
-
- let mut machine = |cont: &mut bool, (i, c): (usize, char)| -> bool {
- let whitespace = if c.is_whitespace() {
- Ws
- } else {
- Cr
- };
- let limit = if (i - slice_start + 1) <= lim {
- UnderLim
- } else {
- OverLim
- };
-
- state = match (state, whitespace, limit) {
- (A, Ws, _) => A,
- (A, Cr, _) => {
- slice_start = i;
- last_start = i;
- B
- }
-
- (B, Cr, UnderLim) => B,
- (B, Cr, OverLim) if (i - last_start + 1) > lim => {
- panic!("word starting with {} longer than limit!",
- &ss[last_start..i + 1])
- }
- (B, Cr, OverLim) => {
- *cont = it(&ss[slice_start..last_end]);
- slice_start = last_start;
- B
- }
- (B, Ws, UnderLim) => {
- last_end = i;
- C
- }
- (B, Ws, OverLim) => {
- last_end = i;
- *cont = it(&ss[slice_start..last_end]);
- A
- }
-
- (C, Cr, UnderLim) => {
- last_start = i;
- B
- }
- (C, Cr, OverLim) => {
- *cont = it(&ss[slice_start..last_end]);
- slice_start = i;
- last_start = i;
- last_end = i;
- B
- }
- (C, Ws, OverLim) => {
- *cont = it(&ss[slice_start..last_end]);
- A
- }
- (C, Ws, UnderLim) => C,
- };
-
- *cont
- };
-
- ss.char_indices().all(|x| machine(&mut cont, x));
-
- // Let the automaton 'run out' by supplying trailing whitespace
- while cont &&
- match state {
- B | C => true,
- A => false,
- } {
- machine(&mut cont, (fake_i, ' '));
- fake_i += 1;
- }
- cont
-}
-
-#[test]
-fn test_split_within() {
- fn t(s: &str, i: usize, u: &[String]) {
- let mut v = Vec::new();
- each_split_within(s, i, |s| {
- v.push(s.to_string());
- true
- });
- assert!(v.iter().zip(u).all(|(a, b)| a == b));
- }
- t("", 0, &[]);
- t("", 15, &[]);
- t("hello", 15, &["hello".to_string()]);
- t("\nMary had a little lamb\nLittle lamb\n",
- 15,
- &["Mary had a".to_string(), "little lamb".to_string(), "Little lamb".to_string()]);
- t("\nMary had a little lamb\nLittle lamb\n",
- ::std::usize::MAX,
- &["Mary had a little lamb\nLittle lamb".to_string()]);
-}
-
-#[cfg(test)]
-mod tests {
- use super::*;
-
- use std::result::Result::{Err, Ok};
- use std::result;
-
- // Tests for reqopt
- #[test]
- fn test_reqopt() {
- let long_args = vec!["--test=20".to_string()];
- let opts = vec![reqopt("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(m.opt_present("test"));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!(m.opt_present("t"));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- }
- _ => {
- panic!("test_reqopt failed (long arg)");
- }
- }
- let short_args = vec!["-t".to_string(), "20".to_string()];
- match getopts(&short_args, &opts) {
- Ok(ref m) => {
- assert!((m.opt_present("test")));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!((m.opt_present("t")));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- }
- _ => {
- panic!("test_reqopt failed (short arg)");
- }
- }
- }
-
- #[test]
- fn test_reqopt_missing() {
- let args = vec!["blah".to_string()];
- let opts = vec![reqopt("t", "test", "testing", "TEST")];
- let rs = getopts(&args, &opts);
- match rs {
- Err(OptionMissing(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_reqopt_no_arg() {
- let long_args = vec!["--test".to_string()];
- let opts = vec![reqopt("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Err(ArgumentMissing(_)) => {}
- _ => panic!(),
- }
- let short_args = vec!["-t".to_string()];
- match getopts(&short_args, &opts) {
- Err(ArgumentMissing(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_reqopt_multi() {
- let args = vec!["--test=20".to_string(), "-t".to_string(), "30".to_string()];
- let opts = vec![reqopt("t", "test", "testing", "TEST")];
- let rs = getopts(&args, &opts);
- match rs {
- Err(OptionDuplicated(_)) => {}
- _ => panic!(),
- }
- }
-
- // Tests for optopt
- #[test]
- fn test_optopt() {
- let long_args = vec!["--test=20".to_string()];
- let opts = vec![optopt("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(m.opt_present("test"));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!((m.opt_present("t")));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- }
- _ => panic!(),
- }
- let short_args = vec!["-t".to_string(), "20".to_string()];
- match getopts(&short_args, &opts) {
- Ok(ref m) => {
- assert!((m.opt_present("test")));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!((m.opt_present("t")));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optopt_missing() {
- let args = vec!["blah".to_string()];
- let opts = vec![optopt("t", "test", "testing", "TEST")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(!m.opt_present("test"));
- assert!(!m.opt_present("t"));
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optopt_no_arg() {
- let long_args = vec!["--test".to_string()];
- let opts = vec![optopt("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Err(ArgumentMissing(_)) => {}
- _ => panic!(),
- }
- let short_args = vec!["-t".to_string()];
- match getopts(&short_args, &opts) {
- Err(ArgumentMissing(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optopt_multi() {
- let args = vec!["--test=20".to_string(), "-t".to_string(), "30".to_string()];
- let opts = vec![optopt("t", "test", "testing", "TEST")];
- let rs = getopts(&args, &opts);
- match rs {
- Err(OptionDuplicated(_)) => {}
- _ => panic!(),
- }
- }
-
- // Tests for optflag
- #[test]
- fn test_optflag() {
- let long_args = vec!["--test".to_string()];
- let opts = vec![optflag("t", "test", "testing")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(m.opt_present("test"));
- assert!(m.opt_present("t"));
- }
- _ => panic!(),
- }
- let short_args = vec!["-t".to_string()];
- match getopts(&short_args, &opts) {
- Ok(ref m) => {
- assert!(m.opt_present("test"));
- assert!(m.opt_present("t"));
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflag_missing() {
- let args = vec!["blah".to_string()];
- let opts = vec![optflag("t", "test", "testing")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(!m.opt_present("test"));
- assert!(!m.opt_present("t"));
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflag_long_arg() {
- let args = vec!["--test=20".to_string()];
- let opts = vec![optflag("t", "test", "testing")];
- let rs = getopts(&args, &opts);
- match rs {
- Err(UnexpectedArgument(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflag_multi() {
- let args = vec!["--test".to_string(), "-t".to_string()];
- let opts = vec![optflag("t", "test", "testing")];
- let rs = getopts(&args, &opts);
- match rs {
- Err(OptionDuplicated(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflag_short_arg() {
- let args = vec!["-t".to_string(), "20".to_string()];
- let opts = vec![optflag("t", "test", "testing")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- // The next variable after the flag is just a free argument
-
- assert!(m.free[0] == "20");
- }
- _ => panic!(),
- }
- }
-
- // Tests for optflagmulti
- #[test]
- fn test_optflagmulti_short1() {
- let args = vec!["-v".to_string()];
- let opts = vec![optflagmulti("v", "verbose", "verbosity")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert_eq!(m.opt_count("v"), 1);
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflagmulti_short2a() {
- let args = vec!["-v".to_string(), "-v".to_string()];
- let opts = vec![optflagmulti("v", "verbose", "verbosity")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert_eq!(m.opt_count("v"), 2);
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflagmulti_short2b() {
- let args = vec!["-vv".to_string()];
- let opts = vec![optflagmulti("v", "verbose", "verbosity")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert_eq!(m.opt_count("v"), 2);
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflagmulti_long1() {
- let args = vec!["--verbose".to_string()];
- let opts = vec![optflagmulti("v", "verbose", "verbosity")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert_eq!(m.opt_count("verbose"), 1);
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflagmulti_long2() {
- let args = vec!["--verbose".to_string(), "--verbose".to_string()];
- let opts = vec![optflagmulti("v", "verbose", "verbosity")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert_eq!(m.opt_count("verbose"), 2);
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optflagmulti_mix() {
- let args = vec!["--verbose".to_string(),
- "-v".to_string(),
- "-vv".to_string(),
- "verbose".to_string()];
- let opts = vec![optflagmulti("v", "verbose", "verbosity")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert_eq!(m.opt_count("verbose"), 4);
- assert_eq!(m.opt_count("v"), 4);
- }
- _ => panic!(),
- }
- }
-
- // Tests for optmulti
- #[test]
- fn test_optmulti() {
- let long_args = vec!["--test=20".to_string()];
- let opts = vec![optmulti("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Ok(ref m) => {
- assert!((m.opt_present("test")));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!((m.opt_present("t")));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- }
- _ => panic!(),
- }
- let short_args = vec!["-t".to_string(), "20".to_string()];
- match getopts(&short_args, &opts) {
- Ok(ref m) => {
- assert!((m.opt_present("test")));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!((m.opt_present("t")));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optmulti_missing() {
- let args = vec!["blah".to_string()];
- let opts = vec![optmulti("t", "test", "testing", "TEST")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(!m.opt_present("test"));
- assert!(!m.opt_present("t"));
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optmulti_no_arg() {
- let long_args = vec!["--test".to_string()];
- let opts = vec![optmulti("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Err(ArgumentMissing(_)) => {}
- _ => panic!(),
- }
- let short_args = vec!["-t".to_string()];
- match getopts(&short_args, &opts) {
- Err(ArgumentMissing(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_optmulti_multi() {
- let args = vec!["--test=20".to_string(), "-t".to_string(), "30".to_string()];
- let opts = vec![optmulti("t", "test", "testing", "TEST")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(m.opt_present("test"));
- assert_eq!(m.opt_str("test").unwrap(), "20");
- assert!(m.opt_present("t"));
- assert_eq!(m.opt_str("t").unwrap(), "20");
- let pair = m.opt_strs("test");
- assert!(pair[0] == "20");
- assert!(pair[1] == "30");
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_unrecognized_option() {
- let long_args = vec!["--untest".to_string()];
- let opts = vec![optmulti("t", "test", "testing", "TEST")];
- let rs = getopts(&long_args, &opts);
- match rs {
- Err(UnrecognizedOption(_)) => {}
- _ => panic!(),
- }
- let short_args = vec!["-u".to_string()];
- match getopts(&short_args, &opts) {
- Err(UnrecognizedOption(_)) => {}
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_combined() {
- let args = vec!["prog".to_string(),
- "free1".to_string(),
- "-s".to_string(),
- "20".to_string(),
- "free2".to_string(),
- "--flag".to_string(),
- "--long=30".to_string(),
- "-f".to_string(),
- "-m".to_string(),
- "40".to_string(),
- "-m".to_string(),
- "50".to_string(),
- "-n".to_string(),
- "-A B".to_string(),
- "-n".to_string(),
- "-60 70".to_string()];
- let opts = vec![optopt("s", "something", "something", "SOMETHING"),
- optflag("", "flag", "a flag"),
- reqopt("", "long", "hi", "LONG"),
- optflag("f", "", "another flag"),
- optmulti("m", "", "mmmmmm", "YUM"),
- optmulti("n", "", "nothing", "NOTHING"),
- optopt("", "notpresent", "nothing to see here", "NOPE")];
- let rs = getopts(&args, &opts);
- match rs {
- Ok(ref m) => {
- assert!(m.free[0] == "prog");
- assert!(m.free[1] == "free1");
- assert_eq!(m.opt_str("s").unwrap(), "20");
- assert!(m.free[2] == "free2");
- assert!((m.opt_present("flag")));
- assert_eq!(m.opt_str("long").unwrap(), "30");
- assert!((m.opt_present("f")));
- let pair = m.opt_strs("m");
- assert!(pair[0] == "40");
- assert!(pair[1] == "50");
- let pair = m.opt_strs("n");
- assert!(pair[0] == "-A B");
- assert!(pair[1] == "-60 70");
- assert!((!m.opt_present("notpresent")));
- }
- _ => panic!(),
- }
- }
-
- #[test]
- fn test_multi() {
- let opts = vec![optopt("e", "", "encrypt", "ENCRYPT"),
- optopt("", "encrypt", "encrypt", "ENCRYPT"),
- optopt("f", "", "flag", "FLAG")];
-
- let args_single = vec!["-e".to_string(), "foo".to_string()];
- let matches_single = &match getopts(&args_single, &opts) {
- result::Result::Ok(m) => m,
- result::Result::Err(_) => panic!(),
- };
- assert!(matches_single.opts_present(&["e".to_string()]));
- assert!(matches_single.opts_present(&["encrypt".to_string(), "e".to_string()]));
- assert!(matches_single.opts_present(&["e".to_string(), "encrypt".to_string()]));
- assert!(!matches_single.opts_present(&["encrypt".to_string()]));
- assert!(!matches_single.opts_present(&["thing".to_string()]));
- assert!(!matches_single.opts_present(&[]));
-
- assert_eq!(matches_single.opts_str(&["e".to_string()]).unwrap(), "foo");
- assert_eq!(matches_single.opts_str(&["e".to_string(), "encrypt".to_string()]).unwrap(),
- "foo");
- assert_eq!(matches_single.opts_str(&["encrypt".to_string(), "e".to_string()]).unwrap(),
- "foo");
-
- let args_both = vec!["-e".to_string(),
- "foo".to_string(),
- "--encrypt".to_string(),
- "foo".to_string()];
- let matches_both = &match getopts(&args_both, &opts) {
- result::Result::Ok(m) => m,
- result::Result::Err(_) => panic!(),
- };
- assert!(matches_both.opts_present(&["e".to_string()]));
- assert!(matches_both.opts_present(&["encrypt".to_string()]));
- assert!(matches_both.opts_present(&["encrypt".to_string(), "e".to_string()]));
- assert!(matches_both.opts_present(&["e".to_string(), "encrypt".to_string()]));
- assert!(!matches_both.opts_present(&["f".to_string()]));
- assert!(!matches_both.opts_present(&["thing".to_string()]));
- assert!(!matches_both.opts_present(&[]));
-
- assert_eq!(matches_both.opts_str(&["e".to_string()]).unwrap(), "foo");
- assert_eq!(matches_both.opts_str(&["encrypt".to_string()]).unwrap(),
- "foo");
- assert_eq!(matches_both.opts_str(&["e".to_string(), "encrypt".to_string()]).unwrap(),
- "foo");
- assert_eq!(matches_both.opts_str(&["encrypt".to_string(), "e".to_string()]).unwrap(),
- "foo");
- }
-
- #[test]
- fn test_nospace() {
- let args = vec!["-Lfoo".to_string(), "-M.".to_string()];
- let opts = vec![optmulti("L", "", "library directory", "LIB"),
- optmulti("M", "", "something", "MMMM")];
- let matches = &match getopts(&args, &opts) {
- result::Result::Ok(m) => m,
- result::Result::Err(_) => panic!(),
- };
- assert!(matches.opts_present(&["L".to_string()]));
- assert_eq!(matches.opts_str(&["L".to_string()]).unwrap(), "foo");
- assert!(matches.opts_present(&["M".to_string()]));
- assert_eq!(matches.opts_str(&["M".to_string()]).unwrap(), ".");
-
- }
-
- #[test]
- fn test_nospace_conflict() {
- let args = vec!["-vvLverbose".to_string(), "-v".to_string()];
- let opts = vec![optmulti("L", "", "library directory", "LIB"),
- optflagmulti("v", "verbose", "Verbose")];
- let matches = &match getopts(&args, &opts) {
- result::Result::Ok(m) => m,
- result::Result::Err(e) => panic!("{}", e),
- };
- assert!(matches.opts_present(&["L".to_string()]));
- assert_eq!(matches.opts_str(&["L".to_string()]).unwrap(), "verbose");
- assert!(matches.opts_present(&["v".to_string()]));
- assert_eq!(3, matches.opt_count("v"));
- }
-
- #[test]
- fn test_long_to_short() {
- let mut short = Opt {
- name: Name::Long("banana".to_string()),
- hasarg: HasArg::Yes,
- occur: Occur::Req,
- aliases: Vec::new(),
- };
- short.aliases = vec![Opt {
- name: Name::Short('b'),
- hasarg: HasArg::Yes,
- occur: Occur::Req,
- aliases: Vec::new(),
- }];
- let verbose = reqopt("b", "banana", "some bananas", "VAL");
-
- assert!(verbose.long_to_short() == short);
- }
-
- #[test]
- fn test_aliases_long_and_short() {
- let opts = vec![optflagmulti("a", "apple", "Desc")];
-
- let args = vec!["-a".to_string(), "--apple".to_string(), "-a".to_string()];
-
- let matches = getopts(&args, &opts).unwrap();
- assert_eq!(3, matches.opt_count("a"));
- assert_eq!(3, matches.opt_count("apple"));
- }
-
- #[test]
- fn test_usage() {
- let optgroups = vec![reqopt("b", "banana", "Desc", "VAL"),
- optopt("a", "012345678901234567890123456789", "Desc", "VAL"),
- optflag("k", "kiwi", "Desc"),
- optflagopt("p", "", "Desc", "VAL"),
- optmulti("l", "", "Desc", "VAL")];
-
- let expected =
-"Usage: fruits
-
-Options:
- -b --banana VAL Desc
- -a --012345678901234567890123456789 VAL
- Desc
- -k --kiwi Desc
- -p [VAL] Desc
- -l VAL Desc
-";
-
- let generated_usage = usage("Usage: fruits", &optgroups);
-
- assert_eq!(generated_usage, expected);
- }
-
- #[test]
- fn test_usage_description_wrapping() {
- // indentation should be 24 spaces
- // lines wrap after 78: or rather descriptions wrap after 54
-
- let optgroups = vec![optflag("k",
- "kiwi",
- // 54
- "This is a long description which won't be wrapped..+.."),
- optflag("a",
- "apple",
- "This is a long description which _will_ be wrapped..+..")];
-
- let expected =
-"Usage: fruits
-
-Options:
- -k --kiwi This is a long description which won't be wrapped..+..
- -a --apple This is a long description which _will_ be
- wrapped..+..
-";
-
- let usage = usage("Usage: fruits", &optgroups);
-
- assert!(usage == expected)
- }
-
- #[test]
- fn test_usage_description_multibyte_handling() {
- let optgroups = vec![optflag("k",
- "k\u{2013}w\u{2013}",
- "The word kiwi is normally spelled with two i's"),
- optflag("a",
- "apple",
- "This \u{201C}description\u{201D} has some characters that \
- could confuse the line wrapping; an apple costs 0.51€ in \
- some parts of Europe.")];
-
- let expected =
-"Usage: fruits
-
-Options:
- -k --k–w– The word kiwi is normally spelled with two i's
- -a --apple This “description” has some characters that could
- confuse the line wrapping; an apple costs 0.51€ in
- some parts of Europe.
-";
-
- let usage = usage("Usage: fruits", &optgroups);
-
- assert!(usage == expected)
- }
-
- #[test]
- fn test_short_usage() {
- let optgroups = vec![reqopt("b", "banana", "Desc", "VAL"),
- optopt("a", "012345678901234567890123456789", "Desc", "VAL"),
- optflag("k", "kiwi", "Desc"),
- optflagopt("p", "", "Desc", "VAL"),
- optmulti("l", "", "Desc", "VAL")];
-
- let expected = "Usage: fruits -b VAL [-a VAL] [-k] [-p [VAL]] [-l VAL]..".to_string();
- let generated_usage = short_usage("fruits", &optgroups);
-
- assert_eq!(generated_usage, expected);
- }
-
- #[test]
- fn test_args_with_equals() {
- let args = vec!["--one".to_string(), "A=B".to_string(),
- "--two=C=D".to_string()];
- let opts = vec![optopt("o", "one", "One", "INFO"),
- optopt("t", "two", "Two", "INFO")];
- let matches = &match getopts(&args, &opts) {
- result::Result::Ok(m) => m,
- result::Result::Err(e) => panic!("{}", e)
- };
- assert_eq!(matches.opts_str(&["o".to_string()]).unwrap(), "A=B");
- assert_eq!(matches.opts_str(&["t".to_string()]).unwrap(), "C=D");
- }
-}