From 3db45d402868ab24177a92eba7c046ee03e05f6f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E8=89=AF?= <841369634@qq.com>
Date: Fri, 19 Apr 2024 11:49:05 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96script.js=EF=BC=9A=E5=8F=AA?=
 =?UTF-8?q?=E8=A6=81=E6=9C=89=E8=84=9A=E6=9C=AC=E5=AD=98=E5=9C=A8=EF=BC=8C?=
 =?UTF-8?q?=E5=B0=B1=E5=8A=A0=E8=BD=BD=20tampermonkey.script?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/lib/interceptor/impl/res/script.js        | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/packages/mitmproxy/src/lib/interceptor/impl/res/script.js b/packages/mitmproxy/src/lib/interceptor/impl/res/script.js
index 0994266f..562d4da3 100644
--- a/packages/mitmproxy/src/lib/interceptor/impl/res/script.js
+++ b/packages/mitmproxy/src/lib/interceptor/impl/res/script.js
@@ -33,12 +33,12 @@ module.exports = {
       keys = [keys]
     }
     try {
-      // 内置脚本
+      // 内置脚本列表
       const scripts = monkey.get(setting.script.dirAbsolutePath)
 
       let tags = ''
       for (const key of keys) {
-        if (key === 'global') {
+        if (key === 'global' || key === 'tampermonkey') {
           continue
         }
 
@@ -51,10 +51,6 @@ module.exports = {
           if (script == null) {
             continue
           }
-          if (key === 'github') {
-            // 如果是github油猴脚本，需要先加载 `global.script`，避免 `GM_xxx` 方法不存在导致报错。
-            tags += '\r\n\t' + getScript('global', scripts.global.script)
-          }
           scriptTag = getScript(key, script.script) // 2.DS内置脚本
         }
 
@@ -62,7 +58,12 @@ module.exports = {
       }
 
       // 如果脚本为空，则不插入
-      if (tags === '') return
+      if (tags === '') {
+        return
+      }
+
+      // 插入油猴脚本浏览器扩展
+      tags = '\r\n\t' + getScript('tampermonkey', scripts.tampermonkey.script) + tags
 
       res.setHeader('DS-Script-Interceptor', 'true')
       log.info('script response intercept: insert script', rOptions.hostname, rOptions.path, ', head:', tags)