# 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

from o2ims.adapter.orm import ocloud, resource, resourcetype, \
    resourcepool, deploymentmanager, subscription
from o2ims.adapter import unit_of_work
# from o2ims.domain.ocloud import Ocloud


def oclouds(uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "oCloudId", "name" FROM ocloud
        #     """,
        # )

        res = uow.session.execute(select(ocloud))
    return [dict(r) for r in res]


def ocloud_one(ocloudid: str, uow: unit_of_work.AbstractUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "oCloudId", "name" FROM ocloud
        #     WHERE "oCloudId" = :oCloudId
        #     """,
        #     dict(oCloudId=ocloudid),
        # )
        res = uow.session.execute(
            select(ocloud).where(ocloud.c.oCloudId == ocloudid))
        first = res.first()
    return None if first is None else dict(first)


def resource_types(uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "resourceTypeId", "oCloudId", "name" FROM resourcetype
        #     """,
        # )
        res = uow.session.execute(select(resourcetype))
    return [dict(r) for r in res]


def resource_type_one(resourceTypeId: str,
                      uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "resourceTypeId", "oCloudId", "name"
        #     FROM resourcetype WHERE "resourceTypeId" = :resourceTypeId
        #     """,
        #     dict(resourceTypeId=resourceTypeId),
        # )
        res = uow.session.execute(select(resourcetype).where(
            resourcetype.c.resourceTypeId == resourceTypeId))
        first = res.first()
    return None if first is None else dict(first)


def resource_pools(uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "resourcePoolId", "oCloudId", "location", "name"
        #     FROM resourcepool
        #     """,
        # )
        res = uow.session.execute(select(resourcepool))
    return [dict(r) for r in res]


def resource_pool_one(resourcePoolId: str,
                      uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "resourcePoolId", "oCloudId", "location", "name"
        #     FROM resourcepool
        #     WHERE "resourcePoolId" = :resourcePoolId
        #     """,
        #     dict(resourcePoolId=resourcePoolId),
        # )
        res = uow.session.execute(select(resourcepool).where(
            resourcepool.c.resourcePoolId == resourcePoolId))
        first = res.first()
    return None if first is None else dict(first)


def resources(resourcePoolId: str, uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "resourceId", "parentId", "resourceTypeId",
        #           "resourcePoolId", "oCloudId"
        #     FROM resource
        #     WHERE "resourcePoolId" = :resourcePoolId
        #     """,
        #     dict(resourcePoolId=resourcePoolId),
        # )
        res = uow.session.execute(select(resource).where(
            resource.c.resourcePoolId == resourcePoolId))
    return [dict(r) for r in res]


def resource_one(resourceId: str, uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "resourceId", "parentId", "resourceTypeId",
        #           "resourcePoolId", "oCloudId"
        #     FROM resource
        #     WHERE "resourceId" = :resourceId
        #     """,
        #     # AND "resourcePoolId" = :resourcePoolId
        #     # dict(resourcePoolId=resourcePoolId,
        #     dict(resourceId=resourceId),
        # )
        res = uow.session.execute(select(resource).where(
            resource.c.resourceId == resourceId))
        first = res.first()
    return None if first is None else dict(first)


def deployment_managers(uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "deploymentManagerId", "oCloudId",
        #           "deploymentManagementServiceEndpoint", "name"
        #     FROM deploymentmanager
        #     """,
        # )
        res = uow.session.execute(select(deploymentmanager))
    return [dict(r) for r in res]


def deployment_manager_one(deploymentManagerId: str,
                           uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        # res = uow.session.execute(
        #     """
        #     SELECT "deploymentManagerId", "oCloudId",
        #           "deploymentManagementServiceEndpoint", "name"
        #     FROM deploymentmanager
        #     WHERE "deploymentManagerId" = :deploymentManagerId
        #     """,
        #     dict(deploymentManagerId=deploymentManagerId),
        # )
        res = uow.session.execute(select(deploymentmanager).where(
            deploymentmanager.c.deploymentManagerId == deploymentManagerId))
        first = res.first()
    return None if first is None else dict(first)


def subscriptions(uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        res = uow.session.execute(select(subscription))
    return [dict(r) for r in res]


def subscription_one(subscriptionId: str,
                     uow: unit_of_work.SqlAlchemyUnitOfWork):
    with uow:
        res = uow.session.execute(select(subscription).where(
            subscription.c.subscriptionId == subscriptionId))
        first = res.first()
    return None if first is None else dict(first)