NotAShelf/stash
1
0
Fork 0
mirror of https://github.com/NotAShelf/stash.git synced 2026-04-18 00:28:12 +00:00
Code Issues Projects Releases Packages Wiki Activity

stash: allow confirming destructive operations with --ask

Browse source 
Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: I6a6a69644c23734a8b088e20473d381390d532b4
This commit is contained in:
raf 2025-08-13 19:30:56 +03:00
commit 86001652cd
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
1 changed files with 111 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

160
src/main.rs
View file

@ -6,6 +6,7 @@ use std::{
}; };
use clap::{CommandFactory, Parser, Subcommand}; use clap::{CommandFactory, Parser, Subcommand};
use inquire::Confirm;
mod commands; mod commands;
mod db; mod db;
@ -39,6 +40,10 @@ struct Cli {
#[arg(long)] #[arg(long)]
db_path: Option<PathBuf>, db_path: Option<PathBuf>,
/// Ask for confirmation before destructive operations
#[arg(long)]
ask: bool,
#[command(flatten)] #[command(flatten)]
verbosity: clap_verbosity_flag::Verbosity, verbosity: clap_verbosity_flag::Verbosity,
} }
@ -67,16 +72,28 @@ enum Command {
/// Explicitly specify type: "id" or "query" /// Explicitly specify type: "id" or "query"
#[arg(long, value_parser = ["id", "query"])] #[arg(long, value_parser = ["id", "query"])]
r#type: Option<String>, r#type: Option<String>,
/// Ask for confirmation before deleting
#[arg(long)]
ask: bool,
}, },
/// Wipe all clipboard history /// Wipe all clipboard history
Wipe, Wipe {
/// Ask for confirmation before wiping
#[arg(long)]
ask: bool,
},
/// Import clipboard data from stdin (default: TSV format) /// Import clipboard data from stdin (default: TSV format)
Import { Import {
/// Explicitly specify format: "tsv" (default) /// Explicitly specify format: "tsv" (default)
#[arg(long, value_parser = ["tsv"])] #[arg(long, value_parser = ["tsv"])]
r#type: Option<String>, r#type: Option<String>,
/// Ask for confirmation before importing
#[arg(long)]
ask: bool,
}, },
/// Watch clipboard for changes and store automatically /// Watch clipboard for changes and store automatically
@ -144,17 +161,16 @@ fn main() {
"Failed to list entries", "Failed to list entries",
); );
} }
"json" => {
// Implement JSON output "json" => match db.list_json() {
match db.list_json() { Ok(json) => {
Ok(json) => { println!("{json}");
println!("{json}");
}
Err(e) => {
log::error!("Failed to list entries as JSON: {e}");
}
} }
} Err(e) => {
log::error!("Failed to list entries as JSON: {e}");
}
},
_ => { _ => {
log::error!("Unsupported format: {format}"); log::error!("Unsupported format: {format}");
} }
@ -166,52 +182,98 @@ fn main() {
"Failed to decode entry", "Failed to decode entry",
); );
} }
Some(Command::Delete { arg, r#type }) => match (arg, r#type.as_deref()) { Some(Command::Delete { arg, r#type, ask }) => {
(Some(s), Some("id")) => { let mut should_proceed = true;
if let Ok(id) = s.parse::<u64>() { if ask {
use std::io::Cursor; should_proceed =
report_error( Confirm::new("Are you sure you want to delete clipboard entries?")
db.delete(Cursor::new(format!("{id}\n"))), .with_default(false)
"Failed to delete entry by id", .prompt()
); .unwrap_or(false);
} else {
log::error!("Argument is not a valid id"); if !should_proceed {
log::info!("Aborted by user.");
} }
} }
(Some(s), Some("query")) => { if should_proceed {
report_error(db.query_delete(&s), "Failed to delete entry by query"); match (arg, r#type.as_deref()) {
} (Some(s), Some("id")) => {
(Some(s), None) => { if let Ok(id) = s.parse::<u64>() {
if let Ok(id) = s.parse::<u64>() { use std::io::Cursor;
use std::io::Cursor; report_error(
report_error( db.delete(Cursor::new(format!("{id}\n"))),
db.delete(Cursor::new(format!("{id}\n"))), "Failed to delete entry by id",
"Failed to delete entry by id", );
); } else {
} else { log::error!("Argument is not a valid id");
report_error(db.query_delete(&s), "Failed to delete entry by query"); }
}
(Some(s), Some("query")) => {
report_error(db.query_delete(&s), "Failed to delete entry by query");
}
(Some(s), None) => {
if let Ok(id) = s.parse::<u64>() {
use std::io::Cursor;
report_error(
db.delete(Cursor::new(format!("{id}\n"))),
"Failed to delete entry by id",
);
} else {
report_error(
db.query_delete(&s),
"Failed to delete entry by query",
);
}
}
(None, _) => {
report_error(
db.delete(io::stdin()),
"Failed to delete entry from stdin",
);
}
(_, Some(_)) => {
log::error!("Unknown type for --type. Use \"id\" or \"query\".");
}
} }
} }
(None, _) => { }
report_error(db.delete(io::stdin()), "Failed to delete entry from stdin"); Some(Command::Wipe { ask }) => {
let mut should_proceed = true;
if ask {
should_proceed =
Confirm::new("Are you sure you want to wipe all clipboard history?")
.with_default(false)
.prompt()
.unwrap_or(false);
if !should_proceed {
log::info!("Aborted by user.");
}
} }
(_, Some(_)) => { if should_proceed {
log::error!("Unknown type for --type. Use \"id\" or \"query\"."); report_error(db.wipe(), "Failed to wipe database");
} }
},
Some(Command::Wipe) => {
report_error(db.wipe(), "Failed to wipe database");
} }
Some(Command::Import { r#type }) => { Some(Command::Import { r#type, ask }) => {
// Default format is TSV (Cliphist compatible) let mut should_proceed = true;
let format = r#type.as_deref().unwrap_or("tsv"); if ask {
match format { should_proceed = Confirm::new("Are you sure you want to import clipboard data? This may overwrite existing entries.")
"tsv" => { .with_default(false)
db.import_tsv(io::stdin()); .prompt()
.unwrap_or(false);
if !should_proceed {
log::info!("Aborted by user.");
} }
_ => { }
log::error!("Unsupported import format: {format}"); if should_proceed {
let format = r#type.as_deref().unwrap_or("tsv");
match format {
"tsv" => {
db.import_tsv(io::stdin());
}
_ => {
log::error!("Unsupported import format: {format}");
}
} }
} }
} }
@ -220,7 +282,7 @@ fn main() {
} }
None => { None => {
if let Err(e) = Cli::command().print_help() { if let Err(e) = Cli::command().print_help() {
eprintln!("Failed to print help: {e}"); log::error!("Failed to print help: {e}");
} }
println!(); println!();
} }