@albinocrabs/o-switcher 0.3.0 → 0.4.0

package/dist/chunk-QKEHCM2F.cjs

@@ -0,0 +1,234 @@
+'use strict';
+
+var chunkBHHR2U5B_cjs = require('./chunk-BHHR2U5B.cjs');
+var promises = require('fs/promises');
+var path = require('path');
+var os = require('os');
+var http = require('http');
+
+var STATE_DIR = path.join(os.homedir(), ".local", "share", "o-switcher");
+var STATE_FILE = path.join(STATE_DIR, "tui-state.json");
+var STATE_TMP = path.join(STATE_DIR, "tui-state.json.tmp");
+var writeStateAtomic = async (state) => {
+  await promises.mkdir(path.dirname(STATE_FILE), { recursive: true });
+  const json = JSON.stringify(state);
+  await promises.writeFile(STATE_TMP, json, "utf8");
+  await promises.rename(STATE_TMP, STATE_FILE);
+};
+var createStateWriter = (debounceMs = 500) => {
+  let pending;
+  let timer;
+  let writePromise = Promise.resolve();
+  const doWrite = () => {
+    if (!pending) return;
+    const snapshot = pending;
+    pending = void 0;
+    writePromise = writeStateAtomic(snapshot).catch(() => void 0);
+  };
+  const write = (state) => {
+    pending = state;
+    if (timer) clearTimeout(timer);
+    timer = setTimeout(doWrite, debounceMs);
+  };
+  const flush = async () => {
+    if (timer) {
+      clearTimeout(timer);
+      timer = void 0;
+    }
+    doWrite();
+    await writePromise;
+  };
+  return { write, flush };
+};
+var DEFAULT_CONFIG = {
+  port: 4444,
+  upstream: "https://api.openai.com",
+  maxRetries: 3,
+  provider: "openai"
+};
+var buildProfilePool = (profiles) => profiles.sort((a, b) => a.consecutiveFailures - b.consecutiveFailures);
+var getBearerToken = (profile) => {
+  const creds = profile.entry.credentials;
+  if (creds.type === "api-key") {
+    return creds.key;
+  }
+  if (creds.type === "oauth") {
+    return creds.access;
+  }
+  return void 0;
+};
+var readBody = (req) => new Promise((resolve, reject) => {
+  const chunks = [];
+  req.on("data", (chunk) => chunks.push(chunk));
+  req.on("end", () => resolve(Buffer.concat(chunks)));
+  req.on("error", reject);
+});
+var forwardRequest = async (method, path, headers, body, upstream) => {
+  const url = `${upstream}${path}`;
+  const reqHeaders = { ...headers };
+  delete reqHeaders["host"];
+  const response = await fetch(url, {
+    method,
+    headers: reqHeaders,
+    body: method !== "GET" && method !== "HEAD" ? body : void 0
+  });
+  const respHeaders = {};
+  response.headers.forEach((value, key) => {
+    respHeaders[key] = value;
+  });
+  if (respHeaders["content-type"]?.includes("text/event-stream") && response.body) {
+    return { status: response.status, headers: respHeaders, body: response.body };
+  }
+  const respBody = Buffer.from(await response.arrayBuffer());
+  return { status: response.status, headers: respHeaders, body: respBody };
+};
+var publishState = (stateWriter, profiles, activeId) => {
+  const targets = profiles.map((p) => ({
+    target_id: p.entry.id,
+    provider_id: p.entry.provider,
+    profile: p.entry.id,
+    state: p.consecutiveFailures > 0 ? "CoolingDown" : "Active",
+    health_score: Math.max(0, 1 - p.consecutiveFailures * 0.25),
+    latency_ema_ms: 0,
+    enabled: true
+  }));
+  const snapshot = {
+    version: 1,
+    updated_at: Date.now(),
+    active_target_id: activeId,
+    targets
+  };
+  stateWriter.write(snapshot);
+};
+var createProxy = async (userConfig = {}) => {
+  const config = { ...DEFAULT_CONFIG, ...userConfig };
+  const logger = chunkBHHR2U5B_cjs.createAuditLogger({ level: "info" });
+  const stateWriter = createStateWriter(300);
+  const store = await chunkBHHR2U5B_cjs.loadProfiles();
+  const providerProfiles = Object.values(store).filter(
+    (p) => p.provider === config.provider
+  );
+  if (providerProfiles.length === 0) {
+    throw new Error(
+      `No API key profiles found for provider "${config.provider}". Run o-switcher profile tools to add profiles first.`
+    );
+  }
+  const profiles = providerProfiles.map((entry) => ({
+    entry,
+    failedAt: null,
+    consecutiveFailures: 0
+  }));
+  logger.info(
+    { provider: config.provider, profiles: profiles.map((p) => p.entry.id), port: config.port },
+    "O-Switcher proxy initialized"
+  );
+  publishState(stateWriter, profiles, profiles[0]?.entry.id);
+  const handleRequest = async (req, res) => {
+    const method = req.method ?? "GET";
+    const path = req.url ?? "/";
+    if (path === "/health") {
+      res.writeHead(200, { "content-type": "application/json" });
+      res.end(JSON.stringify({ status: "ok", profiles: profiles.length }));
+      return;
+    }
+    const body = await readBody(req);
+    const pool = buildProfilePool(profiles);
+    if (pool.length === 0) {
+      res.writeHead(503, { "content-type": "application/json" });
+      res.end(JSON.stringify({ error: "All profiles exhausted" }));
+      return;
+    }
+    const maxAttempts = Math.min(config.maxRetries, pool.length);
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      const profile = pool[attempt];
+      const token = getBearerToken(profile);
+      if (!token) continue;
+      const headers = {};
+      for (const [key, value] of Object.entries(req.headers)) {
+        if (value && typeof value === "string") {
+          headers[key] = value;
+        }
+      }
+      headers["authorization"] = `Bearer ${token}`;
+      logger.info(
+        { profile: profile.entry.id, attempt: attempt + 1, path },
+        "Forwarding request"
+      );
+      try {
+        const result = await forwardRequest(method, path, headers, body, config.upstream);
+        if (result.status === 429) {
+          profile.consecutiveFailures += 1;
+          profile.failedAt = Date.now();
+          logger.warn(
+            { profile: profile.entry.id, attempt: attempt + 1, consecutiveFailures: profile.consecutiveFailures },
+            "Rate limited \u2014 trying next profile"
+          );
+          publishState(stateWriter, profiles, pool[attempt + 1]?.entry.id);
+          continue;
+        }
+        profile.consecutiveFailures = 0;
+        profile.failedAt = null;
+        publishState(stateWriter, profiles, profile.entry.id);
+        res.writeHead(result.status, result.headers);
+        if (result.body instanceof ReadableStream) {
+          const reader = result.body.getReader();
+          const pump = async () => {
+            const { done, value } = await reader.read();
+            if (done) {
+              res.end();
+              return;
+            }
+            res.write(value);
+            await pump();
+          };
+          await pump();
+        } else {
+          res.end(result.body);
+        }
+        return;
+      } catch (err) {
+        profile.consecutiveFailures += 1;
+        profile.failedAt = Date.now();
+        logger.error(
+          { profile: profile.entry.id, attempt: attempt + 1, err },
+          "Request failed \u2014 trying next profile"
+        );
+        publishState(stateWriter, profiles, pool[attempt + 1]?.entry.id);
+        continue;
+      }
+    }
+    res.writeHead(429, { "content-type": "application/json" });
+    res.end(JSON.stringify({
+      error: {
+        message: `All ${maxAttempts} profiles rate-limited. Try again later.`,
+        type: "rate_limit_error",
+        code: "all_profiles_exhausted"
+      }
+    }));
+  };
+  const start = () => new Promise((resolve) => {
+    const server = http.createServer((req, res) => {
+      handleRequest(req, res).catch((err) => {
+        logger.error({ err }, "Unhandled request error");
+        if (!res.headersSent) {
+          res.writeHead(500, { "content-type": "application/json" });
+        }
+        res.end(JSON.stringify({ error: "Internal proxy error" }));
+      });
+    });
+    server.listen(config.port, () => {
+      logger.info({ port: config.port, upstream: config.upstream }, "O-Switcher proxy listening");
+      resolve({
+        port: config.port,
+        close: () => {
+          server.close();
+          stateWriter.flush();
+        }
+      });
+    });
+  });
+  return { start };
+};
+
+exports.createProxy = createProxy;
+exports.createStateWriter = createStateWriter;

package/dist/{chunk-H72U2MNG.cjs → chunk-TGYAV5TV.cjs}

@@ -1,8 +1,7 @@
 'use strict';
 
+var chunkBHHR2U5B_cjs = require('./chunk-BHHR2U5B.cjs');
 var zod = require('zod');
-var pino = require('pino');
-var crypto = require('crypto');
 var eventemitter3 = require('eventemitter3');
 var cockatiel = require('cockatiel');
 var PQueue = require('p-queue');
@@ -13,8 +12,6 @@ var path = require('path');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
-var pino__default = /*#__PURE__*/_interopDefault(pino);
-var crypto__default = /*#__PURE__*/_interopDefault(crypto);
 var PQueue__default = /*#__PURE__*/_interopDefault(PQueue);
 
 // src/config/defaults.ts
@@ -313,37 +310,6 @@ var TARGET_STATES = [
   "Draining",
   "Disabled"
 ];
-var REDACT_PATHS = [
-  "api_key",
-  "token",
-  "secret",
-  "password",
-  "authorization",
-  "credential",
-  "credentials",
-  "*.api_key",
-  "*.token",
-  "*.secret",
-  "*.password",
-  "*.authorization",
-  "*.credential",
-  "*.credentials"
-];
-var createAuditLogger = (options) => {
-  const opts = {
-    level: options?.level ?? "info",
-    redact: {
-      paths: [...REDACT_PATHS],
-      censor: "[Redacted]"
-    }
-  };
-  if (options?.destination) {
-    return pino__default.default(opts, options.destination);
-  }
-  return pino__default.default(opts);
-};
-var createRequestLogger = (baseLogger, requestId) => baseLogger.child({ request_id: requestId });
-var generateCorrelationId = () => crypto__default.default.randomUUID();
 var RateLimitedSchema = zod.z.object({
   class: zod.z.literal("RateLimited"),
   retryable: zod.z.literal(true),
@@ -1435,72 +1401,6 @@ var createOperatorTools = (deps) => ({
     }
   })
 });
-var PROFILES_DIR = path.join(os.homedir(), ".local", "share", "o-switcher");
-var PROFILES_PATH = path.join(PROFILES_DIR, "profiles.json");
-var loadProfiles = async (path = PROFILES_PATH) => {
-  try {
-    const content = await promises.readFile(path, "utf-8");
-    return JSON.parse(content);
-  } catch (err) {
-    const code = err.code;
-    if (code === "ENOENT") {
-      return {};
-    }
-    throw err;
-  }
-};
-var saveProfiles = async (store, path$1 = PROFILES_PATH, logger) => {
-  const dir = path.dirname(path$1);
-  await promises.mkdir(dir, { recursive: true });
-  const tmpPath = `${path$1}.tmp`;
-  await promises.writeFile(tmpPath, JSON.stringify(store, null, 2), "utf-8");
-  await promises.rename(tmpPath, path$1);
-  logger?.info({ path: path$1 }, "Profiles saved to disk");
-};
-var credentialsMatch = (a, b) => {
-  if (a.type !== b.type) return false;
-  if (a.type === "api-key" && b.type === "api-key") {
-    return a.key === b.key;
-  }
-  if (a.type === "oauth" && b.type === "oauth") {
-    return a.refresh === b.refresh && a.access === b.access && a.expires === b.expires && a.accountId === b.accountId;
-  }
-  return false;
-};
-var addProfile = (store, provider, credentials) => {
-  const isDuplicate = Object.values(store).some(
-    (entry2) => entry2.provider === provider && credentialsMatch(entry2.credentials, credentials)
-  );
-  if (isDuplicate) {
-    return store;
-  }
-  const id = nextProfileId(store, provider);
-  const entry = {
-    id,
-    provider,
-    type: credentials.type,
-    credentials,
-    created: (/* @__PURE__ */ new Date()).toISOString()
-  };
-  return { ...store, [id]: entry };
-};
-var removeProfile = (store, id) => {
-  if (store[id] === void 0) {
-    return { store, removed: false };
-  }
-  const { [id]: _removed, ...rest } = store;
-  return { store: rest, removed: true };
-};
-var listProfiles = (store) => {
-  return Object.values(store).sort(
-    (a, b) => new Date(a.created).getTime() - new Date(b.created).getTime()
-  );
-};
-var nextProfileId = (store, provider) => {
-  const prefix = `${provider}-`;
-  const maxN = Object.keys(store).filter((key) => key.startsWith(prefix)).map((key) => Number(key.slice(prefix.length))).filter((n) => !Number.isNaN(n)).reduce((max, n) => Math.max(max, n), 0);
-  return `${provider}-${maxN + 1}`;
-};
 var AUTH_JSON_PATH = path.join(os.homedir(), ".local", "share", "opencode", "auth.json");
 var AUTH_DIR = path.join(os.homedir(), ".local", "share", "opencode");
 var isAuthFile = (filename) => filename === "auth.json" || /^auth-work\d+\.json$/.test(filename);
@@ -1551,14 +1451,14 @@ var entriesEqual = (a, b) => {
 };
 var createAuthWatcher = (options) => {
   const authDir = options?.authJsonPath ? path.dirname(options.authJsonPath) : AUTH_DIR;
-  const profPath = options?.profilesPath ?? PROFILES_PATH;
+  const profPath = options?.profilesPath ?? chunkBHHR2U5B_cjs.PROFILES_PATH;
   const log = options?.logger?.child({ component: "profile-watcher" });
   let lastKnownAuth = {};
   let abortController = null;
   let debounceTimer = null;
   const processChange = async () => {
     const newAuth = await readAllAuthFiles(authDir);
-    let store = await loadProfiles(profPath);
+    let store = await chunkBHHR2U5B_cjs.loadProfiles(profPath);
     let changed = false;
     for (const [compositeKey, entry] of Object.entries(newAuth)) {
       if (!entry) continue;
@@ -1570,7 +1470,7 @@ var createAuthWatcher = (options) => {
           "Credential change detected \u2014 saving previous credential"
         );
         const prevCredential = toCredential(previousEntry);
-        const newStore = addProfile(store, provider, prevCredential);
+        const newStore = chunkBHHR2U5B_cjs.addProfile(store, provider, prevCredential);
         if (newStore !== store) {
           store = newStore;
           changed = true;
@@ -1578,7 +1478,7 @@ var createAuthWatcher = (options) => {
       } else if (previousEntry === void 0) {
         log?.info({ provider, source: compositeKey, action: "new_provider" }, "New auth file detected");
         const credential = toCredential(entry);
-        const newStore = addProfile(store, provider, credential);
+        const newStore = chunkBHHR2U5B_cjs.addProfile(store, provider, credential);
         if (newStore !== store) {
           store = newStore;
           changed = true;
@@ -1586,7 +1486,7 @@ var createAuthWatcher = (options) => {
       }
     }
     if (changed) {
-      await saveProfiles(store, profPath);
+      await chunkBHHR2U5B_cjs.saveProfiles(store, profPath);
       log?.info({ profiles_saved: true }, "Profiles saved to disk");
     }
     lastKnownAuth = newAuth;
@@ -1604,7 +1504,7 @@ var createAuthWatcher = (options) => {
   };
   const start = async () => {
     const currentAuth = await readAllAuthFiles(authDir);
-    const currentStore = await loadProfiles(profPath);
+    const currentStore = await chunkBHHR2U5B_cjs.loadProfiles(profPath);
     log?.info({ auth_keys: Object.keys(currentAuth).length }, "Auth watcher started");
     if (Object.keys(currentStore).length === 0 && Object.keys(currentAuth).length > 0) {
       let store = {};
@@ -1612,9 +1512,9 @@ var createAuthWatcher = (options) => {
         if (!entry) continue;
         const provider = compositeKey.split(":")[0] ?? compositeKey;
         const credential = toCredential(entry);
-        store = addProfile(store, provider, credential);
+        store = chunkBHHR2U5B_cjs.addProfile(store, provider, credential);
       }
-      await saveProfiles(store, profPath);
+      await chunkBHHR2U5B_cjs.saveProfiles(store, profPath);
       log?.info(
         { profiles_initialized: Object.keys(store).length },
         "Initialized profiles from auth files"
@@ -1674,10 +1574,10 @@ var readCurrentAuth = async (authPath) => {
   }
 };
 var switchProfile = async (options) => {
-  const profPath = options.profilesPath ?? PROFILES_PATH;
+  const profPath = options.profilesPath ?? chunkBHHR2U5B_cjs.PROFILES_PATH;
   const authPath = options.authJsonPath ?? AUTH_JSON_PATH;
   const log = options.logger?.child({ component: "profile-switcher" });
-  const store = await loadProfiles(profPath);
+  const store = await chunkBHHR2U5B_cjs.loadProfiles(profPath);
   const target = store[options.targetProfileId];
   if (target === void 0) {
     log?.warn({ profileId: options.targetProfileId }, "Profile not found");
@@ -1708,9 +1608,9 @@ var switchProfile = async (options) => {
   };
 };
 var switchToNextProfile = async (options) => {
-  const profPath = options.profilesPath ?? PROFILES_PATH;
+  const profPath = options.profilesPath ?? chunkBHHR2U5B_cjs.PROFILES_PATH;
   const log = options.logger?.child({ component: "profile-switcher" });
-  const store = await loadProfiles(profPath);
+  const store = await chunkBHHR2U5B_cjs.loadProfiles(profPath);
   const excludeSet = new Set(options.excludeProfileIds ?? []);
   if (options.currentProfileId) {
     excludeSet.add(options.currentProfileId);
@@ -1738,8 +1638,8 @@ var createProfileTools = (options) => ({
     description: "List all saved auth profiles with provider, type, and creation date.",
     args: {},
     async execute() {
-      const store = await loadProfiles(options?.profilesPath);
-      const profiles = listProfiles(store);
+      const store = await chunkBHHR2U5B_cjs.loadProfiles(options?.profilesPath);
+      const profiles = chunkBHHR2U5B_cjs.listProfiles(store);
       const result = profiles.map((entry) => ({
         id: entry.id,
         provider: entry.provider,
@@ -1753,10 +1653,10 @@ var createProfileTools = (options) => ({
     description: "Remove a saved auth profile by ID.",
     args: { id: z4.string().min(1) },
     async execute(args) {
-      const store = await loadProfiles(options?.profilesPath);
-      const { store: newStore, removed } = removeProfile(store, args.id);
+      const store = await chunkBHHR2U5B_cjs.loadProfiles(options?.profilesPath);
+      const { store: newStore, removed } = chunkBHHR2U5B_cjs.removeProfile(store, args.id);
       if (removed) {
-        await saveProfiles(newStore, options?.profilesPath);
+        await chunkBHHR2U5B_cjs.saveProfiles(newStore, options?.profilesPath);
         return JSON.stringify({ success: true, id: args.id });
       }
       return JSON.stringify({
@@ -1785,12 +1685,10 @@ exports.DualBreaker = DualBreaker;
 exports.EXCLUSION_REASONS = EXCLUSION_REASONS;
 exports.ErrorClassSchema = ErrorClassSchema;
 exports.INITIAL_HEALTH_SCORE = INITIAL_HEALTH_SCORE;
-exports.REDACT_PATHS = REDACT_PATHS;
 exports.SwitcherConfigSchema = SwitcherConfigSchema;
 exports.TARGET_STATES = TARGET_STATES;
 exports.TargetConfigSchema = TargetConfigSchema;
 exports.TargetRegistry = TargetRegistry;
-exports.addProfile = addProfile;
 exports.applyConfigDiff = applyConfigDiff;
 exports.checkHardRejects = checkHardRejects;
 exports.computeBackoffMs = computeBackoffMs;
@@ -1798,7 +1696,6 @@ exports.computeConfigDiff = computeConfigDiff;
 exports.computeCooldownMs = computeCooldownMs;
 exports.computeScore = computeScore;
 exports.createAdmissionController = createAdmissionController;
-exports.createAuditLogger = createAuditLogger;
 exports.createAuthWatcher = createAuthWatcher;
 exports.createCircuitBreaker = createCircuitBreaker;
 exports.createConcurrencyTracker = createConcurrencyTracker;
@@ -1808,7 +1705,6 @@ exports.createLogSubscriber = createLogSubscriber;
 exports.createOperatorTools = createOperatorTools;
 exports.createProfileTools = createProfileTools;
 exports.createRegistry = createRegistry;
-exports.createRequestLogger = createRequestLogger;
 exports.createRequestTraceBuffer = createRequestTraceBuffer;
 exports.createRetryPolicy = createRetryPolicy;
 exports.createRoutingEventBus = createRoutingEventBus;
@@ -1816,21 +1712,15 @@ exports.disableTarget = disableTarget;
 exports.discoverTargets = discoverTargets;
 exports.discoverTargetsFromProfiles = discoverTargetsFromProfiles;
 exports.drainTarget = drainTarget;
-exports.generateCorrelationId = generateCorrelationId;
 exports.getExclusionReason = getExclusionReason;
 exports.getTargetStateTransition = getTargetStateTransition;
 exports.inspectRequest = inspectRequest;
 exports.isRetryable = isRetryable;
-exports.listProfiles = listProfiles;
 exports.listTargets = listTargets;
-exports.loadProfiles = loadProfiles;
-exports.nextProfileId = nextProfileId;
 exports.normalizeLatency = normalizeLatency;
 exports.pauseTarget = pauseTarget;
 exports.reloadConfig = reloadConfig;
-exports.removeProfile = removeProfile;
 exports.resumeTarget = resumeTarget;
-exports.saveProfiles = saveProfiles;
 exports.selectTarget = selectTarget;
 exports.switchProfile = switchProfile;
 exports.switchToNextProfile = switchToNextProfile;