from rest_framework import serializers

from apps.accounts.models import Team, TeamMember, User
from apps.ai.models import QualityWord


class QualityWordSerializer(serializers.ModelSerializer):
    class Meta:
        model = QualityWord
        fields = ["id", "stage", "slot", "text", "sort", "enabled", "created_at"]
        read_only_fields = ["id", "created_at"]


class AdminTeamSerializer(serializers.ModelSerializer):
    owner_username = serializers.CharField(source="owner.username", read_only=True, default=None)
    member_count = serializers.SerializerMethodField()
    balance = serializers.SerializerMethodField()

    class Meta:
        model = Team
        fields = ["id", "name", "status", "owner", "owner_username", "member_count", "balance", "created_at"]
        read_only_fields = fields

    def get_member_count(self, obj):
        anno = getattr(obj, "member_count_anno", None)
        return anno if anno is not None else obj.members.count()

    def get_balance(self, obj):
        acct = getattr(obj, "credit_account", None)
        return str(acct.balance) if acct is not None else "0"


class AdminTeamMemberSerializer(serializers.ModelSerializer):
    username = serializers.CharField(source="user.username", read_only=True)
    user_status = serializers.CharField(source="user.status", read_only=True)

    class Meta:
        model = TeamMember
        fields = ["id", "username", "role", "status", "user_status", "monthly_credit_limit"]
        read_only_fields = fields


class AdminUserSerializer(serializers.ModelSerializer):
    teams = serializers.SerializerMethodField()

    class Meta:
        model = User
        fields = ["id", "username", "status", "is_platform_admin", "date_joined", "teams"]
        read_only_fields = fields

    def get_teams(self, obj):
        return [
            {"team_id": str(m.team_id), "team_name": m.team.name, "role": m.role}
            for m in obj.team_memberships.select_related("team").all()
        ]