from django.contrib import admin
from django.contrib.auth.admin import UserAdmin

from .models import (
    AuditLog,
    InviteCode,
    InviteCodeUsage,
    User,
    UserSession,
    VisitorMigration,
)


@admin.register(User)
class HulumathUserAdmin(UserAdmin):
    fieldsets = UserAdmin.fieldsets + (
        ("葫芦数学", {"fields": ("nickname", "avatar_url", "bio", "track", "rating")}),
    )
    add_fieldsets = UserAdmin.add_fieldsets + (
        ("葫芦数学", {"fields": ("nickname", "track")}),
    )
    list_display = ("username", "nickname", "track", "rating", "is_active", "date_joined")


@admin.register(InviteCode)
class InviteCodeAdmin(admin.ModelAdmin):
    list_display = ("code", "group", "used_count", "max_uses", "expires_at", "is_active")
    list_filter = ("is_active", "group")
    search_fields = ("code", "group")


admin.site.register(InviteCodeUsage)
admin.site.register(UserSession)
admin.site.register(VisitorMigration)
admin.site.register(AuditLog)