63 lines
1.8 KiB
Rust
63 lines
1.8 KiB
Rust
use super::Cmd;
|
|
use crate::config;
|
|
use crate::db::{DatabaseFile, Query};
|
|
use crate::error::WriteErrorHandler;
|
|
use crate::fzf::Fzf;
|
|
use crate::util;
|
|
|
|
use anyhow::{bail, Context, Result};
|
|
use clap::Clap;
|
|
|
|
use std::io::Write;
|
|
|
|
/// Remove a directory from the database
|
|
#[derive(Clap, Debug)]
|
|
pub struct Remove {
|
|
#[clap(conflicts_with = "path", long, short, value_name = "keywords")]
|
|
interactive: Option<Vec<String>>,
|
|
#[clap(
|
|
conflicts_with = "interactive",
|
|
required_unless_present = "interactive"
|
|
)]
|
|
path: Option<String>,
|
|
}
|
|
|
|
impl Cmd for Remove {
|
|
fn run(&self) -> Result<()> {
|
|
let data_dir = config::zo_data_dir()?;
|
|
let mut db = DatabaseFile::new(data_dir);
|
|
let mut db = db.open()?;
|
|
|
|
let selection;
|
|
let path = match &self.interactive {
|
|
Some(keywords) => {
|
|
let query = Query::new(keywords);
|
|
let now = util::current_time()?;
|
|
|
|
let mut fzf = Fzf::new()?;
|
|
let handle = fzf.stdin();
|
|
let resolve_symlinks = config::zo_resolve_symlinks();
|
|
for dir in db.iter_matches(&query, now, resolve_symlinks) {
|
|
writeln!(handle, "{}", dir.display_score(now)).handle_err("fzf")?;
|
|
}
|
|
|
|
selection = fzf.wait_select()?;
|
|
selection
|
|
.get(5..selection.len().saturating_sub(1))
|
|
.context("fzf returned invalid output")?
|
|
}
|
|
None => self.path.as_ref().unwrap(),
|
|
};
|
|
|
|
if !db.remove(path) {
|
|
let path = util::resolve_path(&path)?;
|
|
let path = util::path_to_str(&path)?;
|
|
if !db.remove(path) {
|
|
bail!("path not found in database: {}", &path)
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|