@rehearsal-db/core 0.1.0-beta.6 → 0.1.0-beta.8

package/scripts/lib/runtime/postgresql_runtime.mjs

@@ -27,10 +27,14 @@ import {
   summarizeRestoreError,
 } from "../rehearsal/runtime_restore.mjs";
 import { readBoundRuntimeSanitizationPolicy } from "../rehearsal/sanitization_policy.mjs";
+import { ensureLocalContainerRuntime } from "../environment/local_supabase.mjs";
+import { parseRuntimeInvocation } from "./runtime_target.mjs";
 
 const repositoryRoot = process.cwd();
+const invocation = parseRuntimeInvocation();
 const loadedConfiguration = await loadRehearsalConfig({
   projectRoot: repositoryRoot,
+  configPath: invocation.configPath,
 });
 const { config, paths: configuredPaths } = loadedConfiguration;
 const artifactRoot = configuredPaths.artifactDirectory;
@@ -50,7 +54,7 @@ const image = config.postgresql.image;
 const database = config.postgresql.database;
 const databaseUser = config.postgresql.user;
 const databasePort = config.runtime.ports.database;
-const action = process.argv[2] ?? "status";
+const action = invocation.action;
 const applicationMigrations = new URL(
   "./",
   pathToFileURL(`${configuredPaths.migrationDirectory}/`),
@@ -105,11 +109,10 @@ const assertRuntimePath = () => {
 };
 
 const assertDocker = () => {
-  if (!commandSucceeds("docker", ["info"])) {
-    throw new Error(
-      "A running Docker-compatible runtime is required. Start Docker or Colima, then retry.",
-    );
-  }
+  ensureLocalContainerRuntime({
+    cwd: repositoryRoot,
+    autoStartColima: config.containerRuntime.autoStartColima,
+  });
 };
 
 const inspectResourceLabel = (kind, name) =>
@@ -650,9 +653,7 @@ const migrateRuntime = async () => {
   const { baseline } = await readActive();
   await startRuntime();
   const receipt = await readCandidateReceipt(baseline);
-  const confirmation = process.argv
-    .find((argument) => argument.startsWith("--confirm-candidates="))
-    ?.slice("--confirm-candidates=".length);
+  const confirmation = invocation.confirmation;
   if (confirmation !== receipt.candidateSha256) {
     throw new Error(
       "Candidate migration confirmation is missing or does not match the exact SHA-256.",

package/scripts/lib/runtime/runtime_target.mjs

@@ -17,6 +17,31 @@ const TARGETS = Object.freeze({
 
 export const REHEARSAL_RUNTIME_TARGETS = Object.freeze(Object.keys(TARGETS));
 
+export const parseRuntimeInvocation = (arguments_ = process.argv.slice(2)) => {
+  const action = arguments_.find((argument) => !argument.startsWith("--"));
+  const valueAfter = (prefix) =>
+    arguments_
+      .find((argument) => argument.startsWith(prefix))
+      ?.slice(prefix.length);
+  return Object.freeze({
+    action: action ?? "status",
+    configPath: valueAfter("--config="),
+    confirmation: valueAfter("--confirm-candidates="),
+  });
+};
+
+export const buildRuntimeManagerArguments = ({
+  managerPath,
+  action,
+  configPath,
+  confirmation,
+}) => [
+  managerPath,
+  action,
+  ...(configPath ? [`--config=${configPath}`] : []),
+  ...(confirmation ? [`--confirm-candidates=${confirmation}`] : []),
+];
+
 export const resolveRuntimeTarget = (value = "supabase") => {
   if (typeof value !== "string" || !Object.hasOwn(TARGETS, value)) {
     throw new Error(

package/scripts/lib/runtime/supabase_runtime.mjs

@@ -50,10 +50,13 @@ import {
   compareProductionSchemaDumps,
 } from "../rehearsal/schema_snapshot.mjs";
 import { formatCount } from "../rehearsal/human_output.mjs";
+import { parseRuntimeInvocation } from "./runtime_target.mjs";
 
 const repositoryRoot = process.cwd();
+const invocation = parseRuntimeInvocation();
 const loadedConfiguration = await loadRehearsalConfig({
   projectRoot: repositoryRoot,
+  configPath: invocation.configPath,
 });
 const { config, paths: configuredPaths } = loadedConfiguration;
 const artifactRoot = configuredPaths.artifactDirectory;
@@ -76,7 +79,7 @@ const runtimeAdapter = configuredPaths.runtimeAdapter
   ? await import(pathToFileURL(configuredPaths.runtimeAdapter).href)
   : null;
 const environmentKeyPattern = /^[A-Z][A-Z0-9_]*$/u;
-const action = process.argv[2] ?? "status";
+const action = invocation.action;
 
 const readServiceEnvironment = () =>
   readRehearsalServiceEnvironment({
@@ -468,6 +471,7 @@ const resetRuntime = async () => {
       workdir: runtimeWorkdir,
       exclude: ["edge-runtime", "logflare", "vector", "realtime"],
       environment: serviceEnvironment,
+      autoStartColima: config.containerRuntime.autoStartColima,
     });
     const structuralSchemaSha256 = await restoreProductionSchema(active);
     await restoreBaselineRows(active);
@@ -532,6 +536,7 @@ const startRuntime = async () => {
     workdir: runtimeWorkdir,
     exclude: ["edge-runtime", "logflare", "vector", "realtime"],
     environment: serviceEnvironment,
+    autoStartColima: config.containerRuntime.autoStartColima,
   });
   const projectRuntime = configureProjectRuntime({ environment, baseline });
   await writeRuntimeEnvironment({
@@ -660,9 +665,7 @@ const migrateRuntime = async () => {
   console.log(
     `${receipt.candidates.length === 1 ? "Candidate migration" : "Candidate migrations"} (${receipt.candidateSha256}): ${receipt.candidates.map((candidate) => candidate.filename).join(", ")}`,
   );
-  const confirmation = process.argv
-    .find((argument) => argument.startsWith("--confirm-candidates="))
-    ?.slice("--confirm-candidates=".length);
+  const confirmation = invocation.confirmation;
   if (confirmation !== receipt.candidateSha256) {
     throw new Error(
       "Candidate migration confirmation is missing or does not match the exact SHA-256.",

package/scripts/operations/database/manage_rehearsal_database.mjs

@@ -5,7 +5,14 @@
  */
 
 import { loadRehearsalConfig } from "../../lib/rehearsal/configuration.mjs";
-import { runRuntimeTarget } from "../../lib/runtime/runtime_target.mjs";
+import {
+  parseRuntimeInvocation,
+  runRuntimeTarget,
+} from "../../lib/runtime/runtime_target.mjs";
 
-const { config } = await loadRehearsalConfig({ projectRoot: process.cwd() });
+const invocation = parseRuntimeInvocation();
+const { config } = await loadRehearsalConfig({
+  projectRoot: process.cwd(),
+  configPath: invocation.configPath,
+});
 await runRuntimeTarget(config.runtime.target);

package/scripts/operations/rehearsal/rehearsal_cli.mjs

@@ -56,6 +56,11 @@ import {
   suggestPolicyExceptionColumns,
 } from "../../lib/rehearsal/policy_review.mjs";
 import { collectRehearsalSupportReport } from "../../lib/rehearsal/support_report.mjs";
+import {
+  applyRehearsalCleanup,
+  planRehearsalCleanup,
+} from "../../lib/rehearsal/cleanup.mjs";
+import { buildRuntimeManagerArguments } from "../../lib/runtime/runtime_target.mjs";
 
 const packageRoot = fileURLToPath(new URL("../../..", import.meta.url));
 const projectRoot = process.cwd();
@@ -81,6 +86,9 @@ const parseArguments = (arguments_) => {
     ledgerPath: undefined,
     assetsPath: undefined,
     target: undefined,
+    includeRuntime: false,
+    includeImages: false,
+    cleanupConfirmation: undefined,
   };
   const positionals = [];
   for (const argument of arguments_) {
@@ -95,6 +103,8 @@ const parseArguments = (arguments_) => {
       flags.configPath = argument.slice("--config=".length);
     } else if (argument.startsWith("--confirm-candidates=")) {
       flags.confirmation = argument.slice("--confirm-candidates=".length);
+    } else if (argument.startsWith("--confirm-cleanup=")) {
+      flags.cleanupConfirmation = argument.slice("--confirm-cleanup=".length);
     } else if (argument.startsWith("--records=")) {
       flags.recordsPath = argument.slice("--records=".length);
     } else if (argument.startsWith("--ledger=")) {
@@ -103,6 +113,10 @@ const parseArguments = (arguments_) => {
       flags.assetsPath = argument.slice("--assets=".length);
     } else if (argument.startsWith("--target=")) {
       flags.target = argument.slice("--target=".length);
+    } else if (argument === "--include-runtime") {
+      flags.includeRuntime = true;
+    } else if (argument === "--include-images") {
+      flags.includeImages = true;
     } else if (argument.startsWith("--")) {
       throw new Error(`Unknown Rehearsal option: ${argument}.`);
     } else positionals.push(argument);
@@ -131,7 +145,7 @@ const renderPlan = (plan, verbosity) => {
     "Environment",
     `  ✓ ${plan.environment.kind}`,
     `  ✓ Hosted access ${plan.environment.hostedAccess}`,
-    `  ✓ Application/provider-data egress ${plan.environment.outboundNetwork}`,
+    `  ✓ Hosted target configuration ${plan.environment.outboundNetwork}`,
     ...(plan.environment.authenticationProviders.length
       ? [
           `  ✓ External identity providers ${plan.environment.authenticationProviders.join(", ")}`,
@@ -834,6 +848,13 @@ const runGuidedHome = async ({ flags, planOptions }) => {
       command: "manage-runtime",
     });
   }
+  if (state.node.supported && state.config === "valid") {
+    options.push({
+      label: "Clean up disk space",
+      hint: "Preview old baselines, the runtime, and unused images",
+      command: "cleanup-guide",
+    });
+  }
   if (lastGuidedDetails) {
     options.push({
       label: "Show details from the last action",
@@ -982,6 +1003,60 @@ const runGuidedHome = async ({ flags, planOptions }) => {
     }
     return runtimeAction.command;
   }
+  if (selected.command === "cleanup-guide") {
+    const cleanupScope = await promptForChoice({
+      message: "What may Rehearsal include in the cleanup preview?",
+      flags,
+      options: [
+        {
+          label: "Old baselines only",
+          hint: "Keep the configured number of recent generations",
+          command: "baselines",
+        },
+        {
+          label: "Old baselines and this project's runtime",
+          hint: "Also remove the disposable database and its volumes",
+          command: "runtime",
+        },
+        {
+          label: "Old baselines and older unused Supabase images",
+          hint: "Images may be shared and can be downloaded again",
+          command: "images",
+        },
+        {
+          label: "Everything above",
+          command: "all",
+        },
+        { label: "Back", command: "exit" },
+      ],
+    });
+    if (!cleanupScope || cleanupScope.command === "exit") return "refresh";
+    flags.includeRuntime = ["runtime", "all"].includes(cleanupScope.command);
+    flags.includeImages = ["images", "all"].includes(cleanupScope.command);
+    flags.cleanupPlan = await planRehearsalCleanup({
+      ...planOptions,
+      includeRuntime: flags.includeRuntime,
+      includeImages: flags.includeImages,
+    });
+    console.log(
+      `\n${renderCleanup({ mode: "preview", plan: flags.cleanupPlan })}`,
+    );
+    const selectedCount =
+      flags.cleanupPlan.baselines.removed.length +
+      (flags.cleanupPlan.runtime.included && flags.cleanupPlan.runtime.detected
+        ? 1
+        : 0) +
+      flags.cleanupPlan.images.candidates.length;
+    if (selectedCount === 0) return "refresh";
+    const accepted = await promptForConfirmation(
+      "Remove exactly the resources shown above?",
+      flags,
+    );
+    if (!accepted) return "refresh";
+    flags.write = true;
+    flags.cleanupConfirmation = flags.cleanupPlan.digest;
+    return "cleanup";
+  }
   if (["baseline-guide", "baseline-prepare"].includes(selected.command)) {
     console.log(
       [
@@ -1209,10 +1284,12 @@ const emit = ({ command, data, flags, render, status = "success" }) => {
 };
 
 const runManager = ({ action, flags }) => {
-  const args = [managerPath, action];
-  if (flags.confirmation) {
-    args.push(`--confirm-candidates=${flags.confirmation}`);
-  }
+  const args = buildRuntimeManagerArguments({
+    managerPath,
+    action,
+    configPath: flags.configPath,
+    confirmation: flags.confirmation,
+  });
   const environment = Object.fromEntries(
     [
       "CI",
@@ -1641,6 +1718,112 @@ const renderSupportReport = (report) => {
   ].join("\n");
 };
 
+const formatBytes = (bytes) => {
+  if (!Number.isFinite(bytes) || bytes <= 0) return "0 B";
+  const units = ["B", "KiB", "MiB", "GiB", "TiB"];
+  const unit = Math.min(
+    Math.floor(Math.log(bytes) / Math.log(1024)),
+    units.length - 1,
+  );
+  const value = bytes / 1024 ** unit;
+  return `${value >= 10 || unit === 0 ? value.toFixed(0) : value.toFixed(1)} ${units[unit]}`;
+};
+
+const renderCleanup = (result) => {
+  const plan = result.plan;
+  const imageBytes = plan.images.candidates.reduce(
+    (total, image) => total + image.bytes,
+    0,
+  );
+  const dockerImages = plan.docker.resources.find(
+    (resource) => resource.Type === "Images",
+  );
+  const selectedCount =
+    plan.baselines.removed.length +
+    (plan.runtime.included && plan.runtime.detected ? 1 : 0) +
+    plan.images.candidates.length;
+  return [
+    `REHEARSAL CLEANUP — ${result.mode.toUpperCase()}`,
+    `Project: ${plan.project}`,
+    ...(dockerImages
+      ? [
+          `Docker images: ${dockerImages.Size}; ${dockerImages.Reclaimable} reclaimable across Docker`,
+        ]
+      : ["Docker usage: unavailable"]),
+    ...(plan.docker.disk
+      ? [
+          `Container disk: ${formatBytes(plan.docker.disk.usedBytes)} used of ${formatBytes(plan.docker.disk.totalBytes)}; ${formatBytes(plan.docker.disk.availableBytes)} free (${plan.docker.disk.usedPercent}% used)`,
+        ]
+      : []),
+    "",
+    "Project-owned files",
+    plan.baselines.available
+      ? `  ${plan.baselines.removed.length ? "→" : "✓"} ${formatCount(plan.baselines.removed.length, "old baseline generation")} selected; keeping ${plan.baselines.retained.length}`
+      : "  ○ No active baseline found",
+    "",
+    "Disposable runtime",
+    !plan.runtime.detected
+      ? "  ✓ No runtime found"
+      : plan.runtime.included
+        ? "  → Current project runtime selected"
+        : "  ○ Current project runtime preserved (add --include-runtime to select it)",
+    "",
+    "Shared Docker images",
+    !plan.images.requested
+      ? "  ○ Not inspected (add --include-images to inspect older unused Supabase images)"
+      : !plan.images.inspected
+        ? "  ! Docker is unavailable; images could not be inspected"
+        : plan.images.candidates.length
+          ? `  → ${formatCount(plan.images.candidates.length, "older unused Supabase image")} selected; approximately ${formatBytes(imageBytes)}`
+          : "  ✓ No older unused Supabase images found",
+    ...(plan.images.candidates.length
+      ? plan.images.candidates.flatMap((image) =>
+          image.tags.map((tag) => `    ${tag}`),
+        )
+      : []),
+    "",
+    ...(result.mode === "preview"
+      ? selectedCount
+        ? [
+            "Nothing has been removed.",
+            `Cleanup set: ${plan.digest.slice(0, 12)}`,
+            `Exact confirmation: --confirm-cleanup=${plan.digest}`,
+            "Review this list, then rerun the same command with --write and the exact confirmation above.",
+          ]
+        : [
+            "Nothing has been removed, and nothing is currently selected.",
+            "Use --include-runtime or --include-images only when you want those shared resources considered.",
+          ]
+      : [
+          `Removed ${formatCount(result.applied.baselines.removed.length, "old baseline generation")}.`,
+          result.applied.runtimeRemoved
+            ? "Removed this project's disposable runtime and volumes."
+            : "Preserved this project's disposable runtime.",
+          `Removed ${formatCount(result.applied.imagesRemoved.length, "unused image tag")}.`,
+        ]),
+  ].join("\n");
+};
+
+const runCleanup = async ({ flags, planOptions }) => {
+  const plan =
+    flags.cleanupPlan ??
+    (await planRehearsalCleanup({
+      ...planOptions,
+      includeRuntime: flags.includeRuntime,
+      includeImages: flags.includeImages,
+    }));
+  if (!flags.write) return { mode: "preview", plan };
+  const applied = await applyRehearsalCleanup({
+    plan,
+    confirmation: flags.cleanupConfirmation,
+    ...planOptions,
+    removeRuntime: async () => {
+      runManager({ action: "discard", flags });
+    },
+  });
+  return { mode: "written", plan, applied };
+};
+
 const usage = () => `Usage: rehearsal <command> [options]
 
 Commands:
@@ -1663,6 +1846,8 @@ Commands:
   status                     Report the disposable local runtime state
   stop                       Stop only this project's local runtime
   discard                    Remove only this project's disposable runtime
+  cleanup [--include-runtime] [--include-images] [--write] [--confirm-cleanup=]
+                             Preview or apply conservative disk cleanup
   verify                     Verify the current local Rehearsal runtime
 
 Options: --json --verbose --debug --plain --config=<path> --target=supabase|postgresql`;
@@ -1737,6 +1922,11 @@ const executeCommand = async ({ command, flags, planOptions, guided }) => {
     emit({ command, data, flags, render: renderSupportReport });
     return;
   }
+  if (command === "cleanup") {
+    const data = await runCleanup({ flags, planOptions });
+    emit({ command, data, flags, render: renderCleanup });
+    return;
+  }
   if (command === "explain" || (command === "run" && flags.dryRun)) {
     const data = await buildRehearsalPlan(planOptions);
     emit({ command, data, flags, render: renderPlan });
@@ -1872,8 +2062,12 @@ const resetGuidedFlags = (flags) => {
   flags.ledgerPath = undefined;
   flags.assetsPath = undefined;
   flags.target = undefined;
+  flags.includeRuntime = false;
+  flags.includeImages = false;
+  flags.cleanupConfirmation = undefined;
   flags.setupPlan = undefined;
   flags.preparationPlan = undefined;
+  flags.cleanupPlan = undefined;
 };
 
 const installGuidedExitShortcut = (flags) => {