Add workspace mode, enable if in git repo

atuin-client/src/database.rs

@@ -1,4 +1,8 @@
-use std::{env, path::Path, str::FromStr};
+use std::{
+    env,
+    path::{Path, PathBuf},
+    str::FromStr,
+};
 
 use async_trait::async_trait;
 use atuin_common::utils;
@@ -25,6 +29,7 @@ pub struct Context {
     pub cwd: String,
     pub hostname: String,
     pub host_id: String,
+    pub git_root: Option<PathBuf>,
 }
 
 #[derive(Default, Clone)]
@@ -52,12 +57,14 @@ pub fn current_context() -> Context {
     );
     let cwd = utils::get_current_dir();
     let host_id = Settings::host_id().expect("failed to load host ID");
+    let git_root = utils::in_git_repo(cwd.as_str());
 
     Context {
         session,
         hostname,
         cwd,
         host_id,
+        git_root,
     }
 }
 
@@ -260,6 +267,7 @@ impl Database for Sqlite {
             FilterMode::Host => query.and_where_eq("hostname", quote(&context.hostname)),
             FilterMode::Session => query.and_where_eq("session", quote(&context.session)),
             FilterMode::Directory => query.and_where_eq("cwd", quote(&context.cwd)),
+            FilterMode::Workspace => query.and_where_like_any("cwd", context.cwd.clone()),
         };
 
         if unique {
@@ -379,11 +387,18 @@ impl Database for Sqlite {
             sql.order_desc("timestamp");
         }
 
+        let git_root = if let Some(git_root) = context.git_root.clone() {
+            git_root.to_str().unwrap_or("/").to_string()
+        } else {
+            context.cwd.clone()
+        };
+
         match filter {
             FilterMode::Global => &mut sql,
             FilterMode::Host => sql.and_where_eq("hostname", quote(&context.hostname)),
             FilterMode::Session => sql.and_where_eq("session", quote(&context.session)),
             FilterMode::Directory => sql.and_where_eq("cwd", quote(&context.cwd)),
+            FilterMode::Workspace => sql.and_where_like_left("cwd", git_root),
         };
 
         let orig_query = query;

atuin-client/src/settings.rs

@@ -69,6 +69,9 @@ pub enum FilterMode {
 
     #[serde(rename = "directory")]
     Directory = 3,
+
+    #[serde(rename = "workspace")]
+    Workspace = 4,
 }
 
 impl FilterMode {
@@ -78,6 +81,7 @@ impl FilterMode {
             FilterMode::Host => "HOST",
             FilterMode::Session => "SESSION",
             FilterMode::Directory => "DIRECTORY",
+            FilterMode::Workspace => "WORKSPACE",
         }
     }
 }
@@ -160,6 +164,7 @@ pub struct Settings {
     pub history_filter: RegexSet,
     #[serde(with = "serde_regex", default = "RegexSet::empty")]
     pub cwd_filter: RegexSet,
+    pub workspaces: bool,
 
     // This is automatically loaded when settings is created. Do not set in
     // config! Keep secrets and settings apart.
@@ -369,6 +374,7 @@ impl Settings {
             .set_default("scroll_context_lines", 1)?
             .set_default("shell_up_key_binding", false)?
             .set_default("session_token", "")?
+            .set_default("workspaces", false)?
             .add_source(
                 Environment::with_prefix("atuin")
                     .prefix_separator("_")

atuin-common/src/utils.rs

@@ -1,5 +1,5 @@
 use std::env;
-use std::path::PathBuf;
+use std::path::{Path, PathBuf};
 
 use chrono::{Months, NaiveDate};
 use rand::RngCore;
@@ -46,6 +46,31 @@ pub fn uuid_v4() -> String {
     Uuid::new_v4().as_simple().to_string()
 }
 
+pub fn has_git_dir(path: &str) -> bool {
+    let mut gitdir = PathBuf::from(path);
+    gitdir.push(".git");
+
+    gitdir.exists()
+}
+
+// detect if any parent dir has a git repo in it
+// I really don't want to bring in libgit for something simple like this
+// If we start to do anything more advanced, then perhaps
+pub fn in_git_repo(path: &str) -> Option<PathBuf> {
+    let mut gitdir = PathBuf::from(path);
+
+    while gitdir.parent().is_some() && !has_git_dir(gitdir.to_str().unwrap()) {
+        gitdir.pop();
+    }
+
+    // No parent? then we hit root, finding no git
+    if gitdir.parent().is_some() {
+        return Some(gitdir);
+    }
+
+    return None;
+}
+
 // TODO: more reliable, more tested
 // I don't want to use ProjectDirs, it puts config in awkward places on
 // mac. Data too. Seems to be more intended for GUI apps.

atuin/src/command/client/history.rs

@@ -13,7 +13,7 @@ use runtime_format::{FormatKey, FormatKeyError, ParseSegment, ParsedFmt};
 use atuin_client::{
     database::{current_context, Database},
     history::History,
-    settings::Settings,
+    settings::{FilterMode, Settings},
 };
 
 #[cfg(feature = "sync")]

atuin/src/command/client/search.rs

@@ -194,14 +194,21 @@ async fn run_non_interactive(
     let context = current_context();
 
     let opt_filter = OptFilters {
-        cwd: dir,
+        cwd: dir.clone(),
         ..filter_options
     };
 
+    let dir = dir.unwrap_or("/".to_string());
+    let filter_mode = if settings.workspaces && utils::has_git_dir(dir.as_str()) {
+        FilterMode::Workspace
+    } else {
+        settings.filter_mode
+    };
+
     let results = db
         .search(
             settings.search_mode,
-            settings.filter_mode,
+            filter_mode,
             &context,
             query.join(" ").as_str(),
             opt_filter,

atuin/src/command/client/search/interactive.rs

@@ -3,6 +3,7 @@ use std::{
     time::Duration,
 };
 
+use atuin_common::utils;
 use crossterm::{
     event::{self, Event, KeyCode, KeyEvent, KeyModifiers, MouseEvent},
     execute, terminal,
@@ -586,14 +587,16 @@ pub async fn history(
         search_mode: settings.search_mode,
         search: SearchState {
             input,
-            context,
-            filter_mode: if settings.shell_up_key_binding {
+            filter_mode: if settings.workspaces && context.git_root.is_some() {
+                FilterMode::Workspace
+            } else if settings.shell_up_key_binding {
                 settings
                     .filter_mode_shell_up_key_binding
                     .unwrap_or(settings.filter_mode)
             } else {
                 settings.filter_mode
             },
+            context,
         },
         engine: engines::engine(settings.search_mode),
         results_len: 0,

atuin/src/command/client/stats.rs

@@ -80,6 +80,7 @@ impl Cmd {
         } else {
             self.period.join(" ")
         };
+
         let history = if words.as_str() == "all" {
             db.list(FilterMode::Global, &context, None, false).await?
         } else if words.trim() == "today" {