# -*- coding: utf8 -*-
# Copyright (c) 2021 Niklas Rosenstein
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to
# deal in the Software without restriction, including without limitation the
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
# sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.

__author__ = "Niklas Rosenstein <rosensteinniklas@gmail.com>"
__version__ = "2.2.1"
__all__ = [
    "Location",
    "Decoration",
    "Docstring",
    "Argument",
    "ApiObject",
    "Indirection",
    "HasLocation",
    "HasMembers",
    "VariableSemantic",
    "Variable",
    "FunctionSemantic",
    "Function",
    "ClassSemantic",
    "Class",
    "Module",
    "load_module",
    "load_modules",
    "dump_module",
    "filter_visit",
    "visit",
    "get_member",
]

import dataclasses
import enum
import io
import json
import sys
import typing as t
import weakref

import databind.json
import typing_extensions as te
from databind.core.settings import Alias, SerializeDefaults, Union


@dataclasses.dataclass
class Location:
    """
    Represents the location of an #ApiObject by a filename and line number.
    """

    filename: str
    lineno: int

    #: If the location of an entity spans over multiple lines, it can be indicated by specifying at
    #: which line it ends with this property.
    endlineno: t.Optional[int] = None


@dataclasses.dataclass
class HasLocation:
    """Base class for objects that have a #Location."""

    location: Location


@dataclasses.dataclass
class Docstring(HasLocation):
    """
    Represents a docstring for an #APIObject, i.e. it's content and location. This class is a subclass of `str`
    for backwards compatibility reasons. Use the #content property to access the docstring content over the
    #Docstring value directory.
    """

    #: The location of where the docstring is defined.
    location: Location = dataclasses.field(repr=False)

    #: The content of the docstring. While the #Docstring class is a subclass of `str` and holds
    #: the same value as *content*, using the #content property should be preferred as the inheritance
    #: from the `str` class may be removed in future versions.
    content: str


@dataclasses.dataclass
class Decoration(HasLocation):
    """
    Represents a decorator on a #Class or #Function.
    """

    #: The location of the decoration in the source code.
    location: Location = dataclasses.field(repr=False)

    #: The name of the decorator (i.e. the text between the `@` and `(`). In languages that support it,
    #: this may be a piece of code.
    name: str

    #: Decorator arguments as plain code (including the leading and trailing parentheses). This is
    #: `None` when the decorator does not have call arguments. This is deprecated in favor of #arglist.
    #: For backwards compatibility, loaders may populate both the #args and #arglist fields.
    args: t.Optional[str] = None

    #: Decorator arguments, one item per argument. For keyword arguments, the keyword name and equals
    #: sign preceed the argument value expression code.
    arglist: t.Optional[t.List[str]] = None


@dataclasses.dataclass
class Argument(HasLocation):
    """
    Represents a #Function argument.
    """

    class Type(enum.Enum):
        """
        The type of the argument. This is currently very Python-centric, however most other languages should be able
        to represent the various argument types with a subset of these types without additions (e.g. Java or TypeScript
        only support #Positional and #PositionalRemainder arguments).
        """

        #: A positional only argument. Such arguments are denoted in Python like this: `def foo(a, b, /): ...`
        POSITIONAL_ONLY: te.Annotated[int, Alias("POSITIONAL_ONLY", "PositionalOnly")] = 0

        #: A positional argument, which may also be given as a keyword argument. Basically that is just a normal
        #: argument as you would see most commonly in Python function definitions.
        POSITIONAL: te.Annotated[int, Alias("POSITIONAL", "Positional")] = 1

        #: An argument that denotes the capture of additional positional arguments, aka. "args" or "varags".
        POSITIONAL_REMAINDER: te.Annotated[int, Alias("POSITIONAL_REMAINDER", "PositionalRemainder")] = 2

        #: A keyword-only argument is denoted in Python like thisL `def foo(*, kwonly): ...`
        KEYWORD_ONLY: te.Annotated[int, Alias("KEYWORD_ONLY", "KeywordOnly")] = 3

        #: An argument that captures additional keyword arguments, aka. "kwargs".
        KEYWORD_REMAINDER: te.Annotated[int, Alias("KEYWORD_REMAINDER", "KeywordRemainder")] = 4

        # backwards compatibility, < 1.2.0
        PositionalOnly: t.ClassVar["Argument.Type"]
        Positional: t.ClassVar["Argument.Type"]
        PositionalRemainder: t.ClassVar["Argument.Type"]
        KeywordOnly: t.ClassVar["Argument.Type"]
        KeywordRemainder: t.ClassVar["Argument.Type"]

    Type.PositionalOnly = Type.POSITIONAL_ONLY
    Type.Positional = Type.POSITIONAL
    Type.PositionalRemainder = Type.POSITIONAL_REMAINDER
    Type.KeywordOnly = Type.KEYWORD_ONLY
    Type.KeywordRemainder = Type.KEYWORD_REMAINDER

    #: The location of the argument in the source code.
    location: Location = dataclasses.field(repr=False)

    #: The name of the argument.
    name: str

    #: The argument type.
    type: Type

    #: A list of argument decorations. Python does not actually support decorators on function arguments
    #: like for example Java does. This is probably premature to add into the API, but hey, here it is.
    decorations: t.Optional[t.List[Decoration]] = None

    #: The datatype/type annotation of this argument as a code string.
    datatype: t.Optional[str] = None

    #: The default value of the argument as a code string.
    default_value: t.Optional[str] = None


@dataclasses.dataclass
class ApiObject(HasLocation):
    """
    The base class for representing "API Objects". Any API object is any addressable entity in code,
    be that a variable/constant, function, class or module.
    """

    #: The location of the API object, i.e. where it is sourced from/defined in the code.
    location: Location = dataclasses.field(repr=False)

    #: The name of the entity. This is usually relative to the respective parent of the entity,
    #: as opposed to it's fully qualified name/absolute name. However, that is more of a
    #: recommendation than rule. For example the #docspec_python loader by default returns
    #: #Module objects with their full module name (and does not create a module hierarchy).
    name: str

    #: The documentation string of the API object.
    docstring: t.Optional[Docstring] = dataclasses.field(repr=False)

    def __post_init__(self) -> None:
        self._parent: t.Optional["weakref.ReferenceType[HasMembers]"] = None

    @property
    def parent(self) -> t.Optional["HasMembers"]:
        """
        Returns the parent of the #HasMembers. Note that if you make any modifications to the API object tree,
        you will need to call #sync_hierarchy() afterwards because adding to #Class.members or #Module.members
        does not automatically keep the #parent property in sync.
        """

        if self._parent is not None:
            parent = self._parent()
            if parent is None:
                raise RuntimeError("lost reference to parent object")
        else:
            parent = None
        return parent

    @parent.setter
    def parent(self, parent: t.Optional["HasMembers"]) -> None:
        if parent is not None:
            self._parent = weakref.ref(parent)
        else:
            self._parent = None

    @property
    def path(self) -> t.List["ApiObject"]:
        """
        Returns a list of all of this API object's parents, from top to bottom. The list includes *self* as the
        last item.
        """

        result = []
        current: t.Optional[ApiObject] = self
        while current:
            result.append(current)
            current = current.parent
        result.reverse()
        return result

    def sync_hierarchy(self, parent: t.Optional["HasMembers"] = None) -> None:
        """
        Synchronize the hierarchy of this API object and all of it's children. This should be called when the
        #HasMembers.members are updated to ensure that all child objects reference the right #parent. Loaders
        are expected to return #ApiObject#s in a fully synchronized state such that the user does not have to
        call this method unless they are doing modifications to the tree.
        """

        self.parent = parent


class VariableSemantic(enum.Enum):
    """
    A list of well-known properties and behaviour that can be attributed to a variable/constant.
    """

    #: The #Variable object is an instance variable of a class.
    INSTANCE_VARIABLE = 0

    #: The #Variable object is a static variable of a class.
    CLASS_VARIABLE = 1

    #: The #Variable object represents a constant value.
    CONSTANT = 2


@dataclasses.dataclass
class Variable(ApiObject):
    """
    Represents a variable assignment (e.g. for global variables (often used as constants) or class members).
    """

    Semantic: t.ClassVar = VariableSemantic

    #: The datatype associated with the assignment as code.
    datatype: t.Optional[str] = None

    #: The value of the variable as code.
    value: t.Optional[str] = None

    #: A list of language-specific modifiers that were used to declare this #Variable object.
    modifiers: t.List[str] = dataclasses.field(default_factory=list)

    #: A list of hints that express semantics of this #Variable object which are not otherwise
    #: derivable from the context.
    semantic_hints: t.List[VariableSemantic] = dataclasses.field(default_factory=list)


@dataclasses.dataclass
class Indirection(ApiObject):
    """
    Represents an imported name. It can be used to properly find the full name target of a link written with a
    local name.
    """

    target: str


class FunctionSemantic(enum.Enum):
    """
    A list of well-known properties and behaviour that can be attributed to a function.
    """

    #: The function is abstract.
    ABSTRACT = 0

    #: The function is final.
    FINAL = 1

    #: The function is a coroutine.
    COROUTINE = 2

    #: The function does not return.
    NO_RETURN = 3

    #: The function is an instance method.
    INSTANCE_METHOD = 4

    #: The function is a classmethod.
    CLASS_METHOD = 5

    #: The function is a staticmethod.
    STATIC_METHOD = 6

    #: The function is a property getter.
    PROPERTY_GETTER = 7

    #: The function is a property setter.
    PROPERTY_SETTER = 8

    #: The function is a property deleter.
    PROPERTY_DELETER = 9


@dataclasses.dataclass
class Function(ApiObject):
    """
    Represents a function definition. This can be in a #Module for plain functions or in a #Class for methods.
    The #decorations need to be introspected to understand if the function has a special purpose (e.g. is it a
    `@property`, `@classmethod` or `@staticmethod`?).
    """

    Semantic: t.ClassVar = FunctionSemantic

    #: A list of modifiers used in the function definition. For example, the only valid modifier in
    #: Python is "async".
    modifiers: t.Optional[t.List[str]]

    #: A list of the function arguments.
    args: t.List[Argument]

    #: The return type of the function as a code string.
    return_type: t.Optional[str]

    #: A list of decorations used on the function.
    decorations: t.Optional[t.List[Decoration]]

    #: A list of hints that describe the object.
    semantic_hints: t.List[FunctionSemantic] = dataclasses.field(default_factory=list)


@dataclasses.dataclass
class HasMembers(ApiObject):
    """
    Base class for API objects that can have members, e.g. #Class and #Module.
    """

    #: The members of the API object.
    members: t.Sequence[ApiObject]

    def sync_hierarchy(self, parent: t.Optional["HasMembers"] = None) -> None:
        self.parent = parent
        for member in self.members:
            member.sync_hierarchy(self)


class ClassSemantic(enum.Enum):
    """
    A list of well-known properties and behaviour that can be attributed to a class.
    """

    #: The class describes an interface.
    INTERFACE = 0

    #: The class is abstract.
    ABSTRACT = 1

    #: The class is final.
    FINAL = 2

    #: The class is an enumeration.
    ENUM = 3


@dataclasses.dataclass
class Class(HasMembers):
    """
    Represents a class definition.
    """

    Semantic: t.ClassVar = ClassSemantic

    #: The metaclass used in the class definition as a code string.
    metaclass: t.Optional[str]

    #: The list of base classes as code strings.
    bases: t.Optional[t.List[str]]

    #: A list of decorations used in the class definition.
    decorations: t.Optional[t.List[Decoration]]

    #: A list of the classes members. #Function#s in a class are to be considered instance methods of
    #: that class unless some information about the #Function indicates otherwise.
    members: t.List["_MemberType"]

    #: A list of language-specific modifiers that were used to declare this #Variable object.
    modifiers: t.List[str] = dataclasses.field(default_factory=list)

    #: A list of hints that describe the object.
    semantic_hints: t.List[ClassSemantic] = dataclasses.field(default_factory=list)


@dataclasses.dataclass
class Module(HasMembers):
    """
    Represents a module, basically a named container for code/API objects. Modules may be nested in other modules.
    Be aware that for historical reasons, some loaders lile #docspec_python by default do not return nested modules,
    even if nesting would be appropriate (and instead the #Module.name simply contains the fully qualified name).
    """

    #: A list of module members.
    members: t.List["_ModuleMemberType"]


_Members = t.Union[Variable, Function, Class, Indirection]
_MemberType = te.Annotated[
    _Members,
    Union({"data": Variable, "function": Function, "class": Class, "indirection": Indirection}, style=Union.FLAT),
]


_ModuleMembers = t.Union[Variable, Function, Class, Module, Indirection]
_ModuleMemberType = te.Annotated[
    _ModuleMembers,
    Union(
        {"data": Variable, "function": Function, "class": Class, "module": Module, "indirection": Indirection},
        style=Union.FLAT,
    ),
]


def load_module(
    source: t.Union[str, t.TextIO, t.Dict[str, t.Any]],
    filename: t.Optional[str] = None,
    loader: t.Callable[[t.IO[str]], t.Any] = json.load,
) -> Module:
    """
    Loads a #Module from the specified *source*, which may be either a filename,
    a file-like object to read from or plain structured data.

    # Arguments
    source: The JSON source to load the module from.
    filename: The name of the source. This will be displayed in error
      messages if the deserialization fails.
    loader: A function for loading plain structured data from a file-like
      object. Defaults to #json.load().

    # Returns
    The loaded `Module` object.
    """

    filename = filename or getattr(source, "name", None)

    if isinstance(source, str):
        if source == "-":
            return load_module(sys.stdin, source, loader)
        with io.open(source, encoding="utf-8") as fp:
            return load_module(fp, source, loader)
    elif hasattr(source, "read"):
        # we ar sure the type is "IO" since the source has a read attribute.
        source = loader(source)  # type: ignore[arg-type]

    module = databind.json.load(source, Module, filename=filename or "<unknown>")
    module.sync_hierarchy()
    return module


def load_modules(
    source: t.Union[str, t.TextIO, t.Iterable[t.Any]],
    filename: t.Optional[str] = None,
    loader: t.Callable[[t.IO[str]], t.Any] = json.load,
) -> t.Iterable[Module]:
    """
    Loads a stream of modules from the specified *source*. Similar to
    #load_module(), the *source* can be a filename, file-like object or a
    list of plain structured data to deserialize from.
    """

    filename = filename or getattr(source, "name", None)

    if isinstance(source, str):
        with io.open(source, encoding="utf-8") as fp:
            yield from load_modules(fp, source, loader)
        return
    elif hasattr(source, "read"):
        source = (loader(io.StringIO(line)) for line in t.cast(t.IO[str], source))

    for data in source:
        module = databind.json.load(data, Module, filename=filename or "<unknown>")
        module.sync_hierarchy()
        yield module


@t.overload
def dump_module(
    module: Module, target: t.Union[str, t.IO[str]], dumper: t.Callable[[t.Any, t.IO[str]], None] = json.dump
) -> None:
    ...


@t.overload
def dump_module(
    module: Module, target: None = None, dumper: t.Callable[[t.Any, t.IO[str]], None] = json.dump
) -> t.Dict[str, t.Any]:
    ...


def dump_module(
    module: Module,
    target: t.Optional[t.Union[str, t.IO[str]]] = None,
    dumper: t.Callable[[t.Any, t.IO[str]], None] = json.dump,
    serialize_defaults: bool = False,
) -> t.Optional[t.Dict[str, t.Any]]:
    """
    Dumps a module to the specified target or returns it as plain structured data.

    :param module: The module to dump.
    :param target: The target to dump to. If #None, the module will be returned as plain structured data.
    :param dumper: A function for dumping plain structured data to a file-like object. Defaults to #json.dump().
    :param serialize_defaults: If #True, default values will be serialized into the payload. Otherwise, they will be
        omitted. Defaults to #False.
    """

    if isinstance(target, str):
        with io.open(target, "w", encoding="utf-8") as fp:
            dump_module(module, fp, dumper)
        return None

    data = databind.json.dump(module, Module, settings=[SerializeDefaults(serialize_defaults)])
    if target:
        dumper(data, target)
        target.write("\n")
        return None
    else:
        return t.cast(t.Dict[str, t.Any], data)


def filter_visit(
    objects: t.MutableSequence[ApiObject],
    predicate: t.Callable[[ApiObject], bool],
    order: str = "pre",
) -> t.MutableSequence[ApiObject]:
    """
    Visits all *objects* recursively, applying the *predicate* in the specified *order*. If
    the predicate returrns #False, the object will be removed from it's containing list.

    If an object is removed in pre-order, it's members will not be visited.

    :param objects: A list of objects to visit recursively. This list will be modified if
      the *predicate* returns #False for an object.
    :param predicate: The function to apply over all visited objects.
    :param order: The order in which the objects are visited. The default order is `'pre'`
      in which case the *predicate* is called before visiting the object's members. The
      order may also be `'post'`.
    """

    if order not in ("pre", "post"):
        raise ValueError("invalid order: {!r}".format(order))

    offset = 0
    for index in range(len(objects)):
        current = objects[index - offset]
        if order == "pre":
            if not predicate(current):
                del objects[index - offset]
                offset += 1
                continue
        if isinstance(current, HasMembers):
            current.members = filter_visit(list(current.members), predicate, order)
        if order == "post":
            if not predicate(current):
                del objects[index - offset]
                offset += 1

    return objects


def visit(
    objects: t.Sequence[ApiObject],
    func: t.Callable[[ApiObject], t.Any],
    order: str = "pre",
) -> None:
    """
    Visits all *objects*, applying *func* in the specified *order*.
    """

    filter_visit(
        t.cast(t.MutableSequence[ApiObject], objects),  # Sequence does not get mutated in this call
        (lambda obj: func(obj) or True),
        order,
    )


def get_member(obj: ApiObject, name: str) -> t.Optional[ApiObject]:
    """
    Generic function to retrieve a member from an API object. This will always return #None for
    objects that don't support members (eg. #Function and #Variable).
    """

    if isinstance(obj, HasMembers):
        for member in obj.members:
            if member.name == name:
                assert isinstance(member, ApiObject), (name, obj, member)
                return member

    return None