feat(execute_command): enhance command management with update functionality and improve example usage

Author: akosyakov

examples/execute_command.py

@@ -3,7 +3,7 @@
 from gitpod import Gitpod
 from utils.context import get_context_url, get_authenticated_user
 from utils.environment import get_environment_class_id, get_environment_class, create_environment
-from utils.task import create_command, run_command, delete_command, stream_command_logs
+from utils.task import create_command, run_command, update_command, delete_command
 
 def main() -> None:
     """
@@ -33,22 +33,21 @@ def main() -> None:
 
     try:
         # Define command
-        reference = "hello-world"
-        name = "Hello, World!"
-        description = "Prints 'Hello, World!' to the console."
-        task_id = create_command(client, environment_id, reference, name, description)    
+        reference = "greeting"
+        name = "Greeting"
+        description = "Prints a greeting to the console."
+        command = "echo 'Hello, World!'"
+        task_id = create_command(client, environment_id, reference, name, description, command)    
         try:
-            # Run command
-            execution_id = run_command(client, task_id)
-            
-            # Stream logs
-            stream_command_logs(client, environment_id, execution_id)
+            # Run command with first greeting
+            run_command(client, environment_id, task_id)
 
-            # Check command status
-            execution = client.environments.automations.tasks.executions.retrieve(id=execution_id).task_execution
-            print(f"\nTask execution status: {execution.status.phase}")
-            if execution.status.phase == "TASK_EXECUTION_PHASE_FAILED":
-                print(f"Task execution failed: {execution.status.failure_message}")
+            # Update command with second greeting
+            command = "echo 'Howdy, Partner!'"
+            update_command(client, task_id, command)
+            
+            # Run command with second greeting
+            run_command(client, environment_id, task_id)
         finally: 
             # Clean up command
             delete_command(client, task_id)

examples/utils/task.py

@@ -4,7 +4,7 @@
 from utils.environment import ensure_environment_healthy
 
 
-def create_command(client: Gitpod, environment_id: str, reference: str, name: str, description: str) -> str:
+def create_command(client: Gitpod, environment_id: str, reference: str, name: str, description: str, command: str) -> str:
     """
     Create a task in the given environment.
 
@@ -20,7 +20,7 @@ def create_command(client: Gitpod, environment_id: str, reference: str, name: st
     """
     task = client.environments.automations.tasks.create(
         spec={
-            "command": "echo 'Hello, World!'",
+            "command": command,
         },
         environment_id=environment_id,
         metadata={
@@ -32,8 +32,24 @@ def create_command(client: Gitpod, environment_id: str, reference: str, name: st
     print(f"\nCreated task: {task.id}")
     return task.id
 
+def update_command(client: Gitpod, task_id: str, command: str) -> None:
+    """
+    Update the command of an existing task.
+
+    Args:
+        client (Gitpod): The Gitpod client instance.
+        task_id (str): The task ID.
+        new_command (str): The new command to set.
+    """
+    client.environments.automations.tasks.update(
+        id=task_id,
+        spec={
+            "command": command,
+        }
+    )
+    print(f"Updated task: {task_id}")
 
-def run_command(client: Gitpod, task_id: str) -> str:
+def start_command(client: Gitpod, task_id: str) -> str:
     """
     Start a task execution.
 
@@ -127,3 +143,25 @@ def stream_command_logs(client: Gitpod, environment_id: str, execution_id: str)
 
     logsAccessToken = client.environments.create_logs_token(environment_id=environment_id).access_token
     asyncio.run(stream_logs(log_url, logsAccessToken))
+
+def run_command(client: Gitpod, environment_id: str, task_id: str) -> None:
+    """
+    Run a command, stream its logs, and check its status.
+
+    Args:
+        client (Gitpod): The Gitpod client instance.
+        environment_id (str): The environment ID.
+        task_id (str): The task ID.
+    """
+    # Start command
+    execution_id = start_command(client, task_id)
+    
+    # Stream logs
+    stream_command_logs(client, environment_id, execution_id)
+
+    # Check command status
+    execution = client.environments.automations.tasks.executions.retrieve(id=execution_id).task_execution
+    print(f"\nTask execution status: {execution.status.phase}")
+    if execution.status.phase == "TASK_EXECUTION_PHASE_FAILED":
+        print(f"Task execution failed: {execution.status.failure_message}")
+