openclaw/scripts/lib/sqlite-reliability-restore-worker.ts

import fs from "node:fs/promises";
import path from "node:path";
import { pathToFileURL } from "node:url";

type RestoreCrashPoint = "after-publish" | "before-publish";

const SLEEP_BUFFER = new Int32Array(new SharedArrayBuffer(4));

function parseCrashPoint(value: string | undefined): RestoreCrashPoint {
  if (value === "after-publish" || value === "before-publish") {
    return value;
  }
  throw new Error(`invalid SQLite restore crash point: ${String(value)}`);
}

function holdAtCrashPoint(crashPoint: RestoreCrashPoint): never {
  process.send?.({ crashPoint, kind: "crash-point" });
  while (true) {
    Atomics.wait(SLEEP_BUFFER, 0, 0, 60_000);
  }
}

async function main(argv: string[]): Promise<void> {
  const [crashPointValue, repositoryPath, validationRootPath, snapshotPath, targetPath, ...extra] =
    argv;
  if (extra.length > 0 || !repositoryPath || !validationRootPath || !snapshotPath || !targetPath) {
    throw new Error("invalid SQLite restore interruption worker arguments");
  }
  const crashPoint = parseCrashPoint(crashPointValue);
  const resolvedTargetPath = path.resolve(targetPath);

  if (crashPoint === "before-publish") {
    const originalLink = fs.link.bind(fs);
    Object.defineProperty(fs, "link", {
      configurable: true,
      enumerable: true,
      value: async (sourcePath: string, publishedPath: string) => {
        const resolvedSourcePath = path.resolve(sourcePath);
        if (
          path.resolve(publishedPath) === resolvedTargetPath &&
          path.basename(resolvedSourcePath) === "database.sqlite" &&
          path.basename(path.dirname(resolvedSourcePath)).startsWith(".sqlite-publish-")
        ) {
          holdAtCrashPoint(crashPoint);
        }
        await originalLink(sourcePath, publishedPath);
      },
      writable: true,
    });
  } else {
    const originalRmdir = fs.rmdir.bind(fs);
    const restoreParentPath = path.resolve(path.dirname(targetPath));
    Object.defineProperty(fs, "rmdir", {
      configurable: true,
      enumerable: true,
      value: async (directoryPath: string) => {
        const resolvedDirectoryPath = path.resolve(directoryPath);
        if (
          path.dirname(resolvedDirectoryPath) === restoreParentPath &&
          path.basename(resolvedDirectoryPath).startsWith(".tmp-restore-")
        ) {
          holdAtCrashPoint(crashPoint);
        }
        await originalRmdir(directoryPath);
      },
      writable: true,
    });
  }

  const { createLocalSqliteSnapshotProvider } =
    await import("../../src/snapshot/local-repository.js");
  const provider = createLocalSqliteSnapshotProvider({
    repositoryPath,
    validationRootPath,
  });
  process.send?.({ kind: "ready" });
  await provider.restoreFresh({ path: snapshotPath }, targetPath);
  throw new Error(`SQLite restore worker passed ${crashPoint} without being terminated.`);
}

if (import.meta.url === pathToFileURL(process.argv[1] ?? "").href) {
  await main(process.argv.slice(2));
}