swayrouter 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.env.example +43 -0
- package/CHANGELOG.md +132 -0
- package/LICENSE +30 -0
- package/README.id.md +633 -0
- package/README.md +628 -0
- package/cli/cli.js +388 -0
- package/cli/package.json +4 -0
- package/cli/src/cli/commands/xaiVideo.js +280 -0
- package/cli/src/cli/tray/SwayAvatar.svg +20 -0
- package/cli/src/cli/tray/autostart.js +351 -0
- package/cli/src/cli/tray/icon.ico +0 -0
- package/cli/src/cli/tray/icon.png +0 -0
- package/cli/src/cli/tray/tray.js +258 -0
- package/cli/src/cli/tray/tray.ps1 +129 -0
- package/cli/src/cli/tray/trayHost.js +157 -0
- package/cli/src/cli/tray/trayWin.js +83 -0
- package/cli/src/cli/utils/display.js +123 -0
- package/cli/src/cli/utils/input.js +138 -0
- package/cli/src/runtime/discover.js +153 -0
- package/cli/src/runtime/lifecycle.js +333 -0
- package/cli/src/runtime/runMenu.js +250 -0
- package/cli/src/runtime/uiAdapters.js +12 -0
- package/cli/src/runtime/update.js +130 -0
- package/dashboard/dist/assets/CliTools-Ch3gKuNw.js +24 -0
- package/dashboard/dist/assets/Combo-BBCldjWi.js +1 -0
- package/dashboard/dist/assets/ConsoleLog-wywAYcPT.js +6 -0
- package/dashboard/dist/assets/Dashboard-DxzyF41b.js +1 -0
- package/dashboard/dist/assets/Header-C_9Su9BF.js +1 -0
- package/dashboard/dist/assets/Keys-D0ufgcsc.js +1 -0
- package/dashboard/dist/assets/Login-BeKpipS_.js +1 -0
- package/dashboard/dist/assets/ManageApiKey-CVXsmW2B.js +1 -0
- package/dashboard/dist/assets/OAuthCallback-D_UZYcF6.js +1 -0
- package/dashboard/dist/assets/Provider-BU7JfTqm.js +4 -0
- package/dashboard/dist/assets/ProviderModelAccordion-BAifkfVR.js +1 -0
- package/dashboard/dist/assets/Proxy-CU15VJ--.js +1 -0
- package/dashboard/dist/assets/QuotaMonitor-HzGK7bcc.js +1 -0
- package/dashboard/dist/assets/Settings-Dt4y2dC0.js +1 -0
- package/dashboard/dist/assets/Setup-DSDUwU7N.js +1 -0
- package/dashboard/dist/assets/StatusBadge-1nAOyHU2.js +1 -0
- package/dashboard/dist/assets/SwayChat-Bup7up6D.js +10 -0
- package/dashboard/dist/assets/Usage-eWnrPQ_v.js +4 -0
- package/dashboard/dist/assets/accordion-DOOPDDgt.js +1 -0
- package/dashboard/dist/assets/admin-extras-api-CCdu5TFM.js +1 -0
- package/dashboard/dist/assets/api-keys-api-ifAC2Ok3.js +1 -0
- package/dashboard/dist/assets/card-CC0Omjcv.js +1 -0
- package/dashboard/dist/assets/chatStudio-C0zPz_Z_.js +2 -0
- package/dashboard/dist/assets/checkbox-7SZBrmbN.js +1 -0
- package/dashboard/dist/assets/connections-api-_N54UD3S.js +1 -0
- package/dashboard/dist/assets/copy-button-qBvM6H9z.js +1 -0
- package/dashboard/dist/assets/currency-api-SV9CBf5e.js +1 -0
- package/dashboard/dist/assets/flip-button-BGgkjtl0.js +1 -0
- package/dashboard/dist/assets/frame-cUiti6Lq.js +1 -0
- package/dashboard/dist/assets/icons-vendor-CUx7SB5V.js +541 -0
- package/dashboard/dist/assets/index-B6j-9zHy.js +10 -0
- package/dashboard/dist/assets/index-C-_l0yYz.css +1 -0
- package/dashboard/dist/assets/live-mode-CHShLY4E.js +1 -0
- package/dashboard/dist/assets/markdown-vendor-BY8e1gQ8.js +29 -0
- package/dashboard/dist/assets/motion-vendor-g5fpGzmJ.js +9 -0
- package/dashboard/dist/assets/oauth-callback-CqYQzyd7.js +1 -0
- package/dashboard/dist/assets/query-vendor-DVevaSVS.js +1 -0
- package/dashboard/dist/assets/react-vendor-xnYrwzsL.js +61 -0
- package/dashboard/dist/assets/ripple-button-l8aMiVVG.js +1 -0
- package/dashboard/dist/assets/segmented-CqmEnQjZ.js +1 -0
- package/dashboard/dist/assets/select-D8pUtMXp.js +1 -0
- package/dashboard/dist/assets/skeleton-7uJDiIXe.js +1 -0
- package/dashboard/dist/assets/table-D0M60Tu1.js +1 -0
- package/dashboard/dist/assets/tabs-C6YkuulG.js +1 -0
- package/dashboard/dist/assets/tunnel-api-DDPsBaDV.js +1 -0
- package/dashboard/dist/assets/ui-vendor-BuM9f2NY.js +17 -0
- package/dashboard/dist/image2/SwayAvatar.svg +20 -0
- package/dashboard/dist/index.html +32 -0
- package/dashboard/dist/logo/avatar.svg +1 -0
- package/dashboard/dist/logo/sway.svg +20 -0
- package/dashboard/dist/logo/swaychat.svg +20 -0
- package/dashboard/dist/providers/SwayCustom.svg +19 -0
- package/dashboard/dist/providers/alicode.svg +12 -0
- package/dashboard/dist/providers/anthropic.svg +11 -0
- package/dashboard/dist/providers/antigravity.svg +3 -0
- package/dashboard/dist/providers/azure.svg +5 -0
- package/dashboard/dist/providers/baidu.svg +3 -0
- package/dashboard/dist/providers/blackbox.svg +5 -0
- package/dashboard/dist/providers/byteplus.svg +3 -0
- package/dashboard/dist/providers/cerebras.svg +4 -0
- package/dashboard/dist/providers/chutes.svg +4 -0
- package/dashboard/dist/providers/claude.svg +3 -0
- package/dashboard/dist/providers/cline.svg +11 -0
- package/dashboard/dist/providers/cloudflare-ai.svg +4 -0
- package/dashboard/dist/providers/codebuddy.svg +11 -0
- package/dashboard/dist/providers/codex.svg +10 -0
- package/dashboard/dist/providers/commandcode.svg +3 -0
- package/dashboard/dist/providers/cursor.svg +3 -0
- package/dashboard/dist/providers/deepseek.svg +10 -0
- package/dashboard/dist/providers/featherless.svg +3 -0
- package/dashboard/dist/providers/fireworks.svg +3 -0
- package/dashboard/dist/providers/gemini.svg +3 -0
- package/dashboard/dist/providers/geminicli.svg +11 -0
- package/dashboard/dist/providers/github.svg +3 -0
- package/dashboard/dist/providers/grok.svg +3 -0
- package/dashboard/dist/providers/groq.svg +3 -0
- package/dashboard/dist/providers/hermesagent.svg +1 -0
- package/dashboard/dist/providers/kilocode.svg +15 -0
- package/dashboard/dist/providers/kimchi.svg +3 -0
- package/dashboard/dist/providers/kimi.svg +4 -0
- package/dashboard/dist/providers/kiro.svg +10 -0
- package/dashboard/dist/providers/meta.svg +3 -0
- package/dashboard/dist/providers/mimo.svg +10 -0
- package/dashboard/dist/providers/minimax.svg +3 -0
- package/dashboard/dist/providers/mistral.svg +3 -0
- package/dashboard/dist/providers/morph.svg +3 -0
- package/dashboard/dist/providers/nebius.svg +4 -0
- package/dashboard/dist/providers/nvidia.svg +3 -0
- package/dashboard/dist/providers/ollama.svg +3 -0
- package/dashboard/dist/providers/omp.svg +5 -0
- package/dashboard/dist/providers/openai.svg +3 -0
- package/dashboard/dist/providers/opencode.svg +3 -0
- package/dashboard/dist/providers/openrouter.svg +3 -0
- package/dashboard/dist/providers/perplexity.svg +3 -0
- package/dashboard/dist/providers/poolside.svg +36 -0
- package/dashboard/dist/providers/qoder.svg +11 -0
- package/dashboard/dist/providers/sumopod.svg +3 -0
- package/dashboard/dist/providers/tencent.svg +5 -0
- package/dashboard/dist/providers/together.svg +12 -0
- package/dashboard/dist/providers/trae.svg +3 -0
- package/dashboard/dist/providers/venice.svg +4 -0
- package/dashboard/dist/providers/vercel.svg +3 -0
- package/dashboard/dist/providers/vertex.svg +17 -0
- package/dashboard/dist/providers/windsurf.svg +10 -0
- package/dashboard/dist/providers/zai.svg +3 -0
- package/dashboard/tsconfig.json +13 -0
- package/package.json +87 -0
- package/src/config/appConstants.js +153 -0
- package/src/config/defaultThinkingSignature.js +7 -0
- package/src/config/errorConfig.js +117 -0
- package/src/config/grokCli.js +9 -0
- package/src/config/kiroConstants.js +235 -0
- package/src/config/mediaConfig.js +20 -0
- package/src/config/ollamaModels.js +18 -0
- package/src/config/providerModels.js +109 -0
- package/src/config/providers.js +17 -0
- package/src/config/runtimeConfig.js +84 -0
- package/src/dashboardGuard.js +254 -0
- package/src/executors/antigravity.js +389 -0
- package/src/executors/azure.js +49 -0
- package/src/executors/base.js +225 -0
- package/src/executors/codebuddy-cn.js +23 -0
- package/src/executors/codebuddy-intl.js +60 -0
- package/src/executors/codex.js +436 -0
- package/src/executors/commandcode.js +83 -0
- package/src/executors/cursor.js +1075 -0
- package/src/executors/default.js +294 -0
- package/src/executors/gemini-cli.js +88 -0
- package/src/executors/github.js +393 -0
- package/src/executors/grok-cli.js +510 -0
- package/src/executors/grok-web.js +342 -0
- package/src/executors/index.js +86 -0
- package/src/executors/kimchi.js +116 -0
- package/src/executors/kiro.js +1155 -0
- package/src/executors/ollama-local.js +14 -0
- package/src/executors/opencode-go.js +47 -0
- package/src/executors/opencode.js +31 -0
- package/src/executors/qoder.js +433 -0
- package/src/executors/trae.js +318 -0
- package/src/executors/vertex.js +151 -0
- package/src/executors/windsurf.js +555 -0
- package/src/executors/xiaomi-tokenplan.js +16 -0
- package/src/handlers/chatCore/nonStreamingHandler.js +370 -0
- package/src/handlers/chatCore/requestDetail.js +130 -0
- package/src/handlers/chatCore/sseToJsonHandler.js +318 -0
- package/src/handlers/chatCore/streamingHandler.js +250 -0
- package/src/handlers/chatCore.js +652 -0
- package/src/lib/auth/dashboardSession.js +83 -0
- package/src/lib/auth/loginLimiter.js +99 -0
- package/src/lib/cache/ttlCache.js +153 -0
- package/src/lib/consoleLogBuffer.js +127 -0
- package/src/lib/dataDir.js +45 -0
- package/src/lib/db/adapters/betterSqliteAdapter.js +36 -0
- package/src/lib/db/adapters/bunSqliteAdapter.js +47 -0
- package/src/lib/db/adapters/nodeSqliteAdapter.js +65 -0
- package/src/lib/db/adapters/runtimeHelpers.js +23 -0
- package/src/lib/db/adapters/sqljsAdapter.js +103 -0
- package/src/lib/db/backup.js +75 -0
- package/src/lib/db/driver.js +110 -0
- package/src/lib/db/helpers/jsonCol.js +18 -0
- package/src/lib/db/helpers/kvStore.js +48 -0
- package/src/lib/db/helpers/metaStore.js +32 -0
- package/src/lib/db/index.js +473 -0
- package/src/lib/db/migrate.js +267 -0
- package/src/lib/db/migrations/001-initial.js +12 -0
- package/src/lib/db/migrations/002-prune-headroom-pxpipe-settings.js +32 -0
- package/src/lib/db/migrations/004-request-details-trace-id.js +12 -0
- package/src/lib/db/migrations/005-request-details-payload-capture.js +11 -0
- package/src/lib/db/migrations/007-headroom-readd-settings.js +35 -0
- package/src/lib/db/migrations/008-account-model-locks.js +22 -0
- package/src/lib/db/migrations/009-warp-pool-lifecycle.js +5 -0
- package/src/lib/db/migrations/010-api-key-file-permissions.js +7 -0
- package/src/lib/db/migrations/011-remove-upstream-user-agent-profile.js +41 -0
- package/src/lib/db/migrations/012-remove-headroom-settings.js +33 -0
- package/src/lib/db/migrations/013-default-fill-first-routing.js +20 -0
- package/src/lib/db/migrations/014-normalize-fill-first-routing.js +37 -0
- package/src/lib/db/migrations/015-remove-ip-ban-tables.js +8 -0
- package/src/lib/db/migrations/016-prune-legacy-codebuddy-intl-models.js +25 -0
- package/src/lib/db/migrations/index.js +20 -0
- package/src/lib/db/nineRouterMigration.js +785 -0
- package/src/lib/db/paths.js +23 -0
- package/src/lib/db/repos/accountModelLocksRepo.js +74 -0
- package/src/lib/db/repos/aliasRepo.js +108 -0
- package/src/lib/db/repos/apiKeysRepo.js +204 -0
- package/src/lib/db/repos/combosRepo.js +85 -0
- package/src/lib/db/repos/connectionsRepo.js +522 -0
- package/src/lib/db/repos/disabledModelsRepo.js +63 -0
- package/src/lib/db/repos/nodesRepo.js +127 -0
- package/src/lib/db/repos/pricingRepo.js +108 -0
- package/src/lib/db/repos/proxyPoolsRepo.js +104 -0
- package/src/lib/db/repos/requestDetailsRepo.js +345 -0
- package/src/lib/db/repos/settingsRepo.js +135 -0
- package/src/lib/db/repos/usageRepo.js +1335 -0
- package/src/lib/db/schema.js +179 -0
- package/src/lib/db/version.js +32 -0
- package/src/lib/db/writeBehind.js +174 -0
- package/src/lib/disabledModelsDb.js +3 -0
- package/src/lib/env.ts +175 -0
- package/src/lib/errors/publicError.js +18 -0
- package/src/lib/grokBuildConfig.js +243 -0
- package/src/lib/guardRequest.js +51 -0
- package/src/lib/localDb.js +21 -0
- package/src/lib/mcp/stdioSseBridge.js +174 -0
- package/src/lib/network/connectionProxy.js +200 -0
- package/src/lib/network/proxyTest.js +158 -0
- package/src/lib/network/proxyUrl.js +76 -0
- package/src/lib/network/relay.js +71 -0
- package/src/lib/oauth/constants/oauth.js +152 -0
- package/src/lib/oauth/constants/xai.js +43 -0
- package/src/lib/oauth/kiroExternalIdp.js +167 -0
- package/src/lib/oauth/providerHelpers.js +90 -0
- package/src/lib/oauth/providers/_shared.js +12 -0
- package/src/lib/oauth/providers/antigravity.js +126 -0
- package/src/lib/oauth/providers/claude.js +60 -0
- package/src/lib/oauth/providers/cline.js +62 -0
- package/src/lib/oauth/providers/clinepass.js +62 -0
- package/src/lib/oauth/providers/codebuddy-cn.js +75 -0
- package/src/lib/oauth/providers/codebuddy-intl.js +73 -0
- package/src/lib/oauth/providers/codex.js +72 -0
- package/src/lib/oauth/providers/cursor.js +18 -0
- package/src/lib/oauth/providers/gemini-cli.js +84 -0
- package/src/lib/oauth/providers/github.js +96 -0
- package/src/lib/oauth/providers/grok-cli.js +123 -0
- package/src/lib/oauth/providers/index.js +233 -0
- package/src/lib/oauth/providers/kilocode.js +60 -0
- package/src/lib/oauth/providers/kimchi.js +79 -0
- package/src/lib/oauth/providers/kimi.js +79 -0
- package/src/lib/oauth/providers/kiro.js +147 -0
- package/src/lib/oauth/providers/qoder.js +90 -0
- package/src/lib/oauth/providers/trae.js +244 -0
- package/src/lib/oauth/providers/windsurf.js +123 -0
- package/src/lib/oauth/providers/xai.js +95 -0
- package/src/lib/oauth/providers.js +16 -0
- package/src/lib/oauth/services/cursor.js +132 -0
- package/src/lib/oauth/services/kiro.js +351 -0
- package/src/lib/oauth/services/oauth.js +135 -0
- package/src/lib/oauth/services/qoder.js +151 -0
- package/src/lib/oauth/services/xai.js +204 -0
- package/src/lib/oauth/utils/ideDetect.js +55 -0
- package/src/lib/oauth/utils/pkce.js +25 -0
- package/src/lib/oauth/utils/server.js +732 -0
- package/src/lib/oauth/utils/ui.js +43 -0
- package/src/lib/providerNormalization.js +40 -0
- package/src/lib/qoder/constants.js +1 -0
- package/src/lib/requestDetailsDb.js +3 -0
- package/src/lib/resourceProfile.ts +124 -0
- package/src/lib/runtimeSecrets.ts +130 -0
- package/src/lib/security/payloadCapture.js +87 -0
- package/src/lib/tunnel/cloudflare/cloudflared.js +436 -0
- package/src/lib/tunnel/cloudflare/config.js +8 -0
- package/src/lib/tunnel/cloudflare/healthCheck.js +37 -0
- package/src/lib/tunnel/cloudflare/manager.js +149 -0
- package/src/lib/tunnel/cloudflare/pid.js +31 -0
- package/src/lib/tunnel/cloudflare/publicUrl.js +41 -0
- package/src/lib/tunnel/cloudflare/requirements.js +12 -0
- package/src/lib/tunnel/index.js +26 -0
- package/src/lib/tunnel/shared/dnsResolver.js +16 -0
- package/src/lib/tunnel/shared/internetCheck.js +30 -0
- package/src/lib/tunnel/shared/state.js +45 -0
- package/src/lib/tunnel/shared/watchdogConfig.js +6 -0
- package/src/lib/update/installer.js +52 -0
- package/src/lib/update/version.js +116 -0
- package/src/lib/usageDb.js +6 -0
- package/src/lib/usageRefresh.js +88 -0
- package/src/lifecycle/appLifecycle.ts +99 -0
- package/src/models.js +36 -0
- package/src/next/headers.js +12 -0
- package/src/next/server.js +74 -0
- package/src/observability/metrics.js +439 -0
- package/src/observability/tracing.js +107 -0
- package/src/open-sse-index.js +70 -0
- package/src/providers/capabilities.js +299 -0
- package/src/providers/display.ts +142 -0
- package/src/providers/index.js +27 -0
- package/src/providers/models/helpers.js +19 -0
- package/src/providers/models/namePatterns.js +27 -0
- package/src/providers/models/schema.js +32 -0
- package/src/providers/pricing.js +248 -0
- package/src/providers/registry/agentrouter.js +52 -0
- package/src/providers/registry/alicode-intl.js +32 -0
- package/src/providers/registry/alicode.js +33 -0
- package/src/providers/registry/alims-intl.js +30 -0
- package/src/providers/registry/anthropic.js +29 -0
- package/src/providers/registry/antigravity.js +139 -0
- package/src/providers/registry/azure.js +23 -0
- package/src/providers/registry/baidu.js +35 -0
- package/src/providers/registry/blackbox.js +41 -0
- package/src/providers/registry/byteplus.js +36 -0
- package/src/providers/registry/cerebras.js +33 -0
- package/src/providers/registry/chutes.js +26 -0
- package/src/providers/registry/claude.js +80 -0
- package/src/providers/registry/cline.js +53 -0
- package/src/providers/registry/clinepass.js +62 -0
- package/src/providers/registry/cloudflare-ai.js +84 -0
- package/src/providers/registry/codebuddy-cn.js +95 -0
- package/src/providers/registry/codebuddy-intl.js +91 -0
- package/src/providers/registry/codex.js +145 -0
- package/src/providers/registry/commandcode.js +45 -0
- package/src/providers/registry/cursor.js +61 -0
- package/src/providers/registry/deepseek.js +55 -0
- package/src/providers/registry/featherless.js +36 -0
- package/src/providers/registry/fireworks.js +34 -0
- package/src/providers/registry/gemini-cli.js +59 -0
- package/src/providers/registry/gemini.js +76 -0
- package/src/providers/registry/github.js +131 -0
- package/src/providers/registry/glm-cn.js +35 -0
- package/src/providers/registry/glm.js +58 -0
- package/src/providers/registry/grok-cli.js +87 -0
- package/src/providers/registry/grok-web.js +41 -0
- package/src/providers/registry/groq.js +38 -0
- package/src/providers/registry/index.js +77 -0
- package/src/providers/registry/kilo-gateway.js +34 -0
- package/src/providers/registry/kilocode.js +48 -0
- package/src/providers/registry/kimchi.js +49 -0
- package/src/providers/registry/kimi.js +82 -0
- package/src/providers/registry/kiro.js +132 -0
- package/src/providers/registry/meta.js +34 -0
- package/src/providers/registry/minimax-cn.js +90 -0
- package/src/providers/registry/minimax.js +90 -0
- package/src/providers/registry/mistral.js +36 -0
- package/src/providers/registry/morph.js +31 -0
- package/src/providers/registry/nebius.js +26 -0
- package/src/providers/registry/nvidia.js +55 -0
- package/src/providers/registry/ollama-local.js +18 -0
- package/src/providers/registry/ollama.js +38 -0
- package/src/providers/registry/openai.js +114 -0
- package/src/providers/registry/opencode-go.js +41 -0
- package/src/providers/registry/opencode.js +26 -0
- package/src/providers/registry/openrouter.js +36 -0
- package/src/providers/registry/perplexity.js +56 -0
- package/src/providers/registry/poolside.js +30 -0
- package/src/providers/registry/qoder.js +58 -0
- package/src/providers/registry/sumopod.js +84 -0
- package/src/providers/registry/tencent.js +29 -0
- package/src/providers/registry/together.js +38 -0
- package/src/providers/registry/trae.js +71 -0
- package/src/providers/registry/venice.js +69 -0
- package/src/providers/registry/vercel-ai-gateway.js +37 -0
- package/src/providers/registry/vertex-partner.js +31 -0
- package/src/providers/registry/vertex.js +33 -0
- package/src/providers/registry/windsurf.js +129 -0
- package/src/providers/registry/xai.js +46 -0
- package/src/providers/registry/xiaomi-mimo.js +66 -0
- package/src/providers/registry/xiaomi-tokenplan.js +58 -0
- package/src/providers/registryEntry.js +3 -0
- package/src/providers/schema.js +33 -0
- package/src/providers/shared.js +114 -0
- package/src/providers/thinkingLevels.js +49 -0
- package/src/routes/api/auth/login/route.js +82 -0
- package/src/routes/api/auth/logout/route.js +9 -0
- package/src/routes/api/auth/reset-password/route.js +12 -0
- package/src/routes/api/auth/status/route.js +33 -0
- package/src/routes/api/auth/verify-password/route.js +39 -0
- package/src/routes/api/chat/tools/curl/route.js +100 -0
- package/src/routes/api/chat/tools/router/route.js +215 -0
- package/src/routes/api/chat/tools/search/route.js +362 -0
- package/src/routes/api/chat/tools/shell/route.js +228 -0
- package/src/routes/api/cli-tools/all-statuses/route.js +37 -0
- package/src/routes/api/cli-tools/claude-settings/route.js +261 -0
- package/src/routes/api/cli-tools/codex-settings/route.js +220 -0
- package/src/routes/api/cli-tools/cowork-mcp-registry/route.js +80 -0
- package/src/routes/api/cli-tools/cowork-mcp-tools/route.js +96 -0
- package/src/routes/api/cli-tools/cowork-settings/route.js +380 -0
- package/src/routes/api/cli-tools/grok-build-settings/route.js +152 -0
- package/src/routes/api/cli-tools/hermes-settings/route.js +169 -0
- package/src/routes/api/cli-tools/kilo-settings/route.js +131 -0
- package/src/routes/api/cli-tools/model-mappings/route.js +43 -0
- package/src/routes/api/cli-tools/omp-settings/route.js +266 -0
- package/src/routes/api/cli-tools/opencode-settings/route.js +243 -0
- package/src/routes/api/combos/[id]/route.js +73 -0
- package/src/routes/api/combos/route.js +43 -0
- package/src/routes/api/health/live/route.js +7 -0
- package/src/routes/api/health/ready/route.js +19 -0
- package/src/routes/api/health/route.js +17 -0
- package/src/routes/api/inflight/stream/route.js +24 -0
- package/src/routes/api/init/route.js +3 -0
- package/src/routes/api/keys/[id]/route.js +71 -0
- package/src/routes/api/keys/route.js +41 -0
- package/src/routes/api/logs/route.js +19 -0
- package/src/routes/api/mcp/[plugin]/message/route.js +23 -0
- package/src/routes/api/mcp/[plugin]/sse/route.js +35 -0
- package/src/routes/api/models/availability/route.js +104 -0
- package/src/routes/api/models/custom/route.js +58 -0
- package/src/routes/api/models/disabled/route.js +47 -0
- package/src/routes/api/models/route.js +64 -0
- package/src/routes/api/models/test/ping.js +236 -0
- package/src/routes/api/models/test/route.js +14 -0
- package/src/routes/api/oauth/[provider]/[action]/route.js +491 -0
- package/src/routes/api/oauth/codex/bulk-import/route.js +105 -0
- package/src/routes/api/oauth/codex/import-token/route.js +82 -0
- package/src/routes/api/oauth/cursor/auto-import/route.js +250 -0
- package/src/routes/api/oauth/cursor/import/route.js +86 -0
- package/src/routes/api/oauth/kiro/api-key/route.js +60 -0
- package/src/routes/api/oauth/kiro/auto-import/route.js +120 -0
- package/src/routes/api/oauth/kiro/import/route.js +72 -0
- package/src/routes/api/oauth/kiro/import-cli-proxy/route.js +42 -0
- package/src/routes/api/oauth/kiro/social-authorize/route.js +40 -0
- package/src/routes/api/oauth/kiro/social-exchange/route.js +62 -0
- package/src/routes/api/pricing/route.js +96 -0
- package/src/routes/api/provider-nodes/[id]/route.js +106 -0
- package/src/routes/api/provider-nodes/route.js +110 -0
- package/src/routes/api/provider-nodes/validate/route.js +197 -0
- package/src/routes/api/providers/[id]/models/route.js +716 -0
- package/src/routes/api/providers/[id]/route.js +195 -0
- package/src/routes/api/providers/[id]/test/route.js +27 -0
- package/src/routes/api/providers/[id]/test/testUtils.js +912 -0
- package/src/routes/api/providers/[id]/test-models/route.js +211 -0
- package/src/routes/api/providers/catalog/route.js +58 -0
- package/src/routes/api/providers/client/route.js +133 -0
- package/src/routes/api/providers/counts/route.js +43 -0
- package/src/routes/api/providers/kilo/free-models/route.js +58 -0
- package/src/routes/api/providers/route.js +361 -0
- package/src/routes/api/providers/suggested-models/filters.js +24 -0
- package/src/routes/api/providers/suggested-models/route.js +34 -0
- package/src/routes/api/providers/test-batch/route.js +145 -0
- package/src/routes/api/providers/validate/route.js +633 -0
- package/src/routes/api/proxy-pools/[id]/route.js +129 -0
- package/src/routes/api/proxy-pools/[id]/test/route.js +64 -0
- package/src/routes/api/proxy-pools/cloudflare-deploy/route.js +183 -0
- package/src/routes/api/proxy-pools/route.js +85 -0
- package/src/routes/api/proxy-pools/vercel-deploy/route.js +181 -0
- package/src/routes/api/settings/currency/route.js +29 -0
- package/src/routes/api/settings/database/path/route.js +8 -0
- package/src/routes/api/settings/database/route.js +89 -0
- package/src/routes/api/settings/migrate-9router/route.js +81 -0
- package/src/routes/api/settings/require-login/route.js +14 -0
- package/src/routes/api/settings/route.js +186 -0
- package/src/routes/api/shutdown/route.js +25 -0
- package/src/routes/api/tags/route.js +17 -0
- package/src/routes/api/translator/console-logs/route.js +28 -0
- package/src/routes/api/translator/console-logs/stream/route.js +95 -0
- package/src/routes/api/tunnel/disable/route.js +18 -0
- package/src/routes/api/tunnel/enable/route.js +32 -0
- package/src/routes/api/tunnel/status/route.js +23 -0
- package/src/routes/api/update/route.js +42 -0
- package/src/routes/api/usage/[connectionId]/codex-reset-credits/route.js +169 -0
- package/src/routes/api/usage/[connectionId]/route.js +222 -0
- package/src/routes/api/usage/chart/route.js +27 -0
- package/src/routes/api/usage/history/route.js +94 -0
- package/src/routes/api/usage/logs/route.js +14 -0
- package/src/routes/api/usage/providers/route.js +36 -0
- package/src/routes/api/usage/request-details/route.js +61 -0
- package/src/routes/api/usage/request-logs/route.js +15 -0
- package/src/routes/api/usage/stats/route.js +23 -0
- package/src/routes/api/usage/stream/route.js +78 -0
- package/src/routes/api/v1/api/chat/route.js +36 -0
- package/src/routes/api/v1/chat/completions/route.js +28 -0
- package/src/routes/api/v1/images/edits/route.js +70 -0
- package/src/routes/api/v1/images/generations/route.js +50 -0
- package/src/routes/api/v1/images/imageValidation.js +45 -0
- package/src/routes/api/v1/messages/count_tokens/route.js +87 -0
- package/src/routes/api/v1/messages/route.js +26 -0
- package/src/routes/api/v1/models/route.js +440 -0
- package/src/routes/api/v1/responses/compact/route.js +34 -0
- package/src/routes/api/v1/responses/route.js +26 -0
- package/src/routes/api/v1/route.js +1 -0
- package/src/routes/api/version/route.js +11 -0
- package/src/routes/table.js +1141 -0
- package/src/rtk/applyFilter.js +13 -0
- package/src/rtk/autodetect.js +104 -0
- package/src/rtk/caveman.js +6 -0
- package/src/rtk/cavemanPrompts.js +107 -0
- package/src/rtk/constants.js +51 -0
- package/src/rtk/filters/buildOutput.js +120 -0
- package/src/rtk/filters/dedupLog.js +43 -0
- package/src/rtk/filters/find.js +47 -0
- package/src/rtk/filters/gitDiff.js +90 -0
- package/src/rtk/filters/gitLog.js +88 -0
- package/src/rtk/filters/gitStatus.js +101 -0
- package/src/rtk/filters/grep.js +45 -0
- package/src/rtk/filters/ls.js +73 -0
- package/src/rtk/filters/readNumbered.js +22 -0
- package/src/rtk/filters/searchList.js +49 -0
- package/src/rtk/filters/smartTruncate.js +13 -0
- package/src/rtk/filters/tree.js +28 -0
- package/src/rtk/index.js +143 -0
- package/src/rtk/ponytail.js +6 -0
- package/src/rtk/ponytailPrompt.js +49 -0
- package/src/rtk/systemInject.js +89 -0
- package/src/server.ts +474 -0
- package/src/services/accountFallback.js +241 -0
- package/src/services/accountPool.js +389 -0
- package/src/services/capacityAdapter.js +140 -0
- package/src/services/cliModelMapping.js +84 -0
- package/src/services/clinepassModels.js +60 -0
- package/src/services/codebuddyModels.js +250 -0
- package/src/services/combo.js +482 -0
- package/src/services/copilotModels.js +126 -0
- package/src/services/currencyRate.js +56 -0
- package/src/services/cursorModels.js +197 -0
- package/src/services/grokCliModels.js +130 -0
- package/src/services/kimchiModels.js +175 -0
- package/src/services/kiroModels.js +246 -0
- package/src/services/model.js +83 -0
- package/src/services/oauthCredentialManager.js +153 -0
- package/src/services/projectId.js +240 -0
- package/src/services/provider.js +145 -0
- package/src/services/qoderModels.js +274 -0
- package/src/services/tokenRefresh/dedup.js +70 -0
- package/src/services/tokenRefresh/providers.js +626 -0
- package/src/services/tokenRefresh.js +274 -0
- package/src/services/usage/claude.js +134 -0
- package/src/services/usage/codebuddy-cn.js +212 -0
- package/src/services/usage/codex.js +193 -0
- package/src/services/usage/deepseek.js +103 -0
- package/src/services/usage/github.js +94 -0
- package/src/services/usage/google.js +218 -0
- package/src/services/usage/grok-cli.js +368 -0
- package/src/services/usage/grokCliQuotaFrame.js +170 -0
- package/src/services/usage/kimi.js +185 -0
- package/src/services/usage/kiro.js +172 -0
- package/src/services/usage/minimax.js +215 -0
- package/src/services/usage/misc.js +264 -0
- package/src/services/usage/shared.js +59 -0
- package/src/services/usage.js +56 -0
- package/src/shared/clineAuth.js +37 -0
- package/src/shared/constants/config.js +80 -0
- package/src/shared/constants/coworkPlugins.js +69 -0
- package/src/shared/constants/models.js +41 -0
- package/src/shared/constants/providers.js +128 -0
- package/src/shared/constants/providersDisplay.js +1 -0
- package/src/shared/qoder/constants.js +50 -0
- package/src/shared/qoder/cosy.js +134 -0
- package/src/shared/qoder/encoding.js +36 -0
- package/src/shared/services/initializeApp.js +226 -0
- package/src/shared/services/quotaAutoPing.js +310 -0
- package/src/shared/utils/apiKey.js +65 -0
- package/src/shared/utils/clineAuth.js +1 -0
- package/src/shared/utils/index.js +3 -0
- package/src/shared/utils/machineId.js +60 -0
- package/src/shared/utils/ssrfGuard.js +280 -0
- package/src/sse/handlers/chat.js +426 -0
- package/src/sse/oauthLease.js +32 -0
- package/src/sse/recovery.js +235 -0
- package/src/sse/services/auth.js +489 -0
- package/src/sse/services/backgroundTokenRefresh.js +173 -0
- package/src/sse/services/model.js +129 -0
- package/src/sse/services/requestCancellation.js +4 -0
- package/src/sse/services/tokenRefresh.js +264 -0
- package/src/sse/utils/logger.js +142 -0
- package/src/staticServe.ts +95 -0
- package/src/transformer/streamToJsonConverter.js +87 -0
- package/src/translator/canonical.js +81 -0
- package/src/translator/concerns/chunk.js +9 -0
- package/src/translator/concerns/compatFallback.js +147 -0
- package/src/translator/concerns/finishReason.js +59 -0
- package/src/translator/concerns/image.js +105 -0
- package/src/translator/concerns/json.js +4 -0
- package/src/translator/concerns/kiroConversation.js +428 -0
- package/src/translator/concerns/message.js +5 -0
- package/src/translator/concerns/modality.js +134 -0
- package/src/translator/concerns/paramSupport.js +97 -0
- package/src/translator/concerns/prefetch.js +86 -0
- package/src/translator/concerns/reasoning.js +18 -0
- package/src/translator/concerns/thinking.js +40 -0
- package/src/translator/concerns/thinkingUnified.js +319 -0
- package/src/translator/concerns/toolCall.js +134 -0
- package/src/translator/concerns/usage.js +61 -0
- package/src/translator/detection.js +42 -0
- package/src/translator/formats/claude.js +323 -0
- package/src/translator/formats/gemini.js +366 -0
- package/src/translator/formats/maxTokens.js +19 -0
- package/src/translator/formats/openai.js +116 -0
- package/src/translator/formats/responsesApi.js +117 -0
- package/src/translator/formats.js +28 -0
- package/src/translator/index.js +237 -0
- package/src/translator/request/antigravity-to-openai.js +217 -0
- package/src/translator/request/claude-to-kiro.js +288 -0
- package/src/translator/request/claude-to-openai.js +199 -0
- package/src/translator/request/gemini-to-openai.js +142 -0
- package/src/translator/request/openai-responses.js +373 -0
- package/src/translator/request/openai-to-claude.js +260 -0
- package/src/translator/request/openai-to-commandcode.js +161 -0
- package/src/translator/request/openai-to-cursor.js +175 -0
- package/src/translator/request/openai-to-gemini.js +422 -0
- package/src/translator/request/openai-to-kiro.js +360 -0
- package/src/translator/request/openai-to-ollama.js +152 -0
- package/src/translator/request/openai-to-vertex.js +36 -0
- package/src/translator/response/claude-to-openai.js +185 -0
- package/src/translator/response/commandcode-to-openai.js +164 -0
- package/src/translator/response/cursor-to-openai.js +18 -0
- package/src/translator/response/gemini-to-openai.js +127 -0
- package/src/translator/response/kiro-to-claude.js +230 -0
- package/src/translator/response/kiro-to-openai.js +136 -0
- package/src/translator/response/ollama-to-openai.js +106 -0
- package/src/translator/response/openai-responses.js +534 -0
- package/src/translator/response/openai-to-antigravity.js +107 -0
- package/src/translator/response/openai-to-claude.js +231 -0
- package/src/translator/schema/blocks.js +40 -0
- package/src/translator/schema/defaults.js +3 -0
- package/src/translator/schema/finishReasons.js +22 -0
- package/src/translator/schema/index.js +7 -0
- package/src/translator/schema/roles.js +12 -0
- package/src/transport/requestBoundary.ts +223 -0
- package/src/transport/routerAdmission.ts +380 -0
- package/src/types/provider.ts +59 -0
- package/src/utils/bypassHandler.js +272 -0
- package/src/utils/claudeCloaking.js +10 -0
- package/src/utils/claudeSignature.js +35 -0
- package/src/utils/clientDetector.js +183 -0
- package/src/utils/cursorChecksum.js +104 -0
- package/src/utils/cursorProtobuf.js +801 -0
- package/src/utils/debugLog.js +12 -0
- package/src/utils/error.js +120 -0
- package/src/utils/inflight.js +127 -0
- package/src/utils/kiroSessionReplay.js +139 -0
- package/src/utils/ollamaTransform.js +83 -0
- package/src/utils/proxyFetch.js +308 -0
- package/src/utils/reasoningContentInjector.js +71 -0
- package/src/utils/requestLogger.js +272 -0
- package/src/utils/responsesStreamHelpers.js +46 -0
- package/src/utils/sessionManager.js +205 -0
- package/src/utils/sse.js +13 -0
- package/src/utils/sseConstants.js +21 -0
- package/src/utils/stream.js +511 -0
- package/src/utils/streamHandler.js +256 -0
- package/src/utils/streamHelpers.js +113 -0
- package/src/utils/toolDeduper.js +44 -0
- package/src/utils/usageTracking.js +329 -0
- package/tsconfig.json +45 -0
|
@@ -0,0 +1,489 @@
|
|
|
1
|
+
import { getProviderConnectionById, getProviderConnectionsForRouting, validateApiKey, updateProviderConnection, getSettings, getProxyPools } from "@/lib/localDb";
|
|
2
|
+
import { resolveConnectionProxyConfig, pickProxyPoolId } from "@/lib/network/connectionProxy";
|
|
3
|
+
import {
|
|
4
|
+
formatRetryAfter, checkFallbackError, isModelLockActive, buildModelLockUpdate,
|
|
5
|
+
getEarliestModelLockUntil,
|
|
6
|
+
|
|
7
|
+
isDeprioritized, sortWithDeprioritization, pickCacheAffine,
|
|
8
|
+
DEPRIORITIZE_FIELD,
|
|
9
|
+
} from "../../services/accountFallback.js";
|
|
10
|
+
import { MAX_RATE_LIMIT_COOLDOWN_MS, classifyError, ERROR_TIERS } from "../../config/errorConfig.js";
|
|
11
|
+
import { HTTP_STATUS } from "../../config/runtimeConfig.js";
|
|
12
|
+
import { recordLock, clearLock, clearAllLocksForConnection } from "@/lib/db/repos/accountModelLocksRepo.js";
|
|
13
|
+
import { accountFallbackDecisionsTotal, cacheAffineSelectionsTotal } from "@/observability/metrics.js";
|
|
14
|
+
import { resolveProviderId, NO_AUTH_PROVIDERS } from "@/providers/display.js";
|
|
15
|
+
import {
|
|
16
|
+
getCachedProviderConnections,
|
|
17
|
+
invalidateProviderAccountPool,
|
|
18
|
+
pickLeastInflightConnection,
|
|
19
|
+
rankConnectionsByInflight,
|
|
20
|
+
resolveAccountPoolConfig,
|
|
21
|
+
tryAcquireAccountSlot,
|
|
22
|
+
waitForAccountCapacity,
|
|
23
|
+
} from "../../services/accountPool.js";
|
|
24
|
+
import * as log from "../utils/logger.js";
|
|
25
|
+
|
|
26
|
+
function deriveCacheKey(options) {
|
|
27
|
+
const body = options?.cacheKey ?? options?.promptPrefix;
|
|
28
|
+
if (typeof body === "string" && body.length > 0) return body.slice(0, 256);
|
|
29
|
+
return null;
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
let selectionMutex = Promise.resolve();
|
|
33
|
+
|
|
34
|
+
const GITHUB_MONTHLY_USAGE_LIMIT = "you've reached your additional usage limit for your plan";
|
|
35
|
+
|
|
36
|
+
function githubMonthlyResetMs(status, errorText, provider) {
|
|
37
|
+
if (resolveProviderId(provider) !== "github" || Number(status) !== 402) return null;
|
|
38
|
+
if (!String(errorText || "").toLowerCase().includes(GITHUB_MONTHLY_USAGE_LIMIT)) return null;
|
|
39
|
+
const now = new Date();
|
|
40
|
+
return Date.UTC(now.getUTCFullYear(), now.getUTCMonth() + 1, 1);
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
export async function getProviderCredentials(provider, excludeConnectionIds = null, model = null, options = {}) {
|
|
44
|
+
|
|
45
|
+
const excludeSet = excludeConnectionIds instanceof Set
|
|
46
|
+
? excludeConnectionIds
|
|
47
|
+
: (excludeConnectionIds ? new Set([excludeConnectionIds]) : new Set());
|
|
48
|
+
const preferredConnectionId = options?.preferredConnectionId || null;
|
|
49
|
+
|
|
50
|
+
const currentMutex = selectionMutex;
|
|
51
|
+
let resolveMutex;
|
|
52
|
+
selectionMutex = new Promise(resolve => { resolveMutex = resolve; });
|
|
53
|
+
let mutexReleased = false;
|
|
54
|
+
const releaseSelectionMutex = () => {
|
|
55
|
+
if (mutexReleased) return;
|
|
56
|
+
mutexReleased = true;
|
|
57
|
+
resolveMutex?.();
|
|
58
|
+
};
|
|
59
|
+
|
|
60
|
+
try {
|
|
61
|
+
await currentMutex;
|
|
62
|
+
|
|
63
|
+
const providerId = resolveProviderId(provider);
|
|
64
|
+
|
|
65
|
+
if (NO_AUTH_PROVIDERS[providerId]?.noAuth) {
|
|
66
|
+
const settings = await getSettings();
|
|
67
|
+
const override = (settings.providerStrategies || {})[providerId] || {};
|
|
68
|
+
const strategy = override.rotateStrategy || "none";
|
|
69
|
+
let pickedId = override.proxyPoolId || null;
|
|
70
|
+
if (strategy !== "none") {
|
|
71
|
+
const allPools = await getProxyPools({ isActive: true });
|
|
72
|
+
const poolIds = allPools.filter(p => p.proxyUrl).map(p => p.id);
|
|
73
|
+
pickedId = pickProxyPoolId(poolIds, strategy, providerId);
|
|
74
|
+
}
|
|
75
|
+
const resolvedProxy = await resolveConnectionProxyConfig({ proxyPoolId: pickedId || "" });
|
|
76
|
+
return {
|
|
77
|
+
id: "noauth",
|
|
78
|
+
connectionName: "Public",
|
|
79
|
+
isActive: true,
|
|
80
|
+
accessToken: "public",
|
|
81
|
+
providerSpecificData: {
|
|
82
|
+
connectionProxyEnabled: resolvedProxy.connectionProxyEnabled,
|
|
83
|
+
connectionProxyUrl: resolvedProxy.connectionProxyUrl,
|
|
84
|
+
connectionNoProxy: resolvedProxy.connectionNoProxy,
|
|
85
|
+
connectionProxyPoolId: resolvedProxy.proxyPoolId || null,
|
|
86
|
+
proxyPoolUnavailable: resolvedProxy.proxyPoolUnavailable === true,
|
|
87
|
+
proxyPoolError: resolvedProxy.proxyPoolError || "",
|
|
88
|
+
relayUrl: resolvedProxy.relayUrl || "",
|
|
89
|
+
relayToken: resolvedProxy.relayToken || "",
|
|
90
|
+
vercelRelayUrl: resolvedProxy.vercelRelayUrl || "",
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
|
|
95
|
+
const settings = await getSettings();
|
|
96
|
+
|
|
97
|
+
const providerOverride = (settings.providerStrategies || {})[providerId] || {};
|
|
98
|
+
const strategy = providerOverride.fallbackStrategy || settings.fallbackStrategy || "fill-first";
|
|
99
|
+
const poolConfig = resolveAccountPoolConfig(settings, providerOverride);
|
|
100
|
+
|
|
101
|
+
const connections = strategy === "round-robin" || strategy === "cache-affine"
|
|
102
|
+
? await getProviderConnectionsForRouting({ provider: providerId, isActive: true })
|
|
103
|
+
: await getCachedProviderConnections(
|
|
104
|
+
providerId,
|
|
105
|
+
() => getProviderConnectionsForRouting({ provider: providerId, isActive: true }),
|
|
106
|
+
poolConfig.cacheTtlMs,
|
|
107
|
+
);
|
|
108
|
+
log.debug("AUTH", `${provider} | total connections: ${connections.length}, excludeIds: ${excludeSet.size > 0 ? [...excludeSet].join(",") : "none"}, model: ${model || "any"}`);
|
|
109
|
+
|
|
110
|
+
if (connections.length === 0) {
|
|
111
|
+
log.warn("AUTH", `No credentials for ${provider}`);
|
|
112
|
+
return null;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
const availableConnections = connections.filter(c => {
|
|
116
|
+
if (excludeSet.has(c.id)) return false;
|
|
117
|
+
if (isModelLockActive(c, model)) return false;
|
|
118
|
+
return true;
|
|
119
|
+
});
|
|
120
|
+
|
|
121
|
+
log.debug("AUTH", `${provider} | available: ${availableConnections.length}/${connections.length}`);
|
|
122
|
+
connections.forEach(c => {
|
|
123
|
+
const excluded = excludeSet.has(c.id);
|
|
124
|
+
const locked = isModelLockActive(c, model);
|
|
125
|
+
if (excluded || locked) {
|
|
126
|
+
const lockUntil = getEarliestModelLockUntil(c);
|
|
127
|
+
log.debug("AUTH", ` → ${c.id?.slice(0, 8)} | ${excluded ? "excluded" : ""} ${locked ? `modelLocked(${model}) until ${lockUntil}` : ""}`);
|
|
128
|
+
}
|
|
129
|
+
});
|
|
130
|
+
|
|
131
|
+
if (availableConnections.length === 0) {
|
|
132
|
+
|
|
133
|
+
const lockedConns = connections.filter(c => isModelLockActive(c, model));
|
|
134
|
+
const expiries = lockedConns.map(c => getEarliestModelLockUntil(c)).filter(Boolean);
|
|
135
|
+
const earliest = expiries.sort()[0] || null;
|
|
136
|
+
if (earliest) {
|
|
137
|
+
const earliestConn = lockedConns[0];
|
|
138
|
+
log.warn("AUTH", `${provider} | all ${connections.length} accounts locked for ${model || "all"} (${formatRetryAfter(earliest)}) | lastError=${earliestConn?.lastError?.slice(0, 50)}`);
|
|
139
|
+
return {
|
|
140
|
+
allRateLimited: true,
|
|
141
|
+
retryAfter: earliest,
|
|
142
|
+
retryAfterHuman: formatRetryAfter(earliest),
|
|
143
|
+
lastError: earliestConn?.lastError || null,
|
|
144
|
+
lastErrorCode: earliestConn?.errorCode || null
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
log.warn("AUTH", `${provider} | all ${connections.length} accounts unavailable`);
|
|
148
|
+
return null;
|
|
149
|
+
}
|
|
150
|
+
|
|
151
|
+
const rankedConnections = sortWithDeprioritization(availableConnections);
|
|
152
|
+
|
|
153
|
+
let connection;
|
|
154
|
+
|
|
155
|
+
if (preferredConnectionId) {
|
|
156
|
+
connection = rankedConnections.find((c) => c.id === preferredConnectionId);
|
|
157
|
+
if (connection) {
|
|
158
|
+
log.info("AUTH", `${provider} | pinned to ${connection.id?.slice(0, 8)} (${connection.name || connection.email || "unnamed"})`);
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
if (connection && preferredConnectionId) {
|
|
162
|
+
|
|
163
|
+
} else if (strategy === "cache-affine") {
|
|
164
|
+
|
|
165
|
+
const cacheKey = deriveCacheKey(options);
|
|
166
|
+
const { connection: picked, affinity } = pickCacheAffine(rankedConnections, {
|
|
167
|
+
cacheKey,
|
|
168
|
+
excludeIds: excludeSet,
|
|
169
|
+
});
|
|
170
|
+
connection = picked || null;
|
|
171
|
+
try { cacheAffineSelectionsTotal.inc({ result: affinity || "miss" }); } catch {}
|
|
172
|
+
if (connection && cacheKey) {
|
|
173
|
+
log.debug("AUTH", `${provider} | cache-affine(${affinity}) → ${connection.id?.slice(0, 8)}`);
|
|
174
|
+
await updateProviderConnection(connection.id, {
|
|
175
|
+
lastUsedAt: new Date().toISOString(),
|
|
176
|
+
consecutiveUseCount: (connection.consecutiveUseCount || 0) + 1
|
|
177
|
+
});
|
|
178
|
+
}
|
|
179
|
+
} else if (strategy === "round-robin") {
|
|
180
|
+
const stickyLimit = providerOverride.stickyRoundRobinLimit || settings.stickyRoundRobinLimit || 1;
|
|
181
|
+
|
|
182
|
+
const byRecency = [...rankedConnections].sort((a, b) => {
|
|
183
|
+
if (!a.lastUsedAt && !b.lastUsedAt) return (a.priority || 999) - (b.priority || 999);
|
|
184
|
+
if (!a.lastUsedAt) return 1;
|
|
185
|
+
if (!b.lastUsedAt) return -1;
|
|
186
|
+
return new Date(b.lastUsedAt) - new Date(a.lastUsedAt);
|
|
187
|
+
});
|
|
188
|
+
|
|
189
|
+
const current = byRecency[0];
|
|
190
|
+
const currentCount = current?.consecutiveUseCount || 0;
|
|
191
|
+
|
|
192
|
+
if (current && current.lastUsedAt && currentCount < stickyLimit) {
|
|
193
|
+
|
|
194
|
+
connection = current;
|
|
195
|
+
|
|
196
|
+
await updateProviderConnection(connection.id, {
|
|
197
|
+
lastUsedAt: new Date().toISOString(),
|
|
198
|
+
consecutiveUseCount: (connection.consecutiveUseCount || 0) + 1
|
|
199
|
+
});
|
|
200
|
+
} else {
|
|
201
|
+
|
|
202
|
+
const sortedByOldest = [...rankedConnections].sort((a, b) => {
|
|
203
|
+
if (!a.lastUsedAt && !b.lastUsedAt) return (a.priority || 999) - (b.priority || 999);
|
|
204
|
+
if (!a.lastUsedAt) return -1;
|
|
205
|
+
if (!b.lastUsedAt) return 1;
|
|
206
|
+
return new Date(a.lastUsedAt) - new Date(b.lastUsedAt);
|
|
207
|
+
});
|
|
208
|
+
|
|
209
|
+
connection = sortedByOldest[0];
|
|
210
|
+
|
|
211
|
+
await updateProviderConnection(connection.id, {
|
|
212
|
+
lastUsedAt: new Date().toISOString(),
|
|
213
|
+
consecutiveUseCount: 1
|
|
214
|
+
});
|
|
215
|
+
}
|
|
216
|
+
} else if (strategy === "least-inflight") {
|
|
217
|
+
|
|
218
|
+
connection = pickLeastInflightConnection(rankedConnections);
|
|
219
|
+
} else {
|
|
220
|
+
|
|
221
|
+
connection = rankedConnections[0];
|
|
222
|
+
}
|
|
223
|
+
|
|
224
|
+
let candidateConnections = preferredConnectionId && connection
|
|
225
|
+
? [connection]
|
|
226
|
+
: [connection, ...(strategy === "least-inflight" ? [] : rankedConnections)]
|
|
227
|
+
.filter(Boolean)
|
|
228
|
+
.filter((candidate, index, list) => list.findIndex((item) => item.id === candidate.id) === index);
|
|
229
|
+
const candidateIds = availableConnections.map((candidate) => candidate.id);
|
|
230
|
+
const waitCandidateIds = preferredConnectionId && connection ? [connection.id] : candidateIds;
|
|
231
|
+
|
|
232
|
+
let selectedConnection = null;
|
|
233
|
+
let releaseAccountSlot = null;
|
|
234
|
+
let sawStaleConnection = false;
|
|
235
|
+
let expandedLeastInflightFallback = strategy !== "least-inflight" || Boolean(preferredConnectionId);
|
|
236
|
+
let candidateIndex = 0;
|
|
237
|
+
while (candidateIndex < candidateConnections.length) {
|
|
238
|
+
const candidate = candidateConnections[candidateIndex];
|
|
239
|
+
const release = tryAcquireAccountSlot(providerId, candidate.id, poolConfig);
|
|
240
|
+
if (release) {
|
|
241
|
+
|
|
242
|
+
let liveConnection;
|
|
243
|
+
try {
|
|
244
|
+
liveConnection = await getProviderConnectionById(candidate.id);
|
|
245
|
+
} catch (error) {
|
|
246
|
+
release();
|
|
247
|
+
throw error;
|
|
248
|
+
}
|
|
249
|
+
if (!liveConnection || liveConnection.provider !== providerId || liveConnection.isActive !== true || isModelLockActive(liveConnection, model)) {
|
|
250
|
+
sawStaleConnection = true;
|
|
251
|
+
release();
|
|
252
|
+
} else {
|
|
253
|
+
selectedConnection = liveConnection;
|
|
254
|
+
releaseAccountSlot = release;
|
|
255
|
+
break;
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
|
|
259
|
+
candidateIndex += 1;
|
|
260
|
+
|
|
261
|
+
if (
|
|
262
|
+
strategy === "least-inflight"
|
|
263
|
+
&& candidateIndex >= candidateConnections.length
|
|
264
|
+
&& !expandedLeastInflightFallback
|
|
265
|
+
) {
|
|
266
|
+
expandedLeastInflightFallback = true;
|
|
267
|
+
const selectedId = candidateConnections[0]?.id;
|
|
268
|
+
candidateConnections = [
|
|
269
|
+
...candidateConnections,
|
|
270
|
+
...rankConnectionsByInflight(rankedConnections).filter((item) => item.id !== selectedId),
|
|
271
|
+
];
|
|
272
|
+
}
|
|
273
|
+
}
|
|
274
|
+
|
|
275
|
+
if (!selectedConnection) {
|
|
276
|
+
|
|
277
|
+
if (sawStaleConnection && !options._accountPoolRefreshed) {
|
|
278
|
+
invalidateProviderAccountPool(providerId);
|
|
279
|
+
releaseSelectionMutex();
|
|
280
|
+
return getProviderCredentials(provider, excludeSet, model, {
|
|
281
|
+
...options,
|
|
282
|
+
_accountPoolRefreshed: true,
|
|
283
|
+
});
|
|
284
|
+
}
|
|
285
|
+
|
|
286
|
+
releaseSelectionMutex();
|
|
287
|
+
const queueDeadlineAt = Number(options._accountQueueDeadlineAt) || (Date.now() + poolConfig.queueWaitTimeoutMs);
|
|
288
|
+
const remainingWaitMs = Math.max(0, queueDeadlineAt - Date.now());
|
|
289
|
+
if (poolConfig.enabled && candidateIds.length > 0 && remainingWaitMs > 0) {
|
|
290
|
+
const waitResult = await waitForAccountCapacity(
|
|
291
|
+
providerId,
|
|
292
|
+
preferredConnectionId ? waitCandidateIds : null,
|
|
293
|
+
{ ...poolConfig, queueWaitTimeoutMs: remainingWaitMs },
|
|
294
|
+
options.signal || null,
|
|
295
|
+
);
|
|
296
|
+
if (waitResult.ok) {
|
|
297
|
+
return getProviderCredentials(provider, excludeSet, model, {
|
|
298
|
+
...options,
|
|
299
|
+
_accountQueueDeadlineAt: queueDeadlineAt,
|
|
300
|
+
});
|
|
301
|
+
}
|
|
302
|
+
if (waitResult.reason === "aborted") {
|
|
303
|
+
return { allBusy: true, reason: "aborted" };
|
|
304
|
+
}
|
|
305
|
+
}
|
|
306
|
+
|
|
307
|
+
const retryAfter = new Date(Date.now() + 1000).toISOString();
|
|
308
|
+
log.warn("AUTH", `${provider} | account pool busy (${candidateIds.length} candidates, queue=${poolConfig.maxQueuePerProvider})`);
|
|
309
|
+
return {
|
|
310
|
+
allBusy: true,
|
|
311
|
+
retryAfter,
|
|
312
|
+
retryAfterHuman: "account pool busy; retry shortly",
|
|
313
|
+
lastErrorCode: HTTP_STATUS.SERVICE_UNAVAILABLE,
|
|
314
|
+
};
|
|
315
|
+
}
|
|
316
|
+
|
|
317
|
+
releaseSelectionMutex();
|
|
318
|
+
connection = selectedConnection;
|
|
319
|
+
let resolvedProxy;
|
|
320
|
+
try {
|
|
321
|
+
resolvedProxy = await resolveConnectionProxyConfig(connection.providerSpecificData || {});
|
|
322
|
+
} catch (error) {
|
|
323
|
+
releaseAccountSlot?.();
|
|
324
|
+
throw error;
|
|
325
|
+
}
|
|
326
|
+
|
|
327
|
+
return {
|
|
328
|
+
authType: connection.authType,
|
|
329
|
+
apiKey: connection.apiKey,
|
|
330
|
+
accessToken: connection.accessToken,
|
|
331
|
+
refreshToken: connection.refreshToken,
|
|
332
|
+
idToken: connection.idToken,
|
|
333
|
+
expiresAt: connection.expiresAt,
|
|
334
|
+
expiresIn: connection.expiresIn,
|
|
335
|
+
lastRefreshAt: connection.lastRefreshAt,
|
|
336
|
+
projectId: connection.projectId,
|
|
337
|
+
connectionName: connection.displayName || connection.name || connection.email || connection.id,
|
|
338
|
+
copilotToken: connection.providerSpecificData?.copilotToken,
|
|
339
|
+
providerSpecificData: {
|
|
340
|
+
...(connection.providerSpecificData || {}),
|
|
341
|
+
connectionProxyEnabled: resolvedProxy.connectionProxyEnabled,
|
|
342
|
+
connectionProxyUrl: resolvedProxy.connectionProxyUrl,
|
|
343
|
+
connectionNoProxy: resolvedProxy.connectionNoProxy,
|
|
344
|
+
connectionProxyPoolId: resolvedProxy.proxyPoolId || null,
|
|
345
|
+
strictProxy: resolvedProxy.strictProxy === true,
|
|
346
|
+
proxyPoolUnavailable: resolvedProxy.proxyPoolUnavailable === true,
|
|
347
|
+
proxyPoolError: resolvedProxy.proxyPoolError || "",
|
|
348
|
+
relayUrl: resolvedProxy.relayUrl || "",
|
|
349
|
+
relayToken: resolvedProxy.relayToken || "",
|
|
350
|
+
vercelRelayUrl: resolvedProxy.vercelRelayUrl || "",
|
|
351
|
+
},
|
|
352
|
+
connectionId: connection.id,
|
|
353
|
+
|
|
354
|
+
releaseAccountSlot,
|
|
355
|
+
|
|
356
|
+
testStatus: connection.testStatus,
|
|
357
|
+
lastError: connection.lastError,
|
|
358
|
+
|
|
359
|
+
_connection: connection
|
|
360
|
+
};
|
|
361
|
+
} finally {
|
|
362
|
+
releaseSelectionMutex();
|
|
363
|
+
}
|
|
364
|
+
}
|
|
365
|
+
|
|
366
|
+
export async function markAccountUnavailable(connectionId, status, errorText, provider = null, model = null, resetsAtMs = null, opts = {}) {
|
|
367
|
+
if (!connectionId || connectionId === "noauth") return { shouldFallback: false, cooldownMs: 0 };
|
|
368
|
+
const conn = await getProviderConnectionById(connectionId);
|
|
369
|
+
const backoffLevel = conn?.backoffLevel || 0;
|
|
370
|
+
|
|
371
|
+
const githubResetAtMs = githubMonthlyResetMs(status, errorText, provider);
|
|
372
|
+
|
|
373
|
+
let shouldFallback = true;
|
|
374
|
+
let cooldownMs = 0;
|
|
375
|
+
let tier = ERROR_TIERS.T2;
|
|
376
|
+
let action = "retry-same";
|
|
377
|
+
let deprioitizeUntilMs = null;
|
|
378
|
+
if (githubResetAtMs) {
|
|
379
|
+
cooldownMs = githubResetAtMs - Date.now();
|
|
380
|
+
tier = ERROR_TIERS.T1;
|
|
381
|
+
action = "cooldown";
|
|
382
|
+
} else if (resetsAtMs && resetsAtMs > Date.now()) {
|
|
383
|
+
cooldownMs = Math.min(resetsAtMs - Date.now(), MAX_RATE_LIMIT_COOLDOWN_MS);
|
|
384
|
+
tier = ERROR_TIERS.T1;
|
|
385
|
+
action = "cooldown";
|
|
386
|
+
} else {
|
|
387
|
+
|
|
388
|
+
const retryCount = Number.isFinite(opts.retryCount) ? opts.retryCount : backoffLevel;
|
|
389
|
+
const decision = classifyError(status, errorText, retryCount);
|
|
390
|
+
({ tier, action, cooldownMs, deprioitizeUntil: deprioitizeUntilMs } = decision);
|
|
391
|
+
shouldFallback = action !== "retry-same" || cooldownMs > 0;
|
|
392
|
+
}
|
|
393
|
+
if (!shouldFallback) return { shouldFallback: false, cooldownMs: 0 };
|
|
394
|
+
|
|
395
|
+
const reason = typeof errorText === "string" ? errorText.slice(0, 100) : "Provider error";
|
|
396
|
+
const lockUpdate = buildModelLockUpdate(githubResetAtMs ? null : model, cooldownMs);
|
|
397
|
+
|
|
398
|
+
const deprioUpdate = deprioitizeUntilMs ? { [DEPRIORITIZE_FIELD]: deprioitizeUntilMs } : {};
|
|
399
|
+
|
|
400
|
+
await updateProviderConnection(connectionId, {
|
|
401
|
+
...lockUpdate,
|
|
402
|
+
...deprioUpdate,
|
|
403
|
+
testStatus: "unavailable",
|
|
404
|
+
lastError: reason,
|
|
405
|
+
errorCode: status,
|
|
406
|
+
lastErrorAt: new Date().toISOString(),
|
|
407
|
+
backoffLevel: backoffLevel + 1
|
|
408
|
+
});
|
|
409
|
+
invalidateProviderAccountPool(provider);
|
|
410
|
+
|
|
411
|
+
await recordLock(connectionId, githubResetAtMs ? null : model, {
|
|
412
|
+
tier, reason, expiresAt: new Date(Date.now() + cooldownMs).toISOString(),
|
|
413
|
+
}).catch(() => {});
|
|
414
|
+
|
|
415
|
+
try { accountFallbackDecisionsTotal.inc({ tier, action }); } catch {}
|
|
416
|
+
|
|
417
|
+
const lockKey = Object.keys(lockUpdate)[0];
|
|
418
|
+
const connName = conn?.displayName || conn?.name || conn?.email || connectionId.slice(0, 8);
|
|
419
|
+
log.warn("AUTH", `${connName} locked ${lockKey} for ${Math.round(cooldownMs / 1000)}s [${status}][${tier}]`);
|
|
420
|
+
|
|
421
|
+
if (provider && status && reason) {
|
|
422
|
+
console.error(`[${provider}] ERROR ${status}: ${reason}`);
|
|
423
|
+
}
|
|
424
|
+
|
|
425
|
+
return { shouldFallback: true, cooldownMs };
|
|
426
|
+
}
|
|
427
|
+
|
|
428
|
+
export async function clearAccountError(connectionId, currentConnection, model = null) {
|
|
429
|
+
if (!connectionId || connectionId === "noauth") return;
|
|
430
|
+
const conn = currentConnection._connection || currentConnection;
|
|
431
|
+
const now = Date.now();
|
|
432
|
+
const allLockKeys = Object.keys(conn).filter(k => k.startsWith("modelLock_"));
|
|
433
|
+
|
|
434
|
+
if (!conn.testStatus && !conn.lastError && allLockKeys.length === 0 && !conn[DEPRIORITIZE_FIELD]) return;
|
|
435
|
+
|
|
436
|
+
const keysToClear = allLockKeys.filter(k => {
|
|
437
|
+
if (model && k === `modelLock_${model}`) return true;
|
|
438
|
+
if (model && k === "modelLock___all") return true;
|
|
439
|
+
const expiry = conn[k];
|
|
440
|
+
return expiry && new Date(expiry).getTime() <= now;
|
|
441
|
+
});
|
|
442
|
+
|
|
443
|
+
await clearLock(connectionId, model).catch(() => {});
|
|
444
|
+
|
|
445
|
+
if (keysToClear.length === 0 && conn.testStatus !== "unavailable" && !conn.lastError && !conn[DEPRIORITIZE_FIELD]) return;
|
|
446
|
+
|
|
447
|
+
const remainingActiveLocks = allLockKeys.filter(k => {
|
|
448
|
+
if (keysToClear.includes(k)) return false;
|
|
449
|
+
const expiry = conn[k];
|
|
450
|
+
return expiry && new Date(expiry).getTime() > now;
|
|
451
|
+
});
|
|
452
|
+
|
|
453
|
+
const clearObj = Object.fromEntries(keysToClear.map(k => [k, null]));
|
|
454
|
+
|
|
455
|
+
if (remainingActiveLocks.length === 0) {
|
|
456
|
+
Object.assign(clearObj, {
|
|
457
|
+
testStatus: "active",
|
|
458
|
+
lastError: null,
|
|
459
|
+
errorCode: null,
|
|
460
|
+
lastErrorAt: null,
|
|
461
|
+
backoffLevel: 0,
|
|
462
|
+
|
|
463
|
+
[DEPRIORITIZE_FIELD]: null,
|
|
464
|
+
});
|
|
465
|
+
}
|
|
466
|
+
|
|
467
|
+
await updateProviderConnection(connectionId, clearObj);
|
|
468
|
+
invalidateProviderAccountPool(conn.provider);
|
|
469
|
+
}
|
|
470
|
+
|
|
471
|
+
export function extractApiKey(request) {
|
|
472
|
+
|
|
473
|
+
const authHeader = request.headers.get("Authorization");
|
|
474
|
+
if (authHeader?.startsWith("Bearer ")) {
|
|
475
|
+
return authHeader.slice(7);
|
|
476
|
+
}
|
|
477
|
+
|
|
478
|
+
const xApiKey = request.headers.get("x-api-key");
|
|
479
|
+
if (xApiKey) {
|
|
480
|
+
return xApiKey;
|
|
481
|
+
}
|
|
482
|
+
|
|
483
|
+
return null;
|
|
484
|
+
}
|
|
485
|
+
|
|
486
|
+
export async function isValidApiKey(apiKey) {
|
|
487
|
+
if (!apiKey) return false;
|
|
488
|
+
return await validateApiKey(apiKey);
|
|
489
|
+
}
|
|
@@ -0,0 +1,173 @@
|
|
|
1
|
+
import * as log from "../utils/logger.js";
|
|
2
|
+
import { getRefreshLeadMs } from "../../services/tokenRefresh.js";
|
|
3
|
+
import { getCredentialExpiryMs } from "../../services/oauthCredentialManager.js";
|
|
4
|
+
|
|
5
|
+
export const BACKGROUND_REFRESH_LEAD_MS = 30 * 60 * 1000;
|
|
6
|
+
const DEFAULT_INTERVAL_MS = 5 * 60 * 1000;
|
|
7
|
+
const INITIAL_DELAY_MS = 10 * 1000;
|
|
8
|
+
|
|
9
|
+
let started = false;
|
|
10
|
+
let intervalHandle = null;
|
|
11
|
+
let initialTimeoutHandle = null;
|
|
12
|
+
let tickRunning = false;
|
|
13
|
+
|
|
14
|
+
function isTruthyEnv(value) {
|
|
15
|
+
if (value == null || value === "") return false;
|
|
16
|
+
const v = String(value).trim().toLowerCase();
|
|
17
|
+
return v === "1" || v === "true" || v === "yes" || v === "on";
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
function isNonServerRuntime() {
|
|
21
|
+
if (typeof window !== "undefined") return true;
|
|
22
|
+
const phase = process.env.NEXT_PHASE || "";
|
|
23
|
+
if (
|
|
24
|
+
phase === "phase-production-build" ||
|
|
25
|
+
phase === "phase-export" ||
|
|
26
|
+
phase === "phase-static"
|
|
27
|
+
) {
|
|
28
|
+
return true;
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
if (process.env.NEXT_RUNTIME === "edge") return true;
|
|
32
|
+
return false;
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
export function selectConnectionsNeedingRefresh(connections, nowMs = Date.now()) {
|
|
36
|
+
if (!Array.isArray(connections) || connections.length === 0) return [];
|
|
37
|
+
|
|
38
|
+
const out = [];
|
|
39
|
+
for (const conn of connections) {
|
|
40
|
+
if (!conn) continue;
|
|
41
|
+
|
|
42
|
+
const authType = String(conn.authType || "").toLowerCase().replace(/_/g, "");
|
|
43
|
+
if (authType !== "oauth") continue;
|
|
44
|
+
if (!conn.refreshToken) continue;
|
|
45
|
+
|
|
46
|
+
const expiresAtMs = getCredentialExpiryMs(conn);
|
|
47
|
+
if (expiresAtMs === null) continue;
|
|
48
|
+
|
|
49
|
+
const providerLead = getRefreshLeadMs(conn.provider);
|
|
50
|
+
const leadMs = Math.max(
|
|
51
|
+
Number.isFinite(providerLead) ? providerLead : 0,
|
|
52
|
+
BACKGROUND_REFRESH_LEAD_MS
|
|
53
|
+
);
|
|
54
|
+
|
|
55
|
+
if (expiresAtMs - nowMs < leadMs) {
|
|
56
|
+
out.push(conn);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
return out;
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
async function loadActiveConnections() {
|
|
63
|
+
|
|
64
|
+
const { getProviderConnections } = await import("../../lib/db/repos/connectionsRepo.js");
|
|
65
|
+
return getProviderConnections({ isActive: true, authType: "oauth" });
|
|
66
|
+
}
|
|
67
|
+
|
|
68
|
+
async function refreshOne(connection) {
|
|
69
|
+
const { checkAndRefreshToken } = await import("./tokenRefresh.js");
|
|
70
|
+
return checkAndRefreshToken(connection.provider, connection, { force: true });
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
export async function runBackgroundTokenRefreshTick(deps = {}) {
|
|
74
|
+
if (tickRunning) {
|
|
75
|
+
log.debug("BG_TOKEN_REFRESH", "Tick already running, skip");
|
|
76
|
+
return;
|
|
77
|
+
}
|
|
78
|
+
tickRunning = true;
|
|
79
|
+
try {
|
|
80
|
+
const load = deps.loadConnections || loadActiveConnections;
|
|
81
|
+
const refresh = deps.refreshConnection || refreshOne;
|
|
82
|
+
|
|
83
|
+
const connections = await load();
|
|
84
|
+
const due = selectConnectionsNeedingRefresh(connections, Date.now());
|
|
85
|
+
|
|
86
|
+
if (due.length === 0) {
|
|
87
|
+
log.debug("BG_TOKEN_REFRESH", "No connections due for refresh", {
|
|
88
|
+
active: Array.isArray(connections) ? connections.length : 0,
|
|
89
|
+
});
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
log.info("BG_TOKEN_REFRESH", "Refreshing due OAuth connections", {
|
|
94
|
+
due: due.length,
|
|
95
|
+
ids: due.map((c) => c.id).filter(Boolean),
|
|
96
|
+
});
|
|
97
|
+
|
|
98
|
+
await Promise.allSettled(
|
|
99
|
+
due.map(async (conn) => {
|
|
100
|
+
try {
|
|
101
|
+
await refresh(conn);
|
|
102
|
+
log.info("BG_TOKEN_REFRESH", "Connection refresh finished", {
|
|
103
|
+
id: conn.id,
|
|
104
|
+
provider: conn.provider,
|
|
105
|
+
});
|
|
106
|
+
} catch (err) {
|
|
107
|
+
log.warn("BG_TOKEN_REFRESH", "Connection refresh failed (swallowed)", {
|
|
108
|
+
id: conn?.id,
|
|
109
|
+
provider: conn?.provider,
|
|
110
|
+
error: err?.message ?? String(err),
|
|
111
|
+
});
|
|
112
|
+
}
|
|
113
|
+
})
|
|
114
|
+
);
|
|
115
|
+
} catch (err) {
|
|
116
|
+
log.warn("BG_TOKEN_REFRESH", "Tick failed (swallowed)", {
|
|
117
|
+
error: err?.message ?? String(err),
|
|
118
|
+
});
|
|
119
|
+
} finally {
|
|
120
|
+
tickRunning = false;
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
|
|
124
|
+
export function startBackgroundTokenRefresh({ intervalMs } = {}) {
|
|
125
|
+
if (started) return false;
|
|
126
|
+
if (isTruthyEnv(process.env.DISABLE_BACKGROUND_TOKEN_REFRESH)) {
|
|
127
|
+
log.info("BG_TOKEN_REFRESH", "Disabled via DISABLE_BACKGROUND_TOKEN_REFRESH");
|
|
128
|
+
return false;
|
|
129
|
+
}
|
|
130
|
+
if (isNonServerRuntime()) {
|
|
131
|
+
log.debug("BG_TOKEN_REFRESH", "Skip start outside long-running server runtime");
|
|
132
|
+
return false;
|
|
133
|
+
}
|
|
134
|
+
|
|
135
|
+
started = true;
|
|
136
|
+
const period = Number.isFinite(intervalMs) && intervalMs > 0 ? intervalMs : DEFAULT_INTERVAL_MS;
|
|
137
|
+
|
|
138
|
+
const safeTick = () => {
|
|
139
|
+
runBackgroundTokenRefreshTick().catch((err) => {
|
|
140
|
+
log.warn("BG_TOKEN_REFRESH", "Unhandled tick rejection (swallowed)", {
|
|
141
|
+
error: err?.message ?? String(err),
|
|
142
|
+
});
|
|
143
|
+
});
|
|
144
|
+
};
|
|
145
|
+
|
|
146
|
+
initialTimeoutHandle = setTimeout(safeTick, INITIAL_DELAY_MS);
|
|
147
|
+
if (initialTimeoutHandle.unref) initialTimeoutHandle.unref();
|
|
148
|
+
|
|
149
|
+
intervalHandle = setInterval(safeTick, period);
|
|
150
|
+
if (intervalHandle.unref) intervalHandle.unref();
|
|
151
|
+
|
|
152
|
+
log.info("BG_TOKEN_REFRESH", "Scheduler started", {
|
|
153
|
+
intervalMs: period,
|
|
154
|
+
initialDelayMs: INITIAL_DELAY_MS,
|
|
155
|
+
leadMs: BACKGROUND_REFRESH_LEAD_MS,
|
|
156
|
+
});
|
|
157
|
+
return true;
|
|
158
|
+
}
|
|
159
|
+
|
|
160
|
+
export function stopBackgroundTokenRefresh() {
|
|
161
|
+
if (initialTimeoutHandle) {
|
|
162
|
+
clearTimeout(initialTimeoutHandle);
|
|
163
|
+
initialTimeoutHandle = null;
|
|
164
|
+
}
|
|
165
|
+
if (intervalHandle) {
|
|
166
|
+
clearInterval(intervalHandle);
|
|
167
|
+
intervalHandle = null;
|
|
168
|
+
}
|
|
169
|
+
if (started) {
|
|
170
|
+
started = false;
|
|
171
|
+
log.info("BG_TOKEN_REFRESH", "Scheduler stopped");
|
|
172
|
+
}
|
|
173
|
+
}
|