mirror of https://github.com/jumpserver/jumpserver
104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from common.utils import get_object_or_none, random_string
|
|
from users.models import User
|
|
from users.serializers import UserProfileSerializer
|
|
from ..models import AccessKey, TempToken
|
|
|
|
__all__ = [
|
|
'AccessKeySerializer', 'BearerTokenSerializer',
|
|
'SSOTokenSerializer', 'TempTokenSerializer',
|
|
]
|
|
|
|
|
|
class AccessKeySerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = AccessKey
|
|
fields = ['id', 'secret', 'is_active', 'date_created']
|
|
read_only_fields = ['id', 'secret', 'date_created']
|
|
|
|
|
|
class BearerTokenSerializer(serializers.Serializer):
|
|
username = serializers.CharField(allow_null=True, required=False, write_only=True)
|
|
password = serializers.CharField(write_only=True, allow_null=True,
|
|
required=False, allow_blank=True)
|
|
public_key = serializers.CharField(write_only=True, allow_null=True,
|
|
allow_blank=True, required=False)
|
|
token = serializers.CharField(read_only=True)
|
|
keyword = serializers.SerializerMethodField()
|
|
date_expired = serializers.DateTimeField(read_only=True)
|
|
user = UserProfileSerializer(read_only=True)
|
|
|
|
@staticmethod
|
|
def get_keyword(obj):
|
|
return 'Bearer'
|
|
|
|
def update_last_login(self, user):
|
|
user.last_login = timezone.now()
|
|
user.save(update_fields=['last_login'])
|
|
|
|
def get_request_user(self):
|
|
request = self.context.get('request')
|
|
if request.user and request.user.is_authenticated:
|
|
user = request.user
|
|
else:
|
|
user_id = request.session.get('user_id')
|
|
user = get_object_or_none(User, pk=user_id)
|
|
if not user:
|
|
raise serializers.ValidationError(
|
|
"user id {} not exist".format(user_id)
|
|
)
|
|
return user
|
|
|
|
def create(self, validated_data):
|
|
request = self.context.get('request')
|
|
user = self.get_request_user()
|
|
|
|
token, date_expired = user.create_bearer_token(request)
|
|
self.update_last_login(user)
|
|
|
|
instance = {
|
|
"token": token,
|
|
"date_expired": date_expired,
|
|
"user": user
|
|
}
|
|
return instance
|
|
|
|
|
|
class SSOTokenSerializer(serializers.Serializer):
|
|
username = serializers.CharField(write_only=True)
|
|
login_url = serializers.CharField(read_only=True)
|
|
next = serializers.CharField(write_only=True, allow_blank=True, required=False, allow_null=True)
|
|
|
|
|
|
class TempTokenSerializer(serializers.ModelSerializer):
|
|
is_valid = serializers.BooleanField(label=_("Is valid"), read_only=True)
|
|
|
|
class Meta:
|
|
model = TempToken
|
|
fields = [
|
|
'id', 'username', 'secret', 'verified', 'is_valid',
|
|
'date_created', 'date_updated', 'date_verified',
|
|
'date_expired',
|
|
]
|
|
read_only_fields = fields
|
|
|
|
def create(self, validated_data):
|
|
request = self.context.get('request')
|
|
if not request or not request.user:
|
|
raise PermissionError()
|
|
|
|
secret = random_string(36)
|
|
username = request.user.username
|
|
kwargs = {
|
|
'username': username, 'secret': secret,
|
|
'date_expired': timezone.now() + timezone.timedelta(seconds=5*60),
|
|
}
|
|
token = TempToken(**kwargs)
|
|
token.save()
|
|
return token
|