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

from . import models
from .models import brokers as broker_models
from .auth.v1.models import OIDCConfig


class TypeFilter(admin.SimpleListFilter):
    title = 'Type'
    parameter_name = 'user_type'

    def lookups(self, request, model_admin):
        return models.User.TYPE_CHOICES

    def queryset(self, request, queryset):
        if self.value():
            return queryset.filter(user_type=self.value())
        return queryset


class ClientAdmin(UserAdmin):
    list_filter = (TypeFilter,)

admin.site.register(OIDCConfig)
admin.site.register(broker_models.RabbitMQInstance)

admin.site.register(models.User, ClientAdmin)
admin.site.unregister(Group)
admin.site.register(Group)
admin.site.register(models.Site)
admin.site.register(models.Service)
admin.site.register(models.SSHPublicKey)

admin.site.register(models.NewDeployment)
admin.site.register(models.NewDeploymentStateItem)