# Copyright 2023 highstreet technologies GmbH
#
# 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.

# inspired by https://github.com/rochacbruno/python-project-template

import os
import sys

from network_generation.base import NetworkGenerator
from network_generation.model.python.o_ran_network import ORanNetwork
from network_generation.parameter_validator import ParameterValidator
from network_generation.view.network_viewer import NetworkViewer

"""
CLI interface for network_generation project.
Module as entry point to generate an ietf topology json
"""


def main() -> None:  # pragma: no cover
    """
    The main function executes on commands:
    `python -m network_generation`.

    """
    validator: ParameterValidator = ParameterValidator(sys.argv)

    if validator.is_valid():
        configuration: dict = validator.configuration()
        generator: NetworkGenerator = NetworkGenerator(
            configuration["network"]
        )
        network: ORanNetwork = generator.generate()
        viewer: NetworkViewer = NetworkViewer(network)

        output_folder: str = configuration["outputFolder"]
        # If folder doesn't exist, then create it.
        if not os.path.isdir(output_folder):
            os.makedirs(output_folder)

        name: str = str(configuration["network"]["name"]).lower()
        filename: str = "/".join([output_folder, name])

        # topology json
        if configuration["generationTasks"]["topology"]["enabled"] is True:
            viewer.json().save(
                filename,
                configuration["generationTasks"]["topology"]["compressed"]
            )

        # dir structure for day0 configuration
        # Note: compressed option ignored
        if configuration["generationTasks"]["network_dir"]["enabled"] is True:
            viewer.to_directory(
                output_folder
            )

        # svg xml
        if configuration["generationTasks"]["svg"]["enabled"] is True:
            viewer.svg(
                filename,
                configuration["generationTasks"]["svg"]["compressed"]
            )

        # kml/kmz xml
        if configuration["generationTasks"]["kml"]["enabled"] is True:
            viewer.kml(
                filename,
                configuration["generationTasks"]["kml"]["compressed"]
            )

    else:
        print(validator.error_message())