# 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.

from sqlalchemy import select
import uuid
from o2dms.domain.states import NfDeploymentState
from o2common.service import messagebus
from o2common.service import unit_of_work
from o2dms.adapter.orm import nfDeployment
from o2dms.api.dms_dto import DmsLcmNfDeploymentDTO
from o2dms.domain.dms import NfDeployment
from o2common.helper import o2logging
logger = o2logging.get_logger(__name__)


def lcm_nfdeployment_list(
        deploymentManagerID: str, uow: unit_of_work.AbstractUnitOfWork):
    with uow:
        res = uow.session.execute(select(nfDeployment).where(
            nfDeployment.c.deploymentManagerId == deploymentManagerID))
    return [dict(r) for r in res]


def lcm_nfdeployment_one(
        nfdeploymentid: str, uow: unit_of_work.AbstractUnitOfWork):
    with uow:
        res = uow.session.execute(select(nfDeployment).where(
            nfDeployment.c.id == nfdeploymentid))
        first = res.first()
    return None if first is None else dict(first)


def lcm_nfdeployment_create(
        deploymentManagerId: str,
        input: DmsLcmNfDeploymentDTO.
        NfDeployment_create,
        bus: messagebus.MessageBus):

    uow = bus.uow
    id = str(uuid.uuid4())
    with uow:
        _check_duplication(input, uow)
        _check_dependencies(input, uow)
        entity = NfDeployment(
            id, input['name'], deploymentManagerId, input['description'],
            input['descriptorId'], input['parentDeploymentId'])
        uow.nfdeployments.add(entity)
        # entity.transit_state(NfDeploymentState.NotInstalled)

        # to be refactor later according to O2 DMS API design
        entity.transit_state(NfDeploymentState.Installing)
        uow.commit()
    _handle_events(bus)

    return id


def lcm_nfdeployment_update(
        nfdeploymentid: str,
        input: DmsLcmNfDeploymentDTO.NfDeployment_update,
        bus: messagebus.MessageBus):

    uow = bus.uow
    with uow:
        entity: NfDeployment = uow.nfdeployments.get(nfdeploymentid)
        entity.name = input['name']
        entity.description = input['description']
        entity.outputParams = input['parentDeploymentId']
        entity.transit_state(NfDeploymentState.Updating)
        uow.commit()
    _handle_events(bus)
    return True


def _handle_events(bus: messagebus.MessageBus):
    # handle events
    events = bus.uow.collect_new_events()
    for event in events:
        bus.handle(event)
    return True


def lcm_nfdeployment_uninstall(
        nfdeploymentid: str,
        bus: messagebus.MessageBus):

    uow = bus.uow
    with uow:
        entity: NfDeployment = uow.nfdeployments.get(nfdeploymentid)
        if not entity:
            pass
        elif entity.status == NfDeploymentState.Initial:
            bus.uow.nfdeployments.delete(nfdeploymentid)
        # elif entity.status == NfDeploymentState.NotInstalled:
        #     bus.uow.nfdeployments.delete(nfdeploymentid)
        elif entity.status == NfDeploymentState.Installing:
            entity.transit_state(NfDeploymentState.Uninstalling)
        elif entity.status == NfDeploymentState.Installed:
            entity.transit_state(NfDeploymentState.Uninstalling)
        elif entity.status == NfDeploymentState.Updating:
            entity.transit_state(NfDeploymentState.Uninstalling)
        elif entity.status == NfDeploymentState.Uninstalling:
            pass
        elif entity.status == NfDeploymentState.Abnormal:
            bus.uow.nfdeployments.delete(nfdeploymentid)
        else:
            entity.transit_state(NfDeploymentState.Abnormal)
        uow.commit()
    _handle_events(bus)
    return True


# def lcm_nfdeployment_delete(
#         nfdeploymentid: str,
#         bus: messagebus.MessageBus):

#     uow = bus.uow
#     with uow:
#         entity = uow.nfdeployments.get(nfdeploymentid)
#         if entity.status != NfDeploymentState.Initial:
#             raise Exception(
#                 "NfDeployment {} is not in status to delete".format(
#                     nfdeploymentid))
#         uow.nfdeployments.delete(nfdeploymentid)
#         entity.transit_state(NfDeploymentState.Deleting)
#         uow.commit()
#     return True


def _check_duplication(
        input: DmsLcmNfDeploymentDTO,
        uow: unit_of_work.AbstractUnitOfWork):
    name = input['name']
    descriptorId = input['descriptorId']
    if uow.nfdeployments.count(name=name) > 0:
        raise Exception(
            "NfDeployment with name {} exists already".format(name))
    if uow.nfdeployments.count(descriptorId=descriptorId) > 0:
        raise Exception(
            "NfDeployment with descriptorId {} exists already".format(
                descriptorId))


def _check_dependencies(
        input: DmsLcmNfDeploymentDTO,
        uow: unit_of_work.AbstractUnitOfWork):
    descriptorId = input['descriptorId']
    if uow.nfdeployment_descs.count(id=descriptorId) == 0:
        raise Exception(
            "NfDeploymentDescriptor with id {} does not exist".format(
                descriptorId))