# Copyright (c) 2014 Red Hat, Inc.
# All Rights Reserved.
#
#    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 oslo_serialization import jsonutils

from nova.scheduler import rpcapi as scheduler_rpcapi


class SchedulerQueryClient(object):
    """Client class for querying to the scheduler."""

    def __init__(self):
        self.scheduler_rpcapi = scheduler_rpcapi.SchedulerAPI()

    def select_destinations(self, context, spec_obj):
        """Returns destinations(s) best suited for this request_spec and
        filter_properties.

        The result should be a list of dicts with 'host', 'nodename' and
        'limits' as keys.
        """
        # TODO(sbauza): Provide directly the RequestSpec object as an arg
        # once the RPC API is modified for that.
        request_spec = spec_obj.to_legacy_request_spec_dict()
        filter_properties = spec_obj.to_legacy_filter_properties_dict()
        # FIXME(sbauza): Serialize/Unserialize the legacy dict because of
        # oslo.messaging #1529084 to transform datetime values into strings.
        # tl;dr: datetimes in dicts are not accepted as correct values by the
        # rpc fake driver.
        # will be removed in the next patch of that series.
        # Yeah, that's an ugly hack I know, but that's only for not squashing
        # both commits.
        request_spec = jsonutils.loads(jsonutils.dumps(request_spec))
        return self.scheduler_rpcapi.select_destinations(context, request_spec,
                                                         filter_properties)

    def update_aggregates(self, context, aggregates):
        """Updates HostManager internal aggregates information.

        :param aggregates: Aggregate(s) to update
        :type aggregates: :class:`nova.objects.Aggregate`
                          or :class:`nova.objects.AggregateList`
        """
        self.scheduler_rpcapi.update_aggregates(context, aggregates)

    def delete_aggregate(self, context, aggregate):
        """Deletes HostManager internal information about a specific aggregate.

        :param aggregate: Aggregate to delete
        :type aggregate: :class:`nova.objects.Aggregate`
        """
        self.scheduler_rpcapi.delete_aggregate(context, aggregate)

    def update_instance_info(self, context, host_name, instance_info):
        """Updates the HostManager with the current information about the
        instances on a host.

        :param context: local context
        :param host_name: name of host sending the update
        :param instance_info: an InstanceList object.
        """
        self.scheduler_rpcapi.update_instance_info(context, host_name,
                                                   instance_info)

    def delete_instance_info(self, context, host_name, instance_uuid):
        """Updates the HostManager with the current information about an
        instance that has been deleted on a host.

        :param context: local context
        :param host_name: name of host sending the update
        :param instance_uuid: the uuid of the deleted instance
        """
        self.scheduler_rpcapi.delete_instance_info(context, host_name,
                                                   instance_uuid)

    def sync_instance_info(self, context, host_name, instance_uuids):
        """Notifies the HostManager of the current instances on a host by
        sending a list of the uuids for those instances. The HostManager can
        then compare that with its in-memory view of the instances to detect
        when they are out of sync.

        :param context: local context
        :param host_name: name of host sending the update
        :param instance_uuids: a list of UUID strings representing the current
                               instances on the specified host
        """
        self.scheduler_rpcapi.sync_instance_info(context, host_name,
                                                 instance_uuids)