wyxrouter 0.4.71
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/CHANGELOG.md +222 -0
- package/LICENSE +21 -0
- package/README.md +96 -0
- package/README.zh-CN.md +1311 -0
- package/assets/pixel-router-2d.png +0 -0
- package/cli/LICENSE +42 -0
- package/cli/README.md +125 -0
- package/cli/app/package.json +58 -0
- package/cli/cli.js +806 -0
- package/cli/package.json +48 -0
- package/i18n/README.ja-JP.md +1209 -0
- package/i18n/README.ru.md +1311 -0
- package/i18n/README.vi.md +1310 -0
- package/i18n/README.zh-CN.md +1306 -0
- package/images/9router.png +0 -0
- package/jsconfig.json +12 -0
- package/next.config.mjs +71 -0
- package/open-sse/config/appConstants.js +203 -0
- package/open-sse/config/codexInstructions.js +119 -0
- package/open-sse/config/constants.js +4 -0
- package/open-sse/config/defaultThinkingSignature.js +12 -0
- package/open-sse/config/errorConfig.js +85 -0
- package/open-sse/config/googleTtsLanguages.js +62 -0
- package/open-sse/config/kiroConstants.js +322 -0
- package/open-sse/config/models.js +13 -0
- package/open-sse/config/ollamaModels.js +19 -0
- package/open-sse/config/providerModels.js +944 -0
- package/open-sse/config/providers.js +458 -0
- package/open-sse/config/runtimeConfig.js +72 -0
- package/open-sse/config/ttsModels.js +129 -0
- package/open-sse/executors/antigravity.js +504 -0
- package/open-sse/executors/azure.js +57 -0
- package/open-sse/executors/base.js +185 -0
- package/open-sse/executors/codex.js +469 -0
- package/open-sse/executors/commandcode.js +88 -0
- package/open-sse/executors/cursor.js +795 -0
- package/open-sse/executors/default.js +497 -0
- package/open-sse/executors/gemini-cli.js +89 -0
- package/open-sse/executors/github.js +379 -0
- package/open-sse/executors/grok-web.js +345 -0
- package/open-sse/executors/iflow.js +108 -0
- package/open-sse/executors/index.js +75 -0
- package/open-sse/executors/kiro.js +508 -0
- package/open-sse/executors/ollama-local.js +14 -0
- package/open-sse/executors/opencode-go.js +41 -0
- package/open-sse/executors/opencode.js +32 -0
- package/open-sse/executors/perplexity-web.js +507 -0
- package/open-sse/executors/qoder.js +450 -0
- package/open-sse/executors/qwen.js +129 -0
- package/open-sse/executors/vertex.js +131 -0
- package/open-sse/executors/xiaomi-tokenplan.js +19 -0
- package/open-sse/handlers/chatCore/nonStreamingHandler.js +230 -0
- package/open-sse/handlers/chatCore/requestDetail.js +102 -0
- package/open-sse/handlers/chatCore/sseToJsonHandler.js +231 -0
- package/open-sse/handlers/chatCore/streamingHandler.js +103 -0
- package/open-sse/handlers/chatCore.js +287 -0
- package/open-sse/handlers/embeddingProviders/_base.js +4 -0
- package/open-sse/handlers/embeddingProviders/gemini.js +54 -0
- package/open-sse/handlers/embeddingProviders/index.js +23 -0
- package/open-sse/handlers/embeddingProviders/openai.js +39 -0
- package/open-sse/handlers/embeddingProviders/openaiCompatNode.js +13 -0
- package/open-sse/handlers/embeddingsCore.js +126 -0
- package/open-sse/handlers/fetch/index.js +237 -0
- package/open-sse/handlers/imageGenerationCore.js +189 -0
- package/open-sse/handlers/imageProviders/_base.js +31 -0
- package/open-sse/handlers/imageProviders/blackForestLabs.js +43 -0
- package/open-sse/handlers/imageProviders/cloudflareAi.js +178 -0
- package/open-sse/handlers/imageProviders/codex.js +198 -0
- package/open-sse/handlers/imageProviders/comfyui.js +8 -0
- package/open-sse/handlers/imageProviders/falAi.js +41 -0
- package/open-sse/handlers/imageProviders/gemini.js +25 -0
- package/open-sse/handlers/imageProviders/huggingface.js +22 -0
- package/open-sse/handlers/imageProviders/index.js +40 -0
- package/open-sse/handlers/imageProviders/nanobanana.js +58 -0
- package/open-sse/handlers/imageProviders/openai.js +40 -0
- package/open-sse/handlers/imageProviders/runwayml.js +47 -0
- package/open-sse/handlers/imageProviders/sdwebui.js +17 -0
- package/open-sse/handlers/imageProviders/stabilityAi.js +34 -0
- package/open-sse/handlers/responsesHandler.js +103 -0
- package/open-sse/handlers/search/callers.js +371 -0
- package/open-sse/handlers/search/chatSearch.js +409 -0
- package/open-sse/handlers/search/index.js +201 -0
- package/open-sse/handlers/search/normalizers.js +223 -0
- package/open-sse/handlers/sttCore.js +194 -0
- package/open-sse/handlers/ttsCore.js +74 -0
- package/open-sse/handlers/ttsProviders/_base.js +39 -0
- package/open-sse/handlers/ttsProviders/edgeTts.js +89 -0
- package/open-sse/handlers/ttsProviders/elevenlabs.js +48 -0
- package/open-sse/handlers/ttsProviders/gemini.js +117 -0
- package/open-sse/handlers/ttsProviders/genericFormats.js +169 -0
- package/open-sse/handlers/ttsProviders/googleTts.js +54 -0
- package/open-sse/handlers/ttsProviders/index.js +50 -0
- package/open-sse/handlers/ttsProviders/localDevice.js +87 -0
- package/open-sse/handlers/ttsProviders/minimax.js +59 -0
- package/open-sse/handlers/ttsProviders/openai.js +30 -0
- package/open-sse/handlers/ttsProviders/openrouter.js +70 -0
- package/open-sse/index.js +82 -0
- package/open-sse/rtk/applyFilter.js +15 -0
- package/open-sse/rtk/autodetect.js +111 -0
- package/open-sse/rtk/caveman.js +100 -0
- package/open-sse/rtk/cavemanPrompts.js +78 -0
- package/open-sse/rtk/constants.js +55 -0
- package/open-sse/rtk/filters/buildOutput.js +127 -0
- package/open-sse/rtk/filters/dedupLog.js +44 -0
- package/open-sse/rtk/filters/find.js +49 -0
- package/open-sse/rtk/filters/gitDiff.js +92 -0
- package/open-sse/rtk/filters/gitStatus.js +117 -0
- package/open-sse/rtk/filters/grep.js +48 -0
- package/open-sse/rtk/filters/ls.js +79 -0
- package/open-sse/rtk/filters/readNumbered.js +27 -0
- package/open-sse/rtk/filters/searchList.js +52 -0
- package/open-sse/rtk/filters/smartTruncate.js +15 -0
- package/open-sse/rtk/filters/tree.js +32 -0
- package/open-sse/rtk/index.js +155 -0
- package/open-sse/rtk/registry.js +38 -0
- package/open-sse/services/accountFallback.js +238 -0
- package/open-sse/services/combo.js +198 -0
- package/open-sse/services/compact.js +71 -0
- package/open-sse/services/kiroModels.js +332 -0
- package/open-sse/services/model.js +261 -0
- package/open-sse/services/oauthCredentialManager.js +151 -0
- package/open-sse/services/projectId.js +306 -0
- package/open-sse/services/provider.js +356 -0
- package/open-sse/services/qoderModels.js +214 -0
- package/open-sse/services/tokenRefresh.js +939 -0
- package/open-sse/services/usage.js +1496 -0
- package/open-sse/transformer/responsesTransformer.js +439 -0
- package/open-sse/transformer/streamToJsonConverter.js +103 -0
- package/open-sse/translator/formats.js +36 -0
- package/open-sse/translator/helpers/claudeHelper.js +216 -0
- package/open-sse/translator/helpers/geminiHelper.js +372 -0
- package/open-sse/translator/helpers/imageHelper.js +34 -0
- package/open-sse/translator/helpers/maxTokensHelper.js +27 -0
- package/open-sse/translator/helpers/openaiHelper.js +130 -0
- package/open-sse/translator/helpers/responsesApiHelper.js +139 -0
- package/open-sse/translator/helpers/toolCallHelper.js +148 -0
- package/open-sse/translator/index.js +251 -0
- package/open-sse/translator/request/antigravity-to-openai.js +229 -0
- package/open-sse/translator/request/claude-to-openai.js +232 -0
- package/open-sse/translator/request/gemini-to-openai.js +147 -0
- package/open-sse/translator/request/openai-responses.js +318 -0
- package/open-sse/translator/request/openai-to-claude.js +401 -0
- package/open-sse/translator/request/openai-to-commandcode.js +170 -0
- package/open-sse/translator/request/openai-to-cursor.js +183 -0
- package/open-sse/translator/request/openai-to-gemini.js +470 -0
- package/open-sse/translator/request/openai-to-kiro.js +629 -0
- package/open-sse/translator/request/openai-to-kiro.old.js +278 -0
- package/open-sse/translator/request/openai-to-ollama.js +192 -0
- package/open-sse/translator/request/openai-to-vertex.js +42 -0
- package/open-sse/translator/response/claude-to-openai.js +206 -0
- package/open-sse/translator/response/commandcode-to-openai.js +197 -0
- package/open-sse/translator/response/cursor-to-openai.js +30 -0
- package/open-sse/translator/response/gemini-to-openai.js +245 -0
- package/open-sse/translator/response/kiro-to-openai.js +195 -0
- package/open-sse/translator/response/ollama-to-openai.js +152 -0
- package/open-sse/translator/response/openai-responses.js +590 -0
- package/open-sse/translator/response/openai-to-antigravity.js +122 -0
- package/open-sse/translator/response/openai-to-claude.js +266 -0
- package/open-sse/utils/bypassHandler.js +298 -0
- package/open-sse/utils/claudeCloaking.js +155 -0
- package/open-sse/utils/claudeHeaderCache.js +70 -0
- package/open-sse/utils/clientDetector.js +63 -0
- package/open-sse/utils/cursorChecksum.js +149 -0
- package/open-sse/utils/cursorProtobuf.js +904 -0
- package/open-sse/utils/debugLog.js +14 -0
- package/open-sse/utils/error.js +147 -0
- package/open-sse/utils/ollamaTransform.js +85 -0
- package/open-sse/utils/proxyFetch.js +368 -0
- package/open-sse/utils/reasoningContentInjector.js +79 -0
- package/open-sse/utils/requestLogger.js +260 -0
- package/open-sse/utils/responsesStreamHelpers.js +49 -0
- package/open-sse/utils/sessionManager.js +82 -0
- package/open-sse/utils/stream.js +462 -0
- package/open-sse/utils/streamHandler.js +250 -0
- package/open-sse/utils/streamHelpers.js +122 -0
- package/open-sse/utils/toolDeduper.js +49 -0
- package/open-sse/utils/usageTracking.js +347 -0
- package/package.json +100 -0
- package/postcss.config.mjs +12 -0
- package/public/favicon.svg +11 -0
- package/public/file.svg +1 -0
- package/public/globe.svg +1 -0
- package/public/i18n/literals/ar.json +195 -0
- package/public/i18n/literals/bn.json +195 -0
- package/public/i18n/literals/cs.json +195 -0
- package/public/i18n/literals/da.json +195 -0
- package/public/i18n/literals/de.json +195 -0
- package/public/i18n/literals/el.json +195 -0
- package/public/i18n/literals/es.json +195 -0
- package/public/i18n/literals/fi.json +195 -0
- package/public/i18n/literals/fr.json +195 -0
- package/public/i18n/literals/he.json +195 -0
- package/public/i18n/literals/hi.json +195 -0
- package/public/i18n/literals/hu.json +195 -0
- package/public/i18n/literals/id.json +195 -0
- package/public/i18n/literals/it.json +195 -0
- package/public/i18n/literals/ja.json +195 -0
- package/public/i18n/literals/ko.json +195 -0
- package/public/i18n/literals/nl.json +195 -0
- package/public/i18n/literals/no.json +195 -0
- package/public/i18n/literals/pl.json +195 -0
- package/public/i18n/literals/pt-BR.json +195 -0
- package/public/i18n/literals/pt-PT.json +195 -0
- package/public/i18n/literals/ro.json +195 -0
- package/public/i18n/literals/ru.json +195 -0
- package/public/i18n/literals/sv.json +195 -0
- package/public/i18n/literals/th.json +195 -0
- package/public/i18n/literals/tl.json +195 -0
- package/public/i18n/literals/tr.json +195 -0
- package/public/i18n/literals/uk.json +195 -0
- package/public/i18n/literals/ur.json +195 -0
- package/public/i18n/literals/vi.json +195 -0
- package/public/i18n/literals/zh-CN.json +772 -0
- package/public/i18n/literals/zh-TW.json +195 -0
- package/public/icons/discord.svg +4 -0
- package/public/icons/icon-192.svg +4 -0
- package/public/icons/icon-512.svg +4 -0
- package/public/next.svg +1 -0
- package/public/providers/alicode-intl.png +0 -0
- package/public/providers/alicode.png +0 -0
- package/public/providers/amp.png +0 -0
- package/public/providers/anthropic-m.png +0 -0
- package/public/providers/anthropic.png +0 -0
- package/public/providers/antigravity.png +0 -0
- package/public/providers/assemblyai.png +0 -0
- package/public/providers/aws-polly.png +0 -0
- package/public/providers/azure.png +0 -0
- package/public/providers/black-forest-labs.png +0 -0
- package/public/providers/blackbox.png +0 -0
- package/public/providers/brave-search.png +0 -0
- package/public/providers/byteplus.png +0 -0
- package/public/providers/cartesia.png +0 -0
- package/public/providers/cerebras.png +0 -0
- package/public/providers/chutes.png +0 -0
- package/public/providers/claude.png +0 -0
- package/public/providers/cline.png +0 -0
- package/public/providers/cloudflare-ai.png +0 -0
- package/public/providers/codebuddy.svg +37 -0
- package/public/providers/codex.png +0 -0
- package/public/providers/cohere.png +0 -0
- package/public/providers/comfyui.png +0 -0
- package/public/providers/commandcode.png +0 -0
- package/public/providers/continue.png +0 -0
- package/public/providers/copilot.png +0 -0
- package/public/providers/coqui.png +0 -0
- package/public/providers/cursor.png +0 -0
- package/public/providers/deepgram.png +0 -0
- package/public/providers/deepseek-tui.png +0 -0
- package/public/providers/deepseek.png +0 -0
- package/public/providers/droid.png +0 -0
- package/public/providers/edge-tts.png +0 -0
- package/public/providers/elevenlabs.png +0 -0
- package/public/providers/exa.png +0 -0
- package/public/providers/fal-ai.png +0 -0
- package/public/providers/firecrawl.png +0 -0
- package/public/providers/fireworks.png +0 -0
- package/public/providers/gemini-cli.png +0 -0
- package/public/providers/gemini.png +0 -0
- package/public/providers/github.png +0 -0
- package/public/providers/glm-cn.png +0 -0
- package/public/providers/glm.png +0 -0
- package/public/providers/google-pse.png +0 -0
- package/public/providers/google-tts.png +0 -0
- package/public/providers/grok-web.png +0 -0
- package/public/providers/groq.png +0 -0
- package/public/providers/hermes.png +0 -0
- package/public/providers/huggingface.png +0 -0
- package/public/providers/hyperbolic.png +0 -0
- package/public/providers/iflow.png +0 -0
- package/public/providers/inworld.png +0 -0
- package/public/providers/jcode.png +0 -0
- package/public/providers/jina-ai.png +0 -0
- package/public/providers/jina-reader.png +0 -0
- package/public/providers/kilocode.png +0 -0
- package/public/providers/kimi-coding.png +0 -0
- package/public/providers/kimi.png +0 -0
- package/public/providers/kiro.png +0 -0
- package/public/providers/linkup.png +0 -0
- package/public/providers/local-device.png +0 -0
- package/public/providers/minimax-cn.png +0 -0
- package/public/providers/minimax.png +0 -0
- package/public/providers/mistral.png +0 -0
- package/public/providers/nanobanana.png +0 -0
- package/public/providers/nebius.png +0 -0
- package/public/providers/nvidia.png +0 -0
- package/public/providers/oai-cc.png +0 -0
- package/public/providers/oai-r.png +0 -0
- package/public/providers/ollama-local.png +0 -0
- package/public/providers/ollama.png +0 -0
- package/public/providers/openai.png +0 -0
- package/public/providers/openclaw.png +0 -0
- package/public/providers/opencode-go.png +0 -0
- package/public/providers/opencode.png +0 -0
- package/public/providers/openrouter.png +0 -0
- package/public/providers/perplexity-web.png +0 -0
- package/public/providers/perplexity.png +0 -0
- package/public/providers/playht.png +0 -0
- package/public/providers/qoder.png +0 -0
- package/public/providers/qwen.png +0 -0
- package/public/providers/recraft.png +0 -0
- package/public/providers/roo.png +0 -0
- package/public/providers/runwayml.png +0 -0
- package/public/providers/sdwebui.png +0 -0
- package/public/providers/searchapi.png +0 -0
- package/public/providers/searxng.png +0 -0
- package/public/providers/serper.png +0 -0
- package/public/providers/siliconflow.png +0 -0
- package/public/providers/stability-ai.png +0 -0
- package/public/providers/tavily.png +0 -0
- package/public/providers/together.png +0 -0
- package/public/providers/topaz.png +0 -0
- package/public/providers/tortoise.png +0 -0
- package/public/providers/vertex-partner.png +0 -0
- package/public/providers/vertex.png +0 -0
- package/public/providers/volcengine-ark.png +0 -0
- package/public/providers/voyage-ai.png +0 -0
- package/public/providers/xai.png +0 -0
- package/public/providers/xiaomi-mimo.png +0 -0
- package/public/providers/xiaomi-tokenplan.png +0 -0
- package/public/providers/youcom.png +0 -0
- package/public/sw.js +22 -0
- package/public/vercel.svg +1 -0
- package/public/window.svg +1 -0
- package/scripts/compact-request-details.mjs +71 -0
- package/scripts/import-codex-gptjson.mjs +342 -0
- package/scripts/start-standalone.mjs +25 -0
- package/scripts/translate-readme.js +201 -0
- package/src/app/(dashboard)/dashboard/automation/page.js +294 -0
- package/src/app/(dashboard)/dashboard/basic-chat/BasicChatPageClient.js +967 -0
- package/src/app/(dashboard)/dashboard/basic-chat/page.js +5 -0
- package/src/app/(dashboard)/dashboard/cli-tools/CLIToolsPageClient.js +66 -0
- package/src/app/(dashboard)/dashboard/cli-tools/[toolId]/ToolDetailClient.js +173 -0
- package/src/app/(dashboard)/dashboard/cli-tools/[toolId]/page.js +11 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/AntigravityToolCard.js +481 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/ApiKeySelect.js +66 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/BaseUrlSelect.js +174 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/ClaudeToolCard.js +390 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/ClineToolCard.js +301 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/CodexToolCard.js +458 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/CopilotToolCard.js +323 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/CoworkToolCard.js +640 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/DeepSeekTuiToolCard.js +338 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/DefaultToolCard.js +271 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/DroidToolCard.js +410 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/EndpointPresetControl.js +128 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/HermesToolCard.js +317 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/JcodeToolCard.js +380 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/KiloToolCard.js +275 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/MitmLinkCard.js +40 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/MitmServerCard.js +329 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/MitmToolCard.js +318 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/OpenClawToolCard.js +388 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/OpenCodeToolCard.js +500 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/ToolSummaryCard.js +39 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/cliEndpointMatch.js +13 -0
- package/src/app/(dashboard)/dashboard/cli-tools/components/index.js +19 -0
- package/src/app/(dashboard)/dashboard/cli-tools/page.js +7 -0
- package/src/app/(dashboard)/dashboard/combos/page.js +612 -0
- package/src/app/(dashboard)/dashboard/console-log/ConsoleLogClient.js +91 -0
- package/src/app/(dashboard)/dashboard/console-log/page.js +8 -0
- package/src/app/(dashboard)/dashboard/endpoint/EndpointPageClient.js +1555 -0
- package/src/app/(dashboard)/dashboard/endpoint/page.js +7 -0
- package/src/app/(dashboard)/dashboard/media-providers/[kind]/[id]/page.js +1903 -0
- package/src/app/(dashboard)/dashboard/media-providers/[kind]/page.js +289 -0
- package/src/app/(dashboard)/dashboard/media-providers/combo/[id]/page.js +410 -0
- package/src/app/(dashboard)/dashboard/media-providers/web/page.js +209 -0
- package/src/app/(dashboard)/dashboard/mitm/MitmPageClient.js +117 -0
- package/src/app/(dashboard)/dashboard/mitm/page.js +5 -0
- package/src/app/(dashboard)/dashboard/page.js +7 -0
- package/src/app/(dashboard)/dashboard/profile/page.js +1140 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/AddApiKeyModal.js +389 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/AddCustomModelModal.js +125 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/CompatibleModelsSection.js +250 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/ConnectionRow.js +299 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/CooldownTimer.js +42 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/EditCompatibleNodeModal.js +161 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/ModelRow.js +95 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/PassthroughModelsSection.js +183 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/page.js +2053 -0
- package/src/app/(dashboard)/dashboard/providers/[id]/page.new.js +1724 -0
- package/src/app/(dashboard)/dashboard/providers/components/ConnectionsCard.js +497 -0
- package/src/app/(dashboard)/dashboard/providers/components/ModelAvailabilityBadge.js +185 -0
- package/src/app/(dashboard)/dashboard/providers/components/ModelsCard.js +294 -0
- package/src/app/(dashboard)/dashboard/providers/new/page.js +220 -0
- package/src/app/(dashboard)/dashboard/providers/page.js +1365 -0
- package/src/app/(dashboard)/dashboard/proxy-pools/page.js +1092 -0
- package/src/app/(dashboard)/dashboard/quota/page.js +11 -0
- package/src/app/(dashboard)/dashboard/skills/page.js +112 -0
- package/src/app/(dashboard)/dashboard/translator/page.js +303 -0
- package/src/app/(dashboard)/dashboard/usage/components/OverviewCards.js +35 -0
- package/src/app/(dashboard)/dashboard/usage/components/ProviderLimits/ProviderLimitCard.js +185 -0
- package/src/app/(dashboard)/dashboard/usage/components/ProviderLimits/QuotaProgressBar.js +127 -0
- package/src/app/(dashboard)/dashboard/usage/components/ProviderLimits/QuotaTable.js +259 -0
- package/src/app/(dashboard)/dashboard/usage/components/ProviderLimits/index.js +1394 -0
- package/src/app/(dashboard)/dashboard/usage/components/ProviderLimits/utils.js +244 -0
- package/src/app/(dashboard)/dashboard/usage/components/ProviderTopology.js +327 -0
- package/src/app/(dashboard)/dashboard/usage/components/RequestDetailsTab.js +433 -0
- package/src/app/(dashboard)/dashboard/usage/components/UsageChart.js +141 -0
- package/src/app/(dashboard)/dashboard/usage/components/UsageTable.js +247 -0
- package/src/app/(dashboard)/dashboard/usage/page.js +75 -0
- package/src/app/(dashboard)/layout.js +6 -0
- package/src/app/api/auth/login/route.js +76 -0
- package/src/app/api/auth/logout/route.js +12 -0
- package/src/app/api/auth/oidc/callback/route.js +87 -0
- package/src/app/api/auth/oidc/start/route.js +52 -0
- package/src/app/api/auth/oidc/test/route.js +84 -0
- package/src/app/api/auth/status/route.js +45 -0
- package/src/app/api/cli-tools/all-statuses/route.js +46 -0
- package/src/app/api/cli-tools/antigravity-mitm/alias/route.js +53 -0
- package/src/app/api/cli-tools/antigravity-mitm/route.js +202 -0
- package/src/app/api/cli-tools/claude-settings/route.js +203 -0
- package/src/app/api/cli-tools/cline-settings/route.js +133 -0
- package/src/app/api/cli-tools/codex-gateway/accounts/route.js +16 -0
- package/src/app/api/cli-tools/codex-settings/route.js +239 -0
- package/src/app/api/cli-tools/copilot-settings/route.js +148 -0
- package/src/app/api/cli-tools/cowork-mcp-registry/route.js +77 -0
- package/src/app/api/cli-tools/cowork-mcp-tools/route.js +95 -0
- package/src/app/api/cli-tools/cowork-settings/route.js +412 -0
- package/src/app/api/cli-tools/deepseek-tui-settings/route.js +164 -0
- package/src/app/api/cli-tools/droid-settings/route.js +213 -0
- package/src/app/api/cli-tools/hermes-settings/route.js +175 -0
- package/src/app/api/cli-tools/jcode-settings/route.js +216 -0
- package/src/app/api/cli-tools/kilo-settings/route.js +131 -0
- package/src/app/api/cli-tools/openclaw-settings/route.js +292 -0
- package/src/app/api/cli-tools/opencode-settings/route.js +259 -0
- package/src/app/api/combos/[id]/route.js +81 -0
- package/src/app/api/combos/route.js +48 -0
- package/src/app/api/health/route.js +15 -0
- package/src/app/api/init/route.js +4 -0
- package/src/app/api/keys/[id]/route.js +58 -0
- package/src/app/api/keys/route.js +42 -0
- package/src/app/api/locale/route.js +30 -0
- package/src/app/api/mcp/[plugin]/message/route.js +21 -0
- package/src/app/api/mcp/[plugin]/sse/route.js +37 -0
- package/src/app/api/media-providers/tts/deepgram/voices/route.js +65 -0
- package/src/app/api/media-providers/tts/elevenlabs/voices/route.js +71 -0
- package/src/app/api/media-providers/tts/inworld/voices/route.js +61 -0
- package/src/app/api/media-providers/tts/minimax/voices/route.js +113 -0
- package/src/app/api/media-providers/tts/voices/route.js +99 -0
- package/src/app/api/models/alias/route.js +53 -0
- package/src/app/api/models/availability/route.js +103 -0
- package/src/app/api/models/custom/route.js +48 -0
- package/src/app/api/models/disabled/route.js +50 -0
- package/src/app/api/models/route.js +64 -0
- package/src/app/api/models/test/ping.js +191 -0
- package/src/app/api/models/test/route.js +14 -0
- package/src/app/api/oauth/[provider]/[action]/route.js +343 -0
- package/src/app/api/oauth/codebuddy/bulk-import/[jobId]/cancel/route.js +19 -0
- package/src/app/api/oauth/codebuddy/bulk-import/[jobId]/manual/[workerId]/route.js +30 -0
- package/src/app/api/oauth/codebuddy/bulk-import/[jobId]/route.js +23 -0
- package/src/app/api/oauth/codebuddy/bulk-import/latest/route.js +25 -0
- package/src/app/api/oauth/codebuddy/bulk-import/route.js +49 -0
- package/src/app/api/oauth/codebuddy/quota-cookie/route.js +133 -0
- package/src/app/api/oauth/codex/import-token/route.js +96 -0
- package/src/app/api/oauth/cursor/auto-import/route.js +258 -0
- package/src/app/api/oauth/cursor/import/route.js +100 -0
- package/src/app/api/oauth/gitlab/pat/route.js +62 -0
- package/src/app/api/oauth/iflow/cookie/route.js +137 -0
- package/src/app/api/oauth/kiro/auto-import/route.js +85 -0
- package/src/app/api/oauth/kiro/bulk-import/[jobId]/cancel/route.js +18 -0
- package/src/app/api/oauth/kiro/bulk-import/[jobId]/manual/[workerId]/route.js +29 -0
- package/src/app/api/oauth/kiro/bulk-import/[jobId]/route.js +22 -0
- package/src/app/api/oauth/kiro/bulk-import/latest/route.js +25 -0
- package/src/app/api/oauth/kiro/bulk-import/route.js +49 -0
- package/src/app/api/oauth/kiro/import/route.js +110 -0
- package/src/app/api/oauth/kiro/social-authorize/route.js +27 -0
- package/src/app/api/oauth/kiro/social-exchange/route.js +41 -0
- package/src/app/api/pricing/route.js +134 -0
- package/src/app/api/provider-nodes/[id]/route.js +101 -0
- package/src/app/api/provider-nodes/route.js +104 -0
- package/src/app/api/provider-nodes/validate/route.js +201 -0
- package/src/app/api/providers/[id]/models/route.js +526 -0
- package/src/app/api/providers/[id]/route.js +189 -0
- package/src/app/api/providers/[id]/test/route.js +23 -0
- package/src/app/api/providers/[id]/test/testUtils.js +714 -0
- package/src/app/api/providers/[id]/test-models/route.js +66 -0
- package/src/app/api/providers/client/route.js +126 -0
- package/src/app/api/providers/kilo/free-models/route.js +55 -0
- package/src/app/api/providers/route.js +206 -0
- package/src/app/api/providers/suggested-models/filters.js +20 -0
- package/src/app/api/providers/suggested-models/route.js +32 -0
- package/src/app/api/providers/test-batch/route.js +131 -0
- package/src/app/api/providers/validate/route.js +637 -0
- package/src/app/api/proxy-pools/[id]/route.js +123 -0
- package/src/app/api/proxy-pools/[id]/test/route.js +70 -0
- package/src/app/api/proxy-pools/cloudflare-deploy/route.js +145 -0
- package/src/app/api/proxy-pools/deno-deploy/route.js +175 -0
- package/src/app/api/proxy-pools/route.js +93 -0
- package/src/app/api/proxy-pools/vercel-deploy/route.js +142 -0
- package/src/app/api/settings/database/route.js +36 -0
- package/src/app/api/settings/proxy-test/route.js +23 -0
- package/src/app/api/settings/require-login/route.js +15 -0
- package/src/app/api/settings/route.js +100 -0
- package/src/app/api/shutdown/route.js +24 -0
- package/src/app/api/tags/route.js +18 -0
- package/src/app/api/translator/console-logs/route.js +24 -0
- package/src/app/api/translator/console-logs/stream/route.js +79 -0
- package/src/app/api/translator/load/route.js +45 -0
- package/src/app/api/translator/save/route.js +44 -0
- package/src/app/api/translator/send/route.js +94 -0
- package/src/app/api/translator/translate/route.js +90 -0
- package/src/app/api/tunnel/disable/route.js +12 -0
- package/src/app/api/tunnel/enable/route.js +16 -0
- package/src/app/api/tunnel/status/route.js +13 -0
- package/src/app/api/tunnel/tailscale-check/route.js +50 -0
- package/src/app/api/tunnel/tailscale-disable/route.js +12 -0
- package/src/app/api/tunnel/tailscale-enable/route.js +12 -0
- package/src/app/api/tunnel/tailscale-install/route.js +72 -0
- package/src/app/api/usage/[connectionId]/route.js +188 -0
- package/src/app/api/usage/chart/route.js +21 -0
- package/src/app/api/usage/history/route.js +12 -0
- package/src/app/api/usage/logs/route.js +12 -0
- package/src/app/api/usage/providers/route.js +42 -0
- package/src/app/api/usage/request-details/route.js +57 -0
- package/src/app/api/usage/request-logs/route.js +13 -0
- package/src/app/api/usage/stats/route.js +23 -0
- package/src/app/api/usage/stream/route.js +79 -0
- package/src/app/api/v1/api/chat/route.js +37 -0
- package/src/app/api/v1/audio/speech/route.js +16 -0
- package/src/app/api/v1/audio/transcriptions/route.js +19 -0
- package/src/app/api/v1/audio/voices/route.js +68 -0
- package/src/app/api/v1/chat/completions/route.js +35 -0
- package/src/app/api/v1/embeddings/route.js +21 -0
- package/src/app/api/v1/images/generations/route.js +16 -0
- package/src/app/api/v1/messages/count_tokens/route.js +52 -0
- package/src/app/api/v1/messages/route.js +36 -0
- package/src/app/api/v1/models/[kind]/route.js +55 -0
- package/src/app/api/v1/models/info/route.js +110 -0
- package/src/app/api/v1/models/route.js +451 -0
- package/src/app/api/v1/responses/compact/route.js +37 -0
- package/src/app/api/v1/responses/route.js +30 -0
- package/src/app/api/v1/route.js +1 -0
- package/src/app/api/v1/search/route.js +21 -0
- package/src/app/api/v1/web/fetch/route.js +21 -0
- package/src/app/api/v1beta/models/[...path]/route.js +328 -0
- package/src/app/api/v1beta/models/route.js +44 -0
- package/src/app/api/version/route.js +45 -0
- package/src/app/api/version/shutdown/route.js +15 -0
- package/src/app/api/version/update/route.js +21 -0
- package/src/app/callback/page.js +148 -0
- package/src/app/dashboard/settings/pricing/page.js +173 -0
- package/src/app/favicon.ico +0 -0
- package/src/app/globals.css +496 -0
- package/src/app/landing/components/AnimatedBackground.js +57 -0
- package/src/app/landing/components/Features.js +133 -0
- package/src/app/landing/components/FlowAnimation.js +175 -0
- package/src/app/landing/components/Footer.js +61 -0
- package/src/app/landing/components/GetStarted.js +97 -0
- package/src/app/landing/components/HeroSection.js +47 -0
- package/src/app/landing/components/HowItWorks.js +66 -0
- package/src/app/landing/components/Navigation.js +72 -0
- package/src/app/landing/page.js +106 -0
- package/src/app/layout.js +49 -0
- package/src/app/login/page.js +197 -0
- package/src/app/manifest.js +30 -0
- package/src/app/page.js +5 -0
- package/src/dashboardGuard.js +242 -0
- package/src/i18n/RuntimeI18nProvider.js +27 -0
- package/src/i18n/config.js +146 -0
- package/src/i18n/runtime.js +162 -0
- package/src/lib/appUpdater.js +200 -0
- package/src/lib/auth/dashboardSession.js +68 -0
- package/src/lib/auth/loginLimiter.js +52 -0
- package/src/lib/auth/oidc.js +234 -0
- package/src/lib/consoleLogBuffer.js +79 -0
- package/src/lib/dataDir.js +29 -0
- package/src/lib/db/adapters/betterSqliteAdapter.js +55 -0
- package/src/lib/db/adapters/bunSqliteAdapter.js +63 -0
- package/src/lib/db/adapters/nodeSqliteAdapter.js +84 -0
- package/src/lib/db/adapters/sqljsAdapter.js +115 -0
- package/src/lib/db/backup.js +35 -0
- package/src/lib/db/driver.js +85 -0
- package/src/lib/db/helpers/jsonCol.js +9 -0
- package/src/lib/db/helpers/kvStore.js +39 -0
- package/src/lib/db/helpers/metaStore.js +22 -0
- package/src/lib/db/index.js +171 -0
- package/src/lib/db/migrate.js +286 -0
- package/src/lib/db/migrations/001-initial.js +14 -0
- package/src/lib/db/migrations/index.js +10 -0
- package/src/lib/db/paths.js +18 -0
- package/src/lib/db/repos/aliasRepo.js +62 -0
- package/src/lib/db/repos/apiKeysRepo.js +75 -0
- package/src/lib/db/repos/combosRepo.js +73 -0
- package/src/lib/db/repos/connectionsRepo.js +226 -0
- package/src/lib/db/repos/disabledModelsRepo.js +56 -0
- package/src/lib/db/repos/nodesRepo.js +95 -0
- package/src/lib/db/repos/pricingRepo.js +108 -0
- package/src/lib/db/repos/proxyPoolsRepo.js +103 -0
- package/src/lib/db/repos/requestDetailsRepo.js +259 -0
- package/src/lib/db/repos/settingsRepo.js +104 -0
- package/src/lib/db/repos/usageRepo.js +731 -0
- package/src/lib/db/schema.js +157 -0
- package/src/lib/db/version.js +21 -0
- package/src/lib/disabledModelsDb.js +4 -0
- package/src/lib/localDb.js +21 -0
- package/src/lib/mcp/stdioSseBridge.js +198 -0
- package/src/lib/mitmAliasCache.js +46 -0
- package/src/lib/network/connectionProxy.js +160 -0
- package/src/lib/network/initOutboundProxy.js +25 -0
- package/src/lib/network/outboundProxy.js +68 -0
- package/src/lib/network/proxyTest.js +91 -0
- package/src/lib/oauth/constants/oauth.js +284 -0
- package/src/lib/oauth/constants/xai.js +61 -0
- package/src/lib/oauth/providers.js +1506 -0
- package/src/lib/oauth/services/antigravity.js +321 -0
- package/src/lib/oauth/services/claude.js +136 -0
- package/src/lib/oauth/services/codebuddyBulkImportManager.js +454 -0
- package/src/lib/oauth/services/codex.js +144 -0
- package/src/lib/oauth/services/cursor.js +179 -0
- package/src/lib/oauth/services/gemini.js +240 -0
- package/src/lib/oauth/services/github.js +225 -0
- package/src/lib/oauth/services/iflow.js +202 -0
- package/src/lib/oauth/services/index.js +17 -0
- package/src/lib/oauth/services/kiro.js +334 -0
- package/src/lib/oauth/services/kiroBulkImportManager.js +778 -0
- package/src/lib/oauth/services/kiroConnections.js +93 -0
- package/src/lib/oauth/services/kiroGoogleAutomation.js +1136 -0
- package/src/lib/oauth/services/oauth.js +157 -0
- package/src/lib/oauth/services/openai.js +123 -0
- package/src/lib/oauth/services/qoder.js +216 -0
- package/src/lib/oauth/services/qwen.js +170 -0
- package/src/lib/oauth/services/xai.js +238 -0
- package/src/lib/oauth/utils/banner.js +63 -0
- package/src/lib/oauth/utils/pkce.js +39 -0
- package/src/lib/oauth/utils/server.js +415 -0
- package/src/lib/oauth/utils/ui.js +48 -0
- package/src/lib/providerNormalization.js +45 -0
- package/src/lib/qoder/constants.js +64 -0
- package/src/lib/qoder/cosy.js +175 -0
- package/src/lib/qoder/encoding.js +55 -0
- package/src/lib/requestDetailsDb.js +4 -0
- package/src/lib/tunnel/cloudflare/cloudflared.js +449 -0
- package/src/lib/tunnel/cloudflare/config.js +9 -0
- package/src/lib/tunnel/cloudflare/healthCheck.js +29 -0
- package/src/lib/tunnel/cloudflare/manager.js +151 -0
- package/src/lib/tunnel/cloudflare/pid.js +23 -0
- package/src/lib/tunnel/index.js +48 -0
- package/src/lib/tunnel/shared/dnsResolver.js +17 -0
- package/src/lib/tunnel/shared/internetCheck.js +26 -0
- package/src/lib/tunnel/shared/state.js +41 -0
- package/src/lib/tunnel/shared/watchdogConfig.js +8 -0
- package/src/lib/tunnel/tailscale/config.js +7 -0
- package/src/lib/tunnel/tailscale/healthCheck.js +29 -0
- package/src/lib/tunnel/tailscale/manager.js +129 -0
- package/src/lib/tunnel/tailscale/tailscale.js +790 -0
- package/src/lib/updater/updater.js +235 -0
- package/src/lib/usage/fetcher.js +208 -0
- package/src/lib/usageDb.js +7 -0
- package/src/mitm/antigravityIdeVersion.js +50 -0
- package/src/mitm/cert/generate.js +32 -0
- package/src/mitm/cert/install.js +269 -0
- package/src/mitm/cert/rootCA.js +173 -0
- package/src/mitm/config.js +87 -0
- package/src/mitm/dbReader.js +22 -0
- package/src/mitm/dns/dnsConfig.js +266 -0
- package/src/mitm/handlers/antigravity.js +33 -0
- package/src/mitm/handlers/base.js +226 -0
- package/src/mitm/handlers/copilot.js +35 -0
- package/src/mitm/handlers/cursor.js +15 -0
- package/src/mitm/handlers/kiro.js +526 -0
- package/src/mitm/logger.js +106 -0
- package/src/mitm/manager.js +851 -0
- package/src/mitm/paths.js +32 -0
- package/src/mitm/server.js +435 -0
- package/src/mitm/winElevated.js +81 -0
- package/src/models/index.js +38 -0
- package/src/proxy.js +5 -0
- package/src/shared/components/AddCustomEmbeddingModal.js +183 -0
- package/src/shared/components/Avatar.js +88 -0
- package/src/shared/components/Badge.js +54 -0
- package/src/shared/components/BulkAccountAutomationModal.js +508 -0
- package/src/shared/components/Button.js +56 -0
- package/src/shared/components/Card.js +116 -0
- package/src/shared/components/ChangelogModal.js +97 -0
- package/src/shared/components/CodeBuddyQuotaCookieModal.js +109 -0
- package/src/shared/components/ComboFormModal.js +176 -0
- package/src/shared/components/CursorAuthModal.js +212 -0
- package/src/shared/components/DonateModal.js +136 -0
- package/src/shared/components/Drawer.js +82 -0
- package/src/shared/components/EditConnectionModal.js +286 -0
- package/src/shared/components/Footer.js +132 -0
- package/src/shared/components/GitLabAuthModal.js +194 -0
- package/src/shared/components/Header.js +380 -0
- package/src/shared/components/HeaderLanguage.js +46 -0
- package/src/shared/components/HeaderMenu.js +126 -0
- package/src/shared/components/IFlowCookieModal.js +132 -0
- package/src/shared/components/Input.js +65 -0
- package/src/shared/components/KiroAuthModal.js +1171 -0
- package/src/shared/components/KiroOAuthWrapper.js +149 -0
- package/src/shared/components/KiroSocialOAuthModal.js +205 -0
- package/src/shared/components/LanguageSwitcher.js +190 -0
- package/src/shared/components/Loading.js +75 -0
- package/src/shared/components/ManualConfigModal.js +44 -0
- package/src/shared/components/McpMarketplaceModal.js +255 -0
- package/src/shared/components/Modal.js +146 -0
- package/src/shared/components/ModelSelectModal.js +537 -0
- package/src/shared/components/NineRemoteButton.js +23 -0
- package/src/shared/components/NineRemotePromoModal.js +99 -0
- package/src/shared/components/NoAuthProxyCard.js +86 -0
- package/src/shared/components/OAuthModal.js +682 -0
- package/src/shared/components/Pagination.js +150 -0
- package/src/shared/components/PricingModal.js +208 -0
- package/src/shared/components/ProviderIcon.js +63 -0
- package/src/shared/components/ProviderInfoCard.js +82 -0
- package/src/shared/components/RequestLogger.js +121 -0
- package/src/shared/components/SegmentedControl.js +48 -0
- package/src/shared/components/Select.js +67 -0
- package/src/shared/components/Sidebar.js +441 -0
- package/src/shared/components/ThemeProvider.js +15 -0
- package/src/shared/components/ThemeToggle.js +42 -0
- package/src/shared/components/Toggle.js +69 -0
- package/src/shared/components/Tooltip.js +25 -0
- package/src/shared/components/UsageStats.js +505 -0
- package/src/shared/components/index.js +46 -0
- package/src/shared/components/layouts/AuthLayout.js +29 -0
- package/src/shared/components/layouts/DashboardLayout.js +104 -0
- package/src/shared/components/layouts/index.js +4 -0
- package/src/shared/constants/cliTools.js +397 -0
- package/src/shared/constants/colors.js +77 -0
- package/src/shared/constants/config.js +99 -0
- package/src/shared/constants/coworkPlugins.js +75 -0
- package/src/shared/constants/index.js +4 -0
- package/src/shared/constants/locales.js +36 -0
- package/src/shared/constants/mitmToolHosts.js +12 -0
- package/src/shared/constants/models.js +38 -0
- package/src/shared/constants/pricing.js +303 -0
- package/src/shared/constants/providers.js +289 -0
- package/src/shared/constants/skills.js +78 -0
- package/src/shared/constants/ttsProviders.js +138 -0
- package/src/shared/hooks/index.js +2 -0
- package/src/shared/hooks/useCopyToClipboard.js +43 -0
- package/src/shared/hooks/useTheme.js +60 -0
- package/src/shared/services/bootstrap.js +12 -0
- package/src/shared/services/initializeApp.js +268 -0
- package/src/shared/utils/api.js +93 -0
- package/src/shared/utils/apiKey.js +98 -0
- package/src/shared/utils/clineAuth.js +37 -0
- package/src/shared/utils/cn.js +11 -0
- package/src/shared/utils/connectionStatus.js +162 -0
- package/src/shared/utils/index.js +40 -0
- package/src/shared/utils/machine.js +6 -0
- package/src/shared/utils/machineId.js +66 -0
- package/src/shared/utils/providerModelsFetcher.js +30 -0
- package/src/sse/handlers/chat.js +261 -0
- package/src/sse/handlers/embeddings.js +141 -0
- package/src/sse/handlers/fetch.js +213 -0
- package/src/sse/handlers/imageGeneration.js +142 -0
- package/src/sse/handlers/search.js +206 -0
- package/src/sse/handlers/stt.js +88 -0
- package/src/sse/handlers/tts.js +114 -0
- package/src/sse/services/auth.js +346 -0
- package/src/sse/services/codexGateway.js +215 -0
- package/src/sse/services/model.js +99 -0
- package/src/sse/services/tokenRefresh.js +319 -0
- package/src/sse/utils/logger.js +75 -0
- package/src/store/headerSearchStore.js +19 -0
- package/src/store/index.js +6 -0
- package/src/store/notificationStore.js +45 -0
- package/src/store/providerStore.js +55 -0
- package/src/store/settingsStore.js +51 -0
- package/src/store/themeStore.js +54 -0
- package/src/store/userStore.js +20 -0
- package/start.sh +4 -0
|
@@ -0,0 +1,778 @@
|
|
|
1
|
+
import { randomUUID } from "crypto";
|
|
2
|
+
import fs from "node:fs";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import { DATA_DIR } from "../../dataDir.js";
|
|
5
|
+
import { KiroService } from "./kiro.js";
|
|
6
|
+
import { createKiroCallbackMonitor, runKiroGoogleAutomation } from "./kiroGoogleAutomation.js";
|
|
7
|
+
|
|
8
|
+
export const KIRO_BULK_IMPORT_DEFAULT_CONCURRENCY = 4;
|
|
9
|
+
export const KIRO_BULK_IMPORT_MIN_CONCURRENCY = 1;
|
|
10
|
+
export const KIRO_BULK_IMPORT_MAX_CONCURRENCY = 8;
|
|
11
|
+
|
|
12
|
+
const TERMINAL_ACCOUNT_STATUSES = new Set([
|
|
13
|
+
"success",
|
|
14
|
+
"failed",
|
|
15
|
+
"failed_invalid_credentials",
|
|
16
|
+
"failed_exchange",
|
|
17
|
+
"failed_timeout",
|
|
18
|
+
"cancelled",
|
|
19
|
+
]);
|
|
20
|
+
|
|
21
|
+
const MAX_ACCOUNT_LOG_ENTRIES = 40;
|
|
22
|
+
const MAX_JOB_ACTIVITY_ENTRIES = 80;
|
|
23
|
+
const PREVIEW_CAPTURE_INTERVAL_MS = 1500;
|
|
24
|
+
const RECENT_TERMINAL_JOB_WINDOW_MS = 30 * 60_000;
|
|
25
|
+
const KIRO_BULK_IMPORT_DIR = path.join(DATA_DIR, "kiro-bulk-import");
|
|
26
|
+
const KIRO_BULK_IMPORT_META_FILE = path.join(KIRO_BULK_IMPORT_DIR, "meta.json");
|
|
27
|
+
const ACTIVE_JOB_STATUSES = new Set(["queued", "running", "needs_manual"]);
|
|
28
|
+
|
|
29
|
+
function nowIso() {
|
|
30
|
+
return new Date().toISOString();
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
function ensurePersistenceDir(dir = KIRO_BULK_IMPORT_DIR) {
|
|
34
|
+
fs.mkdirSync(dir, { recursive: true });
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
function getJobFile(jobId, dir = KIRO_BULK_IMPORT_DIR) {
|
|
38
|
+
ensurePersistenceDir(dir);
|
|
39
|
+
return path.join(dir, `${jobId}.json`);
|
|
40
|
+
}
|
|
41
|
+
|
|
42
|
+
function readJsonFile(filePath) {
|
|
43
|
+
try {
|
|
44
|
+
if (!fs.existsSync(filePath)) return null;
|
|
45
|
+
return JSON.parse(fs.readFileSync(filePath, "utf8"));
|
|
46
|
+
} catch {
|
|
47
|
+
return null;
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
function writeJsonFile(filePath, payload) {
|
|
52
|
+
ensurePersistenceDir(path.dirname(filePath));
|
|
53
|
+
const tempFile = `${filePath}.${process.pid}.tmp`;
|
|
54
|
+
fs.writeFileSync(tempFile, JSON.stringify(payload, null, 2), "utf8");
|
|
55
|
+
fs.renameSync(tempFile, filePath);
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
function readPersistedLatestJobId(metaFile = KIRO_BULK_IMPORT_META_FILE) {
|
|
59
|
+
return readJsonFile(metaFile)?.latestJobId || null;
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
function writePersistedLatestJobId(jobId, metaFile = KIRO_BULK_IMPORT_META_FILE) {
|
|
63
|
+
writeJsonFile(metaFile, {
|
|
64
|
+
latestJobId: jobId || null,
|
|
65
|
+
updatedAt: nowIso(),
|
|
66
|
+
});
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
function clampConcurrency(value) {
|
|
70
|
+
const parsed = Number.parseInt(value, 10);
|
|
71
|
+
if (!Number.isFinite(parsed)) return KIRO_BULK_IMPORT_DEFAULT_CONCURRENCY;
|
|
72
|
+
return Math.min(KIRO_BULK_IMPORT_MAX_CONCURRENCY, Math.max(KIRO_BULK_IMPORT_MIN_CONCURRENCY, parsed));
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
export function parseKiroBulkAccounts(accounts = []) {
|
|
76
|
+
const lines = Array.isArray(accounts) ? accounts : [];
|
|
77
|
+
const parsed = [];
|
|
78
|
+
const invalidLines = [];
|
|
79
|
+
|
|
80
|
+
lines.forEach((line, index) => {
|
|
81
|
+
const raw = String(line || "").trim();
|
|
82
|
+
if (!raw) return;
|
|
83
|
+
|
|
84
|
+
const [email = "", ...passwordParts] = raw.split("|");
|
|
85
|
+
const normalizedEmail = email.trim();
|
|
86
|
+
const normalizedPassword = passwordParts.join("|").trim();
|
|
87
|
+
|
|
88
|
+
if (!normalizedEmail || !normalizedPassword) {
|
|
89
|
+
invalidLines.push(index + 1);
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
parsed.push({
|
|
94
|
+
line: index + 1,
|
|
95
|
+
email: normalizedEmail,
|
|
96
|
+
password: normalizedPassword,
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
|
|
100
|
+
return {
|
|
101
|
+
parsed,
|
|
102
|
+
invalidLines,
|
|
103
|
+
};
|
|
104
|
+
}
|
|
105
|
+
|
|
106
|
+
function getFailedCount(accounts) {
|
|
107
|
+
return accounts.filter((account) => (
|
|
108
|
+
account.status === "failed"
|
|
109
|
+
|| account.status === "failed_invalid_credentials"
|
|
110
|
+
|| account.status === "failed_exchange"
|
|
111
|
+
|| account.status === "failed_timeout"
|
|
112
|
+
)).length;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
function buildSummary(accounts) {
|
|
116
|
+
return {
|
|
117
|
+
total: accounts.length,
|
|
118
|
+
queued: accounts.filter((account) => account.status === "queued").length,
|
|
119
|
+
running: accounts.filter((account) => account.status === "running").length,
|
|
120
|
+
success: accounts.filter((account) => account.status === "success").length,
|
|
121
|
+
failed: getFailedCount(accounts),
|
|
122
|
+
needs_manual: accounts.filter((account) => account.status === "needs_manual").length,
|
|
123
|
+
};
|
|
124
|
+
}
|
|
125
|
+
|
|
126
|
+
function createLogEntry(step, message, level = "info") {
|
|
127
|
+
return {
|
|
128
|
+
id: randomUUID(),
|
|
129
|
+
at: nowIso(),
|
|
130
|
+
step,
|
|
131
|
+
message,
|
|
132
|
+
level,
|
|
133
|
+
};
|
|
134
|
+
}
|
|
135
|
+
|
|
136
|
+
function appendAccountLog(account, step, message, level = "info") {
|
|
137
|
+
const entry = createLogEntry(step, message, level);
|
|
138
|
+
account.currentStep = step;
|
|
139
|
+
account.updatedAt = entry.at;
|
|
140
|
+
account.logs = account.logs || [];
|
|
141
|
+
account.logs.push(entry);
|
|
142
|
+
if (account.logs.length > MAX_ACCOUNT_LOG_ENTRIES) {
|
|
143
|
+
account.logs.splice(0, account.logs.length - MAX_ACCOUNT_LOG_ENTRIES);
|
|
144
|
+
}
|
|
145
|
+
return entry;
|
|
146
|
+
}
|
|
147
|
+
|
|
148
|
+
function buildJobActivity(accounts) {
|
|
149
|
+
return accounts
|
|
150
|
+
.flatMap((account) => (account.logs || []).map((entry) => ({
|
|
151
|
+
...entry,
|
|
152
|
+
email: account.email,
|
|
153
|
+
line: account.line,
|
|
154
|
+
workerId: account.workerId || null,
|
|
155
|
+
status: account.status,
|
|
156
|
+
})))
|
|
157
|
+
.sort((left, right) => String(left.at).localeCompare(String(right.at)))
|
|
158
|
+
.slice(-MAX_JOB_ACTIVITY_ENTRIES);
|
|
159
|
+
}
|
|
160
|
+
|
|
161
|
+
function sanitizeAccount(account) {
|
|
162
|
+
return {
|
|
163
|
+
email: account.email,
|
|
164
|
+
status: account.status,
|
|
165
|
+
error: account.error || null,
|
|
166
|
+
connectionId: account.connectionId || null,
|
|
167
|
+
workerId: account.workerId || null,
|
|
168
|
+
line: account.line,
|
|
169
|
+
currentStep: account.currentStep || null,
|
|
170
|
+
updatedAt: account.updatedAt || null,
|
|
171
|
+
logs: (account.logs || []).slice(-8),
|
|
172
|
+
manualSessionAvailable: Boolean(account.manualSession?.page) && account.status === "needs_manual",
|
|
173
|
+
manualSessionOpened: Boolean(account.manualSession?.opened),
|
|
174
|
+
};
|
|
175
|
+
}
|
|
176
|
+
|
|
177
|
+
function sanitizeJob(job, extras = {}) {
|
|
178
|
+
return {
|
|
179
|
+
jobId: job.jobId,
|
|
180
|
+
status: job.status,
|
|
181
|
+
summary: buildSummary(job.accounts),
|
|
182
|
+
concurrency: job.concurrency,
|
|
183
|
+
createdAt: job.createdAt,
|
|
184
|
+
startedAt: job.startedAt,
|
|
185
|
+
finishedAt: job.finishedAt,
|
|
186
|
+
accounts: job.accounts.map(sanitizeAccount),
|
|
187
|
+
activity: buildJobActivity(job.accounts),
|
|
188
|
+
error: job.error || null,
|
|
189
|
+
preview: extras.preview || null,
|
|
190
|
+
};
|
|
191
|
+
}
|
|
192
|
+
|
|
193
|
+
function buildPersistedSnapshot(job) {
|
|
194
|
+
return sanitizeJob(job, {
|
|
195
|
+
preview: job.lastPreview || null,
|
|
196
|
+
});
|
|
197
|
+
}
|
|
198
|
+
|
|
199
|
+
function isRecentTerminalJob(job) {
|
|
200
|
+
if (!job || ACTIVE_JOB_STATUSES.has(job.status)) return false;
|
|
201
|
+
const finishedAtMs = job.finishedAt ? Date.parse(job.finishedAt) : NaN;
|
|
202
|
+
if (!Number.isFinite(finishedAtMs)) return false;
|
|
203
|
+
return (Date.now() - finishedAtMs) <= RECENT_TERMINAL_JOB_WINDOW_MS;
|
|
204
|
+
}
|
|
205
|
+
|
|
206
|
+
export function buildLookupResponse(job, extras = {}) {
|
|
207
|
+
if (!job) {
|
|
208
|
+
return {
|
|
209
|
+
found: false,
|
|
210
|
+
stale: Boolean(extras.stale),
|
|
211
|
+
recoverable: false,
|
|
212
|
+
job: null,
|
|
213
|
+
};
|
|
214
|
+
}
|
|
215
|
+
|
|
216
|
+
return {
|
|
217
|
+
found: true,
|
|
218
|
+
stale: false,
|
|
219
|
+
recoverable: ACTIVE_JOB_STATUSES.has(job.status) || isRecentTerminalJob(job),
|
|
220
|
+
job,
|
|
221
|
+
};
|
|
222
|
+
}
|
|
223
|
+
|
|
224
|
+
async function defaultBrowserLauncher() {
|
|
225
|
+
const { chromium } = await import("playwright");
|
|
226
|
+
|
|
227
|
+
return await chromium.launch({
|
|
228
|
+
headless: true,
|
|
229
|
+
});
|
|
230
|
+
}
|
|
231
|
+
|
|
232
|
+
async function defaultSocialExchange(args) {
|
|
233
|
+
const { exchangeAndSaveKiroSocialConnection } = await import("./kiroConnections.js");
|
|
234
|
+
return exchangeAndSaveKiroSocialConnection(args);
|
|
235
|
+
}
|
|
236
|
+
|
|
237
|
+
export async function createFreshContext(browser) {
|
|
238
|
+
const context = await browser.newContext();
|
|
239
|
+
const page = await context.newPage();
|
|
240
|
+
return { context, page };
|
|
241
|
+
}
|
|
242
|
+
|
|
243
|
+
async function revealBrowserWindow(page) {
|
|
244
|
+
if (!page) return false;
|
|
245
|
+
|
|
246
|
+
try {
|
|
247
|
+
const context = page.context?.();
|
|
248
|
+
if (!context?.newCDPSession) {
|
|
249
|
+
await page.bringToFront?.().catch(() => null);
|
|
250
|
+
return true;
|
|
251
|
+
}
|
|
252
|
+
|
|
253
|
+
const session = await context.newCDPSession(page);
|
|
254
|
+
let windowId = null;
|
|
255
|
+
|
|
256
|
+
try {
|
|
257
|
+
const targetInfo = await session.send("Target.getTargetInfo");
|
|
258
|
+
const targetId = targetInfo?.targetInfo?.targetId;
|
|
259
|
+
const windowInfo = await session.send("Browser.getWindowForTarget", targetId ? { targetId } : {});
|
|
260
|
+
windowId = windowInfo?.windowId ?? null;
|
|
261
|
+
} catch {
|
|
262
|
+
windowId = null;
|
|
263
|
+
}
|
|
264
|
+
|
|
265
|
+
if (windowId != null) {
|
|
266
|
+
await session.send("Browser.setWindowBounds", {
|
|
267
|
+
windowId,
|
|
268
|
+
bounds: {
|
|
269
|
+
windowState: "normal",
|
|
270
|
+
left: 80,
|
|
271
|
+
top: 80,
|
|
272
|
+
width: 1280,
|
|
273
|
+
height: 960,
|
|
274
|
+
},
|
|
275
|
+
}).catch(() => null);
|
|
276
|
+
}
|
|
277
|
+
|
|
278
|
+
await page.bringToFront?.().catch(() => null);
|
|
279
|
+
await session.detach?.().catch(() => null);
|
|
280
|
+
return true;
|
|
281
|
+
} catch {
|
|
282
|
+
await page.bringToFront?.().catch(() => null);
|
|
283
|
+
return true;
|
|
284
|
+
}
|
|
285
|
+
}
|
|
286
|
+
|
|
287
|
+
export class KiroBulkImportManager {
|
|
288
|
+
constructor({
|
|
289
|
+
browserLauncher = defaultBrowserLauncher,
|
|
290
|
+
googleAutomation = runKiroGoogleAutomation,
|
|
291
|
+
socialExchange = defaultSocialExchange,
|
|
292
|
+
kiroServiceFactory = () => new KiroService(),
|
|
293
|
+
storageName = "kiro-bulk-import",
|
|
294
|
+
} = {}) {
|
|
295
|
+
this.browserLauncher = browserLauncher;
|
|
296
|
+
this.googleAutomation = googleAutomation;
|
|
297
|
+
this.socialExchange = socialExchange;
|
|
298
|
+
this.kiroServiceFactory = kiroServiceFactory;
|
|
299
|
+
this.storageDir = path.join(DATA_DIR, storageName);
|
|
300
|
+
this.metaFile = path.join(this.storageDir, "meta.json");
|
|
301
|
+
this.jobs = new Map();
|
|
302
|
+
this.latestJobId = readPersistedLatestJobId(this.metaFile);
|
|
303
|
+
}
|
|
304
|
+
|
|
305
|
+
async startJob({ accounts, concurrency }) {
|
|
306
|
+
const { parsed, invalidLines } = parseKiroBulkAccounts(accounts);
|
|
307
|
+
if (!parsed.length) {
|
|
308
|
+
const error = invalidLines.length > 0
|
|
309
|
+
? "Invalid account format. Use one account per line: gmail@example.com|password"
|
|
310
|
+
: "At least one account entry is required";
|
|
311
|
+
const response = { error };
|
|
312
|
+
if (invalidLines.length > 0) response.invalidLines = invalidLines;
|
|
313
|
+
throw Object.assign(new Error(error), response);
|
|
314
|
+
}
|
|
315
|
+
|
|
316
|
+
if (invalidLines.length > 0) {
|
|
317
|
+
const error = "Invalid account format. Use one account per line: gmail@example.com|password";
|
|
318
|
+
throw Object.assign(new Error(error), { error, invalidLines });
|
|
319
|
+
}
|
|
320
|
+
|
|
321
|
+
const jobId = randomUUID();
|
|
322
|
+
const createdAt = nowIso();
|
|
323
|
+
const job = {
|
|
324
|
+
jobId,
|
|
325
|
+
status: "running",
|
|
326
|
+
concurrency: clampConcurrency(concurrency),
|
|
327
|
+
createdAt,
|
|
328
|
+
startedAt: createdAt,
|
|
329
|
+
finishedAt: null,
|
|
330
|
+
error: null,
|
|
331
|
+
cancelRequested: false,
|
|
332
|
+
browser: null,
|
|
333
|
+
nextIndex: 0,
|
|
334
|
+
manualFollowups: new Set(),
|
|
335
|
+
persistPromise: Promise.resolve(),
|
|
336
|
+
lastPreview: null,
|
|
337
|
+
lastPreviewCapturedAt: 0,
|
|
338
|
+
accounts: parsed.map((account) => ({
|
|
339
|
+
line: account.line,
|
|
340
|
+
email: account.email,
|
|
341
|
+
password: account.password,
|
|
342
|
+
status: "queued",
|
|
343
|
+
error: null,
|
|
344
|
+
connectionId: null,
|
|
345
|
+
workerId: null,
|
|
346
|
+
manualSession: null,
|
|
347
|
+
runtimeSession: null,
|
|
348
|
+
currentStep: "queued",
|
|
349
|
+
updatedAt: createdAt,
|
|
350
|
+
logs: [createLogEntry("queued", "Queued and waiting for an available worker")],
|
|
351
|
+
})),
|
|
352
|
+
};
|
|
353
|
+
|
|
354
|
+
this.jobs.set(jobId, job);
|
|
355
|
+
this.latestJobId = jobId;
|
|
356
|
+
writePersistedLatestJobId(jobId, this.metaFile);
|
|
357
|
+
await this.persistJobSnapshot(job, { forcePreview: false });
|
|
358
|
+
void this.runJob(jobId);
|
|
359
|
+
return sanitizeJob(job);
|
|
360
|
+
}
|
|
361
|
+
|
|
362
|
+
getJob(jobId) {
|
|
363
|
+
const job = this.jobs.get(jobId);
|
|
364
|
+
if (job) return sanitizeJob(job, { preview: job.lastPreview || null });
|
|
365
|
+
return readJsonFile(getJobFile(jobId, this.storageDir));
|
|
366
|
+
}
|
|
367
|
+
|
|
368
|
+
async getJobWithPreview(jobId) {
|
|
369
|
+
const job = this.jobs.get(jobId);
|
|
370
|
+
if (!job) return readJsonFile(getJobFile(jobId, this.storageDir));
|
|
371
|
+
const preview = await this.capturePreview(job);
|
|
372
|
+
job.lastPreview = preview || job.lastPreview || null;
|
|
373
|
+
await this.persistJobSnapshot(job, { forcePreview: false });
|
|
374
|
+
return sanitizeJob(job, { preview: job.lastPreview || null });
|
|
375
|
+
}
|
|
376
|
+
|
|
377
|
+
async getLatestJobWithPreview({ includeRecentTerminal = false } = {}) {
|
|
378
|
+
const latestJobId = this.latestJobId || readPersistedLatestJobId(this.metaFile);
|
|
379
|
+
if (!latestJobId) return null;
|
|
380
|
+
const job = await this.getJobWithPreview(latestJobId);
|
|
381
|
+
if (!job) return null;
|
|
382
|
+
if (ACTIVE_JOB_STATUSES.has(job.status)) {
|
|
383
|
+
return job;
|
|
384
|
+
}
|
|
385
|
+
if (includeRecentTerminal && isRecentTerminalJob(job)) {
|
|
386
|
+
return job;
|
|
387
|
+
}
|
|
388
|
+
return null;
|
|
389
|
+
}
|
|
390
|
+
|
|
391
|
+
cancelJob(jobId) {
|
|
392
|
+
const job = this.jobs.get(jobId);
|
|
393
|
+
if (!job) return readJsonFile(getJobFile(jobId, this.storageDir));
|
|
394
|
+
|
|
395
|
+
job.cancelRequested = true;
|
|
396
|
+
if (job.status === "queued") {
|
|
397
|
+
job.status = "cancelled";
|
|
398
|
+
job.finishedAt = nowIso();
|
|
399
|
+
job.accounts.forEach((account) => {
|
|
400
|
+
if (account.status === "queued") account.status = "cancelled";
|
|
401
|
+
});
|
|
402
|
+
}
|
|
403
|
+
|
|
404
|
+
if (job.browser) {
|
|
405
|
+
void job.browser.close().catch(() => null);
|
|
406
|
+
job.browser = null;
|
|
407
|
+
}
|
|
408
|
+
|
|
409
|
+
void this.persistJobSnapshot(job, { forcePreview: true });
|
|
410
|
+
|
|
411
|
+
return sanitizeJob(job);
|
|
412
|
+
}
|
|
413
|
+
|
|
414
|
+
async openManualSession(jobId, workerId) {
|
|
415
|
+
const job = this.jobs.get(jobId);
|
|
416
|
+
if (!job) return null;
|
|
417
|
+
|
|
418
|
+
const numericWorkerId = Number.parseInt(workerId, 10);
|
|
419
|
+
const account = job.accounts.find((entry) => (
|
|
420
|
+
entry.workerId === numericWorkerId
|
|
421
|
+
&& entry.status === "needs_manual"
|
|
422
|
+
&& entry.manualSession?.page
|
|
423
|
+
));
|
|
424
|
+
|
|
425
|
+
if (!account) {
|
|
426
|
+
return {
|
|
427
|
+
ok: false,
|
|
428
|
+
error: "Manual session not found for this worker",
|
|
429
|
+
job: sanitizeJob(job),
|
|
430
|
+
};
|
|
431
|
+
}
|
|
432
|
+
|
|
433
|
+
const opened = await revealBrowserWindow(account.manualSession.page);
|
|
434
|
+
account.manualSession.opened = opened;
|
|
435
|
+
account.manualSession.openedAt = opened
|
|
436
|
+
? (account.manualSession.openedAt || nowIso())
|
|
437
|
+
: account.manualSession.openedAt || null;
|
|
438
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
439
|
+
|
|
440
|
+
return {
|
|
441
|
+
ok: true,
|
|
442
|
+
job: sanitizeJob(job),
|
|
443
|
+
account: sanitizeAccount(account),
|
|
444
|
+
};
|
|
445
|
+
}
|
|
446
|
+
|
|
447
|
+
dequeueAccount(job, workerId) {
|
|
448
|
+
while (job.nextIndex < job.accounts.length) {
|
|
449
|
+
const account = job.accounts[job.nextIndex];
|
|
450
|
+
job.nextIndex += 1;
|
|
451
|
+
if (account.status !== "queued") continue;
|
|
452
|
+
account.status = "running";
|
|
453
|
+
account.workerId = workerId;
|
|
454
|
+
account.error = null;
|
|
455
|
+
appendAccountLog(account, "worker_assigned", `Worker ${workerId} picked up this account`);
|
|
456
|
+
void this.persistJobSnapshot(job, { forcePreview: false });
|
|
457
|
+
return account;
|
|
458
|
+
}
|
|
459
|
+
return null;
|
|
460
|
+
}
|
|
461
|
+
|
|
462
|
+
finalizeAccount(account, status, extras = {}) {
|
|
463
|
+
account.status = status;
|
|
464
|
+
account.error = extras.error || null;
|
|
465
|
+
account.connectionId = extras.connectionId || null;
|
|
466
|
+
if (extras.step || extras.message) {
|
|
467
|
+
appendAccountLog(
|
|
468
|
+
account,
|
|
469
|
+
extras.step || status,
|
|
470
|
+
extras.message || extras.error || status.replaceAll("_", " ")
|
|
471
|
+
);
|
|
472
|
+
}
|
|
473
|
+
return account;
|
|
474
|
+
}
|
|
475
|
+
|
|
476
|
+
setAccountStep(account, step, message, level = "info") {
|
|
477
|
+
appendAccountLog(account, step, message, level);
|
|
478
|
+
}
|
|
479
|
+
|
|
480
|
+
async persistJobSnapshot(job, { forcePreview = false } = {}) {
|
|
481
|
+
if (!job) return;
|
|
482
|
+
|
|
483
|
+
const runPersist = async () => {
|
|
484
|
+
const shouldCapturePreview = forcePreview || (Date.now() - (job.lastPreviewCapturedAt || 0) >= PREVIEW_CAPTURE_INTERVAL_MS);
|
|
485
|
+
if (shouldCapturePreview) {
|
|
486
|
+
const preview = await this.capturePreview(job);
|
|
487
|
+
if (preview) {
|
|
488
|
+
job.lastPreview = preview;
|
|
489
|
+
}
|
|
490
|
+
job.lastPreviewCapturedAt = Date.now();
|
|
491
|
+
}
|
|
492
|
+
|
|
493
|
+
writeJsonFile(getJobFile(job.jobId, this.storageDir), buildPersistedSnapshot(job));
|
|
494
|
+
};
|
|
495
|
+
|
|
496
|
+
job.persistPromise = Promise.resolve(job.persistPromise).catch(() => null).then(runPersist);
|
|
497
|
+
await job.persistPromise;
|
|
498
|
+
}
|
|
499
|
+
|
|
500
|
+
async capturePreview(job) {
|
|
501
|
+
const previewAccount = job.accounts.find((account) => account.status === "running" && account.runtimeSession?.page)
|
|
502
|
+
|| job.accounts.find((account) => account.status === "needs_manual" && account.manualSession?.page);
|
|
503
|
+
|
|
504
|
+
if (!previewAccount) return null;
|
|
505
|
+
|
|
506
|
+
const page = previewAccount.runtimeSession?.page || previewAccount.manualSession?.page;
|
|
507
|
+
if (!page) return null;
|
|
508
|
+
|
|
509
|
+
try {
|
|
510
|
+
const screenshot = await page.screenshot({
|
|
511
|
+
type: "jpeg",
|
|
512
|
+
quality: 55,
|
|
513
|
+
fullPage: false,
|
|
514
|
+
animations: "disabled",
|
|
515
|
+
caret: "hide",
|
|
516
|
+
});
|
|
517
|
+
|
|
518
|
+
return {
|
|
519
|
+
email: previewAccount.email,
|
|
520
|
+
workerId: previewAccount.workerId || null,
|
|
521
|
+
status: previewAccount.status,
|
|
522
|
+
step: previewAccount.currentStep || null,
|
|
523
|
+
updatedAt: previewAccount.updatedAt || nowIso(),
|
|
524
|
+
imageData: `data:image/jpeg;base64,${screenshot.toString("base64")}`,
|
|
525
|
+
};
|
|
526
|
+
} catch {
|
|
527
|
+
return {
|
|
528
|
+
email: previewAccount.email,
|
|
529
|
+
workerId: previewAccount.workerId || null,
|
|
530
|
+
status: previewAccount.status,
|
|
531
|
+
step: previewAccount.currentStep || null,
|
|
532
|
+
updatedAt: previewAccount.updatedAt || nowIso(),
|
|
533
|
+
imageData: null,
|
|
534
|
+
};
|
|
535
|
+
}
|
|
536
|
+
}
|
|
537
|
+
|
|
538
|
+
async runManualFollowup(job, account, workerId, context, callbackPromise, codeVerifier) {
|
|
539
|
+
const followupPromise = (async () => {
|
|
540
|
+
try {
|
|
541
|
+
const callback = await callbackPromise;
|
|
542
|
+
if (job.cancelRequested) {
|
|
543
|
+
this.finalizeAccount(account, "cancelled", {
|
|
544
|
+
error: "Job cancelled",
|
|
545
|
+
step: "cancelled",
|
|
546
|
+
message: "Job cancelled while waiting for manual completion",
|
|
547
|
+
});
|
|
548
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
549
|
+
return;
|
|
550
|
+
}
|
|
551
|
+
|
|
552
|
+
this.setAccountStep(account, "exchanging_tokens", "Exchanging Kiro callback for OAuth tokens");
|
|
553
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
554
|
+
const { connection } = await this.socialExchange({
|
|
555
|
+
code: callback.code,
|
|
556
|
+
codeVerifier,
|
|
557
|
+
provider: "google",
|
|
558
|
+
});
|
|
559
|
+
|
|
560
|
+
this.finalizeAccount(account, "success", {
|
|
561
|
+
connectionId: connection.id,
|
|
562
|
+
step: "connection_saved",
|
|
563
|
+
message: "Kiro connection saved successfully",
|
|
564
|
+
});
|
|
565
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
566
|
+
} catch (error) {
|
|
567
|
+
if (job.cancelRequested) {
|
|
568
|
+
this.finalizeAccount(account, "cancelled", {
|
|
569
|
+
error: "Job cancelled",
|
|
570
|
+
step: "cancelled",
|
|
571
|
+
message: "Job cancelled while waiting for manual completion",
|
|
572
|
+
});
|
|
573
|
+
} else {
|
|
574
|
+
this.finalizeAccount(account, "failed_exchange", {
|
|
575
|
+
error: error.message || "Manual assist flow failed during token exchange.",
|
|
576
|
+
step: "exchange_failed",
|
|
577
|
+
message: error.message || "Manual assist flow failed during token exchange.",
|
|
578
|
+
});
|
|
579
|
+
}
|
|
580
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
581
|
+
} finally {
|
|
582
|
+
account.manualSession = null;
|
|
583
|
+
account.runtimeSession = null;
|
|
584
|
+
await context.close().catch(() => null);
|
|
585
|
+
job.manualFollowups.delete(followupPromise);
|
|
586
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
587
|
+
}
|
|
588
|
+
})();
|
|
589
|
+
|
|
590
|
+
job.manualFollowups.add(followupPromise);
|
|
591
|
+
}
|
|
592
|
+
|
|
593
|
+
async processAccount(job, account, workerId) {
|
|
594
|
+
if (job.cancelRequested || !job.browser) {
|
|
595
|
+
this.finalizeAccount(account, "cancelled", { error: "Job cancelled" });
|
|
596
|
+
return;
|
|
597
|
+
}
|
|
598
|
+
|
|
599
|
+
const kiroService = this.kiroServiceFactory();
|
|
600
|
+
const socialAuth = kiroService.createSocialAuthorization("google");
|
|
601
|
+
const { context, page } = await createFreshContext(job.browser);
|
|
602
|
+
const callbackPromise = createKiroCallbackMonitor(context, page);
|
|
603
|
+
account.runtimeSession = { context, page };
|
|
604
|
+
|
|
605
|
+
try {
|
|
606
|
+
this.setAccountStep(account, "preparing_worker", `Worker ${workerId} is preparing a browser context`);
|
|
607
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
608
|
+
const automationResult = await this.googleAutomation({
|
|
609
|
+
page,
|
|
610
|
+
authUrl: socialAuth.authUrl,
|
|
611
|
+
email: account.email,
|
|
612
|
+
password: account.password,
|
|
613
|
+
callbackPromise,
|
|
614
|
+
onStep: (step, message) => {
|
|
615
|
+
this.setAccountStep(account, step, message);
|
|
616
|
+
void this.persistJobSnapshot(job, { forcePreview: false });
|
|
617
|
+
},
|
|
618
|
+
});
|
|
619
|
+
|
|
620
|
+
if (automationResult.status === "success") {
|
|
621
|
+
this.setAccountStep(account, "exchanging_tokens", "Exchanging Kiro callback for OAuth tokens");
|
|
622
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
623
|
+
const { connection } = await this.socialExchange({
|
|
624
|
+
code: automationResult.code,
|
|
625
|
+
codeVerifier: socialAuth.codeVerifier,
|
|
626
|
+
provider: "google",
|
|
627
|
+
});
|
|
628
|
+
this.finalizeAccount(account, "success", {
|
|
629
|
+
connectionId: connection.id,
|
|
630
|
+
step: "connection_saved",
|
|
631
|
+
message: "Kiro connection saved successfully",
|
|
632
|
+
});
|
|
633
|
+
account.runtimeSession = null;
|
|
634
|
+
await context.close().catch(() => null);
|
|
635
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
636
|
+
return;
|
|
637
|
+
}
|
|
638
|
+
|
|
639
|
+
if (automationResult.status === "needs_manual") {
|
|
640
|
+
account.manualSession = {
|
|
641
|
+
context,
|
|
642
|
+
page,
|
|
643
|
+
opened: false,
|
|
644
|
+
openedAt: null,
|
|
645
|
+
};
|
|
646
|
+
this.setAccountStep(account, "awaiting_manual", "Waiting for manual completion in the browser session");
|
|
647
|
+
this.finalizeAccount(account, "needs_manual", {
|
|
648
|
+
error: automationResult.error,
|
|
649
|
+
step: "awaiting_manual",
|
|
650
|
+
message: automationResult.error,
|
|
651
|
+
});
|
|
652
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
653
|
+
await this.runManualFollowup(
|
|
654
|
+
job,
|
|
655
|
+
account,
|
|
656
|
+
workerId,
|
|
657
|
+
context,
|
|
658
|
+
callbackPromise,
|
|
659
|
+
socialAuth.codeVerifier
|
|
660
|
+
);
|
|
661
|
+
return;
|
|
662
|
+
}
|
|
663
|
+
|
|
664
|
+
const terminalStatus = TERMINAL_ACCOUNT_STATUSES.has(automationResult.status)
|
|
665
|
+
? automationResult.status
|
|
666
|
+
: "failed";
|
|
667
|
+
this.finalizeAccount(account, terminalStatus, {
|
|
668
|
+
error: automationResult.error || "Kiro Google automation failed.",
|
|
669
|
+
step: terminalStatus,
|
|
670
|
+
message: automationResult.error || "Kiro Google automation failed.",
|
|
671
|
+
});
|
|
672
|
+
account.runtimeSession = null;
|
|
673
|
+
await context.close().catch(() => null);
|
|
674
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
675
|
+
} catch (error) {
|
|
676
|
+
this.finalizeAccount(account, "failed", {
|
|
677
|
+
error: error.message || "Unexpected Kiro bulk import failure.",
|
|
678
|
+
step: "failed",
|
|
679
|
+
message: error.message || "Unexpected Kiro bulk import failure.",
|
|
680
|
+
});
|
|
681
|
+
account.runtimeSession = null;
|
|
682
|
+
await context.close().catch(() => null);
|
|
683
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
684
|
+
} finally {
|
|
685
|
+
account.password = undefined;
|
|
686
|
+
}
|
|
687
|
+
}
|
|
688
|
+
|
|
689
|
+
async runWorker(job, workerId) {
|
|
690
|
+
while (!job.cancelRequested) {
|
|
691
|
+
const account = this.dequeueAccount(job, workerId);
|
|
692
|
+
if (!account) return;
|
|
693
|
+
await this.processAccount(job, account, workerId);
|
|
694
|
+
}
|
|
695
|
+
}
|
|
696
|
+
|
|
697
|
+
async runJob(jobId) {
|
|
698
|
+
const job = this.jobs.get(jobId);
|
|
699
|
+
if (!job) return;
|
|
700
|
+
|
|
701
|
+
try {
|
|
702
|
+
job.browser = await this.browserLauncher();
|
|
703
|
+
job.accounts.forEach((account) => {
|
|
704
|
+
if (account.status === "queued" && (account.logs || []).length === 1) {
|
|
705
|
+
this.setAccountStep(account, "waiting_for_worker", "Waiting for a free worker");
|
|
706
|
+
}
|
|
707
|
+
});
|
|
708
|
+
await this.persistJobSnapshot(job, { forcePreview: false });
|
|
709
|
+
const workerCount = Math.min(job.concurrency, Math.max(job.accounts.length, 1));
|
|
710
|
+
const workers = Array.from({ length: workerCount }, (_, index) => this.runWorker(job, index + 1));
|
|
711
|
+
|
|
712
|
+
await Promise.allSettled(workers);
|
|
713
|
+
|
|
714
|
+
if (job.manualFollowups.size > 0) {
|
|
715
|
+
await Promise.allSettled([...job.manualFollowups]);
|
|
716
|
+
}
|
|
717
|
+
|
|
718
|
+
if (job.cancelRequested) {
|
|
719
|
+
job.status = "cancelled";
|
|
720
|
+
job.accounts.forEach((account) => {
|
|
721
|
+
if (account.status === "queued" || account.status === "running") {
|
|
722
|
+
this.finalizeAccount(account, "cancelled", {
|
|
723
|
+
error: "Job cancelled",
|
|
724
|
+
step: "cancelled",
|
|
725
|
+
message: "Job cancelled before completion",
|
|
726
|
+
});
|
|
727
|
+
}
|
|
728
|
+
});
|
|
729
|
+
} else {
|
|
730
|
+
job.status = "completed";
|
|
731
|
+
}
|
|
732
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
733
|
+
} catch (error) {
|
|
734
|
+
job.status = "failed";
|
|
735
|
+
job.error = error.message || "Failed to start Kiro bulk import job.";
|
|
736
|
+
job.accounts.forEach((account) => {
|
|
737
|
+
if (account.status === "queued" || account.status === "running") {
|
|
738
|
+
this.finalizeAccount(account, "failed", {
|
|
739
|
+
error: job.error,
|
|
740
|
+
step: "failed",
|
|
741
|
+
message: job.error,
|
|
742
|
+
});
|
|
743
|
+
account.password = undefined;
|
|
744
|
+
}
|
|
745
|
+
});
|
|
746
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
747
|
+
} finally {
|
|
748
|
+
if (job.browser) {
|
|
749
|
+
await job.browser.close().catch(() => null);
|
|
750
|
+
job.browser = null;
|
|
751
|
+
}
|
|
752
|
+
job.finishedAt = nowIso();
|
|
753
|
+
await this.persistJobSnapshot(job, { forcePreview: true });
|
|
754
|
+
}
|
|
755
|
+
}
|
|
756
|
+
}
|
|
757
|
+
|
|
758
|
+
function getSingletonStore() {
|
|
759
|
+
if (!globalThis.__kiroBulkImportSingleton) {
|
|
760
|
+
globalThis.__kiroBulkImportSingleton = {
|
|
761
|
+
manager: new KiroBulkImportManager(),
|
|
762
|
+
};
|
|
763
|
+
}
|
|
764
|
+
return globalThis.__kiroBulkImportSingleton;
|
|
765
|
+
}
|
|
766
|
+
|
|
767
|
+
export function getKiroBulkImportManager() {
|
|
768
|
+
return getSingletonStore().manager;
|
|
769
|
+
}
|
|
770
|
+
|
|
771
|
+
export const __test__ = {
|
|
772
|
+
clampConcurrency,
|
|
773
|
+
parseKiroBulkAccounts,
|
|
774
|
+
sanitizeJob,
|
|
775
|
+
buildSummary,
|
|
776
|
+
isRecentTerminalJob,
|
|
777
|
+
buildLookupResponse,
|
|
778
|
+
};
|