#
# 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 Agent RPC API."""

import oslo_messaging as messaging

from oslo_config import cfg
from oslo_log import log

from cyborg.common import constants
from cyborg.common import rpc
from cyborg.objects import base as objects_base


LOG = log.getLogger(__name__)


CONF = cfg.CONF


class AgentAPI:
    """Client side of the Agent RPC API.

    API version history:

    |    1.0 - Initial version.

    """

    RPC_API_VERSION = '1.0'

    def __init__(self, topic=None):
        super().__init__()
        self.topic = topic or constants.AGENT_TOPIC
        target = messaging.Target(topic=self.topic, version='1.0')
        serializer = objects_base.CyborgObjectSerializer()
        self.client = rpc.get_client(
            target, version_cap=self.RPC_API_VERSION, serializer=serializer
        )

    def fpga_program(
        self, context, hostname, controlpath_id, bitstream_uuid, driver_name
    ):
        LOG.info(
            'Agent fpga_program: hostname: (%s) bitstream_id: (%s)',
            hostname,
            bitstream_uuid,
        )
        version = '1.0'
        cctxt = self.client.prepare(server=hostname, version=version)
        return cctxt.call(
            context,
            'fpga_program',
            controlpath_id=controlpath_id,
            bitstream_uuid=bitstream_uuid,
            driver_name=driver_name,
        )

    def create_vgpu_mdev(
        self, context, hostname, pci_addr, asked_type, ah_uuid
    ):
        LOG.debug(
            'Agent create_vgpu_mdev: hostname: (%s) , pci_address: (%s)'
            'gpu_id: (%s)',
            hostname,
            pci_addr,
            ah_uuid,
        )
        version = '1.0'
        cctxt = self.client.prepare(server=hostname, version=version)
        return cctxt.call(
            context,
            'create_vgpu_mdev',
            pci_addr=pci_addr,
            asked_type=asked_type,
            ah_uuid=ah_uuid,
        )

    def remove_vgpu_mdev(
        self, context, hostname, pci_addr, asked_type, ah_uuid
    ):
        LOG.debug(
            'Agent remove_vgpu_mdev: hostname: (%s) gpu_id: (%s)',
            hostname,
            ah_uuid,
        )
        version = '1.0'
        cctxt = self.client.prepare(server=hostname, version=version)
        return cctxt.call(
            context,
            'remove_vgpu_mdev',
            pci_addr=pci_addr,
            asked_type=asked_type,
            ah_uuid=ah_uuid,
        )
