# Copyright (c) 2020 Mirantis 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.

"""
Openstack Client - Main File
"""

import os_client_config
from oslo_log import log as logging
from oslo_utils import importutils

LOG = logging.getLogger(__name__)


class OpenStackClientException(Exception):
    '''
    Custom Exception
    '''


def init_profiling(os_profile):
    """
    Initialize Profiling
    """
    if os_profile:
        osprofiler_profiler = importutils.try_import("osprofiler.profiler")

        if osprofiler_profiler:  # lib is present
            osprofiler_profiler.init(os_profile)
            trace_id = osprofiler_profiler.get().get_base_id()
            LOG.info('Profiling is enabled, trace id: %s', trace_id)
        else:  # param is set, but lib is not present
            LOG.warning('Profiling could not be enabled. To enable profiling '
                        'please install "osprofiler" library')


class OpenStackClient():
    """
    Client Class
    """
    def __init__(self, openstack_params):
        """
        Initialize
        """
        LOG.debug('Establishing connection to OpenStack')

        init_profiling(openstack_params.get('os_profile'))

        config = os_client_config.OpenStackConfig()
        cloud_config = config.get_one_cloud(**openstack_params)
        if openstack_params['os_insecure']:
            cloud_config.config['verify'] = False
            cloud_config.config['cacert'] = None
        self.keystone_session = cloud_config.get_session()
        self.nova = cloud_config.get_legacy_client('compute')
        self.neutron = cloud_config.get_legacy_client('network')
        self.glance = cloud_config.get_legacy_client('image')

        # heat client wants endpoint to be always set
        endpoint = cloud_config.get_session_endpoint('orchestration')
        if not endpoint:
            raise OpenStackClientException(
                'Endpoint for orchestration service is not found')
        self.heat = cloud_config.get_legacy_client('orchestration',
                                                   endpoint=endpoint)

        # Ping OpenStack
        self.keystone_session.get_token()

        LOG.info('Connection to OpenStack is initialized')