# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
"""An autorest MD to RST plugin."""
import logging
from typing import Any, Dict, Set, Union

import m2r2
from pygen import YamlUpdatePlugin

from . import YamlUpdatePluginAutorest


_LOGGER = logging.getLogger(__name__)


class GeneratorRenderer(m2r2.RestRenderer):
    """Redefine the concept of inline HTML in the renderer, we don't want to define a new format
    in the description/summary.
    """

    def inline_html(self, html: str) -> str:
        """Do not render inline HTML with a role definition."""
        return r"\ :code:`{}`".format(html)


class M2R(YamlUpdatePlugin):
    """A plugin to convert any description and summary from MD to RST."""

    def update_yaml(self, yaml_data: Dict[str, Any]) -> None:
        """Convert in place the YAML str."""
        self._convert_docstring_no_cycles(yaml_data, set())

    def _convert_docstring_no_cycles(self, yaml_data: Union[Dict[str, Any], str], node_list: Set[int]) -> None:
        """Walk the YAML tree to convert MD to RST."""
        if id(yaml_data) in node_list:
            return
        node_list.add(id(yaml_data))

        if isinstance(yaml_data, list):
            for elt in yaml_data:
                self._convert_docstring_no_cycles(elt, node_list)
        elif isinstance(yaml_data, dict):
            for key, value in yaml_data.items():
                if key in ["description", "summary"]:
                    yaml_data[key] = self.convert_to_rst(value)
                    continue
                self._convert_docstring_no_cycles(value, node_list)

    @staticmethod
    def convert_to_rst(string_to_convert: str) -> str:
        """Convert that string from MD to RST."""
        try:
            return m2r2.convert(string_to_convert, renderer=GeneratorRenderer()).strip()
        except Exception:  # pylint: disable=broad-except
            return string_to_convert


class M2RAutorest(YamlUpdatePluginAutorest, M2R):
    def get_options(self) -> Dict[str, Any]:
        return {}