51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from schemas import LoginSchema
|
|
from repositories import UserRepository
|
|
from mapper import UserMapper
|
|
from google.oauth2 import id_token
|
|
from google.auth.transport import requests
|
|
from configs import Config
|
|
from exception import AuthException
|
|
from flask import current_app
|
|
|
|
|
|
class AuthService:
|
|
def __init__(self, userRepository: UserRepository):
|
|
self.user_repository = userRepository
|
|
|
|
def verify_google_id_token(self, id_token_str):
|
|
payload = id_token.verify_oauth2_token(
|
|
id_token_str, requests.Request(), Config.GOOGLE_CLIENT_ID
|
|
)
|
|
|
|
if not payload:
|
|
raise AuthException("Invalid Google ID Token")
|
|
|
|
google_id = payload.get("sub")
|
|
email = payload.get("email")
|
|
|
|
existing_user = self.user_repository.get_by_google_id(google_id)
|
|
if existing_user:
|
|
if existing_user.email == email:
|
|
return existing_user
|
|
raise AuthException("Email not match")
|
|
|
|
new_user = UserMapper.from_google_payload(google_id, email, payload)
|
|
|
|
user_id = self.user_repository.insert_user(user_data=new_user)
|
|
|
|
return self.user_repository.get_user_by_id(user_id=user_id)
|
|
|
|
def login(self, data: LoginSchema):
|
|
|
|
current_app.logger.info(f"request data: {data}")
|
|
user_data = self.user_repository.get_user_by_email(data.email)
|
|
|
|
current_app.logger.info(f"user_data: {user_data}")
|
|
if user_data == None:
|
|
return None
|
|
|
|
if user_data.password == data.password:
|
|
del user_data.password
|
|
return user_data
|
|
return None
|