From 9d4c4e82442e4a41f14eb945f002f8748f2a6c94 Mon Sep 17 00:00:00 2001
From: Tiago <70700766+tiagorangel1@users.noreply.github.com>
Date: Sat, 10 May 2025 10:27:00 +0100
Subject: [PATCH] standalone: js libs storage

---
 standalone/package.json     |  4 +-
 standalone/public/test.html |  3 +-
 standalone/src/index.js     | 81 ++++++++++++++++++++++++++++++-------
 widget/src/src/cap.js       |  2 -
 4 files changed, 70 insertions(+), 20 deletions(-)

diff --git a/standalone/package.json b/standalone/package.json
index e52fe40..38d0036 100644
--- a/standalone/package.json
+++ b/standalone/package.json
@@ -1,6 +1,6 @@
 {
-  "name": "app",
-  "version": "1.0.8",
+  "name": "cap-standalone",
+  "version": "1.0.9",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",
     "dev": "bun run --watch src/index.js",
diff --git a/standalone/public/test.html b/standalone/public/test.html
index 19bbb6f..3d4f057 100644
--- a/standalone/public/test.html
+++ b/standalone/public/test.html
@@ -99,7 +99,8 @@
       <canvas id="progressChart"></canvas>
     </div>
 
-    <script src="https://cdn.jsdelivr.net/npm/@cap.js/widget"></script>
+    <script>window.CAP_CUSTOM_WASM_URL="/"</script>
+    <script src="/assets/widget.js"></script>
 
     <script>
       const logsElement = document.querySelector(".logs");
diff --git a/standalone/src/index.js b/standalone/src/index.js
index c62b693..768f100 100644
--- a/standalone/src/index.js
+++ b/standalone/src/index.js
@@ -1,6 +1,6 @@
 import { Elysia, file, NotFoundError } from "elysia";
 import { staticPlugin } from "@elysiajs/static";
-import { cors } from '@elysiajs/cors'
+import { cors } from "@elysiajs/cors";
 import { rateLimit } from "elysia-rate-limit";
 import Cap from "@cap.js/server";
 import crypto from "crypto";
@@ -44,6 +44,47 @@ if (ADMIN_KEY.length < 20) {
 let keys = [];
 const capInstances = new Map();
 
+const updateCache = async () => {
+  let cacheConfig = {};
+
+  try {
+    cacheConfig = JSON.parse(await fs.readFile(path.join(dataDir, "assets-cache.json"), "utf-8"));
+  } catch {}
+
+  const lastUpdate = cacheConfig["lastUpdate"] || 0;
+  const currentTime = Date.now();
+  const updateInterval = 1000 * 60 * 60 * 24; // 1 day
+
+  if (!(currentTime - lastUpdate > updateInterval)) return;
+
+  try {
+    const [widgetSource, floatingSource, wasmSource, wasmLoaderSource] = await Promise.all([
+      fetch("https://cdn.jsdelivr.net/npm/@cap.js/widget@latest").then((r) => r.text()),
+      fetch("https://cdn.jsdelivr.net/npm/@cap.js/widget/cap-floating.min.js").then((r) =>
+        r.text()
+      ),
+      fetch("https://cdn.jsdelivr.net/npm/@cap.js/wasm@0.0.3/browser/cap_wasm_bg.wasm").then((r) =>
+        r.arrayBuffer()
+      ),
+      fetch("https://cdn.jsdelivr.net/npm/@cap.js/wasm@0.0.3/browser/cap_wasm.min.js").then((r) =>
+        r.text()
+      ),
+    ]);
+
+    cacheConfig["lastUpdate"] = currentTime;
+    await fs.writeFile(path.join(dataDir, "assets-cache.json"), JSON.stringify(cacheConfig));
+
+    await fs.writeFile(path.join(dataDir, "assets-widget.js"), widgetSource);
+    await fs.writeFile(path.join(dataDir, "assets-floating.js"), floatingSource);
+    await fs.writeFile(path.join(dataDir, "assets-cap_wasm_bg.wasm"), Buffer.from(wasmSource));
+    await fs.writeFile(path.join(dataDir, "assets-cap_wasm.js"), wasmLoaderSource);
+
+    console.log("[asset server] updated assets cache");
+  } catch (e) {
+    console.error("[asset server] error updating assets cache, trying to load them might fail:", e);
+  }
+};
+
 const init = async () => {
   try {
     await fs.mkdir(dataDir, { recursive: true });
@@ -65,6 +106,8 @@ const init = async () => {
     keys = [];
     capInstances.clear();
   }
+
+  await updateCache();
 };
 
 const saveKeys = async () => {
@@ -182,14 +225,7 @@ const internal = new Elysia({ prefix: "/internal" })
   .post(
     "/editKey",
     async ({
-      body: {
-        publicKey,
-        keyName,
-        challengesCount,
-        challengeSize,
-        challengeDifficulty,
-        expiresMs,
-      },
+      body: { publicKey, keyName, challengesCount, challengeSize, challengeDifficulty, expiresMs },
       set,
     }) => {
       const keyIndex = keys.findIndex((key) => key.publicKey === publicKey);
@@ -240,10 +276,7 @@ const internal = new Elysia({ prefix: "/internal" })
       await fs.unlink(tokenFilePath);
     } catch (error) {
       if (error.code !== "ENOENT") {
-        console.warn(
-          `Could not delete token file ${tokenFilePath}:`,
-          error.message
-        );
+        console.warn(`Could not delete token file ${tokenFilePath}:`, error.message);
       }
     }
 
@@ -320,15 +353,33 @@ const api = new Elysia({ prefix: "/:key" })
     return await capInstance.validateToken(response);
   });
 
+const assetsServer = new Elysia({ prefix: "/assets" })
+  .get("/widget.js", ({ set }) => {
+    set.headers["Content-Type"] = "text/javascript";
+    return file(path.join(dataDir, "assets-widget.js"));
+  })
+  .get("/floating.js", ({ set }) => {
+    set.headers["Content-Type"] = "text/javascript";
+    return file(path.join(dataDir, "assets-floating.js"));
+  })
+  .get("/cap_wasm_bg.wasm", ({ set }) => {
+    set.headers["Content-Type"] = "application/wasm";
+    return file(path.join(dataDir, "assets-cap_wasm_bg.wasm"));
+  })
+  .get("/cap_wasm.js", ({ set }) => {
+    set.headers["Content-Type"] = "text/javascript";
+    return file(path.join(dataDir, "assets-cap_wasm.js"));
+  });
+
 new Elysia()
   .use(staticPlugin())
   .use(auth)
   .use(internal)
   .use(api)
+  .use(assetsServer)
   .get("/", async ({ cookie }) => {
     const authCookie = cookie["cap-admin-key"]?.value;
-    const isAuthed =
-      authCookie && (await Bun.password.verify(ADMIN_KEY, authCookie));
+    const isAuthed = authCookie && (await Bun.password.verify(ADMIN_KEY, authCookie));
 
     return file(isAuthed ? "./public/index.html" : "./public/lock.html");
   })
diff --git a/widget/src/src/cap.js b/widget/src/src/cap.js
index 056e635..4d3fe47 100644
--- a/widget/src/src/cap.js
+++ b/widget/src/src/cap.js
@@ -466,8 +466,6 @@
     let wasmCacheUrl, solve_pow_function;
 
     self.onmessage = async ({ data: { salt, target, wasmUrl } }) => {
-      console.log("[cap] worker message", { salt, target, wasmUrl });
-
       if (wasmCacheUrl !== wasmUrl) {
         wasmCacheUrl = wasmUrl;
         await import(wasmUrl)