feat: 사용자의 설정에 따라 특정 도메인에서만 작동하도록 변경

Author: khw-dev

content.js

@@ -1,4 +1,27 @@
-const script = document.createElement("script");
-script.src = chrome.runtime.getURL("inject.js");
-script.onload = () => script.remove();
-(document.head || document.documentElement).appendChild(script);
+(async () => {
+  const url = location.href;
+
+  const toReg = (g) =>
+    new RegExp(
+      "^" + g.replace(/[.+^${}()|[\]\\]/g, "\\$&").replace(/\*/g, ".*") + "$"
+    );
+
+  const inject = () => {
+    const s = document.createElement("script");
+    s.src = chrome.runtime.getURL("inject.js");
+    (document.head || document.documentElement).append(s);
+  };
+
+  const pref = await chrome.storage.sync.get({
+    enabled: true,
+    enableOnThis: false,
+    domains: [],
+  });
+
+  let should = false;
+  if (pref.enabled) should = true;
+  else if (pref.enableOnThis) should = true;
+  else if (pref.domains.some((d) => toReg(d).test(url))) should = true;
+
+  if (should) inject();
+})();

inject.js

@@ -12,76 +12,61 @@
   const cleanse = (text) =>
     watermarkChars.reduce((t, ch) => t.split(ch).join(" "), text);
 
-  // sol 1
+  let lastCleanTime = 0;
+  const THRESHOLD = 50;
+
+  const shouldSkip = () => {
+    const now = Date.now();
+    if (now - lastCleanTime < THRESHOLD) return true;
+    lastCleanTime = now;
+    return false;
+  };
+
   if (navigator.clipboard?.writeText) {
-    const origWriteText = navigator.clipboard.writeText.bind(
-      navigator.clipboard
-    );
+    const orig = navigator.clipboard.writeText.bind(navigator.clipboard);
     navigator.clipboard.writeText = async (data) => {
+      if (shouldSkip()) return orig(data);
       const cleaned = cleanse(data);
-      console.log(`Text watermarks removed.`);
-      return origWriteText(cleaned);
+      console.log("[T.W.R.] all text watermarks removed");
+      return orig(cleaned);
     };
   }
 
-  // sol 2
   if (navigator.clipboard?.write) {
-    const origWrite = navigator.clipboard.write.bind(navigator.clipboard);
+    const orig = navigator.clipboard.write.bind(navigator.clipboard);
     navigator.clipboard.write = async (items) => {
+      if (shouldSkip()) return orig(items);
       const newItems = await Promise.all(
         items.map(async (item) => {
           const blobs = {};
           for (const type of item.types) {
-            const blob = await item.getType(type);
+            const b = await item.getType(type);
             if (type === "text/plain") {
-              const text = await blob.text();
-              const cleaned = cleanse(text);
-              console.log(`Text watermarks removed.`);
-              blobs[type] = new Blob([cleaned], { type });
-            } else blobs[type] = blob;
+              const text = await b.text();
+              blobs[type] = new Blob([cleanse(text)], { type });
+            } else blobs[type] = b;
           }
+          console.log("[T.W.R.] all text watermarks removed");
           return new ClipboardItem(blobs);
         })
       );
-      return origWrite(newItems);
-    };
-  }
-
-  // sol 3
-  {
-    const origExec = Document.prototype.execCommand;
-    Document.prototype.execCommand = (cmd, ...args) => {
-      if (cmd.toLowerCase() === "copy") {
-        const sel = window.getSelection().toString();
-        const cleaned = cleanse(sel);
-        const ta = document.createElement("textarea");
-        ta.value = cleaned;
-        ta.style.position = "fixed";
-        ta.style.opacity = "0";
-        document.body.appendChild(ta);
-        ta.select();
-        const result = origExec.call(this, "copy", ...args);
-        document.body.removeChild(ta);
-        console.log(`Text watermarks removed.`);
-        return result;
-      }
-      return origExec.call(this, cmd, ...args);
+      return orig(newItems);
     };
   }
 
-  // sol 4
   document.addEventListener(
     "copy",
     (e) => {
+      if (shouldSkip()) return;
       const sel = window.getSelection().toString();
       const cleaned = cleanse(sel);
       e.clipboardData.setData("text/plain", cleaned);
       e.clipboardData.setData("text/html", cleaned);
       e.preventDefault();
-      console.log(`Text watermarks removed.`);
+      console.log("[T.W.R.] all text watermarks removed");
     },
     true
   );
 
-  console.log("[T.W.R.] Injected all content scripts!");
+  console.log("[T.W.R.] all interceptors installed");
 })();