extern crate shrust;
use shrust::{Shell, ShellIO};
use std::io::prelude::*;
use std::collections::HashMap;
use std::str::FromStr;
fn main() {
let map = HashMap::new();
let mut shell = Shell::new(map);
shell.new_command("put", "Insert a value", 2, |_, map, args| {
map.insert(usize::from_str(args[0])?, args[1].to_string());
Ok(())
});
shell.new_command("get", "Get a value", 1, |io, map, args| {
match map.get(&usize::from_str(args[0])?) {
Some(val) => writeln!(io, "{}", val).unwrap(),
None => writeln!(io, "Not found").unwrap()
};
Ok(())
});
shell.new_command("remove", "Remove a value", 1, |_, map, args| {
map.remove(&usize::from_str(args[0])?);
Ok(())
});
shell.new_command("list", "List all values", 0, |io, map, _| {
for (k, v) in map {
writeln!(io, "{} = {}", k, v).unwrap();
}
Ok(())
});
shell.new_command_noargs("clear", "Clear all values", |_, map| {
map.clear();
Ok(())
});
shell.run_loop(&mut ShellIO::default());
}