<reponame>rajkumarpatel2602/cortx-multisite
#
# Copyright (c) 2021 Seagate Technology LLC and/or its Affiliates
#
# 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.
#
# For any questions about this software or licensing,
# please email <EMAIL> or <EMAIL>.
#
from aiohttp import web
import asyncio
import logging
import sys
from .config import Config
from .distributor import JobDistributor
from s3replicationcommon.log import setup_logger
from .job_routes import routes as job_routes
from .subscriber_routes import routes as subscriber_routes
from s3replicationcommon.jobs import Jobs
from .subscribers import Subscribers

_logger = logging.getLogger("s3replicationmanager")


async def on_startup(app):
    _logger.debug("Starting server...")

    distributor = JobDistributor(app)
    app["job_distributor"] = distributor
    asyncio.ensure_future(distributor.start())


async def on_shutdown(app):
    _logger.debug("Performing cleanup on shutdown...")
    app["job_distributor"].stop()
    await app['subscribers'].close()


class ReplicationManagerApp:
    def __init__(self, config_file, log_config_file):
        """Initialise logger and configuration."""
        self._config = Config(config_file)
        if self._config.load() is None:
            print("Failed to load configuration.\n")
            sys.exit(-1)

        # Setup logging.
        self._logger = setup_logger('s3replicationmanager', log_config_file)
        if self._logger is None:
            print("Failed to configure logging.\n")
            sys.exit(-1)

        self._config.print_with(self._logger)

        self._jobs = Jobs(self._logger, "all-jobs")
        self._subscribers = Subscribers()

    def run(self):
        app = web.Application()

        # Setup the global context store.
        # https://docs.aiohttp.org/en/stable/web_advanced.html#application-s-config
        app["config"] = self._config
        # Jobs are in 3 phases, 1. Queued by FDMI, 2. InProgress - Sent for
        # processing to replicator and 3. Completed and acknowledged by
        # replicator.
        app['all_jobs'] = self._jobs
        app['subscribers'] = self._subscribers

        # Setup application routes.
        app.add_routes([
            *job_routes,
            *subscriber_routes
        ])

        # Setup startup/shutdown handlers
        app.on_startup.append(on_startup)
        app.on_shutdown.append(on_shutdown)

        # Start the REST server.
        web.run_app(app, host=self._config.host, port=self._config.port)
