From 6bcf2946354f6ab1bdb0067bddbb4843f66335e9 Mon Sep 17 00:00:00 2001
From: Gabe Yuan <yugang2002@gmail.com>
Date: Sat, 26 Aug 2023 12:11:21 +0800
Subject: [PATCH] userscript in iframe

---
 src/userscript.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/userscript.js b/src/userscript.js
index 8cf1291..1517019 100644
--- a/src/userscript.js
+++ b/src/userscript.js
@@ -26,9 +26,10 @@ import { isIframe } from "./libs/iframe";
   }
 
   // 翻译页面
+  const href = isIframe ? document.referrer : document.location.href;
   const setting = await getSetting();
   const rules = await getRules();
-  const rule = await matchRule(rules, document.location.href, setting);
+  const rule = await matchRule(rules, href, setting);
   const translator = new Translator(rule, setting);
 
   if (isIframe) {