Add searchState compatible js

huhu · Jun 20, 2024 · 7a0793b · 7a0793b
1 parent 2b40082
commit 7a0793b
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 3 deletions.
diff --git a/extension/content-script-bundle.js b/extension/content-script-bundle.js
diff --git a/extension/index-setter.js b/extension/index-setter.js
@@ -9,6 +9,10 @@ export default class IndexSetter {
         storage.setItem('index-std-nightly', index);
     }
 
+    static setDescShards(crate, shards) {
+        storage.setItem(`index-desc-shards-${crate}`, shards);
+    }
+
     static setBookIndex(index) {
         storage.setItem('index-book', index);
     }

diff --git a/extension/script/add-search-index.js b/extension/script/add-search-index.js
@@ -22,7 +22,7 @@
         return list;
     }
     async function loadDescShard(crates) {
-        if (!window.searchState.descShards) return;
+        if (!window.searchState.descShards) return null;
 
         let crateDescsShard = {};
         for (let crate of crates) {
@@ -35,6 +35,7 @@
         }
 
         console.log('load desc shard:', crateDescsShard);
+        return crateDescsShard;
     }
     async function sendSearchIndex() {
         if (location.hostname === "docs.rs") { // docs.rs pages
@@ -91,11 +92,11 @@
             STD_CRATES.forEach(crate => {
                 searchIndex[crate] = rawSearchIndex[crate];
             });
-            await loadDescShard(STD_CRATES);
             window.postMessage({
                 direction: `rust-search-extension:std`,
                 message: {
                     searchIndex,
+                    descShards: await loadDescShard(STD_CRATES),
                 },
             }, "*");
         }

diff --git a/extension/script/doc-rust-lang-org.js b/extension/script/doc-rust-lang-org.js
@@ -30,10 +30,17 @@ window.addEventListener("message", function (event) {
         event.data &&
         event.data.direction === "rust-search-extension:std") {
         let searchIndex = event.data.message.searchIndex;
+        let descShards = event.data.message.descShards;
         if (TARGET === 'stable') {
             rse.IndexSetter.setStdStableIndex(searchIndex);
+            if (descShards) {
+                rse.IndexSetter.setDescShards("std-stable", descShards);
+            }
         } else {
             rse.IndexSetter.setStdNightlyIndex(searchIndex);
+            if (descShards) {
+                rse.IndexSetter.setDescShards("std-nightly", descShards);
+            }
         }
         let now = new Date();
         let version = `${now.getFullYear()}-${now.getMonth() + 1}-${now.getDate()}`;

diff --git a/extension/search/docs/desc-shard.js b/extension/search/docs/desc-shard.js
@@ -0,0 +1,30 @@
+import CrateDocManager from "../../crate-manager.js";
+import IndexManager from "../../index-manager.js";
+
+class DescShardManager {
+    constructor() {
+        // A crate -> desc shard map.
+        this.descShards = new Map();
+    }
+
+    async initDescShards() {
+        for (const crate of Object.keys(await CrateDocManager.getCrates())) {
+            const descShards = await IndexManager.getDescShards(crate);
+            this.descShards.set(crate, descShards);
+        }
+    }
+
+
+    // Load a single desc shard.
+    // Compatible with librustdoc main.js.
+    async loadDesc({ descShard, descIndex }) {
+        let crateDescShard = this.descShards.get(descShard.crate);
+        if (!crateDescShard || crateDescShard.length === 0) {
+            return null;
+        }
+        return crateDescShard[descShard.shard][descIndex];
+    }
+}
+
+const searchState = new DescShardManager();
+export default searchState;