# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import logging
from typing import Any, Dict
from pathlib import Path
from jinja2 import Environment, PackageLoader
from pygen import Plugin
from .. import PluginAutorest

_LOGGER = logging.getLogger(__name__)


class MultiClientPlugin(Plugin):  # pylint: disable=abstract-method
    def process(self) -> bool:
        _LOGGER.info("Generating files for multi client")

        env = Environment(
            loader=PackageLoader("autorest.multiclient", "templates"),
            keep_trailing_newline=True,
            line_statement_prefix="##",
            line_comment_prefix="###",
            trim_blocks=True,
            lstrip_blocks=True,
        )

        # __init__.py
        template = env.get_template("init.py.jinja2")
        self.write_file(Path("__init__.py"), template.render())

        # _version.py
        template = env.get_template("version.py.jinja2")
        self.write_file(
            Path("_version.py"),
            template.render(package_version=self.options.get("package-version") or "1.0.0b1"),
        )

        # py.typed
        self.write_file(Path("py.typed"), "# Marker file for PEP 561.")

        _LOGGER.info("Generating Done for multi client!")
        return True


class MultiClientPluginAutorest(MultiClientPlugin, PluginAutorest):
    def get_options(self) -> Dict[str, Any]:
        return {"package-version": self._autorestapi.get_value("package-version")}