###
# ============LICENSE_START=======================================================
# Netconf Server
# ================================================================================
# Copyright (C) 2021 Nokia. 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.
# ============LICENSE_END=========================================================
###
import asyncio
import sys
import logging
from retry import retry

from netconf_server.netconf_app_configuration import NetconfAppConfiguration

from netconf_server.netconf_change_listener import NetconfChangeListener
from netconf_server.netconf_change_listener_factory import NetconfChangeListenerFactory, NetconfChangeListenerException
from netconf_server.sysrepo_configuration.sysrepo_configuration_loader import SysrepoConfigurationLoader, \
    ConfigLoadingException
from netconf_server.sysrepo_interface.sysrepo_client import SysrepoClient

logging.basicConfig(
    handlers=[logging.StreamHandler(), logging.FileHandler("/logs/netconf_change_listener.log")],
    level=logging.INFO
)
logger = logging.getLogger("netconf_change_listener")


def run_server_forever(session, connection, change_listener: NetconfChangeListener):
    change_listener.run(session)
    asyncio.get_event_loop().run_forever()


def create_change_listener(app_configuration: NetconfAppConfiguration) -> NetconfChangeListener:
    configuration = SysrepoConfigurationLoader.load_configuration(app_configuration.module_configuration_file_path)
    return subscribe_to_kafka(app_configuration, configuration)


@retry(NetconfChangeListenerException, tries=10, delay=10)
def subscribe_to_kafka(app_configuration, configuration):
    return NetconfChangeListenerFactory(configuration.models_to_subscribe_to, app_configuration).create()


if __name__ == "__main__":
    app_configuration, error = NetconfAppConfiguration.get_configuration(sys.argv)  # type: NetconfAppConfiguration, str

    if app_configuration:
        logger.info("Netconf change listener application configuration: {}".format(app_configuration))
        try:
            netconf_change_listener = create_change_listener(app_configuration)
            SysrepoClient().run_in_session(run_server_forever, netconf_change_listener)
        except ConfigLoadingException:
            logger.error("File to load configuration from file %s" % app_configuration.module_configuration_file_path)
    else:
        logger.error(error)