Adding command param in debug command

Author: Shivam Kumar

src/containerapp/azext_containerapp/_ssh_utils.py

@@ -9,23 +9,28 @@
     SSH_DEFAULT_ENCODING, read_ssh
 from azure.cli.core.commands.client_factory import get_subscription_id
 
+import urllib
 from knack.log import get_logger
 
 logger = get_logger(__name__)
 
 
 class DebugWebSocketConnection(WebSocketConnection):
-    def __init__(self, cmd, resource_group_name, name, revision, replica, container):
+    def __init__(self, cmd, resource_group_name, name, revision, replica, container, command):
         super(DebugWebSocketConnection, self).__init__(cmd, resource_group_name, name, revision, replica, container, "")
+        self.command = urllib.parse.quote_plus(command) if command else None
 
     def _get_url(self, cmd, resource_group_name, name, revision, replica, container, startup_command):
         sub = get_subscription_id(cmd.cli_ctx)
         base_url = self._logstream_endpoint
         proxy_api_url = base_url[:base_url.index("/subscriptions/")].replace("https://", "")
 
-        return (f"wss://{proxy_api_url}/subscriptions/{sub}/resourceGroups/{resource_group_name}/containerApps/{name}"
+        debug_url = (f"wss://{proxy_api_url}/subscriptions/{sub}/resourceGroups/{resource_group_name}/containerApps/{name}"
                 f"/revisions/{revision}/replicas/{replica}/debug"
                 f"?targetContainer={container}")
+        if self.command:
+            debug_url += f"&command={self.command}"
+        return debug_url
 
 
 def read_debug_ssh(connection: WebSocketConnection, response_encodings):

src/containerapp/azext_containerapp/custom.py

@@ -3591,15 +3591,16 @@ def list_maintenance_config(cmd, resource_group_name, env_name):
     return r
 
 
-def containerapp_debug(cmd, resource_group_name, name, container=None, revision=None, replica=None):
+def containerapp_debug(cmd, resource_group_name, name, container=None, revision=None, replica=None, command=None):
     logger.warning("Connecting...")
     conn = DebugWebSocketConnection(
         cmd=cmd,
         resource_group_name=resource_group_name,
         name=name,
         revision=revision,
         replica=replica,
-        container=container
+        container=container,
+        command=command
     )
 
     encodings = [SSH_DEFAULT_ENCODING, SSH_BACKUP_ENCODING]