# Copyright (c) 2013 OpenStack Foundation
# 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.

"""
service interface
"""

from cinderclient import api_versions
from cinderclient import base


class Service(base.Resource):

    def __repr__(self):
        return "<Service: binary=%s host=%s>" % (self.binary, self.host)


class LogLevel(base.Resource):
    def __repr__(self):
        return '<LogLevel: binary=%s host=%s prefix=%s level=%s>' % (
            self.binary, self.host, self.prefix, self.level)


class ServiceManagerBase(base.ManagerWithFind):
    resource_class = Service

    def list(self, host=None, binary=None):
        """
        Describes service list for host.

        :param host: destination host name.
        :param binary: service binary.
        """
        url = "/os-services"
        filters = []
        if host:
            filters.append("host=%s" % host)
        if binary:
            filters.append("binary=%s" % binary)
        if filters:
            url = "%s?%s" % (url, "&".join(filters))
        return self._list(url, "services")

    def enable(self, host, binary):
        """Enable the service specified by hostname and binary."""
        body = {"host": host, "binary": binary}
        result = self._update("/os-services/enable", body)
        return self.resource_class(self, result, resp=result.request_ids)

    def disable(self, host, binary):
        """Disable the service specified by hostname and binary."""
        body = {"host": host, "binary": binary}
        result = self._update("/os-services/disable", body)
        return self.resource_class(self, result, resp=result.request_ids)

    def disable_log_reason(self, host, binary, reason):
        """Disable the service with reason."""
        body = {"host": host, "binary": binary, "disabled_reason": reason}
        result = self._update("/os-services/disable-log-reason", body)
        return self.resource_class(self, result, resp=result.request_ids)

    def freeze_host(self, host):
        """Freeze the service specified by hostname."""
        body = {"host": host}
        return self._update("/os-services/freeze", body)

    def thaw_host(self, host):
        """Thaw the service specified by hostname."""
        body = {"host": host}
        return self._update("/os-services/thaw", body)

    def failover_host(self, host, backend_id):
        """Failover a replicated backend by hostname."""
        body = {"host": host, "backend_id": backend_id}
        return self._update("/os-services/failover_host", body)


class ServiceManager(ServiceManagerBase):
    @api_versions.wraps("3.0")
    def server_api_version(self):
        """Returns the API Version supported by the server.

        :return: Returns response obj for a server that supports microversions.
                 Returns an empty list for Liberty and prior Cinder servers.
        """

        try:
            return self._get_with_base_url("", response_key='versions')
        except LookupError:
            return []

    @api_versions.wraps("3.32")
    def set_log_levels(self, level, binary, server, prefix):
        """Set log level for services."""
        body = {'level': level, 'binary': binary, 'server': server,
                'prefix': prefix}
        return self._update("/os-services/set-log", body)

    @api_versions.wraps("3.32")
    def get_log_levels(self, binary, server, prefix):
        """Get log levels for services."""
        body = {'binary': binary, 'server': server, 'prefix': prefix}
        response = self._update("/os-services/get-log", body)

        log_levels = []
        for entry in response['log_levels']:
            entry_levels = sorted(entry['levels'].items(), key=lambda x: x[0])
            for prefix, level in entry_levels:
                log_dict = {'binary': entry['binary'], 'host': entry['host'],
                            'prefix': prefix, 'level': level}
                log_levels.append(LogLevel(self, log_dict, loaded=True))
        return log_levels