feat(search-pro): improve worker, close #2841

vuepress-theme-hope · Feb 28, 2023 · ef06dff · ef06dff
1 parent cf4706f
commit ef06dff
Show file tree

Hide file tree

Showing 5 changed files with 36 additions and 5 deletions.
diff --git a/packages/search-pro/rollup.config.ts b/packages/search-pro/rollup.config.ts
@@ -15,7 +15,7 @@ export default [
       copy: [["client/styles", "client"]],
     }
   ),
-  ...bundle("worker/index", {
+  ...bundle("worker/minify", {
     resolve: true,
     dts: false,
     external: [/^@internal\//],
@@ -24,4 +24,12 @@ export default [
       __VUEPRESS_SSR__: false,
     },
   }),
+  ...bundle("worker/original", {
+    resolve: true,
+    dts: false,
+    replace: {
+      // eslint-disable-next-line @typescript-eslint/naming-convention
+      __VUEPRESS_SSR__: false,
+    },
+  }),
 ];
diff --git a/packages/search-pro/src/node/generateWorker.ts b/packages/search-pro/src/node/generateWorker.ts
@@ -4,21 +4,29 @@ import { utoa } from "vuepress-shared/node";
 
 import { getSearchIndex } from "./generateIndex.js";
 import { type SearchProOptions } from "./options.js";
-import { WORKER_PATH } from "./utils.js";
+import { WORKER_FOLDER } from "./utils.js";
+
+const CONTENT_THRESHOLD = 8192;
 
 export const generateWorker = async (
   app: App,
   options: SearchProOptions
 ): Promise<void> => {
   const workerFilePath = app.dir.dest(options.worker || "search-pro.worker.js");
-  const workerFileContent = await fs.readFile(WORKER_PATH, "utf8");
+  const searchIndexContent = JSON.stringify(getSearchIndex(app, options));
+  const shouldMinify = searchIndexContent.length > CONTENT_THRESHOLD;
+  const workerPath = `${WORKER_FOLDER}${
+    shouldMinify ? "minify" : "original"
+  }.js`;
+
+  const workerFileContent = await fs.readFile(workerPath, "utf8");
 
   await fs.ensureDir(path.dirname(workerFilePath));
   await fs.writeFile(
     workerFilePath,
     workerFileContent.replace(
       "SEARCH_PRO_INDEX",
-      `"${utoa(JSON.stringify(getSearchIndex(app, options)))}"`
+      `"${shouldMinify ? utoa(searchIndexContent) : searchIndexContent}"`
     )
   );
 };
diff --git a/packages/search-pro/src/node/utils.ts b/packages/search-pro/src/node/utils.ts
@@ -8,7 +8,9 @@ export const CLIENT_FOLDER = ensureEndingSlash(
   path.resolve(__dirname, "../client/")
 );
 
-export const WORKER_PATH = path.resolve(__dirname, "../worker/index.js");
+export const WORKER_FOLDER = ensureEndingSlash(
+  path.resolve(__dirname, "../worker/")
+);
 
 export const PLUGIN_NAME = "vuepress-plugin-search-pro";
 

diff --git a/packages/search-pro/src/worker/index.ts → packages/search-pro/src/worker/minify.ts b/packages/search-pro/src/worker/index.ts → packages/search-pro/src/worker/minify.ts
diff --git a/packages/search-pro/src/worker/original.ts b/packages/search-pro/src/worker/original.ts
@@ -0,0 +1,13 @@
+import { getResults } from "../client/utils/index.js";
+import { type SearchIndex } from "../shared/index.js";
+
+const searchIndex: SearchIndex = <SearchIndex>JSON.parse(
+  // @ts-ignore
+  SEARCH_PRO_INDEX as string
+);
+
+self.onmessage = ({
+  data,
+}: MessageEvent<{ query: string; routeLocale: string }>): void => {
+  self.postMessage(getResults(data.query, searchIndex[data.routeLocale]));
+};