tentative changes testing

Author: vageorge00

src/azure-cli/azure/cli/command_modules/appservice/_help.py

@@ -1917,6 +1917,28 @@
     text: az webapp create-remote-connection --name MyWebApp --resource-group MyResourceGroup
 """
 
+helps['webapp exec'] = """
+type: command
+short-summary: Execute a command in a web app container.
+long-summary: >
+    Execute commands remotely in your web app container. The command runs asynchronously 
+    and returns immediately without command output. Requires SCM Basic Auth Publishing 
+    Credentials to be enabled.
+examples:
+  - name: Execute a simple command on a web app
+    text: >
+        az webapp exec -g MyResourceGroup -n MyWebapp --cmd "ls -la /home"
+  - name: Execute a command in a specific working directory on a specific instance
+    text: >
+        az webapp exec -g MyResourceGroup -n MyWebapp --cmd "npm install" --cwd /home/site/wwwroot --instance MyInstanceId
+  - name: Execute a command on all instances
+    text: >
+        az webapp exec -g MyResourceGroup -n MyWebapp --cmd "systemctl restart nginx" --instance all
+  - name: Execute a command on a deployment slot
+    text: >
+        az webapp exec -g MyResourceGroup -n MyWebapp -s staging --cmd "python manage.py migrate"
+"""
+
 helps['webapp delete'] = """
 type: command
 short-summary: Delete a web app.

src/azure-cli/azure/cli/command_modules/appservice/_params.py

@@ -1502,3 +1502,15 @@ def load_arguments(self, _):
         c.argument('environment_name', help="Name of the environment of static site")
     with self.argument_context('staticwebapp enterprise-edge') as c:
         c.argument("no_register", help="Don't try to register the Microsoft.CDN provider. Registration can be done manually with: az provider register --wait --namespace Microsoft.CDN. For more details, please review the documentation available at https://go.microsoft.com/fwlink/?linkid=2184995 .", default=False)
+    with self.argument_context('webapp exec') as c:
+        c.argument('name', arg_type=webapp_name_arg_type, id_part=None)
+        c.argument('command', options_list=['--command', '--cmd'], help='The command to execute in the container.')
+        c.argument('mode', 
+                    help='Execution mode. \'execute\': Starts command execution and returns immediately without returning command output.', 
+                    arg_type=get_enum_type(['execute']), default='execute')
+        c.argument('working_directory', options_list=['--working-directory', '--cwd'], 
+                    help='Working directory for command execution')
+        c.argument('instance', options_list=['--instance', '-i'], 
+                    help='Webapp instance(s) to target. Specify a comma-separated list of instance IDs (use "az webapp list-instances" to get IDs) or "all" for all instances. Defaults to a random instance.')
+        c.argument('slot', options_list=['--slot', '-s'], 
+                    help='Name of the web app slot. Default to the production slot if not specified.')

src/azure-cli/azure/cli/command_modules/appservice/commands.py

@@ -125,6 +125,7 @@ def load_command_table(self, _):
         g.custom_command('create', 'create_webapp', exception_handler=ex_handler_factory(), validator=validate_vnet_integration)
         g.custom_command('up', 'webapp_up', exception_handler=ex_handler_factory(), validator=validate_webapp_up)
         g.custom_command('ssh', 'ssh_webapp', exception_handler=ex_handler_factory(), is_preview=True)
+        g.custom_command('exec', 'webapp_exec', exception_handler=ex_handler_factory(), is_preview=True)
         g.custom_command('list', 'list_webapp', table_transformer=transform_web_list_output)
         g.custom_show_command('show', 'show_app', table_transformer=transform_web_output)
         g.custom_command('delete', 'delete_webapp')

src/azure-cli/azure/cli/command_modules/appservice/custom.py

@@ -10947,3 +10947,94 @@ def _compute_checksum(input_bytes):
         logger.info("Computing the checksum of the file failed with exception:'%s'", ex)
 
     return file_hash
+
+
+def _execute_command_on_instance(scm_url, headers, cookies, command, working_directory=None):
+    from azure.cli.core.util import should_disable_connection_verify
+    import requests
+    
+    # Build the exec endpoint URL
+    exec_url = f"{scm_url}/exec/execute"
+    
+    # Build request body
+    body = {"Command": command}
+    if working_directory:
+        body["WorkingDirectory"] = working_directory
+    
+    # Make the request
+    response = requests.post(
+        exec_url,
+        json=body,
+        headers=headers,
+        cookies=cookies,
+        verify=not should_disable_connection_verify()
+    )
+    
+    # Handle response
+    if response.status_code == 202:
+        # Success - command accepted (no response body expected)
+        return None
+    else:
+        # Error - response contains JSON with ErrorMessage
+        raise CLIError(f"Command execution failed with status code {response.status_code}: {response.text}")
+
+
+def webapp_exec(cmd, resource_group_name, name, command, mode='execute', working_directory=None, instance=None, slot=None):
+    # Validate Linux App
+    webapp = _generic_site_operation(cmd.cli_ctx, resource_group_name, name, 'get', slot)
+    if not webapp:
+        raise ResourceNotFoundError("Unable to find resource '{}' in ResourceGroup '{}'.".format(name, resource_group_name))
+
+    is_linux = webapp.reserved
+    if not is_linux:
+        raise ValidationError("Only Linux App Service Plans supported.")
+
+    # Allow mode 'execute' only
+    if mode.lower() != 'execute': 
+        raise ValidationError("Only 'execute' mode is supported currently.")
+
+    # Validate command is provided if mode is 'execute'
+    if mode.lower() == 'execute' and not command:
+        raise ValidationError("Command is required.")
+
+    # Get scm site and authorization
+    scm_url = _get_scm_url(cmd, resource_group_name, name, slot)
+    headers = get_scm_site_headers(cmd.cli_ctx, name, resource_group_name, slot)
+
+    # Get target instances if specified
+    target_instances = []
+    if instance is not None:
+        # If command execution on all instances
+        if instance.lower() == "all":
+            instances = list_instances(cmd, resource_group_name, name, slot=slot)
+            target_instances = [i.name for i in instances]
+            if not target_instances:
+                raise ValidationError("No instances found for this web app.")
+        else:
+            # Validate listed instances
+            instances = list_instances(cmd, resource_group_name, name, slot=slot)
+            requested_instances = [i.strip() for i in instance.split(',')]
+            instance_names = set(i.name for i in instances)
+            invalid_instances = [inst for inst in requested_instances if inst not in instance_names]
+            if invalid_instances:
+                raise ValidationError("The following instances are not valid for this webapp: {}. Valid instances: {}".format(
+                    ', '.join(invalid_instances), ', '.join(sorted(instance_names))))
+            target_instances = requested_instances
+
+        # Execute command on target instances
+        results = []
+        for target in target_instances if target_instances else [None]:
+            cookies = {}
+            if target is not None:
+                cookies['ARRAffinity'] = target
+
+            try:
+                result = _execute_command_on_instance(scm_url, headers, cookies, command, working_directory)
+                results.append({'instance': target or 'default', 'status': 'success', 'result': result})
+            except CLIError as e:
+                results.append({'instance': target or 'default', 'status': 'failed', 'error': str(e)})
+
+        return results if len(results) > 1 else results[0] if results else None
+
+        
+