tweak dynamic url rewriting logic

handlers/proxy.go

@@ -33,7 +33,7 @@ func NewProxySiteHandler(opts *ProxyOptions) fiber.Handler {
 			SetRequestModifications(
 				// rx.SpoofJA3fingerprint(ja3, "Googlebot"),
 				// rx.MasqueradeAsFacebookBot(),
-				//rx.MasqueradeAsGoogleBot(),
+				rx.MasqueradeAsGoogleBot(),
 				rx.DeleteOutgoingCookies(),
 				rx.ForwardRequestHeaders(),
 				//rx.SpoofReferrerFromGoogleSearch(),
@@ -42,6 +42,7 @@ func NewProxySiteHandler(opts *ProxyOptions) fiber.Handler {
 				// rx.RequestArchiveIs(),
 			).
 			AddResponseModifications(
+				tx.InjectScriptBeforeDOMContentLoaded(`(() => {let d = document.createElement("div"); d.id = "adb-check"; document.body.append(d) })()`),
 				tx.ForwardResponseHeaders(),
 				tx.BypassCORS(),
 				tx.BypassContentSecurityPolicy(),

proxychain/responsemodifers/patch_dynamic_resource_urls.js

@@ -67,8 +67,11 @@
 
     //console.log(`proxychain: origin: ${origin} // proxyOrigin: ${proxyOrigin} // original: ${oldUrl}`)
 
+    //originDomain = origin.replace("https://", "");
+    let scheme = origin.split(":")[0];
+
     if (url.startsWith("//")) {
-            url = `/${origin}/${encodeURIComponent(url.substring(2))}`;
+      url = `/${scheme}://${encodeURIComponent(url.substring(2))}`;
     } else if (url.startsWith("/")) {
       url = `/${origin}/${encodeURIComponent(url.substring(1))}`;
     } else if (
@@ -278,6 +281,19 @@
     }
   });
 
+  // monkey-patching Element.setAttribute
+  const originalSetAttribute = Element.prototype.setAttribute;
+  Element.prototype.setAttribute = function (name, value) {
+    const isMatchingElement = elements.some((element) => {
+      return this.tagName.toLowerCase() === element.tag &&
+        name.toLowerCase() === element.attribute;
+    });
+    if (isMatchingElement) {
+      value = rewriteURL(value);
+    }
+    originalSetAttribute.call(this, name, value);
+  };
+
   // sometimes, libraries will set the Element.innerHTML or Element.outerHTML directly with a string instead of setters.
   // in this case, we intercept it, create a fake DOM, parse it and then rewrite all attributes that could
   // contain a URL. Then we return the replacement innerHTML/outerHTML with redirected links.

proxychain/responsemodifers/patch_tracker_scripts.go

@@ -66,6 +66,7 @@ func PatchTrackerScripts() proxychain.ResponseModification {
 		// preflight checks
 		reqURL := chain.Request.URL.String()
 		isTracker := false
+		//
 
 		var surrogateScript io.ReadCloser
 		for domain, domainRules := range rules {