pi-antigravity-rotator 2.2.2 → 2.3.0

package/src/cli.ts

@@ -8,56 +8,72 @@
 
 import { getConfigDir } from "./paths.js";
 
-const args = process.argv.slice(2).filter((a) => !a.startsWith("--config-dir") && a !== process.argv[process.argv.indexOf("--config-dir") + 1]);
+const args = process.argv
+  .slice(2)
+  .filter(
+    (a) =>
+      !a.startsWith("--config-dir") &&
+      a !== process.argv[process.argv.indexOf("--config-dir") + 1],
+  );
 const command = args[0] || "start";
 
 console.log(`Config dir: ${getConfigDir()}`);
 console.log();
 
 switch (command) {
-	case "start": {
-		// Dynamic import to avoid loading everything for help
-		const { main } = await import("./index.js");
-		main();
-		break;
-	}
-	case "login": {
-		const { runLogin } = await import("./login.js");
-		runLogin();
-		break;
-	}
-	case "status": {
-		try {
-			const port = 51200;
-			const res = await fetch(`http://localhost:${port}/api/status`);
-			const data = await res.json();
-			console.log(JSON.stringify(data, null, 2));
-		} catch {
-			console.error("Rotator is not running or unreachable on port 51200");
-			process.exit(1);
-		}
-		break;
-	}
-	case "doctor": {
-		const { printDoctorReport, runDoctor } = await import("./doctor.js");
-		const result = runDoctor();
-		printDoctorReport(result);
-		process.exit(result.ok ? 0 : 1);
-		break;
-	}
-	default:
-		console.log("Pi Antigravity Rotator");
-		console.log();
-		console.log("Usage:");
-		console.log("  pi-antigravity-rotator start     Start the proxy (default)");
-		console.log("  pi-antigravity-rotator login     Add a new Google account");
-		console.log("  pi-antigravity-rotator status    Show account status (JSON)");
-		console.log("  pi-antigravity-rotator doctor    Validate config and local state");
-		console.log();
-		console.log("Options:");
-		console.log("  --config-dir <path>    Config directory (default: ~/.pi-antigravity-rotator/)");
-		console.log();
-		console.log("Environment:");
-		console.log("  PI_ROTATOR_DIR         Config directory override");
-		process.exit(command === "help" || command === "--help" ? 0 : 1);
+  case "start": {
+    // Dynamic import to avoid loading everything for help
+    const { main } = await import("./index.js");
+    await main();
+    break;
+  }
+  case "login": {
+    const { initDb } = await import("./db-store.js");
+    await initDb();
+    const { runLogin } = await import("./login.js");
+    await runLogin();
+    break;
+  }
+  case "status": {
+    try {
+      const port = 51200;
+      const res = await fetch(`http://localhost:${port}/api/status`);
+      const data = await res.json();
+      console.log(JSON.stringify(data, null, 2));
+    } catch {
+      console.error("Rotator is not running or unreachable on port 51200");
+      process.exit(1);
+    }
+    break;
+  }
+  case "doctor": {
+    const { initDb } = await import("./db-store.js");
+    await initDb();
+    const { printDoctorReport, runDoctor } = await import("./doctor.js");
+    const result = await runDoctor();
+    printDoctorReport(result);
+    process.exit(result.ok ? 0 : 1);
+    break;
+  }
+  default:
+    console.log("Pi Antigravity Rotator");
+    console.log();
+    console.log("Usage:");
+    console.log("  pi-antigravity-rotator start     Start the proxy (default)");
+    console.log("  pi-antigravity-rotator login     Add a new Google account");
+    console.log(
+      "  pi-antigravity-rotator status    Show account status (JSON)",
+    );
+    console.log(
+      "  pi-antigravity-rotator doctor    Validate config and local state",
+    );
+    console.log();
+    console.log("Options:");
+    console.log(
+      "  --config-dir <path>    Config directory (default: ~/.pi-antigravity-rotator/)",
+    );
+    console.log();
+    console.log("Environment:");
+    console.log("  PI_ROTATOR_DIR         Config directory override");
+    process.exit(command === "help" || command === "--help" ? 0 : 1);
 }

package/src/config-defaults.ts

@@ -0,0 +1,64 @@
+import type { Config } from "./types.js";
+
+export function applyConfigDefaults(config: Config): Config {
+	return {
+		proxyPort: config.proxyPort || 51200,
+		bindHost: config.bindHost || process.env.PI_ROTATOR_BIND_HOST || "0.0.0.0",
+		routingPolicy: config.routingPolicy || "timer-first",
+		requestsPerRotation: config.requestsPerRotation || 5,
+		rotateOnQuotaDrop: config.rotateOnQuotaDrop ?? 20,
+		quotaPollIntervalMs: config.quotaPollIntervalMs || 300000,
+		maxConcurrentRequestsPerAccount: config.maxConcurrentRequestsPerAccount ?? 1,
+		maxConcurrentRequestsPerProjectModel: config.maxConcurrentRequestsPerProjectModel ?? 1,
+		projectCircuitBreaker429Threshold: config.projectCircuitBreaker429Threshold ?? 3,
+		projectCircuitBreakerWindowMs: config.projectCircuitBreakerWindowMs ?? 10 * 60 * 1000,
+		projectCircuitBreakerCooldownMs: config.projectCircuitBreakerCooldownMs ?? 60 * 60 * 1000,
+		modelCircuitBreaker429Threshold: config.modelCircuitBreaker429Threshold ?? 3,
+		modelCircuitBreakerCooldownMs: config.modelCircuitBreakerCooldownMs ?? 6 * 60 * 60 * 1000,
+		dailyAccountSlowRequests: config.dailyAccountSlowRequests ?? 250,
+		dailyAccountStopRequests: config.dailyAccountStopRequests ?? 350,
+		dailyProjectSlowRequests: config.dailyProjectSlowRequests ?? 900,
+		dailyProjectStopRequests: config.dailyProjectStopRequests ?? 1200,
+		slowModeJitterMinMs: config.slowModeJitterMinMs ?? 8_000,
+		slowModeJitterMaxMs: config.slowModeJitterMaxMs ?? 25_000,
+		protectivePauseMs: config.protectivePauseMs ?? 21600000,
+		useRequestCountRotationWhenQuotaUnknownOnly: config.useRequestCountRotationWhenQuotaUnknownOnly ?? true,
+		tokenBucketEnabled: config.tokenBucketEnabled ?? false,
+		tokenBucketMaxTokens: config.tokenBucketMaxTokens ?? 50,
+		tokenBucketRefillPerMinute: config.tokenBucketRefillPerMinute ?? 6,
+		tokenBucketInitialTokens: config.tokenBucketInitialTokens ?? (config.tokenBucketMaxTokens ?? 50),
+		accounts: config.accounts ? config.accounts.map((account) => ({
+			...account,
+			tier: account.tier || "unknown",
+		})) : [],
+	};
+}
+
+export function getDefaultConfig(): Config {
+	return applyConfigDefaults({
+		proxyPort: 51200,
+		accounts: [],
+		requestsPerRotation: 5,
+		rotateOnQuotaDrop: 20,
+		quotaPollIntervalMs: 300000,
+		maxConcurrentRequestsPerAccount: 1,
+		maxConcurrentRequestsPerProjectModel: 1,
+		projectCircuitBreaker429Threshold: 3,
+		projectCircuitBreakerWindowMs: 10 * 60 * 1000,
+		projectCircuitBreakerCooldownMs: 60 * 60 * 1000,
+		modelCircuitBreaker429Threshold: 3,
+		modelCircuitBreakerCooldownMs: 6 * 60 * 60 * 1000,
+		dailyAccountSlowRequests: 250,
+		dailyAccountStopRequests: 350,
+		dailyProjectSlowRequests: 900,
+		dailyProjectStopRequests: 1200,
+		slowModeJitterMinMs: 8_000,
+		slowModeJitterMaxMs: 25_000,
+		protectivePauseMs: 21600000,
+		useRequestCountRotationWhenQuotaUnknownOnly: true,
+		tokenBucketEnabled: false,
+		tokenBucketMaxTokens: 50,
+		tokenBucketRefillPerMinute: 6,
+		tokenBucketInitialTokens: 50,
+	});
+}

package/src/config-storage.ts

@@ -0,0 +1,21 @@
+import type { Config } from "./types.js";
+import { getCachedConfig, setCachedConfig } from "./db-store.js";
+import { getDefaultConfig } from "./config-defaults.js";
+
+export function loadConfig(): Config {
+  const cached = getCachedConfig();
+  if (cached) return cached;
+  return getDefaultConfig();
+}
+
+export function loadOrCreateAccountsConfig(): Config {
+  try {
+    return loadConfig();
+  } catch {
+    return getDefaultConfig();
+  }
+}
+
+export function saveAccountsConfig(config: Config): void {
+  setCachedConfig(config);
+}