...

src/virt/buildah/builder.rs

@@ -355,6 +355,64 @@ impl Builder {
         }
     }
     
+    /// Set the entrypoint for the container
+    ///
+    /// # Arguments
+    ///
+    /// * `entrypoint` - The entrypoint command
+    ///
+    /// # Returns
+    ///
+    /// * `Result<CommandResult, BuildahError>` - Command result or error
+    pub fn set_entrypoint(&self, entrypoint: &str) -> Result<CommandResult, BuildahError> {
+        if let Some(container_id) = &self.container_id {
+            // Save the current debug flag
+            let previous_debug = thread_local_debug();
+            
+            // Set the thread-local debug flag from the Builder's debug flag
+            set_thread_local_debug(self.debug);
+            
+            // Execute the command
+            let result = execute_buildah_command(&["config", "--entrypoint", entrypoint, container_id]);
+            
+            // Restore the previous debug flag
+            set_thread_local_debug(previous_debug);
+            
+            result
+        } else {
+            Err(BuildahError::Other("No container ID available".to_string()))
+        }
+    }
+    
+    /// Set the default command for the container
+    ///
+    /// # Arguments
+    ///
+    /// * `cmd` - The default command
+    ///
+    /// # Returns
+    ///
+    /// * `Result<CommandResult, BuildahError>` - Command result or error
+    pub fn set_cmd(&self, cmd: &str) -> Result<CommandResult, BuildahError> {
+        if let Some(container_id) = &self.container_id {
+            // Save the current debug flag
+            let previous_debug = thread_local_debug();
+            
+            // Set the thread-local debug flag from the Builder's debug flag
+            set_thread_local_debug(self.debug);
+            
+            // Execute the command
+            let result = execute_buildah_command(&["config", "--cmd", cmd, container_id]);
+            
+            // Restore the previous debug flag
+            set_thread_local_debug(previous_debug);
+            
+            result
+        } else {
+            Err(BuildahError::Other("No container ID available".to_string()))
+        }
+    }
+    
     /// List images in local storage
     ///
     /// # Returns