src/cli/command.rs: simplify cli handler
This commit is contained in:
parent
61834c1cbf
commit
33256db6dd
@ -185,10 +185,5 @@ fn main() {
|
|||||||
.into()
|
.into()
|
||||||
);
|
);
|
||||||
|
|
||||||
if let Err(err) = run_cli_command(&cmd_def.into()) {
|
run_cli_command(&cmd_def.into());
|
||||||
eprintln!("Error: {}", err);
|
|
||||||
print_cli_usage();
|
|
||||||
std::process::exit(-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -236,12 +236,5 @@ fn main() {
|
|||||||
.insert("garbage-collect".to_owned(), garbage_collect_cmd_def.into())
|
.insert("garbage-collect".to_owned(), garbage_collect_cmd_def.into())
|
||||||
.insert("list".to_owned(), list_cmd_def.into());
|
.insert("list".to_owned(), list_cmd_def.into());
|
||||||
|
|
||||||
if let Err(err) = run_cli_command(&cmd_def.into()) {
|
run_cli_command(&cmd_def.into());
|
||||||
eprintln!("Error: {}", err);
|
|
||||||
if err.downcast::<UsageError>().is_ok() {
|
|
||||||
print_cli_usage();
|
|
||||||
}
|
|
||||||
std::process::exit(-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -51,10 +51,5 @@ fn main() {
|
|||||||
.insert("datastore".to_owned(), datastore_commands())
|
.insert("datastore".to_owned(), datastore_commands())
|
||||||
.insert("garbage-collection".to_owned(), garbage_collection_commands());
|
.insert("garbage-collection".to_owned(), garbage_collection_commands());
|
||||||
|
|
||||||
if let Err(err) = run_cli_command(&cmd_def.into()) {
|
run_cli_command(&cmd_def.into());
|
||||||
eprintln!("Error: {}", err);
|
|
||||||
print_cli_usage();
|
|
||||||
std::process::exit(-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -2,7 +2,7 @@ use failure::*;
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
|
|
||||||
use serde_json::Value;
|
//use serde_json::Value;
|
||||||
|
|
||||||
use crate::api_schema::*;
|
use crate::api_schema::*;
|
||||||
use crate::api_schema::router::*;
|
use crate::api_schema::router::*;
|
||||||
@ -11,27 +11,39 @@ use super::environment::CliEnvironment;
|
|||||||
|
|
||||||
use super::getopts;
|
use super::getopts;
|
||||||
|
|
||||||
pub fn print_cli_usage() {
|
fn print_simple_usage(prefix: &str, _cli_cmd: &CliCommand, err: Error) {
|
||||||
|
|
||||||
eprintln!("Usage: TODO");
|
eprintln!("Error: {}", err);
|
||||||
|
eprintln!("\nUsage: {} <TODO>", prefix);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Fail)]
|
fn handle_simple_command(prefix: &str, cli_cmd: &CliCommand, args: Vec<String>) {
|
||||||
#[fail(display = "Usage error: {}", _0)]
|
|
||||||
pub struct UsageError(Error);
|
|
||||||
|
|
||||||
pub struct Invocation<'a>(&'a CliCommand, Value);
|
let (params, rest) = match getopts::parse_arguments(
|
||||||
|
&args, &cli_cmd.arg_param, &cli_cmd.info.parameters) {
|
||||||
fn handle_simple_command(cli_cmd: &CliCommand, args: Vec<String>) -> Result<Invocation, Error> {
|
Ok((p, r)) => (p, r),
|
||||||
|
Err(err) => {
|
||||||
let (params, rest) = getopts::parse_arguments(
|
print_simple_usage(prefix, cli_cmd, err.into());
|
||||||
&args, &cli_cmd.arg_param, &cli_cmd.info.parameters)?;
|
std::process::exit(-1);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
if !rest.is_empty() {
|
if !rest.is_empty() {
|
||||||
bail!("got additional arguments: {:?}", rest);
|
let err = format_err!("got additional arguments: {:?}", rest);
|
||||||
|
print_simple_usage(prefix, cli_cmd, err);
|
||||||
|
std::process::exit(-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Invocation(cli_cmd, params))
|
let mut rpcenv = CliEnvironment::new();
|
||||||
|
|
||||||
|
match (cli_cmd.info.handler)(params, &cli_cmd.info, &mut rpcenv) {
|
||||||
|
Ok(value) => {
|
||||||
|
println!("Result: {}", serde_json::to_string_pretty(&value).unwrap());
|
||||||
|
}
|
||||||
|
Err(err) => {
|
||||||
|
eprintln!("Error: {}", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn find_command<'a>(def: &'a CliCommandMap, name: &str) -> Option<&'a CommandLineInterface> {
|
fn find_command<'a>(def: &'a CliCommandMap, name: &str) -> Option<&'a CommandLineInterface> {
|
||||||
@ -56,7 +68,14 @@ fn find_command<'a>(def: &'a CliCommandMap, name: &str) -> Option<&'a CommandLin
|
|||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handle_nested_command(def: &CliCommandMap, mut args: Vec<String>) -> Result<Invocation, Error> {
|
fn print_nested_usage(prefix: &str, _def: &CliCommandMap, err: Error) {
|
||||||
|
|
||||||
|
eprintln!("Error: {}", err);
|
||||||
|
eprintln!("\nUsage: {} <TODO>", prefix);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_nested_command(prefix: &str, def: &CliCommandMap, mut args: Vec<String>) {
|
||||||
|
|
||||||
if args.len() < 1 {
|
if args.len() < 1 {
|
||||||
let mut cmds: Vec<&String> = def.commands.keys().collect();
|
let mut cmds: Vec<&String> = def.commands.keys().collect();
|
||||||
@ -68,22 +87,30 @@ fn handle_nested_command(def: &CliCommandMap, mut args: Vec<String>) -> Result<I
|
|||||||
s
|
s
|
||||||
});
|
});
|
||||||
|
|
||||||
bail!("expected command argument, but no command specified.\nPossible commands: {}", list);
|
let err = format_err!("no command specified.\nPossible commands: {}", list);
|
||||||
|
print_nested_usage(prefix, def, err);
|
||||||
|
std::process::exit(-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
let command = args.remove(0);
|
let command = args.remove(0);
|
||||||
|
|
||||||
let sub_cmd = match find_command(def, &command) {
|
let sub_cmd = match find_command(def, &command) {
|
||||||
Some(cmd) => cmd,
|
Some(cmd) => cmd,
|
||||||
None => bail!("no such command '{}'", command),
|
None => {
|
||||||
|
let err = format_err!("no such command '{}'", command);
|
||||||
|
print_nested_usage(prefix, def, err);
|
||||||
|
std::process::exit(-1);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let new_prefix = format!("{} {}", prefix, command);
|
||||||
|
|
||||||
match sub_cmd {
|
match sub_cmd {
|
||||||
CommandLineInterface::Simple(cli_cmd) => {
|
CommandLineInterface::Simple(cli_cmd) => {
|
||||||
handle_simple_command(cli_cmd, args)
|
handle_simple_command(&new_prefix, cli_cmd, args);
|
||||||
}
|
}
|
||||||
CommandLineInterface::Nested(map) => {
|
CommandLineInterface::Nested(map) => {
|
||||||
handle_nested_command(map, args)
|
handle_nested_command(&new_prefix, map, args);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -248,30 +275,24 @@ pub fn print_bash_completion(def: &CommandLineInterface) {
|
|||||||
print_nested_completion(def, args);
|
print_nested_completion(def, args);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn run_cli_command(def: &CommandLineInterface) -> Result<(), Error> {
|
pub fn run_cli_command(def: &CommandLineInterface) {
|
||||||
|
|
||||||
let args: Vec<String> = std::env::args().skip(1).collect();
|
let mut args = std::env::args();
|
||||||
|
|
||||||
|
let prefix = args.next().unwrap();
|
||||||
|
let prefix = prefix.rsplit('/').next().unwrap(); // without path
|
||||||
|
|
||||||
|
let args: Vec<String> = args.collect();
|
||||||
|
|
||||||
if !args.is_empty() && args[0] == "bashcomplete" {
|
if !args.is_empty() && args[0] == "bashcomplete" {
|
||||||
print_bash_completion(def);
|
print_bash_completion(def);
|
||||||
return Ok(());
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
let invocation = match def {
|
match def {
|
||||||
CommandLineInterface::Simple(cli_cmd) => handle_simple_command(cli_cmd, args),
|
CommandLineInterface::Simple(cli_cmd) => handle_simple_command(&prefix, cli_cmd, args),
|
||||||
CommandLineInterface::Nested(map) => handle_nested_command(map, args),
|
CommandLineInterface::Nested(map) => handle_nested_command(&prefix, map, args),
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut rpcenv = CliEnvironment::new();
|
|
||||||
|
|
||||||
let res = match invocation {
|
|
||||||
Err(e) => return Err(UsageError(e).into()),
|
|
||||||
Ok(invocation) => (invocation.0.info.handler)(invocation.1, &invocation.0.info, &mut rpcenv)?,
|
|
||||||
};
|
|
||||||
|
|
||||||
println!("Result: {}", serde_json::to_string_pretty(&res).unwrap());
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub type CompletionFunction = fn(&str) -> Vec<String>;
|
pub type CompletionFunction = fn(&str) -> Vec<String>;
|
||||||
|
Loading…
Reference in New Issue
Block a user