from django.contrib import admin

from .models import ClusterSettings, Worker, WorkerStatus


@admin.register(ClusterSettings)
class ClusterSettingsAdmin(admin.ModelAdmin):
    list_display = ('name', 'enabled', 'cluster_mode', 'restart_mode', 'config_version', 'created', 'updated')
    list_filter = ('enabled', 'cluster_mode', 'restart_mode', 'primary_enable_wireguard')


@admin.register(Worker)
class WorkerAdmin(admin.ModelAdmin):
    list_display = ('name', 'enabled', 'ip_address', 'country', 'city', 'ip_lock', 'created', 'updated')
    list_filter = ('enabled', 'ip_lock', 'country', 'force_reload', 'force_restart')


@admin.register(WorkerStatus)
class WorkerStatusAdmin(admin.ModelAdmin):
    list_display = ('worker', 'last_seen', 'config_version', 'config_pending', 'last_reload', 'last_restart')
    list_filter = ('config_pending', 'last_seen', 'last_reload', 'last_restart')