adds agent root command for migrate (#2418)

Author: dingfeli

crates/chat-cli/src/cli/agent/legacy/mod.rs

@@ -25,9 +25,9 @@ use crate::util::directories;
 ///
 /// Returns [Some] with the newly migrated agents if the migration was performed, [None] if the
 /// migration was already done previously.
-pub async fn migrate(os: &mut Os) -> eyre::Result<Option<Vec<Agent>>> {
+pub async fn migrate(os: &mut Os, force: bool) -> eyre::Result<Option<Vec<Agent>>> {
     let has_migrated = os.database.get_has_migrated()?;
-    if has_migrated.is_some_and(|has_migrated| has_migrated) {
+    if !force && has_migrated.is_some_and(|has_migrated| has_migrated) {
         return Ok(None);
     }
 
@@ -102,22 +102,27 @@ pub async fn migrate(os: &mut Os) -> eyre::Result<Option<Vec<Agent>>> {
     }
 
     let labels = vec!["Yes", "No"];
-    let selection: Option<_> = match Select::with_theme(&crate::util::dialoguer_theme())
-        .with_prompt("Legacy profiles detected. Would you like to migrate them?")
-        .items(&labels)
-        .default(1)
-        .interact_on_opt(&dialoguer::console::Term::stdout())
-    {
-        Ok(sel) => {
-            let _ = crossterm::execute!(
-                std::io::stdout(),
-                crossterm::style::SetForegroundColor(crossterm::style::Color::Magenta)
-            );
-            sel
-        },
-        // Ctrl‑C -> Err(Interrupted)
-        Err(dialoguer::Error::IO(ref e)) if e.kind() == std::io::ErrorKind::Interrupted => None,
-        Err(e) => bail!("Failed to choose an option: {e}"),
+    let selection: Option<_> = if !force {
+        match Select::with_theme(&crate::util::dialoguer_theme())
+            .with_prompt("Legacy profiles detected. Would you like to migrate them?")
+            .items(&labels)
+            .default(1)
+            .interact_on_opt(&dialoguer::console::Term::stdout())
+        {
+            Ok(sel) => {
+                let _ = crossterm::execute!(
+                    std::io::stdout(),
+                    crossterm::style::SetForegroundColor(crossterm::style::Color::Magenta)
+                );
+                sel
+            },
+            // Ctrl‑C -> Err(Interrupted)
+            Err(dialoguer::Error::IO(ref e)) if e.kind() == std::io::ErrorKind::Interrupted => None,
+            Err(e) => bail!("Failed to choose an option: {e}"),
+        }
+    } else {
+        // Yes
+        Some(0)
     };
 
     if selection.is_none() || selection == Some(1) {

crates/chat-cli/src/cli/agent/mod.rs

@@ -399,7 +399,7 @@ impl Agents {
         };
 
         let new_agents = if !skip_migration {
-            match legacy::migrate(os).await {
+            match legacy::migrate(os, false).await {
                 Ok(Some(new_agents)) => {
                     let migrated_count = new_agents.len();
                     info!(migrated_count, "Profile migration successful");

crates/chat-cli/src/cli/agent/root_command_args.rs

@@ -21,6 +21,7 @@ use super::{
     Agent,
     Agents,
     McpServerConfig,
+    legacy,
 };
 use crate::database::settings::Setting;
 use crate::os::Os;
@@ -62,6 +63,13 @@ pub enum AgentSubcommands {
         #[arg(long, short)]
         path: String,
     },
+    /// Migrate profiles to agent
+    /// Note that doing this is potentially destructive to agents that are already in the global
+    /// agent directories
+    Migrate {
+        #[arg(long)]
+        force: bool,
+    },
 }
 
 #[derive(Debug, Clone, PartialEq, Eq, Default, Args)]
@@ -201,7 +209,61 @@ impl AgentArgs {
 
                 stderr.flush()?;
             },
+            Some(AgentSubcommands::Migrate { force }) => {
+                if !force {
+                    let _ = queue!(
+                        stderr,
+                        style::SetForegroundColor(Color::Yellow),
+                        style::Print("WARNING: "),
+                        style::ResetColor,
+                        style::Print(
+                            "manual migrate is potentially destructive to existing agent configs with name collision. Use"
+                        ),
+                        style::SetForegroundColor(Color::Cyan),
+                        style::Print(" --force "),
+                        style::ResetColor,
+                        style::Print("to run"),
+                        style::Print("\n"),
+                    );
+                    return Ok(ExitCode::SUCCESS);
+                }
+
+                match legacy::migrate(os, force).await {
+                    Ok(Some(new_agents)) => {
+                        let migrated_count = new_agents.len();
+                        let _ = queue!(
+                            stderr,
+                            style::SetForegroundColor(Color::Green),
+                            style::Print("✓ Success: "),
+                            style::ResetColor,
+                            style::Print(format!(
+                                "Profile migration successful. Migrated {} agent(s)\n",
+                                migrated_count
+                            )),
+                        );
+                    },
+                    Ok(None) => {
+                        let _ = queue!(
+                            stderr,
+                            style::SetForegroundColor(Color::Blue),
+                            style::Print("Info: "),
+                            style::ResetColor,
+                            style::Print("Migration was not performed. Nothing to migrate\n"),
+                        );
+                    },
+                    Err(e) => {
+                        let _ = queue!(
+                            stderr,
+                            style::SetForegroundColor(Color::Red),
+                            style::Print("Error: "),
+                            style::ResetColor,
+                            style::Print(format!("Migration did not happen for the following reason: {e}\n")),
+                        );
+                    },
+                }
+            },
         }
+
         Ok(ExitCode::SUCCESS)
     }
 }