opencode-nim-rotator 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,201 @@
+# opencode-nim-rotator
+
+An [OpenCode](https://opencode.ai) plugin for managing and rotating multiple NVIDIA NIM API keys.
+
+## How It Works
+
+The plugin uses OpenCode's `auth` hook with a custom `fetch` function to intercept every NVIDIA NIM API call and inject a rotated API key into the `Authorization` header. This is the same pattern used by the codex and github-copilot plugins in production.
+
+**Key rotation happens per-request**: each LLM call uses the next key in the rotation. If a key returns 401/403, it automatically increments the failure count and retries with the next key.
+
+## Install
+
+```bash
+npm install -g opencode-nim-rotator
+```
+
+The postinstall script automatically adds the plugin to your `~/.config/opencode/opencode.json`.
+
+## Setup
+
+### 1. Add API keys
+
+Run the TUI manager:
+
+```bash
+bun opencode-nim-rotator
+```
+
+Or manually — add at least one key via OpenCode's auth system:
+
+```bash
+opencode /connect nvidia
+```
+
+Select "Enter NVIDIA NIM API Key" and paste your key.
+
+### 2. Add more keys via the TUI
+
+```bash
+bun opencode-nim-rotator
+```
+
+The TUI lets you:
+
+- **Add** keys with a friendly name
+- **Rename** keys
+- **Delete** keys
+- **Toggle** keys on/off
+- **Reset** failure counts
+- **Switch** rotation strategy (round-robin or least-failures)
+
+### 3. Restart OpenCode
+
+After adding keys, restart opencode. The plugin's `auth` loader will fire on startup and provide the custom `fetch` that rotates keys.
+
+## Configuration
+
+### Environment Variables
+
+| Variable                   | Description                         | Default                                    |
+| -------------------------- | ----------------------------------- | ------------------------------------------ |
+| `NIM_ROTATOR_STORE_PATH`   | Path to key store JSON file         | `~/.config/opencode/nim-rotator-keys.json` |
+| `NIM_ROTATOR_MAX_FAILURES` | Max failures before disabling a key | `5`                                        |
+
+### opencode.json Options
+
+```json
+{
+  "plugin": [
+    [
+      "opencode-nim-rotator",
+      {
+        "rotationStrategy": "round-robin",
+        "storePath": "/custom/path/to/keys.json"
+      }
+    ]
+  ]
+}
+```
+
+### Rotation Strategies
+
+- **`round-robin`** (default): Cycles through keys in order
+- **`least-failures`**: Always uses the key with the fewest failures
+
+## How Key Rotation Works
+
+1. On startup, the `auth` hook `loader` registers a custom `fetch` for the `nvidia` provider
+2. Every NVIDIA API call goes through this custom fetch
+3. The fetch selects the next key based on the rotation strategy
+4. The `Authorization: Bearer <key>` header is replaced with the rotated key
+5. If the request returns 401/403, the failure count is incremented and the next key is tried
+6. Keys that exceed `NIM_ROTATOR_MAX_FAILURES` are automatically skipped
+7. Successful requests reset the key's failure count to 0
+
+## Key Store Format
+
+Keys are stored in `~/.config/opencode/nim-rotator-keys.json` with file mode `0600`:
+
+```json
+{
+  "keys": [
+    {
+      "id": "uuid",
+      "name": "work-key",
+      "key": "nvapi-...",
+      "createdAt": 1700000000000,
+      "lastUsedAt": 1700000100000,
+      "failureCount": 0,
+      "enabled": true
+    }
+  ],
+  "currentIndex": 0,
+  "rotationStrategy": "round-robin",
+  "updatedAt": 1700000000000
+}
+```
+
+## Themes
+
+The TUI supports multiple color themes that match OpenCode's built-in themes. By default, the rotator **syncs with your `opencode.json` theme setting** — when you change your theme in OpenCode, the rotator picks it up automatically.
+
+### Available Themes
+
+| ID           | Name               |
+| ------------ | ------------------ |
+| `opencode`   | OpenCode (default) |
+| `catppuccin` | Catppuccin Mocha   |
+| `dracula`    | Dracula            |
+| `gruvbox`    | Gruvbox            |
+| `kanagawa`   | Kanagawa           |
+| `nord`       | Nord               |
+| `one-dark`   | One Dark           |
+| `rosepine`   | Rose Pine          |
+| `solarized`  | Solarized          |
+| `tokyonight` | Tokyonight         |
+
+### Setting a Theme
+
+From the TUI main menu, select **Theme** to pick a theme or switch back to syncing with `opencode.json`.
+
+To override via `opencode.json`:
+
+```json
+{
+  "theme": "dracula"
+}
+```
+
+To override independently (saved in the key store):
+
+```json
+{
+  "theme": "kanagawa",
+  "keys": [...]
+}
+```
+
+Set the store `theme` field to `""` or remove it to revert to syncing with `opencode.json`.
+
+## TUI
+
+The TUI is built with [OpenTUI](https://opentui.com) and provides a menu-driven interface:
+
+- Main menu with Add, Manage, Strategy, and Theme options
+- Key selector showing name, masked key, failure count, last used
+- Key actions: toggle, rename, delete
+- Add key flow: enter name, then enter key
+- Theme selector with sync-to-opencode option
+
+## Development
+
+```bash
+# Install dependencies
+bun install
+
+# Run TUI locally
+bun run tui
+
+# Build TypeScript
+bun run build
+```
+
+## Uninstall
+
+To remove the plugin and clean up all associated data:
+
+```bash
+npm uninstall -g opencode-nim-rotator
+```
+
+The uninstaller will automatically:
+
+1. Remove `opencode-nim-rotator` from your `~/.config/opencode/opencode.json` plugin list
+2. Delete your key store file at `~/.config/opencode/nim-rotator-keys.json`
+
+**Note:** This will permanently delete all stored API keys, so back them up first if needed. After uninstalling, restart opencode to apply the changes.
+
+## License
+
+MIT

package/bin/opencode-nim-rotator.ts

@@ -0,0 +1,10 @@
+#!/usr/bin/env bun
+
+import { createCliRenderer } from "@opentui/core";
+import { state } from "../src/tui/state.js";
+import { initApp } from "../src/tui/app.js";
+
+const renderer = await createCliRenderer({ exitOnCtrlC: false });
+
+state.renderer = renderer;
+initApp();

package/dist/index.d.ts

@@ -0,0 +1,4 @@
+import type { Plugin } from "@opencode-ai/plugin";
+export declare const NvidiaNimKeyRotator: Plugin;
+export default NvidiaNimKeyRotator;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAsB,MAAM,qBAAqB,CAAC;AAkCtE,eAAO,MAAM,mBAAmB,EAAE,MAyFjC,CAAC;AAEF,eAAe,mBAAmB,CAAC"}

package/dist/index.js

@@ -0,0 +1,103 @@
+import { loadStore, saveStore, addKey, getNextKey, recordFailure, getActiveKeys, } from "./storage.js";
+const PROVIDER_ID = "nvidia";
+const NIM_BASE_URL = "https://integrate.api.nvidia.com";
+const VALID_STRATEGIES = ["round-robin", "least-failures"];
+function isValidStrategy(val) {
+    return typeof val === "string" && VALID_STRATEGIES.includes(val);
+}
+function isAuthError(obj) {
+    if (typeof obj === "object" && obj !== null) {
+        const code = obj.code;
+        const status = obj.status;
+        if (code === 401 || code === 403 || status === 401 || status === 403)
+            return true;
+    }
+    const msg = String(obj).toLowerCase();
+    return (msg.includes("401") || msg.includes("403") || msg.includes("unauthorized"));
+}
+export const NvidiaNimKeyRotator = async (input, options) => {
+    const config = {
+        storePath: options?.storePath,
+        rotationStrategy: isValidStrategy(options?.rotationStrategy)
+            ? options.rotationStrategy
+            : "round-robin",
+    };
+    const store = loadStore(config);
+    const activeKeys = getActiveKeys(store, config);
+    // Seed an env key if the store is empty
+    if (activeKeys.length === 0) {
+        const envKey = process.env.NVIDIA_API_KEY;
+        if (envKey) {
+            const existing = store.keys.find((k) => k.name === "env-default");
+            if (!existing) {
+                addKey(store, "env-default", envKey);
+                saveStore(store, config);
+            }
+        }
+    }
+    const hooks = {
+        auth: {
+            provider: PROVIDER_ID,
+            // We deliberately omit auth.loader. The fetch we used
+            // to return was never called by opencode; it only uses
+            // the apiKey string from the authorize step.  Rotation is
+            // handled in chat.headers instead.
+            methods: [
+                {
+                    type: "api",
+                    label: "Enter NVIDIA NIM API Key",
+                    async authorize(inputs) {
+                        const key = inputs?.["apiKey"];
+                        if (!key)
+                            return { type: "failed" };
+                        try {
+                            const res = await fetch(`${NIM_BASE_URL}/v1/models`, {
+                                headers: { Authorization: `Bearer ${key}` },
+                            });
+                            if (!res.ok)
+                                return { type: "failed" };
+                        }
+                        catch {
+                            return { type: "failed" };
+                        }
+                        return {
+                            type: "success",
+                            key,
+                            provider: PROVIDER_ID,
+                        };
+                    },
+                },
+            ],
+        },
+        // Rotate the API key on every outgoing request by mutating the
+        // Authorization header.  This is the hook that actually runs
+        // before each LLM call, unlike auth.loader which only runs once.
+        "chat.headers": async (_input, _output) => {
+            const next = getNextKey(store, config);
+            if (next) {
+                _output.headers["Authorization"] = `Bearer ${next.key.key}`;
+                saveStore(store, config);
+            }
+        },
+        "shell.env": async (_input, output) => {
+            if (output.env.NVIDIA_API_KEY !== undefined) {
+                const next = getNextKey(store, config);
+                if (next) {
+                    output.env.NVIDIA_API_KEY = next.key.key;
+                    saveStore(store, config);
+                }
+            }
+        },
+        event: async ({ event }) => {
+            if (event.type === "session.error" && isAuthError(event.error)) {
+                if (store.lastUsedKeyId) {
+                    recordFailure(store, store.lastUsedKeyId);
+                    saveStore(store, config);
+                }
+            }
+        },
+    };
+    return hooks;
+};
+export default NvidiaNimKeyRotator;
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,SAAS,EACT,SAAS,EACT,MAAM,EACN,UAAU,EACV,aAAa,EACb,aAAa,GACd,MAAM,cAAc,CAAC;AAGtB,MAAM,WAAW,GAAG,QAAQ,CAAC;AAC7B,MAAM,YAAY,GAAG,kCAAkC,CAAC;AACxD,MAAM,gBAAgB,GAAG,CAAC,aAAa,EAAE,gBAAgB,CAAU,CAAC;AAEpE,SAAS,eAAe,CACtB,GAAY;IAEZ,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,gBAAgB,CAAC,QAAQ,CAAC,GAAU,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;QAC5C,MAAM,IAAI,GAAI,GAAW,CAAC,IAAI,CAAC;QAC/B,MAAM,MAAM,GAAI,GAAW,CAAC,MAAM,CAAC;QACnC,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG;YAClE,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;IACtC,OAAO,CACL,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC,CAC3E,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAW,KAAK,EAC9C,KAAkB,EAClB,OAAiC,EACjC,EAAE;IACF,MAAM,MAAM,GAAmB;QAC7B,SAAS,EAAE,OAAO,EAAE,SAA+B;QACnD,gBAAgB,EAAE,eAAe,CAAC,OAAO,EAAE,gBAAgB,CAAC;YAC1D,CAAC,CAAC,OAAQ,CAAC,gBAAgB;YAC3B,CAAC,CAAC,aAAa;KAClB,CAAC;IAEF,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAEhD,wCAAwC;IACxC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAC1C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YAClE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;gBACrC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAU;QACnB,IAAI,EAAE;YACJ,QAAQ,EAAE,WAAW;YACrB,sDAAsD;YACtD,uDAAuD;YACvD,0DAA0D;YAC1D,mCAAmC;YACnC,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,KAAK;oBACX,KAAK,EAAE,0BAA0B;oBACjC,KAAK,CAAC,SAAS,CAAC,MAAM;wBACpB,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,CAAC;wBAC/B,IAAI,CAAC,GAAG;4BAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;wBAEpC,IAAI,CAAC;4BACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,YAAY,YAAY,EAAE;gCACnD,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,GAAG,EAAE,EAAE;6BAC5C,CAAC,CAAC;4BACH,IAAI,CAAC,GAAG,CAAC,EAAE;gCAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;wBACzC,CAAC;wBAAC,MAAM,CAAC;4BACP,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;wBAC5B,CAAC;wBAED,OAAO;4BACL,IAAI,EAAE,SAAS;4BACf,GAAG;4BACH,QAAQ,EAAE,WAAW;yBACtB,CAAC;oBACJ,CAAC;iBACF;aACF;SACF;QACD,+DAA+D;QAC/D,6DAA6D;QAC7D,iEAAiE;QACjE,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YACxC,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACvC,IAAI,IAAI,EAAE,CAAC;gBACT,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;gBAC5D,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QACD,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;YACpC,IAAI,MAAM,CAAC,GAAG,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;gBAC5C,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBACvC,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,CAAC,GAAG,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;oBACzC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;QACD,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;YACzB,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,IAAI,WAAW,CAAE,KAAa,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxE,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;oBACxB,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;oBAC1C,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,eAAe,mBAAmB,CAAC"}

package/dist/storage.d.ts

@@ -0,0 +1,18 @@
+import type { ApiKeyEntry, KeyStore, KeyStoreConfig } from "./types.js";
+export declare const DEFAULT_MAX_FAILURES = 5;
+export declare function resolveStorePath(config?: KeyStoreConfig): string;
+export declare function loadStore(config?: KeyStoreConfig): KeyStore;
+export declare function saveStore(store: KeyStore, config?: KeyStoreConfig): void;
+export declare function addKey(store: KeyStore, name: string, key: string): void;
+export declare function removeKey(store: KeyStore, id: string): void;
+export declare function renameKey(store: KeyStore, id: string, newName: string): void;
+export declare function toggleKey(store: KeyStore, id: string, enabled?: boolean): void;
+export declare function getMaxFailures(config?: KeyStoreConfig): number;
+export declare function getActiveKeys(store: KeyStore, config?: KeyStoreConfig): ApiKeyEntry[];
+export declare function getNextKey(store: KeyStore, config?: KeyStoreConfig): {
+    key: ApiKeyEntry;
+    index: number;
+} | null;
+export declare function recordFailure(store: KeyStore, keyId: string): void;
+export declare function resetFailures(store: KeyStore, keyId?: string): void;
+//# sourceMappingURL=storage.d.ts.map

package/dist/storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../src/storage.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAQxE,eAAO,MAAM,oBAAoB,IAAI,CAAC;AAWtC,wBAAgB,gBAAgB,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,MAAM,CAMhE;AAED,wBAAgB,SAAS,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,QAAQ,CAwB3D;AAED,wBAAgB,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,cAAc,GAAG,IAAI,CAUxE;AAED,wBAAgB,MAAM,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,IAAI,CAUvE;AAED,wBAAgB,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,GAAG,IAAI,CAO3D;AAED,wBAAgB,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAG5E;AAED,wBAAgB,SAAS,CACvB,KAAK,EAAE,QAAQ,EACf,EAAE,EAAE,MAAM,EACV,OAAO,CAAC,EAAE,OAAO,GAChB,IAAI,CAGN;AAED,wBAAgB,cAAc,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,MAAM,CAK9D;AAED,wBAAgB,aAAa,CAC3B,KAAK,EAAE,QAAQ,EACf,MAAM,CAAC,EAAE,cAAc,GACtB,WAAW,EAAE,CAGf;AAED,wBAAgB,UAAU,CACxB,KAAK,EAAE,QAAQ,EACf,MAAM,CAAC,EAAE,cAAc,GACtB;IAAE,GAAG,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAyB5C;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAGlE;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAOnE"}

package/dist/storage.js

@@ -0,0 +1,129 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { join, dirname } from "path";
+import { homedir } from "os";
+const DEFAULT_STORE_PATH = join(homedir(), ".config", "opencode", "nim-rotator-keys.json");
+export const DEFAULT_MAX_FAILURES = 5;
+function getDefaultStore() {
+    return {
+        keys: [],
+        currentIndex: 0,
+        rotationStrategy: "round-robin",
+        updatedAt: Date.now(),
+    };
+}
+export function resolveStorePath(config) {
+    return (config?.storePath ??
+        process.env.NIM_ROTATOR_STORE_PATH ??
+        DEFAULT_STORE_PATH);
+}
+export function loadStore(config) {
+    const storePath = resolveStorePath(config);
+    try {
+        if (existsSync(storePath)) {
+            const raw = readFileSync(storePath, "utf-8");
+            const data = JSON.parse(raw);
+            if (!data.keys || !Array.isArray(data.keys)) {
+                console.warn(`[nim-rotator] Store at "${storePath}" has invalid keys format, using defaults`);
+                return getDefaultStore();
+            }
+            return {
+                ...getDefaultStore(),
+                ...data,
+            };
+        }
+    }
+    catch (err) {
+        console.error(`[nim-rotator] Failed to load store at "${storePath}":`, err);
+        console.warn("[nim-rotator] Starting with a fresh store. Your existing keys are preserved on disk.");
+    }
+    return getDefaultStore();
+}
+export function saveStore(store, config) {
+    const storePath = resolveStorePath(config);
+    const dir = dirname(storePath);
+    if (!existsSync(dir)) {
+        mkdirSync(dir, { recursive: true });
+    }
+    store.updatedAt = Date.now();
+    writeFileSync(storePath, JSON.stringify(store, null, 2) + "\n", {
+        mode: 0o600,
+    });
+}
+export function addKey(store, name, key) {
+    const entry = {
+        id: crypto.randomUUID(),
+        name,
+        key,
+        createdAt: Date.now(),
+        failureCount: 0,
+        enabled: true,
+    };
+    store.keys.push(entry);
+}
+export function removeKey(store, id) {
+    const index = store.keys.findIndex((k) => k.id === id);
+    if (index === -1)
+        return;
+    store.keys.splice(index, 1);
+    if (store.currentIndex >= store.keys.length) {
+        store.currentIndex = 0;
+    }
+}
+export function renameKey(store, id, newName) {
+    const entry = store.keys.find((k) => k.id === id);
+    if (entry)
+        entry.name = newName;
+}
+export function toggleKey(store, id, enabled) {
+    const entry = store.keys.find((k) => k.id === id);
+    if (entry)
+        entry.enabled = enabled ?? !entry.enabled;
+}
+export function getMaxFailures(config) {
+    return (config?.maxFailuresBeforeDisable ??
+        (Number(process.env.NIM_ROTATOR_MAX_FAILURES) || DEFAULT_MAX_FAILURES));
+}
+export function getActiveKeys(store, config) {
+    const maxFailures = getMaxFailures(config);
+    return store.keys.filter((k) => k.enabled && k.failureCount < maxFailures);
+}
+export function getNextKey(store, config) {
+    const active = getActiveKeys(store, config);
+    if (active.length === 0)
+        return null;
+    const strategy = config?.rotationStrategy ?? store.rotationStrategy ?? "round-robin";
+    if (strategy === "least-failures") {
+        const sorted = [...active].sort((a, b) => a.failureCount - b.failureCount);
+        const best = sorted[0];
+        const idx = store.keys.indexOf(best);
+        store.currentIndex = idx;
+        store.lastUsedKeyId = best.id;
+        best.lastUsedAt = Date.now();
+        return { key: best, index: idx };
+    }
+    // round-robin
+    let idx = store.currentIndex % active.length;
+    const selected = active[idx];
+    const realIdx = store.keys.indexOf(selected);
+    store.currentIndex = (idx + 1) % active.length;
+    store.lastUsedKeyId = selected.id;
+    selected.lastUsedAt = Date.now();
+    return { key: selected, index: realIdx };
+}
+export function recordFailure(store, keyId) {
+    const entry = store.keys.find((k) => k.id === keyId);
+    if (entry)
+        entry.failureCount++;
+}
+export function resetFailures(store, keyId) {
+    if (keyId) {
+        const entry = store.keys.find((k) => k.id === keyId);
+        if (entry)
+            entry.failureCount = 0;
+    }
+    else {
+        for (const k of store.keys)
+            k.failureCount = 0;
+    }
+}
+//# sourceMappingURL=storage.js.map

package/dist/storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.js","sourceRoot":"","sources":["../src/storage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAG7B,MAAM,kBAAkB,GAAG,IAAI,CAC7B,OAAO,EAAE,EACT,SAAS,EACT,UAAU,EACV,uBAAuB,CACxB,CAAC;AACF,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAEtC,SAAS,eAAe;IACtB,OAAO;QACL,IAAI,EAAE,EAAE;QACR,YAAY,EAAE,CAAC;QACf,gBAAgB,EAAE,aAAa;QAC/B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAuB;IACtD,OAAO,CACL,MAAM,EAAE,SAAS;QACjB,OAAO,CAAC,GAAG,CAAC,sBAAsB;QAClC,kBAAkB,CACnB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,MAAuB;IAC/C,MAAM,SAAS,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,CAAC;QACH,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1B,MAAM,GAAG,GAAG,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAa,CAAC;YACzC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,OAAO,CAAC,IAAI,CACV,2BAA2B,SAAS,2CAA2C,CAChF,CAAC;gBACF,OAAO,eAAe,EAAE,CAAC;YAC3B,CAAC;YACD,OAAO;gBACL,GAAG,eAAe,EAAE;gBACpB,GAAG,IAAI;aACR,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,0CAA0C,SAAS,IAAI,EAAE,GAAG,CAAC,CAAC;QAC5E,OAAO,CAAC,IAAI,CACV,sFAAsF,CACvF,CAAC;IACJ,CAAC;IACD,OAAO,eAAe,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAe,EAAE,MAAuB;IAChE,MAAM,SAAS,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IAC/B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACrB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtC,CAAC;IACD,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE;QAC9D,IAAI,EAAE,KAAK;KACZ,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,KAAe,EAAE,IAAY,EAAE,GAAW;IAC/D,MAAM,KAAK,GAAgB;QACzB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;QACvB,IAAI;QACJ,GAAG;QACH,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,YAAY,EAAE,CAAC;QACf,OAAO,EAAE,IAAI;KACd,CAAC;IACF,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAe,EAAE,EAAU;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,CAAC,CAAC;QAAE,OAAO;IACzB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC5B,IAAI,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QAC5C,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC;IACzB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAe,EAAE,EAAU,EAAE,OAAe;IACpE,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,IAAI,KAAK;QAAE,KAAK,CAAC,IAAI,GAAG,OAAO,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,SAAS,CACvB,KAAe,EACf,EAAU,EACV,OAAiB;IAEjB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,IAAI,KAAK;QAAE,KAAK,CAAC,OAAO,GAAG,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAuB;IACpD,OAAO,CACL,MAAM,EAAE,wBAAwB;QAChC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,IAAI,oBAAoB,CAAC,CACvE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,KAAe,EACf,MAAuB;IAEvB,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IAC3C,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,YAAY,GAAG,WAAW,CAAC,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,UAAU,CACxB,KAAe,EACf,MAAuB;IAEvB,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC5C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAErC,MAAM,QAAQ,GACZ,MAAM,EAAE,gBAAgB,IAAI,KAAK,CAAC,gBAAgB,IAAI,aAAa,CAAC;IAEtE,IAAI,QAAQ,KAAK,gBAAgB,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;QAC3E,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACvB,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrC,KAAK,CAAC,YAAY,GAAG,GAAG,CAAC;QACzB,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;IACnC,CAAC;IAED,cAAc;IACd,IAAI,GAAG,GAAG,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7C,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC7C,KAAK,CAAC,YAAY,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IAC/C,KAAK,CAAC,aAAa,GAAG,QAAQ,CAAC,EAAE,CAAC;IAClC,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACjC,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAe,EAAE,KAAa;IAC1D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;IACrD,IAAI,KAAK;QAAE,KAAK,CAAC,YAAY,EAAE,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAe,EAAE,KAAc;IAC3D,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;QACrD,IAAI,KAAK;YAAE,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC;IACpC,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,IAAI;YAAE,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC;IACjD,CAAC;AACH,CAAC"}

package/dist/themes.d.ts

@@ -0,0 +1,64 @@
+export interface SelectThemeColors {
+    backgroundColor: string;
+    focusedBackgroundColor: string;
+    focusedTextColor: string;
+    selectedBackgroundColor: string;
+    selectedTextColor: string;
+    textColor: string;
+    descriptionColor: string;
+    selectedDescriptionColor: string;
+}
+export interface InputThemeColors {
+    backgroundColor: string;
+    focusedBackgroundColor: string;
+    textColor: string;
+    cursorColor: string;
+}
+export declare function selectColors(theme: RotatorTheme): SelectThemeColors;
+export declare function inputColors(theme: RotatorTheme): InputThemeColors;
+export declare function dangerSelectColors(theme: RotatorTheme): SelectThemeColors;
+export declare function applySelectColors(target: {
+    backgroundColor: string;
+    focusedBackgroundColor: string;
+    focusedTextColor: string;
+    selectedBackgroundColor: string;
+    selectedTextColor: string;
+    textColor: string;
+    descriptionColor: string;
+    selectedDescriptionColor: string;
+}, colors: SelectThemeColors): void;
+export interface RotatorTheme {
+    id: string;
+    name: string;
+    background: string;
+    backgroundPanel: string;
+    backgroundElement: string;
+    text: string;
+    textMuted: string;
+    primary: string;
+    primaryMuted: string;
+    accent: string;
+    error: string;
+    errorBg: string;
+    success: string;
+    warning: string;
+    selectedBg: string;
+    selectedText: string;
+    border: string;
+    borderActive: string;
+    inputBg: string;
+    inputFocusedBg: string;
+    cursor: string;
+    description: string;
+    selectedDescription: string;
+}
+export declare function getTheme(id: string): RotatorTheme;
+export declare function listThemes(): RotatorTheme[];
+export declare function getThemeIdFromOpenCodeConfig(): string | null;
+export declare function getResolvedTheme(): RotatorTheme;
+export declare function getThemeOverride(): string | null;
+export declare function saveThemeOverride(themeId: string): void;
+export declare function setPreviewTheme(themeId: string | null): void;
+export declare function getPreviewTheme(): string | null;
+export declare function getActiveTheme(): RotatorTheme;
+//# sourceMappingURL=themes.d.ts.map

package/dist/themes.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"themes.d.ts","sourceRoot":"","sources":["../src/themes.ts"],"names":[],"mappings":"AA6EA,MAAM,WAAW,iBAAiB;IAChC,eAAe,EAAE,MAAM,CAAC;IACxB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,gBAAgB,EAAE,MAAM,CAAC;IACzB,uBAAuB,EAAE,MAAM,CAAC;IAChC,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,wBAAwB,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,gBAAgB;IAC/B,eAAe,EAAE,MAAM,CAAC;IACxB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,YAAY,GAAG,iBAAiB,CAWnE;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,YAAY,GAAG,gBAAgB,CAOjE;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,YAAY,GAAG,iBAAiB,CAWzE;AAED,wBAAgB,iBAAiB,CAC/B,MAAM,EAAE;IACN,eAAe,EAAE,MAAM,CAAC;IACxB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,gBAAgB,EAAE,MAAM,CAAC;IACzB,uBAAuB,EAAE,MAAM,CAAC;IAChC,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,wBAAwB,EAAE,MAAM,CAAC;CAClC,EACD,MAAM,EAAE,iBAAiB,GACxB,IAAI,CASN;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,MAAM,CAAC;IACxB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AA+PD,wBAAgB,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,YAAY,CAEjD;AAED,wBAAgB,UAAU,IAAI,YAAY,EAAE,CAE3C;AAED,wBAAgB,4BAA4B,IAAI,MAAM,GAAG,IAAI,CAG5D;AAED,wBAAgB,gBAAgB,IAAI,YAAY,CAM/C;AAED,wBAAgB,gBAAgB,IAAI,MAAM,GAAG,IAAI,CAShD;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAmBvD;AAID,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI,CAE5D;AAED,wBAAgB,eAAe,IAAI,MAAM,GAAG,IAAI,CAE/C;AAED,wBAAgB,cAAc,IAAI,YAAY,CAS7C"}