openclaw/test/helpers/temp-dir.ts

// Test temp directory helper creates and cleans up temporary directories.
import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { afterEach } from "vitest";

// Synchronous temporary directory helpers for tests.

export type TempDirCollection = string[] | Set<string>;

export interface TestTempDirTracker {
  readonly dirs: ReadonlySet<string>;
  make(prefix: string): string;
  cleanup(): void;
}

export interface AutoCleanupTempDirTracker {
  readonly dirs: ReadonlySet<string>;
  make(prefix: string): string;
}

/** Create a temp dir and register it in an array or set for cleanup. */
export function makeTempDir(tempDirs: TempDirCollection, prefix: string): string {
  const dir = fs.mkdtempSync(path.join(os.tmpdir(), prefix));
  if (Array.isArray(tempDirs)) {
    tempDirs.push(dir);
  } else {
    tempDirs.add(dir);
  }
  return dir;
}

/** Remove all tracked temporary directories and clear the tracker. */
export function cleanupTempDirs(tempDirs: TempDirCollection): void {
  const dirs = Array.isArray(tempDirs) ? tempDirs.splice(0) : [...tempDirs];
  for (const dir of dirs) {
    fs.rmSync(dir, { recursive: true, force: true, maxRetries: 5, retryDelay: 20 });
  }
  if (!Array.isArray(tempDirs)) {
    tempDirs.clear();
  }
}

export function createTempDirTracker(): TestTempDirTracker {
  const dirs = new Set<string>();
  return {
    dirs,
    make(prefix: string): string {
      return makeTempDir(dirs, prefix);
    },
    cleanup(): void {
      cleanupTempDirs(dirs);
    },
  };
}

/** Create a temp dir tracker that Vitest cleans up after each test. */
export function useAutoCleanupTempDirTracker(): AutoCleanupTempDirTracker {
  const tracker = createTempDirTracker();
  afterEach(() => {
    tracker.cleanup();
  });
  return {
    dirs: tracker.dirs,
    make(prefix: string): string {
      return tracker.make(prefix);
    },
  };
}