Merge: 카카오 로그인, 구글 소셜로그인 기능 구현

[Feature/social-login] 카카오 로그인, 구글 소셜로그인 기능 구현

Author: yyysolhhh

apps/chat/consumers.py

@@ -40,10 +40,10 @@ async def connect(self) -> None:
             chatroom = await self.get_chatroom(self.chatroom_id)
 
             if chatroom is None:
-                await self.close(code=1008, reason="해당 채팅방이 존재하지 않습니다.")
+                raise ValueError("해당 채팅방이 존재하지 않습니다.")
             user = self.scope["user"]
             if not await database_sync_to_async(check_entered_chatroom)(chatroom, user):
-                await self.close(code=1008, reason="해당 채팅방에 존재하는 유저가 아닙니다.")
+                raise ValueError("해당 채팅방에 존재하는 유저가 아닙니다.")
             await self.channel_layer.group_add(self.chat_group_name, self.channel_name)
             await self.accept()
             if check_opponent_online(self.chat_group_name):
@@ -62,6 +62,9 @@ async def connect(self) -> None:
                         "opponent_state": "offline",
                     },
                 )
+        except ValueError as e:
+            logger.error("예외 발생: %s", e, exc_info=True)
+            await self.close(code=1008, reason=str(e))
         except Exception as e:
             logger.error("예외 발생: %s", e, exc_info=True)
             await self.close(code=1011, reason=str(e))

apps/chat/utils.py

@@ -159,33 +159,33 @@ def get_chatroom_message(chatroom_id: int) -> Any:
         # 레디스에 저장된 메시지가 30개가 넘으면 가장 마지막에 저장된 메시지부터 30개를 가져옴
         if stored_message_num >= 30:
             stored_messages = redis_conn.lrange(key, 0, 29)
-            messages = [json.loads(msg) for msg in stored_messages]
+            messages = [json.loads(msg) for msg in reversed(stored_messages)]
             return messages
 
         # 30개가 넘지않으면 레디스에 저장된 메시지들을 가져오고
         stored_messages = redis_conn.lrange(key, 0, -1)
-        messages = [json.loads(msg) for msg in stored_messages]
+        messages = [json.loads(msg) for msg in reversed(stored_messages)]
 
         # 데이터베이스에서 30 - stored_message_num을 뺀 개수만큼 가져옴
-        db_messages = Message.objects.filter(chatroom_id=chatroom_id).order_by("-created_at")
+        db_messages = Message.objects.filter(chatroom_id=chatroom_id).order_by("created_at")
 
         # 디비에 저장된 메시지가 30-stored_message_num 보다 많으면 슬라이싱해서 필요한 만큼의 데이터를 가져옴
         if db_messages.count() >= 30 - stored_message_num:
             serialized_messages = MessageSerializer(db_messages[: 30 - stored_message_num], many=True).data
-            return messages + serialized_messages  # type: ignore
+            return serialized_messages + messages  # type: ignore
 
         # 디비에 저장된 메시지가 30-stored_message_num 보다 적으면 db에 저장된 채팅방의 모든 메시지를 가져옴
-        serialized_messages = MessageSerializer(db_messages, many=True).data
-        return messages + serialized_messages  # type: ignore
+        serialized_messages = MessageSerializer(db_messages.order_by("created_at"), many=True).data
+        return serialized_messages + messages  # type: ignore
 
     # 레디스에 해당 채팅방 그룹 네임으로 지정된 키값이 없으면 데이터베이스에서 채팅 메시지를 가져옴
     db_messages = Message.objects.filter(chatroom_id=chatroom_id)
     if db_messages:
         if db_messages.count() >= 30:
-            serialized_messages = MessageSerializer(db_messages.order_by("-created_at")[:30], many=True).data
+            serialized_messages = MessageSerializer(db_messages.order_by("created_at")[:30], many=True).data
             return serialized_messages
 
-        serialized_messages = MessageSerializer(db_messages, many=True).data
+        serialized_messages = MessageSerializer(db_messages.order_by("created_at"), many=True).data
         return serialized_messages
 
     # 어디에도 데이터가 존재하지않으면 None을 반환

apps/user/urls.py

@@ -17,6 +17,8 @@
     CustomLoginView,
     CustomSignupView,
     DeleteUserView,
+    KakaoLoginView,
+    GoogleLoginView,
     SendCodeView,
 )
 
@@ -32,6 +34,8 @@
     path("password/reset/confirm/", PasswordResetConfirmView.as_view(), name="rest_password_reset_confirm"),
     # path('login/', LoginView.as_view(), name='rest_login'),
     path("login/", CustomLoginView.as_view(), name="login"),
+    path("login/social/kakao/", KakaoLoginView.as_view(), name="kakao_login"),
+    path("login/social/google/", GoogleLoginView.as_view(), name="google_login"),
     path("logout/", LogoutView.as_view(), name="rest_logout"),
     path("detail/", UserDetailsView.as_view(), name="rest_user_details"),
     path("leave/", DeleteUserView.as_view(), name="delete-user"),
@@ -42,5 +46,4 @@
     # re_path(r"^confirm-email/(?P<key>[-:\w]+)/$", CustomConfirmEmailView.as_view(), name="account_confirm_email"),
     path("send-code/", SendCodeView.as_view(), name="send-code"),
     path("confirm-email/", ConfirmEmailView.as_view(), name="confirm-email"),
-    # path("google/", GoogleLogin.as_view(), name="google_login")
 ]

apps/user/views.py

@@ -1,40 +1,23 @@
-import logging
 from typing import Any
+from urllib.request import urlopen
 
-from allauth.account import app_settings
-from allauth.account import app_settings as allauth_account_settings
-from allauth.account import app_settings as allauth_settings
-from allauth.account.models import (
-    EmailConfirmation,
-    EmailConfirmationHMAC,
-    get_emailconfirmation_model,
-)
-from allauth.socialaccount.providers.google.views import GoogleOAuth2Adapter
-from allauth.socialaccount.providers.oauth2.client import OAuth2Client
+import requests
 from dj_rest_auth.app_settings import api_settings
-from dj_rest_auth.models import get_token_model
-from dj_rest_auth.registration.views import RegisterView, SocialLoginView
+from dj_rest_auth.jwt_auth import set_jwt_cookies
+from dj_rest_auth.registration.views import RegisterView
 from dj_rest_auth.utils import jwt_encode
 from dj_rest_auth.views import LoginView
 from django.conf import settings
 from django.core.cache import cache
-from django.http import Http404, HttpResponse, HttpResponseRedirect
-from django.shortcuts import render
+from django.core.files.base import ContentFile
 from django.utils import timezone
-from drf_spectacular.utils import (
-    extend_schema,
-    extend_schema_serializer,
-    extend_schema_view,
-    inline_serializer,
-)
-from rest_framework import permissions, serializers, status
-from rest_framework.exceptions import NotFound
-from rest_framework.generics import CreateAPIView
-from rest_framework.permissions import AllowAny
+from drf_spectacular.utils import extend_schema
+from rest_framework import permissions, status
 from rest_framework.request import Request
 from rest_framework.response import Response
 from rest_framework.views import APIView
 
+from apps.common.utils import uuid4_generator
 from apps.user.api_schema import (
     ConfirmRequestSchema,
     ConfirmResponseSchema,
@@ -49,11 +32,6 @@
 from apps.user.serializers import ConfirmEmailSerializer, SendCodeSerializer
 from apps.user.utils import generate_confirmation_code, send_email
 
-# class GoogleLogin(SocialLoginView):
-#     adapter_class = GoogleOAuth2Adapter
-#     callback_url = api_settings.GOOGLE_OAUTH2_URL
-#     client_class = OAuth2Client
-
 
 @extend_schema(request=SignupRequestSchema, responses=SignupResponseSchema)
 class CustomSignupView(RegisterView):  # type: ignore
@@ -125,8 +103,6 @@ def get_response(self) -> Response:
         data.pop("user", None)
         response = Response(data, status=status.HTTP_200_OK)
         if api_settings.USE_JWT:
-            from dj_rest_auth.jwt_auth import set_jwt_cookies
-
             set_jwt_cookies(response, self.access_token, self.refresh_token)
         return response
 
@@ -174,6 +150,188 @@ def post(self, request: Response, *args: Any, **kwargs: Any) -> Response:
         return Response({"message": "Email confirmation successful."}, status=status.HTTP_200_OK)
 
 
+class KakaoLoginView(APIView):
+    permission_classes = [permissions.AllowAny]
+
+    def post(self, request: Request, *args: Any, **kwargs: Any) -> Response:
+        code = request.data.get("code")  # 프론트에서 보내준 코드
+        # 카카오 oauth 토큰 발급 url로 code가 담긴 post 요청을 보내 응답을 받는다.
+        CLIENT_ID = settings.KAKAO_CLIENT_ID
+        REDIRECT_URI = settings.REDIRECT_URI
+        token_response = requests.post(
+            "https://kauth.kakao.com/oauth/token",
+            headers={"Content-Type": "application/x-www-form-urlencoded"},
+            data={
+                "grant_type": "authorization_code",
+                "code": code,
+                "redirect_uri": REDIRECT_URI,
+                "client_id": CLIENT_ID,
+            },
+        )
+
+        if token_response.status_code != status.HTTP_200_OK:
+            return Response(
+                {"msg": "카카오 서버로 부터 토큰을 받아오는데 실패하였습니다."},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
+        # 응답으로부터 액세스 토큰을 가져온다.
+        access_token = token_response.json().get("access_token")
+        response = requests.get(
+            "https://kapi.kakao.com/v2/user/me",
+            headers={
+                "Authorization": f"Bearer {access_token}",
+                "Content-type": "application/x-www-form-urlencoded;charset=utf-8",
+            },
+        )
+
+        if response.status_code != status.HTTP_200_OK:
+            return Response(
+                {"msg": "카카오 서버로 부터 프로필 데이터를 받아오는데 실패하였습니다."},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
+        response_data = response.json()
+        kakao_account = response_data["kakao_account"]
+        profile = kakao_account.get("profile")
+        requests.post("https://kapi.kakao.com/v1/user/logout", headers={"Authorization": f"Bearer {access_token}"})
+        try:
+            user = Account.objects.get(email=kakao_account.get("email"))
+            access_token, refresh_token = jwt_encode(user)
+            response = Response(
+                {
+                    "access": str(access_token),
+                    "refresh": str(refresh_token),  # type: ignore
+                    "email": user.email,
+                    "nickname": user.nickname,
+                    "profile_image": user.profile_img.url,
+                },
+                status=status.HTTP_200_OK,
+            )
+            # set_jwt_cookies(response, access_token, refresh_token)
+            return response  # type: ignore
+
+        except Account.DoesNotExist:
+            # 이미지를 다운로드하여 파일 객체로 가져옴
+            image_response = urlopen(profile.get("profile_image_url"))
+            image_content = image_response.read()
+            kakao_profile_image = ContentFile(image_content, name=f"kakao-profile-{uuid4_generator(8)}.jpg")
+            user = Account.objects.create(
+                email=kakao_account.get("email"),
+                nickname=profile.get("nickname"),
+                profile_img=kakao_profile_image,
+            )
+            user.set_unusable_password()
+            access_token, refresh_token = jwt_encode(user)
+            response = Response(
+                {
+                    "access": str(access_token),
+                    "refresh": str(refresh_token),  # type: ignore
+                    "email": user.email,
+                    "nickname": user.nickname,
+                    "profile_image": user.profile_img.url,
+                },
+                status=status.HTTP_200_OK,
+            )
+            # set_jwt_cookies(response, access_token, refresh_token)
+            return response  # type: ignore
+        except Exception as e:
+            return Response({"msg": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+
+class GoogleLoginView(APIView):
+    permission_classes = [permissions.AllowAny]
+
+    def post(self, request: Request) -> Response:
+        code = request.data.get("code")
+
+        client_id = settings.GOOGLE_CLIENT_ID
+        client_secret = settings.GOOGLE_SECRET
+        redirect_uri = settings.REDIRECT_URI
+
+        if not code:
+            return Response({"msg": "인가코드가 필요합니다."}, status=status.HTTP_400_BAD_REQUEST)
+
+        # 인가코드를 통해 토큰을 가져오는 요청
+        token_req = requests.post(
+            # f"https://oauth2.googleapis.com/token?client_id={client_id}&client_secret={client_secret}&code={code}&grant_type=authorization_code&redirect_uri={redirect_uri}"
+            "https://oauth2.googleapis.com/token",
+            headers={
+                "Content-type": "application/x-www-form-urlencoded;charset=utf-8",
+            },
+            data={
+                "client_id": client_id,
+                "client_secret": client_secret,
+                "code": code,
+                "grant_type": "authorization_code",
+                "redirect_uri": redirect_uri
+            },
+        )
+        # 요청의 응답을 json 파싱
+        token_req_json = token_req.json()
+        if token_req_json.status_code != 200:
+            return Response({"msg": token_req_json.get("error")}, status=status.HTTP_400_BAD_REQUEST)
+        # 파싱된 데이터중 액세스 토큰을 가져옴
+        google_access_token = token_req_json.get("access_token")
+
+        # 가져온 액세스토큰을 통해 사용자 정보에 접근하는 요청
+        info_response = requests.get(
+            f"https://www.googleapis.com/oauth2/v1/userinfo?access_token={google_access_token}"
+        )
+
+        # 상태코드로 요청이 실패했는지 확인
+        if info_response.status_code != 200:
+            return Response(
+                {"message": "구글 api로부터 액세스토큰을 받아오는데 실패했습니다."},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
+
+        # 요청의 응답을 json 파싱
+        res_json = info_response.json()
+        # 파싱된 데이터중 이메일값을 가져옴
+        email = res_json.get("email")
+        # 파싱된 데이터중 닉네임을 가져옴
+        nickname = res_json.get("nickname")
+        try:
+            user = Account.objects.get(email=email)
+            access_token, refresh_token = jwt_encode(user)
+            response_data = {
+                "access": str(access_token),
+                "refresh": str(refresh_token),
+                "email": user.email,
+                "nickname": user.nickname
+            }
+            if user.profile_img:
+                response_data["profile_image"] = user.profile_img.url
+            response = Response(response_data, status=status.HTTP_200_OK)
+            # if api_settings.USE_JWT:
+            #     set_jwt_cookies(response, access_token, refresh_token)
+            return response
+        except Account.DoesNotExist:
+            # 파싱된 데이터에서 프로필 이미지 url을 가져와서 파일로 변환
+            image_response = urlopen(res_json.get("picture"))
+            image_content = image_response.read()
+            google_profile_image = ContentFile(image_content, name=f"google-profile-{uuid4_generator(8)}.jpg")
+            # 가져온 이메일, 닉네임, 프로필 이미지를 통해 유저 생성
+            user = Account.objects.create(email=email, nickname=nickname, profile_img=google_profile_image)
+            user.set_unusable_password()
+            access_token, refresh_token = jwt_encode(user)
+            response_data = {
+                "access": str(access_token),
+                "refresh": str(refresh_token),
+                "email": user.email,
+                "nickname": user.nickname
+            }
+            if user.profile_img:
+                response_data["profile_image"] = user.profile_img.url
+            response = Response(response_data, status=status.HTTP_200_OK)
+            # if api_settings.USE_JWT:
+            #     set_jwt_cookies(response, access_token, refresh_token)
+            return response
+
+        except Exception as e:
+            # 가입이 필요한 회원
+            return Response({"msg": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+
 # class CustomConfirmEmailView(APIView):
 #     permission_classes = [AllowAny]
 #

config/settings/base.py

@@ -59,9 +59,6 @@
     "allauth.socialaccount",
     "dj_rest_auth",
     "dj_rest_auth.registration",
-    "allauth.socialaccount.providers.google",
-    # "allauth.socialaccount.providers.naver",
-    # "allauth.socialaccount.providers.kakao",
     "django_cleanup.apps.CleanupConfig",
     "corsheaders",
     "django_filters",

config/settings/local.py

@@ -206,3 +206,7 @@
 EMAIL_CODE_TIMEOUT = env("EMAIL_CODE_TIMEOUT")
 DJANGO_SUPERUSER_EMAIL = env("DJANGO_SUPERUSER_EMAIL")
 DJANGO_SUPERUSER_PASSWORD = env("DJANGO_SUPERUSER_PASSWORD")
+REDIRECT_URI = env("REDIRECT_URI")
+KAKAO_CLIENT_ID = env("KAKAO_CLIENT_ID")
+GOOGLE_CLIENT_ID = env("GOOGLE_CLIENT_ID")
+GOOGLE_SECRET = env("GOOGLE_SECRET")

config/settings/prod.py

@@ -148,3 +148,7 @@
 EMAIL_CODE_TIMEOUT = ENV["EMAIL_CODE_TIMEOUT"]
 DJANGO_SUPERUSER_EMAIL = ENV["DJANGO_SUPERUSER_EMAIL"]
 DJANGO_SUPERUSER_PASSWORD = ENV["DJANGO_SUPERUSER_PASSWORD"]
+REDIRECT_URI = ENV["REDIRECT_URI"]
+KAKAO_CLIENT_ID = ENV["KAKAO_CLIENT_ID"]
+GOOGLE_CLIENT_ID = ENV["GOOGLE_CLIENT_ID"]
+GOOGLE_SECRET = ENV["GOOGLE_SECRET"]

pyproject.toml

@@ -31,6 +31,7 @@ daphne = "^4.1.2"
 django-stubs-ext = "^5.0.0"
 django-filter = "^24.2"
 whitenoise = "^6.6.0"
+requests = "^2.32.0"
 
 
 [tool.poetry.group.dev.dependencies]