# Copyright 2013, 2016 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.

import json
import re

from config_tempest import constants as C
from config_tempest.services.base import ServiceError
from config_tempest.services.base import VersionedService
from config_tempest import utils

from tempest.lib import exceptions


class VolumeService(VersionedService):
    def set_extensions(self):
        body = self.do_get(self.service_url + '/extensions')
        body = json.loads(body)
        self.extensions = list(map(lambda x: x['alias'], body['extensions']))

    def set_versions(self):
        url_without_version = utils.get_base_url(self.service_url)
        body = self.do_get(url_without_version, top_level=False)
        self.versions_body = json.loads(body)
        self.versions = self.deserialize_versions(self.versions_body)

    def get_volume_pools(self):
        body = self.do_get(self.service_url + '/scheduler-stats/get_pools')
        body = json.loads(body)
        return body

    def set_default_tempest_options(self, conf):
        if 'v3' in self.service_url:
            m_vs = self.filter_api_microversions()
            conf.set('volume', 'min_microversion', m_vs['min_microversion'])
            conf.set('volume', 'max_microversion', m_vs['max_microversion'])

        try:
            pools = self.get_volume_pools()['pools']
        except exceptions.Forbidden:
            C.LOG.warning("User has no permissions to list back-end storage "
                          "pools - storage back-ends can't be discovered.")
            return
        except ServiceError:
            C.LOG.warning("Volume backend doesn't exist.")
            return
        if pools:
            backends = [
                re.findall(r'(\w*)@(\w*)', pool['name'])[0][1]
                for pool in pools
            ]
            conf.set('volume', 'backend_names', ','.join(backends))
            if len(backends) > 1:
                conf.set('volume-feature-enabled', 'multi_backend', 'True')

    def get_service_extension_key(self):
        return 'api_extensions'

    @staticmethod
    def get_codename():
        return 'cinder'

    def get_feature_name(self):
        return 'volume'

    def get_unversioned_service_type(self):
        return 'volume'

    @staticmethod
    def get_service_type():
        return ['volumev3']

    def post_configuration(self, conf, is_service):
        # Verify if the cinder backup service is enabled
        if not is_service(name=self.name):
            C.LOG.info("No volume service found, "
                       "skipping backup service check")
            conf.set('volume-feature-enabled', 'backup', 'False')
            return
        try:
            params = {'binary': 'cinder-backup'}
            is_backup = self.client.list_services(**params)
        except exceptions.Forbidden:
            C.LOG.warning("User has no permissions to list services - "
                          "cinder-backup service can't be discovered.")
            conf.set('volume-feature-enabled', 'backup', 'False')
            return

        if is_backup:
            services = is_backup['services']
            backup_enabled = 'False'
            for service in services:
                if service['state'] == 'up':
                    backup_enabled = 'True'
                    break

            # post_configuration method is called with every volume (v2,
            # v3) service, therefore set the value with priority so that it
            # can't be overrided by this method called from other instance
            # of volume service
            conf.set('volume-feature-enabled', 'backup', backup_enabled,
                     priority=True)
