from django.contrib import admin
from django.contrib.auth import get_user_model

from lunarlink.models import Project

# Register your models here.

Users = get_user_model()


class ProjectAdmin(admin.ModelAdmin):
    # 列表要显示的字段
    list_display = (
        "name",
        "responsible",
        "get_groups",
        "get_creator_name",
        "get_updater_name",
        "create_time",
        "update_time",
    )

    def get_creator_name(self, obj):
        """返回项目创建者的名字"""
        return obj.creator.name if obj.creator else "-"

    def get_updater_name(self, obj):
        """返回项目创建者的名字"""
        user = Users.objects.filter(id=obj.updater).first()
        return user.name if user else "-"

    get_creator_name.short_description = "创建人"
    get_updater_name.short_description = "更新人"

    def get_groups(self, obj):
        """返回项目所属的所有分组"""
        return ", ".join([group.name for group in obj.groups.all()])

    get_groups.short_description = "所属分组"

    def has_add_permission(self, request):
        return False  # 移除增加按钮

    def has_delete_permission(self, request, obj=None):
        return False  # 移除删除按钮

    # 指定在编辑页面上要显示的字段
    fields = [
        "name",
        "desc",
        "responsible",
        "yapi_base_url",
        "yapi_openapi_token",
        "jira_project_key",
        "jira_bearer_token",
        "groups",
    ]

    # 指定不可编辑的字段
    readonly_fields = ("responsible",)

    # 为ManyToMany字段提供一个水平滚动选择器
    filter_horizontal = ("groups",)

    def get_queryset(self, request):
        # 获取原始的queryset
        qs = super().get_queryset(request)

        # 如果是超级管理员，返回所有项目
        if request.user.is_superuser:
            return qs

        # 获取当前用户所在的所有分组
        user_groups = request.user.groups.all()

        # 基于分组来过滤项目
        return qs.filter(groups__in=user_groups).distinct()


admin.site.register(Project, ProjectAdmin)