@hot-updater/server 0.32.0 → 0.33.0

package/src/db/requestBundleIdentityMap.ts

@@ -0,0 +1,61 @@
+import type { Bundle } from "@hot-updater/core";
+import type { HotUpdaterContext } from "@hot-updater/plugin-core";
+
+type LoadBundleById<TContext> = (
+  bundleId: string,
+  context?: HotUpdaterContext<TContext>,
+) => Promise<Bundle | null>;
+
+type RequestBundleIdentityMapOptions<TContext> = {
+  readonly context?: HotUpdaterContext<TContext>;
+  readonly loadBundleById: LoadBundleById<TContext>;
+  readonly seeds: readonly (Bundle | null | undefined)[];
+};
+
+export const createRequestBundleIdentityMap = <TContext = unknown>({
+  context,
+  loadBundleById,
+  seeds,
+}: RequestBundleIdentityMapOptions<TContext>) => {
+  const bundles = new Map<string, Bundle>();
+  const pendingBundles = new Map<string, Promise<Bundle | null>>();
+
+  for (const seed of seeds) {
+    if (seed) {
+      bundles.set(seed.id, seed);
+    }
+  }
+
+  const get = async (bundleId: string): Promise<Bundle | null> => {
+    const cachedBundle = bundles.get(bundleId);
+    if (cachedBundle) {
+      return cachedBundle;
+    }
+
+    const pendingBundle = pendingBundles.get(bundleId);
+    if (pendingBundle) {
+      return pendingBundle;
+    }
+
+    const lookup = loadBundleById(bundleId, context).then(
+      (bundle) => {
+        pendingBundles.delete(bundleId);
+        if (bundle) {
+          bundles.set(bundle.id, bundle);
+        }
+        return bundle;
+      },
+      (error: unknown) => {
+        pendingBundles.delete(bundleId);
+        throw error;
+      },
+    );
+    pendingBundles.set(bundleId, lookup);
+    return lookup;
+  };
+
+  const peek = (bundleId: string): Bundle | null =>
+    bundles.get(bundleId) ?? null;
+
+  return { get, peek };
+};

package/src/db/types.ts

@@ -5,6 +5,7 @@ import type {
   UpdateInfo,
 } from "@hot-updater/core";
 import type {
+  DatabaseDiagnostics,
   DatabaseBundleQueryOptions,
   DatabasePlugin,
   HotUpdaterContext,
@@ -108,6 +109,7 @@ export interface DatabaseAPI<TContext = unknown> {
     bundleId: string,
     context?: HotUpdaterContext<TContext>,
   ): Promise<void>;
+  diagnostics?: DatabaseDiagnostics<TContext>;
 }
 
 export type StoragePluginFactory<TContext = unknown> =

package/src/runtime.spec.ts

@@ -706,6 +706,45 @@ describe("runtime createHotUpdater", () => {
     });
   });
 
+  it("keeps optional maintenance capabilities lazy", () => {
+    const factory = vi.fn(() => ({
+      async getBundleById() {
+        return null;
+      },
+      async getBundles() {
+        return {
+          data: [],
+          pagination: {
+            hasNextPage: false,
+            hasPreviousPage: false,
+            currentPage: 1,
+            totalPages: 1,
+            total: 0,
+          },
+        };
+      },
+      async getChannels() {
+        return [];
+      },
+      async commitBundle() {},
+    }));
+    const database = createDatabasePlugin({
+      name: "lazyRuntimePlugin",
+      factory,
+    })({});
+
+    const hotUpdater = createHotUpdater({
+      database,
+      basePath: "/api/check-update",
+    });
+
+    expect(factory).not.toHaveBeenCalled();
+    expect(hotUpdater.diagnostics).toBeUndefined();
+    expect(factory).not.toHaveBeenCalled();
+    expect(hotUpdater.diagnostics).toBeUndefined();
+    expect(factory).not.toHaveBeenCalled();
+  });
+
   it("clears pending plugin changes after a failed mutation commit", async () => {
     const committedBundles = new Map<string, Bundle>();
     let commitAttempt = 0;

package/src/runtime.ts

@@ -73,14 +73,10 @@ export function createHotUpdater<TContext = unknown>(
       : { readStorageText },
   );
 
-  const api = {
-    ...core.api,
-    handler: createHandler(core.api, {
-      basePath,
-      routes: options.routes,
-    }),
-    adapterName: core.adapterName,
-  };
+  const internalHandler = createHandler(core.api, {
+    basePath,
+    routes: options.routes,
+  });
 
   // Some framework adapters strip the mounted base path or pass extra
   // bindings/execution context arguments. Ignore those extras here so the
@@ -91,17 +87,19 @@ export function createHotUpdater<TContext = unknown>(
     ...extraArgs: unknown[]
   ) => {
     if (extraArgs.length > 0) {
-      return api.handler(request);
+      return internalHandler(request);
     }
 
-    return api.handler(request, context);
+    return internalHandler(request, context);
   };
 
-  return {
-    ...api,
+  const api = {
     basePath,
+    adapterName: core.adapterName,
     handler,
   };
+  Object.defineProperties(api, Object.getOwnPropertyDescriptors(core.api));
+  return api as HotUpdaterAPI<TContext>;
 }
 
 export { createHandler };