generated from PlexSheep/baserepo
idk why this isn't in master yet #94
|
@ -0,0 +1,133 @@
|
|||
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(())
|
||||
}
|
Reference in New Issue