92 lines
2.4 KiB
Rust
92 lines
2.4 KiB
Rust
|
use crate::err::Error;
|
||
|
|
use reqwest::blocking::Client;
|
||
|
|
use reqwest::blocking::Response;
|
||
|
|
use reqwest::header::CONTENT_TYPE;
|
||
|
|
use rustyline::error::ReadlineError;
|
||
|
|
use rustyline::Editor;
|
||
|
|
use serde_json::Value;
|
||
|
|
|
||
|
|
pub fn init(matches: &clap::ArgMatches) -> Result<(), Error> {
|
||
|
|
// Set the default logging level
|
||
|
|
crate::cli::log::init(3);
|
||
|
|
// Parse all other cli arguments
|
||
|
|
let user = matches.value_of("user").unwrap();
|
||
|
|
let pass = matches.value_of("pass").unwrap();
|
||
|
|
let conn = matches.value_of("conn").unwrap();
|
||
|
|
let ns = matches.value_of("ns").unwrap();
|
||
|
|
let db = matches.value_of("db").unwrap();
|
||
|
|
// If we should pretty-print responses
|
||
|
|
let pretty = matches.is_present("pretty");
|
||
|
|
// Set the correct import URL
|
||
|
|
let conn = format!("{}/sql", conn);
|
||
|
|
// Create a new terminal REPL
|
||
|
|
let mut rl = Editor::<()>::new().unwrap();
|
||
|
|
// Load the command-line history
|
||
|
|
let _ = rl.load_history("history.txt");
|
||
|
|
// Loop over each command-line input
|
||
|
|
loop {
|
||
|
|
// Prompt the user to input SQL
|
||
|
|
let readline = rl.readline("> ");
|
||
|
|
// Check the user input
|
||
|
|
match readline {
|
||
|
|
// The user typed a query
|
||
|
|
Ok(line) => {
|
||
|
|
// Add the entry to the history
|
||
|
|
rl.add_history_entry(line.as_str());
|
||
|
|
// Make a new remote request
|
||
|
|
let res = Client::new()
|
||
|
|
.post(&conn)
|
||
|
|
.header(CONTENT_TYPE, "application/json")
|
||
|
|
.basic_auth(user, Some(pass))
|
||
|
|
.header("NS", ns)
|
||
|
|
.header("DB", db)
|
||
|
|
.body(line)
|
||
|
|
.send();
|
||
|
|
// Get the request response
|
||
|
|
match process(pretty, res) {
|
||
|
|
Ok(v) => println!("{}", v),
|
||
|
|
Err(e) => eprintln!("{}", e),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// The user types CTRL-C
|
||
|
|
Err(ReadlineError::Interrupted) => {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
// The user typed CTRL-D
|
||
|
|
Err(ReadlineError::Eof) => {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
// There was en error
|
||
|
|
Err(err) => {
|
||
|
|
eprintln!("Error: {:?}", err);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// Save the inputs to the history
|
||
|
|
let _ = rl.save_history("history.txt");
|
||
|
|
// Everything OK
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
fn process(pretty: bool, res: reqwest::Result<Response>) -> Result<String, Error> {
|
||
|
|
// Catch any errors
|
||
|
|
let res = res?;
|
||
|
|
// Process the TEXT response
|
||
|
|
let res = res.text()?;
|
||
|
|
// Check if we should prettify
|
||
|
|
match pretty {
|
||
|
|
// Don't prettify the response
|
||
|
|
false => Ok(res),
|
||
|
|
// Yes prettify the response
|
||
|
|
true => {
|
||
|
|
// Parse the JSON response
|
||
|
|
let res: Value = serde_json::from_str(&res)?;
|
||
|
|
// Pretty the JSON response
|
||
|
|
let res = serde_json::to_string_pretty(&res)?;
|
||
|
|
// Everything processed OK
|
||
|
|
Ok(res)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|