@uniqbit/mate-core 0.15.0-canary.3 → 0.15.0-canary.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uniqbit/mate-core",
-  "version": "0.15.0-canary.3",
+  "version": "0.15.0-canary.6",
   "license": "MIT",
   "files": [
     "src/",

package/src/cli/main.ts

@@ -1,5 +1,6 @@
 import { getActiveDistribution } from "../distribution";
 import {
+  enforceUpdateIfRequired,
   scheduleBackgroundCheck,
   showUpdateBannerIfAvailable,
   UpdateStateStore,
@@ -56,8 +57,15 @@ export async function main(argv = process.argv, deps: MainDeps = mainDeps): Prom
     command === "cap" && findPluginCliCommand(subcommand, rest[0]) !== undefined;
   if (!isPluginCommand) {
     const updateStore = new UpdateStateStore();
-    await showUpdateBannerIfAvailable(updateStore);
     scheduleBackgroundCheck(updateStore);
+    if (
+      !isInstallRecoveryCommand(command, subcommand) &&
+      (await enforceUpdateIfRequired(updateStore))
+    ) {
+      process.exitCode = 1;
+      return;
+    }
+    await showUpdateBannerIfAvailable(updateStore);
   }
 
   if (!command || command === "help" || command === "--help" || command === "-h") {

package/src/distribution.ts

@@ -1,10 +1,17 @@
 import type { PluginRegistry } from "./tools/setup/registry";
 
 export interface DistributionUpdateConfig {
-  /** npm package whose versions are checked and installed by `mate update`. */
-  packageName: string;
-  /** npm registry used for update checks and downloads. */
-  registry: string;
+  /** npm package whose versions are checked and installed by `mate update`. Defaults to `@uniqbit/<name>`. */
+  packageName?: string;
+  /** npm registry used for update checks and downloads. Defaults to public npm. */
+  registry?: string;
+  /**
+   * Refuse to run commands while a newer version is available, instead of
+   * only printing a banner. Install-recovery commands (`install`, `update`,
+   * `doctor`, `help`, `companion setup|link`) stay available so the user can
+   * actually upgrade.
+   */
+  enforce?: boolean;
 }
 
 /** Identity and version of the distribution assembled through `createMate`. */

package/src/lib/update-checker.ts

@@ -13,7 +13,7 @@ interface UpdateState {
   latestVersion: string | null;
 }
 
-const TWELVE_HOURS_MS = 12 * 60 * 60 * 1000;
+const SIX_HOURS_MS = 6 * 60 * 60 * 1000;
 
 export const updateCheckerDeps = {
   now: () => Date.now(),
@@ -34,11 +34,12 @@ export function getCurrentVersion(): string {
   return getActiveDistribution().config.version;
 }
 
-export function getUpdateConfig(): DistributionUpdateConfig {
+export function getUpdateConfig(): Required<DistributionUpdateConfig> {
   const { name, update } = getActiveDistribution().config;
   return {
     packageName: update?.packageName ?? `@uniqbit/${name}`,
     registry: update?.registry ?? PUBLIC_NPM_REGISTRY,
+    enforce: update?.enforce ?? false,
   };
 }
 
@@ -69,6 +70,28 @@ export async function showUpdateBannerIfAvailable(store: UpdateStateStore): Prom
   }
 }
 
+/**
+ * Distributions that set `update.enforce` refuse to run further commands while
+ * a cached newer version is available. Returns true when the current command
+ * must stop; state-load failures never block.
+ */
+export async function enforceUpdateIfRequired(store: UpdateStateStore): Promise<boolean> {
+  if (!getUpdateConfig().enforce) return false;
+  try {
+    const state = await store.load();
+    if (!state.latestVersion) return false;
+    const current = getCurrentVersion();
+    if (!isNewer(state.latestVersion, current)) return false;
+    process.stderr.write(
+      `\n${frameworkConfig.name}: update required (${current} → ${state.latestVersion})\n`,
+    );
+    process.stderr.write(`  Run \`${frameworkConfig.name} update\` before continuing.\n\n`);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
 export async function fetchLatestVersion(): Promise<string> {
   const { packageName, registry } = getUpdateConfig();
   return fetchPublicPackageVersion(packageName, registry);
@@ -79,7 +102,7 @@ export function scheduleBackgroundCheck(store: UpdateStateStore): void {
     try {
       const state = await store.load();
       const lastChecked = state.lastChecked ? new Date(state.lastChecked).getTime() : 0;
-      if (updateCheckerDeps.now() - lastChecked < TWELVE_HOURS_MS) return;
+      if (updateCheckerDeps.now() - lastChecked < SIX_HOURS_MS) return;
       const latestVersion = await fetchLatestVersion();
       await store.save({ lastChecked: updateCheckerDeps.toIsoString(), latestVersion });
     } catch {