from django.contrib import admin
from .models import File, Message, PublicMessage, PublicFile, Friendship

@admin.register(File)
class FileAdmin(admin.ModelAdmin):
    list_display = ('file', 'owner', 'created_at')
    list_filter = ('owner', 'created_at')
    search_fields = ('file', 'description')
    raw_id_fields = ('shared_with',)

@admin.register(Message)
class MessageAdmin(admin.ModelAdmin):
    list_display = ('sender', 'created_at')
    list_filter = ('sender', 'created_at')
    search_fields = ('content',)
    raw_id_fields = ('recipients',)

@admin.register(PublicMessage)
class PublicMessageAdmin(admin.ModelAdmin):
    list_display = ('name', 'created_at', 'ip_address')
    list_filter = ('created_at',)
    search_fields = ('content',)

@admin.register(PublicFile)
class PublicFileAdmin(admin.ModelAdmin):
    list_display = ('name', 'file', 'created_at', 'ip_address')
    list_filter = ('created_at',)
    search_fields = ('file', 'description')

@admin.register(Friendship)
class FriendshipAdmin(admin.ModelAdmin):
    list_display = ('from_user', 'to_user', 'status', 'created_at')
    list_filter = ('status', 'created_at')
    search_fields = ('from_user__username', 'to_user__username')