# Copyright 2017 Huawei Technologies Co.,LTD.
# 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 oslo_messaging as messaging

from oslo_concurrency import processutils
from oslo_log import log
from oslo_service import service
from oslo_service import wsgi
from oslo_utils import importutils

from cyborg import context
from cyborg import objects
from cyborg.api import app
from cyborg.common import config
from cyborg.common import exception
from cyborg.common import rpc
from cyborg.common.i18n import _
from cyborg.conf import CONF
from cyborg.objects import base as objects_base


LOG = log.getLogger(__name__)


class RPCService(service.Service):
    def __init__(self, manager_module, manager_class, topic, host=None):
        super().__init__()
        self.topic = topic
        self.host = host or CONF.host
        manager_module = importutils.try_import(manager_module)
        manager_class = getattr(manager_module, manager_class)
        self.manager = manager_class(self.topic, self.host)
        self.rpcserver = None

    def start(self):
        super().start()

        target = messaging.Target(topic=self.topic, server=self.host)
        endpoints = [self.manager]
        serializer = objects_base.CyborgObjectSerializer()
        self.rpcserver = rpc.get_server(target, endpoints, serializer)
        self.rpcserver.start()

        admin_context = context.get_admin_context()
        # Use the _args variant to pass kwargs to the callback.
        self.tg.add_dynamic_timer_args(
            self.manager.periodic_tasks,
            kwargs={"context": admin_context},
            periodic_interval_max=CONF.periodic_interval,
        )

        LOG.info(
            'Created RPC server for service %(service)s on host %(host)s.',
            {'service': self.topic, 'host': self.host},
        )

    def stop(self, graceful=True):
        try:
            self.rpcserver.stop()
            self.rpcserver.wait()
        except Exception as e:
            LOG.exception(
                'Service error occurred when stopping the '
                'RPC server. Error: %s',
                e,
            )

        super().stop(graceful=graceful)
        LOG.info(
            'Stopped RPC server for service %(service)s on host %(host)s.',
            {'service': self.topic, 'host': self.host},
        )


def prepare_service(argv=None):
    log.register_options(CONF)
    log.set_defaults(default_log_levels=CONF.default_log_levels)

    argv = argv or []
    config.parse_args(argv)

    log.setup(CONF, 'cyborg')
    objects.register_all()


def process_launcher():
    return service.ProcessLauncher(CONF, restart_method='mutate')


class WSGIService(service.ServiceBase):
    """Provides ability to launch cyborg API from wsgi app."""

    def __init__(self, name, use_ssl=False):
        """Initialize, but do not start the WSGI server.

        :param name: The name of the WSGI server given to the loader.
        :param use_ssl: Wraps the socket in an SSL context if True.
        :returns: None
        """
        self.name = name
        self.app = app.load_app()
        self.workers = CONF.api.api_workers or processutils.get_worker_count()
        if self.workers and self.workers < 1:
            raise exception.ConfigInvalid(
                _(
                    "api_workers value of %d is invalid, "
                    "must be greater than 0."
                )
                % self.workers
            )

        self.server = wsgi.Server(
            CONF,
            self.name,
            self.app,
            host=CONF.api.host_ip,
            port=CONF.api.port,
            use_ssl=use_ssl,
        )

    def start(self):
        """Start serving this service using loaded configuration.

        :returns: None
        """
        self.server.start()

    def stop(self):
        """Stop serving this API.

        :returns: None
        """
        self.server.stop()

    def wait(self):
        """Wait for the service to stop serving this API.

        :returns: None
        """
        self.server.wait()

    def reset(self):
        """Reset server greenpool size to default.

        :returns: None
        """
        self.server.reset()
