@hot-updater/server 0.36.7 → 1.0.0-rc.0

package/src/db/pluginCore.ts

@@ -1,475 +0,0 @@
-import type {
-  AppUpdateAvailableInfo,
-  AppVersionGetBundlesArgs,
-  Bundle,
-  FingerprintGetBundlesArgs,
-  GetBundlesArgs,
-  Platform,
-  UpdateInfo,
-} from "@hot-updater/core";
-import { isCohortEligibleForUpdate, NIL_UUID } from "@hot-updater/core";
-import {
-  type DatabaseBundleQueryOptions,
-  type DatabaseBundleQueryOrder,
-  type DatabaseBundleQueryWhere,
-  type DatabasePlugin,
-  createRequestUpdateBundleResolver,
-  type HotUpdaterContext,
-  semverSatisfies,
-} from "@hot-updater/plugin-core";
-
-import { assertBundlePersistenceConstraints } from "./schemaEnhancements";
-import type { DatabaseAPI } from "./types";
-import { resolveManifestArtifacts } from "./updateArtifacts";
-
-const PAGE_SIZE = 100;
-const DESC_ORDER = { field: "id", direction: "desc" } as const;
-
-const bundleMatchesQueryWhere = (
-  bundle: Bundle,
-  where: DatabaseBundleQueryWhere | undefined,
-) => {
-  if (!where) return true;
-  if (where.channel !== undefined && bundle.channel !== where.channel)
-    return false;
-  if (where.platform !== undefined && bundle.platform !== where.platform)
-    return false;
-  if (where.enabled !== undefined && bundle.enabled !== where.enabled)
-    return false;
-  if (where.id?.eq !== undefined && bundle.id !== where.id.eq) return false;
-  if (where.id?.gt !== undefined && bundle.id.localeCompare(where.id.gt) <= 0)
-    return false;
-  if (where.id?.gte !== undefined && bundle.id.localeCompare(where.id.gte) < 0)
-    return false;
-  if (where.id?.lt !== undefined && bundle.id.localeCompare(where.id.lt) >= 0)
-    return false;
-  if (where.id?.lte !== undefined && bundle.id.localeCompare(where.id.lte) > 0)
-    return false;
-  if (where.id?.in && !where.id.in.includes(bundle.id)) return false;
-  if (where.targetAppVersionNotNull && bundle.targetAppVersion === null) {
-    return false;
-  }
-  if (
-    where.targetAppVersion !== undefined &&
-    bundle.targetAppVersion !== where.targetAppVersion
-  ) {
-    return false;
-  }
-  if (
-    where.targetAppVersionIn &&
-    !where.targetAppVersionIn.includes(bundle.targetAppVersion ?? "")
-  ) {
-    return false;
-  }
-  if (
-    where.fingerprintHash !== undefined &&
-    bundle.fingerprintHash !== where.fingerprintHash
-  ) {
-    return false;
-  }
-  return true;
-};
-
-const sortBundles = (
-  bundles: Bundle[],
-  orderBy: DatabaseBundleQueryOrder | undefined,
-) => {
-  const direction = orderBy?.direction ?? "desc";
-  return bundles.slice().sort((a, b) => {
-    const result = a.id.localeCompare(b.id);
-    return direction === "asc" ? result : -result;
-  });
-};
-
-const makeResponse = (
-  bundle: Bundle,
-  status: "UPDATE" | "ROLLBACK",
-): UpdateInfo => ({
-  id: bundle.id,
-  message: bundle.message,
-  shouldForceUpdate: status === "ROLLBACK" ? true : bundle.shouldForceUpdate,
-  status,
-  storageUri: bundle.storageUri,
-  fileHash: bundle.fileHash,
-});
-
-const INIT_BUNDLE_ROLLBACK_UPDATE_INFO: UpdateInfo = {
-  message: null,
-  id: NIL_UUID,
-  shouldForceUpdate: true,
-  status: "ROLLBACK",
-  storageUri: null,
-  fileHash: null,
-};
-
-export function createPluginDatabaseCore<TContext = unknown>(
-  getPlugin: () => DatabasePlugin<TContext>,
-  resolveFileUrl: (
-    storageUri: string | null,
-    context?: HotUpdaterContext<TContext>,
-  ) => Promise<string | null>,
-  options?: {
-    createMutationPlugin?: () => DatabasePlugin<TContext>;
-    cleanupMutationPlugin?: (
-      plugin: DatabasePlugin<TContext>,
-    ) => Promise<void> | void;
-    beforeOperation?: () => Promise<void>;
-    readStorageText?: (
-      storageUri: string,
-      context?: HotUpdaterContext<TContext>,
-    ) => Promise<string | null>;
-  },
-): {
-  api: DatabaseAPI<TContext>;
-  adapterName: string;
-  createMigrator: () => never;
-  generateSchema: () => never;
-} {
-  const coreOptions = options;
-  const runWithMutationPlugin = async <T>(
-    operation: (plugin: DatabasePlugin<TContext>) => Promise<T>,
-  ): Promise<T> => {
-    const plugin = coreOptions?.createMutationPlugin?.() ?? getPlugin();
-
-    try {
-      return await operation(plugin);
-    } finally {
-      if (coreOptions?.createMutationPlugin) {
-        await coreOptions.cleanupMutationPlugin?.(plugin);
-      }
-    }
-  };
-
-  const getSortedBundlePage = async (
-    options: DatabaseBundleQueryOptions,
-    context?: HotUpdaterContext<TContext>,
-  ): Promise<Awaited<ReturnType<DatabasePlugin<TContext>["getBundles"]>>> => {
-    const result = await getPlugin().getBundles(
-      {
-        ...options,
-        orderBy: options.orderBy ?? DESC_ORDER,
-      },
-      context,
-    );
-
-    return {
-      ...result,
-      data: sortBundles(result.data, options.orderBy ?? DESC_ORDER),
-    };
-  };
-
-  const isEligibleForUpdate = (
-    bundle: Bundle,
-    cohort: string | undefined,
-  ): boolean => {
-    return isCohortEligibleForUpdate(
-      bundle.id,
-      cohort,
-      bundle.rolloutCohortCount,
-      bundle.targetCohorts,
-    );
-  };
-
-  const findUpdateInfoByScanning = async ({
-    args,
-    queryWhere,
-    isCandidate,
-    context,
-  }: {
-    args: AppVersionGetBundlesArgs | FingerprintGetBundlesArgs;
-    queryWhere: DatabaseBundleQueryWhere;
-    isCandidate: (bundle: Bundle) => boolean;
-    context?: HotUpdaterContext<TContext>;
-  }): Promise<UpdateInfo | null> => {
-    let after: string | undefined;
-
-    while (true) {
-      const { data, pagination } = await getSortedBundlePage(
-        {
-          where: queryWhere,
-          limit: PAGE_SIZE,
-          orderBy: DESC_ORDER,
-          ...(after
-            ? {
-                cursor: {
-                  after,
-                },
-              }
-            : {}),
-        },
-        context,
-      );
-
-      for (const bundle of data) {
-        if (
-          !bundleMatchesQueryWhere(bundle, queryWhere) ||
-          !isCandidate(bundle)
-        ) {
-          continue;
-        }
-
-        if (args.bundleId === NIL_UUID) {
-          if (isEligibleForUpdate(bundle, args.cohort)) {
-            return makeResponse(bundle, "UPDATE");
-          }
-          continue;
-        }
-
-        const compareResult = bundle.id.localeCompare(args.bundleId);
-
-        if (compareResult > 0) {
-          if (isEligibleForUpdate(bundle, args.cohort)) {
-            return makeResponse(bundle, "UPDATE");
-          }
-          continue;
-        }
-
-        if (compareResult === 0) {
-          if (isEligibleForUpdate(bundle, args.cohort)) {
-            return null;
-          }
-          continue;
-        }
-
-        return makeResponse(bundle, "ROLLBACK");
-      }
-
-      if (!pagination.hasNextPage) {
-        break;
-      }
-
-      after = data.at(-1)?.id;
-      if (!after) {
-        break;
-      }
-    }
-
-    if (args.bundleId === NIL_UUID) {
-      return null;
-    }
-
-    if (
-      args.minBundleId &&
-      args.bundleId.localeCompare(args.minBundleId) <= 0
-    ) {
-      return null;
-    }
-
-    return INIT_BUNDLE_ROLLBACK_UPDATE_INFO;
-  };
-
-  const getBaseWhere = ({
-    platform,
-    channel,
-    minBundleId,
-  }: {
-    platform: Platform;
-    channel: string;
-    minBundleId: string;
-  }): DatabaseBundleQueryWhere => ({
-    platform,
-    channel,
-    enabled: true,
-    id: {
-      gte: minBundleId,
-    },
-  });
-
-  const api: DatabaseAPI<TContext> = {
-    async getBundleById(
-      id: string,
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<Bundle | null> {
-      await coreOptions?.beforeOperation?.();
-      return getPlugin().getBundleById(id, context);
-    },
-
-    async getUpdateInfo(
-      args: GetBundlesArgs,
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<UpdateInfo | null> {
-      await coreOptions?.beforeOperation?.();
-      const plugin = getPlugin();
-      const directGetUpdateInfo = plugin.getUpdateInfo;
-      if (directGetUpdateInfo) {
-        return context === undefined
-          ? await directGetUpdateInfo(args)
-          : await directGetUpdateInfo(args, context);
-      }
-
-      const channel = args.channel ?? "production";
-      const minBundleId = args.minBundleId ?? NIL_UUID;
-      const baseWhere = getBaseWhere({
-        platform: args.platform,
-        channel,
-        minBundleId,
-      });
-
-      if (args._updateStrategy === "fingerprint") {
-        return findUpdateInfoByScanning({
-          args,
-          queryWhere: {
-            ...baseWhere,
-            fingerprintHash: args.fingerprintHash,
-          },
-          context,
-          isCandidate: (bundle) => {
-            return (
-              bundle.enabled &&
-              bundle.platform === args.platform &&
-              bundle.channel === channel &&
-              bundle.id.localeCompare(minBundleId) >= 0 &&
-              bundle.fingerprintHash === args.fingerprintHash
-            );
-          },
-        });
-      }
-
-      return findUpdateInfoByScanning({
-        args,
-        queryWhere: {
-          ...baseWhere,
-        },
-        context,
-        isCandidate: (bundle) => {
-          return (
-            bundle.enabled &&
-            bundle.platform === args.platform &&
-            bundle.channel === channel &&
-            bundle.id.localeCompare(minBundleId) >= 0 &&
-            !!bundle.targetAppVersion &&
-            semverSatisfies(bundle.targetAppVersion, args.appVersion)
-          );
-        },
-      });
-    },
-
-    async getAppUpdateInfo(
-      args: GetBundlesArgs,
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<AppUpdateAvailableInfo | null> {
-      const info = await this.getUpdateInfo(args, context);
-      if (!info) {
-        return null;
-      }
-      const { storageUri, ...rest } = info;
-
-      const readStorageText = coreOptions?.readStorageText;
-      if (info.id === NIL_UUID || !readStorageText) {
-        const fileUrl = await resolveFileUrl(storageUri ?? null, context);
-        const baseResponse: AppUpdateAvailableInfo = { ...rest, fileUrl };
-        return baseResponse;
-      }
-
-      const requestBundles = createRequestUpdateBundleResolver(context);
-      const getCurrentBundle = () => {
-        if (args.bundleId === NIL_UUID) {
-          return null;
-        }
-
-        const seededCurrentBundle = requestBundles.peek(args.bundleId);
-        if (seededCurrentBundle || requestBundles.hasSeededBundles()) {
-          return seededCurrentBundle;
-        }
-
-        return requestBundles.getById(args.bundleId, () =>
-          getPlugin().getBundleById(args.bundleId, context),
-        );
-      };
-      const [fileUrl, targetBundle, currentBundle] = await Promise.all([
-        resolveFileUrl(storageUri ?? null, context),
-        requestBundles.getById(info.id, () =>
-          getPlugin().getBundleById(info.id, context),
-        ),
-        getCurrentBundle(),
-      ]);
-      const baseResponse: AppUpdateAvailableInfo = { ...rest, fileUrl };
-      const manifestArtifacts = await resolveManifestArtifacts({
-        currentBundle,
-        resolveFileUrl,
-        readStorageText,
-        targetBundle,
-        context,
-      });
-      if (!manifestArtifacts) {
-        return baseResponse;
-      }
-
-      return {
-        ...baseResponse,
-        ...manifestArtifacts,
-      };
-    },
-
-    async getChannels(
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<string[]> {
-      await coreOptions?.beforeOperation?.();
-      return getPlugin().getChannels(context);
-    },
-
-    async getBundles(options, context?: HotUpdaterContext<TContext>) {
-      await coreOptions?.beforeOperation?.();
-      return getPlugin().getBundles(options, context);
-    },
-
-    async insertBundle(
-      bundle: Bundle,
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<void> {
-      await coreOptions?.beforeOperation?.();
-      assertBundlePersistenceConstraints(bundle);
-      await runWithMutationPlugin(async (plugin) => {
-        await plugin.appendBundle(bundle, context);
-        await plugin.commitBundle(context);
-      });
-    },
-
-    async updateBundleById(
-      bundleId: string,
-      newBundle: Partial<Bundle>,
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<void> {
-      await coreOptions?.beforeOperation?.();
-      await runWithMutationPlugin(async (plugin) => {
-        const current = await plugin.getBundleById(bundleId, context);
-        if (!current) {
-          throw new Error("targetBundleId not found");
-        }
-        assertBundlePersistenceConstraints({ ...current, ...newBundle });
-        await plugin.updateBundle(bundleId, newBundle, context);
-        await plugin.commitBundle(context);
-      });
-    },
-
-    async deleteBundleById(
-      bundleId: string,
-      context?: HotUpdaterContext<TContext>,
-    ): Promise<void> {
-      await coreOptions?.beforeOperation?.();
-      await runWithMutationPlugin(async (plugin) => {
-        const bundle = await plugin.getBundleById(bundleId, context);
-        if (!bundle) {
-          return;
-        }
-        await plugin.deleteBundle(bundle, context);
-        await plugin.commitBundle(context);
-      });
-    },
-  };
-
-  return {
-    api,
-    adapterName: getPlugin().name,
-    createMigrator: () => {
-      throw new Error(
-        "createMigrator is only available for Kysely/MongoDB database adapters.",
-      );
-    },
-    generateSchema: () => {
-      throw new Error(
-        "generateSchema is only available for Drizzle/Prisma database adapters.",
-      );
-    },
-  };
-}

package/src/db/pluginUpdateCheck.bench.ts

@@ -1,254 +0,0 @@
-import { bench, describe } from "vitest";
-
-import type {
-  DatabaseBundleQueryOptions,
-  DatabaseBundleQueryWhere,
-  DatabasePlugin,
-} from "../../../../plugins/plugin-core/src";
-import {
-  paginateBundles,
-  semverSatisfies,
-} from "../../../../plugins/plugin-core/src";
-import type {
-  AppVersionGetBundlesArgs,
-  Bundle,
-  Platform,
-  UpdateInfo,
-} from "../../../core/src";
-import {
-  DEFAULT_ROLLOUT_COHORT_COUNT,
-  isCohortEligibleForUpdate,
-  NIL_UUID,
-} from "../../../core/src";
-import { createPluginDatabaseCore } from "./pluginCore";
-
-const BUNDLE_COUNT = 20_000;
-const BENCH_APP_VERSION = "1.0.0";
-const BENCH_PLATFORM = "ios" as const;
-const BENCH_CHANNEL = "production";
-
-const cloneBundle = (bundle: Bundle): Bundle => ({
-  ...bundle,
-  metadata: bundle.metadata
-    ? structuredClone(bundle.metadata)
-    : bundle.metadata,
-  targetCohorts: bundle.targetCohorts ? [...bundle.targetCohorts] : null,
-});
-
-const bundleMatchesWhere = (
-  bundle: Bundle,
-  where: DatabaseBundleQueryWhere | undefined,
-) => {
-  if (!where) return true;
-  if (where.channel !== undefined && bundle.channel !== where.channel)
-    return false;
-  if (where.platform !== undefined && bundle.platform !== where.platform)
-    return false;
-  if (where.enabled !== undefined && bundle.enabled !== where.enabled)
-    return false;
-  if (where.id?.eq !== undefined && bundle.id !== where.id.eq) return false;
-  if (where.id?.gt !== undefined && bundle.id.localeCompare(where.id.gt) <= 0)
-    return false;
-  if (where.id?.gte !== undefined && bundle.id.localeCompare(where.id.gte) < 0)
-    return false;
-  if (where.id?.lt !== undefined && bundle.id.localeCompare(where.id.lt) >= 0)
-    return false;
-  if (where.id?.lte !== undefined && bundle.id.localeCompare(where.id.lte) > 0)
-    return false;
-  if (where.id?.in && !where.id.in.includes(bundle.id)) return false;
-  if (where.targetAppVersionNotNull && bundle.targetAppVersion === null) {
-    return false;
-  }
-  if (
-    where.targetAppVersion !== undefined &&
-    bundle.targetAppVersion !== where.targetAppVersion
-  ) {
-    return false;
-  }
-  if (
-    where.targetAppVersionIn &&
-    !where.targetAppVersionIn.includes(bundle.targetAppVersion ?? "")
-  ) {
-    return false;
-  }
-  if (
-    where.fingerprintHash !== undefined &&
-    bundle.fingerprintHash !== where.fingerprintHash
-  ) {
-    return false;
-  }
-  return true;
-};
-
-const createBundle = (
-  index: number,
-  {
-    platform = BENCH_PLATFORM,
-    channel = BENCH_CHANNEL,
-    targetAppVersion = "*",
-    enabled = true,
-  }: {
-    platform?: Platform;
-    channel?: string;
-    targetAppVersion?: string | null;
-    enabled?: boolean;
-  } = {},
-): Bundle => ({
-  id: `00000000-0000-0000-0000-${String(index).padStart(12, "0")}`,
-  platform,
-  shouldForceUpdate: false,
-  enabled,
-  fileHash: `hash-${index}`,
-  gitCommitHash: `commit-${index}`,
-  message: `bundle-${index}`,
-  channel,
-  storageUri: `s3://bench/bundles/${index}.zip`,
-  targetAppVersion,
-  fingerprintHash: `fingerprint-${index % 10}`,
-  metadata: { app_version: String(index) },
-  rolloutCohortCount: DEFAULT_ROLLOUT_COHORT_COUNT,
-  targetCohorts: null,
-});
-
-const createBenchPlugin = (bundles: Bundle[]): DatabasePlugin => {
-  const bundlesById = new Map(bundles.map((bundle) => [bundle.id, bundle]));
-
-  const sortByDirection = (direction: "asc" | "desc" | undefined): Bundle[] => {
-    const sorted = bundles.slice().sort((a, b) => a.id.localeCompare(b.id));
-    return direction === "asc" ? sorted : sorted.reverse();
-  };
-
-  return {
-    name: "bench-plugin",
-    async getBundleById(bundleId) {
-      return bundlesById.get(bundleId) ?? null;
-    },
-    async getBundles(options: DatabaseBundleQueryOptions) {
-      const { where, limit, cursor, orderBy } = options;
-      const source = sortByDirection(orderBy?.direction);
-      const matched = source.filter((bundle) =>
-        bundleMatchesWhere(bundle, where),
-      );
-      const paginated = paginateBundles({
-        bundles: matched.map(cloneBundle),
-        limit,
-        cursor,
-        orderBy,
-      });
-
-      return paginated;
-    },
-    async getChannels() {
-      return [...new Set(bundles.map((bundle) => bundle.channel))];
-    },
-    async updateBundle() {
-      throw new Error("Not implemented for benchmark");
-    },
-    async appendBundle() {
-      throw new Error("Not implemented for benchmark");
-    },
-    async commitBundle() {},
-    async deleteBundle() {
-      throw new Error("Not implemented for benchmark");
-    },
-  };
-};
-
-const oldPluginCoreGetUpdateInfo = async (
-  plugin: DatabasePlugin,
-  args: AppVersionGetBundlesArgs,
-): Promise<UpdateInfo | null> => {
-  const where: DatabaseBundleQueryWhere = {
-    channel: args.channel ?? BENCH_CHANNEL,
-    platform: args.platform,
-  };
-
-  const { pagination } = await plugin.getBundles({
-    where,
-    limit: 1,
-  });
-
-  if (pagination.total === 0) {
-    return null;
-  }
-
-  const { data } = await plugin.getBundles({
-    where,
-    limit: pagination.total,
-  });
-
-  for (const bundle of data) {
-    if (!bundle.enabled) {
-      continue;
-    }
-    if (bundle.platform !== args.platform) {
-      continue;
-    }
-    if (bundle.channel !== (args.channel ?? BENCH_CHANNEL)) {
-      continue;
-    }
-    if (!bundle.targetAppVersion) {
-      continue;
-    }
-    if (!semverSatisfies(bundle.targetAppVersion, args.appVersion)) {
-      continue;
-    }
-    if (
-      !isCohortEligibleForUpdate(
-        bundle.id,
-        args.cohort,
-        bundle.rolloutCohortCount,
-        bundle.targetCohorts,
-      )
-    ) {
-      continue;
-    }
-
-    return {
-      id: bundle.id,
-      message: bundle.message,
-      shouldForceUpdate: bundle.shouldForceUpdate,
-      status: "UPDATE",
-      storageUri: bundle.storageUri,
-      fileHash: bundle.fileHash,
-    };
-  }
-
-  return null;
-};
-
-describe("plugin update check benchmark", () => {
-  const bundles = Array.from({ length: BUNDLE_COUNT }, (_, index) =>
-    createBundle(index + 1),
-  );
-
-  const args: AppVersionGetBundlesArgs = {
-    _updateStrategy: "appVersion",
-    appVersion: BENCH_APP_VERSION,
-    bundleId: NIL_UUID,
-    platform: BENCH_PLATFORM,
-    channel: BENCH_CHANNEL,
-  };
-
-  const plugin = createBenchPlugin(bundles);
-  const currentApi = createPluginDatabaseCore(
-    () => plugin,
-    async () => null,
-  ).api;
-
-  bench(
-    "pluginCore legacy full fetch",
-    async () => {
-      await oldPluginCoreGetUpdateInfo(plugin, args);
-    },
-    { warmupIterations: 5, iterations: 20 },
-  );
-
-  bench(
-    "pluginCore current paged fetch",
-    async () => {
-      await currentApi.getUpdateInfo(args);
-    },
-    { warmupIterations: 5, iterations: 20 },
-  );
-});