lib.rs

// 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: #5516 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: #5516 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: #5516 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");
    }
}