from django.contrib import admin
from django.contrib.gis.admin import GISModelAdmin
from .models.company import Company
from .models.rides import Ride
from .models.location import RideLocation


class ManufacturerAdmin(admin.ModelAdmin):
    list_display = ("name", "headquarters", "website", "rides_count")
    search_fields = ("name",)

    def get_queryset(self, request):
        return super().get_queryset(request).filter(roles__contains=["MANUFACTURER"])


class DesignerAdmin(admin.ModelAdmin):
    list_display = ("name", "headquarters", "website")
    search_fields = ("name",)

    def get_queryset(self, request):
        return super().get_queryset(request).filter(roles__contains=["DESIGNER"])


class RideLocationInline(admin.StackedInline):
    """Inline admin for RideLocation"""

    model = RideLocation
    extra = 0
    fields = (
        "park_area",
        "point",
        "entrance_notes",
        "accessibility_notes",
    )


class RideLocationAdmin(GISModelAdmin):
    """Admin for standalone RideLocation management"""

    list_display = ("ride", "park_area", "has_coordinates", "created_at")
    list_filter = ("park_area", "created_at")
    search_fields = ("ride__name", "park_area", "entrance_notes")
    readonly_fields = (
        "latitude",
        "longitude",
        "coordinates",
        "created_at",
        "updated_at",
    )
    fieldsets = (
        ("Ride", {"fields": ("ride",)}),
        (
            "Location Information",
            {
                "fields": (
                    "park_area",
                    "point",
                    "latitude",
                    "longitude",
                    "coordinates",
                ),
                "description": "Optional coordinates - not all rides need precise location tracking",
            },
        ),
        (
            "Navigation Notes",
            {
                "fields": ("entrance_notes", "accessibility_notes"),
            },
        ),
        (
            "Metadata",
            {"fields": ("created_at", "updated_at"), "classes": ("collapse",)},
        ),
    )

    def latitude(self, obj):
        return obj.latitude

    latitude.short_description = "Latitude"

    def longitude(self, obj):
        return obj.longitude

    longitude.short_description = "Longitude"


class RideAdmin(admin.ModelAdmin):
    """Enhanced Ride admin with location inline"""

    inlines = [RideLocationInline]


admin.site.register(Company)
admin.site.register(Ride, RideAdmin)
admin.site.register(RideLocation, RideLocationAdmin)