From 40092d8227dd55c184a879e3b5bf8a460e4e7f51 Mon Sep 17 00:00:00 2001
From: DarkCat09 <gh@dc09.ru>
Date: Fri, 22 Sep 2023 13:00:16 +0400
Subject: [PATCH] replace-href: code cleanup

---
 src/utils/replace-href.ts | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/src/utils/replace-href.ts b/src/utils/replace-href.ts
index bd16d33..1ac4f3c 100644
--- a/src/utils/replace-href.ts
+++ b/src/utils/replace-href.ts
@@ -8,10 +8,7 @@ export default function replaceHref(
   engine?: string,
   redirectPath: string = "get",
 ) {
-  const bytag =
-    (dom: JSDOM, tag: string) => dom.window.document.getElementsByTagName(tag);
-  const bycss =
-    (dom: JSDOM, css: string) => dom.window.document.querySelectorAll(css);
+  const doc = dom.window.document;
 
   const parserUrl = (href: string) => generateParserUrl(
     requestUrl,
@@ -25,30 +22,30 @@ export default function replaceHref(
   );
 
   modifyLinks(
-    bytag(dom, "a"),
+    doc.getElementsByTagName("a"),
     "href",
     parserUrl,
   );
   modifyLinks(
-    bycss(dom, "frame,iframe"),
+    doc.querySelectorAll("frame,iframe"),
     "src",
     parserUrl,
   );
 
   if (getConfig().proxy_res) {
     modifyLinks(
-      bycss(dom, "img,image,video,audio,embed,track,source"),
+      doc.querySelectorAll("img,image,video,audio,embed,track,source"),
       "src",
       proxyUrl,
     );
 
     modifyLinks(
-      bytag(dom, "object"),
+      doc.getElementsByTagName("object"),
       "data",
       proxyUrl,
     );
 
-    const sources = bytag(dom, "source");
+    const sources = doc.getElementsByTagName("source");
     for (const source of sources) {
       // split srcset by comma
       // @ts-ignore