pi-antigravity-rotator 2.2.2 → 2.3.0

package/src/db-store.ts

@@ -0,0 +1,155 @@
+// Convenience façade over the settings repository.
+//
+// Maintains full backward compatibility with the previous module API while
+// delegating all persistence to an ISettingsRepository implementation.
+// Both PostgresSettingsRepository and FileSettingsRepository are fully
+// self-contained — callers no longer need to branch on isDbConfigured().
+
+import type { Config, PersistedState, TokenUsageTiered } from "./types.js";
+import type { PersistedResponsesStore } from "./responses-store.js";
+import { validateConfig } from "./validators.js";
+import { applyConfigDefaults } from "./config-defaults.js";
+import {
+  type ISettingsRepository,
+  PostgresSettingsRepository,
+  FileSettingsRepository,
+} from "./settings-repository.js";
+
+export type { PersistedResponsesStore } from "./responses-store.js";
+
+// ----- Repository strategy -----
+
+function createRepository(): ISettingsRepository {
+  if (process.env.PI_ROTATOR_DATABASE_URL || process.env.DATABASE_URL) {
+    return new PostgresSettingsRepository();
+  }
+  return new FileSettingsRepository();
+}
+
+const repository: ISettingsRepository = createRepository();
+
+let initialized = false;
+
+function assertInitialized(): void {
+  if (!initialized) {
+    throw new Error(
+      "db-store: repository not initialized. Call initDb() before accessing settings.",
+    );
+  }
+}
+
+// ----- Backward-compatible public API -----
+
+/**
+ * Whether persistence is backed by PostgreSQL (true) or disk files (false).
+ * Callers that need to know the storage backend (e.g. for diagnostics or
+ * doctor output) can use this, but most callers should NOT need to branch
+ * on this — both repositories handle their own I/O.
+ */
+export function isDbConfigured(): boolean {
+  return !!(process.env.PI_ROTATOR_DATABASE_URL || process.env.DATABASE_URL);
+}
+
+export async function initDb(): Promise<void> {
+  await repository.init();
+  initialized = true;
+}
+
+export async function closeDb(): Promise<void> {
+  await repository.close();
+  initialized = false;
+}
+
+// --- Accounts config ---
+
+export function getCachedConfig(): Config | null {
+  assertInitialized();
+  const raw = repository.get("accounts_json");
+  if (!raw) return null;
+  try {
+    const parsed = JSON.parse(raw);
+    const validation = validateConfig(parsed);
+    if (validation.ok && validation.value) {
+      return applyConfigDefaults(validation.value);
+    }
+  } catch (err) {
+    console.error(`Failed to parse accounts config from repository: ${err}`);
+  }
+  return null;
+}
+
+export function setCachedConfig(config: Config): void {
+  assertInitialized();
+  const withDefaults = applyConfigDefaults(config);
+  repository.set("accounts_json", JSON.stringify(withDefaults, null, 2));
+}
+
+// --- Admin token ---
+
+export function getCachedAdminToken(): string | null {
+  assertInitialized();
+  const raw = repository.get("admin_token");
+  return raw ? raw.trim() : null;
+}
+
+export function setCachedAdminToken(token: string): void {
+  assertInitialized();
+  repository.set("admin_token", token.trim());
+}
+
+// --- Rotator state ---
+
+export function getCachedState(): PersistedState | null {
+  assertInitialized();
+  const raw = repository.get("rotator_state");
+  if (!raw) return null;
+  try {
+    return JSON.parse(raw) as PersistedState;
+  } catch (err) {
+    console.error(`Failed to parse rotator state from repository: ${err}`);
+    return null;
+  }
+}
+
+export function setCachedState(state: PersistedState): void {
+  assertInitialized();
+  repository.set("rotator_state", JSON.stringify(state));
+}
+
+// --- Token usage ---
+
+export function getCachedTokenUsage(): TokenUsageTiered | null {
+  assertInitialized();
+  const raw = repository.get("token_usage");
+  if (!raw) return null;
+  try {
+    return JSON.parse(raw) as TokenUsageTiered;
+  } catch (err) {
+    console.error(`Failed to parse token usage from repository: ${err}`);
+    return null;
+  }
+}
+
+export function setCachedTokenUsage(usage: TokenUsageTiered): void {
+  assertInitialized();
+  repository.set("token_usage", JSON.stringify(usage));
+}
+
+// --- Responses store ---
+
+export function getCachedResponsesStore(): PersistedResponsesStore | null {
+  assertInitialized();
+  const raw = repository.get("responses_store");
+  if (!raw) return null;
+  try {
+    return JSON.parse(raw) as PersistedResponsesStore;
+  } catch (err) {
+    console.error(`Failed to parse responses store from repository: ${err}`);
+    return null;
+  }
+}
+
+export function setCachedResponsesStore(store: PersistedResponsesStore): void {
+  assertInitialized();
+  repository.set("responses_store", JSON.stringify(store));
+}

package/src/doctor.ts

@@ -1,97 +1,115 @@
 import { accessSync, constants, existsSync } from "node:fs";
-import { join } from "node:path";
 import { getConfiguredAdminToken } from "./admin-auth.js";
-import { getAccountsPath, getConfigDir, getStatePath } from "./paths.js";
-import { getTokenUsagePath, loadConfigFromDisk } from "./account-store.js";
-import { listBackups, readJsonFile } from "./storage.js";
-import type { PersistedState, TokenUsageTiered } from "./types.js";
+import { getConfigDir } from "./paths.js";
+import { listBackups } from "./storage.js";
+import {
+  isDbConfigured,
+  getCachedConfig,
+  getCachedState,
+  getCachedTokenUsage,
+} from "./db-store.js";
 
 function checkWritable(path: string): boolean {
-	try {
-		accessSync(path, constants.R_OK | constants.W_OK);
-		return true;
-	} catch {
-		return false;
-	}
+  try {
+    accessSync(path, constants.R_OK | constants.W_OK);
+    return true;
+  } catch {
+    return false;
+  }
 }
 
 export interface DoctorResult {
-	ok: boolean;
-	warnings: string[];
-	errors: string[];
-	info: Record<string, unknown>;
+  ok: boolean;
+  warnings: string[];
+  errors: string[];
+  info: Record<string, unknown>;
 }
 
-export function runDoctor(env: NodeJS.ProcessEnv = process.env): DoctorResult {
-	const warnings: string[] = [];
-	const errors: string[] = [];
-	const configDir = getConfigDir();
-	const accountsPath = getAccountsPath();
-	const statePath = getStatePath();
-	const tokenUsagePath = getTokenUsagePath();
+export async function runDoctor(
+  env: NodeJS.ProcessEnv = process.env,
+): Promise<DoctorResult> {
+  const warnings: string[] = [];
+  const errors: string[] = [];
+  const configDir = getConfigDir();
 
-	if (!getConfiguredAdminToken(env)) {
-		warnings.push("PI_ROTATOR_ADMIN_TOKEN is not configured; dashboard and /api/* remain open on the bound interface.");
-	}
+  if (!getConfiguredAdminToken(env)) {
+    warnings.push(
+      "PI_ROTATOR_ADMIN_TOKEN is not configured; dashboard and /api/* remain open on the bound interface.",
+    );
+  }
 
-	if (!existsSync(configDir)) {
-		errors.push(`Config directory missing: ${configDir}`);
-	}
-	if (!checkWritable(configDir)) {
-		errors.push(`Config directory is not writable: ${configDir}`);
-	}
+  const dbConfigured = isDbConfigured();
 
-	try {
-		loadConfigFromDisk();
-	} catch (err) {
-		errors.push(`Config validation failed: ${err instanceof Error ? err.message : String(err)}`);
-	}
+  if (dbConfigured) {
+    // When using PostgreSQL, configDir is not required for data persistence
+    if (!existsSync(configDir)) {
+      warnings.push(
+        `Config directory does not exist (${configDir}), but storage backend is PostgreSQL — this is expected in containerised environments.`,
+      );
+    }
+  } else {
+    if (!existsSync(configDir)) {
+      errors.push(`Config directory missing: ${configDir}`);
+    } else if (!checkWritable(configDir)) {
+      errors.push(`Config directory is not writable: ${configDir}`);
+    }
+  }
 
-	try {
-		if (existsSync(statePath)) readJsonFile<PersistedState>(statePath);
-	} catch (err) {
-		errors.push(`State file is corrupted: ${err instanceof Error ? err.message : String(err)}`);
-	}
+  // Validate config — read from repository (handles both DB and file)
+  const cfg = getCachedConfig();
+  if (!cfg) {
+    warnings.push(
+      "No accounts config found. Run 'pi-antigravity-rotator login' to add your first account.",
+    );
+  }
 
-	try {
-		if (existsSync(tokenUsagePath)) readJsonFile<TokenUsageTiered>(tokenUsagePath);
-	} catch (err) {
-		errors.push(`Token usage file is corrupted: ${err instanceof Error ? err.message : String(err)}`);
-	}
+  // Validate state
+  const state = getCachedState();
+  if (state && typeof state !== "object") {
+    warnings.push("Rotator state data is corrupted.");
+  }
 
-	return {
-		ok: errors.length === 0,
-		warnings,
-		errors,
-		info: {
-			configDir,
-			accountsPath,
-			statePath,
-			tokenUsagePath,
-			backupCount: listBackups().length,
-			firstBackup: listBackups()[0] ?? null,
-			adminTokenConfigured: !!getConfiguredAdminToken(env),
-			bindHost: env.PI_ROTATOR_BIND_HOST ?? null,
-		},
-	};
+  // Validate token usage
+  const usage = getCachedTokenUsage();
+  if (usage && typeof usage !== "object") {
+    warnings.push("Token usage data is corrupted.");
+  }
+
+  const backups = dbConfigured ? [] : listBackups();
+
+  return {
+    ok: errors.length === 0,
+    warnings,
+    errors,
+    info: {
+      configDir,
+      backupCount: backups.length,
+      firstBackup: backups[0] ?? null,
+      adminTokenConfigured: !!getConfiguredAdminToken(env),
+      bindHost: env.PI_ROTATOR_BIND_HOST ?? null,
+      storageBackend: dbConfigured ? "postgresql" : "file",
+    },
+  };
 }
 
 export function printDoctorReport(result: DoctorResult): void {
-	console.log("Pi Antigravity Rotator Doctor");
-	console.log();
-	console.log(`Status: ${result.ok ? "OK" : "ERROR"}`);
-	console.log();
-	for (const [key, value] of Object.entries(result.info)) {
-		console.log(`${key}: ${typeof value === "string" || typeof value === "number" || typeof value === "boolean" || value === null ? value : JSON.stringify(value)}`);
-	}
-	if (result.warnings.length > 0) {
-		console.log();
-		console.log("Warnings:");
-		for (const warning of result.warnings) console.log(`- ${warning}`);
-	}
-	if (result.errors.length > 0) {
-		console.log();
-		console.log("Errors:");
-		for (const error of result.errors) console.log(`- ${error}`);
-	}
+  console.log("Pi Antigravity Rotator Doctor");
+  console.log();
+  console.log(`Status: ${result.ok ? "OK" : "ERROR"}`);
+  console.log();
+  for (const [key, value] of Object.entries(result.info)) {
+    console.log(
+      `${key}: ${typeof value === "string" || typeof value === "number" || typeof value === "boolean" || value === null ? value : JSON.stringify(value)}`,
+    );
+  }
+  if (result.warnings.length > 0) {
+    console.log();
+    console.log("Warnings:");
+    for (const warning of result.warnings) console.log(`- ${warning}`);
+  }
+  if (result.errors.length > 0) {
+    console.log();
+    console.log("Errors:");
+    for (const error of result.errors) console.log(`- ${error}`);
+  }
 }