수정/test 완료

addon/init.py

@@ -1,6 +1,7 @@
 from mitmproxy import http
 import asyncio
 from pkce_check import PKCEDowngradeChecker
+from ScopeDetection import ScopeDetection
 
 class PKCEAddon:
     def __init__(self):
@@ -13,4 +14,25 @@ class PKCEAddon:
         except Exception as e:
             print(f"[ERROR] Addon failed: {e}")
 
-addons = [PKCEAddon()]
+class ScopeAddon:
+    def __init__(self):
+        self.checker = ScopeDetection()
+        self._flow_map = {}  # 요청 정보를 저장
+
+    async def request(self, flow: http.HTTPFlow):
+        self._flow_map[flow.id] = {
+            "method": flow.request.method,
+            "url": flow.request.pretty_url,
+            "query": flow.request.query,
+        }
+
+    async def response(self, flow: http.HTTPFlow):
+        try:
+            await self.checker.test(flow)
+        except Exception as e:
+            print(f"[ERROR] ScopeDetection failed: {e}")
+
+
+addons = [PKCEAddon(),
+          ScopeAddon()
+]