define a global capture dirctory

Author: evanliu048

crates/chat-cli/src/cli/chat/capture.rs

@@ -23,7 +23,7 @@ use crate::cli::ConversationState;
 use crate::os::Os;
 
 // The shadow repo path that MUST be appended with a session-specific directory
-pub const SHADOW_REPO_DIR: &str = "/Users/kiranbug/.amazonq/captures/";
+// pub const SHADOW_REPO_DIR: &str = "/Users/kiranbug/.amazonq/captures/";
 
 // CURRENT APPROACH:
 // We only enable automatically enable checkpoints when the user is already in a git repo.

crates/chat-cli/src/cli/chat/cli/capture.rs

@@ -1,5 +1,4 @@
 use std::io::Write;
-use std::path::PathBuf;
 
 use clap::Subcommand;
 use crossterm::style::{
@@ -16,14 +15,14 @@ use eyre::Result;
 use crate::cli::chat::capture::{
     Capture,
     CaptureManager,
-    SHADOW_REPO_DIR,
 };
 use crate::cli::chat::{
     ChatError,
     ChatSession,
     ChatState,
 };
 use crate::os::Os;
+use crate::util::directories::get_shadow_repo_dir;
 
 #[derive(Debug, PartialEq, Subcommand)]
 pub enum CaptureSubcommand {
@@ -68,7 +67,8 @@ impl CaptureSubcommand {
                     )
                 )?;
             } else {
-                let path = PathBuf::from(SHADOW_REPO_DIR).join(session.conversation.conversation_id());
+                let path = get_shadow_repo_dir(os, session.conversation.conversation_id().to_string())
+                    .map_err(|e| ChatError::Custom(e.to_string().into()))?;
                 let start = std::time::Instant::now();
                 session.conversation.capture_manager = Some(
                     CaptureManager::manual_init(os, path)

crates/chat-cli/src/util/directories.rs

@@ -43,6 +43,7 @@ pub enum DirectoryError {
 type Result<T, E = DirectoryError> = std::result::Result<T, E>;
 
 const WORKSPACE_AGENT_DIR_RELATIVE: &str = ".amazonq/cli-agents";
+const GLOBAL_SHADOW_REPO_DIR: &str = ".aws/.amazonq/cli-captures";
 const GLOBAL_AGENT_DIR_RELATIVE_TO_HOME: &str = ".aws/amazonq/cli-agents";
 const CLI_BASH_HISTORY_PATH: &str = ".aws/amazonq/.cli_bash_history";
 
@@ -249,6 +250,10 @@ pub fn get_mcp_auth_dir(os: &Os) -> Result<PathBuf> {
     Ok(home_dir(os)?.join(".aws").join("sso").join("cache"))
 }
 
+pub fn get_shadow_repo_dir(os: &Os, conversation_id: String) -> Result<PathBuf> {
+    Ok(home_dir(os)?.join(GLOBAL_SHADOW_REPO_DIR).join(conversation_id))
+}
+
 /// Generate a unique identifier for an agent based on its path and name
 fn generate_agent_unique_id(agent: &crate::cli::Agent) -> String {
     use std::collections::hash_map::DefaultHasher;