# Copyright (C) 2017, Red Hat, Inc.
#
# 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.

import argparse
import collections.abc
import fnmatch
import importlib
import inspect
import re
import sys
import typing as ty

from docutils import nodes
from docutils.parsers import rst
from docutils.parsers.rst import directives
from docutils import statemachine
import sphinx.application

from cliff import app
from cliff import command
from cliff import commandmanager


def _indent(text: str) -> str:
    """Indent by four spaces."""
    prefix = ' ' * 4

    def prefixed_lines() -> collections.abc.Iterable[str]:
        for line in text.splitlines(True):
            yield (prefix + line if line.strip() else line)

    return ''.join(prefixed_lines())


def _format_description(
    parser: argparse.ArgumentParser,
) -> collections.abc.Iterable[str]:
    """Get parser description.

    We parse this as reStructuredText, allowing users to embed rich
    information in their help messages if they so choose.
    """
    if parser.description is None:
        return
        yield

    yield from statemachine.string2lines(
        parser.description, tab_width=4, convert_whitespace=True
    )


def _format_usage(parser: argparse.ArgumentParser) -> list[str]:
    """Get usage without a prefix."""
    fmt = argparse.HelpFormatter(parser.prog)

    optionals = parser._get_optional_actions()
    positionals = parser._get_positional_actions()
    groups = parser._mutually_exclusive_groups

    # hacked variant of the regex used by the actual argparse module. Unlike
    # that version, this one attempts to group long and short opts with their
    # optional arguments ensuring that, for example, '--format <FORMAT>'
    # becomes ['--format <FORMAT>'] and not ['--format', '<FORMAT>'].
    # Yes, they really do use regexes to break apart and rewrap their help
    # string. Don't ask me why.
    part_regexp = re.compile(
        r"""
        \(.*?\)+ |
        \[.*?\]+ |
        (?:(?:-\w|--\w+(?:-\w+)*)(?:\s+<?\w[\w-]*>?)?) |
        \S+
    """,
        re.VERBOSE,
    )

    opt_usage = fmt._format_actions_usage(optionals, groups)
    pos_usage = fmt._format_actions_usage(positionals, groups)

    opt_parts = part_regexp.findall(opt_usage)
    pos_parts = part_regexp.findall(pos_usage)
    parts = opt_parts + pos_parts

    if len(' '.join([parser.prog] + parts)) < 72:
        return [' '.join([parser.prog] + parts)]

    return [parser.prog] + [_indent(x) for x in parts]


def _format_epilog(
    parser: argparse.ArgumentParser,
) -> collections.abc.Iterable[str]:
    """Get parser epilog.

    We parse this as reStructuredText, allowing users to embed rich
    information in their help messages if they so choose.
    """
    if parser.epilog is None:
        return
        yield

    yield from statemachine.string2lines(
        parser.epilog, tab_width=4, convert_whitespace=True
    )


def _format_positional_action(
    action: argparse.Action,
) -> collections.abc.Iterable[str]:
    """Format a positional action."""
    if action.help == argparse.SUPPRESS:
        return

    if action.metavar:
        # metavar can be a tuple if nargs > 1
        # https://docs.python.org/3/library/argparse.html#metavar
        if isinstance(action.metavar, tuple):
            metavar = action.metavar[0]
        else:
            metavar = action.metavar
    else:
        metavar = action.dest

    # NOTE(stephenfin): We strip all types of brackets from 'metavar' because
    # the 'option' directive dictates that only option argument names should be
    # surrounded by angle brackets
    yield '.. option:: {}'.format(metavar.strip('<>[]() '))
    if action.help:
        yield ''
        for line in statemachine.string2lines(
            action.help, tab_width=4, convert_whitespace=True
        ):
            yield _indent(line)


def _format_optional_action(
    action: argparse.Action,
) -> collections.abc.Iterable[str]:
    """Format an optional action."""
    if action.help == argparse.SUPPRESS or action.option_strings is None:
        return

    if action.nargs == 0:
        yield '.. option:: {}'.format(', '.join(action.option_strings))
    else:
        if action.metavar:
            # metavar can be a tuple if nargs > 1
            # https://docs.python.org/3/library/argparse.html#metavar
            if isinstance(action.metavar, tuple):
                metavar = action.metavar[0]
            else:
                metavar = action.metavar
        else:
            metavar = f'<{action.dest.upper()}>'
        # TODO(stephenfin): At some point, we may wish to provide more
        # information about the options themselves, for example, if nargs is
        # specified
        option_strings = [
            ' '.join([x, metavar]) for x in action.option_strings
        ]
        yield '.. option:: {}'.format(', '.join(option_strings))

    if action.help:
        yield ''
        for line in statemachine.string2lines(
            action.help, tab_width=4, convert_whitespace=True
        ):
            yield _indent(line)


def _format_parser(
    parser: argparse.ArgumentParser,
) -> collections.abc.Iterable[str]:
    """Format the output of an argparse 'ArgumentParser' object.

    Given the following parser::

      >>> import argparse
      >>> parser = argparse.ArgumentParser(prog='hello-world', \
              description='This is my description.',
              epilog='This is my epilog')
      >>> parser.add_argument('name', help='User name', metavar='<name>')
      >>> parser.add_argument('--language', action='store', dest='lang', \
              help='Greeting language')

    Returns the following::

      This is my description.

      .. program:: hello-world
      .. code:: shell

          hello-world [-h] [--language LANG] <name>

      .. option:: name

          User name

      .. option:: --language LANG

          Greeting language

      .. option:: -h, --help

          Show this help message and exit

      This is my epilog.
    """
    if parser.description:
        for line in _format_description(parser):
            yield line
        yield ''

    yield f'.. program:: {parser.prog}'

    yield '.. code-block:: shell'
    yield ''
    for line in _format_usage(parser):
        yield _indent(line)
    yield ''

    # In argparse, all arguments and parameters are known as "actions".
    # Optional actions are what would be known as flags or options in other
    # libraries, while positional actions would generally be known as
    # arguments. We present these slightly differently.

    for action in parser._get_optional_actions():
        for line in _format_optional_action(action):
            yield line
        yield ''

    for action in parser._get_positional_actions():
        for line in _format_positional_action(action):
            yield line
        yield ''

    if parser.epilog:
        for line in _format_epilog(parser):
            yield line
        yield ''


class AutoprogramCliffDirective(rst.Directive):
    """Auto-document a subclass of `cliff.command.Command`."""

    has_content = False
    required_arguments = 1
    option_spec = {
        'command': directives.unchanged,
        'arguments': directives.unchanged,
        'ignored': directives.unchanged,
        'application': directives.unchanged,
    }

    def _get_ignored_opts(self) -> list[str]:
        global_ignored = self.env.config.autoprogram_cliff_ignored
        local_ignored = self.options.get('ignored', '')
        local_ignored = [
            x.strip() for x in local_ignored.split(',') if x.strip()
        ]
        return list(set(global_ignored + local_ignored))

    def _drop_ignored_options(
        self, parser: argparse.ArgumentParser, ignored_opts: list[str]
    ) -> None:
        for action in list(parser._actions):
            for option_string in action.option_strings:
                if option_string in ignored_opts:
                    del parser._actions[parser._actions.index(action)]
                    break

    def _load_app(self) -> app.App | None:
        mod_str, _sep, class_str = self.arguments[0].rpartition('.')
        if not mod_str:
            return None

        try:
            importlib.import_module(mod_str)
        except ImportError:
            return None

        try:
            cliff_app_class = getattr(sys.modules[mod_str], class_str)
        except AttributeError:
            return None

        if not inspect.isclass(cliff_app_class):
            return None

        if not issubclass(cliff_app_class, app.App):
            return None

        app_arguments = self.options.get('arguments', '').split()
        return cliff_app_class(*app_arguments)

    def _load_command(
        self,
        manager: commandmanager.CommandManager,
        command_name: str,
    ) -> type[command.Command]:
        """Load a command using an instance of a `CommandManager`."""
        try:
            # find_command expects the value of argv so split to emulate that
            return manager.find_command(command_name.split())[0]
        except ValueError:
            raise self.error(
                f'"{command_name}" is not a valid command in the '
                f'"{manager.namespace}" namespace'
            )

    def _load_commands(self) -> dict[str, type[command.Command]]:
        # TODO(sfinucan): We should probably add this wildcarding functionality
        # to the CommandManager itself to allow things like "show me the
        # commands like 'foo *'"
        command_pattern = self.options.get('command')
        manager = commandmanager.CommandManager(self.arguments[0])
        if command_pattern:
            commands = [
                x
                for x in manager.commands
                if fnmatch.fnmatch(x, command_pattern)
            ]
        else:
            commands = list(manager.commands.keys())

        if not commands:
            msg = 'No commands found in the "{}" namespace'
            if command_pattern:
                msg += ' using the "{}" command name/pattern'
            msg += (
                '. Are you sure this is correct and the application being '
                'documented is installed?'
            )
            raise self.warning(msg.format(self.arguments[0], command_pattern))

        return dict(
            (name, self._load_command(manager, name)) for name in commands
        )

    def _generate_app_node(
        self, app: app.App, application_name: str
    ) -> list[nodes.Node]:
        ignored_opts = self._get_ignored_opts()

        parser = app.parser

        self._drop_ignored_options(parser, ignored_opts)

        parser.prog = application_name

        source_name = f'<{app.__class__.__name__}>'
        result = statemachine.StringList()
        for line in _format_parser(parser):
            result.append(line, source_name)

        section = nodes.section()
        self.state.nested_parse(result, 0, section)

        return section.children

    def _generate_nodes_per_command(
        self,
        title: str,
        command_name: str,
        command_class: type[command.Command],
        ignored_opts: list[str],
    ) -> list[nodes.Node]:
        """Generate the relevant Sphinx nodes.

        This doesn't bother using raw docutils nodes as they simply don't offer
        the power of directives, like Sphinx's 'option' directive. Instead, we
        generate reStructuredText and parse this in a nested context (to obtain
        correct header levels). Refer to [1] for more information.

        [1] http://www.sphinx-doc.org/en/stable/extdev/markupapi.html

        :param title: Title of command
        :param command_name: Name of command, as used on the command line
        :param command_class: Subclass of :py:class:`cliff.command.Command`
        :param prefix: Prefix to apply before command, if any
        :param ignored_opts: A list of options to exclude from output, if any
        :returns: A list of nested docutil nodes
        """
        # TODO(stephenfin): Pass proper arguments here
        command = command_class(None, None)  # type: ignore
        if not getattr(command, 'app_dist_name', None):
            command.app_dist_name = (
                self.env.config.autoprogram_cliff_app_dist_name
            )
        parser = command.get_parser(command_name)
        ignored_opts = ignored_opts or []

        self._drop_ignored_options(parser, ignored_opts)

        section = nodes.section(
            '',
            nodes.title(text=title),
            ids=[nodes.make_id(title)],
            names=[nodes.fully_normalize_name(title)],
        )

        source_name = f'<{command.__class__.__name__}>'
        result = statemachine.StringList()

        for line in _format_parser(parser):
            result.append(line, source_name)

        self.state.nested_parse(result, 0, section)

        return [section]

    def _generate_command_nodes(
        self, commands: dict[str, type[command.Command]], application_name: str
    ) -> list[nodes.Node]:
        ignored_opts = self._get_ignored_opts()
        output = []
        for command_name in sorted(commands):
            command_class = commands[command_name]
            title = command_name
            if application_name:
                command_name = ' '.join([application_name, command_name])

            output.extend(
                self._generate_nodes_per_command(
                    title, command_name, command_class, ignored_opts
                )
            )

        return output

    def run(self) -> list[nodes.Node]:
        self.env = self.state.document.settings.env

        application_name = (
            self.options.get('application')
            or self.env.config.autoprogram_cliff_application
        )

        app = self._load_app()
        if app:
            return self._generate_app_node(app, application_name)

        commands = self._load_commands()
        return self._generate_command_nodes(commands, application_name)


def setup(app: sphinx.application.Sphinx) -> dict[str, ty.Any]:
    app.add_directive('autoprogram-cliff', AutoprogramCliffDirective)
    app.add_config_value('autoprogram_cliff_application', '', 'env')
    app.add_config_value('autoprogram_cliff_ignored', ['--help'], 'env')
    app.add_config_value('autoprogram_cliff_app_dist_name', None, 'env')

    return {
        'parallel_read_safe': True,
        'parallel_write_safe': True,
    }
