"""Core way to access configuration"""

import copy
from typing import TYPE_CHECKING, Any, Dict, List, Tuple, Type, Union

import pgtrigger
from django.apps import apps
from django.conf import settings
from django.db import models
from django.utils.module_loading import import_string

from pghistory import constants

if TYPE_CHECKING:
    from json import JSONEncoder

    from django.db.models import Model, QuerySet

    from pghistory.admin import EventsAdmin
    from pghistory.core import Tracker


def default_trackers() -> Union[Tuple["Tracker", ...], None]:
    """
    The default event trackers.

    Returns:
        Default pghistory trackers
    """
    default_trackers = getattr(settings, "PGHISTORY_DEFAULT_TRACKERS", None)

    # Copy the default trackers, otherwise we end up with the same instances used across
    # all models which causes issues
    if default_trackers:
        default_trackers = copy.deepcopy(default_trackers)

    return default_trackers


def append_only() -> bool:
    """If event models should be append-only by default.

    Protection triggers are installed for update and delete operations if `True`.

    Returns:
        `True` if event models should be append-only by default.
    """
    return getattr(settings, "PGHISTORY_APPEND_ONLY", False)


def middleware_methods() -> Tuple[str, ...]:
    """
    Methods tracked by the pghistory middleware.

    Returns:
        The HTTP methods
    """
    return getattr(
        settings, "PGHISTORY_MIDDLEWARE_METHODS", ("GET", "POST", "PUT", "PATCH", "DELETE")
    )


def install_context_func_on_migrate() -> bool:
    """True if the pghistory context tracking function is installed after migration.

    Defaults to `False`.

    Returns:
        `True` if the _pgh_attach_context() Postgres func is installed after migration.
    """
    return getattr(settings, "PGHISTORY_INSTALL_CONTEXT_FUNC_ON_MIGRATE", False)


def json_encoder() -> Type["JSONEncoder"]:
    """The JSON encoder when tracking context

    Returns:
        The JSON encoder class
    """
    encoder = getattr(
        settings, "PGHISTORY_JSON_ENCODER", "django.core.serializers.json.DjangoJSONEncoder"
    )

    if isinstance(encoder, str):  # pragma: no branch
        encoder = import_string(encoder)

    return encoder


def base_model() -> Type["Model"]:
    """The base model for event models.

    Returns:
        The model class
    """
    event_model = import_string("pghistory.models.Event")
    base_model = getattr(settings, "PGHISTORY_BASE_MODEL", event_model)

    if isinstance(base_model, str):  # pragma: no cover
        base_model = import_string(base_model)

    assert issubclass(base_model, event_model)
    return base_model


def field() -> "Field":
    """The default configuration for all fields in event models.

    Returns:
        The pghistory `Field` class
    """
    field = getattr(settings, "PGHISTORY_FIELD", Field())
    assert isinstance(field, Field)
    return field


def related_field() -> "RelatedField":
    """The default configuration for related fields in event models.

    Returns:
        The pghistory `RelatedField` class
    """
    related_field = getattr(settings, "PGHISTORY_RELATED_FIELD", RelatedField())
    assert isinstance(related_field, RelatedField)
    return related_field


def foreign_key_field() -> "ForeignKey":
    """The default configuration for foreign keys in event models.

    Returns:
        The pghistory `ForeignKey` class
    """
    foreign_key_field = getattr(settings, "PGHISTORY_FOREIGN_KEY_FIELD", ForeignKey())
    assert isinstance(foreign_key_field, ForeignKey)
    return foreign_key_field


def context_field() -> Union["ContextForeignKey", "ContextJSONField"]:
    """The default field config to use for context in event models.

    Returns:
        The pghistory field class
    """

    # Note: We will be changing the default context field to have on_delete=PROTECT
    # in version 3.
    context_field = getattr(settings, "PGHISTORY_CONTEXT_FIELD", ContextForeignKey())
    assert isinstance(context_field, (ContextForeignKey, ContextJSONField))
    return context_field


def context_id_field() -> "ContextUUIDField":
    """
    The default field config to use for context ID in event models when context is denormalized.

    Returns:
        The pghistory context UUID field
    """

    context_id_field = getattr(settings, "PGHISTORY_CONTEXT_ID_FIELD", ContextUUIDField())
    assert isinstance(context_id_field, ContextUUIDField)
    return context_id_field


def obj_field() -> "ObjForeignKey":
    """The default field config to use for object references in event models.

    Returns:
        The pghistory object field
    """

    obj_field = getattr(settings, "PGHISTORY_OBJ_FIELD", ObjForeignKey())
    assert isinstance(obj_field, ObjForeignKey)
    return obj_field


def exclude_field_kwargs() -> Dict["Field", List[str]]:
    """
    Provide a mapping of field classes to a list of keyword args to ignore
    when instantiating the field on the event model.

    For example, a field may not allow `unique` as a keyword argument.
    If so, set `settings.PGHISTORY_EXCLUDE_FIELD_KWARGS = {"field.FieldClass": ["unique"]}`

    Returns:
        The mapping of field classes to kwargs
    """
    exclude_field_kwargs = getattr(settings, "PGHISTORY_EXCLUDE_FIELD_KWARGS", {})
    assert isinstance(exclude_field_kwargs, dict)
    for val in exclude_field_kwargs.values():
        assert isinstance(val, (list, tuple))

    # Import strings
    exclude_field_kwargs = {
        import_string(key) if isinstance(key, str) else key: value
        for key, value in exclude_field_kwargs.items()
    }

    return exclude_field_kwargs


def created_at_function() -> str:
    """The default PostgreSQL function used to populate the "pgh_created_at" field.

    Use `clock_timestamp()` to set the actual current time and not the time of the transaction.

    Returns:
        PostgreSQL function to be used as a default value for the "pgh_created_at" field.
    """
    return getattr(settings, "PGHISTORY_CREATED_AT_FUNCTION", "NOW()")


def admin_ordering() -> List[str]:
    """The default ordering for the events admin.

    Returns:
        The fields over which to order by
    """
    ordering = getattr(settings, "PGHISTORY_ADMIN_ORDERING", "-pgh_created_at") or []

    if not isinstance(ordering, (list, tuple)):
        ordering = [ordering]

    return ordering


def admin_model() -> Type["Model"]:
    """The default list display for the events admin.

    Returns:
        The model class
    """
    return apps.get_model(getattr(settings, "PGHISTORY_ADMIN_MODEL", "pghistory.Events"))


def admin_queryset() -> "QuerySet":
    """The default queryset for the events admin.

    Returns:
        The queryset
    """
    return getattr(
        settings, "PGHISTORY_ADMIN_QUERYSET", admin_model().objects.order_by(*admin_ordering())
    )


def admin_class() -> Type["EventsAdmin"]:
    """The admin class to use for the events admin.

    Returns:
        The pghistory.admin.EventsAdmin class
    """

    events_admin = import_string("pghistory.admin.EventsAdmin")
    admin_class = getattr(settings, "PGHISTORY_ADMIN_CLASS", events_admin)

    if isinstance(admin_class, str):  # pragma: no cover
        admin_class = import_string(admin_class)

    assert issubclass(admin_class, events_admin)

    return admin_class


def admin_all_events() -> bool:
    """True if all events should be shown in the admin when there are no filters.

    Returns:
        The bool setting
    """
    return getattr(settings, "PGHISTORY_ADMIN_ALL_EVENTS", True)


def admin_list_display() -> List[str]:
    """The default list display for the events admin.

    Returns:
        The field display order list
    """
    defaults = ["pgh_created_at", "pgh_obj_model", "pgh_obj_id", "pgh_diff"]

    if admin_queryset().model._meta.label == "pghistory.MiddlewareEvents":
        defaults.extend(["user", "url"])

    return getattr(settings, "PGHISTORY_ADMIN_LIST_DISPLAY", defaults)


def level() -> pgtrigger.Level:
    """The default level for the event model.

    Returns:
        A level, either `pghistory.Row` or `pghistory.Statement`
    """
    return getattr(settings, "PGHISTORY_LEVEL", pgtrigger.Row)


def _get_kwargs(vals):
    return {
        key: val
        for key, val in vals.items()
        if key not in ("self", "kwargs", "__class__") and val is not constants.UNSET
    }


class Field:
    """Configuration for fields.

    The default values for the attributes ensure that
    event models don't have unnecessary uniqueness constraints
    carried over from the tracked model.

    Attributes:
        primary_key (bool, default=False): True if a primary key
        unique (bool, default=False): True if unique
        blank (bool): True if blank
        null (bool): True if null
        db_index (bool): True if indexed
        editable (bool): True if editable
        unique_for_date (bool): True if unique for date
        unique_for_month (bool): True if unique for the month
        unique_for_year (bool): True if unique for the year
    """

    def __init__(
        self,
        *,
        primary_key: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        unique: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        blank: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        null: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        db_index: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        editable: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        unique_for_date: Union[bool, None, constants.Unset, constants.Default] = constants.UNSET,
        unique_for_month: Union[bool, None, constants.Unset, constants.Default] = constants.UNSET,
        unique_for_year: Union[bool, None, constants.Unset, constants.Default] = constants.UNSET,
    ):
        self._kwargs = _get_kwargs(locals())
        self._finalized = False

    @property
    def kwargs(self):
        return {
            key: val
            for key, val in {**self.get_default_kwargs(), **self._kwargs}.items()
            if val is not constants.DEFAULT
        }

    def get_default_kwargs(self):
        return {
            **Field(
                primary_key=False,
                unique=False,
                db_index=False,
                unique_for_date=None,
                unique_for_month=None,
                unique_for_year=None,
            )._kwargs,
            **field()._kwargs,
        }


class RelatedField(Field):
    """Configuration for related fields.

    By default, related names are stripped to avoid
    unnecessary clashes.

    Note that all arguments from `Field` can also be supplied.

    Attributes:
        related_name (str, default="+"): The related_name to use
        related_query_name (str, default="+"): The related_query_name
            to use
    """

    def __init__(
        self,
        *,
        related_name: Union[str, constants.Unset, constants.Default] = constants.UNSET,
        related_query_name: Union[str, constants.Unset, constants.Default] = constants.UNSET,
        **kwargs: Any,
    ):
        super().__init__(**kwargs)
        self._kwargs.update(_get_kwargs(locals()))

    def get_default_kwargs(self):
        return {
            **super().get_default_kwargs(),
            **RelatedField(related_name="+", related_query_name="+")._kwargs,
            **related_field()._kwargs,
        }


class ForeignKey(RelatedField):
    """Configuration for foreign keys.

    Arguments for `RelatedField` and `Field` can also be supplied.
    Note that db_index is overridden to `True` for all foreign keys

    Attributes:
        on_delete (default=models.DO_NOTHING): Django's on_delete property
        db_constraint (bool, default=False): True to use a datbase constraint
            for the foreign key
    """

    def __init__(
        self,
        *,
        on_delete: Any = constants.UNSET,
        db_constraint: Union[bool, constants.Unset, constants.Default] = constants.UNSET,
        **kwargs: Any,
    ):
        super().__init__(**kwargs)
        self._kwargs.update(_get_kwargs(locals()))

    def get_default_kwargs(self):
        return {
            **super().get_default_kwargs(),
            **ForeignKey(on_delete=models.DO_NOTHING, db_index=True, db_constraint=False)._kwargs,
            **foreign_key_field()._kwargs,
        }


class ContextForeignKey(ForeignKey):
    """Configuration for the `pgh_context` field when a foreign key is used.

    Overrides null to `True`.

    Attributes:
        null (bool, default=True): True if nullable context is allowed
        related_query_name (str): The related_query_name to use
    """

    def __init__(
        self,
        *,
        null: bool = True,
        related_query_name: Union[str, constants.Unset, constants.Default] = constants.DEFAULT,
        **kwargs: Any,
    ):
        super().__init__(null=null, related_query_name=related_query_name, **kwargs)
        self._kwargs.update(_get_kwargs(locals()))


class ContextJSONField(Field):
    """Configuration for the `pgh_context` field when denormalized context is used.

    Attributes:
        null (bool, default=True): True if nullable context is allowed
    """

    def __init__(self, *, null: bool = True, **kwargs: Any):
        super().__init__(null=null, **kwargs)
        self._kwargs.update(_get_kwargs(locals()))


class ContextUUIDField(Field):
    """Configuration for the `pgh_context_id` field when denormalized context is used.

    Attributes:
        null (bool, default=True): True if nullable context is allowed
    """

    def __init__(self, *, null: bool = True, **kwargs: Any):
        super().__init__(null=null, **kwargs)
        self._kwargs.update(_get_kwargs(locals()))


class ObjForeignKey(ForeignKey):
    """Configuration for the `pgh_obj` field

    Attributes:
        related_name (str): The related_name to use
        related_query_name (str): The related_query_name to use
    """

    def __init__(
        self,
        *,
        related_name: Union[str, constants.Unset, constants.Default] = constants.DEFAULT,
        related_query_name: Union[str, constants.Unset, constants.Default] = constants.DEFAULT,
        **kwargs,
    ):
        # Note: We will be changing the default object field to nullable with on_delete=SET_NULL
        # in version 3. related_name will also default to "+"
        super().__init__(
            related_name=related_name, related_query_name=related_query_name, **kwargs
        )
        self._kwargs.update(_get_kwargs(locals()))
