pirate/src/opts.rs

77 lines
2.2 KiB
Rust
Raw Normal View History

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-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-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);
}
opts.insert(String::from("-h", false));
opts.insert(String::from("--help", false));
2015-06-08 16:26:12 -04:00
Ok(Opts {
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()
}
pub fn arg_len(&self) -> usize {
self.args.len()
}
}