From 40867acb26db3b2a90e0cac7afea4490ea2b21f1 Mon Sep 17 00:00:00 2001
From: imnyang <imnyang@proton.me>
Date: Sun, 15 Jun 2025 12:53:07 +0900
Subject: [PATCH] =?UTF-8?q?feat:=20=ED=99=98=EA=B2=BD=20=EB=B3=80=EC=88=98?=
 =?UTF-8?q?=EB=A5=BC=20=EC=84=A4=EC=A0=95=ED=95=98=EA=B3=A0=20Google=20?=
 =?UTF-8?q?=EB=A1=9C=EA=B7=B8=EC=9D=B8=20=ED=9E=8C=ED=8A=B8=20=EA=B8=B0?=
 =?UTF-8?q?=EB=8A=A5=EC=9D=84=20=EC=B6=94=EA=B0=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .github/workflows/ci.yml |  4 ++++
 addon/init.py            | 13 ++++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index d990669..bf5d5a9 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -26,6 +26,10 @@ jobs:
 
       - name: Install dependencies
         run: uv sync
+      
+      - name: Set up environment variables
+        run: |
+          echo "GOOGLE_ID=bot.imnya.ng@gmail.com" > .env
 
       - name: Start application and run proxy test
         run: |
diff --git a/addon/init.py b/addon/init.py
index afb3db8..b022811 100644
--- a/addon/init.py
+++ b/addon/init.py
@@ -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: