Feat DescShardManager

content-script-bundle.js

@@ -3,10 +3,12 @@ import storage from "./extension/core/storage.js";
 import settings from "./extension/settings.js";
 import IndexSetter from "./extension/index-setter.js";
 import CrateDocManager from "./extension/crate-manager.js";
+import { DescShardManager } from "./extension/search/docs/desc-shard.js";
 
 export {
     storage,
     settings,
     IndexSetter,
-    CrateDocManager
+    CrateDocManager,
+    DescShardManager,
 }

extension/index-setter.js

@@ -9,10 +9,6 @@ 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);
     }

extension/script/doc-rust-lang-org.js

@@ -34,12 +34,12 @@ window.addEventListener("message", function (event) {
         if (TARGET === 'stable') {
             rse.IndexSetter.setStdStableIndex(searchIndex);
             if (descShards) {
-                rse.IndexSetter.setDescShards("std-stable", descShards);
+                rse.DescShardManager.setDescShards("std-stable", descShards);
             }
         } else {
             rse.IndexSetter.setStdNightlyIndex(searchIndex);
             if (descShards) {
-                rse.IndexSetter.setDescShards("std-nightly", descShards);
+                rse.DescShardManager.setDescShards("std-nightly", descShards);
             }
         }
         let now = new Date();

extension/search/docs/desc-shard.js

@@ -1,15 +1,18 @@
 import CrateDocManager from "../../crate-manager.js";
-import IndexManager from "../../index-manager.js";
+import storage from "../../core/storage.js";
 
-class DescShardManager {
+export class DescShardManager {
     constructor() {
         // A crate -> desc shard map.
         this.descShards = new Map();
+        this.initDescShards();
     }
 
     async initDescShards() {
+        const stdDescShards = await DescShardManager.getDescShards('std-stable');
+        this.descShards = new Map(Object.entries(stdDescShards));
         for (const crate of Object.keys(await CrateDocManager.getCrates())) {
-            const descShards = await IndexManager.getDescShards(crate);
+            const descShards = await DescShardManager.getDescShards(crate);
             this.descShards.set(crate, descShards);
         }
     }
@@ -24,6 +27,14 @@ class DescShardManager {
         }
         return crateDescShard[descShard.shard][descIndex];
     }
+
+    static setDescShards(crate, shards) {
+        storage.setItem(`desc-shards-${crate}`, shards);
+    }
+
+    static async getDescShards(crate) {
+        return await storage.getItem(`desc-shards-${crate}`) || {};
+    }
 }
 
 const searchState = new DescShardManager();