feat: add env command (#1829)

Author: tusharmath

crates/forge_main/src/cli.rs

@@ -102,6 +102,11 @@ pub enum TopLevelCommand {
         porcelain: bool,
     },
 
+    /// Display environment information
+    ///
+    /// Example: forge env
+    Env,
+
     /// Configuration management commands
     Config(ConfigCommandGroup),
 

crates/forge_main/src/info.rs

@@ -75,7 +75,13 @@ impl From<&Environment> for Info {
             None => "(not in a git repository)".to_string(),
         };
 
-        let mut info = Info::new().add_title("PATHS");
+        let mut info = Info::new()
+            .add_title("ENVIRONMENT")
+            .add_key_value("Version", VERSION)
+            .add_key_value("Working Directory", format_path_for_display(env, &env.cwd))
+            .add_key_value("Shell", &env.shell)
+            .add_key_value("Git Branch", branch_info)
+            .add_title("PATHS");
 
         // Only show logs path if the directory exists
         let log_path = env.log_path();
@@ -84,9 +90,8 @@ impl From<&Environment> for Info {
         }
 
         let agent_path = env.agent_path();
-        info = info.add_key_value("Agents", format_path_for_display(env, &agent_path));
-
         info = info
+            .add_key_value("Agents", format_path_for_display(env, &agent_path))
             .add_key_value("History", format_path_for_display(env, &env.history_path()))
             .add_key_value(
                 "Checkpoints",
@@ -95,12 +100,7 @@ impl From<&Environment> for Info {
             .add_key_value(
                 "Policies",
                 format_path_for_display(env, &env.permissions_path()),
-            )
-            .add_title("ENVIRONMENT")
-            .add_key_value("Version", VERSION)
-            .add_key_value("Working Directory", format_path_for_display(env, &env.cwd))
-            .add_key_value("Shell", &env.shell)
-            .add_key_value("Git Branch", branch_info);
+            );
 
         info
     }
@@ -215,7 +215,7 @@ impl fmt::Display for Info {
                         }
                     } else {
                         // Show value-only items
-                        writeln!(f, "    ⦿ {}", value)?;
+                        writeln!(f, "    {} {}", "⦿".cyan(), value)?;
                     }
                 }
             }

crates/forge_main/src/model.rs

@@ -329,6 +329,7 @@ impl ForgeCommandManager {
             "/compact" => Ok(SlashCommand::Compact),
             "/new" => Ok(SlashCommand::New),
             "/info" => Ok(SlashCommand::Info),
+            "/env" => Ok(SlashCommand::Env),
             "/usage" => Ok(SlashCommand::Usage),
             "/exit" => Ok(SlashCommand::Exit),
             "/update" => Ok(SlashCommand::Update),
@@ -417,6 +418,9 @@ pub enum SlashCommand {
     /// Display usage information (tokens & requests).
     #[strum(props(usage = "Shows usage information (tokens & requests)"))]
     Usage,
+    /// Display environment information.
+    #[strum(props(usage = "Display environment information"))]
+    Env,
     /// Exit the application without any further action.
     #[strum(props(usage = "Exit the application"))]
     Exit,
@@ -495,6 +499,7 @@ impl SlashCommand {
             SlashCommand::Message(_) => "message",
             SlashCommand::Update => "update",
             SlashCommand::Info => "info",
+            SlashCommand::Env => "env",
             SlashCommand::Usage => "usage",
             SlashCommand::Exit => "exit",
             SlashCommand::Forge => "forge",

crates/forge_main/src/ui.rs

@@ -400,6 +400,10 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
                 self.on_info(porcelain, conversation_id).await?;
                 return Ok(());
             }
+            TopLevelCommand::Env => {
+                self.on_env().await?;
+                return Ok(());
+            }
             TopLevelCommand::Banner => {
                 banner::display(true)?;
                 return Ok(());
@@ -663,6 +667,7 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
         // Define base commands with their descriptions
         info = info
             .add_key_value("info", "Print session information")
+            .add_key_value("env", "Display environment information")
             .add_key_value("provider", "Switch the providers")
             .add_key_value("model", "Switch the models")
             .add_key_value("new", "Start new conversation")
@@ -793,7 +798,7 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
         porcelain: bool,
         conversation_id: Option<ConversationId>,
     ) -> anyhow::Result<()> {
-        let mut info = Info::from(&self.api.environment());
+        let mut info = Info::new();
 
         // Fetch conversation
         let conversation = match conversation_id {
@@ -876,6 +881,13 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
         Ok(())
     }
 
+    async fn on_env(&mut self) -> anyhow::Result<()> {
+        let env = self.api.environment();
+        let info = Info::from(&env);
+        self.writeln(info)?;
+        Ok(())
+    }
+
     async fn on_zsh_prompt(&self) -> anyhow::Result<()> {
         println!("{}", include_str!("../../../shell-plugin/forge.plugin.zsh"));
         Ok(())
@@ -981,6 +993,9 @@ impl<A: API + 'static, F: Fn() -> A> UI<A, F> {
             SlashCommand::Info => {
                 self.on_info(false, None).await?;
             }
+            SlashCommand::Env => {
+                self.on_env().await?;
+            }
             SlashCommand::Usage => {
                 self.on_usage().await?;
             }

shell-plugin/forge.plugin.zsh

@@ -237,6 +237,13 @@ function _forge_action_info() {
     _forge_reset
 }
 
+# Action handler: Show environment info
+function _forge_action_env() {
+    echo
+    _forge_exec env
+    _forge_reset
+}
+
 # Action handler: Dump conversation
 function _forge_action_dump() {
     local input_text="$1"
@@ -428,6 +435,9 @@ function forge-accept-line() {
         info|i)
             _forge_action_info
         ;;
+        env|e)
+            _forge_action_env
+        ;;
         dump)
             _forge_action_dump "$input_text"
         ;;