src/cli/command.rs: move code into separate file
This commit is contained in:
parent
211fabd795
commit
b7329c8a1a
@ -1,95 +1,9 @@
|
||||
extern crate apitest;
|
||||
|
||||
use failure::*;
|
||||
use std::collections::HashMap;
|
||||
|
||||
//use std::sync::Arc;
|
||||
|
||||
use apitest::api::schema::*;
|
||||
use apitest::api::router::*;
|
||||
use apitest::api::config::*;
|
||||
use apitest::getopts;
|
||||
|
||||
use apitest::api3;
|
||||
|
||||
fn print_cli_usage() {
|
||||
|
||||
eprintln!("Usage: TODO");
|
||||
}
|
||||
|
||||
|
||||
fn handle_simple_command(cli_cmd: &CliCommand, args: Vec<String>) -> Result<(), Error> {
|
||||
|
||||
let (params, rest) = getopts::parse_arguments(
|
||||
&args, &cli_cmd.arg_param, &cli_cmd.info.parameters)?;
|
||||
|
||||
if !rest.is_empty() {
|
||||
bail!("got additional arguments: {:?}", rest);
|
||||
}
|
||||
|
||||
let res = (cli_cmd.info.handler)(params, &cli_cmd.info)?;
|
||||
|
||||
println!("Result: {}", serde_json::to_string_pretty(&res).unwrap());
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_nested_command(def: &HashMap<String, CmdDef>, mut args: Vec<String>) -> Result<(), Error> {
|
||||
|
||||
if args.len() < 1 {
|
||||
let mut cmds: Vec<&String> = def.keys().collect();
|
||||
cmds.sort();
|
||||
|
||||
let list = cmds.iter().fold(String::new(),|mut s,item| {
|
||||
if !s.is_empty() { s+= ", "; }
|
||||
s += item;
|
||||
s
|
||||
});
|
||||
|
||||
bail!("expected command argument, but no command specified.\nPossible commands: {}", list);
|
||||
}
|
||||
|
||||
let command = args.remove(0);
|
||||
|
||||
let sub_cmd = match def.get(&command) {
|
||||
Some(cmd) => cmd,
|
||||
None => {
|
||||
bail!("no such command '{}'", command);
|
||||
}
|
||||
};
|
||||
|
||||
match sub_cmd {
|
||||
CmdDef::Simple(cli_cmd) => {
|
||||
handle_simple_command(cli_cmd, args)?;
|
||||
}
|
||||
CmdDef::Nested(map) => {
|
||||
handle_nested_command(map, args)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn run_cli_command(def: &CmdDef) -> Result<(), Error> {
|
||||
|
||||
let args: Vec<String> = std::env::args().skip(1).collect();
|
||||
|
||||
match def {
|
||||
CmdDef::Simple(cli_cmd) => handle_simple_command(cli_cmd, args),
|
||||
CmdDef::Nested(map) => handle_nested_command(map, args),
|
||||
}
|
||||
}
|
||||
|
||||
struct CliCommand {
|
||||
info: ApiMethod,
|
||||
arg_param: Vec<&'static str>,
|
||||
fixed_param: Vec<&'static str>,
|
||||
}
|
||||
|
||||
enum CmdDef {
|
||||
Simple(CliCommand),
|
||||
Nested(HashMap<String, CmdDef>),
|
||||
}
|
||||
use apitest::cli::command::*;
|
||||
|
||||
fn datastore_commands() -> CmdDef {
|
||||
|
||||
|
85
src/cli/command.rs
Normal file
85
src/cli/command.rs
Normal file
@ -0,0 +1,85 @@
|
||||
use failure::*;
|
||||
use std::collections::HashMap;
|
||||
|
||||
use crate::api::schema::*;
|
||||
use crate::api::router::*;
|
||||
use crate::api::config::*;
|
||||
use crate::getopts;
|
||||
|
||||
pub fn print_cli_usage() {
|
||||
|
||||
eprintln!("Usage: TODO");
|
||||
}
|
||||
|
||||
fn handle_simple_command(cli_cmd: &CliCommand, args: Vec<String>) -> Result<(), Error> {
|
||||
|
||||
let (params, rest) = getopts::parse_arguments(
|
||||
&args, &cli_cmd.arg_param, &cli_cmd.info.parameters)?;
|
||||
|
||||
if !rest.is_empty() {
|
||||
bail!("got additional arguments: {:?}", rest);
|
||||
}
|
||||
|
||||
let res = (cli_cmd.info.handler)(params, &cli_cmd.info)?;
|
||||
|
||||
println!("Result: {}", serde_json::to_string_pretty(&res).unwrap());
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_nested_command(def: &HashMap<String, CmdDef>, mut args: Vec<String>) -> Result<(), Error> {
|
||||
|
||||
if args.len() < 1 {
|
||||
let mut cmds: Vec<&String> = def.keys().collect();
|
||||
cmds.sort();
|
||||
|
||||
let list = cmds.iter().fold(String::new(),|mut s,item| {
|
||||
if !s.is_empty() { s+= ", "; }
|
||||
s += item;
|
||||
s
|
||||
});
|
||||
|
||||
bail!("expected command argument, but no command specified.\nPossible commands: {}", list);
|
||||
}
|
||||
|
||||
let command = args.remove(0);
|
||||
|
||||
let sub_cmd = match def.get(&command) {
|
||||
Some(cmd) => cmd,
|
||||
None => {
|
||||
bail!("no such command '{}'", command);
|
||||
}
|
||||
};
|
||||
|
||||
match sub_cmd {
|
||||
CmdDef::Simple(cli_cmd) => {
|
||||
handle_simple_command(cli_cmd, args)?;
|
||||
}
|
||||
CmdDef::Nested(map) => {
|
||||
handle_nested_command(map, args)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn run_cli_command(def: &CmdDef) -> Result<(), Error> {
|
||||
|
||||
let args: Vec<String> = std::env::args().skip(1).collect();
|
||||
|
||||
match def {
|
||||
CmdDef::Simple(cli_cmd) => handle_simple_command(cli_cmd, args),
|
||||
CmdDef::Nested(map) => handle_nested_command(map, args),
|
||||
}
|
||||
}
|
||||
|
||||
pub struct CliCommand {
|
||||
pub info: ApiMethod,
|
||||
pub arg_param: Vec<&'static str>,
|
||||
pub fixed_param: Vec<&'static str>,
|
||||
}
|
||||
|
||||
pub enum CmdDef {
|
||||
Simple(CliCommand),
|
||||
Nested(HashMap<String, CmdDef>),
|
||||
}
|
@ -50,4 +50,10 @@ pub mod storage {
|
||||
|
||||
pub mod getopts;
|
||||
|
||||
pub mod cli {
|
||||
|
||||
pub mod command;
|
||||
}
|
||||
|
||||
|
||||
pub mod api3;
|
||||
|
Loading…
Reference in New Issue
Block a user