# -*- coding: utf-8 -*-
#
from django.utils.translation import ugettext_lazy as _

from rest_framework import serializers
from rest_framework_bulk import BulkListSerializer, BulkSerializerMixin

from common.utils import signer, validate_ssh_public_key
from .models import User, UserGroup


# class UserDetailSerializer(BulkSerializerMixin, serializers.ModelSerializer):
#     class Meta:
#         model = User
#         fields = ['avatar', 'wechat', 'phone', 'enable_otp', 'comment', 'is_active', 'name']


class UserSerializer(BulkSerializerMixin, serializers.ModelSerializer):
    group_display = serializers.SerializerMethodField()
    active_display = serializers.SerializerMethodField()
    groups = serializers.PrimaryKeyRelatedField(many=True, queryset=UserGroup.objects.all())

    class Meta:
        model = User
        list_serializer_class = BulkListSerializer
        exclude = ['first_name', 'last_name', 'password', '_private_key', '_public_key']

    def get_field_names(self, declared_fields, info):
        fields = super(UserSerializer, self).get_field_names(declared_fields, info)
        fields.extend(['group_display', 'get_role_display'])
        return fields

    @staticmethod
    def get_group_display(obj):
        return " ".join([group.name for group in obj.groups.all()])

    @staticmethod
    def get_active_display(obj):
        # TODO: user active state
        return not (obj.is_expired and obj.is_active)


class UserPKUpdateSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = ['id', '_public_key']

    @staticmethod
    def validate__public_key(value):
        if not validate_ssh_public_key(value):
            print('Not a valid key')
            print(value)
            raise serializers.ValidationError(_('Not a valid ssh public key'))
        return value


class UserAndGroupSerializer(serializers.ModelSerializer):
    groups = serializers.PrimaryKeyRelatedField(many=True, queryset=UserGroup.objects.all())

    class Meta:
        model = User
        fields = ['id', 'groups']


# class GroupDetailSerializer(serializers.ModelSerializer):
#     class Meta:
#         model = UserGroup
#         fields = ['id', 'name', 'comment', 'date_created', 'created_by', 'users']


# class UserBulkUpdateSerializer(BulkSerializerMixin, serializers.ModelSerializer):
#     group_display = serializers.SerializerMethodField()
#     active_display = serializers.SerializerMethodField()
#     groups = serializers.PrimaryKeyRelatedField(many=True, queryset=UserGroup.objects.all())
#
#     class Meta(object):
#         model = User
#         list_serializer_class = BulkListSerializer
#         fields = ['id', 'is_active', 'username', 'name', 'email', 'role', 'avatar',
#                   'enable_otp', 'comment', 'groups', 'get_role_display',
#                   'group_display', 'active_display']
#
#     @staticmethod
#     def get_group_display(obj):
#         return " ".join([group.name for group in obj.groups.all()])
#
#     @staticmethod
#     def get_active_display(obj):
#         TODO: user active state
        # return not (obj.is_expired and obj.is_active)
#
#
# class GroupBulkUpdateSerializer(BulkSerializerMixin, serializers.ModelSerializer):
#     user_amount = serializers.SerializerMethodField()
#
#     class Meta:
#         model = UserGroup
#         list_serializer_class = BulkListSerializer
#         fields = ['id', 'name', 'comment', 'user_amount']
#
#     @staticmethod
#     def get_user_amount(obj):
#         return obj.users.count()
#