feat: 환경 변수를 설정하고 Google 로그인 힌트 기능을 추가

This commit is contained in:
imnyang 2025-06-15 12:53:07 +09:00
commit 40867acb26
2 changed files with 16 additions and 1 deletions

View file

@ -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: