fixo-cli 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/LICENSE +201 -0
- package/README.md +530 -0
- package/dist/agent/agent-client.d.ts +108 -0
- package/dist/agent/agent-client.d.ts.map +1 -0
- package/dist/agent/agent-client.js +1247 -0
- package/dist/agent/agent-client.js.map +1 -0
- package/dist/agent/agent-pool.d.ts +20 -0
- package/dist/agent/agent-pool.d.ts.map +1 -0
- package/dist/agent/agent-pool.js +217 -0
- package/dist/agent/agent-pool.js.map +1 -0
- package/dist/agent/background-awareness.d.ts +55 -0
- package/dist/agent/background-awareness.d.ts.map +1 -0
- package/dist/agent/background-awareness.js +104 -0
- package/dist/agent/background-awareness.js.map +1 -0
- package/dist/agent/command-parser.d.ts +33 -0
- package/dist/agent/command-parser.d.ts.map +1 -0
- package/dist/agent/command-parser.js +120 -0
- package/dist/agent/command-parser.js.map +1 -0
- package/dist/agent/context-budget.d.ts +91 -0
- package/dist/agent/context-budget.d.ts.map +1 -0
- package/dist/agent/context-budget.js +219 -0
- package/dist/agent/context-budget.js.map +1 -0
- package/dist/agent/conversation.d.ts +190 -0
- package/dist/agent/conversation.d.ts.map +1 -0
- package/dist/agent/conversation.js +547 -0
- package/dist/agent/conversation.js.map +1 -0
- package/dist/agent/hooks.d.ts +72 -0
- package/dist/agent/hooks.d.ts.map +1 -0
- package/dist/agent/hooks.js +214 -0
- package/dist/agent/hooks.js.map +1 -0
- package/dist/agent/mcp-bridge.d.ts +13 -0
- package/dist/agent/mcp-bridge.d.ts.map +1 -0
- package/dist/agent/mcp-bridge.js +86 -0
- package/dist/agent/mcp-bridge.js.map +1 -0
- package/dist/agent/mcp-client.d.ts +24 -0
- package/dist/agent/mcp-client.d.ts.map +1 -0
- package/dist/agent/mcp-client.js +146 -0
- package/dist/agent/mcp-client.js.map +1 -0
- package/dist/agent/mcp-manager.d.ts +13 -0
- package/dist/agent/mcp-manager.d.ts.map +1 -0
- package/dist/agent/mcp-manager.js +84 -0
- package/dist/agent/mcp-manager.js.map +1 -0
- package/dist/agent/mcp-registry.d.ts +45 -0
- package/dist/agent/mcp-registry.d.ts.map +1 -0
- package/dist/agent/mcp-registry.js +98 -0
- package/dist/agent/mcp-registry.js.map +1 -0
- package/dist/agent/orchestrator.d.ts +14 -0
- package/dist/agent/orchestrator.d.ts.map +1 -0
- package/dist/agent/orchestrator.js +118 -0
- package/dist/agent/orchestrator.js.map +1 -0
- package/dist/agent/parser-adapter.d.ts +120 -0
- package/dist/agent/parser-adapter.d.ts.map +1 -0
- package/dist/agent/parser-adapter.js +265 -0
- package/dist/agent/parser-adapter.js.map +1 -0
- package/dist/agent/parsers/imports.d.ts +11 -0
- package/dist/agent/parsers/imports.d.ts.map +1 -0
- package/dist/agent/parsers/imports.js +94 -0
- package/dist/agent/parsers/imports.js.map +1 -0
- package/dist/agent/parsers/shell.d.ts +23 -0
- package/dist/agent/parsers/shell.d.ts.map +1 -0
- package/dist/agent/parsers/shell.js +200 -0
- package/dist/agent/parsers/shell.js.map +1 -0
- package/dist/agent/parsers/symbols.d.ts +17 -0
- package/dist/agent/parsers/symbols.d.ts.map +1 -0
- package/dist/agent/parsers/symbols.js +103 -0
- package/dist/agent/parsers/symbols.js.map +1 -0
- package/dist/agent/permissions.d.ts +65 -0
- package/dist/agent/permissions.d.ts.map +1 -0
- package/dist/agent/permissions.js +219 -0
- package/dist/agent/permissions.js.map +1 -0
- package/dist/agent/predictive-gate.d.ts +69 -0
- package/dist/agent/predictive-gate.d.ts.map +1 -0
- package/dist/agent/predictive-gate.js +128 -0
- package/dist/agent/predictive-gate.js.map +1 -0
- package/dist/agent/provider-cooldown.d.ts +144 -0
- package/dist/agent/provider-cooldown.d.ts.map +1 -0
- package/dist/agent/provider-cooldown.js +300 -0
- package/dist/agent/provider-cooldown.js.map +1 -0
- package/dist/agent/providers-manager.d.ts +109 -0
- package/dist/agent/providers-manager.d.ts.map +1 -0
- package/dist/agent/providers-manager.js +464 -0
- package/dist/agent/providers-manager.js.map +1 -0
- package/dist/agent/repo-map.d.ts +6 -0
- package/dist/agent/repo-map.d.ts.map +1 -0
- package/dist/agent/repo-map.js +221 -0
- package/dist/agent/repo-map.js.map +1 -0
- package/dist/agent/retry.d.ts +103 -0
- package/dist/agent/retry.d.ts.map +1 -0
- package/dist/agent/retry.js +276 -0
- package/dist/agent/retry.js.map +1 -0
- package/dist/agent/search/index.d.ts +61 -0
- package/dist/agent/search/index.d.ts.map +1 -0
- package/dist/agent/search/index.js +314 -0
- package/dist/agent/search/index.js.map +1 -0
- package/dist/agent/single-agent.d.ts +76 -0
- package/dist/agent/single-agent.d.ts.map +1 -0
- package/dist/agent/single-agent.js +697 -0
- package/dist/agent/single-agent.js.map +1 -0
- package/dist/agent/skills.d.ts +22 -0
- package/dist/agent/skills.d.ts.map +1 -0
- package/dist/agent/skills.js +139 -0
- package/dist/agent/skills.js.map +1 -0
- package/dist/agent/stream-glue.d.ts +85 -0
- package/dist/agent/stream-glue.d.ts.map +1 -0
- package/dist/agent/stream-glue.js +120 -0
- package/dist/agent/stream-glue.js.map +1 -0
- package/dist/agent/subagent.d.ts +72 -0
- package/dist/agent/subagent.d.ts.map +1 -0
- package/dist/agent/subagent.js +193 -0
- package/dist/agent/subagent.js.map +1 -0
- package/dist/agent/telemetry.d.ts +192 -0
- package/dist/agent/telemetry.d.ts.map +1 -0
- package/dist/agent/telemetry.js +400 -0
- package/dist/agent/telemetry.js.map +1 -0
- package/dist/agent/tokenizer.d.ts +42 -0
- package/dist/agent/tokenizer.d.ts.map +1 -0
- package/dist/agent/tokenizer.js +107 -0
- package/dist/agent/tokenizer.js.map +1 -0
- package/dist/agent/tool-executor.d.ts +289 -0
- package/dist/agent/tool-executor.d.ts.map +1 -0
- package/dist/agent/tool-executor.js +2519 -0
- package/dist/agent/tool-executor.js.map +1 -0
- package/dist/agent/web-impl.d.ts +2 -0
- package/dist/agent/web-impl.d.ts.map +1 -0
- package/dist/agent/web-impl.js +34 -0
- package/dist/agent/web-impl.js.map +1 -0
- package/dist/agent/web.d.ts +8 -0
- package/dist/agent/web.d.ts.map +1 -0
- package/dist/agent/web.js +8 -0
- package/dist/agent/web.js.map +1 -0
- package/dist/agent/worker-agent.d.ts +27 -0
- package/dist/agent/worker-agent.d.ts.map +1 -0
- package/dist/agent/worker-agent.js +503 -0
- package/dist/agent/worker-agent.js.map +1 -0
- package/dist/config.d.ts +162 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +138 -0
- package/dist/config.js.map +1 -0
- package/dist/context/fixo-md-watcher.d.ts +42 -0
- package/dist/context/fixo-md-watcher.d.ts.map +1 -0
- package/dist/context/fixo-md-watcher.js +126 -0
- package/dist/context/fixo-md-watcher.js.map +1 -0
- package/dist/context/fixo-md.d.ts +50 -0
- package/dist/context/fixo-md.d.ts.map +1 -0
- package/dist/context/fixo-md.js +118 -0
- package/dist/context/fixo-md.js.map +1 -0
- package/dist/context/todo.d.ts +65 -0
- package/dist/context/todo.d.ts.map +1 -0
- package/dist/context/todo.js +194 -0
- package/dist/context/todo.js.map +1 -0
- package/dist/git/git-manager.d.ts +33 -0
- package/dist/git/git-manager.d.ts.map +1 -0
- package/dist/git/git-manager.js +293 -0
- package/dist/git/git-manager.js.map +1 -0
- package/dist/git/git-ops.d.ts +10 -0
- package/dist/git/git-ops.d.ts.map +1 -0
- package/dist/git/git-ops.js +131 -0
- package/dist/git/git-ops.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +352 -0
- package/dist/index.js.map +1 -0
- package/dist/indexer.d.ts +30 -0
- package/dist/indexer.d.ts.map +1 -0
- package/dist/indexer.js +273 -0
- package/dist/indexer.js.map +1 -0
- package/dist/lsp/lsp-client.d.ts +24 -0
- package/dist/lsp/lsp-client.d.ts.map +1 -0
- package/dist/lsp/lsp-client.js +205 -0
- package/dist/lsp/lsp-client.js.map +1 -0
- package/dist/lsp/lsp-manager.d.ts +17 -0
- package/dist/lsp/lsp-manager.d.ts.map +1 -0
- package/dist/lsp/lsp-manager.js +154 -0
- package/dist/lsp/lsp-manager.js.map +1 -0
- package/dist/lsp/lsp-pre-save.d.ts +137 -0
- package/dist/lsp/lsp-pre-save.d.ts.map +1 -0
- package/dist/lsp/lsp-pre-save.js +245 -0
- package/dist/lsp/lsp-pre-save.js.map +1 -0
- package/dist/lsp/syntax-fallback.d.ts +83 -0
- package/dist/lsp/syntax-fallback.d.ts.map +1 -0
- package/dist/lsp/syntax-fallback.js +275 -0
- package/dist/lsp/syntax-fallback.js.map +1 -0
- package/dist/model-outcomes.d.ts +12 -0
- package/dist/model-outcomes.d.ts.map +1 -0
- package/dist/model-outcomes.js +46 -0
- package/dist/model-outcomes.js.map +1 -0
- package/dist/planner.d.ts +32 -0
- package/dist/planner.d.ts.map +1 -0
- package/dist/planner.js +163 -0
- package/dist/planner.js.map +1 -0
- package/dist/project-memory.d.ts +29 -0
- package/dist/project-memory.d.ts.map +1 -0
- package/dist/project-memory.js +349 -0
- package/dist/project-memory.js.map +1 -0
- package/dist/review.d.ts +2 -0
- package/dist/review.d.ts.map +1 -0
- package/dist/review.js +61 -0
- package/dist/review.js.map +1 -0
- package/dist/runtime/background-jobs.d.ts +97 -0
- package/dist/runtime/background-jobs.d.ts.map +1 -0
- package/dist/runtime/background-jobs.js +331 -0
- package/dist/runtime/background-jobs.js.map +1 -0
- package/dist/runtime/credential-vault.d.ts +124 -0
- package/dist/runtime/credential-vault.d.ts.map +1 -0
- package/dist/runtime/credential-vault.js +184 -0
- package/dist/runtime/credential-vault.js.map +1 -0
- package/dist/runtime/loop-trap.d.ts +197 -0
- package/dist/runtime/loop-trap.d.ts.map +1 -0
- package/dist/runtime/loop-trap.js +420 -0
- package/dist/runtime/loop-trap.js.map +1 -0
- package/dist/runtime/policy.d.ts +15 -0
- package/dist/runtime/policy.d.ts.map +1 -0
- package/dist/runtime/policy.js +60 -0
- package/dist/runtime/policy.js.map +1 -0
- package/dist/runtime/redaction.d.ts +66 -0
- package/dist/runtime/redaction.d.ts.map +1 -0
- package/dist/runtime/redaction.js +155 -0
- package/dist/runtime/redaction.js.map +1 -0
- package/dist/runtime/session-snapshots.d.ts +76 -0
- package/dist/runtime/session-snapshots.d.ts.map +1 -0
- package/dist/runtime/session-snapshots.js +166 -0
- package/dist/runtime/session-snapshots.js.map +1 -0
- package/dist/runtime/staging.d.ts +205 -0
- package/dist/runtime/staging.d.ts.map +1 -0
- package/dist/runtime/staging.js +526 -0
- package/dist/runtime/staging.js.map +1 -0
- package/dist/runtime/task-session.d.ts +95 -0
- package/dist/runtime/task-session.d.ts.map +1 -0
- package/dist/runtime/task-session.js +263 -0
- package/dist/runtime/task-session.js.map +1 -0
- package/dist/runtime/worktree.d.ts +55 -0
- package/dist/runtime/worktree.d.ts.map +1 -0
- package/dist/runtime/worktree.js +175 -0
- package/dist/runtime/worktree.js.map +1 -0
- package/dist/setup-wizard.d.ts +8 -0
- package/dist/setup-wizard.d.ts.map +1 -0
- package/dist/setup-wizard.js +73 -0
- package/dist/setup-wizard.js.map +1 -0
- package/dist/shared/content.d.ts +43 -0
- package/dist/shared/content.d.ts.map +1 -0
- package/dist/shared/content.js +61 -0
- package/dist/shared/content.js.map +1 -0
- package/dist/shared/types.d.ts +217 -0
- package/dist/shared/types.d.ts.map +1 -0
- package/dist/shared/types.js +3 -0
- package/dist/shared/types.js.map +1 -0
- package/dist/test-runner.d.ts +5 -0
- package/dist/test-runner.d.ts.map +1 -0
- package/dist/test-runner.js +42 -0
- package/dist/test-runner.js.map +1 -0
- package/dist/types.d.ts +85 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/ui/ascii.d.ts +23 -0
- package/dist/ui/ascii.d.ts.map +1 -0
- package/dist/ui/ascii.js +45 -0
- package/dist/ui/ascii.js.map +1 -0
- package/dist/ui/colors.d.ts +111 -0
- package/dist/ui/colors.d.ts.map +1 -0
- package/dist/ui/colors.js +166 -0
- package/dist/ui/colors.js.map +1 -0
- package/dist/ui/image-attach.d.ts +27 -0
- package/dist/ui/image-attach.d.ts.map +1 -0
- package/dist/ui/image-attach.js +100 -0
- package/dist/ui/image-attach.js.map +1 -0
- package/dist/ui/index.d.ts +18 -0
- package/dist/ui/index.d.ts.map +1 -0
- package/dist/ui/index.js +18 -0
- package/dist/ui/index.js.map +1 -0
- package/dist/ui/markdown-stream.d.ts +91 -0
- package/dist/ui/markdown-stream.d.ts.map +1 -0
- package/dist/ui/markdown-stream.js +524 -0
- package/dist/ui/markdown-stream.js.map +1 -0
- package/dist/ui/plan-renderer.d.ts +36 -0
- package/dist/ui/plan-renderer.d.ts.map +1 -0
- package/dist/ui/plan-renderer.js +79 -0
- package/dist/ui/plan-renderer.js.map +1 -0
- package/dist/ui/prompt.d.ts +11 -0
- package/dist/ui/prompt.d.ts.map +1 -0
- package/dist/ui/prompt.js +1960 -0
- package/dist/ui/prompt.js.map +1 -0
- package/dist/ui/render-primitives.d.ts +117 -0
- package/dist/ui/render-primitives.d.ts.map +1 -0
- package/dist/ui/render-primitives.js +322 -0
- package/dist/ui/render-primitives.js.map +1 -0
- package/dist/ui/render.d.ts +133 -0
- package/dist/ui/render.d.ts.map +1 -0
- package/dist/ui/render.js +547 -0
- package/dist/ui/render.js.map +1 -0
- package/dist/ui/session-header.d.ts +30 -0
- package/dist/ui/session-header.d.ts.map +1 -0
- package/dist/ui/session-header.js +74 -0
- package/dist/ui/session-header.js.map +1 -0
- package/dist/workspace-guard.d.ts +68 -0
- package/dist/workspace-guard.d.ts.map +1 -0
- package/dist/workspace-guard.js +168 -0
- package/dist/workspace-guard.js.map +1 -0
- package/dist/workspace-lock.d.ts +27 -0
- package/dist/workspace-lock.d.ts.map +1 -0
- package/dist/workspace-lock.js +95 -0
- package/dist/workspace-lock.js.map +1 -0
- package/package.json +63 -0
|
@@ -0,0 +1,214 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* hooks.ts — PreToolUse / PostToolUse hook engine.
|
|
3
|
+
*
|
|
4
|
+
* Phase 3.4 spec:
|
|
5
|
+
* - Two hook events supported: `PreToolUse` and `PostToolUse`.
|
|
6
|
+
* No `Stop`, `Notification`, or `SessionStart` (PRD §3.4
|
|
7
|
+
* hard rule).
|
|
8
|
+
* - Storage: `<cwd>/.fixo/hooks.json`. Schema is a map of
|
|
9
|
+
* event name → list of hook commands.
|
|
10
|
+
* - Hooks are *sync* `spawn` calls. The hook command's
|
|
11
|
+
* stdin receives a JSON document of the form
|
|
12
|
+
* `{event, tool, args, sessionId}`. The command's stdout
|
|
13
|
+
* (if any) is parsed as JSON
|
|
14
|
+
* `{decision, reason?, modifiedArgs?}`. A non-zero exit
|
|
15
|
+
* is treated as a deny with a default reason.
|
|
16
|
+
* - Pre-hook decisions: `allow` (pass), `deny` (block),
|
|
17
|
+
* `modify` (replace args before re-validation).
|
|
18
|
+
* - Post-hook decisions: `allow` (no-op), `deny` (force
|
|
19
|
+
* the result to be marked as denied — the tool has
|
|
20
|
+
* already executed, but the post-hook can flag it).
|
|
21
|
+
* `modifiedArgs` on a post-hook is not honoured (the
|
|
22
|
+
* tool has already run).
|
|
23
|
+
* - After a pre-hook `modify`, the modified args are
|
|
24
|
+
* re-validated through `WorkspaceGuard.resolve()` to
|
|
25
|
+
* guard against hook escape attempts.
|
|
26
|
+
* - A 5-second wall-clock timeout protects against
|
|
27
|
+
* runaway hooks.
|
|
28
|
+
* - Every fired hook emits a `hook_fired` telemetry event
|
|
29
|
+
* with the decision and the duration.
|
|
30
|
+
*/
|
|
31
|
+
import { spawnSync } from 'node:child_process';
|
|
32
|
+
import fs from 'node:fs';
|
|
33
|
+
import path from 'node:path';
|
|
34
|
+
import { recordTelemetry, telemetry } from './telemetry.js';
|
|
35
|
+
import { WorkspaceGuard } from '../workspace-guard.js';
|
|
36
|
+
const DEFAULT_TIMEOUT_MS = 5000;
|
|
37
|
+
const HOOKS_PATH_FOR = (cwd) => path.join(cwd, '.fixo', 'hooks.json');
|
|
38
|
+
/** Read `.fixo/hooks.json` from cwd. Returns null if absent. */
|
|
39
|
+
export function loadHooksFile(cwd) {
|
|
40
|
+
const p = HOOKS_PATH_FOR(cwd);
|
|
41
|
+
if (!fs.existsSync(p))
|
|
42
|
+
return null;
|
|
43
|
+
try {
|
|
44
|
+
const raw = JSON.parse(fs.readFileSync(p, 'utf-8'));
|
|
45
|
+
if (raw.version !== 1 || !Array.isArray(raw.hooks))
|
|
46
|
+
return null;
|
|
47
|
+
return raw;
|
|
48
|
+
}
|
|
49
|
+
catch {
|
|
50
|
+
return null;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
/** Save the hooks file. Used by /hooks slash command. */
|
|
54
|
+
export function saveHooksFile(cwd, file) {
|
|
55
|
+
const p = HOOKS_PATH_FOR(cwd);
|
|
56
|
+
try {
|
|
57
|
+
fs.mkdirSync(path.dirname(p), { recursive: true });
|
|
58
|
+
fs.writeFileSync(p, JSON.stringify(file, null, 2));
|
|
59
|
+
return { ok: true };
|
|
60
|
+
}
|
|
61
|
+
catch (err) {
|
|
62
|
+
return { ok: false, error: err.message };
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
/** Hooks for a given event. */
|
|
66
|
+
export function getHooksForEvent(cwd, event) {
|
|
67
|
+
const file = loadHooksFile(cwd);
|
|
68
|
+
if (!file)
|
|
69
|
+
return [];
|
|
70
|
+
return file.hooks.filter(h => h.event === event && (h.enabled ?? true));
|
|
71
|
+
}
|
|
72
|
+
/**
|
|
73
|
+
* Run a single hook synchronously with a 5s default timeout.
|
|
74
|
+
* The hook's stdin receives the JSON payload; the stdout
|
|
75
|
+
* (if present) is parsed as a HookCommandResult.
|
|
76
|
+
*/
|
|
77
|
+
function runHook(spec, payload, timeoutMs) {
|
|
78
|
+
const payloadJson = JSON.stringify(payload);
|
|
79
|
+
let result;
|
|
80
|
+
try {
|
|
81
|
+
result = spawnSync(spec.command, spec.args ?? [], {
|
|
82
|
+
input: payloadJson,
|
|
83
|
+
encoding: 'utf-8',
|
|
84
|
+
timeout: timeoutMs,
|
|
85
|
+
maxBuffer: 1024 * 1024, // 1 MiB stdout cap
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
catch (err) {
|
|
89
|
+
return { decision: 'deny', reason: `hook failed to spawn: ${err.message}` };
|
|
90
|
+
}
|
|
91
|
+
if (result.error) {
|
|
92
|
+
return { decision: 'deny', reason: `hook error: ${result.error.message}` };
|
|
93
|
+
}
|
|
94
|
+
if (result.status !== 0) {
|
|
95
|
+
const stderrText = typeof result.stderr === 'string' ? result.stderr : '';
|
|
96
|
+
return {
|
|
97
|
+
decision: 'deny',
|
|
98
|
+
reason: `hook exited with status ${result.status}${stderrText ? `: ${stderrText.slice(0, 200)}` : ''}`,
|
|
99
|
+
};
|
|
100
|
+
}
|
|
101
|
+
const stdout = (typeof result.stdout === 'string' ? result.stdout : '').trim();
|
|
102
|
+
if (stdout.length === 0) {
|
|
103
|
+
// Hook succeeded with no output → default allow.
|
|
104
|
+
return { decision: 'allow' };
|
|
105
|
+
}
|
|
106
|
+
try {
|
|
107
|
+
const parsed = JSON.parse(stdout);
|
|
108
|
+
const decision = parsed.decision === 'allow' || parsed.decision === 'deny' || parsed.decision === 'modify'
|
|
109
|
+
? parsed.decision
|
|
110
|
+
: 'allow';
|
|
111
|
+
return {
|
|
112
|
+
decision,
|
|
113
|
+
reason: typeof parsed.reason === 'string' ? parsed.reason : undefined,
|
|
114
|
+
modifiedArgs: parsed.modifiedArgs && typeof parsed.modifiedArgs === 'object'
|
|
115
|
+
? parsed.modifiedArgs
|
|
116
|
+
: undefined,
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
catch {
|
|
120
|
+
// Hook wrote garbage — treat as a soft allow (the
|
|
121
|
+
// contract is "stdout is JSON; if not, ignore").
|
|
122
|
+
return { decision: 'allow' };
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
/**
|
|
126
|
+
* Fire all hooks for an event, in declaration order. The
|
|
127
|
+
* first non-`allow` decision wins. Returns the aggregated
|
|
128
|
+
* result.
|
|
129
|
+
*/
|
|
130
|
+
export function fireHooks(cwd, event, payload) {
|
|
131
|
+
const start = Date.now();
|
|
132
|
+
const specs = getHooksForEvent(cwd, event);
|
|
133
|
+
let aggDecision = 'allow';
|
|
134
|
+
let aggReason;
|
|
135
|
+
let aggModifiedArgs;
|
|
136
|
+
let lastHookId = null;
|
|
137
|
+
for (const spec of specs) {
|
|
138
|
+
const res = runHook(spec, { ...payload, event, cwd }, spec.timeoutMs ?? DEFAULT_TIMEOUT_MS);
|
|
139
|
+
lastHookId = spec.id;
|
|
140
|
+
recordTelemetry(telemetry.hookFired({
|
|
141
|
+
hook: `${spec.id}:${payload.tool}:${res.decision}`,
|
|
142
|
+
phase: event === 'PreToolUse' ? 'pre' : 'post',
|
|
143
|
+
matched: res.decision !== 'allow',
|
|
144
|
+
durationMs: Date.now() - start,
|
|
145
|
+
}));
|
|
146
|
+
if (res.decision === 'deny') {
|
|
147
|
+
return {
|
|
148
|
+
fired: true,
|
|
149
|
+
decision: 'deny',
|
|
150
|
+
reason: res.reason ?? `hook ${spec.id} denied`,
|
|
151
|
+
durationMs: Date.now() - start,
|
|
152
|
+
hookId: spec.id,
|
|
153
|
+
};
|
|
154
|
+
}
|
|
155
|
+
if (res.decision === 'modify') {
|
|
156
|
+
aggDecision = 'modify';
|
|
157
|
+
if (res.modifiedArgs)
|
|
158
|
+
aggModifiedArgs = res.modifiedArgs;
|
|
159
|
+
if (res.reason)
|
|
160
|
+
aggReason = res.reason;
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
return {
|
|
164
|
+
fired: specs.length > 0,
|
|
165
|
+
decision: aggDecision,
|
|
166
|
+
reason: aggReason,
|
|
167
|
+
modifiedArgs: aggModifiedArgs,
|
|
168
|
+
durationMs: Date.now() - start,
|
|
169
|
+
hookId: lastHookId,
|
|
170
|
+
};
|
|
171
|
+
}
|
|
172
|
+
/**
|
|
173
|
+
* Apply a pre-hook's `modify` decision: re-validate
|
|
174
|
+
* `modifiedArgs` through `WorkspaceGuard.resolve()` to
|
|
175
|
+
* prevent an attacker-controlled hook from writing outside
|
|
176
|
+
* the workspace. The check is a *best-effort* scan: any
|
|
177
|
+
* `path`/`filePath`/`cwd` field in the args must resolve
|
|
178
|
+
* inside `cwd` (relative paths are fine).
|
|
179
|
+
*/
|
|
180
|
+
export function applyModifiedArgs(cwd, original, modified) {
|
|
181
|
+
const guard = new WorkspaceGuard(cwd);
|
|
182
|
+
// We re-validate any path-shaped field.
|
|
183
|
+
const pathKeys = ['path', 'filePath', 'filepath', 'src', 'dst', 'cwd', 'outputPath', 'inputPath'];
|
|
184
|
+
for (const key of pathKeys) {
|
|
185
|
+
if (!(key in modified))
|
|
186
|
+
continue;
|
|
187
|
+
const value = modified[key];
|
|
188
|
+
if (typeof value !== 'string')
|
|
189
|
+
continue;
|
|
190
|
+
// If the value is a relative path or equals cwd, it
|
|
191
|
+
// resolves inside the workspace by construction. Only
|
|
192
|
+
// absolute paths are checked.
|
|
193
|
+
if (!path.isAbsolute(value))
|
|
194
|
+
continue;
|
|
195
|
+
if (value === cwd)
|
|
196
|
+
continue;
|
|
197
|
+
try {
|
|
198
|
+
guard.resolve(value, 'modified-args');
|
|
199
|
+
}
|
|
200
|
+
catch (err) {
|
|
201
|
+
return {
|
|
202
|
+
ok: false,
|
|
203
|
+
args: original,
|
|
204
|
+
reason: `pre-hook attempted to inject out-of-workspace path: ${err.message}`,
|
|
205
|
+
};
|
|
206
|
+
}
|
|
207
|
+
}
|
|
208
|
+
return { ok: true, args: modified };
|
|
209
|
+
}
|
|
210
|
+
/** Get the hooks file path for a cwd (test-only introspection). */
|
|
211
|
+
export function getHooksPath(cwd) {
|
|
212
|
+
return HOOKS_PATH_FOR(cwd);
|
|
213
|
+
}
|
|
214
|
+
//# sourceMappingURL=hooks.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.js","sourceRoot":"","sources":["../../src/agent/hooks.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAkDvD,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAEhC,MAAM,cAAc,GAAG,CAAC,GAAW,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;AAE9E,gEAAgE;AAChE,MAAM,UAAU,aAAa,CAAC,GAAW;IACvC,MAAM,CAAC,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IACnC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,OAAO,CAAC,CAAc,CAAC;QACjE,IAAI,GAAG,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAChE,OAAO,GAAG,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,yDAAyD;AACzD,MAAM,UAAU,aAAa,CAAC,GAAW,EAAE,IAAe;IACxD,MAAM,CAAC,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC;QACH,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACnD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;IACtB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC;IACtD,CAAC;AACH,CAAC;AAED,+BAA+B;AAC/B,MAAM,UAAU,gBAAgB,CAAC,GAAW,EAAE,KAAgB;IAC5D,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;IAChC,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IACrB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC;AAC1E,CAAC;AAED;;;;GAIG;AACH,SAAS,OAAO,CAAC,IAAc,EAAE,OAAoB,EAAE,SAAiB;IACtE,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC5C,IAAI,MAAoC,CAAC;IACzC,IAAI,CAAC;QACH,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE;YAChD,KAAK,EAAE,WAAW;YAClB,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,SAAS;YAClB,SAAS,EAAE,IAAI,GAAG,IAAI,EAAE,mBAAmB;SAC5C,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,yBAA0B,GAAa,CAAC,OAAO,EAAE,EAAE,CAAC;IACzF,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;IAC7E,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,UAAU,GAAG,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,OAAO;YACL,QAAQ,EAAE,MAAM;YAChB,MAAM,EAAE,2BAA2B,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;SACvG,CAAC;IACJ,CAAC;IACD,MAAM,MAAM,GAAG,CAAC,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC/E,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,iDAAiD;QACjD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAC/B,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAA+B,CAAC;QAChE,MAAM,QAAQ,GAAiB,MAAM,CAAC,QAAQ,KAAK,OAAO,IAAI,MAAM,CAAC,QAAQ,KAAK,MAAM,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ;YACtH,CAAC,CAAC,MAAM,CAAC,QAAQ;YACjB,CAAC,CAAC,OAAO,CAAC;QACZ,OAAO;YACL,QAAQ;YACR,MAAM,EAAE,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;YACrE,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ;gBAC1E,CAAC,CAAE,MAAM,CAAC,YAAwC;gBAClD,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,kDAAkD;QAClD,iDAAiD;QACjD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAC/B,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS,CACvB,GAAW,EACX,KAAgB,EAChB,OAA2C;IAE3C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACzB,MAAM,KAAK,GAAG,gBAAgB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC3C,IAAI,WAAW,GAAiB,OAAO,CAAC;IACxC,IAAI,SAA6B,CAAC;IAClC,IAAI,eAAoD,CAAC;IACzD,IAAI,UAAU,GAAkB,IAAI,CAAC;IACrC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,EAAE,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC,SAAS,IAAI,kBAAkB,CAAC,CAAC;QAC5F,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC;QACrB,eAAe,CACb,SAAS,CAAC,SAAS,CAAC;YAClB,IAAI,EAAE,GAAG,IAAI,CAAC,EAAE,IAAI,OAAO,CAAC,IAAI,IAAI,GAAG,CAAC,QAAQ,EAAE;YAClD,KAAK,EAAE,KAAK,KAAK,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM;YAC9C,OAAO,EAAE,GAAG,CAAC,QAAQ,KAAK,OAAO;YACjC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;SAC/B,CAAC,CACH,CAAC;QACF,IAAI,GAAG,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO;gBACL,KAAK,EAAE,IAAI;gBACX,QAAQ,EAAE,MAAM;gBAChB,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,QAAQ,IAAI,CAAC,EAAE,SAAS;gBAC9C,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;gBAC9B,MAAM,EAAE,IAAI,CAAC,EAAE;aAChB,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC9B,WAAW,GAAG,QAAQ,CAAC;YACvB,IAAI,GAAG,CAAC,YAAY;gBAAE,eAAe,GAAG,GAAG,CAAC,YAAY,CAAC;YACzD,IAAI,GAAG,CAAC,MAAM;gBAAE,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC;QACzC,CAAC;IACH,CAAC;IACD,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC;QACvB,QAAQ,EAAE,WAAW;QACrB,MAAM,EAAE,SAAS;QACjB,YAAY,EAAE,eAAe;QAC7B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;QAC9B,MAAM,EAAE,UAAU;KACnB,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,iBAAiB,CAC/B,GAAW,EACX,QAAiC,EACjC,QAAiC;IAEjC,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,GAAG,CAAC,CAAC;IACtC,wCAAwC;IACxC,MAAM,QAAQ,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC;IAClG,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,GAAG,IAAI,QAAQ,CAAC;YAAE,SAAS;QACjC,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,SAAS;QACxC,oDAAoD;QACpD,sDAAsD;QACtD,8BAA8B;QAC9B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;YAAE,SAAS;QACtC,IAAI,KAAK,KAAK,GAAG;YAAE,SAAS;QAC5B,IAAI,CAAC;YACH,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,uDAAwD,GAAa,CAAC,OAAO,EAAE;aACxF,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;AACtC,CAAC;AAED,mEAAmE;AACnE,MAAM,UAAU,YAAY,CAAC,GAAW;IACtC,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { ChatToolDefinition } from '../shared/types.js';
|
|
2
|
+
export declare class McpBridgeManager {
|
|
3
|
+
private clients;
|
|
4
|
+
private toolsMap;
|
|
5
|
+
private registeredTools;
|
|
6
|
+
initialize(cwd: string): Promise<void>;
|
|
7
|
+
private loadProjectMcpConfig;
|
|
8
|
+
getTools(): ChatToolDefinition[];
|
|
9
|
+
hasTool(name: string): boolean;
|
|
10
|
+
executeTool(name: string, args: any): Promise<string>;
|
|
11
|
+
shutdown(): void;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=mcp-bridge.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-bridge.d.ts","sourceRoot":"","sources":["../../src/agent/mcp-bridge.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAE7D,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,OAAO,CAAgC;IAC/C,OAAO,CAAC,QAAQ,CAAkE;IAClF,OAAO,CAAC,eAAe,CAA4B;IAE7C,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA8B5C,OAAO,CAAC,oBAAoB;IAgB5B,QAAQ,IAAI,kBAAkB,EAAE;IAIhC,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIxB,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;IAiB3D,QAAQ,IAAI,IAAI;CAQjB"}
|
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import yaml from 'js-yaml';
|
|
4
|
+
import { McpClient } from './mcp-client.js';
|
|
5
|
+
export class McpBridgeManager {
|
|
6
|
+
clients = new Map();
|
|
7
|
+
toolsMap = new Map();
|
|
8
|
+
registeredTools = [];
|
|
9
|
+
async initialize(cwd) {
|
|
10
|
+
const config = this.loadProjectMcpConfig(cwd);
|
|
11
|
+
if (!config || !config.mcpServers)
|
|
12
|
+
return;
|
|
13
|
+
for (const [name, serverConfig] of Object.entries(config.mcpServers)) {
|
|
14
|
+
const client = new McpClient(name, serverConfig);
|
|
15
|
+
const ok = await client.start();
|
|
16
|
+
if (ok) {
|
|
17
|
+
this.clients.set(name, client);
|
|
18
|
+
try {
|
|
19
|
+
const tools = await client.listTools();
|
|
20
|
+
for (const tool of tools) {
|
|
21
|
+
const registeredName = `mcp_local_${name}_${tool.name}`;
|
|
22
|
+
this.toolsMap.set(registeredName, { client, originalName: tool.name });
|
|
23
|
+
this.registeredTools.push({
|
|
24
|
+
type: 'function',
|
|
25
|
+
function: {
|
|
26
|
+
name: registeredName,
|
|
27
|
+
description: tool.description || '',
|
|
28
|
+
parameters: tool.inputSchema || { type: 'object', properties: {} },
|
|
29
|
+
},
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
catch (e) {
|
|
34
|
+
console.error(`[MCP Bridge] Failed to load tools for local server ${name}:`, e);
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
loadProjectMcpConfig(cwd) {
|
|
40
|
+
const ymlPath = path.join(cwd, '.fixo.yml');
|
|
41
|
+
const yamlPath = path.join(cwd, '.fixo.yaml');
|
|
42
|
+
let configPath = fs.existsSync(ymlPath) ? ymlPath : fs.existsSync(yamlPath) ? yamlPath : null;
|
|
43
|
+
if (configPath) {
|
|
44
|
+
try {
|
|
45
|
+
const content = fs.readFileSync(configPath, 'utf-8');
|
|
46
|
+
return yaml.load(content);
|
|
47
|
+
}
|
|
48
|
+
catch (e) {
|
|
49
|
+
console.warn(`[MCP Bridge] Failed to parse project MCP config:`, e);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
return null;
|
|
53
|
+
}
|
|
54
|
+
getTools() {
|
|
55
|
+
return this.registeredTools;
|
|
56
|
+
}
|
|
57
|
+
hasTool(name) {
|
|
58
|
+
return this.toolsMap.has(name);
|
|
59
|
+
}
|
|
60
|
+
async executeTool(name, args) {
|
|
61
|
+
const mapping = this.toolsMap.get(name);
|
|
62
|
+
if (!mapping) {
|
|
63
|
+
throw new Error(`Local MCP tool ${name} not registered`);
|
|
64
|
+
}
|
|
65
|
+
const result = await mapping.client.callTool(mapping.originalName, args);
|
|
66
|
+
if (result && Array.isArray(result.content)) {
|
|
67
|
+
return result.content
|
|
68
|
+
.map((c) => {
|
|
69
|
+
if (c.type === 'text')
|
|
70
|
+
return c.text;
|
|
71
|
+
return JSON.stringify(c);
|
|
72
|
+
})
|
|
73
|
+
.join('\n');
|
|
74
|
+
}
|
|
75
|
+
return JSON.stringify(result);
|
|
76
|
+
}
|
|
77
|
+
shutdown() {
|
|
78
|
+
for (const client of this.clients.values()) {
|
|
79
|
+
client.stop();
|
|
80
|
+
}
|
|
81
|
+
this.clients.clear();
|
|
82
|
+
this.toolsMap.clear();
|
|
83
|
+
this.registeredTools = [];
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
//# sourceMappingURL=mcp-bridge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-bridge.js","sourceRoot":"","sources":["../../src/agent/mcp-bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,IAAI,MAAM,SAAS,CAAC;AAC3B,OAAO,EAAE,SAAS,EAAwB,MAAM,iBAAiB,CAAC;AAGlE,MAAM,OAAO,gBAAgB;IACnB,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;IACvC,QAAQ,GAAG,IAAI,GAAG,EAAuD,CAAC;IAC1E,eAAe,GAAyB,EAAE,CAAC;IAEnD,KAAK,CAAC,UAAU,CAAC,GAAW;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU;YAAE,OAAO;QAE1C,KAAK,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YACrE,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,IAAI,EAAE,YAA+B,CAAC,CAAC;YACpE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,EAAE,EAAE,CAAC;gBACP,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;oBACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,MAAM,cAAc,GAAG,aAAa,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACxD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;wBACvE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;4BACxB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACR,IAAI,EAAE,cAAc;gCACpB,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gCACnC,UAAU,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;6BACnE;yBACF,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,OAAO,CAAC,KAAK,CAAC,sDAAsD,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,GAAW;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;QAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QAC9C,IAAI,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC;QAE9F,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACrD,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,kDAAkD,EAAE,CAAC,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,OAAO,CAAC,IAAY;QAClB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,IAAS;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzE,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5C,OAAO,MAAM,CAAC,OAAO;iBAClB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE;gBACd,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;oBAAE,OAAO,CAAC,CAAC,IAAI,CAAC;gBACrC,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAC3B,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,QAAQ;QACN,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;IAC5B,CAAC;CACF"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
export interface McpServerConfig {
|
|
2
|
+
command: string;
|
|
3
|
+
args?: string[];
|
|
4
|
+
env?: Record<string, string>;
|
|
5
|
+
}
|
|
6
|
+
export declare class McpClient {
|
|
7
|
+
private serverName;
|
|
8
|
+
private config;
|
|
9
|
+
private process;
|
|
10
|
+
private nextId;
|
|
11
|
+
private pendingRequests;
|
|
12
|
+
private rl;
|
|
13
|
+
private initialized;
|
|
14
|
+
constructor(serverName: string, config: McpServerConfig);
|
|
15
|
+
start(): Promise<boolean>;
|
|
16
|
+
private handleMessage;
|
|
17
|
+
private sendRequest;
|
|
18
|
+
private sendNotification;
|
|
19
|
+
private initializeConnection;
|
|
20
|
+
listTools(): Promise<any[]>;
|
|
21
|
+
callTool(name: string, args: any): Promise<any>;
|
|
22
|
+
stop(): void;
|
|
23
|
+
}
|
|
24
|
+
//# sourceMappingURL=mcp-client.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../../src/agent/mcp-client.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC9B;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,MAAM,CAAK;IACnB,OAAO,CAAC,eAAe,CAAoF;IAC3G,OAAO,CAAC,EAAE,CAAmC;IAC7C,OAAO,CAAC,WAAW,CAAS;gBAEhB,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe;IAKjD,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC;IAsC/B,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,WAAW;IA4BnB,OAAO,CAAC,gBAAgB;YAUV,oBAAoB;IAS5B,SAAS,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAW3B,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IAQrD,IAAI;CAcL"}
|
|
@@ -0,0 +1,146 @@
|
|
|
1
|
+
import { spawn } from 'child_process';
|
|
2
|
+
import readline from 'readline';
|
|
3
|
+
export class McpClient {
|
|
4
|
+
serverName;
|
|
5
|
+
config;
|
|
6
|
+
process = null;
|
|
7
|
+
nextId = 1;
|
|
8
|
+
pendingRequests = new Map();
|
|
9
|
+
rl = null;
|
|
10
|
+
initialized = false;
|
|
11
|
+
constructor(serverName, config) {
|
|
12
|
+
this.serverName = serverName;
|
|
13
|
+
this.config = config;
|
|
14
|
+
}
|
|
15
|
+
async start() {
|
|
16
|
+
try {
|
|
17
|
+
const env = { ...process.env, ...this.config.env };
|
|
18
|
+
this.process = spawn(this.config.command, this.config.args || [], {
|
|
19
|
+
env,
|
|
20
|
+
stdio: ['pipe', 'pipe', 'inherit'],
|
|
21
|
+
});
|
|
22
|
+
this.process.on('error', (err) => {
|
|
23
|
+
console.error(`[MCP Client] Failed to start server ${this.serverName}:`, err);
|
|
24
|
+
});
|
|
25
|
+
this.process.on('exit', (code, signal) => {
|
|
26
|
+
if (code !== 0 && code !== null) {
|
|
27
|
+
console.warn(`[MCP Client] Server ${this.serverName} exited with code ${code}`);
|
|
28
|
+
}
|
|
29
|
+
});
|
|
30
|
+
this.rl = readline.createInterface({
|
|
31
|
+
input: this.process.stdout,
|
|
32
|
+
terminal: false,
|
|
33
|
+
});
|
|
34
|
+
this.rl.on('line', (line) => {
|
|
35
|
+
this.handleMessage(line);
|
|
36
|
+
});
|
|
37
|
+
// Initialize connection
|
|
38
|
+
await this.initializeConnection();
|
|
39
|
+
this.initialized = true;
|
|
40
|
+
return true;
|
|
41
|
+
}
|
|
42
|
+
catch (e) {
|
|
43
|
+
console.error(`[MCP Client] Error starting server ${this.serverName}:`, e);
|
|
44
|
+
this.stop();
|
|
45
|
+
return false;
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
handleMessage(line) {
|
|
49
|
+
try {
|
|
50
|
+
const msg = JSON.parse(line);
|
|
51
|
+
if (msg.jsonrpc === '2.0' && msg.id !== undefined) {
|
|
52
|
+
const pending = this.pendingRequests.get(msg.id);
|
|
53
|
+
if (pending) {
|
|
54
|
+
this.pendingRequests.delete(msg.id);
|
|
55
|
+
if (msg.error) {
|
|
56
|
+
pending.reject(new Error(msg.error.message || JSON.stringify(msg.error)));
|
|
57
|
+
}
|
|
58
|
+
else {
|
|
59
|
+
pending.resolve(msg.result);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
catch (e) {
|
|
65
|
+
// Ignored / malformed line
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
sendRequest(method, params) {
|
|
69
|
+
return new Promise((resolve, reject) => {
|
|
70
|
+
if (!this.process || !this.process.stdin) {
|
|
71
|
+
return reject(new Error(`MCP server ${this.serverName} process not running`));
|
|
72
|
+
}
|
|
73
|
+
const id = this.nextId++;
|
|
74
|
+
const req = {
|
|
75
|
+
jsonrpc: '2.0',
|
|
76
|
+
id,
|
|
77
|
+
method,
|
|
78
|
+
params,
|
|
79
|
+
};
|
|
80
|
+
const timeoutId = setTimeout(() => {
|
|
81
|
+
if (this.pendingRequests.has(id)) {
|
|
82
|
+
this.pendingRequests.delete(id);
|
|
83
|
+
reject(new Error(`MCP request timeout for method ${method} on server ${this.serverName}`));
|
|
84
|
+
}
|
|
85
|
+
}, 20000);
|
|
86
|
+
this.pendingRequests.set(id, {
|
|
87
|
+
resolve: (val) => { clearTimeout(timeoutId); resolve(val); },
|
|
88
|
+
reject: (err) => { clearTimeout(timeoutId); reject(err); }
|
|
89
|
+
});
|
|
90
|
+
this.process.stdin.write(JSON.stringify(req) + '\n');
|
|
91
|
+
});
|
|
92
|
+
}
|
|
93
|
+
sendNotification(method, params) {
|
|
94
|
+
if (!this.process || !this.process.stdin)
|
|
95
|
+
return;
|
|
96
|
+
const notif = {
|
|
97
|
+
jsonrpc: '2.0',
|
|
98
|
+
method,
|
|
99
|
+
params,
|
|
100
|
+
};
|
|
101
|
+
this.process.stdin.write(JSON.stringify(notif) + '\n');
|
|
102
|
+
}
|
|
103
|
+
async initializeConnection() {
|
|
104
|
+
await this.sendRequest('initialize', {
|
|
105
|
+
protocolVersion: '2024-11-05',
|
|
106
|
+
capabilities: {},
|
|
107
|
+
clientInfo: { name: 'fixo-cli', version: '1.0.0' },
|
|
108
|
+
});
|
|
109
|
+
this.sendNotification('notifications/initialized');
|
|
110
|
+
}
|
|
111
|
+
async listTools() {
|
|
112
|
+
if (!this.initialized)
|
|
113
|
+
return [];
|
|
114
|
+
try {
|
|
115
|
+
const res = await this.sendRequest('tools/list', {});
|
|
116
|
+
return res.tools || [];
|
|
117
|
+
}
|
|
118
|
+
catch (e) {
|
|
119
|
+
console.error(`[MCP Client] Failed to list tools for ${this.serverName}:`, e);
|
|
120
|
+
return [];
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
async callTool(name, args) {
|
|
124
|
+
if (!this.initialized)
|
|
125
|
+
throw new Error(`MCP Client ${this.serverName} not initialized`);
|
|
126
|
+
return this.sendRequest('tools/call', {
|
|
127
|
+
name,
|
|
128
|
+
arguments: args,
|
|
129
|
+
});
|
|
130
|
+
}
|
|
131
|
+
stop() {
|
|
132
|
+
if (this.rl) {
|
|
133
|
+
this.rl.close();
|
|
134
|
+
this.rl = null;
|
|
135
|
+
}
|
|
136
|
+
if (this.process) {
|
|
137
|
+
this.process.kill();
|
|
138
|
+
this.process = null;
|
|
139
|
+
}
|
|
140
|
+
for (const [id, pending] of this.pendingRequests.entries()) {
|
|
141
|
+
pending.reject(new Error(`MCP client ${this.serverName} stopped`));
|
|
142
|
+
}
|
|
143
|
+
this.pendingRequests.clear();
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
//# sourceMappingURL=mcp-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../../src/agent/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,eAAe,CAAC;AACzD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAQhC,MAAM,OAAO,SAAS;IACZ,UAAU,CAAS;IACnB,MAAM,CAAkB;IACxB,OAAO,GAAwB,IAAI,CAAC;IACpC,MAAM,GAAG,CAAC,CAAC;IACX,eAAe,GAAG,IAAI,GAAG,EAAyE,CAAC;IACnG,EAAE,GAA8B,IAAI,CAAC;IACrC,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,UAAkB,EAAE,MAAuB;QACrD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YACnD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE;gBAChE,GAAG;gBACH,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC;aACnC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC/B,OAAO,CAAC,KAAK,CAAC,uCAAuC,IAAI,CAAC,UAAU,GAAG,EAAE,GAAG,CAAC,CAAC;YAChF,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;gBACvC,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;oBAChC,OAAO,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,UAAU,qBAAqB,IAAI,EAAE,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;gBACjC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,MAAO;gBAC3B,QAAQ,EAAE,KAAK;aAChB,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBAC1B,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,wBAAwB;YACxB,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAClC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,sCAAsC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,CAAC,CAAC;YAC3E,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK,IAAI,GAAG,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;gBAClD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACjD,IAAI,OAAO,EAAE,CAAC;oBACZ,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACpC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;wBACd,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC5E,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,2BAA2B;QAC7B,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,MAAc,EAAE,MAAW;QAC7C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACzC,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,UAAU,sBAAsB,CAAC,CAAC,CAAC;YAChF,CAAC;YACD,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG;gBACV,OAAO,EAAE,KAAK;gBACd,EAAE;gBACF,MAAM;gBACN,MAAM;aACP,CAAC;YAEF,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,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,kCAAkC,MAAM,cAAc,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAC7F,CAAC;YACH,CAAC,EAAE,KAAK,CAAC,CAAC;YAEV,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE;gBAC3B,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC5D,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;aAC3D,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,MAAc,EAAE,MAAY;QACnD,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK;YAAE,OAAO;QACjD,MAAM,KAAK,GAAG;YACZ,OAAO,EAAE,KAAK;YACd,MAAM;YACN,MAAM;SACP,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;IACzD,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;YACnC,eAAe,EAAE,YAAY;YAC7B,YAAY,EAAE,EAAE;YAChB,UAAU,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE;SACnD,CAAC,CAAC;QACH,IAAI,CAAC,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,SAAS;QACb,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YACrD,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,yCAAyC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,CAAC,CAAC;YAC9E,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAY,EAAE,IAAS;QACpC,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,UAAU,kBAAkB,CAAC,CAAC;QACxF,OAAO,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;YACpC,IAAI;YACJ,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;IACL,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACtB,CAAC;QACD,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,CAAC;YAC3D,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,UAAU,UAAU,CAAC,CAAC,CAAC;QACrE,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { ChatToolDefinition } from '../shared/types.js';
|
|
2
|
+
export declare class McpManager {
|
|
3
|
+
private clients;
|
|
4
|
+
private toolsMap;
|
|
5
|
+
private registeredTools;
|
|
6
|
+
initialize(): Promise<void>;
|
|
7
|
+
private loadMcpConfig;
|
|
8
|
+
getTools(): ChatToolDefinition[];
|
|
9
|
+
hasTool(name: string): boolean;
|
|
10
|
+
executeTool(name: string, args: any): Promise<string>;
|
|
11
|
+
shutdown(): void;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=mcp-manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-manager.d.ts","sourceRoot":"","sources":["../../src/agent/mcp-manager.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAE7D,qBAAa,UAAU;IACrB,OAAO,CAAC,OAAO,CAAgC;IAC/C,OAAO,CAAC,QAAQ,CAAkE;IAClF,OAAO,CAAC,eAAe,CAA4B;IAE7C,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA8BjC,OAAO,CAAC,aAAa;IAarB,QAAQ,IAAI,kBAAkB,EAAE;IAIhC,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIxB,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;IAiB3D,QAAQ,IAAI,IAAI;CAQjB"}
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
import os from 'os';
|
|
2
|
+
import path from 'path';
|
|
3
|
+
import fs from 'fs';
|
|
4
|
+
import { McpClient } from './mcp-client.js';
|
|
5
|
+
export class McpManager {
|
|
6
|
+
clients = new Map();
|
|
7
|
+
toolsMap = new Map();
|
|
8
|
+
registeredTools = [];
|
|
9
|
+
async initialize() {
|
|
10
|
+
const config = this.loadMcpConfig();
|
|
11
|
+
if (!config || !config.mcpServers)
|
|
12
|
+
return;
|
|
13
|
+
for (const [name, serverConfig] of Object.entries(config.mcpServers)) {
|
|
14
|
+
const client = new McpClient(name, serverConfig);
|
|
15
|
+
const ok = await client.start();
|
|
16
|
+
if (ok) {
|
|
17
|
+
this.clients.set(name, client);
|
|
18
|
+
try {
|
|
19
|
+
const tools = await client.listTools();
|
|
20
|
+
for (const tool of tools) {
|
|
21
|
+
const registeredName = `mcp_${name}_${tool.name}`;
|
|
22
|
+
this.toolsMap.set(registeredName, { client, originalName: tool.name });
|
|
23
|
+
this.registeredTools.push({
|
|
24
|
+
type: 'function',
|
|
25
|
+
function: {
|
|
26
|
+
name: registeredName,
|
|
27
|
+
description: tool.description || '',
|
|
28
|
+
parameters: tool.inputSchema || { type: 'object', properties: {} },
|
|
29
|
+
},
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
catch (e) {
|
|
34
|
+
console.error(`[MCP Manager] Failed to load tools for server ${name}:`, e);
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
loadMcpConfig() {
|
|
40
|
+
const homeDir = os.homedir();
|
|
41
|
+
const configPath = path.join(homeDir, '.freellmapi', 'mcp.json');
|
|
42
|
+
if (fs.existsSync(configPath)) {
|
|
43
|
+
try {
|
|
44
|
+
return JSON.parse(fs.readFileSync(configPath, 'utf-8'));
|
|
45
|
+
}
|
|
46
|
+
catch (e) {
|
|
47
|
+
console.warn(`[MCP Manager] Failed to parse MCP config:`, e);
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
return null;
|
|
51
|
+
}
|
|
52
|
+
getTools() {
|
|
53
|
+
return this.registeredTools;
|
|
54
|
+
}
|
|
55
|
+
hasTool(name) {
|
|
56
|
+
return this.toolsMap.has(name);
|
|
57
|
+
}
|
|
58
|
+
async executeTool(name, args) {
|
|
59
|
+
const mapping = this.toolsMap.get(name);
|
|
60
|
+
if (!mapping) {
|
|
61
|
+
throw new Error(`MCP tool ${name} not registered`);
|
|
62
|
+
}
|
|
63
|
+
const result = await mapping.client.callTool(mapping.originalName, args);
|
|
64
|
+
if (result && Array.isArray(result.content)) {
|
|
65
|
+
return result.content
|
|
66
|
+
.map((c) => {
|
|
67
|
+
if (c.type === 'text')
|
|
68
|
+
return c.text;
|
|
69
|
+
return JSON.stringify(c);
|
|
70
|
+
})
|
|
71
|
+
.join('\n');
|
|
72
|
+
}
|
|
73
|
+
return JSON.stringify(result);
|
|
74
|
+
}
|
|
75
|
+
shutdown() {
|
|
76
|
+
for (const client of this.clients.values()) {
|
|
77
|
+
client.stop();
|
|
78
|
+
}
|
|
79
|
+
this.clients.clear();
|
|
80
|
+
this.toolsMap.clear();
|
|
81
|
+
this.registeredTools = [];
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
//# sourceMappingURL=mcp-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-manager.js","sourceRoot":"","sources":["../../src/agent/mcp-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,SAAS,EAAwB,MAAM,iBAAiB,CAAC;AAGlE,MAAM,OAAO,UAAU;IACb,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;IACvC,QAAQ,GAAG,IAAI,GAAG,EAAuD,CAAC;IAC1E,eAAe,GAAyB,EAAE,CAAC;IAEnD,KAAK,CAAC,UAAU;QACd,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACpC,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU;YAAE,OAAO;QAE1C,KAAK,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YACrE,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,IAAI,EAAE,YAA+B,CAAC,CAAC;YACpE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,EAAE,EAAE,CAAC;gBACP,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;oBACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,MAAM,cAAc,GAAG,OAAO,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBAClD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;wBACvE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;4BACxB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACR,IAAI,EAAE,cAAc;gCACpB,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gCACnC,UAAU,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;6BACnE;yBACF,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,OAAO,CAAC,KAAK,CAAC,iDAAiD,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC;gBAC7E,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEO,aAAa;QACnB,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,UAAU,CAAC,CAAC;QACjE,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1D,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,2CAA2C,EAAE,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,OAAO,CAAC,IAAY;QAClB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,IAAS;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,YAAY,IAAI,iBAAiB,CAAC,CAAC;QACrD,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzE,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5C,OAAO,MAAM,CAAC,OAAO;iBAClB,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE;gBACd,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;oBAAE,OAAO,CAAC,CAAC,IAAI,CAAC;gBACrC,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAC3B,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,QAAQ;QACN,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;IAC5B,CAAC;CACF"}
|