# Copyright (C) 2020 by Animath
# SPDX-License-Identifier: GPL-3.0-or-later

from django.contrib import admin
from django.utils.translation import gettext_lazy as _

from .models import Note, Participation, Passage, Pool, Solution, Synthesis, Team, Tournament, Tweak


class ParticipationInline(admin.StackedInline):
    model = Participation
    extra = 0
    autocomplete_fields = ('team', 'tournament',)
    show_change_link = True


class ParticipationTabularInline(admin.TabularInline):
    model = Participation
    extra = 0
    fields = ('team', 'valid', 'final',)
    readonly_fields = ('team',)
    ordering = ('final', 'valid', 'team__trigram',)
    autocomplete_fields = ('tournament',)
    show_change_link = True


class SolutionInline(admin.TabularInline):
    model = Solution
    extra = 0
    ordering = ('problem',)
    autocomplete_fields = ('participation',)
    show_change_link = True


class SynthesisInline(admin.TabularInline):
    model = Synthesis
    extra = 0
    ordering = ('passage__solution_number', 'type',)
    autocomplete_fields = ('passage',)
    show_change_link = True


class PoolInline(admin.TabularInline):
    model = Pool
    extra = 0
    autocomplete_fields = ('tournament', 'participations', 'jury_president', 'juries',)
    show_change_link = True


class PassageInline(admin.TabularInline):
    model = Passage
    extra = 0
    ordering = ('position',)
    autocomplete_fields = ('defender', 'opponent', 'reporter',)
    show_change_link = True


class NoteInline(admin.TabularInline):
    model = Note
    extra = 0
    autocomplete_fields = ('jury',)
    show_change_link = True


class TweakInline(admin.TabularInline):
    model = Tweak
    extra = 0
    autocomplete_fields = ('participation', 'pool',)
    show_change_link = True


@admin.register(Team)
class TeamAdmin(admin.ModelAdmin):
    list_display = ('name', 'trigram', 'tournament', 'valid', 'final',)
    search_fields = ('name', 'trigram',)
    list_filter = ('participation__valid', 'participation__tournament', 'participation__final',)
    inlines = (ParticipationInline,)

    @admin.display(description=_("tournament"))
    def tournament(self, record):
        return record.participation.tournament

    @admin.display(description=_("valid"), boolean=True)
    def valid(self, team):
        return team.participation.valid

    @admin.display(description=_("selected for final"), boolean=True)
    def final(self, team):
        return team.participation.final


@admin.register(Participation)
class ParticipationAdmin(admin.ModelAdmin):
    list_display = ('team', 'tournament', 'valid', 'final',)
    search_fields = ('team__name', 'team__trigram',)
    list_filter = ('valid', 'tournament',)
    autocomplete_fields = ('team', 'tournament',)
    inlines = (SolutionInline, SynthesisInline,)


@admin.register(Pool)
class PoolAdmin(admin.ModelAdmin):
    list_display = ('__str__', 'tournament', 'round', 'letter', 'room', 'teams', 'jury_president',)
    list_filter = ('tournament', 'round', 'letter', 'room',)
    search_fields = ('participations__team__name', 'participations__team__trigram',)
    autocomplete_fields = ('tournament', 'participations', 'jury_president', 'juries',)
    inlines = (PassageInline, TweakInline,)

    @admin.display(description=_("teams"))
    def teams(self, record: Pool):
        return ', '.join(p.team.trigram for p in record.participations.all())


@admin.register(Passage)
class PassageAdmin(admin.ModelAdmin):
    list_display = ('__str__', 'defender_trigram', 'solution_number', 'opponent_trigram', 'reporter_trigram',
                    'pool_abbr', 'position', 'tournament')
    list_filter = ('pool__tournament', 'pool__round', 'pool__letter', 'solution_number',)
    search_fields = ('pool__participations__team__name', 'pool__participations__team__trigram',)
    ordering = ('pool__tournament', 'pool__round', 'pool__letter', 'position',)
    autocomplete_fields = ('pool', 'defender', 'opponent', 'reporter',)
    inlines = (NoteInline,)

    @admin.display(description=_("defender"), ordering='defender__team__trigram')
    def defender_trigram(self, record: Passage):
        return record.defender.team.trigram

    @admin.display(description=_("opponent"), ordering='opponent__team__trigram')
    def opponent_trigram(self, record: Passage):
        return record.opponent.team.trigram

    @admin.display(description=_("reporter"), ordering='reporter__team__trigram')
    def reporter_trigram(self, record: Passage):
        return record.reporter.team.trigram

    @admin.display(description=_("pool"), ordering='pool__letter')
    def pool_abbr(self, record):
        return f"{record.pool.short_name}"

    @admin.display(description=_("tournament"), ordering='pool__tournament__name')
    def tournament(self, record: Passage):
        return record.pool.tournament


@admin.register(Note)
class NoteAdmin(admin.ModelAdmin):
    list_display = ('passage', 'pool', 'jury', 'defender_writing', 'defender_oral',
                    'opponent_writing', 'opponent_oral', 'reporter_writing', 'reporter_oral',)
    list_filter = ('passage__pool__letter', 'passage__solution_number', 'jury',
                   'defender_writing', 'defender_oral', 'opponent_writing', 'opponent_oral',
                   'reporter_writing', 'reporter_oral')
    search_fields = ('jury__user__last_name', 'jury__user__first_name', 'passage__defender__team__trigram',)
    autocomplete_fields = ('jury', 'passage',)

    @admin.display(description=_("pool"))
    def pool(self, record):
        return record.passage.pool.short_name


@admin.register(Solution)
class SolutionAdmin(admin.ModelAdmin):
    list_display = ('team', 'tournament', 'problem', 'final_solution',)
    list_filter = ('problem', 'participation__tournament', 'final_solution',)
    search_fields = ('participation__team__name', 'participation__team__trigram',)
    autocomplete_fields = ('participation',)

    @admin.display(ordering='participation__team', description=_("team"))
    def team(self, record):
        return record.participation.team

    @admin.display(ordering='participation__tournament__name', description=_("tournament"))
    def tournament(self, record):
        return Tournament.final_tournament() if record.final_solution else record.participation.tournament


@admin.register(Synthesis)
class SynthesisAdmin(admin.ModelAdmin):
    list_display = ('participation', 'type', 'defender', 'passage',)
    list_filter = ('participation__tournament', 'type', 'passage__solution_number',)
    search_fields = ('participation__team__name', 'participation__team__trigram',)
    autocomplete_fields = ('participation', 'passage',)

    @admin.display(description=_("defender"))
    def defender(self, record: Synthesis):
        return record.passage.defender

    @admin.display(description=_("problem"))
    def problem(self, record: Synthesis):
        return record.passage.solution_number


@admin.register(Tournament)
class TournamentAdmin(admin.ModelAdmin):
    list_display = ('name', 'date_start', 'date_end',)
    search_fields = ('name',)
    ordering = ('date_start', 'name',)
    autocomplete_fields = ('organizers',)
    inlines = (ParticipationTabularInline, PoolInline,)


@admin.register(Tweak)
class TweakAdmin(admin.ModelAdmin):
    list_display = ('participation', 'pool', 'diff',)
    list_filter = ('pool__tournament', 'pool__round',)
    search_fields = ('participation__team__name', 'participation__team__trigram',)
    autocomplete_fields = ('participation', 'pool',)