# Copyright (c) 2017-2024 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.

"""
Client side of the DC Orchestrator RPC API.
"""

from dcorch.common import consts
from dcorch.common import messaging


class EngineClient(object):
    """Client side of the DC orchestrator engine rpc API.

    Version History:
     1.0 - Initial version
    """

    BASE_RPC_API_VERSION = "1.0"

    def __init__(self):
        self._client = messaging.get_rpc_client(
            topic=consts.TOPIC_ORCH_ENGINE, version=self.BASE_RPC_API_VERSION
        )

    @staticmethod
    def make_msg(method, **kwargs):
        return method, kwargs

    def call(self, ctxt, msg, version=None):
        method, kwargs = msg
        if version is not None:
            client = self._client.prepare(version=version)
        else:
            client = self._client
        return client.call(ctxt, method, **kwargs)

    def cast(self, ctxt, msg, fanout=None, version=None):
        method, kwargs = msg
        if version or fanout:
            client = self._client.prepare(fanout=fanout, version=version)
        else:
            client = self._client
        return client.cast(ctxt, method, **kwargs)

    def get_usage_for_project_and_user(
        self, ctxt, endpoint_type, project_id, user_id=None
    ):
        return self.call(
            ctxt,
            self.make_msg(
                "get_usage_for_project_and_user",
                endpoint_type=endpoint_type,
                project_id=project_id,
                user_id=user_id,
            ),
        )

    def quota_sync_for_project(self, ctxt, project_id, user_id):
        return self.cast(
            ctxt,
            self.make_msg(
                "quota_sync_for_project", project_id=project_id, user_id=user_id
            ),
        )


class EngineWorkerClient(object):
    """Client side of the DC orchestrator engine worker rpc API.

    Version History:
     1.0 - Initial version
    """

    BASE_RPC_API_VERSION = "1.0"

    def __init__(self):
        self._client = messaging.get_rpc_client(
            topic=consts.TOPIC_ORCH_ENGINE_WORKER, version=self.BASE_RPC_API_VERSION
        )

    @staticmethod
    def make_msg(method, **kwargs):
        return method, kwargs

    def call(self, ctxt, msg, version=None):
        method, kwargs = msg
        if version is not None:
            client = self._client.prepare(version=version)
        else:
            client = self._client
        return client.call(ctxt, method, **kwargs)

    def cast(self, ctxt, msg, fanout=None, version=None):
        method, kwargs = msg
        if version or fanout:
            client = self._client.prepare(fanout=fanout, version=version)
        else:
            client = self._client
        return client.cast(ctxt, method, **kwargs)

    def keypair_sync_for_user(self, ctxt, job_id, payload):
        return self.cast(
            ctxt, self.make_msg("keypair_sync_for_user", job_id=job_id, payload=payload)
        )

    def image_sync(self, ctxt, job_id, payload):
        return self.cast(
            ctxt, self.make_msg("image_sync", job_id=job_id, payload=payload)
        )

    def add_subcloud(self, ctxt, subcloud_name, sw_version, management_ip):
        return self.call(
            ctxt,
            self.make_msg(
                "add_subcloud",
                subcloud_name=subcloud_name,
                sw_version=sw_version,
                management_ip=management_ip,
            ),
        )

    def del_subcloud(self, ctxt, subcloud_name):
        return self.call(
            ctxt, self.make_msg("del_subcloud", subcloud_name=subcloud_name)
        )

    def update_subcloud_states(
        self, ctxt, subcloud_name, management_state, availability_status
    ):
        return self.call(
            ctxt,
            self.make_msg(
                "update_subcloud_states",
                subcloud_name=subcloud_name,
                management_state=management_state,
                availability_status=availability_status,
            ),
        )

    def add_subcloud_sync_endpoint_type(self, ctxt, subcloud_name, endpoint_type_list):
        return self.cast(
            ctxt,
            self.make_msg(
                "add_subcloud_sync_endpoint_type",
                subcloud_name=subcloud_name,
                endpoint_type_list=endpoint_type_list,
            ),
        )

    def remove_subcloud_sync_endpoint_type(
        self, ctxt, subcloud_name, endpoint_type_list
    ):
        return self.cast(
            ctxt,
            self.make_msg(
                "remove_subcloud_sync_endpoint_type",
                subcloud_name=subcloud_name,
                endpoint_type_list=endpoint_type_list,
            ),
        )

    def sync_subclouds(self, ctxt, subcloud_sync_list):
        return self.cast(
            ctxt, self.make_msg("sync_subclouds", subcloud_sync_list=subcloud_sync_list)
        )

    def run_sync_audit(self, ctxt, subcloud_sync_list):
        return self.cast(
            ctxt, self.make_msg("run_sync_audit", subcloud_sync_list=subcloud_sync_list)
        )

    def initial_sync_subclouds(self, ctxt, subcloud_capabilities):
        return self.cast(
            ctxt,
            self.make_msg(
                "initial_sync_subclouds", subcloud_capabilities=subcloud_capabilities
            ),
        )

    def update_subcloud_version(self, ctxt, subcloud_name, sw_version):
        return self.call(
            ctxt,
            self.make_msg(
                "update_subcloud_version",
                subcloud_name=subcloud_name,
                sw_version=sw_version,
            ),
        )

    def update_subcloud_management_ip(self, ctxt, subcloud_name, management_ip):
        return self.call(
            ctxt,
            self.make_msg(
                "update_subcloud_management_ip",
                subcloud_name=subcloud_name,
                management_ip=management_ip,
            ),
        )

    # The sync job info has been written to the DB, alert the sync engine
    # that there is work to do.
    def sync_request(self, ctxt, endpoint_type):
        return self.cast(
            ctxt, self.make_msg("sync_request", endpoint_type=endpoint_type)
        )