Option to query only into sub directories in zoxide query

src/cmd/cmd.rs

@@ -117,6 +117,14 @@ pub struct Query {
     #[clap(long, short)]
     pub score: bool,
 
+    /// Search only through current working directory
+    #[clap(long, short, conflicts_with = "homedir")]
+    pub currentdir: bool,
+
+    /// Search only through home directory
+    #[clap(long, short, conflicts_with = "currentdir", short = 'm')]
+    pub homedir: bool,
+
     /// Exclude a path from results
     #[clap(long, value_hint = ValueHint::DirPath, value_name = "path")]
     pub exclude: Option<String>,

src/cmd/query.rs

@@ -4,7 +4,7 @@ use anyhow::{Context, Result};
 
 use crate::cmd::{Query, Run};
 use crate::config;
-use crate::db::{Database, DatabaseFile};
+use crate::db::{Database, DatabaseFile, WorkingMode};
 use crate::error::BrokenPipeHandler;
 use crate::util::{self, Fzf};
 
@@ -30,6 +30,19 @@ impl Query {
             stream = stream.with_exclude(path);
         }
 
+        match stream.working_mode {
+            WorkingMode::NoKeywordPass => {
+                if self.currentdir {
+                    stream = stream.with_currentdir();
+                } else if self.homedir {
+                    stream = stream.with_homedir();
+                } else {
+                    stream = stream.with_globaldir();
+                }
+            }
+            WorkingMode::Current | WorkingMode::Home | WorkingMode::Global => {}
+        }
+
         if self.interactive {
             let mut fzf = Fzf::new(false)?;
             let stdin = fzf.stdin();

src/db/mod.rs

@@ -17,6 +17,14 @@ pub struct Database<'file> {
     pub data_dir: &'file Path,
 }
 
+#[derive(Debug)]
+pub enum WorkingMode {
+    Current,
+    Home,
+    Global,
+    NoKeywordPass,
+}
+
 impl<'file> Database<'file> {
     pub fn save(&mut self) -> Result<()> {
         if !self.modified {

src/db/stream.rs

@@ -2,10 +2,14 @@ use std::iter::Rev;
 use std::ops::Range;
 use std::{fs, path};
 
-use crate::db::{Database, Dir, Epoch};
+use crate::db::{Database, Dir, Epoch, WorkingMode};
 use crate::util;
 
+use std::env;
+
 pub struct Stream<'db, 'file> {
+    pub working_mode: WorkingMode,
+
     db: &'db mut Database<'file>,
     idxs: Rev<Range<usize>>,
 
@@ -34,6 +38,7 @@ impl<'db, 'file> Stream<'db, 'file> {
             check_exists: false,
             expire_below,
             resolve_symlinks: false,
+            working_mode: WorkingMode::Global,
             exclude_path: None,
         }
     }
@@ -43,14 +48,42 @@ impl<'db, 'file> Stream<'db, 'file> {
         self
     }
 
+    pub fn with_globaldir(mut self) -> Self {
+        self.working_mode = WorkingMode::Global;
+        self
+    }
+
+    pub fn with_homedir(mut self) -> Self {
+        self.working_mode = WorkingMode::Home;
+        self
+    }
+
+    pub fn with_currentdir(mut self) -> Self {
+        self.working_mode = WorkingMode::Current;
+        self
+    }
+
     pub fn with_exists(mut self, resolve_symlinks: bool) -> Self {
         self.check_exists = true;
         self.resolve_symlinks = resolve_symlinks;
         self
     }
 
     pub fn with_keywords<S: AsRef<str>>(mut self, keywords: &[S]) -> Self {
-        self.keywords = keywords.iter().map(util::to_lowercase).collect();
+        let mut keywords_iter = keywords.iter();
+
+        let workingdir_mode = keywords[0].as_ref();
+        if workingdir_mode == "." {
+            self.working_mode = WorkingMode::Current;
+            keywords_iter.next();
+        } else if workingdir_mode == env::var("HOME").unwrap_or_else(|_| String::from("~")) {
+            self.working_mode = WorkingMode::Home;
+            keywords_iter.next();
+        } else {
+            self.working_mode = WorkingMode::NoKeywordPass;
+        }
+
+        self.keywords = keywords_iter.map(util::to_lowercase).collect();
         self
     }
 
@@ -74,6 +107,29 @@ impl<'db, 'file> Stream<'db, 'file> {
                 continue;
             }
 
+            match self.working_mode {
+                WorkingMode::Current => {
+                    if !dir.path.starts_with(env::current_dir().unwrap_or_default().to_str().unwrap_or_default()) {
+                        {
+                            continue;
+                        }
+                    }
+                }
+
+                WorkingMode::Home => {
+                    if !dir
+                        .path
+                        .starts_with(&env::var("HOME").unwrap_or_else(|_| env::var("UserProfile").unwrap_or_default()))
+                    {
+                        {
+                            continue;
+                        }
+                    }
+                }
+
+                WorkingMode::Global | WorkingMode::NoKeywordPass => {}
+            }
+
             let dir = &self.db.dirs[idx];
             return Some(dir);
         }

templates/bash.txt

@@ -93,6 +93,12 @@ function __zoxide_zi() {
     result="$(\command zoxide query -i -- "$@")" && __zoxide_cd "${result}"
 }
 
+# Jump to sub directories of current directory.
+function __zoxide_zw() {
+    \builtin local result
+    result="$(\command zoxide query --workingdir --exclude "$(__zoxide_pwd)" -- "$@")" && __zoxide_cd "${result}"
+}
+
 {{ section }}
 # Commands for zoxide. Disable these using --no-cmd.
 #
@@ -110,6 +116,11 @@ function {{cmd}}i() {
     __zoxide_zi "$@"
 }
 
+\builtin unalias {{cmd}}w &>/dev/null || \builtin true
+function {{cmd}}w() {
+    __zoxide_zw "$@"
+}
+
 # Load completions.
 # - Bash 4.4+ is required to use `@Q`.
 # - Completions require line editing. Since Bash supports only two modes of

templates/fish.txt

@@ -106,6 +106,12 @@ function __zoxide_zi
     and __zoxide_cd $result
 end
 
+# Jump to sub directories of current directory.
+function __zoxide_zw
+    set -l result (command zoxide query --workingdir -- $argv)
+    and __zoxide_cd $result
+end
+
 {{ section }}
 # Commands for zoxide. Disable these using --no-cmd.
 #
@@ -126,6 +132,12 @@ function {{cmd}}i
     __zoxide_zi $argv
 end
 
+abbr --erase {{cmd}}w &>/dev/null
+complete --command {{cmd}}w --erase
+function {{cmd}}w
+    __zoxide_zw $argv
+end
+
 {%- when None %}
 
 {{ not_configured }}

templates/zsh.txt

@@ -79,6 +79,12 @@ function __zoxide_zi() {
     result="$(\command zoxide query -i -- "$@")" && __zoxide_cd "${result}"
 }
 
+# Jump to sub directories of current directory.
+function __zoxide_zw() {
+    \builtin local result
+    result="$(\command zoxide query --workingdir -- "$@")" && __zoxide_cd "${result}"
+}
+
 {{ section }}
 # Commands for zoxide. Disable these using --no-cmd.
 #
@@ -96,6 +102,11 @@ function {{cmd}}i() {
     __zoxide_zi "$@"
 }
 
+\builtin unalias {{cmd}}w &>/dev/null || \builtin true
+function {{cmd}}w() {
+    __zoxide_zw "$@"
+}
+
 if [[ -o zle ]]; then
     function __zoxide_z_complete() {
         # Only show completions when the cursor is at the end of the line.