mirror of
https://github.com/j93es/oauth-backend.git
synced 2026-06-04 05:21:51 +09:00
feat: 환경 변수를 설정하고 Google 로그인 힌트 기능을 추가
This commit is contained in:
parent
c311aaad71
commit
40867acb26
2 changed files with 16 additions and 1 deletions
|
|
@ -1,3 +1,4 @@
|
|||
from json import load
|
||||
from mitmproxy import http
|
||||
import asyncio
|
||||
from pkce_check import PKCEDowngradeChecker
|
||||
|
|
@ -7,6 +8,10 @@ from nonce_check import NonceChecker
|
|||
from redirect_uri_check import RedirectBypassChecker
|
||||
from access_token import AccessTokenScanner
|
||||
from GoogleLoginHint import GoogleLoginHint
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv(override=True)
|
||||
|
||||
class PKCEAddon:
|
||||
def __init__(self):
|
||||
|
|
@ -90,8 +95,14 @@ class RedirectBypassAddon:
|
|||
|
||||
class GoogleLoginHintAddon():
|
||||
def __init__(self) -> None:
|
||||
self.checker = GoogleLoginHint()
|
||||
if os.getenv('GOOGLE_ID'):
|
||||
self.checker = GoogleLoginHint()
|
||||
else:
|
||||
self.checker = None
|
||||
|
||||
def request(self, flow: http.HTTPFlow):
|
||||
if self.checker is None:
|
||||
return
|
||||
try:
|
||||
asyncio.run(self.checker.request(flow))
|
||||
except Exception as e:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue