feat(helmbroker): add provision

Author: duanhongyi

.gitignore

@@ -33,6 +33,7 @@ htmlcov/
 
 # python virtual environments for testing
 venv/
+.venv/
 
 # vendored go source code
 vendor/

rootfs/helmbroker/broker.py

@@ -1,15 +1,18 @@
+import os
+import shutil
 from typing import Union, List
-
-import openbrokerapi
 from openbrokerapi import api
 from openbrokerapi.api import ServiceBroker
 from openbrokerapi.catalog import ServicePlan
 from openbrokerapi.service_broker import *
 
+from .utils import get_instance_path, get_chart_path, get_plan_path, get_addon_path
+from .tasks import provision
 
 class HelmServiceBroker(ServiceBroker):
+
     def catalog(self) -> Union[Service, List[Service]]:
-        return Service(
+        return [Service(
             id='server',
             name='server',
             description='service description',
@@ -27,16 +30,27 @@ def catalog(self) -> Union[Service, List[Service]]:
                 )
             ],
             plan_updateable=True,
-        )
+        )]
 
     def provision(self,
                   instance_id: str,
                   details: ProvisionDetails,
                   async_allowed: bool,
                   **kwargs) -> ProvisionedServiceSpec:
-        # Create service instance
-        # ...
-        return ProvisionedServiceSpec()
+        instance_path = get_instance_path(instance_id)
+        if os.path.exists(instance_path):
+            return ProvisionedServiceSpec(
+                state=ProvisionState.IDENTICAL_ALREADY_EXISTS
+            )
+        os.makedirs(instance_path, exist_ok=True)
+        chart_path, plan_path =get_chart_path(instance_id), get_plan_path(instance_id)
+        addon_chart_path, addon_plan_path = get_addon_path(details.service_id, details.plan_id)
+        shutil.copy(addon_chart_path, chart_path)
+        shutil.copy(addon_plan_path, plan_path)
+        if async_allowed:
+            provision.delay(instance_id, details. async_allowed, **kwargs)
+            return ProvisionedServiceSpec(state=ProvisionState.IS_ASYNC)
+        return provision(instance_id, details. async_allowed, **kwargs)
 
     def get_binding(self,
                     instance_id: str,

rootfs/helmbroker/config.py

@@ -1,10 +1,12 @@
+import os
 import yaml
 
-HELMBROKER_ROOT = '/etc/helmbroker'
-ADDONS_PATH = HELMBROKER_ROOT + '/addons'
-CONFIG_PATH = HELMBROKER_ROOT + '/config'
+HELMBROKER_ROOT = os.environ.get("HELMBROKER_CELERY_BROKER", '/etc/helmbroker')
 
+ADDONS_PATH = os.path.join(HELMBROKER_ROOT, 'addons')
+CONFIG_PATH = os.path.join(HELMBROKER_ROOT, 'config')
+INSTANCES_PATH = os.path.join(HELMBROKER_ROOT, 'instances')
 
 class Config:
     with open(CONFIG_PATH, 'r') as f:
-        repositorie = yaml.load(f.read())
+        repository = yaml.load(f.read())

rootfs/helmbroker/tasks.py

@@ -1,6 +1,37 @@
+import os
+import shutil
 from .celery import app
+from .utils import command, get_plan_path, get_chart_path, get_or_create_instance_meta
+from openbrokerapi.service_broker import *
 
 
-@app.task
-def mul(x, y):
-    return x * y
+def provision(iself,
+                  instance_id: str,
+                  details: ProvisionDetails,
+                  async_allowed: bool,
+                  **kwargs) -> ProvisionedServiceSpec:
+    chart_path = get_chart_path(instance_id)
+    values_file =  os.path.join(get_plan_path(instance_id), "values.yaml")
+    args = [ 
+        "install",
+        details.context["instance_name"],
+        chart_path,
+        "--namespace",
+        details.context["namespace"],
+        "--create-namespace",
+        "--output",
+        "--wait",
+        "--timeout 30m0s"
+        "-f",
+        values_file
+    ]
+    status, output = command("helm", *args)
+    if status != 0:
+        return ProvisionedServiceSpec(state="status error: %s" % status, operation=output)
+    else:
+        config = get_or_create_instance_meta(instance_id)
+        return ProvisionedServiceSpec(
+            dashboard_url=config.get("dashboard_url", None),
+            state=ProvisionState.SUCCESSFUL_CREATED,
+            operation=config.get("operation", None),
+        )

rootfs/helmbroker/utils.py

@@ -1,12 +1,27 @@
+import os
 import yaml
 import json
 import subprocess
 
+from .config import INSTANCES_PATH
+
 
 def command(cmd, *args, output_type="text"):
-    output = subprocess.getoutput("%s %s" % (cmd, " ".join(args)))
+    status, output = subprocess.getstatusoutput("%s %s" % (cmd, " ".join(args)))
     if output_type == "yaml":
         return yaml.load(output, Loader=yaml.Loader)
     elif output_type == "json":
         return json.loads(output)
-    return output
+    return status, output
+
+
+get_instance_path = lambda instance_id: os.path.join(INSTANCES_PATH, instance_id)
+get_chart_path = lambda instance_id: os.path.join(get_instance_path(instance_id), "chart")
+get_plan_path = lambda instance_id: os.path.join(get_instance_path(instance_id), "plan")
+
+def get_addon_path(service_id, plan_id):
+    return ""
+
+
+def get_or_create_instance_meta(instance_id):
+    pass