代码重构：将自动生成script拦截器配置的代码，转移到 script.js 中。

7 months ago · 4267e20148
parent e90dbae814
commit 4267e20148
2 changed files with 98 additions and 89 deletions
--- a/packages/mitmproxy/src/lib/interceptor/impl/res/script.js
+++ b/packages/mitmproxy/src/lib/interceptor/impl/res/script.js
@ -1,8 +1,14 @@
-const contextPath = '/____ds_script____/'
 const monkey = require('../../../monkey')
 const CryptoJs = require('crypto-js')
+const lodash = require('lodash')
+const log = require('../../../../utils/util.log')
+
+const SCRIPT_URL_PRE = '/____ds_script____/' // 内置脚本的请求地址前缀
+const SCRIPT_PROXY_URL_PRE = '/____ds_script_proxy____/' // 绝对地址脚本的伪脚本地址前缀
+const REMOVE = '[remove]' // 标记需要移除的头信息
+
 function getScript (key, script) {
-  const scriptUrl = contextPath + key
+  const scriptUrl = SCRIPT_URL_PRE + key

  const hash = CryptoJs.SHA256(script).toString(CryptoJs.enc.Base64)
  return `<script crossorigin="anonymous" defer="defer" type="application/javascript" src="${scriptUrl}" integrity="sha256-${hash}"></script>`
@ -60,5 +66,91 @@ module.exports = {
  },
  is (interceptOpt) {
    return interceptOpt.script
+  },
+  // 处理拦截配置：自动生成script拦截器所需的辅助配置，降低使用`script拦截器`配置绝对地址和相对地址时的门槛
+  handleScriptInterceptConfig (intercepts) {
+    // 为了简化 script 拦截器配置脚本绝对地址，这里特殊处理一下
+    for (const hostnamePattern in intercepts) {
+      const hostnameConfig = intercepts[hostnamePattern]
+
+      const scriptProxy = {}
+      for (const pathPattern in hostnameConfig) {
+        const pathConfig = hostnameConfig[pathPattern]
+        if (typeof pathConfig.script === 'object' && pathConfig.script.length > 0) {
+          for (let i = 0; i < pathConfig.script.length; i++) {
+            const script = pathConfig.script[i]
+            if (script.indexOf('https:') === 0 || script.indexOf('http:') === 0) {
+              // 绝对地址
+              const scriptKey = SCRIPT_PROXY_URL_PRE + script.replace('.js', '').replace(/\W/g, '') + '.js' // 伪脚本地址：移除 script 中可能存在的特殊字符，并转为相对地址
+              scriptProxy[scriptKey] = script
+              log.info(`替换script配置值：'${pathConfig.script[i]}' -> '${scriptKey}'`)
+              pathConfig.script[i] = scriptKey
+            } else if (script.indexOf('/') === 0) {
+              // 相对地址
+              scriptProxy[script] = script
+            }
+          }
+        }
+      }
+
+      // 自动创建脚本
+      if (!lodash.isEmpty(scriptProxy)) {
+        for (const scriptKey in scriptProxy) {
+          if (scriptKey.indexOf(SCRIPT_PROXY_URL_PRE) === 0) {
+            // 绝对地址：新增代理配置
+            const scriptUrl = scriptProxy[scriptKey]
+
+            const pathPattern = `^${scriptKey.replace(/\./g, '\\.')}$`
+            if (hostnameConfig[pathPattern]) {
+              continue // 配置已经存在，按自定义配置优先
+            }
+            hostnameConfig[pathPattern] = {
+              proxy: scriptUrl,
+              // 移除部分请求头，避免触发目标站点的拦截策略
+              requestReplace: {
+                headers: {
+                  host: REMOVE,
+                  referer: REMOVE,
+                  cookie: REMOVE
+                }
+              },
+              // 替换和移除部分响应头，避免触发目标站点的阻止脚本加载策略
+              responseReplace: {
+                headers: {
+                  'content-type': 'application/javascript; charset=utf-8',
+                  'set-cookie': REMOVE,
+                  server: REMOVE
+                }
+              },
+              cacheDays: 7,
+              desc: "为伪脚本文件设置代理地址，并设置响应头 `content-type: 'application/javascript; charset=utf-8'`，同时缓存7天。"
+            }
+
+            const obj = {}
+            obj[pathPattern] = hostnameConfig[pathPattern]
+            log.info(`域名 '${hostnamePattern}' 拦截配置中，新增伪脚本地址的代理配置:`, obj)
+          } else {
+            // 相对地址：新增响应头Content-Type替换配置
+            if (hostnameConfig[scriptKey]) {
+              continue // 配置已经存在，按自定义配置优先
+            }
+
+            hostnameConfig[scriptKey] = {
+              responseReplace: {
+                headers: {
+                  'content-type': 'application/javascript; charset=utf-8'
+                }
+              },
+              cacheDays: 7,
+              desc: "为脚本设置响应头 `content-type: 'application/javascript; charset=utf-8'`，同时缓存7天。"
+            }
+
+            const obj = {}
+            obj[scriptKey] = hostnameConfig[scriptKey]
+            log.info(`域名 '${hostnamePattern}' 拦截配置中，新增目标脚本地址的响应头替换配置:`, obj)
+          }
+        }
+      }
+    }
  }
 }
--- a/packages/mitmproxy/src/options.js
+++ b/packages/mitmproxy/src/options.js
@ -3,97 +3,14 @@ const dnsUtil = require('./lib/dns')
 const log = require('./utils/util.log')
 const matchUtil = require('./utils/util.match')
 const path = require('path')
-const lodash = require('lodash')
-const SCRIPT_URL_PRE = '/____ds_script_proxy____/'
-const REMOVE = '[remove]'
+const scriptInterceptor = require('./lib/interceptor/impl/res/script')

 const createOverwallMiddleware = require('./lib/proxy/middleware/overwall')

-// 处理拦截配置：目前，只自动生成了script拦截器所需的辅助配置，降低script拦截器配置绝对地址和相对地址的门槛
+// 处理拦截配置
 function buildIntercepts (intercepts) {
-  // 为了简化 script 拦截器配置脚本绝对地址，这里特殊处理一下
-  for (const hostnamePattern in intercepts) {
-    const hostnameConfig = intercepts[hostnamePattern]
-
-    const scriptProxy = {}
-    for (const pathPattern in hostnameConfig) {
-      const pathConfig = hostnameConfig[pathPattern]
-      if (typeof pathConfig.script === 'object' && pathConfig.script.length > 0) {
-        for (let i = 0; i < pathConfig.script.length; i++) {
-          const script = pathConfig.script[i]
-          if (script.indexOf('https:') === 0 || script.indexOf('http:') === 0) {
-            // 绝对地址
-            const scriptKey = SCRIPT_URL_PRE + script.replace('.js', '').replace(/\W/g, '') + '.js' // 伪脚本地址：移除 script 中可能存在的特殊字符，并转为相对地址
-            scriptProxy[scriptKey] = script
-            log.info(`替换script配置值：'${pathConfig.script[i]}' -> '${scriptKey}'`)
-            pathConfig.script[i] = scriptKey
-          } else if (script.indexOf('/') === 0) {
-            // 相对地址
-            scriptProxy[script] = script
-          }
-        }
-      }
-    }
-
-    // 自动创建脚本
-    if (!lodash.isEmpty(scriptProxy)) {
-      for (const scriptKey in scriptProxy) {
-        if (scriptKey.indexOf(SCRIPT_URL_PRE) === 0) {
-          // 绝对地址：新增代理配置
-          const scriptUrl = scriptProxy[scriptKey]
-
-          const pathPattern = `^${scriptKey.replace(/\./g, '\\.')}$`
-          if (hostnameConfig[pathPattern]) {
-            continue // 配置已经存在，按自定义配置优先
-          }
-          hostnameConfig[pathPattern] = {
-            proxy: scriptUrl,
-            // 移除部分请求头，避免触发目标站点的拦截策略
-            requestReplace: {
-              headers: {
-                host: REMOVE,
-                referer: REMOVE,
-                cookie: REMOVE
-              }
-            },
-            // 替换和移除部分响应头，避免触发目标站点的阻止脚本加载策略
-            responseReplace: {
-              headers: {
-                'content-type': 'application/javascript; charset=utf-8',
-                'set-cookie': REMOVE,
-                server: REMOVE
-              }
-            },
-            cacheDays: 7,
-            desc: "为伪脚本文件设置代理地址，并设置响应头 `content-type: 'application/javascript; charset=utf-8'`，同时缓存7天。"
-          }
-
-          const obj = {}
-          obj[pathPattern] = hostnameConfig[pathPattern]
-          log.info(`域名 '${hostnamePattern}' 拦截配置中，新增伪脚本地址的代理配置:`, obj)
-        } else {
-          // 相对地址：新增响应头Content-Type替换配置
-          if (hostnameConfig[scriptKey]) {
-            continue // 配置已经存在，按自定义配置优先
-          }
-
-          hostnameConfig[scriptKey] = {
-            responseReplace: {
-              headers: {
-                'content-type': 'application/javascript; charset=utf-8'
-              }
-            },
-            cacheDays: 7,
-            desc: "为脚本设置响应头 `content-type: 'application/javascript; charset=utf-8'`，同时缓存7天。"
-          }
-
-          const obj = {}
-          obj[scriptKey] = hostnameConfig[scriptKey]
-          log.info(`域名 '${hostnamePattern}' 拦截配置中，新增目标脚本地址的响应头替换配置:`, obj)
-        }
-      }
-    }
-  }
+  // 自动生成script拦截器所需的辅助配置，降低使用`script拦截器`配置绝对地址和相对地址时的门槛
+  scriptInterceptor.handleScriptInterceptConfig(intercepts)

  return intercepts
 }