# -*- coding: utf-8 -*-
import logging

from django.db.models import Q
from django.utils.html import escape, format_html

from .mixins import JSONResponseView

logger = logging.getLogger(__name__)


class DatatableMixin(object):
    """JSON data for datatables"""

    model = None
    columns = []
    _columns = []  # internal cache for columns definition
    order_columns = []
    max_display_length = 100  # max limit of records returned, do not allow to kill our server by huge sets of data
    pre_camel_case_notation = (
        False  # datatables 1.10 changed query string parameter names
    )
    none_string = ""
    escape_values = (
        True  # if set to true then values returned by render_column will be escaped
    )
    columns_data = []
    """
    "is_data_list" determines the type of results. If JavaScript passes data attribute that is not an integer then
    it expects dictionary with specific fields in the response, see:
    https://datatables.net/reference/option/columns.data
    """
    is_data_list = True

    FILTER_ISTARTSWITH = "istartswith"
    FILTER_ICONTAINS = "icontains"

    @property
    def _querydict(self):
        if self.request.method == "POST":
            return self.request.POST
        else:
            return self.request.GET

    def get_filter_method(self):
        """Returns preferred filter method"""
        return self.FILTER_ISTARTSWITH

    def initialize(self, *args, **kwargs):
        """Determine which version of DataTables is being used - there are differences in parameters sent by
        DataTables < 1.10.x
        """
        if "iSortingCols" in self._querydict:
            self.pre_camel_case_notation = True

    def get_order_columns(self):
        """Return list of columns used for ordering.
        By default returns self.order_columns but if these are not defined it tries to get columns
        from the request using the columns[i][name] attribute. This requires proper client side definition of
        columns, eg:
            columns: [
                {
                    name: 'username',
                    data: 'username',
                    orderable: true,
                },
                {
                    name: 'email',
                    data: 'email',
                    orderable: false
                }
            ]
        """
        if self.order_columns or self.pre_camel_case_notation:
            return self.order_columns

        # try to build list of order_columns using request data
        order_columns = []
        for column_def in self.columns_data:
            if column_def["name"] or not self.is_data_list:
                # if self.is_data_list is False then we have a column name in the 'data' attribute, otherwise
                # 'data' attribute is an integer with column index
                if column_def["orderable"]:
                    if self.is_data_list:
                        order_columns.append(column_def["name"])
                    else:
                        order_columns.append(column_def.get("data"))
                else:
                    order_columns.append("")
            else:
                # fallback to columns
                order_columns = self._columns
                break

        self.order_columns = order_columns
        return order_columns

    def get_columns(self):
        """Returns the list of columns to be returned in the result set.
        By default returns self.columns but if these are not defined it tries to get columns
        from the request using the columns[i][data] or columns[i][name] attribute.
        This requires proper client side definition of
        columns, eg:

            columns: [
                {
                    data: 'username'
                },
                {
                    data: 'email'
                }
            ]
        """
        if self.columns or self.pre_camel_case_notation:
            return self.columns

        columns = []
        for column_def in self.columns_data:
            if self.is_data_list:
                # if self.is_data_list is true then 'data' atribute is an integer - column index, so we
                # cannot use it as a column name, let's try 'name' attribute instead
                col_name = column_def["name"]
            else:
                col_name = column_def["data"]

            if col_name:
                columns.append(col_name)
            else:
                return self.columns

        return columns

    @staticmethod
    def _column_value(obj, key):
        """Returns the value from a queryset item"""
        if isinstance(obj, dict):
            return obj.get(key, None)

        return getattr(obj, key, None)

    def _render_column(self, row, column):
        """Renders a column on a row. column can be given in a module notation eg. document.invoice.type"""
        # try to find rightmost object
        obj = row
        parts = column.split(".")
        for part in parts[:-1]:
            if obj is None:
                break
            obj = getattr(obj, part)

        # try using get_OBJECT_display for choice fields
        if hasattr(obj, "get_%s_display" % parts[-1]):
            value = getattr(obj, "get_%s_display" % parts[-1])()
        else:
            value = self._column_value(obj, parts[-1])

        if value is None:
            value = self.none_string

        if self.escape_values:
            value = escape(value)

        return value

    def render_column(self, row, column):
        """Renders a column on a row. column can be given in a module notation eg. document.invoice.type"""
        value = self._render_column(row, column)
        if value and hasattr(row, "get_absolute_url"):
            return format_html('<a href="{}">{}</a>', row.get_absolute_url(), value)
        return value

    def ordering(self, qs):
        """Get parameters from the request and prepare order by clause"""

        # Number of columns that are used in sorting
        sorting_cols = 0
        if self.pre_camel_case_notation:
            try:
                sorting_cols = int(self._querydict.get("iSortingCols", 0))
            except ValueError:
                sorting_cols = 0
        else:
            sort_key = "order[{0}][column]".format(sorting_cols)
            while sort_key in self._querydict:
                sorting_cols += 1
                sort_key = "order[{0}][column]".format(sorting_cols)

        order = []
        order_columns = self.get_order_columns()

        for i in range(sorting_cols):
            # sorting column
            sort_dir = "asc"
            try:
                if self.pre_camel_case_notation:
                    sort_col = int(self._querydict.get("iSortCol_{0}".format(i)))
                    # sorting order
                    sort_dir = self._querydict.get("sSortDir_{0}".format(i))
                else:
                    sort_col = int(self._querydict.get("order[{0}][column]".format(i)))
                    # sorting order
                    sort_dir = self._querydict.get("order[{0}][dir]".format(i))
            except ValueError:
                sort_col = 0

            sdir = "-" if sort_dir == "desc" else ""
            sortcol = order_columns[sort_col]

            if isinstance(sortcol, list):
                for sc in sortcol:
                    order.append("{0}{1}".format(sdir, sc.replace(".", "__")))
            else:
                order.append("{0}{1}".format(sdir, sortcol.replace(".", "__")))

        if order:
            return qs.order_by(*order)
        return qs

    def paging(self, qs):
        """Paging"""
        if self.pre_camel_case_notation:
            limit = min(
                int(self._querydict.get("iDisplayLength", 10)), self.max_display_length
            )
            start = int(self._querydict.get("iDisplayStart", 0))
        else:
            limit = min(int(self._querydict.get("length", 10)), self.max_display_length)
            start = int(self._querydict.get("start", 0))

        # if pagination is disabled ("paging": false)
        if limit == -1:
            return qs

        offset = start + limit

        return qs[start:offset]

    def get_initial_queryset(self):
        if not self.model:
            raise NotImplementedError(
                "Need to provide a model or implement get_initial_queryset!"
            )
        return self.model.objects.all()

    def extract_datatables_column_data(self):
        """Helper method to extract columns data from request as passed by Datatables 1.10+"""
        request_dict = self._querydict
        col_data = []
        if not self.pre_camel_case_notation:
            counter = 0
            data_name_key = "columns[{0}][name]".format(counter)
            while data_name_key in request_dict:
                searchable = (
                    True
                    if request_dict.get("columns[{0}][searchable]".format(counter))
                    == "true"
                    else False
                )
                orderable = (
                    True
                    if request_dict.get("columns[{0}][orderable]".format(counter))
                    == "true"
                    else False
                )

                col_data.append(
                    {
                        "name": request_dict.get(data_name_key),
                        "data": request_dict.get("columns[{0}][data]".format(counter)),
                        "searchable": searchable,
                        "orderable": orderable,
                        "search.value": request_dict.get(
                            "columns[{0}][search][value]".format(counter)
                        ),
                        "search.regex": request_dict.get(
                            "columns[{0}][search][regex]".format(counter)
                        ),
                    }
                )
                counter += 1
                data_name_key = "columns[{0}][name]".format(counter)
        return col_data

    def filter_queryset(self, qs):
        """If search['value'] is provided then filter all searchable columns using filter_method (istartswith
        by default).

        Automatic filtering only works for Datatables 1.10+. For older versions override this method
        """
        columns = self._columns
        if not self.pre_camel_case_notation:
            # get global search value
            search = self._querydict.get("search[value]", None)
            q = Q()
            filter_method = self.get_filter_method()
            for col_no, col in enumerate(self.columns_data):
                # col['data'] - https://datatables.net/reference/option/columns.data
                data_field = col["data"]
                try:
                    data_field = int(data_field)
                except ValueError:
                    pass
                if isinstance(data_field, int):
                    column = columns[
                        data_field
                    ]  # by index so we need columns definition in self._columns
                else:
                    column = data_field
                column = column.replace(".", "__")
                # apply global search to all searchable columns
                if search and col["searchable"]:
                    q |= Q(**{"{0}__{1}".format(column, filter_method): search})

                # column specific filter
                if col["search.value"]:
                    qs = qs.filter(
                        **{
                            "{0}__{1}".format(column, filter_method): col[
                                "search.value"
                            ]
                        }
                    )
            qs = qs.filter(q)
        return qs

    def prepare_results(self, qs):
        data = []
        for item in qs:
            if self.is_data_list:
                data.append(
                    [self.render_column(item, column) for column in self._columns]
                )
            else:
                row = {
                    col_data["data"]: self.render_column(item, col_data["data"])
                    for col_data in self.columns_data
                }
                data.append(row)

        return data

    def handle_exception(self, e):
        logger.exception(str(e))
        raise e

    def count_records(self, qs):
        """
        Calculates the number of records in the queryset.
        This method can be overwritten to optimise page loading speed.
        """
        return qs.count()

    def get_context_data(self, *args, **kwargs):
        try:
            self.initialize(*args, **kwargs)

            # prepare columns data (for DataTables 1.10+)
            self.columns_data = self.extract_datatables_column_data()

            # determine the response type based on the 'data' field passed from JavaScript
            # https://datatables.net/reference/option/columns.data
            # col['data'] can be an integer (return list) or string (return dictionary)
            # we only check for the first column definition here as there is no way to return list and dictionary
            # at once
            self.is_data_list = True
            if self.columns_data:
                self.is_data_list = False
                try:
                    int(self.columns_data[0]["data"])
                    self.is_data_list = True
                except ValueError:
                    pass

            # prepare list of columns to be returned
            self._columns = self.get_columns()

            # prepare initial queryset
            qs = self.get_initial_queryset()

            # store the total number of records (before filtering)
            total_records = self.count_records(qs=qs)

            # apply filters
            qs = self.filter_queryset(qs)

            # number of records after filtering
            total_display_records = self.count_records(qs=qs)

            # apply ordering
            qs = self.ordering(qs)

            # apply pagintion
            qs = self.paging(qs)

            # prepare output data
            if self.pre_camel_case_notation:
                aa_data = self.prepare_results(qs)

                ret = {
                    "sEcho": int(self._querydict.get("sEcho", 0)),
                    "iTotalRecords": total_records,
                    "iTotalDisplayRecords": total_display_records,
                    "aaData": aa_data,
                }
            else:
                data = self.prepare_results(qs)

                ret = {
                    "draw": int(self._querydict.get("draw", 0)),
                    "recordsTotal": total_records,
                    "recordsFiltered": total_display_records,
                    "data": data,
                }
            return ret
        except Exception as e:
            return self.handle_exception(e)


class BaseDatatableView(DatatableMixin, JSONResponseView):
    pass