generated from PlexSheep/baserepo
134 lines
3.5 KiB
Rust
134 lines
3.5 KiB
Rust
use libpt_cli::repl::REPL_HELP_TEMPLATE;
|
|
use libpt_cli::{clap, dialoguer, printing};
|
|
use libpt_log::{debug, trace, Level, Logger};
|
|
|
|
use clap::{Parser, Subcommand};
|
|
|
|
/// This is the help menu of the repl
|
|
///
|
|
/// More text here
|
|
#[derive(Parser, Debug)]
|
|
#[command(multicall = true)]
|
|
pub struct Repl {
|
|
/// the command you want to execute, along with its args
|
|
#[command(subcommand)]
|
|
command: ReplCommand,
|
|
}
|
|
|
|
#[derive(Subcommand, Debug)]
|
|
enum ReplCommand {
|
|
/// wait for LEN seconds
|
|
Wait {
|
|
/// wait so long
|
|
len: u64,
|
|
},
|
|
/// echo the given texts
|
|
Echo {
|
|
/// the text you want to print
|
|
text: Vec<String>,
|
|
/// print with a fancy border and colors
|
|
#[arg(short, long)]
|
|
fancy: bool,
|
|
},
|
|
/// hello world
|
|
Hello,
|
|
/// leave the repl
|
|
Exit,
|
|
}
|
|
|
|
// TODO: somehow autogenerate this!!!
|
|
pub struct MyCompletion {
|
|
options: Vec<String>,
|
|
}
|
|
impl Default for MyCompletion {
|
|
fn default() -> Self {
|
|
MyCompletion {
|
|
options: vec![
|
|
"help".to_string(),
|
|
"?".to_string(),
|
|
"list".to_string(),
|
|
"publish".to_string(),
|
|
"unpublish".to_string(),
|
|
"delete".to_string(),
|
|
"read".to_string(),
|
|
"show".to_string(),
|
|
"new".to_string(),
|
|
"ls".to_string(),
|
|
],
|
|
}
|
|
}
|
|
}
|
|
|
|
impl dialoguer::Completion for MyCompletion {
|
|
/// Simple completion implementation based on substring
|
|
fn get(&self, input: &str) -> Option<String> {
|
|
let matches = self
|
|
.options
|
|
.iter()
|
|
.filter(|option| option.starts_with(input))
|
|
.collect::<Vec<_>>();
|
|
|
|
if matches.len() == 1 {
|
|
Some(matches[0].to_string())
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() -> anyhow::Result<()> {
|
|
let _logger = Logger::builder()
|
|
.show_time(false)
|
|
.max_level(Level::DEBUG)
|
|
.build();
|
|
|
|
let mut buf: String = String::new();
|
|
let mut buf_preparsed: Vec<String>;
|
|
let completion = MyCompletion::default();
|
|
let mut history = dialoguer::BasicHistory::new();
|
|
|
|
debug!("entering the repl");
|
|
loop {
|
|
buf.clear();
|
|
|
|
buf = dialoguer::Input::with_theme(&dialoguer::theme::ColorfulTheme::default())
|
|
.completion_with(&completion)
|
|
.history_with(&mut history)
|
|
.interact_text()?;
|
|
|
|
buf_preparsed = Vec::new();
|
|
buf_preparsed.extend(shlex::split(&buf).unwrap_or_default());
|
|
|
|
trace!("read input: {buf_preparsed:?}");
|
|
|
|
let options = match Repl::try_parse_from(buf_preparsed) {
|
|
Ok(c) => c,
|
|
Err(e) => {
|
|
println!("{e}");
|
|
continue;
|
|
}
|
|
};
|
|
|
|
match options.command {
|
|
ReplCommand::Exit => break,
|
|
ReplCommand::Wait { len } => {
|
|
debug!("len: {len}");
|
|
let spinner = indicatif::ProgressBar::new_spinner();
|
|
spinner.enable_steady_tick(std::time::Duration::from_millis(100));
|
|
std::thread::sleep(std::time::Duration::from_secs(len));
|
|
spinner.finish();
|
|
}
|
|
ReplCommand::Hello => println!("Hello!"),
|
|
ReplCommand::Echo { text, fancy } => {
|
|
if !fancy {
|
|
println!("{}", text.concat())
|
|
}
|
|
else {
|
|
printing::blockprint(text.concat(), console::Color::Cyan)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|