Enhance: Enable O2 DMS by exposing k8s API endpoint

[pti/o2.git] / o2ims / service / auditor / dms_handler.py

diff --git a/o2ims/service/auditor/dms_handler.py b/o2ims/service/auditor/dms_handler.py
index 9ef0a77..89811c6 100644 (file)
--- a/o2ims/service/auditor/dms_handler.py
+++ b/o2ims/service/auditor/dms_handler.py
@@ -15,6 +15,9 @@
 # pylint: disable=unused-argument
 from __future__ import annotations
 
+import base64
+import json
+
 from o2ims.domain.stx_object import StxGenericModel
 # from dataclasses import asdict
 # from typing import List, Dict, Callable, Type
@@ -43,7 +46,7 @@ def update_dms(
     with uow:
         dms = uow.deployment_managers.get(stxobj.id)
         if not dms:
-            logger.info("add dms:" + stxobj.name
+            logger.info("add dms: " + stxobj.name
                         + " update_at: " + str(stxobj.updatetime)
                         + " id: " + str(stxobj.id)
                         + " hash: " + str(stxobj.hash))
@@ -56,7 +59,7 @@ def update_dms(
         else:
             localmodel = dms
             if is_outdated(localmodel, stxobj):
-                logger.info("update a dms:" + stxobj.name
+                logger.info("update a dms: " + stxobj.name
                             + " update_at: " + str(stxobj.updatetime)
                             + " id: " + str(stxobj.id)
                             + " hash: " + str(stxobj.hash))
@@ -83,9 +86,10 @@ def create_by(stxobj: StxGenericModel, parentid: str) -> DeploymentManager:
     supportedLocations = ''
     capabilities = ''
     capacity = ''
+    profile = _convert_content(stxobj.content)
     localmodel = DeploymentManager(
         stxobj.id, stxobj.name, ocloudid, dmsendpoint, description,
-        supportedLocations, capabilities, capacity)
+        supportedLocations, capabilities, capacity, profile)
     localmodel.createtime = stxobj.createtime
     localmodel.updatetime = stxobj.updatetime
     localmodel.hash = stxobj.hash
@@ -100,8 +104,38 @@ def update_by(target: DeploymentManager, stxobj: StxGenericModel,
     target.name = stxobj.name
     target.createtime = stxobj.createtime
     target.updatetime = stxobj.updatetime
-    # ocloud.content = stxobj.content
     target.hash = stxobj.hash
     target.oCloudId = parentid
     target.version_number = target.version_number + 1
+    target.profile = _convert_content(stxobj.content)
     target.events = []
+
+
+def _convert_content(stxobj_content: str):
+    # Convert api retrun content to dict object
+    content = json.loads(stxobj_content)
+    admin_user = content["admin_user"]
+    cluster_api_endpoint = content["cluster_api_endpoint"]
+    cluster_ca_cert = _b64_encode_str(content["cluster_ca_cert"])
+    admin_client_cert = _b64_encode_str(content["admin_client_cert"])
+    admin_client_key = _b64_encode_str(content["admin_client_key"])
+    # admin_client_cert = base64.b64encode(
+    #     bytes(content["admin_client_cert"], "utf-8"))
+    # admin_client_key = base64.b64encode(
+    #     bytes(content["admin_client_key"], "utf-8"))
+    profile = {
+        "admin_user": admin_user,
+        "cluster_api_endpoint": cluster_api_endpoint,
+        "cluster_ca_cert": cluster_ca_cert,
+        "admin_client_cert": admin_client_cert,
+        "admin_client_key": admin_client_key
+    }
+
+    return json.dumps(profile)
+
+
+def _b64_encode_str(msg: str, encode: str = 'utf-8') -> str:
+    msg_bytes = msg.encode('utf-8')
+    base64_bytes = base64.b64encode(msg_bytes)
+    base64_msg = base64_bytes.decode('utf-8')
+    return base64_msg