2015-04-28 01:10:37 -04:00
|
|
|
/* Pirate - A command-line arrrrguments parser, written in Rust.
|
|
|
|
* Copyright (C) 2015 Zachary Dziura
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
use std::collections::HashMap;
|
|
|
|
use std::collections::VecDeque;
|
|
|
|
|
2015-06-08 16:26:12 -04:00
|
|
|
use errors::Error;
|
|
|
|
use token::Token;
|
2015-06-04 15:37:59 -04:00
|
|
|
|
2015-04-28 01:10:37 -04:00
|
|
|
pub struct Opts {
|
|
|
|
pub opts: HashMap<String, bool>,
|
|
|
|
pub args: VecDeque<String>,
|
2015-06-08 16:26:12 -04:00
|
|
|
tokens: Vec<Token>
|
2015-04-28 01:10:37 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Opts {
|
2015-06-08 16:26:12 -04:00
|
|
|
pub fn new(options: &[&str]) -> Result<Opts, Error> {
|
|
|
|
let mut opts: Hashmap<String, bool> = HashMap::new();
|
|
|
|
let mut args: VecDeque<String> = VecDeque::new();
|
|
|
|
let mut tokens: Vec<Token> = Vec::new();
|
2015-06-04 15:37:59 -04:00
|
|
|
|
2015-06-08 16:26:12 -04:00
|
|
|
for opt in options.iter() {
|
|
|
|
let token = match Token::new(opt) {
|
|
|
|
Ok(t) => t,
|
|
|
|
Err(why) => return Err(why)
|
|
|
|
};
|
|
|
|
|
|
|
|
if !token.is_group {
|
|
|
|
if token.is_arg {
|
|
|
|
args.push_back(String::from(token.name()));
|
|
|
|
} else {
|
|
|
|
opts.insert(String::from(token.name()), token.has_arg);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
tokens.push(token);
|
2015-06-04 15:37:59 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
opts.insert(String::from("-h", false));
|
|
|
|
opts.insert(String::from("--help", false));
|
|
|
|
|
2015-06-08 16:26:12 -04:00
|
|
|
Ok(Opts {
|
2015-06-04 15:37:59 -04:00
|
|
|
opts: opts,
|
2015-06-08 16:26:12 -04:00
|
|
|
args: args,
|
|
|
|
tokens: tokens
|
|
|
|
})
|
2015-04-28 01:10:37 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn get_opt(&self, opt_name: &String) -> Option<&bool> {
|
|
|
|
self.opts.get(opt_name)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn contains_opt(&self, opt: &String) -> bool {
|
|
|
|
self.opts.contains_key(opt)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn get_arg(&mut self) -> Option<String> {
|
|
|
|
self.args.pop_front()
|
|
|
|
}
|
2015-04-30 15:14:34 +00:00
|
|
|
|
|
|
|
pub fn arg_len(&self) -> usize {
|
|
|
|
self.args.len()
|
|
|
|
}
|
2015-06-04 15:37:59 -04:00
|
|
|
}
|