poe-code 0.1.26 → 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/README.md +110 -29
- package/dist/bin/poe-claude.js +16 -0
- package/dist/bin/poe-codex.js +16 -0
- package/dist/bin/poe-kimi-cli.js +16 -0
- package/dist/bin/poe-opencode.js +16 -0
- package/dist/cli/binary-aliases.d.ts +7 -0
- package/dist/cli/binary-aliases.js +26 -0
- package/dist/cli/binary-aliases.js.map +1 -0
- package/dist/cli/bootstrap.d.ts +4 -0
- package/dist/cli/bootstrap.js +78 -0
- package/dist/cli/bootstrap.js.map +1 -0
- package/dist/cli/command-runner.d.ts +1 -1
- package/dist/cli/command-runner.js +16 -2
- package/dist/cli/command-runner.js.map +1 -1
- package/dist/cli/commands/configure-payload.d.ts +15 -0
- package/dist/cli/commands/configure-payload.js +31 -0
- package/dist/cli/commands/configure-payload.js.map +1 -0
- package/dist/cli/commands/configure.d.ts +4 -3
- package/dist/cli/commands/configure.js +102 -101
- package/dist/cli/commands/configure.js.map +1 -1
- package/dist/cli/commands/ensure-isolated-config.d.ts +12 -0
- package/dist/cli/commands/ensure-isolated-config.js +53 -0
- package/dist/cli/commands/ensure-isolated-config.js.map +1 -0
- package/dist/cli/commands/generate.d.ts +9 -0
- package/dist/cli/commands/generate.js +323 -0
- package/dist/cli/commands/generate.js.map +1 -0
- package/dist/cli/commands/install.d.ts +4 -0
- package/dist/cli/commands/install.js +40 -0
- package/dist/cli/commands/install.js.map +1 -0
- package/dist/cli/commands/login.js +62 -8
- package/dist/cli/commands/login.js.map +1 -1
- package/dist/cli/commands/{root.d.ts → mcp.d.ts} +1 -1
- package/dist/cli/commands/mcp.js +168 -0
- package/dist/cli/commands/mcp.js.map +1 -0
- package/dist/cli/commands/shared.d.ts +16 -11
- package/dist/cli/commands/shared.js +45 -24
- package/dist/cli/commands/shared.js.map +1 -1
- package/dist/cli/commands/spawn.d.ts +14 -4
- package/dist/cli/commands/spawn.js +92 -77
- package/dist/cli/commands/spawn.js.map +1 -1
- package/dist/cli/commands/test.d.ts +6 -4
- package/dist/cli/commands/test.js +111 -16
- package/dist/cli/commands/test.js.map +1 -1
- package/dist/cli/commands/unconfigure.d.ts +7 -0
- package/dist/cli/commands/unconfigure.js +116 -0
- package/dist/cli/commands/unconfigure.js.map +1 -0
- package/dist/cli/commands/version.d.ts +3 -0
- package/dist/cli/commands/version.js +31 -0
- package/dist/cli/commands/version.js.map +1 -0
- package/dist/cli/commands/wrap.d.ts +3 -0
- package/dist/cli/commands/wrap.js +61 -0
- package/dist/cli/commands/wrap.js.map +1 -0
- package/dist/cli/constants.d.ts +23 -9
- package/dist/cli/constants.js +42 -9
- package/dist/cli/constants.js.map +1 -1
- package/dist/cli/container.d.ts +4 -10
- package/dist/cli/container.js +18 -22
- package/dist/cli/container.js.map +1 -1
- package/dist/cli/context.d.ts +9 -13
- package/dist/cli/context.js +45 -152
- package/dist/cli/context.js.map +1 -1
- package/dist/cli/environment.d.ts +2 -0
- package/dist/cli/environment.js +80 -0
- package/dist/cli/environment.js.map +1 -1
- package/dist/cli/error-logger.d.ts +18 -2
- package/dist/cli/error-logger.js +25 -7
- package/dist/cli/error-logger.js.map +1 -1
- package/dist/cli/errors.d.ts +14 -10
- package/dist/cli/errors.js +25 -12
- package/dist/cli/errors.js.map +1 -1
- package/dist/cli/exit-signals.d.ts +4 -0
- package/dist/cli/exit-signals.js +8 -0
- package/dist/cli/exit-signals.js.map +1 -0
- package/dist/cli/http.d.ts +1 -0
- package/dist/cli/isolated-env-runner.d.ts +10 -0
- package/dist/cli/isolated-env-runner.js +46 -0
- package/dist/cli/isolated-env-runner.js.map +1 -0
- package/dist/cli/isolated-env.d.ts +22 -0
- package/dist/cli/isolated-env.js +176 -0
- package/dist/cli/isolated-env.js.map +1 -0
- package/dist/cli/logger.d.ts +11 -1
- package/dist/cli/logger.js +101 -16
- package/dist/cli/logger.js.map +1 -1
- package/dist/cli/mcp-output-format.d.ts +2 -0
- package/dist/cli/mcp-output-format.js +22 -0
- package/dist/cli/mcp-output-format.js.map +1 -0
- package/dist/cli/mcp-server.d.ts +21 -0
- package/dist/cli/mcp-server.js +286 -0
- package/dist/cli/mcp-server.js.map +1 -0
- package/dist/cli/options.d.ts +18 -3
- package/dist/cli/options.js +39 -12
- package/dist/cli/options.js.map +1 -1
- package/dist/cli/poe-code-command-runner.d.ts +6 -0
- package/dist/cli/poe-code-command-runner.js +57 -0
- package/dist/cli/poe-code-command-runner.js.map +1 -0
- package/dist/cli/program.js +134 -23
- package/dist/cli/program.js.map +1 -1
- package/dist/cli/prompt-runner.d.ts +10 -0
- package/dist/cli/prompt-runner.js +79 -0
- package/dist/cli/prompt-runner.js.map +1 -0
- package/dist/cli/prompts.d.ts +23 -7
- package/dist/cli/prompts.js +17 -33
- package/dist/cli/prompts.js.map +1 -1
- package/dist/cli/service-registry.d.ts +86 -21
- package/dist/cli/service-registry.js +60 -12
- package/dist/cli/service-registry.js.map +1 -1
- package/dist/cli/ui/service-menu.d.ts +7 -0
- package/dist/cli/ui/service-menu.js +43 -0
- package/dist/cli/ui/service-menu.js.map +1 -0
- package/dist/index.d.ts +6 -2
- package/dist/index.js +7 -74
- package/dist/index.js.map +1 -1
- package/dist/providers/claude-code.d.ts +13 -0
- package/dist/providers/claude-code.js +136 -0
- package/dist/providers/claude-code.js.map +1 -0
- package/dist/providers/codex.d.ts +17 -0
- package/dist/providers/codex.js +144 -0
- package/dist/providers/codex.js.map +1 -0
- package/dist/providers/create-provider.d.ts +29 -0
- package/dist/providers/create-provider.js +63 -0
- package/dist/providers/create-provider.js.map +1 -0
- package/dist/providers/index.d.ts +2 -2
- package/dist/providers/index.js +5 -10
- package/dist/providers/index.js.map +1 -1
- package/dist/providers/kimi.d.ts +4 -0
- package/dist/providers/kimi.js +123 -0
- package/dist/providers/kimi.js.map +1 -0
- package/dist/providers/opencode.d.ts +3 -0
- package/dist/providers/opencode.js +119 -0
- package/dist/providers/opencode.js.map +1 -0
- package/dist/providers/spawn-options.d.ts +12 -0
- package/dist/providers/spawn-options.js +2 -0
- package/dist/providers/spawn-options.js.map +1 -0
- package/dist/sdk/container.d.ts +18 -0
- package/dist/sdk/container.js +127 -0
- package/dist/sdk/container.js.map +1 -0
- package/dist/sdk/credentials.d.ts +9 -0
- package/dist/sdk/credentials.js +35 -0
- package/dist/sdk/credentials.js.map +1 -0
- package/dist/sdk/generate.d.ts +5 -0
- package/dist/sdk/generate.js +97 -0
- package/dist/sdk/generate.js.map +1 -0
- package/dist/sdk/spawn-core.d.ts +23 -0
- package/dist/sdk/spawn-core.js +105 -0
- package/dist/sdk/spawn-core.js.map +1 -0
- package/dist/sdk/spawn.d.ts +23 -0
- package/dist/sdk/spawn.js +41 -0
- package/dist/sdk/spawn.js.map +1 -0
- package/dist/sdk/types.d.ts +38 -0
- package/dist/sdk/types.js +2 -0
- package/dist/sdk/types.js.map +1 -0
- package/dist/services/client-instance.d.ts +12 -0
- package/dist/services/client-instance.js +36 -0
- package/dist/services/client-instance.js.map +1 -0
- package/dist/services/credentials.d.ts +13 -0
- package/dist/services/credentials.js +132 -14
- package/dist/services/credentials.js.map +1 -1
- package/dist/services/llm-client.d.ts +22 -0
- package/dist/services/llm-client.js +138 -0
- package/dist/services/llm-client.js.map +1 -0
- package/dist/services/media-download.d.ts +17 -0
- package/dist/services/media-download.js +42 -0
- package/dist/services/media-download.js.map +1 -0
- package/dist/services/model-strategy.d.ts +2 -2
- package/dist/services/model-strategy.js +36 -31
- package/dist/services/model-strategy.js.map +1 -1
- package/dist/services/mutation-events.d.ts +4 -0
- package/dist/services/mutation-events.js +53 -0
- package/dist/services/mutation-events.js.map +1 -0
- package/dist/services/service-install.d.ts +5 -4
- package/dist/services/service-install.js +36 -13
- package/dist/services/service-install.js.map +1 -1
- package/dist/services/version.d.ts +11 -0
- package/dist/services/version.js +28 -0
- package/dist/services/version.js.map +1 -0
- package/dist/templates/codex/config.toml.hbs +3 -3
- package/dist/tools/label-generator.d.ts +11 -0
- package/dist/tools/label-generator.js +78 -0
- package/dist/tools/label-generator.js.map +1 -0
- package/dist/utils/command-checks.d.ts +45 -0
- package/dist/utils/command-checks.js +120 -0
- package/dist/utils/command-checks.js.map +1 -0
- package/dist/utils/dry-run.d.ts +1 -0
- package/dist/utils/dry-run.js +116 -7
- package/dist/utils/dry-run.js.map +1 -1
- package/dist/utils/execution-context.d.ts +40 -0
- package/dist/utils/execution-context.js +128 -0
- package/dist/utils/execution-context.js.map +1 -0
- package/dist/utils/json.js +3 -0
- package/dist/utils/json.js.map +1 -1
- package/dist/utils/templates.d.ts +1 -0
- package/dist/utils/templates.js +3 -4
- package/dist/utils/templates.js.map +1 -1
- package/package.json +52 -46
- package/dist/cli/api-client.d.ts +0 -14
- package/dist/cli/api-client.js +0 -140
- package/dist/cli/api-client.js.map +0 -1
- package/dist/cli/chat.d.ts +0 -32
- package/dist/cli/chat.js +0 -2
- package/dist/cli/chat.js.map +0 -1
- package/dist/cli/commands/agent.d.ts +0 -18
- package/dist/cli/commands/agent.js +0 -84
- package/dist/cli/commands/agent.js.map +0 -1
- package/dist/cli/commands/configure-agents.d.ts +0 -3
- package/dist/cli/commands/configure-agents.js +0 -72
- package/dist/cli/commands/configure-agents.js.map +0 -1
- package/dist/cli/commands/init.d.ts +0 -8
- package/dist/cli/commands/init.js +0 -45
- package/dist/cli/commands/init.js.map +0 -1
- package/dist/cli/commands/interactive.d.ts +0 -3
- package/dist/cli/commands/interactive.js +0 -17
- package/dist/cli/commands/interactive.js.map +0 -1
- package/dist/cli/commands/logout.d.ts +0 -3
- package/dist/cli/commands/logout.js +0 -31
- package/dist/cli/commands/logout.js.map +0 -1
- package/dist/cli/commands/prerequisites.d.ts +0 -3
- package/dist/cli/commands/prerequisites.js +0 -23
- package/dist/cli/commands/prerequisites.js.map +0 -1
- package/dist/cli/commands/query.d.ts +0 -7
- package/dist/cli/commands/query.js +0 -46
- package/dist/cli/commands/query.js.map +0 -1
- package/dist/cli/commands/remove.d.ts +0 -6
- package/dist/cli/commands/remove.js +0 -80
- package/dist/cli/commands/remove.js.map +0 -1
- package/dist/cli/commands/root.js +0 -7
- package/dist/cli/commands/root.js.map +0 -1
- package/dist/cli/commands/spawn-worktree.d.ts +0 -6
- package/dist/cli/commands/spawn-worktree.js +0 -91
- package/dist/cli/commands/spawn-worktree.js.map +0 -1
- package/dist/cli/file-mentions.d.ts +0 -13
- package/dist/cli/file-mentions.js +0 -87
- package/dist/cli/file-mentions.js.map +0 -1
- package/dist/cli/file-picker-state.d.ts +0 -14
- package/dist/cli/file-picker-state.js +0 -23
- package/dist/cli/file-picker-state.js.map +0 -1
- package/dist/cli/interactive-command-runner.d.ts +0 -11
- package/dist/cli/interactive-command-runner.js +0 -142
- package/dist/cli/interactive-command-runner.js.map +0 -1
- package/dist/cli/interactive-launcher.d.ts +0 -2
- package/dist/cli/interactive-launcher.js +0 -404
- package/dist/cli/interactive-launcher.js.map +0 -1
- package/dist/cli/interactive-tasks.d.ts +0 -7
- package/dist/cli/interactive-tasks.js +0 -199
- package/dist/cli/interactive-tasks.js.map +0 -1
- package/dist/cli/interactive.d.ts +0 -19
- package/dist/cli/interactive.js +0 -479
- package/dist/cli/interactive.js.map +0 -1
- package/dist/cli/markdown-renderer.d.ts +0 -27
- package/dist/cli/markdown-renderer.js +0 -178
- package/dist/cli/markdown-renderer.js.map +0 -1
- package/dist/cli/markdown-renderer.test.d.ts +0 -1
- package/dist/cli/markdown-renderer.test.js +0 -105
- package/dist/cli/markdown-renderer.test.js.map +0 -1
- package/dist/cli/telemetry.d.ts +0 -15
- package/dist/cli/telemetry.js +0 -45
- package/dist/cli/telemetry.js.map +0 -1
- package/dist/commands/init.d.ts +0 -9
- package/dist/commands/init.js +0 -43
- package/dist/commands/init.js.map +0 -1
- package/dist/commands/publish-placeholder.d.ts +0 -8
- package/dist/commands/publish-placeholder.js +0 -50
- package/dist/commands/publish-placeholder.js.map +0 -1
- package/dist/commands/spawn-worktree.d.ts +0 -25
- package/dist/commands/spawn-worktree.js +0 -139
- package/dist/commands/spawn-worktree.js.map +0 -1
- package/dist/providers/claude-code-adapter.d.ts +0 -20
- package/dist/providers/claude-code-adapter.js +0 -48
- package/dist/providers/claude-code-adapter.js.map +0 -1
- package/dist/providers/codex-adapter.d.ts +0 -19
- package/dist/providers/codex-adapter.js +0 -49
- package/dist/providers/codex-adapter.js.map +0 -1
- package/dist/providers/opencode-adapter.d.ts +0 -18
- package/dist/providers/opencode-adapter.js +0 -45
- package/dist/providers/opencode-adapter.js.map +0 -1
- package/dist/providers/roo-code-adapter.d.ts +0 -19
- package/dist/providers/roo-code-adapter.js +0 -70
- package/dist/providers/roo-code-adapter.js.map +0 -1
- package/dist/services/agent-config-manager.d.ts +0 -32
- package/dist/services/agent-config-manager.js +0 -130
- package/dist/services/agent-config-manager.js.map +0 -1
- package/dist/services/agent-registry.d.ts +0 -27
- package/dist/services/agent-registry.js +0 -89
- package/dist/services/agent-registry.js.map +0 -1
- package/dist/services/agent-session.d.ts +0 -24
- package/dist/services/agent-session.js +0 -107
- package/dist/services/agent-session.js.map +0 -1
- package/dist/services/agent-task-registry.d.ts +0 -131
- package/dist/services/agent-task-registry.js +0 -491
- package/dist/services/agent-task-registry.js.map +0 -1
- package/dist/services/chat.d.ts +0 -94
- package/dist/services/chat.js +0 -241
- package/dist/services/chat.js.map +0 -1
- package/dist/services/claude-code.d.ts +0 -28
- package/dist/services/claude-code.js +0 -155
- package/dist/services/claude-code.js.map +0 -1
- package/dist/services/codex.d.ts +0 -27
- package/dist/services/codex.js +0 -231
- package/dist/services/codex.js.map +0 -1
- package/dist/services/mcp-client.d.ts +0 -38
- package/dist/services/mcp-client.js +0 -170
- package/dist/services/mcp-client.js.map +0 -1
- package/dist/services/mcp-manager.d.ts +0 -28
- package/dist/services/mcp-manager.js +0 -157
- package/dist/services/mcp-manager.js.map +0 -1
- package/dist/services/opencode.d.ts +0 -25
- package/dist/services/opencode.js +0 -161
- package/dist/services/opencode.js.map +0 -1
- package/dist/services/poe-code.d.ts +0 -7
- package/dist/services/poe-code.js +0 -5
- package/dist/services/poe-code.js.map +0 -1
- package/dist/services/roo-code.d.ts +0 -22
- package/dist/services/roo-code.js +0 -134
- package/dist/services/roo-code.js.map +0 -1
- package/dist/services/service-manifest.d.ts +0 -122
- package/dist/services/service-manifest.js +0 -368
- package/dist/services/service-manifest.js.map +0 -1
- package/dist/services/task-logger.d.ts +0 -24
- package/dist/services/task-logger.js +0 -73
- package/dist/services/task-logger.js.map +0 -1
- package/dist/services/task-runner.d.ts +0 -19
- package/dist/services/task-runner.js +0 -157
- package/dist/services/task-runner.js.map +0 -1
- package/dist/services/tools.d.ts +0 -76
- package/dist/services/tools.js +0 -870
- package/dist/services/tools.js.map +0 -1
- package/dist/templates/claude-code/anthropic_key.sh.hbs +0 -2
- package/dist/utils/e2e-flags.d.ts +0 -1
- package/dist/utils/e2e-flags.js +0 -9
- package/dist/utils/e2e-flags.js.map +0 -1
- package/dist/utils/prerequisites.d.ts +0 -41
- package/dist/utils/prerequisites.js +0 -92
- package/dist/utils/prerequisites.js.map +0 -1
- package/dist/utils/toml.d.ts +0 -8
- package/dist/utils/toml.js +0 -30
- package/dist/utils/toml.js.map +0 -1
- package/dist/utils/worktree.d.ts +0 -40
- package/dist/utils/worktree.js +0 -179
- package/dist/utils/worktree.js.map +0 -1
- package/shared/conversation-layout.cjs +0 -11
- package/shared/conversation-layout.d.ts +0 -7
- package/shared/conversation-layout.js +0 -11
- package/shared/package.json +0 -13
package/dist/services/codex.js
DELETED
|
@@ -1,231 +0,0 @@
|
|
|
1
|
-
import path from "node:path";
|
|
2
|
-
import { createBinaryExistsCheck, formatCommandRunnerResult } from "../utils/prerequisites.js";
|
|
3
|
-
import { createBackupMutation, ensureDirectory, runServiceConfigure, runServiceRemove } from "./service-manifest.js";
|
|
4
|
-
import { parseTomlDocument, serializeTomlDocument, mergeTomlTables, isTomlTable } from "../utils/toml.js";
|
|
5
|
-
import { runServiceInstall } from "./service-install.js";
|
|
6
|
-
import { renderTemplate } from "../utils/templates.js";
|
|
7
|
-
const CODEX_PROVIDER_ID = "poe";
|
|
8
|
-
const CODEX_BASE_URL = "https://api.poe.com/v1";
|
|
9
|
-
const CODEX_TOP_LEVEL_FIELDS = [
|
|
10
|
-
"model",
|
|
11
|
-
"model_reasoning_effort"
|
|
12
|
-
];
|
|
13
|
-
const CODEX_CONFIG_TEMPLATE_ID = "codex/config.toml.hbs";
|
|
14
|
-
const CODEX_MANIFEST = {
|
|
15
|
-
id: "codex",
|
|
16
|
-
summary: "Configure Codex to use Poe as the model provider.",
|
|
17
|
-
prerequisites: {
|
|
18
|
-
after: ["codex-cli-health"]
|
|
19
|
-
},
|
|
20
|
-
configure: [
|
|
21
|
-
ensureDirectory({
|
|
22
|
-
path: ({ options }) => path.dirname(options.configPath),
|
|
23
|
-
label: "Ensure Codex config directory"
|
|
24
|
-
}),
|
|
25
|
-
createBackupMutation({
|
|
26
|
-
target: ({ options }) => options.configPath,
|
|
27
|
-
timestamp: ({ options }) => options.timestamp,
|
|
28
|
-
label: "Create Codex config backup"
|
|
29
|
-
}),
|
|
30
|
-
{
|
|
31
|
-
kind: "transformFile",
|
|
32
|
-
target: ({ options }) => options.configPath,
|
|
33
|
-
label: "Merge Codex configuration",
|
|
34
|
-
async transform({ content, context }) {
|
|
35
|
-
let document;
|
|
36
|
-
if (content == null) {
|
|
37
|
-
document = {};
|
|
38
|
-
}
|
|
39
|
-
else {
|
|
40
|
-
try {
|
|
41
|
-
document = parseTomlDocument(content);
|
|
42
|
-
}
|
|
43
|
-
catch {
|
|
44
|
-
document = {};
|
|
45
|
-
}
|
|
46
|
-
}
|
|
47
|
-
const rendered = await renderTemplate(CODEX_CONFIG_TEMPLATE_ID, { ...context.options });
|
|
48
|
-
const templateDocument = parseTomlDocument(rendered);
|
|
49
|
-
const nextTable = mergeTomlTables(document, templateDocument);
|
|
50
|
-
const nextContent = serializeTomlDocument(nextTable);
|
|
51
|
-
return {
|
|
52
|
-
content: nextContent,
|
|
53
|
-
changed: nextContent !== (content ?? "")
|
|
54
|
-
};
|
|
55
|
-
}
|
|
56
|
-
}
|
|
57
|
-
],
|
|
58
|
-
remove: [
|
|
59
|
-
{
|
|
60
|
-
kind: "transformFile",
|
|
61
|
-
target: ({ options }) => options.configPath,
|
|
62
|
-
label: "Remove Codex managed configuration",
|
|
63
|
-
transform({ content }) {
|
|
64
|
-
if (content == null) {
|
|
65
|
-
return { content: null, changed: false };
|
|
66
|
-
}
|
|
67
|
-
let document;
|
|
68
|
-
try {
|
|
69
|
-
document = parseTomlDocument(content);
|
|
70
|
-
}
|
|
71
|
-
catch {
|
|
72
|
-
return { content, changed: false };
|
|
73
|
-
}
|
|
74
|
-
const result = stripCodexConfiguration(document);
|
|
75
|
-
if (!result.changed) {
|
|
76
|
-
return { content, changed: false };
|
|
77
|
-
}
|
|
78
|
-
if (result.empty) {
|
|
79
|
-
return { content: null, changed: true };
|
|
80
|
-
}
|
|
81
|
-
const nextContent = serializeTomlDocument(document);
|
|
82
|
-
return {
|
|
83
|
-
content: nextContent,
|
|
84
|
-
changed: nextContent !== content
|
|
85
|
-
};
|
|
86
|
-
}
|
|
87
|
-
}
|
|
88
|
-
]
|
|
89
|
-
};
|
|
90
|
-
const CODEX_INSTALL_DEFINITION = {
|
|
91
|
-
id: "codex",
|
|
92
|
-
summary: "Codex CLI",
|
|
93
|
-
check: createCodexBinaryCheck(),
|
|
94
|
-
steps: [
|
|
95
|
-
{
|
|
96
|
-
id: "install-codex-cli-npm",
|
|
97
|
-
description: "Install Codex CLI via npm",
|
|
98
|
-
command: "npm",
|
|
99
|
-
args: ["install", "-g", "@openai/codex"]
|
|
100
|
-
}
|
|
101
|
-
],
|
|
102
|
-
postChecks: [createCodexVersionCheck()],
|
|
103
|
-
successMessage: "Installed Codex CLI via npm."
|
|
104
|
-
};
|
|
105
|
-
function stripCodexConfiguration(document) {
|
|
106
|
-
if (!isTomlTable(document)) {
|
|
107
|
-
return { changed: false, empty: false };
|
|
108
|
-
}
|
|
109
|
-
if (document["model_provider"] !== CODEX_PROVIDER_ID) {
|
|
110
|
-
return { changed: false, empty: false };
|
|
111
|
-
}
|
|
112
|
-
const providers = document["model_providers"];
|
|
113
|
-
if (!isTomlTable(providers)) {
|
|
114
|
-
return { changed: false, empty: false };
|
|
115
|
-
}
|
|
116
|
-
const poeConfig = providers[CODEX_PROVIDER_ID];
|
|
117
|
-
if (!isTomlTable(poeConfig) || !matchesExpectedProviderConfig(poeConfig)) {
|
|
118
|
-
return { changed: false, empty: false };
|
|
119
|
-
}
|
|
120
|
-
for (const field of CODEX_TOP_LEVEL_FIELDS) {
|
|
121
|
-
if (typeof document[field] !== "string") {
|
|
122
|
-
return { changed: false, empty: false };
|
|
123
|
-
}
|
|
124
|
-
}
|
|
125
|
-
delete document["model_provider"];
|
|
126
|
-
for (const field of CODEX_TOP_LEVEL_FIELDS) {
|
|
127
|
-
delete document[field];
|
|
128
|
-
}
|
|
129
|
-
delete providers[CODEX_PROVIDER_ID];
|
|
130
|
-
if (isTableEmpty(providers)) {
|
|
131
|
-
delete document["model_providers"];
|
|
132
|
-
}
|
|
133
|
-
return {
|
|
134
|
-
changed: true,
|
|
135
|
-
empty: isTableEmpty(document)
|
|
136
|
-
};
|
|
137
|
-
}
|
|
138
|
-
function matchesExpectedProviderConfig(table) {
|
|
139
|
-
if (table["name"] !== "poe") {
|
|
140
|
-
return false;
|
|
141
|
-
}
|
|
142
|
-
if (table["base_url"] !== CODEX_BASE_URL) {
|
|
143
|
-
return false;
|
|
144
|
-
}
|
|
145
|
-
if (table["wire_api"] !== "chat") {
|
|
146
|
-
return false;
|
|
147
|
-
}
|
|
148
|
-
const envKey = table["env_key"];
|
|
149
|
-
if (envKey != null &&
|
|
150
|
-
envKey !== "OPENAI_API_KEY" &&
|
|
151
|
-
envKey !== "POE_API_KEY") {
|
|
152
|
-
return false;
|
|
153
|
-
}
|
|
154
|
-
const bearer = table["experimental_bearer_token"];
|
|
155
|
-
if (bearer != null && typeof bearer !== "string") {
|
|
156
|
-
return false;
|
|
157
|
-
}
|
|
158
|
-
return true;
|
|
159
|
-
}
|
|
160
|
-
function isTableEmpty(value) {
|
|
161
|
-
return isTomlTable(value) && Object.keys(value).length === 0;
|
|
162
|
-
}
|
|
163
|
-
export async function configureCodex(options, runOptions) {
|
|
164
|
-
await runServiceConfigure(CODEX_MANIFEST, {
|
|
165
|
-
fs: options.fs,
|
|
166
|
-
options
|
|
167
|
-
}, runOptions);
|
|
168
|
-
}
|
|
169
|
-
export async function spawnCodex(options) {
|
|
170
|
-
const args = buildCodexExecArgs(options.prompt, options.args);
|
|
171
|
-
return options.runCommand("codex", args);
|
|
172
|
-
}
|
|
173
|
-
export async function removeCodex(options, runOptions) {
|
|
174
|
-
return runServiceRemove(CODEX_MANIFEST, {
|
|
175
|
-
fs: options.fs,
|
|
176
|
-
options
|
|
177
|
-
}, runOptions);
|
|
178
|
-
}
|
|
179
|
-
export async function installCodex(context) {
|
|
180
|
-
return runServiceInstall(CODEX_INSTALL_DEFINITION, context);
|
|
181
|
-
}
|
|
182
|
-
export function registerCodexPrerequisites(prerequisites) {
|
|
183
|
-
prerequisites.registerAfter(createCodexCliHealthCheck());
|
|
184
|
-
}
|
|
185
|
-
const CODEX_DEFAULT_EXEC_ARGS = ["--full-auto"];
|
|
186
|
-
export function buildCodexExecArgs(prompt, extraArgs = []) {
|
|
187
|
-
return ["exec", prompt, ...CODEX_DEFAULT_EXEC_ARGS, ...extraArgs];
|
|
188
|
-
}
|
|
189
|
-
function createCodexBinaryCheck() {
|
|
190
|
-
return createBinaryExistsCheck("codex", "codex-cli-binary", "Codex CLI binary must exist");
|
|
191
|
-
}
|
|
192
|
-
function createCodexVersionCheck() {
|
|
193
|
-
return {
|
|
194
|
-
id: "codex-cli-version",
|
|
195
|
-
description: "Codex CLI responds to --version",
|
|
196
|
-
async run({ runCommand }) {
|
|
197
|
-
const result = await runCommand("codex", ["--version"]);
|
|
198
|
-
if (result.exitCode !== 0) {
|
|
199
|
-
throw new Error(`Codex CLI --version failed with exit code ${result.exitCode}.`);
|
|
200
|
-
}
|
|
201
|
-
}
|
|
202
|
-
};
|
|
203
|
-
}
|
|
204
|
-
function createCodexCliHealthCheck() {
|
|
205
|
-
return {
|
|
206
|
-
id: "codex-cli-health",
|
|
207
|
-
description: "Codex CLI health check must succeed",
|
|
208
|
-
async run({ runCommand }) {
|
|
209
|
-
const result = await spawnCodex({
|
|
210
|
-
prompt: "Output exactly: CODEX_OK",
|
|
211
|
-
runCommand
|
|
212
|
-
});
|
|
213
|
-
if (result.exitCode !== 0) {
|
|
214
|
-
const detail = formatCommandRunnerResult(result);
|
|
215
|
-
throw new Error([
|
|
216
|
-
`Codex CLI health check failed with exit code ${result.exitCode}.`,
|
|
217
|
-
detail
|
|
218
|
-
].join("\n"));
|
|
219
|
-
}
|
|
220
|
-
const output = result.stdout.trim();
|
|
221
|
-
if (output !== "CODEX_OK") {
|
|
222
|
-
const detail = formatCommandRunnerResult(result);
|
|
223
|
-
throw new Error([
|
|
224
|
-
`Codex CLI health check failed: expected "CODEX_OK" but received "${output}".`,
|
|
225
|
-
detail
|
|
226
|
-
].join("\n"));
|
|
227
|
-
}
|
|
228
|
-
}
|
|
229
|
-
};
|
|
230
|
-
}
|
|
231
|
-
//# sourceMappingURL=codex.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"codex.js","sourceRoot":"","sources":["../../src/services/codex.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAQ7B,OAAO,EACL,uBAAuB,EACvB,yBAAyB,EAC1B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EACnB,gBAAgB,EAGjB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,eAAe,EACf,WAAW,EAEZ,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,iBAAiB,EAGlB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAsBvD,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAChC,MAAM,cAAc,GAAG,wBAAwB,CAAC;AAChD,MAAM,sBAAsB,GAAG;IAC7B,OAAO;IACP,wBAAwB;CAChB,CAAC;AACX,MAAM,wBAAwB,GAAG,uBAAuB,CAAC;AAEzD,MAAM,cAAc,GAGhB;IACF,EAAE,EAAE,OAAO;IACX,OAAO,EAAE,mDAAmD;IAC5D,aAAa,EAAE;QACb,KAAK,EAAE,CAAC,kBAAkB,CAAC;KAC5B;IACD,SAAS,EAAE;QACT,eAAe,CAAC;YACd,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;YACvD,KAAK,EAAE,+BAA+B;SACvC,CAAC;QACF,oBAAoB,CAAC;YACnB,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;YAC3C,SAAS,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS;YAC7C,KAAK,EAAE,4BAA4B;SACpC,CAAC;QACF;YACE,IAAI,EAAE,eAAe;YACrB,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;YAC3C,KAAK,EAAE,2BAA2B;YAClC,KAAK,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE;gBAClC,IAAI,QAAmB,CAAC;gBACxB,IAAI,OAAO,IAAI,IAAI,EAAE,CAAC;oBACpB,QAAQ,GAAG,EAAE,CAAC;gBAChB,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC;wBACH,QAAQ,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;oBACxC,CAAC;oBAAC,MAAM,CAAC;wBACP,QAAQ,GAAG,EAAE,CAAC;oBAChB,CAAC;gBACH,CAAC;gBACD,MAAM,QAAQ,GAAG,MAAM,cAAc,CACnC,wBAAwB,EACxB,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CACvB,CAAC;gBACF,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBACrD,MAAM,SAAS,GAAG,eAAe,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;gBAC9D,MAAM,WAAW,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;gBACrD,OAAO;oBACL,OAAO,EAAE,WAAW;oBACpB,OAAO,EAAE,WAAW,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;iBACzC,CAAC;YACJ,CAAC;SACF;KACF;IACD,MAAM,EAAE;QACN;YACE,IAAI,EAAE,eAAe;YACrB,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;YAC3C,KAAK,EAAE,oCAAoC;YAC3C,SAAS,CAAC,EAAE,OAAO,EAAE;gBACnB,IAAI,OAAO,IAAI,IAAI,EAAE,CAAC;oBACpB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAC3C,CAAC;gBAED,IAAI,QAAmB,CAAC;gBACxB,IAAI,CAAC;oBACH,QAAQ,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBACxC,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBACrC,CAAC;gBAED,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;gBACjD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBACrC,CAAC;gBAED,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBAC1C,CAAC;gBAED,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;gBACpD,OAAO;oBACL,OAAO,EAAE,WAAW;oBACpB,OAAO,EAAE,WAAW,KAAK,OAAO;iBACjC,CAAC;YACJ,CAAC;SACF;KACF;CACF,CAAC;AAEF,MAAM,wBAAwB,GAA6B;IACzD,EAAE,EAAE,OAAO;IACX,OAAO,EAAE,WAAW;IACpB,KAAK,EAAE,sBAAsB,EAAE;IAC/B,KAAK,EAAE;QACL;YACE,EAAE,EAAE,uBAAuB;YAC3B,WAAW,EAAE,2BAA2B;YACxC,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,eAAe,CAAC;SACzC;KACF;IACD,UAAU,EAAE,CAAC,uBAAuB,EAAE,CAAC;IACvC,cAAc,EAAE,8BAA8B;CAC/C,CAAC;AAEF,SAAS,uBAAuB,CAC9B,QAAmB;IAEnB,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAC1C,CAAC;IAED,IAAI,QAAQ,CAAC,gBAAgB,CAAC,KAAK,iBAAiB,EAAE,CAAC;QACrD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAC1C,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IAC9C,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAC1C,CAAC;IAED,MAAM,SAAS,GAAG,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,6BAA6B,CAAC,SAAS,CAAC,EAAE,CAAC;QACzE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAC1C,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,sBAAsB,EAAE,CAAC;QAC3C,IAAI,OAAO,QAAQ,CAAC,KAAK,CAAC,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC,gBAAgB,CAAC,CAAC;IAElC,KAAK,MAAM,KAAK,IAAI,sBAAsB,EAAE,CAAC;QAC3C,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAEpC,IAAI,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,OAAO;QACL,OAAO,EAAE,IAAI;QACb,KAAK,EAAE,YAAY,CAAC,QAAQ,CAAC;KAC9B,CAAC;AACJ,CAAC;AAED,SAAS,6BAA6B,CAAC,KAAgB;IACrD,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,KAAK,EAAE,CAAC;QAC5B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,cAAc,EAAE,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,MAAM,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;IAChC,IACE,MAAM,IAAI,IAAI;QACd,MAAM,KAAK,gBAAgB;QAC3B,MAAM,KAAK,aAAa,EACxB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAClD,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,WAAW,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC;AAC/D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAA8B,EAC9B,UAA8B;IAE9B,MAAM,mBAAmB,CACvB,cAAc,EACd;QACE,EAAE,EAAE,OAAO,CAAC,EAAE;QACd,OAAO;KACR,EACD,UAAU,CACX,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAA0B;IAE1B,MAAM,IAAI,GAAG,kBAAkB,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9D,OAAO,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,OAA2B,EAC3B,UAA8B;IAE9B,OAAO,gBAAgB,CACrB,cAAc,EACd;QACE,EAAE,EAAE,OAAO,CAAC,EAAE;QACd,OAAO;KACR,EACD,UAAU,CACX,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAuB;IAEvB,OAAO,iBAAiB,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,aAAkC;IAElC,aAAa,CAAC,aAAa,CAAC,yBAAyB,EAAE,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,uBAAuB,GAAG,CAAC,aAAa,CAAU,CAAC;AAEzD,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,YAAsB,EAAE;IAExB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,EAAE,GAAG,SAAS,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,sBAAsB;IAC7B,OAAO,uBAAuB,CAC5B,OAAO,EACP,kBAAkB,EAClB,6BAA6B,CAC9B,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB;IAC9B,OAAO;QACL,EAAE,EAAE,mBAAmB;QACvB,WAAW,EAAE,iCAAiC;QAC9C,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE;YACtB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;YACxD,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CACb,6CAA6C,MAAM,CAAC,QAAQ,GAAG,CAChE,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB;IAChC,OAAO;QACL,EAAE,EAAE,kBAAkB;QACtB,WAAW,EAAE,qCAAqC;QAClD,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE;YACtB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;gBAC9B,MAAM,EAAE,0BAA0B;gBAClC,UAAU;aACX,CAAC,CAAC;YACH,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,MAAM,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;gBACjD,MAAM,IAAI,KAAK,CACb;oBACE,gDAAgD,MAAM,CAAC,QAAQ,GAAG;oBAClE,MAAM;iBACP,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACpC,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;gBAC1B,MAAM,MAAM,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;gBACjD,MAAM,IAAI,KAAK,CACb;oBACE,oEAAoE,MAAM,IAAI;oBAC9E,MAAM;iBACP,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -1,38 +0,0 @@
|
|
|
1
|
-
import { EventEmitter } from "node:events";
|
|
2
|
-
export interface McpTool {
|
|
3
|
-
name: string;
|
|
4
|
-
description: string;
|
|
5
|
-
inputSchema: {
|
|
6
|
-
type: "object";
|
|
7
|
-
properties: Record<string, unknown>;
|
|
8
|
-
required?: string[];
|
|
9
|
-
};
|
|
10
|
-
}
|
|
11
|
-
export interface McpServerConfig {
|
|
12
|
-
name: string;
|
|
13
|
-
command: string;
|
|
14
|
-
args: string[];
|
|
15
|
-
env?: Record<string, string>;
|
|
16
|
-
}
|
|
17
|
-
export declare class McpClient extends EventEmitter {
|
|
18
|
-
private process;
|
|
19
|
-
private requestId;
|
|
20
|
-
private pendingRequests;
|
|
21
|
-
private buffer;
|
|
22
|
-
private tools;
|
|
23
|
-
private config;
|
|
24
|
-
private initialized;
|
|
25
|
-
constructor(config: McpServerConfig);
|
|
26
|
-
connect(): Promise<void>;
|
|
27
|
-
private initialize;
|
|
28
|
-
private discoverTools;
|
|
29
|
-
getTools(): McpTool[];
|
|
30
|
-
callTool(name: string, args: Record<string, unknown>): Promise<string>;
|
|
31
|
-
private handleData;
|
|
32
|
-
private handleMessage;
|
|
33
|
-
private sendRequest;
|
|
34
|
-
disconnect(): Promise<void>;
|
|
35
|
-
private cleanup;
|
|
36
|
-
isConnected(): boolean;
|
|
37
|
-
getName(): string;
|
|
38
|
-
}
|
|
@@ -1,170 +0,0 @@
|
|
|
1
|
-
import { spawn } from "node:child_process";
|
|
2
|
-
import { EventEmitter } from "node:events";
|
|
3
|
-
export class McpClient extends EventEmitter {
|
|
4
|
-
constructor(config) {
|
|
5
|
-
super();
|
|
6
|
-
this.process = null;
|
|
7
|
-
this.requestId = 0;
|
|
8
|
-
this.pendingRequests = new Map();
|
|
9
|
-
this.buffer = "";
|
|
10
|
-
this.tools = [];
|
|
11
|
-
this.initialized = false;
|
|
12
|
-
this.config = config;
|
|
13
|
-
}
|
|
14
|
-
async connect() {
|
|
15
|
-
if (this.process) {
|
|
16
|
-
throw new Error("Already connected");
|
|
17
|
-
}
|
|
18
|
-
return new Promise((resolve, reject) => {
|
|
19
|
-
this.process = spawn(this.config.command, this.config.args, {
|
|
20
|
-
env: { ...process.env, ...this.config.env },
|
|
21
|
-
stdio: ["pipe", "pipe", "pipe"]
|
|
22
|
-
});
|
|
23
|
-
this.process.on("error", (error) => {
|
|
24
|
-
this.emit("error", error);
|
|
25
|
-
reject(error);
|
|
26
|
-
});
|
|
27
|
-
if (this.process.stdout) {
|
|
28
|
-
this.process.stdout.on("data", (data) => {
|
|
29
|
-
this.handleData(data.toString());
|
|
30
|
-
});
|
|
31
|
-
}
|
|
32
|
-
if (this.process.stderr) {
|
|
33
|
-
this.process.stderr.on("data", (data) => {
|
|
34
|
-
this.emit("log", `[${this.config.name}] ${data.toString()}`);
|
|
35
|
-
});
|
|
36
|
-
}
|
|
37
|
-
this.process.on("close", (code) => {
|
|
38
|
-
this.emit("close", code);
|
|
39
|
-
this.cleanup();
|
|
40
|
-
});
|
|
41
|
-
// Initialize the connection
|
|
42
|
-
this.initialize()
|
|
43
|
-
.then(() => resolve())
|
|
44
|
-
.catch(reject);
|
|
45
|
-
});
|
|
46
|
-
}
|
|
47
|
-
async initialize() {
|
|
48
|
-
const response = await this.sendRequest("initialize", {
|
|
49
|
-
protocolVersion: "2024-11-05",
|
|
50
|
-
capabilities: {
|
|
51
|
-
tools: {}
|
|
52
|
-
},
|
|
53
|
-
clientInfo: {
|
|
54
|
-
name: "poe-code",
|
|
55
|
-
version: "0.1.0"
|
|
56
|
-
}
|
|
57
|
-
});
|
|
58
|
-
this.initialized = true;
|
|
59
|
-
this.emit("initialized", response);
|
|
60
|
-
// Discover available tools
|
|
61
|
-
await this.discoverTools();
|
|
62
|
-
}
|
|
63
|
-
async discoverTools() {
|
|
64
|
-
const response = (await this.sendRequest("tools/list", {}));
|
|
65
|
-
this.tools = response.tools || [];
|
|
66
|
-
this.emit("tools-discovered", this.tools);
|
|
67
|
-
}
|
|
68
|
-
getTools() {
|
|
69
|
-
return [...this.tools];
|
|
70
|
-
}
|
|
71
|
-
async callTool(name, args) {
|
|
72
|
-
if (!this.initialized) {
|
|
73
|
-
throw new Error("Client not initialized");
|
|
74
|
-
}
|
|
75
|
-
const response = (await this.sendRequest("tools/call", {
|
|
76
|
-
name,
|
|
77
|
-
arguments: args
|
|
78
|
-
}));
|
|
79
|
-
// Extract text content from response
|
|
80
|
-
if (response.content && Array.isArray(response.content)) {
|
|
81
|
-
const textContent = response.content
|
|
82
|
-
.filter((c) => c.type === "text" && c.text)
|
|
83
|
-
.map((c) => c.text)
|
|
84
|
-
.join("\n");
|
|
85
|
-
return textContent;
|
|
86
|
-
}
|
|
87
|
-
return JSON.stringify(response);
|
|
88
|
-
}
|
|
89
|
-
handleData(data) {
|
|
90
|
-
this.buffer += data;
|
|
91
|
-
// Process complete JSON-RPC messages (newline-delimited)
|
|
92
|
-
const lines = this.buffer.split("\n");
|
|
93
|
-
this.buffer = lines.pop() || "";
|
|
94
|
-
for (const line of lines) {
|
|
95
|
-
if (!line.trim())
|
|
96
|
-
continue;
|
|
97
|
-
try {
|
|
98
|
-
const message = JSON.parse(line);
|
|
99
|
-
this.handleMessage(message);
|
|
100
|
-
}
|
|
101
|
-
catch (error) {
|
|
102
|
-
this.emit("error", new Error(`Failed to parse message: ${line}, error: ${error}`));
|
|
103
|
-
}
|
|
104
|
-
}
|
|
105
|
-
}
|
|
106
|
-
handleMessage(message) {
|
|
107
|
-
const pending = this.pendingRequests.get(message.id);
|
|
108
|
-
if (!pending) {
|
|
109
|
-
this.emit("error", new Error(`Unexpected message id: ${message.id}`));
|
|
110
|
-
return;
|
|
111
|
-
}
|
|
112
|
-
this.pendingRequests.delete(message.id);
|
|
113
|
-
if (message.error) {
|
|
114
|
-
pending.reject(new Error(`MCP Error: ${message.error.message} (${message.error.code})`));
|
|
115
|
-
}
|
|
116
|
-
else {
|
|
117
|
-
pending.resolve(message.result);
|
|
118
|
-
}
|
|
119
|
-
}
|
|
120
|
-
sendRequest(method, params) {
|
|
121
|
-
if (!this.process || !this.process.stdin) {
|
|
122
|
-
return Promise.reject(new Error("Not connected"));
|
|
123
|
-
}
|
|
124
|
-
const id = ++this.requestId;
|
|
125
|
-
const request = {
|
|
126
|
-
jsonrpc: "2.0",
|
|
127
|
-
id,
|
|
128
|
-
method,
|
|
129
|
-
params
|
|
130
|
-
};
|
|
131
|
-
return new Promise((resolve, reject) => {
|
|
132
|
-
this.pendingRequests.set(id, { resolve, reject });
|
|
133
|
-
const requestStr = JSON.stringify(request) + "\n";
|
|
134
|
-
this.process.stdin.write(requestStr, (error) => {
|
|
135
|
-
if (error) {
|
|
136
|
-
this.pendingRequests.delete(id);
|
|
137
|
-
reject(error);
|
|
138
|
-
}
|
|
139
|
-
});
|
|
140
|
-
// Add timeout
|
|
141
|
-
setTimeout(() => {
|
|
142
|
-
if (this.pendingRequests.has(id)) {
|
|
143
|
-
this.pendingRequests.delete(id);
|
|
144
|
-
reject(new Error(`Request timeout: ${method}`));
|
|
145
|
-
}
|
|
146
|
-
}, 30000);
|
|
147
|
-
});
|
|
148
|
-
}
|
|
149
|
-
async disconnect() {
|
|
150
|
-
if (!this.process) {
|
|
151
|
-
return;
|
|
152
|
-
}
|
|
153
|
-
this.process.kill();
|
|
154
|
-
this.cleanup();
|
|
155
|
-
}
|
|
156
|
-
cleanup() {
|
|
157
|
-
this.process = null;
|
|
158
|
-
this.initialized = false;
|
|
159
|
-
this.tools = [];
|
|
160
|
-
this.pendingRequests.clear();
|
|
161
|
-
this.buffer = "";
|
|
162
|
-
}
|
|
163
|
-
isConnected() {
|
|
164
|
-
return this.process !== null && this.initialized;
|
|
165
|
-
}
|
|
166
|
-
getName() {
|
|
167
|
-
return this.config.name;
|
|
168
|
-
}
|
|
169
|
-
}
|
|
170
|
-
//# sourceMappingURL=mcp-client.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../../src/services/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAqC3C,MAAM,OAAO,SAAU,SAAQ,YAAY;IAezC,YAAY,MAAuB;QACjC,KAAK,EAAE,CAAC;QAfF,YAAO,GAAwB,IAAI,CAAC;QACpC,cAAS,GAAG,CAAC,CAAC;QACd,oBAAe,GAAG,IAAI,GAAG,EAM9B,CAAC;QACI,WAAM,GAAG,EAAE,CAAC;QACZ,UAAK,GAAc,EAAE,CAAC;QAEtB,gBAAW,GAAG,KAAK,CAAC;QAI1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;gBAC1D,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;gBAC3C,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;aAChC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBAC1B,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACxB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;oBAC9C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACnC,CAAC,CAAC,CAAC;YACL,CAAC;YAED,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBACxB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;oBAC9C,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAC/D,CAAC,CAAC,CAAC;YACL,CAAC;YAED,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;gBAChC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBACzB,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;YAEH,4BAA4B;YAC5B,IAAI,CAAC,UAAU,EAAE;iBACd,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;iBACrB,KAAK,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;YACpD,eAAe,EAAE,YAAY;YAC7B,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE;aACV;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,OAAO;aACjB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAEnC,2BAA2B;QAC3B,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,MAAM,QAAQ,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,CAAC,CAEzD,CAAC;QAEF,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC;IAED,QAAQ;QACN,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAY,EAAE,IAA6B;QACxD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,QAAQ,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;YACrD,IAAI;YACJ,SAAS,EAAE,IAAI;SAChB,CAAC,CAED,CAAC;QAEF,qCAAqC;QACrC,IAAI,QAAQ,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACxD,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO;iBACjC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC;iBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;IAEO,UAAU,CAAC,IAAY;QAC7B,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC;QAEpB,yDAAyD;QACzD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEhC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAE3B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoB,CAAC;gBACpD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,IAAI,CACP,OAAO,EACP,IAAI,KAAK,CAAC,4BAA4B,IAAI,YAAY,KAAK,EAAE,CAAC,CAC/D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,OAAwB;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,0BAA0B,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YACtE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAExC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,MAAM,CACZ,IAAI,KAAK,CAAC,cAAc,OAAO,CAAC,KAAK,CAAC,OAAO,KAAK,OAAO,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CACzE,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,MAAc,EAAE,MAAe;QACjD,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACzC,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC;QAC5B,MAAM,OAAO,GAAmB;YAC9B,OAAO,EAAE,KAAK;YACd,EAAE;YACF,MAAM;YACN,MAAM;SACP,CAAC;QAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAElD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;YAClD,IAAI,CAAC,OAAQ,CAAC,KAAM,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC/C,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAChC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,cAAc;YACd,UAAU,CAAC,GAAG,EAAE;gBACd,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACjC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAChC,MAAM,CAAC,IAAI,KAAK,CAAC,oBAAoB,MAAM,EAAE,CAAC,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC,EAAE,KAAK,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,OAAO,KAAK,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC;IACnD,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -1,28 +0,0 @@
|
|
|
1
|
-
import { type McpServerConfig } from "./mcp-client.js";
|
|
2
|
-
import type { Tool } from "./chat.js";
|
|
3
|
-
import type { FileSystem } from "../utils/file-system.js";
|
|
4
|
-
export interface McpManagerConfig {
|
|
5
|
-
servers: McpServerConfig[];
|
|
6
|
-
}
|
|
7
|
-
export declare class McpManager {
|
|
8
|
-
private clients;
|
|
9
|
-
private fs;
|
|
10
|
-
private configPath;
|
|
11
|
-
constructor(fs: FileSystem, homeDir: string);
|
|
12
|
-
loadConfig(): Promise<McpManagerConfig>;
|
|
13
|
-
saveConfig(config: McpManagerConfig): Promise<void>;
|
|
14
|
-
addServer(config: McpServerConfig): Promise<void>;
|
|
15
|
-
removeServer(name: string): Promise<boolean>;
|
|
16
|
-
connectServer(name: string): Promise<void>;
|
|
17
|
-
disconnectServer(name: string): Promise<void>;
|
|
18
|
-
connectAll(): Promise<void>;
|
|
19
|
-
disconnectAll(): Promise<void>;
|
|
20
|
-
getAllTools(): Tool[];
|
|
21
|
-
executeTool(toolName: string, args: Record<string, unknown>): Promise<string>;
|
|
22
|
-
getConnectedServers(): string[];
|
|
23
|
-
listServers(): Promise<Array<{
|
|
24
|
-
name: string;
|
|
25
|
-
connected: boolean;
|
|
26
|
-
toolCount: number;
|
|
27
|
-
}>>;
|
|
28
|
-
}
|
|
@@ -1,157 +0,0 @@
|
|
|
1
|
-
import { McpClient } from "./mcp-client.js";
|
|
2
|
-
import path from "node:path";
|
|
3
|
-
export class McpManager {
|
|
4
|
-
constructor(fs, homeDir) {
|
|
5
|
-
this.clients = new Map();
|
|
6
|
-
this.fs = fs;
|
|
7
|
-
this.configPath = path.join(homeDir, ".poe-code", "mcp-servers.json");
|
|
8
|
-
}
|
|
9
|
-
async loadConfig() {
|
|
10
|
-
try {
|
|
11
|
-
const content = await this.fs.readFile(this.configPath, "utf8");
|
|
12
|
-
return JSON.parse(content);
|
|
13
|
-
}
|
|
14
|
-
catch {
|
|
15
|
-
// Config doesn't exist, return empty
|
|
16
|
-
return { servers: [] };
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
async saveConfig(config) {
|
|
20
|
-
const dir = path.dirname(this.configPath);
|
|
21
|
-
try {
|
|
22
|
-
await this.fs.stat(dir);
|
|
23
|
-
}
|
|
24
|
-
catch {
|
|
25
|
-
await this.fs.mkdir(dir, { recursive: true });
|
|
26
|
-
}
|
|
27
|
-
await this.fs.writeFile(this.configPath, JSON.stringify(config, null, 2), { encoding: "utf8" });
|
|
28
|
-
}
|
|
29
|
-
async addServer(config) {
|
|
30
|
-
const currentConfig = await this.loadConfig();
|
|
31
|
-
// Check if server already exists
|
|
32
|
-
const existingIndex = currentConfig.servers.findIndex((s) => s.name === config.name);
|
|
33
|
-
if (existingIndex >= 0) {
|
|
34
|
-
currentConfig.servers[existingIndex] = config;
|
|
35
|
-
}
|
|
36
|
-
else {
|
|
37
|
-
currentConfig.servers.push(config);
|
|
38
|
-
}
|
|
39
|
-
await this.saveConfig(currentConfig);
|
|
40
|
-
}
|
|
41
|
-
async removeServer(name) {
|
|
42
|
-
const currentConfig = await this.loadConfig();
|
|
43
|
-
const initialLength = currentConfig.servers.length;
|
|
44
|
-
currentConfig.servers = currentConfig.servers.filter((s) => s.name !== name);
|
|
45
|
-
if (currentConfig.servers.length < initialLength) {
|
|
46
|
-
await this.saveConfig(currentConfig);
|
|
47
|
-
// Also disconnect the client if running
|
|
48
|
-
await this.disconnectServer(name);
|
|
49
|
-
return true;
|
|
50
|
-
}
|
|
51
|
-
return false;
|
|
52
|
-
}
|
|
53
|
-
async connectServer(name) {
|
|
54
|
-
const config = await this.loadConfig();
|
|
55
|
-
const serverConfig = config.servers.find((s) => s.name === name);
|
|
56
|
-
if (!serverConfig) {
|
|
57
|
-
throw new Error(`MCP server "${name}" not found in configuration`);
|
|
58
|
-
}
|
|
59
|
-
if (this.clients.has(name)) {
|
|
60
|
-
throw new Error(`MCP server "${name}" is already connected`);
|
|
61
|
-
}
|
|
62
|
-
const client = new McpClient(serverConfig);
|
|
63
|
-
client.on("error", (error) => {
|
|
64
|
-
console.error(`[MCP ${name}] Error:`, error);
|
|
65
|
-
});
|
|
66
|
-
client.on("log", (message) => {
|
|
67
|
-
console.log(`[MCP ${name}]`, message);
|
|
68
|
-
});
|
|
69
|
-
client.on("close", () => {
|
|
70
|
-
this.clients.delete(name);
|
|
71
|
-
});
|
|
72
|
-
await client.connect();
|
|
73
|
-
this.clients.set(name, client);
|
|
74
|
-
}
|
|
75
|
-
async disconnectServer(name) {
|
|
76
|
-
const client = this.clients.get(name);
|
|
77
|
-
if (client) {
|
|
78
|
-
await client.disconnect();
|
|
79
|
-
this.clients.delete(name);
|
|
80
|
-
}
|
|
81
|
-
}
|
|
82
|
-
async connectAll() {
|
|
83
|
-
const config = await this.loadConfig();
|
|
84
|
-
for (const serverConfig of config.servers) {
|
|
85
|
-
try {
|
|
86
|
-
if (!this.clients.has(serverConfig.name)) {
|
|
87
|
-
await this.connectServer(serverConfig.name);
|
|
88
|
-
}
|
|
89
|
-
}
|
|
90
|
-
catch (error) {
|
|
91
|
-
console.error(`Failed to connect to MCP server "${serverConfig.name}":`, error);
|
|
92
|
-
}
|
|
93
|
-
}
|
|
94
|
-
}
|
|
95
|
-
async disconnectAll() {
|
|
96
|
-
const names = Array.from(this.clients.keys());
|
|
97
|
-
for (const name of names) {
|
|
98
|
-
await this.disconnectServer(name);
|
|
99
|
-
}
|
|
100
|
-
}
|
|
101
|
-
getAllTools() {
|
|
102
|
-
const tools = [];
|
|
103
|
-
for (const [serverName, client] of this.clients.entries()) {
|
|
104
|
-
if (!client.isConnected())
|
|
105
|
-
continue;
|
|
106
|
-
const mcpTools = client.getTools();
|
|
107
|
-
for (const mcpTool of mcpTools) {
|
|
108
|
-
tools.push({
|
|
109
|
-
type: "function",
|
|
110
|
-
function: {
|
|
111
|
-
name: `mcp__${serverName}__${mcpTool.name}`,
|
|
112
|
-
description: `[MCP: ${serverName}] ${mcpTool.description}`,
|
|
113
|
-
parameters: {
|
|
114
|
-
type: "object",
|
|
115
|
-
properties: mcpTool.inputSchema.properties,
|
|
116
|
-
required: mcpTool.inputSchema.required
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
});
|
|
120
|
-
}
|
|
121
|
-
}
|
|
122
|
-
return tools;
|
|
123
|
-
}
|
|
124
|
-
async executeTool(toolName, args) {
|
|
125
|
-
// Tool name format: mcp__<servername>__<toolname>
|
|
126
|
-
if (!toolName.startsWith("mcp__")) {
|
|
127
|
-
throw new Error(`Not an MCP tool: ${toolName}`);
|
|
128
|
-
}
|
|
129
|
-
const parts = toolName.substring(5).split("__");
|
|
130
|
-
if (parts.length !== 2) {
|
|
131
|
-
throw new Error(`Invalid MCP tool name format: ${toolName}`);
|
|
132
|
-
}
|
|
133
|
-
const [serverName, actualToolName] = parts;
|
|
134
|
-
const client = this.clients.get(serverName);
|
|
135
|
-
if (!client || !client.isConnected()) {
|
|
136
|
-
throw new Error(`MCP server "${serverName}" is not connected`);
|
|
137
|
-
}
|
|
138
|
-
return await client.callTool(actualToolName, args);
|
|
139
|
-
}
|
|
140
|
-
getConnectedServers() {
|
|
141
|
-
return Array.from(this.clients.keys()).filter((name) => this.clients.get(name)?.isConnected());
|
|
142
|
-
}
|
|
143
|
-
async listServers() {
|
|
144
|
-
const config = await this.loadConfig();
|
|
145
|
-
return config.servers.map((server) => {
|
|
146
|
-
const client = this.clients.get(server.name);
|
|
147
|
-
const connected = client?.isConnected() || false;
|
|
148
|
-
const toolCount = connected ? client.getTools().length : 0;
|
|
149
|
-
return {
|
|
150
|
-
name: server.name,
|
|
151
|
-
connected,
|
|
152
|
-
toolCount
|
|
153
|
-
};
|
|
154
|
-
});
|
|
155
|
-
}
|
|
156
|
-
}
|
|
157
|
-
//# sourceMappingURL=mcp-manager.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"mcp-manager.js","sourceRoot":"","sources":["../../src/services/mcp-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAwB,MAAM,iBAAiB,CAAC;AAGlE,OAAO,IAAI,MAAM,WAAW,CAAC;AAM7B,MAAM,OAAO,UAAU;IAKrB,YAAY,EAAc,EAAE,OAAe;QAJnC,YAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;QAK7C,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,kBAAkB,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAChE,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QAAC,MAAM,CAAC;YACP,qCAAqC;YACrC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAwB;QACvC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,IAAI,CAAC,EAAE,CAAC,SAAS,CACrB,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAC/B,EAAE,QAAQ,EAAE,MAAM,EAAE,CACrB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAuB;QACrC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAE9C,iCAAiC;QACjC,MAAM,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,SAAS,CACnD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAC9B,CAAC;QAEF,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;YACvB,aAAa,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC;QAChD,CAAC;aAAM,CAAC;YACN,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC;QAEnD,aAAa,CAAC,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,CAClD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CACvB,CAAC;QAEF,IAAI,aAAa,CAAC,OAAO,CAAC,MAAM,GAAG,aAAa,EAAE,CAAC;YACjD,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YACrC,wCAAwC;YACxC,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAClC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,IAAY;QAC9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACvC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAEjE,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,8BAA8B,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,wBAAwB,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;QAE3C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC3B,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,UAAU,EAAE,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YAC3B,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,GAAG,EAAE,OAAO,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACvB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAC1B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvC,KAAK,MAAM,YAAY,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1C,IAAI,CAAC;gBACH,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;oBACzC,MAAM,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,oCAAoC,YAAY,CAAC,IAAI,IAAI,EACzD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,WAAW;QACT,MAAM,KAAK,GAAW,EAAE,CAAC;QAEzB,KAAK,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YAC1D,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE;gBAAE,SAAS;YAEpC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YACnC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,IAAI,EAAE,QAAQ,UAAU,KAAK,OAAO,CAAC,IAAI,EAAE;wBAC3C,WAAW,EAAE,SAAS,UAAU,KAAK,OAAO,CAAC,WAAW,EAAE;wBAC1D,UAAU,EAAE;4BACV,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE,OAAO,CAAC,WAAW,CAAC,UAAU;4BAC1C,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,QAAQ;yBACvC;qBACF;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,WAAW,CACf,QAAgB,EAChB,IAA6B;QAE7B,kDAAkD;QAClD,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,iCAAiC,QAAQ,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,CAAC,UAAU,EAAE,cAAc,CAAC,GAAG,KAAK,CAAC;QAE3C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,oBAAoB,CAAC,CAAC;QACjE,CAAC;QAED,OAAO,MAAM,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC;IAED,mBAAmB;QACjB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACrD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,CACtC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW;QAGf,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvC,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,MAAM,EAAE,WAAW,EAAE,IAAI,KAAK,CAAC;YACjD,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,MAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAE5D,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS;gBACT,SAAS;aACV,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|