@cueloop/daemon 0.1.0-alpha.9 → 0.1.0-alpha.90

package/src/retention.test.ts

@@ -0,0 +1,191 @@
+/** Retention: the cleanup period defaults to 30, honors [cleanup] period_days, and prune drops only the sessions and reports past the window. */
+
+import { describe, expect, test } from "bun:test";
+import { mkdirSync, mkdtempSync, utimesSync, writeFileSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import type { Thread, SessionStatus } from "@cueloop/schema";
+import { ThreadStore } from "./store";
+import { reportsDir } from "./paths";
+import {
+  DEFAULT_CLEANUP_PERIOD_DAYS,
+  LATEST_REPORT_FILENAME,
+  isExpired,
+  parseRefineState,
+  pruneExpiredReports,
+  pruneExpiredSessions,
+  resolveCleanupPeriodDays,
+} from "./retention";
+
+const DAY_MS = 24 * 60 * 60 * 1000;
+const NOW_MS = Date.parse("2026-08-26T00:00:00Z");
+
+function tempDir(prefix: string): string {
+  return mkdtempSync(join(tmpdir(), prefix));
+}
+
+function daysAgo(days: number): string {
+  return new Date(NOW_MS - days * DAY_MS).toISOString();
+}
+
+function session(id: string, createdAt: string, status: SessionStatus = "resolved"): Thread {
+  return {
+    schemaVersion: "1",
+    id,
+    workspace: { repoRoot: "/repo", branch: "main" },
+    artifact: { type: "plan", content: "", meta: {} },
+    revisions: [],
+    annotations: [],
+    message: null,
+    status,
+    createdAt,
+  };
+}
+
+describe("resolveCleanupPeriodDays", () => {
+  test("defaults to 30 when no config file is present", () => {
+    // Arrange
+    const missing = join(tempDir("cueloop-retention-"), "config.toml");
+
+    // Act
+    const days = resolveCleanupPeriodDays({ CUELOOP_CONFIG: missing });
+
+    // Assert
+    expect(days).toBe(DEFAULT_CLEANUP_PERIOD_DAYS);
+  });
+
+  test("honors [cleanup] period_days", () => {
+    // Arrange
+    const path = join(tempDir("cueloop-retention-"), "config.toml");
+
+    writeFileSync(path, "[cleanup]\nperiod_days = 7\n");
+
+    // Act
+    const days = resolveCleanupPeriodDays({ CUELOOP_CONFIG: path });
+
+    // Assert
+    expect(days).toBe(7);
+  });
+
+  test("falls back to the default when the value is not a number", () => {
+    // Arrange
+    const path = join(tempDir("cueloop-retention-"), "config.toml");
+
+    writeFileSync(path, '[cleanup]\nperiod_days = "soon"\n');
+
+    // Act
+    const days = resolveCleanupPeriodDays({ CUELOOP_CONFIG: path });
+
+    // Assert
+    expect(days).toBe(DEFAULT_CLEANUP_PERIOD_DAYS);
+  });
+});
+
+describe("parseRefineState", () => {
+  test("reads the analyzed fingerprint map", () => {
+    // Assert
+    expect([...parseRefineState({ analyzed: { ses_a: "1:2:x" } })]).toEqual([["ses_a", "1:2:x"]]);
+  });
+
+  test("returns an empty map for the legacy or malformed shape", () => {
+    // Assert
+    expect(parseRefineState({ seenSessionIds: ["ses_a"] }).size).toBe(0);
+    expect(parseRefineState({ analyzed: { ses_a: 5 } }).size).toBe(0);
+    expect(parseRefineState(null).size).toBe(0);
+  });
+});
+
+describe("isExpired", () => {
+  test("is true past the window and false inside it", () => {
+    // Assert
+    expect(isExpired(daysAgo(40), 30, NOW_MS)).toBe(true);
+    expect(isExpired(daysAgo(10), 30, NOW_MS)).toBe(false);
+  });
+
+  test("a period of zero disables expiry", () => {
+    // Assert
+    expect(isExpired(daysAgo(999), 0, NOW_MS)).toBe(false);
+  });
+});
+
+describe("pruneExpiredSessions", () => {
+  test("deletes resolved sessions past the window and keeps recent ones", () => {
+    // Arrange
+    const home = tempDir("cueloop-retention-home-");
+    const store = new ThreadStore(home);
+
+    store.upsert(session("ses_old", daysAgo(40)));
+    store.upsert(session("ses_new", daysAgo(2)));
+
+    // Act
+    const pruned = pruneExpiredSessions(store, 30, NOW_MS);
+
+    // Assert
+    expect(pruned).toEqual(["ses_old"]);
+    expect(store.get("ses_old")).toBeUndefined();
+    expect(store.get("ses_new")).toBeDefined();
+  });
+
+  test("keeps an expired resolved Thread while its Message is awaiting delivery", () => {
+    const store = new ThreadStore(tempDir("cueloop-retention-home-"));
+
+    store.upsert(session("ses_waiting", daysAgo(40)));
+
+    expect(pruneExpiredSessions(store, 30, NOW_MS, new Set(["ses_waiting"]))).toEqual([]);
+    expect(store.get("ses_waiting")).toBeDefined();
+  });
+
+  test("never deletes an active pending session however old", () => {
+    // Arrange
+    const home = tempDir("cueloop-retention-home-");
+    const store = new ThreadStore(home);
+
+    store.upsert(session("ses_active", daysAgo(400), "pending"));
+
+    // Act
+    const pruned = pruneExpiredSessions(store, 30, NOW_MS);
+
+    // Assert
+    expect(pruned).toEqual([]);
+    expect(store.get("ses_active")).toBeDefined();
+  });
+
+  test("a period of zero prunes nothing", () => {
+    // Arrange
+    const home = tempDir("cueloop-retention-home-");
+    const store = new ThreadStore(home);
+
+    store.upsert(session("ses_old", daysAgo(999)));
+
+    // Act
+    const pruned = pruneExpiredSessions(store, 0, NOW_MS);
+
+    // Assert
+    expect(pruned).toEqual([]);
+    expect(store.get("ses_old")).toBeDefined();
+  });
+});
+
+describe("pruneExpiredReports", () => {
+  test("removes timestamped reports past the window but keeps the latest report", () => {
+    // Arrange
+    const home = tempDir("cueloop-retention-home-");
+    const directory = reportsDir(home);
+
+    mkdirSync(directory, { recursive: true });
+    const latest = join(directory, LATEST_REPORT_FILENAME);
+    const stale = join(directory, "refine-2026-06-01T00-00-00-000Z.md");
+    const fresh = join(directory, "refine-2026-08-25T00-00-00-000Z.md");
+
+    for (const path of [latest, stale, fresh]) writeFileSync(path, "report");
+    const staleSeconds = (NOW_MS - 60 * DAY_MS) / 1000;
+
+    utimesSync(stale, staleSeconds, staleSeconds);
+
+    // Act
+    const pruned = pruneExpiredReports(directory, 30, NOW_MS);
+
+    // Assert
+    expect(pruned).toEqual(["refine-2026-06-01T00-00-00-000Z.md"]);
+  });
+});

package/src/retention.ts

@@ -0,0 +1,123 @@
+import { readdirSync, readFileSync, rmSync, statSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import * as v from "valibot";
+import type { ThreadStore } from "./store";
+
+export const DEFAULT_CLEANUP_PERIOD_DAYS = 30;
+export const LATEST_REPORT_FILENAME = "report.md";
+export const TIMESTAMPED_REPORT_PREFIX = "refine-";
+
+const MILLISECONDS_PER_DAY = 24 * 60 * 60 * 1000;
+
+const CleanupConfigSchema = v.object({
+  cleanup: v.optional(v.object({ period_days: v.optional(v.number()) })),
+});
+
+const RefineStateSchema = v.object({
+  analyzed: v.optional(v.record(v.string(), v.string()), {}),
+});
+
+export type RefineState = v.InferOutput<typeof RefineStateSchema>;
+
+export function parseRefineState(raw: Parameters<typeof v.safeParse>[1]): Map<string, string> {
+  const parsed = v.safeParse(RefineStateSchema, raw);
+
+  if (!parsed.success) return new Map();
+
+  return new Map(Object.entries(parsed.output.analyzed));
+}
+
+function userConfigPath(env: NodeJS.ProcessEnv): string {
+  const explicit = env.CUELOOP_CONFIG;
+
+  if (explicit && explicit.trim()) return explicit;
+  const xdgConfigHome = env.XDG_CONFIG_HOME;
+  const base = xdgConfigHome && xdgConfigHome.trim() ? xdgConfigHome : join(homedir(), ".config");
+
+  return join(base, "cueloop", "config.toml");
+}
+
+function readConfiguredPeriodDays(path: string): number | undefined {
+  let raw: unknown;
+
+  try {
+    raw = Bun.TOML.parse(readFileSync(path, "utf8"));
+  } catch {
+    return undefined;
+  }
+  const parsed = v.safeParse(CleanupConfigSchema, raw);
+
+  if (!parsed.success) return undefined;
+  const value = parsed.output.cleanup?.period_days;
+
+  return value !== undefined && Number.isFinite(value) && value >= 0 ? value : undefined;
+}
+
+export function resolveCleanupPeriodDays(env: NodeJS.ProcessEnv = process.env): number {
+  return readConfiguredPeriodDays(userConfigPath(env)) ?? DEFAULT_CLEANUP_PERIOD_DAYS;
+}
+
+export function isExpired(createdAt: string, periodDays: number, nowMs: number): boolean {
+  if (periodDays <= 0) return false;
+  const createdMs = Date.parse(createdAt);
+
+  if (Number.isNaN(createdMs)) return false;
+
+  return nowMs - createdMs > periodDays * MILLISECONDS_PER_DAY;
+}
+
+export function pruneExpiredSessions(
+  store: ThreadStore,
+  periodDays: number,
+  nowMs: number,
+  protectedThreadIds: ReadonlySet<string> = new Set(),
+): string[] {
+  if (periodDays <= 0) return [];
+  const pruned: string[] = [];
+
+  for (const session of store.list()) {
+    if (session.status !== "resolved" || protectedThreadIds.has(session.id)) continue;
+    if (isExpired(session.createdAt, periodDays, nowMs) && store.delete(session.id)) {
+      pruned.push(session.id);
+    }
+  }
+
+  return pruned;
+}
+
+export function timestampedReportFilename(nowMs: number): string {
+  const stamp = new Date(nowMs).toISOString().replace(/[:.]/g, "-");
+
+  return `${TIMESTAMPED_REPORT_PREFIX}${stamp}.md`;
+}
+
+export function pruneExpiredReports(
+  reportsDirectory: string,
+  periodDays: number,
+  nowMs: number,
+): string[] {
+  if (periodDays <= 0) return [];
+  let entries: string[];
+
+  try {
+    entries = readdirSync(reportsDirectory);
+  } catch {
+    return [];
+  }
+  const cutoffMs = nowMs - periodDays * MILLISECONDS_PER_DAY;
+  const pruned: string[] = [];
+
+  for (const entry of entries) {
+    if (entry === LATEST_REPORT_FILENAME) continue;
+    if (!entry.startsWith(TIMESTAMPED_REPORT_PREFIX)) continue;
+    const path = join(reportsDirectory, entry);
+
+    if (statSync(path).mtimeMs < cutoffMs) {
+      rmSync(path, { force: true });
+      pruned.push(entry);
+    }
+  }
+
+  return pruned;
+}