icurfer
c93ae27f32
All checks were successful
Build And Test / build-and-push (push) Successful in 2m52s
121 lines
5.3 KiB
Python
121 lines
5.3 KiB
Python
# views.py
|
|
import logging # ✅ 로그 추가
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework_simplejwt.views import TokenObtainPairView
|
|
|
|
from .serializers import RegisterSerializer, CustomTokenObtainPairSerializer
|
|
|
|
logger = logging.getLogger(__name__) # ✅ 로그 추가
|
|
|
|
class RegisterView(APIView):
|
|
def post(self, request):
|
|
logger.info("RegisterView POST called. data=%s", request.data) # ✅ 로그 추가
|
|
serializer = RegisterSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
user = serializer.save()
|
|
logger.info("User registered: %s", user.email if hasattr(user, "email") else str(user)) # ✅ 로그 추가
|
|
return Response({"message": "User registered successfully."}, status=status.HTTP_201_CREATED)
|
|
logger.warning("Registration failed: %s", serializer.errors) # ✅ 로그 추가
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class MeView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get(self, request):
|
|
logger.debug("MeView GET called by user: %s", request.user) # ✅ 로그 추가
|
|
user = request.user
|
|
serializer = RegisterSerializer(user)
|
|
return Response(serializer.data)
|
|
|
|
def put(self, request):
|
|
logger.info("MeView PUT called by user: %s", request.user) # ✅ 로그 추가
|
|
user = request.user
|
|
serializer = RegisterSerializer(user, data=request.data, partial=True)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
logger.info("User profile updated: %s", request.user) # ✅ 로그 추가
|
|
return Response(serializer.data)
|
|
logger.warning("Profile update failed: %s", serializer.errors) # ✅ 로그 추가
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class CustomTokenObtainPairView(TokenObtainPairView):
|
|
serializer_class = CustomTokenObtainPairSerializer
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
logger.info("Login attempt: %s", request.data.get("email")) # ✅ 로그 추가 (비밀번호 제외)
|
|
return super().post(request, *args, **kwargs)
|
|
|
|
|
|
class SSHKeyUploadView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def post(self, request):
|
|
logger.info("SSHKeyUploadView POST called by user: %s", request.user) # ✅ 로그 추가
|
|
private_key = request.data.get("private_key")
|
|
key_name = request.data.get("key_name")
|
|
|
|
if not private_key or not key_name:
|
|
logger.warning("Missing SSH key or key name in request") # ✅ 로그 추가
|
|
return Response(
|
|
{"error": "private_key와 key_name 모두 필요합니다."},
|
|
status=status.HTTP_400_BAD_REQUEST
|
|
)
|
|
|
|
user = request.user
|
|
try:
|
|
user.save_private_key(private_key)
|
|
user.encrypted_private_key_name = key_name
|
|
user.save(update_fields=["encrypted_private_key", "encrypted_private_key_name"])
|
|
logger.info("SSH key saved for user: %s", user) # ✅ 로그 추가
|
|
return Response({"message": "SSH key 저장 완료."}, status=201)
|
|
except Exception as e:
|
|
logger.exception("SSH key 저장 중 오류 발생") # ✅ 로그 추가
|
|
return Response({"error": f"암호화 또는 저장 실패: {str(e)}"}, status=500)
|
|
|
|
def delete(self, request):
|
|
logger.info("SSHKeyUploadView DELETE called by user: %s", request.user) # ✅ 로그 추가
|
|
user = request.user
|
|
user.encrypted_private_key = None
|
|
user.encrypted_private_key_name = None
|
|
user.last_used_at = None
|
|
user.save(update_fields=["encrypted_private_key", "encrypted_private_key_name", "last_used_at"])
|
|
logger.info("SSH key deleted for user: %s", user) # ✅ 로그 추가
|
|
return Response({"message": "SSH key deleted."}, status=200)
|
|
|
|
|
|
class SSHKeyInfoView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get(self, request):
|
|
logger.debug("SSHKeyInfoView GET called by user: %s", request.user) # ✅ 로그 추가
|
|
user = request.user
|
|
return Response({
|
|
"has_key": bool(user.encrypted_private_key),
|
|
"encrypted_private_key_name": user.encrypted_private_key_name,
|
|
"last_used_at": user.last_used_at
|
|
})
|
|
|
|
|
|
class SSHKeyRetrieveView(APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get(self, request):
|
|
logger.info("SSHKeyRetrieveView GET called by user: %s", request.user) # ✅ 로그 추가
|
|
user = request.user
|
|
if not user.encrypted_private_key:
|
|
logger.warning("User tried to retrieve non-existent SSH key: %s", user) # ✅ 로그 추가
|
|
return Response({"error": "SSH 키가 등록되어 있지 않습니다."}, status=404)
|
|
|
|
try:
|
|
decrypted_key = user.decrypt_private_key()
|
|
logger.info("SSH key successfully decrypted for user: %s", user) # ✅ 로그 추가
|
|
return Response({"ssh_key": decrypted_key})
|
|
except Exception as e:
|
|
logger.exception("SSH key 복호화 실패") # ✅ 로그 추가
|
|
return Response({"error": f"복호화 실패: {str(e)}"}, status=500)
|