poe-code 0.1.26

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.
Files changed (245) hide show
  1. package/README.md +64 -0
  2. package/dist/cli/api-client.d.ts +14 -0
  3. package/dist/cli/api-client.js +140 -0
  4. package/dist/cli/api-client.js.map +1 -0
  5. package/dist/cli/chat.d.ts +32 -0
  6. package/dist/cli/chat.js +2 -0
  7. package/dist/cli/chat.js.map +1 -0
  8. package/dist/cli/command-runner.d.ts +2 -0
  9. package/dist/cli/command-runner.js +37 -0
  10. package/dist/cli/command-runner.js.map +1 -0
  11. package/dist/cli/commands/agent.d.ts +18 -0
  12. package/dist/cli/commands/agent.js +84 -0
  13. package/dist/cli/commands/agent.js.map +1 -0
  14. package/dist/cli/commands/configure-agents.d.ts +3 -0
  15. package/dist/cli/commands/configure-agents.js +72 -0
  16. package/dist/cli/commands/configure-agents.js.map +1 -0
  17. package/dist/cli/commands/configure.d.ts +11 -0
  18. package/dist/cli/commands/configure.js +150 -0
  19. package/dist/cli/commands/configure.js.map +1 -0
  20. package/dist/cli/commands/init.d.ts +8 -0
  21. package/dist/cli/commands/init.js +45 -0
  22. package/dist/cli/commands/init.js.map +1 -0
  23. package/dist/cli/commands/interactive.d.ts +3 -0
  24. package/dist/cli/commands/interactive.js +17 -0
  25. package/dist/cli/commands/interactive.js.map +1 -0
  26. package/dist/cli/commands/login.d.ts +6 -0
  27. package/dist/cli/commands/login.js +51 -0
  28. package/dist/cli/commands/login.js.map +1 -0
  29. package/dist/cli/commands/logout.d.ts +3 -0
  30. package/dist/cli/commands/logout.js +31 -0
  31. package/dist/cli/commands/logout.js.map +1 -0
  32. package/dist/cli/commands/prerequisites.d.ts +3 -0
  33. package/dist/cli/commands/prerequisites.js +23 -0
  34. package/dist/cli/commands/prerequisites.js.map +1 -0
  35. package/dist/cli/commands/query.d.ts +7 -0
  36. package/dist/cli/commands/query.js +46 -0
  37. package/dist/cli/commands/query.js.map +1 -0
  38. package/dist/cli/commands/remove.d.ts +6 -0
  39. package/dist/cli/commands/remove.js +80 -0
  40. package/dist/cli/commands/remove.js.map +1 -0
  41. package/dist/cli/commands/root.d.ts +3 -0
  42. package/dist/cli/commands/root.js +7 -0
  43. package/dist/cli/commands/root.js.map +1 -0
  44. package/dist/cli/commands/shared.d.ts +24 -0
  45. package/dist/cli/commands/shared.js +61 -0
  46. package/dist/cli/commands/shared.js.map +1 -0
  47. package/dist/cli/commands/spawn-worktree.d.ts +6 -0
  48. package/dist/cli/commands/spawn-worktree.js +91 -0
  49. package/dist/cli/commands/spawn-worktree.js.map +1 -0
  50. package/dist/cli/commands/spawn.d.ts +7 -0
  51. package/dist/cli/commands/spawn.js +112 -0
  52. package/dist/cli/commands/spawn.js.map +1 -0
  53. package/dist/cli/commands/test.d.ts +6 -0
  54. package/dist/cli/commands/test.js +26 -0
  55. package/dist/cli/commands/test.js.map +1 -0
  56. package/dist/cli/constants.d.ts +10 -0
  57. package/dist/cli/constants.js +11 -0
  58. package/dist/cli/constants.js.map +1 -0
  59. package/dist/cli/container.d.ts +48 -0
  60. package/dist/cli/container.js +94 -0
  61. package/dist/cli/container.js.map +1 -0
  62. package/dist/cli/context.d.ts +37 -0
  63. package/dist/cli/context.js +198 -0
  64. package/dist/cli/context.js.map +1 -0
  65. package/dist/cli/environment.d.ts +19 -0
  66. package/dist/cli/environment.js +26 -0
  67. package/dist/cli/environment.js.map +1 -0
  68. package/dist/cli/error-logger.d.ts +46 -0
  69. package/dist/cli/error-logger.js +137 -0
  70. package/dist/cli/error-logger.js.map +1 -0
  71. package/dist/cli/errors.d.ts +91 -0
  72. package/dist/cli/errors.js +128 -0
  73. package/dist/cli/errors.js.map +1 -0
  74. package/dist/cli/file-mentions.d.ts +13 -0
  75. package/dist/cli/file-mentions.js +87 -0
  76. package/dist/cli/file-mentions.js.map +1 -0
  77. package/dist/cli/file-picker-state.d.ts +14 -0
  78. package/dist/cli/file-picker-state.js +23 -0
  79. package/dist/cli/file-picker-state.js.map +1 -0
  80. package/dist/cli/http.d.ts +11 -0
  81. package/dist/cli/http.js +2 -0
  82. package/dist/cli/http.js.map +1 -0
  83. package/dist/cli/interactive-command-runner.d.ts +11 -0
  84. package/dist/cli/interactive-command-runner.js +142 -0
  85. package/dist/cli/interactive-command-runner.js.map +1 -0
  86. package/dist/cli/interactive-launcher.d.ts +2 -0
  87. package/dist/cli/interactive-launcher.js +404 -0
  88. package/dist/cli/interactive-launcher.js.map +1 -0
  89. package/dist/cli/interactive-tasks.d.ts +7 -0
  90. package/dist/cli/interactive-tasks.js +199 -0
  91. package/dist/cli/interactive-tasks.js.map +1 -0
  92. package/dist/cli/interactive.d.ts +19 -0
  93. package/dist/cli/interactive.js +479 -0
  94. package/dist/cli/interactive.js.map +1 -0
  95. package/dist/cli/logger.d.ts +26 -0
  96. package/dist/cli/logger.js +87 -0
  97. package/dist/cli/logger.js.map +1 -0
  98. package/dist/cli/markdown-renderer.d.ts +27 -0
  99. package/dist/cli/markdown-renderer.js +178 -0
  100. package/dist/cli/markdown-renderer.js.map +1 -0
  101. package/dist/cli/markdown-renderer.test.d.ts +1 -0
  102. package/dist/cli/markdown-renderer.test.js +105 -0
  103. package/dist/cli/markdown-renderer.test.js.map +1 -0
  104. package/dist/cli/options.d.ts +30 -0
  105. package/dist/cli/options.js +84 -0
  106. package/dist/cli/options.js.map +1 -0
  107. package/dist/cli/program.d.ts +4 -0
  108. package/dist/cli/program.js +64 -0
  109. package/dist/cli/program.js.map +1 -0
  110. package/dist/cli/prompts.d.ts +23 -0
  111. package/dist/cli/prompts.js +58 -0
  112. package/dist/cli/prompts.js.map +1 -0
  113. package/dist/cli/service-registry.d.ts +33 -0
  114. package/dist/cli/service-registry.js +35 -0
  115. package/dist/cli/service-registry.js.map +1 -0
  116. package/dist/cli/telemetry.d.ts +15 -0
  117. package/dist/cli/telemetry.js +45 -0
  118. package/dist/cli/telemetry.js.map +1 -0
  119. package/dist/cli/types.d.ts +2 -0
  120. package/dist/cli/types.js +2 -0
  121. package/dist/cli/types.js.map +1 -0
  122. package/dist/commands/init.d.ts +9 -0
  123. package/dist/commands/init.js +43 -0
  124. package/dist/commands/init.js.map +1 -0
  125. package/dist/commands/publish-placeholder.d.ts +8 -0
  126. package/dist/commands/publish-placeholder.js +50 -0
  127. package/dist/commands/publish-placeholder.js.map +1 -0
  128. package/dist/commands/spawn-worktree.d.ts +25 -0
  129. package/dist/commands/spawn-worktree.js +139 -0
  130. package/dist/commands/spawn-worktree.js.map +1 -0
  131. package/dist/index.d.ts +4 -0
  132. package/dist/index.js +81 -0
  133. package/dist/index.js.map +1 -0
  134. package/dist/providers/claude-code-adapter.d.ts +20 -0
  135. package/dist/providers/claude-code-adapter.js +48 -0
  136. package/dist/providers/claude-code-adapter.js.map +1 -0
  137. package/dist/providers/codex-adapter.d.ts +19 -0
  138. package/dist/providers/codex-adapter.js +49 -0
  139. package/dist/providers/codex-adapter.js.map +1 -0
  140. package/dist/providers/index.d.ts +2 -0
  141. package/dist/providers/index.js +13 -0
  142. package/dist/providers/index.js.map +1 -0
  143. package/dist/providers/opencode-adapter.d.ts +18 -0
  144. package/dist/providers/opencode-adapter.js +45 -0
  145. package/dist/providers/opencode-adapter.js.map +1 -0
  146. package/dist/providers/roo-code-adapter.d.ts +19 -0
  147. package/dist/providers/roo-code-adapter.js +70 -0
  148. package/dist/providers/roo-code-adapter.js.map +1 -0
  149. package/dist/services/agent-config-manager.d.ts +32 -0
  150. package/dist/services/agent-config-manager.js +130 -0
  151. package/dist/services/agent-config-manager.js.map +1 -0
  152. package/dist/services/agent-registry.d.ts +27 -0
  153. package/dist/services/agent-registry.js +89 -0
  154. package/dist/services/agent-registry.js.map +1 -0
  155. package/dist/services/agent-session.d.ts +24 -0
  156. package/dist/services/agent-session.js +107 -0
  157. package/dist/services/agent-session.js.map +1 -0
  158. package/dist/services/agent-task-registry.d.ts +131 -0
  159. package/dist/services/agent-task-registry.js +491 -0
  160. package/dist/services/agent-task-registry.js.map +1 -0
  161. package/dist/services/chat.d.ts +94 -0
  162. package/dist/services/chat.js +241 -0
  163. package/dist/services/chat.js.map +1 -0
  164. package/dist/services/claude-code.d.ts +28 -0
  165. package/dist/services/claude-code.js +155 -0
  166. package/dist/services/claude-code.js.map +1 -0
  167. package/dist/services/codex.d.ts +27 -0
  168. package/dist/services/codex.js +231 -0
  169. package/dist/services/codex.js.map +1 -0
  170. package/dist/services/credentials.d.ts +11 -0
  171. package/dist/services/credentials.js +44 -0
  172. package/dist/services/credentials.js.map +1 -0
  173. package/dist/services/mcp-client.d.ts +38 -0
  174. package/dist/services/mcp-client.js +170 -0
  175. package/dist/services/mcp-client.js.map +1 -0
  176. package/dist/services/mcp-manager.d.ts +28 -0
  177. package/dist/services/mcp-manager.js +157 -0
  178. package/dist/services/mcp-manager.js.map +1 -0
  179. package/dist/services/model-strategy.d.ts +99 -0
  180. package/dist/services/model-strategy.js +182 -0
  181. package/dist/services/model-strategy.js.map +1 -0
  182. package/dist/services/opencode.d.ts +25 -0
  183. package/dist/services/opencode.js +161 -0
  184. package/dist/services/opencode.js.map +1 -0
  185. package/dist/services/poe-code.d.ts +7 -0
  186. package/dist/services/poe-code.js +5 -0
  187. package/dist/services/poe-code.js.map +1 -0
  188. package/dist/services/roo-code.d.ts +22 -0
  189. package/dist/services/roo-code.js +134 -0
  190. package/dist/services/roo-code.js.map +1 -0
  191. package/dist/services/service-install.d.ts +21 -0
  192. package/dist/services/service-install.js +47 -0
  193. package/dist/services/service-install.js.map +1 -0
  194. package/dist/services/service-manifest.d.ts +122 -0
  195. package/dist/services/service-manifest.js +368 -0
  196. package/dist/services/service-manifest.js.map +1 -0
  197. package/dist/services/task-logger.d.ts +24 -0
  198. package/dist/services/task-logger.js +73 -0
  199. package/dist/services/task-logger.js.map +1 -0
  200. package/dist/services/task-runner.d.ts +19 -0
  201. package/dist/services/task-runner.js +157 -0
  202. package/dist/services/task-runner.js.map +1 -0
  203. package/dist/services/tools.d.ts +76 -0
  204. package/dist/services/tools.js +870 -0
  205. package/dist/services/tools.js.map +1 -0
  206. package/dist/templates/claude-code/anthropic_key.sh.hbs +2 -0
  207. package/dist/templates/codex/config.toml.hbs +9 -0
  208. package/dist/templates/python/env.hbs +3 -0
  209. package/dist/templates/python/main.py.hbs +17 -0
  210. package/dist/templates/python/requirements.txt.hbs +2 -0
  211. package/dist/utils/backup.d.ts +5 -0
  212. package/dist/utils/backup.js +61 -0
  213. package/dist/utils/backup.js.map +1 -0
  214. package/dist/utils/command-line.d.ts +1 -0
  215. package/dist/utils/command-line.js +54 -0
  216. package/dist/utils/command-line.js.map +1 -0
  217. package/dist/utils/dry-run.d.ts +38 -0
  218. package/dist/utils/dry-run.js +211 -0
  219. package/dist/utils/dry-run.js.map +1 -0
  220. package/dist/utils/e2e-flags.d.ts +1 -0
  221. package/dist/utils/e2e-flags.js +9 -0
  222. package/dist/utils/e2e-flags.js.map +1 -0
  223. package/dist/utils/file-system.d.ts +21 -0
  224. package/dist/utils/file-system.js +2 -0
  225. package/dist/utils/file-system.js.map +1 -0
  226. package/dist/utils/json.d.ts +12 -0
  227. package/dist/utils/json.js +42 -0
  228. package/dist/utils/json.js.map +1 -0
  229. package/dist/utils/prerequisites.d.ts +41 -0
  230. package/dist/utils/prerequisites.js +92 -0
  231. package/dist/utils/prerequisites.js.map +1 -0
  232. package/dist/utils/templates.d.ts +4 -0
  233. package/dist/utils/templates.js +22 -0
  234. package/dist/utils/templates.js.map +1 -0
  235. package/dist/utils/toml.d.ts +8 -0
  236. package/dist/utils/toml.js +30 -0
  237. package/dist/utils/toml.js.map +1 -0
  238. package/dist/utils/worktree.d.ts +40 -0
  239. package/dist/utils/worktree.js +179 -0
  240. package/dist/utils/worktree.js.map +1 -0
  241. package/package.json +84 -0
  242. package/shared/conversation-layout.cjs +11 -0
  243. package/shared/conversation-layout.d.ts +7 -0
  244. package/shared/conversation-layout.js +11 -0
  245. package/shared/package.json +13 -0
@@ -0,0 +1,231 @@
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
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,11 @@
1
+ import type { FileSystem } from "../utils/file-system.js";
2
+ export interface CredentialsStoreOptions {
3
+ fs: FileSystem;
4
+ filePath: string;
5
+ }
6
+ export interface SaveCredentialsOptions extends CredentialsStoreOptions {
7
+ apiKey: string;
8
+ }
9
+ export declare function saveCredentials(options: SaveCredentialsOptions): Promise<void>;
10
+ export declare function loadCredentials(options: CredentialsStoreOptions): Promise<string | null>;
11
+ export declare function deleteCredentials(options: CredentialsStoreOptions): Promise<boolean>;
@@ -0,0 +1,44 @@
1
+ import path from "node:path";
2
+ export async function saveCredentials(options) {
3
+ const { fs, filePath, apiKey } = options;
4
+ await fs.mkdir(path.dirname(filePath), { recursive: true });
5
+ const payload = JSON.stringify({ apiKey }, null, 2);
6
+ await fs.writeFile(`${filePath}`, `${payload}\n`, { encoding: "utf8" });
7
+ }
8
+ export async function loadCredentials(options) {
9
+ const { fs, filePath } = options;
10
+ try {
11
+ const raw = await fs.readFile(filePath, "utf8");
12
+ const parsed = JSON.parse(raw);
13
+ if (typeof parsed.apiKey === "string" && parsed.apiKey.length > 0) {
14
+ return parsed.apiKey;
15
+ }
16
+ return null;
17
+ }
18
+ catch (error) {
19
+ if (isNotFound(error)) {
20
+ return null;
21
+ }
22
+ throw error;
23
+ }
24
+ }
25
+ export async function deleteCredentials(options) {
26
+ const { fs, filePath } = options;
27
+ try {
28
+ await fs.unlink(filePath);
29
+ return true;
30
+ }
31
+ catch (error) {
32
+ if (isNotFound(error)) {
33
+ return false;
34
+ }
35
+ throw error;
36
+ }
37
+ }
38
+ function isNotFound(error) {
39
+ return (typeof error === "object" &&
40
+ error !== null &&
41
+ "code" in error &&
42
+ error.code === "ENOENT");
43
+ }
44
+ //# sourceMappingURL=credentials.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"credentials.js","sourceRoot":"","sources":["../../src/services/credentials.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAY7B,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,OAA+B;IAE/B,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IACzC,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACpD,MAAM,EAAE,CAAC,SAAS,CAAC,GAAG,QAAQ,EAAE,EAAE,GAAG,OAAO,IAAI,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,OAAgC;IAEhC,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IACjC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAyB,CAAC;QACvD,IAAI,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClE,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAgC;IAEhC,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IACjC,IAAI,CAAC;QACH,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,KAAc;IAChC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,MAAM,IAAI,KAAK;QACd,KAA2B,CAAC,IAAI,KAAK,QAAQ,CAC/C,CAAC;AACJ,CAAC"}
@@ -0,0 +1,38 @@
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
+ }
@@ -0,0 +1,170 @@
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
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,28 @@
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
+ }