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

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', '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 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',)