from django.contrib import admin
from django.utils.html import format_html
from .models import M3UAccount, M3UFilter, ServerGroup, UserAgent
import json


class M3UFilterInline(admin.TabularInline):
    model = M3UFilter
    extra = 1
    verbose_name = "M3U Filter"
    verbose_name_plural = "M3U Filters"


@admin.register(M3UAccount)
class M3UAccountAdmin(admin.ModelAdmin):
    list_display = (
        "name",
        "server_url",
        "server_group",
        "max_streams",
        "priority",
        "is_active",
        "user_agent_display",
        "uploaded_file_link",
        "created_at",
        "updated_at",
    )
    list_filter = ("is_active", "server_group")
    search_fields = ("name", "server_url", "server_group__name")
    inlines = [M3UFilterInline]
    actions = ["activate_accounts", "deactivate_accounts"]

    # Handle both ForeignKey and ManyToManyField cases for UserAgent
    def user_agent_display(self, obj):
        if hasattr(obj, "user_agent"):  # ForeignKey case
            return obj.user_agent.user_agent if obj.user_agent else "None"
        elif hasattr(obj, "user_agents"):  # ManyToManyField case
            return ", ".join([ua.user_agent for ua in obj.user_agents.all()]) or "None"
        return "None"

    user_agent_display.short_description = "User Agent(s)"

    def vod_enabled_display(self, obj):
        """Display whether VOD is enabled for this account"""
        if obj.custom_properties:
            custom_props = obj.custom_properties or {}
            return "Yes" if custom_props.get('enable_vod', False) else "No"
        return "No"
    vod_enabled_display.short_description = "VOD Enabled"
    vod_enabled_display.boolean = True

    def uploaded_file_link(self, obj):
        if obj.uploaded_file:
            return format_html(
                "<a href='{}' target='_blank'>Download M3U</a>", obj.uploaded_file.url
            )
        return "No file uploaded"

    uploaded_file_link.short_description = "Uploaded File"

    @admin.action(description="Activate selected accounts")
    def activate_accounts(self, request, queryset):
        queryset.update(is_active=True)

    @admin.action(description="Deactivate selected accounts")
    def deactivate_accounts(self, request, queryset):
        queryset.update(is_active=False)

    # Add ManyToManyField for Django Admin (if applicable)
    if hasattr(M3UAccount, "user_agents"):
        filter_horizontal = ("user_agents",)  # Only for ManyToManyField


@admin.register(M3UFilter)
class M3UFilterAdmin(admin.ModelAdmin):
    list_display = ("m3u_account", "filter_type", "regex_pattern", "exclude")
    list_filter = ("filter_type", "exclude")
    search_fields = ("regex_pattern",)
    ordering = ("m3u_account",)


@admin.register(ServerGroup)
class ServerGroupAdmin(admin.ModelAdmin):
    list_display = ("name",)
    search_fields = ("name",)