Karenchen/enable draftv2

src/aks-preview/azext_aks_preview/_help.py

@@ -1572,3 +1572,27 @@
     type: command
     short-summary: Delete a snapshot.
 """
+
+helps['aks app init'] = """
+    type: command
+    short-summary: Generate a Dockerfile and the minimum required Kubernetes deployment files (helm, kustomize, manifests) for your project directory.
+    parameters:
+        - name: --path
+          type: string
+          short-summary: Specify the path to the project directory.
+        - name: --app-name
+          type: string
+          short-summary: Specify the name of the helm release.
+        - name: --language
+          type: string
+          short-summary: Specify the language used to create the Kubernetes deployment.
+        - name: --create-config
+          type: string
+          short-summary: Specify the path to the configuration file.
+        - name: --dockerfile-only
+          type: string
+          short-summary: Only generate Dockerfile for the Kubernetes deployment.
+        - name: --deployment-only
+          type: string
+          short-summary: Only generate deployment files (helm, kustomize, manifests) for the Kubernetes deployment.
+"""

src/aks-preview/azext_aks_preview/aks_app_commands/aksappcommands.py

@@ -0,0 +1,165 @@
+# pylint: disable=too-many-lines
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+
+import sre_parse
+from typing import List, Optional
+import shutil
+import subprocess
+import requests
+import os
+import platform
+from pathlib import Path
+from knack.prompting import prompt_y_n
+
+
+def aks_draft_app_init(deployment_path: str,
+                       app_name: str,
+                       language: str,
+                       create_config: str,
+                       dockerfile_only: str,
+                       deployment_only: str) -> None:
+    file_path = _binary_pre_check()
+    if not file_path:
+        raise ValueError('Binary check was NOT executed successfully')
+
+    arguments = _build_arguments(app_name, language, create_config, dockerfile_only, deployment_only)
+    run_successful = _run(file_path, deployment_path, arguments)
+    if run_successful:
+        _cmd_finish()
+    else:
+        raise ValueError('\'az aks app init\' was NOT executed successfully')
+
+
+# If setup is valid this method returns the correct binary path to execute
+def _binary_pre_check() -> str:
+    print('The DraftV2 setup is in progress...')
+    draftv2_binary_path = _get_existing_path()
+
+    if draftv2_binary_path:  # use found binary
+        return draftv2_binary_path
+    else:  # prompt user to download binary
+        return _download_binary()
+
+
+def _get_filename() -> Optional[str]:
+    operating_system = platform.system().lower()
+    architecture = platform.machine().lower()
+
+    if architecture == 'x86_64':
+        architecture = 'amd64'
+    if architecture not in ['arm64', 'amd64']:
+        print('Cannot find a suitable download for the current system architecture. Draft only supports AMD64 and ARM64.')
+        return None
+
+    return f'draftv2-{operating_system}-{architecture}'
+
+
+# Returns path to existing draftv2 binary and None otherwise
+def _get_existing_path() -> Optional[str]:
+    print('Checking if DraftV2 binary exists locally...')
+
+    filename = _get_filename()
+    if not filename:
+        return None
+
+    paths = _get_potential_paths()
+    if not paths:
+        print('List of potential DraftV2 paths is empty')
+        return None
+
+    for path in paths:
+        binary_file_path = path + '/' + filename
+        if os.path.exists(binary_file_path):
+            print('Existing binary found at: ' + binary_file_path)
+            return binary_file_path
+    return None
+
+
+# Returns a list of potential draftV2 binary paths
+def _get_potential_paths() -> List[str]:
+    result = [str(Path.home()) + '/' + '.aksapp']
+    paths = os.environ['PATH'].split(':')
+
+    for path in paths:
+        if 'draftv2' in path:
+            result.append(path)
+    return result
+
+
+def _build_arguments(app_name: str,
+                     language: str,
+                     create_config: str,
+                     dockerfile_only: str,
+                     deployment_only: str) -> List[str]:
+    options = {
+        'app-name': app_name,
+        'language': language,
+        'create-config': create_config,
+        'dockerfile-only': dockerfile_only,
+        'deployment-only': deployment_only
+    }
+    args_list = []
+    for arg, val in options.items():
+        if val:
+            args_list.append(f'--{arg}={val}')
+    return args_list
+
+
+def _run(binary_path: str, deployment_path: str, arguments: List[str]) -> bool:
+    if binary_path is None:
+        raise ValueError('The given Binary path was null or empty')
+
+    print('Running DraftV2 Binary ...')
+    cmd = [binary_path, 'create', deployment_path] + arguments
+    process = subprocess.Popen(cmd)
+    exit_code = process.wait()
+    return exit_code == 0
+
+
+def _cmd_finish():
+    # Clean up logic can go here if needed
+    print('Finishing running \'az aks app init\'')
+
+
+def _download_binary() -> Optional[str]:
+    # prompt user to download binary. If users says no, we error out and tell them that this requires the binary
+    msg = 'The required binary was not found. Would you like us to download the required binary for you?'
+
+    if not prompt_y_n(msg, default='n'):
+        raise ValueError('`az aks app` requires the missing dependency')
+
+    print('Attempting to download dependency...')
+
+    filename = _get_filename()
+    if not filename:
+        return None
+
+    url = f'https://github.com/Azure/aks-app/releases/latest/download/{filename}'
+    headers = {'Accept': 'application/octet-stream'}
+
+    dir_name = '.aksapp'
+    # Downloading the file by sending the request to the URL
+    req = requests.get(url, headers=headers)
+    binary_path = str(Path.home()) + '/' + dir_name
+
+    # Directory
+    if os.path.exists(binary_path) is False:
+        os.chdir(str(Path.home()))
+        os.mkdir(dir_name)
+        print(f'Directory {dir_name} was created inside of your HOME directory')
+
+    if req.ok:
+        # Split URL to get the file name 'draftv2-darwin-amd64'
+        os.chdir(binary_path)
+        # Writing the file to the local file system
+        with open(filename, 'wb') as output_file:
+            output_file.write(req.content)
+        print('Download of DraftV2 binary was successful with a status code: ' + str(req.status_code))
+        os.chmod(binary_path + '/' + filename, 0o755)
+        return binary_path + '/' + filename
+
+    print('Download of DraftV2 binary was unsuccessful with a status code: ' + str(req.status_code))
+    return None

src/aks-preview/azext_aks_preview/commands.py

@@ -110,6 +110,11 @@ def load_command_table(self, _):
         g.custom_command('stop', 'aks_agentpool_stop', supports_no_wait=True)
         g.custom_command('start', 'aks_agentpool_start', supports_no_wait=True)
 
+    # AKS App commands
+    with self.command_group('aks app', managed_clusters_sdk, client_factory=cf_managed_clusters) as g:
+        g.custom_command('init', 'aks_app_init')
+        # g.custom_command('up', 'aks_app_up')
+
     # AKS pod identity commands
     with self.command_group('aks pod-identity', managed_clusters_sdk, client_factory=cf_managed_clusters) as g:
         g.custom_command('add', 'aks_pod_identity_add')

src/aks-preview/azext_aks_preview/custom.py

@@ -147,6 +147,10 @@
     UserAssignedIdentity,
 )
 
+from .aks_app_commands.aksappcommands import (
+    aks_draft_app_init
+)
+
 logger = get_logger(__name__)
 
 
@@ -2771,7 +2775,7 @@ def _get_linux_os_config(file_path):
 
 def _get_http_proxy_config(file_path):
     if not os.path.isfile(file_path):
-        raise CLIError("{} is not valid file, or not accessable.".format(file_path))
+        raise CLIError("{} is not valid file, or is not accessible.".format(file_path))
     hp_config = get_file_json(file_path)
     if not isinstance(hp_config, dict):
         raise CLIError(
@@ -2785,13 +2789,21 @@ def _get_http_proxy_config(file_path):
     return config_object
 
 
+def aks_app_init(path,
+                 app_name=None,
+                 language=None,
+                 create_config=None,
+                 dockerfile_only=None,
+                 deployment_only=None):
+    aks_draft_app_init(path, app_name, language, create_config, dockerfile_only, deployment_only)
+
+
 def aks_pod_identity_add(cmd, client, resource_group_name, cluster_name,
                          identity_name, identity_namespace, identity_resource_id,
                          binding_selector=None,
                          no_wait=False):  # pylint: disable=unused-argument
     instance = client.get(resource_group_name, cluster_name)
     _ensure_pod_identity_addon_is_enabled(instance)
-
     user_assigned_identity = _get_user_assigned_identity(
         cmd.cli_ctx, identity_resource_id)
     _ensure_managed_identity_operator_permission(