open-session: add shell completions (#136)

* open-session: add shell completions * update dependencies
jrmoulton · Dec 2, 2024 · 75ef696 · 75ef696
1 parent db71f3b
commit 75ef696
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 66 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/flake.lock b/flake.lock
diff --git a/src/cli.rs b/src/cli.rs
@@ -16,6 +16,7 @@ use crate::{
     Result, TmsError,
 };
 use clap::{Args, Parser, Subcommand};
+use clap_complete::{ArgValueCandidates, CompletionCandidate};
 use error_stack::ResultExt;
 use git2::{build::RepoBuilder, FetchOptions, RemoteCallbacks, Repository};
 use ratatui::style::Color;
@@ -145,6 +146,7 @@ pub struct BookmarkCommand {
 
 #[derive(Debug, Args)]
 pub struct OpenSessionCommand {
+    #[arg(add = ArgValueCandidates::new(open_session_completion_candidates))]
     /// Name of the session to open.
     session: Box<str>,
 }
@@ -762,6 +764,20 @@ fn open_session_command(args: &OpenSessionCommand, config: Config, tmux: &Tmux)
     }
 }
 
+fn open_session_completion_candidates() -> Vec<CompletionCandidate> {
+    Config::new()
+        .change_context(TmsError::ConfigError)
+        .and_then(|config| create_sessions(&config))
+        .map(|sessions| {
+            sessions
+                .list()
+                .iter()
+                .map(CompletionCandidate::new)
+                .collect::<Vec<_>>()
+        })
+        .unwrap_or_default()
+}
+
 pub enum SubCommandGiven {
     Yes,
     No(Box<Config>),