parent
192a7a6ed6
commit
dca0c79c4a
GPG Key ID:
284488987C31F630
@ -0,0 +1,96 @@
|
|||||||
|
//! Parsing args
|
||||||
|
use super::*;
|
||||||
|
use ext::*;
|
||||||
|
use std::iter;
|
||||||
|
|
||||||
|
/// Arg state
|
||||||
|
#[derive(Debug, Default)]
|
||||||
|
struct State
|
||||||
|
{
|
||||||
|
is_server: bool,
|
||||||
|
is_sending: bool,
|
||||||
|
|
||||||
|
enc: Option<bool>,
|
||||||
|
comp: Option<bool>,
|
||||||
|
bufsz: Option<usize>,
|
||||||
|
arc: Option<bool>,
|
||||||
|
contin: Option<bool>,
|
||||||
|
oneshot: Option<bool>,
|
||||||
|
|
||||||
|
inter: Option<bool>,
|
||||||
|
|
||||||
|
files: Vec<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl State
|
||||||
|
{
|
||||||
|
fn mode(&self) -> impl fmt::Display + 'static
|
||||||
|
{
|
||||||
|
let send = r#if!(self.is_sending, "send", "recv");
|
||||||
|
let serve = r#if!(self.is_server, "server", "client");
|
||||||
|
lazy_format!("{} ({})", send, serve)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_schain<I>(state: &mut State, single: I) -> eyre::Result<()>
|
||||||
|
where I: IntoIterator<Item=char>
|
||||||
|
{
|
||||||
|
for ch in single.into_iter().map(char::to_lowercase).flatten()
|
||||||
|
{
|
||||||
|
match ch {
|
||||||
|
'e' => state.enc = Some(true),
|
||||||
|
'c' => state.comp = Some(true),
|
||||||
|
'a' => state.arc = Some(true),
|
||||||
|
'k' => state.contin = Some(true),
|
||||||
|
'1' if state.is_server => state.oneshot = Some(true),
|
||||||
|
'i' if !state.is_sending => state.inter = Some(true),
|
||||||
|
x => return Err(eyre!("Unknown option for mode {}", state.mode()))
|
||||||
|
.with_section(move || x.header("Option was"))
|
||||||
|
.with_note(move || "Some options are only valid for certain modes"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Try to parse an iterator of strings (usually the command-line arguments) into an `Op`.
|
||||||
|
pub fn parse_iter<I>(args: &mut I) -> eyre::Result<Op>
|
||||||
|
where I: Iterator<Item= String> + ?Sized
|
||||||
|
{
|
||||||
|
let state = parse_iter_raw(args)
|
||||||
|
.wrap_err(eyre!("Invalid arguments"))
|
||||||
|
.with_suggestion(|| "Try passing `--help`")?; // Send help message here, since it's unlikely to be helpful when returning from state's validation compared to here.
|
||||||
|
|
||||||
|
todo!("TODO: `impl TryFrom<State> for Op`, etc")
|
||||||
|
}
|
||||||
|
fn parse_iter_raw<I>(args: &mut I) -> eyre::Result<State>
|
||||||
|
where I: Iterator<Item= String> + ?Sized
|
||||||
|
{
|
||||||
|
let mut state = State::default();
|
||||||
|
//TODO: Parse modes before this.
|
||||||
|
while let Some(arg) = args.next()
|
||||||
|
{
|
||||||
|
let mut chars = arg.chars();
|
||||||
|
match (&mut chars).take(2).collect_array::<2>() {
|
||||||
|
['-', '-'] => {
|
||||||
|
// Long option
|
||||||
|
let opt = &arg[2..];
|
||||||
|
match opt {
|
||||||
|
"--" => break,
|
||||||
|
//TODO: Long options, pulling option param from `args` if needed, etc.
|
||||||
|
unknown => return Err(eyre!("Unknown option for mode {}", state.mode()))
|
||||||
|
.with_section(|| format!("--{}", unknown).header("Option was"))
|
||||||
|
.with_note(|| "Some options are only valid for certain modes"),
|
||||||
|
}
|
||||||
|
},
|
||||||
|
['-', n] => {
|
||||||
|
// Small option
|
||||||
|
parse_schain(&mut state, iter::once(n).chain(chars))?;
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
// Not an option
|
||||||
|
state.files.push(arg);
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(state)
|
||||||
|
}
|
||||||
Loading…
Reference in new issue