@hot-updater/server 0.28.0 → 0.29.0

package/dist/db/ormCore.mjs

@@ -0,0 +1,386 @@
+import { calculatePagination } from "../calculatePagination.mjs";
+import { v0_21_0 } from "../schema/v0_21_0.mjs";
+import { v0_29_0 } from "../schema/v0_29_0.mjs";
+import { DEFAULT_ROLLOUT_COHORT_COUNT, NIL_UUID, isCohortEligibleForUpdate } from "@hot-updater/core";
+import { semverSatisfies } from "@hot-updater/plugin-core";
+import { fumadb } from "fumadb";
+//#region src/db/ormCore.ts
+const parseTargetCohorts = (value) => {
+	if (!value) return null;
+	if (Array.isArray(value)) return value.filter((v) => typeof v === "string");
+	if (typeof value === "string") try {
+		const parsed = JSON.parse(value);
+		if (Array.isArray(parsed)) return parsed.filter((v) => typeof v === "string");
+	} catch {
+		return null;
+	}
+	return null;
+};
+const schemas = [v0_21_0, v0_29_0];
+const getLastItem = (items) => items.at(-1);
+const HotUpdaterDB = fumadb({
+	namespace: "hot_updater",
+	schemas
+});
+function createOrmDatabaseCore({ database, resolveFileUrl }) {
+	const client = HotUpdaterDB.client(database);
+	const UPDATE_CHECK_PAGE_SIZE = 100;
+	const isMongoAdapter = client.adapter.name.toLowerCase().includes("mongodb");
+	const ensureORM = async () => {
+		const lastSchemaVersion = getLastItem(schemas).version;
+		try {
+			const currentVersion = await client.createMigrator().getVersion();
+			if (currentVersion === void 0) throw new Error("Database is not initialized. Please run 'npx hot-updater migrate' to set up the database schema.");
+			if (currentVersion !== lastSchemaVersion) throw new Error(`Database schema version mismatch. Expected version ${lastSchemaVersion}, but database is on version ${currentVersion}. Please run 'npx hot-updater migrate' to update your database schema.`);
+			return client.orm(lastSchemaVersion);
+		} catch (error) {
+			if (error instanceof Error && error.message.includes("doesn't support migration")) return client.orm(lastSchemaVersion);
+			throw error;
+		}
+	};
+	const buildBundleWhere = (where) => (b) => {
+		if (where?.id?.in && where.id.in.length === 0) return false;
+		if (where?.targetAppVersionIn && where.targetAppVersionIn.length === 0) return false;
+		const conditions = [];
+		if (where?.channel !== void 0) conditions.push(b("channel", "=", where.channel));
+		if (where?.platform !== void 0) conditions.push(b("platform", "=", where.platform));
+		if (where?.enabled !== void 0) conditions.push(b("enabled", "=", where.enabled));
+		if (where?.id?.eq !== void 0) conditions.push(b("id", "=", where.id.eq));
+		if (where?.id?.gt !== void 0) conditions.push(b("id", ">", where.id.gt));
+		if (where?.id?.gte !== void 0) conditions.push(b("id", ">=", where.id.gte));
+		if (where?.id?.lt !== void 0) conditions.push(b("id", "<", where.id.lt));
+		if (where?.id?.lte !== void 0) conditions.push(b("id", "<=", where.id.lte));
+		if (where?.id?.in) conditions.push(b("id", "in", where.id.in));
+		if (where?.targetAppVersionNotNull) conditions.push(b.isNotNull("target_app_version"));
+		if (where?.targetAppVersion !== void 0) conditions.push(where.targetAppVersion === null ? b.isNull("target_app_version") : b("target_app_version", "=", where.targetAppVersion));
+		if (where?.targetAppVersionIn) conditions.push(b("target_app_version", "in", where.targetAppVersionIn));
+		if (where?.fingerprintHash !== void 0) conditions.push(where.fingerprintHash === null ? b.isNull("fingerprint_hash") : b("fingerprint_hash", "=", where.fingerprintHash));
+		return conditions.length > 0 ? b.and(...conditions) : true;
+	};
+	return {
+		api: {
+			async getBundleById(id) {
+				const result = await (await ensureORM()).findFirst("bundles", {
+					select: [
+						"id",
+						"platform",
+						"should_force_update",
+						"enabled",
+						"file_hash",
+						"git_commit_hash",
+						"message",
+						"channel",
+						"storage_uri",
+						"target_app_version",
+						"fingerprint_hash",
+						"metadata",
+						"rollout_cohort_count",
+						"target_cohorts"
+					],
+					where: (b) => b("id", "=", id)
+				});
+				if (!result) return null;
+				return {
+					id: result.id,
+					platform: result.platform,
+					shouldForceUpdate: Boolean(result.should_force_update),
+					enabled: Boolean(result.enabled),
+					fileHash: result.file_hash,
+					gitCommitHash: result.git_commit_hash ?? null,
+					message: result.message ?? null,
+					channel: result.channel,
+					storageUri: result.storage_uri,
+					targetAppVersion: result.target_app_version ?? null,
+					fingerprintHash: result.fingerprint_hash ?? null,
+					rolloutCohortCount: result.rollout_cohort_count ?? DEFAULT_ROLLOUT_COHORT_COUNT,
+					targetCohorts: parseTargetCohorts(result.target_cohorts)
+				};
+			},
+			async getUpdateInfo(args) {
+				const orm = await ensureORM();
+				const toUpdateInfo = (row, status) => ({
+					id: row.id,
+					shouldForceUpdate: status === "ROLLBACK" ? true : Boolean(row.should_force_update),
+					message: row.message ?? null,
+					status,
+					storageUri: row.storage_uri ?? null,
+					fileHash: row.file_hash ?? null
+				});
+				const INIT_BUNDLE_ROLLBACK_UPDATE_INFO = {
+					id: NIL_UUID,
+					message: null,
+					shouldForceUpdate: true,
+					status: "ROLLBACK",
+					storageUri: null,
+					fileHash: null
+				};
+				const isEligibleForUpdate = (row, cohort) => {
+					return isCohortEligibleForUpdate(row.id, cohort, row.rollout_cohort_count ?? null, parseTargetCohorts(row.target_cohorts));
+				};
+				const findUpdateInfoByScanning = async ({ args, where, isCandidate }) => {
+					if (isMongoAdapter) {
+						const rows = await orm.findMany("bundles", {
+							select: [
+								"id",
+								"should_force_update",
+								"message",
+								"storage_uri",
+								"file_hash",
+								"rollout_cohort_count",
+								"target_cohorts",
+								"target_app_version",
+								"fingerprint_hash"
+							],
+							where: buildBundleWhere(where)
+						});
+						rows.sort((a, b) => b.id.localeCompare(a.id));
+						for (const row of rows) {
+							if (!isCandidate(row)) continue;
+							if (args.bundleId === NIL_UUID) {
+								if (isEligibleForUpdate(row, args.cohort)) return toUpdateInfo(row, "UPDATE");
+								continue;
+							}
+							const compareResult = row.id.localeCompare(args.bundleId);
+							if (compareResult > 0) {
+								if (isEligibleForUpdate(row, args.cohort)) return toUpdateInfo(row, "UPDATE");
+								continue;
+							}
+							if (compareResult === 0) {
+								if (isEligibleForUpdate(row, args.cohort)) return null;
+								continue;
+							}
+							return toUpdateInfo(row, "ROLLBACK");
+						}
+						if (args.bundleId === NIL_UUID) return null;
+						if (args.minBundleId && args.bundleId.localeCompare(args.minBundleId) <= 0) return null;
+						return INIT_BUNDLE_ROLLBACK_UPDATE_INFO;
+					}
+					let offset = 0;
+					while (true) {
+						const rows = await orm.findMany("bundles", {
+							select: [
+								"id",
+								"should_force_update",
+								"message",
+								"storage_uri",
+								"file_hash",
+								"rollout_cohort_count",
+								"target_cohorts",
+								"target_app_version",
+								"fingerprint_hash"
+							],
+							where: buildBundleWhere(where),
+							orderBy: [["id", "desc"]],
+							limit: UPDATE_CHECK_PAGE_SIZE,
+							offset
+						});
+						for (const row of rows) {
+							if (!isCandidate(row)) continue;
+							if (args.bundleId === NIL_UUID) {
+								if (isEligibleForUpdate(row, args.cohort)) return toUpdateInfo(row, "UPDATE");
+								continue;
+							}
+							const compareResult = row.id.localeCompare(args.bundleId);
+							if (compareResult > 0) {
+								if (isEligibleForUpdate(row, args.cohort)) return toUpdateInfo(row, "UPDATE");
+								continue;
+							}
+							if (compareResult === 0) {
+								if (isEligibleForUpdate(row, args.cohort)) return null;
+								continue;
+							}
+							return toUpdateInfo(row, "ROLLBACK");
+						}
+						if (rows.length < UPDATE_CHECK_PAGE_SIZE) break;
+						offset += UPDATE_CHECK_PAGE_SIZE;
+					}
+					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 appVersionStrategy = async ({ platform, appVersion, bundleId, minBundleId = NIL_UUID, channel = "production", cohort }) => {
+					return findUpdateInfoByScanning({
+						args: {
+							_updateStrategy: "appVersion",
+							platform,
+							appVersion,
+							bundleId,
+							minBundleId,
+							channel,
+							cohort
+						},
+						where: {
+							enabled: true,
+							platform,
+							channel,
+							id: { gte: minBundleId },
+							targetAppVersionNotNull: true
+						},
+						isCandidate: (row) => !!row.target_app_version && semverSatisfies(row.target_app_version, appVersion)
+					});
+				};
+				const fingerprintStrategy = async ({ platform, fingerprintHash, bundleId, minBundleId = NIL_UUID, channel = "production", cohort }) => {
+					return findUpdateInfoByScanning({
+						args: {
+							_updateStrategy: "fingerprint",
+							platform,
+							fingerprintHash,
+							bundleId,
+							minBundleId,
+							channel,
+							cohort
+						},
+						where: {
+							enabled: true,
+							platform,
+							channel,
+							id: { gte: minBundleId },
+							fingerprintHash
+						},
+						isCandidate: (row) => row.fingerprint_hash === fingerprintHash
+					});
+				};
+				if (args._updateStrategy === "appVersion") return appVersionStrategy(args);
+				if (args._updateStrategy === "fingerprint") return fingerprintStrategy(args);
+				return null;
+			},
+			async getAppUpdateInfo(args, context) {
+				const info = await this.getUpdateInfo(args);
+				if (!info) return null;
+				const { storageUri, ...rest } = info;
+				const fileUrl = await resolveFileUrl(storageUri ?? null, context);
+				return {
+					...rest,
+					fileUrl
+				};
+			},
+			async getChannels() {
+				const rows = await (await ensureORM()).findMany("bundles", {
+					select: ["channel"],
+					orderBy: [["channel", "asc"]]
+				});
+				const set = new Set(rows?.map((r) => r.channel) ?? []);
+				return Array.from(set);
+			},
+			async getBundles(options) {
+				const orm = await ensureORM();
+				const { where, limit, offset, orderBy } = options;
+				const total = await orm.count("bundles", { where: buildBundleWhere(where) });
+				const selectedColumns = [
+					"id",
+					"platform",
+					"should_force_update",
+					"enabled",
+					"file_hash",
+					"git_commit_hash",
+					"message",
+					"channel",
+					"storage_uri",
+					"target_app_version",
+					"fingerprint_hash",
+					"metadata",
+					"rollout_cohort_count",
+					"target_cohorts"
+				];
+				return {
+					data: (isMongoAdapter ? (await orm.findMany("bundles", {
+						select: selectedColumns,
+						where: buildBundleWhere(where)
+					})).sort((a, b) => {
+						const direction = orderBy?.direction ?? "desc";
+						const result = a.id.localeCompare(b.id);
+						return direction === "asc" ? result : -result;
+					}).slice(offset, offset + limit) : await orm.findMany("bundles", {
+						select: selectedColumns,
+						where: buildBundleWhere(where),
+						orderBy: [[orderBy?.field ?? "id", orderBy?.direction ?? "desc"]],
+						limit,
+						offset
+					})).map((r) => ({
+						id: r.id,
+						platform: r.platform,
+						shouldForceUpdate: Boolean(r.should_force_update),
+						enabled: Boolean(r.enabled),
+						fileHash: r.file_hash,
+						gitCommitHash: r.git_commit_hash ?? null,
+						message: r.message ?? null,
+						channel: r.channel,
+						storageUri: r.storage_uri,
+						targetAppVersion: r.target_app_version ?? null,
+						fingerprintHash: r.fingerprint_hash ?? null,
+						rolloutCohortCount: r.rollout_cohort_count ?? DEFAULT_ROLLOUT_COHORT_COUNT,
+						targetCohorts: parseTargetCohorts(r.target_cohorts)
+					})),
+					pagination: calculatePagination(total, {
+						limit,
+						offset
+					})
+				};
+			},
+			async insertBundle(bundle) {
+				const orm = await ensureORM();
+				const values = {
+					id: bundle.id,
+					platform: bundle.platform,
+					should_force_update: bundle.shouldForceUpdate,
+					enabled: bundle.enabled,
+					file_hash: bundle.fileHash,
+					git_commit_hash: bundle.gitCommitHash,
+					message: bundle.message,
+					channel: bundle.channel,
+					storage_uri: bundle.storageUri,
+					target_app_version: bundle.targetAppVersion,
+					fingerprint_hash: bundle.fingerprintHash,
+					metadata: bundle.metadata ?? {},
+					rollout_cohort_count: bundle.rolloutCohortCount ?? DEFAULT_ROLLOUT_COHORT_COUNT,
+					target_cohorts: bundle.targetCohorts ?? null
+				};
+				const { id, ...updateValues } = values;
+				await orm.upsert("bundles", {
+					where: (b) => b("id", "=", id),
+					create: values,
+					update: updateValues
+				});
+			},
+			async updateBundleById(bundleId, newBundle) {
+				const orm = await ensureORM();
+				const current = await this.getBundleById(bundleId);
+				if (!current) throw new Error("targetBundleId not found");
+				const merged = {
+					...current,
+					...newBundle
+				};
+				const values = {
+					id: merged.id,
+					platform: merged.platform,
+					should_force_update: merged.shouldForceUpdate,
+					enabled: merged.enabled,
+					file_hash: merged.fileHash,
+					git_commit_hash: merged.gitCommitHash,
+					message: merged.message,
+					channel: merged.channel,
+					storage_uri: merged.storageUri,
+					target_app_version: merged.targetAppVersion,
+					fingerprint_hash: merged.fingerprintHash,
+					metadata: merged.metadata ?? {},
+					rollout_cohort_count: merged.rolloutCohortCount ?? DEFAULT_ROLLOUT_COHORT_COUNT,
+					target_cohorts: merged.targetCohorts ?? null
+				};
+				const { id: id2, ...updateValues2 } = values;
+				await orm.upsert("bundles", {
+					where: (b) => b("id", "=", id2),
+					create: values,
+					update: updateValues2
+				});
+			},
+			async deleteBundleById(bundleId) {
+				await (await ensureORM()).deleteMany("bundles", { where: (b) => b("id", "=", bundleId) });
+			}
+		},
+		adapterName: client.adapter.name,
+		createMigrator: () => client.createMigrator(),
+		generateSchema: client.generateSchema
+	};
+}
+//#endregion
+export { HotUpdaterDB, createOrmDatabaseCore };

package/dist/db/pluginCore.cjs

@@ -1,60 +1,166 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
-let __hot_updater_js = require("@hot-updater/js");
-__hot_updater_js = require_rolldown_runtime.__toESM(__hot_updater_js);
-
+let _hot_updater_core = require("@hot-updater/core");
+let _hot_updater_plugin_core = require("@hot-updater/plugin-core");
 //#region src/db/pluginCore.ts
+const PAGE_SIZE = 100;
+const DESC_ORDER = {
+	field: "id",
+	direction: "desc"
+};
+const bundleMatchesQueryWhere = (bundle, where) => {
+	if (!where) return true;
+	if (where.channel !== void 0 && bundle.channel !== where.channel) return false;
+	if (where.platform !== void 0 && bundle.platform !== where.platform) return false;
+	if (where.enabled !== void 0 && bundle.enabled !== where.enabled) return false;
+	if (where.id?.eq !== void 0 && bundle.id !== where.id.eq) return false;
+	if (where.id?.gt !== void 0 && bundle.id.localeCompare(where.id.gt) <= 0) return false;
+	if (where.id?.gte !== void 0 && bundle.id.localeCompare(where.id.gte) < 0) return false;
+	if (where.id?.lt !== void 0 && bundle.id.localeCompare(where.id.lt) >= 0) return false;
+	if (where.id?.lte !== void 0 && 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 !== void 0 && bundle.targetAppVersion !== where.targetAppVersion) return false;
+	if (where.targetAppVersionIn && !where.targetAppVersionIn.includes(bundle.targetAppVersion ?? "")) return false;
+	if (where.fingerprintHash !== void 0 && bundle.fingerprintHash !== where.fingerprintHash) return false;
+	return true;
+};
+const sortBundles = (bundles, orderBy) => {
+	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, status) => ({
+	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 = {
+	message: null,
+	id: _hot_updater_core.NIL_UUID,
+	shouldForceUpdate: true,
+	status: "ROLLBACK",
+	storageUri: null,
+	fileHash: null
+};
 function createPluginDatabaseCore(plugin, resolveFileUrl) {
+	const getSortedBundlePage = async (options, context) => {
+		const result = await plugin.getBundles({
+			...options,
+			orderBy: options.orderBy ?? DESC_ORDER
+		}, context);
+		return {
+			...result,
+			data: sortBundles(result.data, options.orderBy ?? DESC_ORDER)
+		};
+	};
+	const isEligibleForUpdate = (bundle, cohort) => {
+		return (0, _hot_updater_core.isCohortEligibleForUpdate)(bundle.id, cohort, bundle.rolloutCohortCount, bundle.targetCohorts);
+	};
+	const findUpdateInfoByScanning = async ({ args, queryWhere, isCandidate, context }) => {
+		let offset = 0;
+		while (true) {
+			const { data, pagination } = await getSortedBundlePage({
+				where: queryWhere,
+				limit: PAGE_SIZE,
+				offset,
+				orderBy: DESC_ORDER
+			}, context);
+			for (const bundle of data) {
+				if (!bundleMatchesQueryWhere(bundle, queryWhere) || !isCandidate(bundle)) continue;
+				if (args.bundleId === _hot_updater_core.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;
+			offset += PAGE_SIZE;
+		}
+		if (args.bundleId === _hot_updater_core.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,
+		channel,
+		enabled: true,
+		id: { gte: minBundleId }
+	});
 	return {
 		api: {
-			async getBundleById(id) {
-				return plugin.getBundleById(id);
+			async getBundleById(id, context) {
+				return plugin.getBundleById(id, context);
 			},
-			async getUpdateInfo(args) {
-				const where = {};
-				if ("platform" in args && args.platform) where.platform = args.platform;
-				where.channel = "channel" in args && args.channel ? args.channel : "production";
-				const { pagination } = await plugin.getBundles({
-					where,
-					limit: 1,
-					offset: 0
+			async getUpdateInfo(args, context) {
+				const channel = args.channel ?? "production";
+				const minBundleId = args.minBundleId ?? _hot_updater_core.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;
+					}
 				});
-				if (pagination.total === 0) return (0, __hot_updater_js.getUpdateInfo)([], args);
-				const { data } = await plugin.getBundles({
-					where,
-					limit: pagination.total,
-					offset: 0
+				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 && (0, _hot_updater_plugin_core.semverSatisfies)(bundle.targetAppVersion, args.appVersion);
+					}
 				});
-				return (0, __hot_updater_js.getUpdateInfo)(data, args);
 			},
-			async getAppUpdateInfo(args) {
-				const info = await this.getUpdateInfo(args);
+			async getAppUpdateInfo(args, context) {
+				const info = await this.getUpdateInfo(args, context);
 				if (!info) return null;
-				const { storageUri,...rest } = info;
-				const fileUrl = await resolveFileUrl(storageUri ?? null);
+				const { storageUri, ...rest } = info;
+				const fileUrl = await resolveFileUrl(storageUri ?? null, context);
 				return {
 					...rest,
 					fileUrl
 				};
 			},
-			async getChannels() {
-				return plugin.getChannels();
+			async getChannels(context) {
+				return plugin.getChannels(context);
 			},
-			async getBundles(options) {
-				return plugin.getBundles(options);
+			async getBundles(options, context) {
+				return plugin.getBundles(options, context);
 			},
-			async insertBundle(bundle) {
-				await plugin.appendBundle(bundle);
-				await plugin.commitBundle();
+			async insertBundle(bundle, context) {
+				await plugin.appendBundle(bundle, context);
+				await plugin.commitBundle(context);
 			},
-			async updateBundleById(bundleId, newBundle) {
-				await plugin.updateBundle(bundleId, newBundle);
-				await plugin.commitBundle();
+			async updateBundleById(bundleId, newBundle, context) {
+				await plugin.updateBundle(bundleId, newBundle, context);
+				await plugin.commitBundle(context);
 			},
-			async deleteBundleById(bundleId) {
-				const bundle = await plugin.getBundleById(bundleId);
+			async deleteBundleById(bundleId, context) {
+				const bundle = await plugin.getBundleById(bundleId, context);
 				if (!bundle) return;
-				await plugin.deleteBundle(bundle);
-				await plugin.commitBundle();
+				await plugin.deleteBundle(bundle, context);
+				await plugin.commitBundle(context);
 			}
 		},
 		adapterName: plugin.name,
@@ -66,6 +172,5 @@ function createPluginDatabaseCore(plugin, resolveFileUrl) {
 		}
 	};
 }
-
 //#endregion
-exports.createPluginDatabaseCore = createPluginDatabaseCore;
+exports.createPluginDatabaseCore = createPluginDatabaseCore;