zoxide/src/subcommand/query.rs

use crate::env::Env;
use crate::util;
use anyhow::{bail, Result};
use std::path::Path;
use structopt::StructOpt;

#[derive(Debug, StructOpt)]
#[structopt(about = "Search for a directory")]
pub struct Query {
    keywords: Vec<String>,
    #[structopt(short, long, help = "Opens an interactive selection menu using fzf")]
    interactive: bool,
}

impl Query {
    pub fn run(mut self, env: &Env) -> Result<()> {
        let path_opt = if self.interactive {
            self.query_interactive(env)?
        } else {
            self.query(env)?
        };

        match path_opt {
            Some(path) => println!("query: {}", path.trim()),
            None => bail!("no match found"),
        };

        Ok(())
    }

    fn query(&mut self, env: &Env) -> Result<Option<String>> {
        if let [path] = self.keywords.as_slice() {
            if Path::new(path).is_dir() {
                return Ok(Some(path.to_string()));
            }
        }

        for keyword in &mut self.keywords {
            keyword.make_ascii_lowercase();
        }

        let now = util::get_current_time()?;

        if let Some(dir) = util::get_db(env)?.query(&self.keywords, now) {
            Ok(Some(dir.path))
        } else {
            Ok(None)
        }
    }

    fn query_interactive(&mut self, env: &Env) -> Result<Option<String>> {
        let now = util::get_current_time()?;

        for keyword in &mut self.keywords {
            keyword.make_ascii_lowercase();
        }

        let dirs = util::get_db(env)?.query_all(&self.keywords);
        util::fzf_helper(now, dirs)
    }
}