From 411813aa4c22a8600bfc4e639de9c665bde80bde Mon Sep 17 00:00:00 2001
From: MotorTruck1221 <motortruck1221@protonmail.com>
Date: Sat, 14 Jun 2025 22:32:10 -0600
Subject: [PATCH 1/2] Add the reactive parts

---
 src/components/ui/Dropdown.astro |  5 +++--
 src/pages/settings/index.astro   | 27 +++++++++++++++++++++++++--
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/src/components/ui/Dropdown.astro b/src/components/ui/Dropdown.astro
index b0a59e7..751d95e 100644
--- a/src/components/ui/Dropdown.astro
+++ b/src/components/ui/Dropdown.astro
@@ -2,12 +2,13 @@
 import type { DropdownOptions } from "@utils/types";
 interface Props {
     id: string;
+    classes?: string;
     options: DropdownOptions[];
 }
 
-const { id, options } = Astro.props;
+const { id, options, classes } = Astro.props;
 ---
-<select id=`dropdownBox-${id}` class="cursor-pointer flex h-10 w-[180px] items-center justify-between text-(--foreground) background-(--background) rounded-lg border border-(--border) px-3 py-2">
+<select id=`dropdownBox-${id}` class=`${classes} cursor-pointer flex h-10 w-[180px] items-center justify-between text-(--foreground) background-(--background) rounded-lg border border-(--border) px-3 py-2`>
     {options.map((el) => 
         <option class="w-full bg-(--accent) rounded-sm p-1" value={el.value}>{el.name}</option>
     )}
diff --git a/src/pages/settings/index.astro b/src/pages/settings/index.astro
index b487ee6..0f4100a 100644
--- a/src/pages/settings/index.astro
+++ b/src/pages/settings/index.astro
@@ -37,6 +37,15 @@ Object.keys(SearchEngines).forEach((k) =>
             <div>
                 <p> Wisp Server </p>
                 <Input id="wispServerSwitcher" placeholder="Wisp server URL (EX: wss://radiusproxy.app/wisp/" />
+                <div class="mt-2 hidden" id="adBlocking">
+                    <p> Ad Blocking </p>
+                    <Dropdown id="adBlocking" options={
+                        [
+                            { name: 'Enabled', value: 'enabled', default: true },
+                            { name: 'Disabled', value: 'disabled' }
+                        ]
+                    } />
+                </div>
             </div>
             <div class="mt-2 mb-2 hidden" id="wispServerInfo">
                 <p class="text-blue-500" id="wispServerInfo-inner"> Checking URL... </p>
@@ -90,9 +99,10 @@ Object.keys(SearchEngines).forEach((k) =>
         const wispServerInfoInner = document.getElementById("wispServerInfo-inner") as HTMLParagraphElement;
         const wispServerSave = document.getElementById("wispServerSave") as HTMLButtonElement;
         const wispServerReset = document.getElementById("wispServerReset") as HTMLButtonElement;
+        const adBlocking = document.getElementById("adBlocking") as HTMLDivElement;
 
         wispServerSwitcher.value = opts.storageManager.getVal("wispServer");
-        
+        const resetVal = `${(location.protocol === "https:" ? "wss://" : "ws://")}${location.host}/wisp/` 
         const reset = (hide: boolean = true) => {
             if (hide) wispServerInfo.classList.add("hidden");
             wispServerInfoInner.innerText = "Checking URL..."
@@ -100,6 +110,16 @@ Object.keys(SearchEngines).forEach((k) =>
             wispServerInfoInner.classList.remove("text-green-500");
         };
 
+        const adBlockingFunc = () => {
+            if (wispServerSwitcher.value === resetVal) {
+                adBlocking.classList.remove("hidden");
+            }
+            else {
+                adBlocking.classList.add("hidden");
+            }
+        }
+        adBlockingFunc();
+
         wispServerSave.addEventListener("click", async () => {
             const server = wispServerSwitcher.value;
             wispServerInfo.classList.remove("hidden"); 
@@ -114,7 +134,9 @@ Object.keys(SearchEngines).forEach((k) =>
                 reset(false);
                 wispServerInfoInner.innerText = "Wisp Server Set!";
                 wispServerInfoInner.classList.add("text-green-500");
+                //adBlocking.classList.contains("hidden") ? "" : adBlocking.classList.add("hidden");
                 await opts.sw.wispServer(wispServerSwitcher.value, true);
+                adBlockingFunc();
             }
 
             // We reset this after 4 seconds (any errors OR success)
@@ -125,9 +147,10 @@ Object.keys(SearchEngines).forEach((k) =>
             wispServerInfo.classList.remove("hidden");
             wispServerInfoInner.innerText = "Wisp Server Reset!";
             wispServerInfoInner.classList.add("text-green-500");
-            await opts.sw.wispServer(`${(location.protocol === "https:" ? "wss://" : "ws://")}${location.host}/wisp/`, true);
+            await opts.sw.wispServer(resetVal, true);
             wispServerSwitcher.value = opts.storageManager.getVal("wispServer");
             setTimeout(reset, 4000);
+            adBlockingFunc();
         });
     }
 

From d2bc8b748dacf8026ca5f1fd7f72ea33f63a207d Mon Sep 17 00:00:00 2001
From: MotorTruck1221 <motortruck1221@protonmail.com>
Date: Sat, 14 Jun 2025 23:08:25 -0600
Subject: [PATCH 2/2] Add adblocking

---
 astro.config.ts                |  2 +-
 server/index.ts                |  3 ++-
 src/components/Loader.astro    |  1 +
 src/pages/settings/index.astro | 10 ++++++++++
 src/utils/proxy.ts             | 14 +++++++++++---
 src/utils/settings.ts          |  9 +++++++++
 6 files changed, 34 insertions(+), 5 deletions(-)

diff --git a/astro.config.ts b/astro.config.ts
index 07ab795..b474e0c 100644
--- a/astro.config.ts
+++ b/astro.config.ts
@@ -18,7 +18,7 @@ const viteWispServer = (): Plugin => {
         name: "vite-wisp-server",
         configureServer(server) {
             server.httpServer?.on("upgrade", (req, socket, head) => {
-                req.url.startsWith("/wisp") ? wisp.routeRequest(req, socket, head) : undefined;
+                req.url.startsWith("/wisp") || req.url.startsWith("/adblock") ? wisp.routeRequest(req, socket, head) : undefined;
             });
         }
     };
diff --git a/server/index.ts b/server/index.ts
index ca58c6d..4dab568 100644
--- a/server/index.ts
+++ b/server/index.ts
@@ -23,7 +23,8 @@ const serverFactory: FastifyServerFactory = (
             handler(req, res);
         })
         .on("upgrade", (req, socket, head) => {
-            if (req.url?.endsWith("/wisp/")) {
+            if (req.url?.endsWith("/wisp/") || req.url?.endsWith("/adblock/")) {
+                console.log(req.url);
                 wisp.routeRequest(req, socket as Socket, head);
             }
         });
diff --git a/src/components/Loader.astro b/src/components/Loader.astro
index 2141ae6..5c2ccb5 100644
--- a/src/components/Loader.astro
+++ b/src/components/Loader.astro
@@ -7,6 +7,7 @@
     const init = async () => {
         settings.searchEngine();
         settings.proxy();
+        settings.adBlock();
         await sw.wispServer();
     }
     
diff --git a/src/pages/settings/index.astro b/src/pages/settings/index.astro
index 0f4100a..d539f1a 100644
--- a/src/pages/settings/index.astro
+++ b/src/pages/settings/index.astro
@@ -111,11 +111,21 @@ Object.keys(SearchEngines).forEach((k) =>
         };
 
         const adBlockingFunc = () => {
+            const adBlockingDropdown = document.getElementById("dropdownBox-adBlocking") as HTMLSelectElement;
+            adBlockingDropdown.addEventListener("change", () => {
+                opts.settings.adBlock(adBlockingDropdown.value === "enabled" ? true : false);
+            });
+
+            adBlockingDropdown.value = opts.storageManager.getVal("adBlock") === "true" ? "enabled" : "disabled";
+
             if (wispServerSwitcher.value === resetVal) {
                 adBlocking.classList.remove("hidden");
+                opts.settings.adBlock(true);
+                adBlockingDropdown.value = "enabled";
             }
             else {
                 adBlocking.classList.add("hidden");
+                opts.settings.adBlock(false);
             }
         }
         adBlockingFunc();
diff --git a/src/utils/proxy.ts b/src/utils/proxy.ts
index ab5ae7b..aa3608c 100644
--- a/src/utils/proxy.ts
+++ b/src/utils/proxy.ts
@@ -60,6 +60,13 @@ class SW {
 
     async setTransport(transport?: "epoxy" | "libcurl", get?: boolean) {
         console.log("Setting transport");
+        const wispServer = (): string => {
+            const wispServerVal = this.#storageManager.getVal("wispServer");
+            if (this.#storageManager.getVal("adBlock") === "true") {
+                return wispServerVal.replace("/wisp/", "/adblock/");
+            }
+            return wispServerVal;
+        }
         if (get) return this.#storageManager.getVal("transport");
         this.#storageManager.setVal(
             "transport",
@@ -68,23 +75,24 @@ class SW {
         switch (transport) {
             case "epoxy": {
                 await this.#baremuxConn!.setTransport("/epoxy/index.mjs", [
-                    { wisp: this.#storageManager.getVal("wispServer") }
+                    { wisp: wispServer() }
                 ]);
             }
             case "libcurl": {
                 await this.#baremuxConn!.setTransport("/libcurl/index.mjs", [
-                    { wisp: this.#storageManager.getVal("wispServer") }
+                    { wisp: wispServer() }
                 ]);
             }
             default: {
                 await this.#baremuxConn!.setTransport("/epoxy/index.mjs", [
-                    { wisp: this.#storageManager.getVal("wispServer") }
+                    { wisp: wispServer() }
                 ]);
             }
         }
     }
 
     async wispServer(wispServer?: string, set?: true) {
+        console.log(wispServer?.replace("/wisp/", "/adblock/"));
         this.#storageManager.setVal(
             "wispServer",
             wispServer ||
diff --git a/src/utils/settings.ts b/src/utils/settings.ts
index a3bddcf..c55ba2c 100644
--- a/src/utils/settings.ts
+++ b/src/utils/settings.ts
@@ -122,6 +122,15 @@ class Settings {
         }
     }
 
+    adBlock(enabled?: boolean) {
+        if (enabled === true || enabled === false) {
+            this.#storageManager.setVal("adBlock", enabled.valueOf().toString());
+        }
+        else {
+            this.#storageManager.setVal("adBlock", "true");
+        }
+    }
+
     async *#init() {
         yield this.theme(this.#storageManager.getVal("theme") || "default");
     }