# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
#
import json
from django.contrib import admin
from django.db.models.functions import Now
from django.db.models import Q
from django import forms
from django.utils.html import conditional_escape
from django.utils.safestring import mark_safe
from fnpdjango.actions import export_as_csv_action
from modeltranslation.admin import TranslationAdmin
import annoy.models
from wolnelektury.utils import YesNoFilter
from . import models


class SingleAmountInline(admin.TabularInline):
    model = models.SingleAmount


class MonthlyAmountInline(admin.TabularInline):
    model = models.MonthlyAmount


@admin.register(models.Club)
class ClubAdmin(admin.ModelAdmin):
    inlines = [
        SingleAmountInline,
        MonthlyAmountInline
    ]


class PayUOrderInline(admin.TabularInline):
    model = models.PayUOrder
    fields = ['order_id', 'status', 'customer_ip']
    readonly_fields = fields
    extra = 0
    show_change_link = True
    can_delete = False

    def has_add_permission(self, request, obj):
        return False


class PayUCardTokenInline(admin.TabularInline):
    model = models.PayUCardToken
    fields = ['created_at', 'disposable_token', 'reusable_token']
    readonly_fields = fields
    extra = 0
    show_change_link = True
    can_delete = False
    show_change_link = True

    def has_add_permission(self, request, obj):
        return False


class PayedFilter(YesNoFilter):
    title = 'płatność zakończona'
    parameter_name = 'payed'
    q = ~Q(payed_at=None)


class ExpiredFilter(YesNoFilter):
    title = 'harmonogram przedawniony'
    parameter_name = 'expired'
    q = Q(expires_at__isnull=False, expires_at__lt=Now())


class ActiveFilter(YesNoFilter):
    title = 'płatność aktualna'
    parameter_name = 'active'
    q = Q(expires_at__gt=Now())


class ScheduleForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['email'].required = False
        self.fields['method'].required = False
        self.fields['consent'].required = False

    class Meta:
        model = models.Schedule
        fields = '__all__'


class SourceFilter(admin.SimpleListFilter):
    title = 'Źródło' # display title
    parameter_name = 'source'
    template = "admin/long_filter.html"

    def lookups(self, request, model_admin):
        lookups = [
            (m, m) for m in
            model_admin.model.objects.exclude(source='').values_list('source', flat=True).distinct()[:10]
        ]
        return lookups

    def queryset(self, request, queryset):
        return queryset
    

class CrisisFilter(admin.SimpleListFilter):
    title = 'czas zbiórki kryzysowej'
    parameter_name = 'crisis'

    def lookups(self, request, model_admin):
        lookups = [
            (b.id, '%s — %s' % (b.since, b.until)) for b in
            annoy.models.Banner.objects.filter(place='crisis')
        ]
        return lookups

    def queryset(self, request, queryset):
        bid = self.value()
        if not bid:
            return
        try:
            b = annoy.models.Banner.objects.get(id=self.value())
        except annoy.models.Banner.DoesNotExist:
            return
        return queryset.filter(
            started_at__gte=b.since,
            started_at__lte=b.until
        )



class ScheduleAdmin(admin.ModelAdmin):
    form = ScheduleForm

    list_display = [
        'email', 'started_at', 'payed_at', 'expires_at', 'amount', 'monthly', 'yearly', 'is_cancelled',
        'method'
    ]
    list_display_links = ['email', 'started_at']
    search_fields = ['email', 'source']
    list_filter = [
        'is_cancelled', 'monthly', 'yearly', 'method',
        PayedFilter, ActiveFilter, ExpiredFilter,
        SourceFilter, CrisisFilter
    ]
    filter_horizontal = ['consent']
    date_hierarchy = 'started_at'
    raw_id_fields = ['membership']
    inlines = [PayUOrderInline, PayUCardTokenInline]
    actions = [export_as_csv_action()]

admin.site.register(models.Schedule, ScheduleAdmin)


class ScheduleInline(admin.TabularInline):
    model = models.Schedule
    fields = ['email', 'amount', 'is_cancelled', 'started_at', 'payed_at', 'expires_at', 'email_sent']
    readonly_fields = fields
    extra = 0
    show_change_link = True
    can_delete = False

    def has_add_permission(self, request, obj):
        return False


class MembershipAdmin(admin.ModelAdmin):
    list_display = ['user', 'manual', 'updated_at', 'notes']
    list_filter = ['manual']
    date_hierarchy = 'updated_at'
    raw_id_fields = ['user']
    search_fields = ['user__username', 'user__email', 'schedule__email', 'notes']
    inlines = [ScheduleInline]

admin.site.register(models.Membership, MembershipAdmin)


admin.site.register(models.ReminderEmail, TranslationAdmin)


class PayUNotificationAdmin(admin.ModelAdmin):
    list_display = ['received_at', 'order']
    fields = ['received_at', 'order', 'body_']
    readonly_fields = ['received_at', 'body_']
    raw_id_fields = ['order']

    def body_(self, obj):
        return mark_safe(
                "<pre>" +
                conditional_escape(json.dumps(json.loads(obj.body), indent=4))
                + "</pre>")


admin.site.register(models.PayUNotification, PayUNotificationAdmin)


class PayUNotificationInline(admin.TabularInline):
    model = models.PayUNotification
    fields = ['received_at', 'body_']
    readonly_fields = fields
    extra = 0
    show_change_link = True
    can_delete = False

    def body_(self, obj):
        return mark_safe(
                "<pre>" +
                conditional_escape(json.dumps(json.loads(obj.body), indent=4))
                + "</pre>")

    def has_add_permission(self, request, obj):
        return False


class PayUOrderAdmin(admin.ModelAdmin):
    list_display = ['schedule']
    raw_id_fields = ['schedule']
    inlines = [PayUNotificationInline]


admin.site.register(models.PayUOrder, PayUOrderAdmin)


admin.site.register(models.Ambassador)


    

@admin.register(models.Consent)
class ConsentAdmin(admin.ModelAdmin):
    list_display = ['text', 'order', 'active', 'required']

    def get_readonly_fields(self, request, obj=None):
        if obj:
            return ['text']
        else:
            return []