# Copyright (C) 2021 Wind River Systems, Inc.
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

# pylint: disable=unused-argument
from __future__ import annotations
import base64
import json

from o2ims.domain import commands, events
from o2ims.domain.stx_object import StxGenericModel
from o2ims.domain.subscription_obj import NotificationEventEnum
from o2common.service.unit_of_work import AbstractUnitOfWork
from o2ims.domain.resource_type import MismatchedModel
from o2ims.domain.ocloud import DeploymentManager
# if TYPE_CHECKING:
#     from . import unit_of_work

from o2common.helper import o2logging
logger = o2logging.get_logger(__name__)


class InvalidResourceType(Exception):
    pass


def update_dms(
    cmd: commands.UpdateDms,
    uow: AbstractUnitOfWork
):
    stxobj = cmd.data
    with uow:
        dms = uow.deployment_managers.get(stxobj.id)
        if not dms:
            logger.info("add dms: " + stxobj.name
                        + " update_at: " + str(stxobj.updatetime)
                        + " id: " + str(stxobj.id)
                        + " hash: " + str(stxobj.hash))
            # ocloud = uow.oclouds.get(cmd.parent.oCloudId)
            localmodel = create_by(stxobj, cmd.parentid)
            uow.deployment_managers.add(localmodel)

            logger.info("Add a dms: " + stxobj.id
                        + ", name: " + stxobj.name)
        else:
            localmodel = dms
            if is_outdated(localmodel, stxobj):
                logger.info("update a dms: " + stxobj.name
                            + " update_at: " + str(stxobj.updatetime)
                            + " id: " + str(stxobj.id)
                            + " hash: " + str(stxobj.hash))
                update_by(localmodel, stxobj, cmd.parentid)
                uow.deployment_managers.update(localmodel)

            logger.info("Update a dms: " + stxobj.id
                        + ", name: " + stxobj.name)
        uow.commit()


def is_outdated(ocloud: DeploymentManager, stxobj: StxGenericModel):
    # if stxobj.updatetime:
    #     return True if Ocloud.updatetime < stxobj.updatetime else False
    # else:
    return True if ocloud.hash != stxobj.hash else False


def create_by(stxobj: StxGenericModel, parentid: str) -> DeploymentManager:
    description = "A DMS"
    ocloudid = parentid
    content = json.loads(stxobj.content)
    # logger.info(stxobj)
    # logger.info(content)
    supportedLocations = ''
    capabilities = content['capabilities']
    capacity = ''
    dmsendpoint = content['cluster_api_endpoint']
    profile = _convert_content(content)
    localmodel = DeploymentManager(
        stxobj.id, stxobj.name, ocloudid, dmsendpoint, description,
        supportedLocations, capabilities, capacity, profile)
    localmodel.createtime = stxobj.createtime
    localmodel.updatetime = stxobj.updatetime
    localmodel.hash = stxobj.hash

    localmodel.events.append(events.DmsChanged(
        id=stxobj.id,
        notificationEventType=NotificationEventEnum.CREATE,
        updatetime=stxobj.updatetime
    ))

    return localmodel


def update_by(target: DeploymentManager, stxobj: StxGenericModel,
              parentid: str) -> None:
    if target.deploymentManagerId != stxobj.id:
        raise MismatchedModel("Mismatched Id")
    content = json.loads(stxobj.content)
    logger.info(content)
    target.name = stxobj.name
    target.createtime = stxobj.createtime
    target.updatetime = stxobj.updatetime
    target.hash = stxobj.hash
    target.oCloudId = parentid
    target.capabilities = content['capabilities']
    target.version_number = target.version_number + 1
    target.profile = _convert_content(content)

    target.events.append(events.DmsChanged(
        id=stxobj.id,
        notificationEventType=NotificationEventEnum.MODIFY,
        updatetime=stxobj.updatetime
    ))


def _convert_content(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