from django.utils import timezone
from rest_framework.response import Response
from rest_framework.decorators import action

from rbac.permissions import RBACPermission
from common.api import JMSModelViewSet
from common.permissions import IsValidUser
from ..serializers import SSHKeySerializer
from users.notifications import ResetPublicKeySuccessMsg


class SSHkeyViewSet(JMSModelViewSet):
    serializer_class = SSHKeySerializer
    permission_classes = [IsValidUser]
    filterset_fields = ('name', 'is_active')
    search_fields = ('name',)

    def get_queryset(self):
        return self.request.user.ssh_keys.all()

    def perform_update(self, serializer):
        super().perform_update(serializer)
        ResetPublicKeySuccessMsg(self.request.user, self.request).publish_async()