@tachu/extensions 1.0.0-beta.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +77 -0
- package/LICENSE +201 -0
- package/README.md +825 -0
- package/README_ZH.md +815 -0
- package/dist/backends/file.d.ts +18 -0
- package/dist/backends/file.d.ts.map +1 -0
- package/dist/backends/file.js +69 -0
- package/dist/backends/file.js.map +1 -0
- package/dist/backends/index.d.ts +4 -0
- package/dist/backends/index.d.ts.map +1 -0
- package/dist/backends/index.js +4 -0
- package/dist/backends/index.js.map +1 -0
- package/dist/backends/terminal.d.ts +18 -0
- package/dist/backends/terminal.d.ts.map +1 -0
- package/dist/backends/terminal.js +57 -0
- package/dist/backends/terminal.js.map +1 -0
- package/dist/backends/web.d.ts +18 -0
- package/dist/backends/web.d.ts.map +1 -0
- package/dist/backends/web.js +53 -0
- package/dist/backends/web.js.map +1 -0
- package/dist/common/net.d.ts +31 -0
- package/dist/common/net.d.ts.map +1 -0
- package/dist/common/net.js +162 -0
- package/dist/common/net.js.map +1 -0
- package/dist/common/path.d.ts +18 -0
- package/dist/common/path.d.ts.map +1 -0
- package/dist/common/path.js +34 -0
- package/dist/common/path.js.map +1 -0
- package/dist/common/process.d.ts +19 -0
- package/dist/common/process.d.ts.map +1 -0
- package/dist/common/process.js +67 -0
- package/dist/common/process.js.map +1 -0
- package/dist/index.d.ts +9 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +9 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/index.d.ts +3 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +3 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/sse-adapter.d.ts +70 -0
- package/dist/mcp/sse-adapter.d.ts.map +1 -0
- package/dist/mcp/sse-adapter.js +176 -0
- package/dist/mcp/sse-adapter.js.map +1 -0
- package/dist/mcp/stdio-adapter.d.ts +73 -0
- package/dist/mcp/stdio-adapter.d.ts.map +1 -0
- package/dist/mcp/stdio-adapter.js +178 -0
- package/dist/mcp/stdio-adapter.js.map +1 -0
- package/dist/observability/index.d.ts +3 -0
- package/dist/observability/index.d.ts.map +1 -0
- package/dist/observability/index.js +3 -0
- package/dist/observability/index.js.map +1 -0
- package/dist/observability/jsonl-emitter.d.ts +58 -0
- package/dist/observability/jsonl-emitter.d.ts.map +1 -0
- package/dist/observability/jsonl-emitter.js +96 -0
- package/dist/observability/jsonl-emitter.js.map +1 -0
- package/dist/observability/otel-emitter.d.ts +52 -0
- package/dist/observability/otel-emitter.d.ts.map +1 -0
- package/dist/observability/otel-emitter.js +143 -0
- package/dist/observability/otel-emitter.js.map +1 -0
- package/dist/providers/anthropic.d.ts +54 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +298 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/index.d.ts +4 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +4 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/mock.d.ts +38 -0
- package/dist/providers/mock.d.ts.map +1 -0
- package/dist/providers/mock.js +79 -0
- package/dist/providers/mock.js.map +1 -0
- package/dist/providers/openai.d.ts +61 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +299 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/rules/index.d.ts +9 -0
- package/dist/rules/index.d.ts.map +1 -0
- package/dist/rules/index.js +15 -0
- package/dist/rules/index.js.map +1 -0
- package/dist/rules/no-hallucination.md +11 -0
- package/dist/rules/no-sensitive-output.md +11 -0
- package/dist/rules/prefer-concise-response.md +11 -0
- package/dist/rules/require-tool-verification.md +11 -0
- package/dist/tools/apply-patch/descriptor.md +27 -0
- package/dist/tools/apply-patch/executor.d.ts +19 -0
- package/dist/tools/apply-patch/executor.d.ts.map +1 -0
- package/dist/tools/apply-patch/executor.js +190 -0
- package/dist/tools/apply-patch/executor.js.map +1 -0
- package/dist/tools/fetch-url/descriptor.md +38 -0
- package/dist/tools/fetch-url/executor.d.ts +20 -0
- package/dist/tools/fetch-url/executor.d.ts.map +1 -0
- package/dist/tools/fetch-url/executor.js +34 -0
- package/dist/tools/fetch-url/executor.js.map +1 -0
- package/dist/tools/index.d.ts +12 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +191 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/list-dir/descriptor.md +29 -0
- package/dist/tools/list-dir/executor.d.ts +22 -0
- package/dist/tools/list-dir/executor.d.ts.map +1 -0
- package/dist/tools/list-dir/executor.js +46 -0
- package/dist/tools/list-dir/executor.js.map +1 -0
- package/dist/tools/read-file/descriptor.md +28 -0
- package/dist/tools/read-file/executor.d.ts +15 -0
- package/dist/tools/read-file/executor.d.ts.map +1 -0
- package/dist/tools/read-file/executor.js +22 -0
- package/dist/tools/read-file/executor.js.map +1 -0
- package/dist/tools/run-shell/descriptor.md +39 -0
- package/dist/tools/run-shell/executor.d.ts +20 -0
- package/dist/tools/run-shell/executor.d.ts.map +1 -0
- package/dist/tools/run-shell/executor.js +76 -0
- package/dist/tools/run-shell/executor.js.map +1 -0
- package/dist/tools/search-code/descriptor.md +31 -0
- package/dist/tools/search-code/executor.d.ts +23 -0
- package/dist/tools/search-code/executor.d.ts.map +1 -0
- package/dist/tools/search-code/executor.js +113 -0
- package/dist/tools/search-code/executor.js.map +1 -0
- package/dist/tools/shared.d.ts +21 -0
- package/dist/tools/shared.d.ts.map +1 -0
- package/dist/tools/shared.js +12 -0
- package/dist/tools/shared.js.map +1 -0
- package/dist/tools/write-file/descriptor.md +30 -0
- package/dist/tools/write-file/executor.d.ts +16 -0
- package/dist/tools/write-file/executor.d.ts.map +1 -0
- package/dist/tools/write-file/executor.js +18 -0
- package/dist/tools/write-file/executor.js.map +1 -0
- package/dist/transformers/document-to-text.d.ts +23 -0
- package/dist/transformers/document-to-text.d.ts.map +1 -0
- package/dist/transformers/document-to-text.js +69 -0
- package/dist/transformers/document-to-text.js.map +1 -0
- package/dist/transformers/image-to-text.d.ts +38 -0
- package/dist/transformers/image-to-text.d.ts.map +1 -0
- package/dist/transformers/image-to-text.js +81 -0
- package/dist/transformers/image-to-text.js.map +1 -0
- package/dist/transformers/index.d.ts +3 -0
- package/dist/transformers/index.d.ts.map +1 -0
- package/dist/transformers/index.js +3 -0
- package/dist/transformers/index.js.map +1 -0
- package/dist/vector/index.d.ts +3 -0
- package/dist/vector/index.d.ts.map +1 -0
- package/dist/vector/index.js +3 -0
- package/dist/vector/index.js.map +1 -0
- package/dist/vector/local-fs.d.ts +76 -0
- package/dist/vector/local-fs.d.ts.map +1 -0
- package/dist/vector/local-fs.js +153 -0
- package/dist/vector/local-fs.js.map +1 -0
- package/dist/vector/qdrant.d.ts +65 -0
- package/dist/vector/qdrant.d.ts.map +1 -0
- package/dist/vector/qdrant.js +176 -0
- package/dist/vector/qdrant.js.map +1 -0
- package/package.json +74 -0
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { ToolExecutor } from "../shared";
|
|
2
|
+
interface ApplyPatchInput {
|
|
3
|
+
patch: string;
|
|
4
|
+
basePath?: string;
|
|
5
|
+
}
|
|
6
|
+
interface ApplyPatchOutput {
|
|
7
|
+
applied: Array<{
|
|
8
|
+
file: string;
|
|
9
|
+
status: "ok" | "failed";
|
|
10
|
+
reason?: string;
|
|
11
|
+
}>;
|
|
12
|
+
success: boolean;
|
|
13
|
+
}
|
|
14
|
+
/**
|
|
15
|
+
* 应用 unified diff Tool 执行器。
|
|
16
|
+
*/
|
|
17
|
+
export declare const applyPatchExecutor: ToolExecutor<ApplyPatchInput, ApplyPatchOutput>;
|
|
18
|
+
export {};
|
|
19
|
+
//# sourceMappingURL=executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../../src/tools/apply-patch/executor.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAI9C,UAAU,eAAe;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,UAAU,gBAAgB;IACxB,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,IAAI,GAAG,QAAQ,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC3E,OAAO,EAAE,OAAO,CAAC;CAClB;AA2JD;;GAEG;AACH,eAAO,MAAM,kBAAkB,EAAE,YAAY,CAAC,eAAe,EAAE,gBAAgB,CA0E9E,CAAC"}
|
|
@@ -0,0 +1,190 @@
|
|
|
1
|
+
import { mkdir, readFile, writeFile, rm } from "node:fs/promises";
|
|
2
|
+
import { dirname } from "node:path";
|
|
3
|
+
import { ValidationError } from "@tachu/core";
|
|
4
|
+
import { assertNotAborted } from "../shared";
|
|
5
|
+
import { resolveWorkspacePath, toWorkspaceRelativePath } from "../../common/path";
|
|
6
|
+
const normalizePatchPath = (value) => value.replace(/^[ab]\//, "").trim();
|
|
7
|
+
const splitContent = (content) => {
|
|
8
|
+
if (content.length === 0) {
|
|
9
|
+
return { lines: [], trailingNewline: false };
|
|
10
|
+
}
|
|
11
|
+
const normalized = content.replaceAll("\r\n", "\n");
|
|
12
|
+
const trailingNewline = normalized.endsWith("\n");
|
|
13
|
+
const lines = normalized.split("\n");
|
|
14
|
+
if (trailingNewline) {
|
|
15
|
+
lines.pop();
|
|
16
|
+
}
|
|
17
|
+
return { lines, trailingNewline };
|
|
18
|
+
};
|
|
19
|
+
const parsePatch = (rawPatch) => {
|
|
20
|
+
const lines = rawPatch.replaceAll("\r\n", "\n").split("\n");
|
|
21
|
+
const files = [];
|
|
22
|
+
let index = 0;
|
|
23
|
+
while (index < lines.length) {
|
|
24
|
+
const line = lines[index] ?? "";
|
|
25
|
+
if (!line.startsWith("--- ")) {
|
|
26
|
+
index += 1;
|
|
27
|
+
continue;
|
|
28
|
+
}
|
|
29
|
+
const oldPath = normalizePatchPath(line.slice(4));
|
|
30
|
+
const newLine = lines[index + 1] ?? "";
|
|
31
|
+
if (!newLine.startsWith("+++ ")) {
|
|
32
|
+
throw new ValidationError("VALIDATION_PATCH_FORMAT", "patch 缺少 +++ 文件头");
|
|
33
|
+
}
|
|
34
|
+
const newPath = normalizePatchPath(newLine.slice(4));
|
|
35
|
+
index += 2;
|
|
36
|
+
const hunks = [];
|
|
37
|
+
while (index < lines.length) {
|
|
38
|
+
const hunkHeader = lines[index] ?? "";
|
|
39
|
+
if (hunkHeader.startsWith("--- ")) {
|
|
40
|
+
break;
|
|
41
|
+
}
|
|
42
|
+
if (!hunkHeader.startsWith("@@ ")) {
|
|
43
|
+
index += 1;
|
|
44
|
+
continue;
|
|
45
|
+
}
|
|
46
|
+
const match = /^@@ -(\d+)(?:,\d+)? \+(\d+)(?:,\d+)? @@/.exec(hunkHeader);
|
|
47
|
+
if (!match) {
|
|
48
|
+
throw new ValidationError("VALIDATION_PATCH_FORMAT", `无效 hunk 头: ${hunkHeader}`);
|
|
49
|
+
}
|
|
50
|
+
const oldStart = Number(match[1]);
|
|
51
|
+
const newStart = Number(match[2]);
|
|
52
|
+
index += 1;
|
|
53
|
+
const hunkLines = [];
|
|
54
|
+
while (index < lines.length) {
|
|
55
|
+
const hunkLine = lines[index] ?? "";
|
|
56
|
+
if (hunkLine.startsWith("@@ ") || hunkLine.startsWith("--- ")) {
|
|
57
|
+
break;
|
|
58
|
+
}
|
|
59
|
+
if (hunkLine === "\") {
|
|
60
|
+
index += 1;
|
|
61
|
+
continue;
|
|
62
|
+
}
|
|
63
|
+
const marker = hunkLine[0];
|
|
64
|
+
if (marker !== " " && marker !== "+" && marker !== "-") {
|
|
65
|
+
throw new ValidationError("VALIDATION_PATCH_FORMAT", `非法 hunk 行: ${hunkLine}`);
|
|
66
|
+
}
|
|
67
|
+
hunkLines.push({ type: marker, content: hunkLine.slice(1) });
|
|
68
|
+
index += 1;
|
|
69
|
+
}
|
|
70
|
+
hunks.push({ oldStart, newStart, lines: hunkLines });
|
|
71
|
+
}
|
|
72
|
+
files.push({ oldPath, newPath, hunks });
|
|
73
|
+
}
|
|
74
|
+
if (files.length === 0) {
|
|
75
|
+
throw new ValidationError("VALIDATION_PATCH_EMPTY", "patch 不包含任何文件");
|
|
76
|
+
}
|
|
77
|
+
return files;
|
|
78
|
+
};
|
|
79
|
+
const applyFilePatch = (original, patch) => {
|
|
80
|
+
const { lines: sourceLines, trailingNewline } = splitContent(original);
|
|
81
|
+
const result = [];
|
|
82
|
+
let cursor = 0;
|
|
83
|
+
for (const hunk of patch.hunks) {
|
|
84
|
+
const hunkStart = Math.max(0, hunk.oldStart - 1);
|
|
85
|
+
if (hunkStart < cursor) {
|
|
86
|
+
throw new ValidationError("VALIDATION_PATCH_CONFLICT", "patch hunk 重叠,无法应用");
|
|
87
|
+
}
|
|
88
|
+
result.push(...sourceLines.slice(cursor, hunkStart));
|
|
89
|
+
let sourceIndex = hunkStart;
|
|
90
|
+
for (const line of hunk.lines) {
|
|
91
|
+
if (line.type === " ") {
|
|
92
|
+
const actual = sourceLines[sourceIndex];
|
|
93
|
+
if (actual !== line.content) {
|
|
94
|
+
throw new ValidationError("VALIDATION_PATCH_CONFLICT", `上下文不匹配: expected "${line.content}" got "${actual ?? "<eof>"}"`);
|
|
95
|
+
}
|
|
96
|
+
result.push(actual);
|
|
97
|
+
sourceIndex += 1;
|
|
98
|
+
continue;
|
|
99
|
+
}
|
|
100
|
+
if (line.type === "-") {
|
|
101
|
+
const actual = sourceLines[sourceIndex];
|
|
102
|
+
if (actual !== line.content) {
|
|
103
|
+
throw new ValidationError("VALIDATION_PATCH_CONFLICT", `删除行不匹配: expected "${line.content}" got "${actual ?? "<eof>"}"`);
|
|
104
|
+
}
|
|
105
|
+
sourceIndex += 1;
|
|
106
|
+
continue;
|
|
107
|
+
}
|
|
108
|
+
result.push(line.content);
|
|
109
|
+
}
|
|
110
|
+
cursor = sourceIndex;
|
|
111
|
+
}
|
|
112
|
+
result.push(...sourceLines.slice(cursor));
|
|
113
|
+
return { content: result.join("\n"), trailingNewline };
|
|
114
|
+
};
|
|
115
|
+
/**
|
|
116
|
+
* 应用 unified diff Tool 执行器。
|
|
117
|
+
*/
|
|
118
|
+
export const applyPatchExecutor = async (input, context) => {
|
|
119
|
+
assertNotAborted(context.abortSignal);
|
|
120
|
+
const baseRoot = resolveWorkspacePath(context.workspaceRoot, input.basePath ?? ".");
|
|
121
|
+
const filePatches = parsePatch(input.patch);
|
|
122
|
+
const backups = new Map();
|
|
123
|
+
const applied = [];
|
|
124
|
+
const rollback = async () => {
|
|
125
|
+
for (const [path, snapshot] of backups.entries()) {
|
|
126
|
+
if (!snapshot.exists) {
|
|
127
|
+
await rm(path, { force: true }).catch(() => undefined);
|
|
128
|
+
continue;
|
|
129
|
+
}
|
|
130
|
+
await mkdir(dirname(path), { recursive: true });
|
|
131
|
+
const suffix = snapshot.trailingNewline ? "\n" : "";
|
|
132
|
+
await writeFile(path, `${snapshot.content}${suffix}`);
|
|
133
|
+
}
|
|
134
|
+
};
|
|
135
|
+
try {
|
|
136
|
+
for (const patch of filePatches) {
|
|
137
|
+
assertNotAborted(context.abortSignal);
|
|
138
|
+
const logicalTarget = patch.newPath === "/dev/null" ? patch.oldPath : patch.newPath;
|
|
139
|
+
const absoluteTarget = resolveWorkspacePath(baseRoot, logicalTarget);
|
|
140
|
+
if (!backups.has(absoluteTarget)) {
|
|
141
|
+
const existing = await readFile(absoluteTarget, "utf8").catch(() => null);
|
|
142
|
+
if (existing === null) {
|
|
143
|
+
backups.set(absoluteTarget, { exists: false, content: "", trailingNewline: false });
|
|
144
|
+
}
|
|
145
|
+
else {
|
|
146
|
+
const split = splitContent(existing);
|
|
147
|
+
backups.set(absoluteTarget, {
|
|
148
|
+
exists: true,
|
|
149
|
+
content: split.lines.join("\n"),
|
|
150
|
+
trailingNewline: split.trailingNewline,
|
|
151
|
+
});
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
const original = backups.get(absoluteTarget)?.exists
|
|
155
|
+
? (await readFile(absoluteTarget, "utf8").catch(() => "") ?? "")
|
|
156
|
+
: "";
|
|
157
|
+
const next = applyFilePatch(original, patch);
|
|
158
|
+
if (patch.newPath === "/dev/null") {
|
|
159
|
+
await rm(absoluteTarget, { force: true });
|
|
160
|
+
}
|
|
161
|
+
else {
|
|
162
|
+
await mkdir(dirname(absoluteTarget), { recursive: true });
|
|
163
|
+
const suffix = next.trailingNewline ? "\n" : "";
|
|
164
|
+
await writeFile(absoluteTarget, `${next.content}${suffix}`);
|
|
165
|
+
}
|
|
166
|
+
applied.push({
|
|
167
|
+
file: toWorkspaceRelativePath(context.workspaceRoot, absoluteTarget),
|
|
168
|
+
status: "ok",
|
|
169
|
+
});
|
|
170
|
+
}
|
|
171
|
+
return { applied, success: true };
|
|
172
|
+
}
|
|
173
|
+
catch (error) {
|
|
174
|
+
await rollback();
|
|
175
|
+
const reason = error instanceof Error ? error.message : String(error);
|
|
176
|
+
if (applied.length === 0) {
|
|
177
|
+
applied.push({ file: "<none>", status: "failed", reason });
|
|
178
|
+
}
|
|
179
|
+
else {
|
|
180
|
+
const previous = applied[applied.length - 1];
|
|
181
|
+
applied[applied.length - 1] = {
|
|
182
|
+
file: previous?.file ?? "<unknown>",
|
|
183
|
+
status: "failed",
|
|
184
|
+
reason,
|
|
185
|
+
};
|
|
186
|
+
}
|
|
187
|
+
return { applied, success: false };
|
|
188
|
+
}
|
|
189
|
+
};
|
|
190
|
+
//# sourceMappingURL=executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../src/tools/apply-patch/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAE9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAmClF,MAAM,kBAAkB,GAAG,CAAC,KAAa,EAAU,EAAE,CACnD,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AAEtC,MAAM,YAAY,GAAG,CAAC,OAAe,EAAiD,EAAE;IACtF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC;IAC/C,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACpD,MAAM,eAAe,GAAG,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAClD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,eAAe,EAAE,CAAC;QACpB,KAAK,CAAC,GAAG,EAAE,CAAC;IACd,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;AACpC,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAe,EAAE;IACnD,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAgB,EAAE,CAAC;IAC9B,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,KAAK,IAAI,CAAC,CAAC;YACX,SAAS;QACX,CAAC;QACD,MAAM,OAAO,GAAG,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,eAAe,CAAC,yBAAyB,EAAE,kBAAkB,CAAC,CAAC;QAC3E,CAAC;QACD,MAAM,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,KAAK,IAAI,CAAC,CAAC;QAEX,MAAM,KAAK,GAAW,EAAE,CAAC;QACzB,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACtC,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAClC,MAAM;YACR,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAClC,KAAK,IAAI,CAAC,CAAC;gBACX,SAAS;YACX,CAAC;YACD,MAAM,KAAK,GAAG,yCAAyC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzE,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,IAAI,eAAe,CAAC,yBAAyB,EAAE,cAAc,UAAU,EAAE,CAAC,CAAC;YACnF,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAClC,KAAK,IAAI,CAAC,CAAC;YAEX,MAAM,SAAS,GAAe,EAAE,CAAC;YACjC,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;gBACpC,IAAI,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC9D,MAAM;gBACR,CAAC;gBACD,IAAI,QAAQ,KAAK,8BAA8B,EAAE,CAAC;oBAChD,KAAK,IAAI,CAAC,CAAC;oBACX,SAAS;gBACX,CAAC;gBACD,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;gBAC3B,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;oBACvD,MAAM,IAAI,eAAe,CAAC,yBAAyB,EAAE,cAAc,QAAQ,EAAE,CAAC,CAAC;gBACjF,CAAC;gBACD,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC7D,KAAK,IAAI,CAAC,CAAC;YACb,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,eAAe,CAAC,wBAAwB,EAAE,eAAe,CAAC,CAAC;IACvE,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,CACrB,QAAgB,EAChB,KAAgB,EAC+B,EAAE;IACjD,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACvE,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QACjD,IAAI,SAAS,GAAG,MAAM,EAAE,CAAC;YACvB,MAAM,IAAI,eAAe,CAAC,2BAA2B,EAAE,oBAAoB,CAAC,CAAC;QAC/E,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;QACrD,IAAI,WAAW,GAAG,SAAS,CAAC;QAE5B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;gBACxC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;oBAC5B,MAAM,IAAI,eAAe,CACvB,2BAA2B,EAC3B,qBAAqB,IAAI,CAAC,OAAO,UAAU,MAAM,IAAI,OAAO,GAAG,CAChE,CAAC;gBACJ,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpB,WAAW,IAAI,CAAC,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;gBACxC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;oBAC5B,MAAM,IAAI,eAAe,CACvB,2BAA2B,EAC3B,qBAAqB,IAAI,CAAC,OAAO,UAAU,MAAM,IAAI,OAAO,GAAG,CAChE,CAAC;gBACJ,CAAC;gBACD,WAAW,IAAI,CAAC,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,CAAC;QACD,MAAM,GAAG,WAAW,CAAC;IACvB,CAAC;IACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAC1C,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,CAAC;AACzD,CAAC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAoD,KAAK,EACtF,KAAK,EACL,OAAO,EACP,EAAE;IACF,gBAAgB,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,oBAAoB,CAAC,OAAO,CAAC,aAAa,EAAE,KAAK,CAAC,QAAQ,IAAI,GAAG,CAAC,CAAC;IACpF,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC5C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAChD,MAAM,OAAO,GAAsE,EAAE,CAAC;IAEtF,MAAM,QAAQ,GAAG,KAAK,IAAmB,EAAE;QACzC,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACjD,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACrB,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBACvD,SAAS;YACX,CAAC;YACD,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YACpD,MAAM,SAAS,CAAC,IAAI,EAAE,GAAG,QAAQ,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC,CAAC;QACxD,CAAC;IACH,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YAChC,gBAAgB,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;YACpF,MAAM,cAAc,GAAG,oBAAoB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;YACrE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;gBACjC,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBAC1E,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtF,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;oBACrC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE;wBAC1B,MAAM,EAAE,IAAI;wBACZ,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;wBAC/B,eAAe,EAAE,KAAK,CAAC,eAAe;qBACvC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,MAAM;gBAClD,CAAC,CAAC,CAAE,MAAM,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAY,IAAI,EAAE,CAAC;gBAC5E,CAAC,CAAC,EAAE,CAAC;YAEP,MAAM,IAAI,GAAG,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC7C,IAAI,KAAK,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;gBAClC,MAAM,EAAE,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC1D,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChD,MAAM,SAAS,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC,CAAC;YAC9D,CAAC;YACD,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC,aAAa,EAAE,cAAc,CAAC;gBACpE,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACpC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,QAAQ,EAAE,CAAC;QACjB,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACtE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7D,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7C,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG;gBAC5B,IAAI,EAAE,QAAQ,EAAE,IAAI,IAAI,WAAW;gBACnC,MAAM,EAAE,QAAQ;gBAChB,MAAM;aACP,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IACrC,CAAC;AACH,CAAC,CAAC"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
---
|
|
2
|
+
kind: tool
|
|
3
|
+
name: fetch-url
|
|
4
|
+
description: 发送 HTTP 请求并返回响应内容
|
|
5
|
+
sideEffect: readonly
|
|
6
|
+
idempotent: false
|
|
7
|
+
requiresApproval: false
|
|
8
|
+
timeout: 15000
|
|
9
|
+
inputSchema:
|
|
10
|
+
type: object
|
|
11
|
+
properties:
|
|
12
|
+
url:
|
|
13
|
+
type: string
|
|
14
|
+
method:
|
|
15
|
+
type: string
|
|
16
|
+
enum: [GET, POST]
|
|
17
|
+
headers:
|
|
18
|
+
type: object
|
|
19
|
+
body:
|
|
20
|
+
type: string
|
|
21
|
+
timeoutMs:
|
|
22
|
+
type: number
|
|
23
|
+
required: [url]
|
|
24
|
+
outputSchema:
|
|
25
|
+
type: object
|
|
26
|
+
properties:
|
|
27
|
+
status:
|
|
28
|
+
type: number
|
|
29
|
+
headers:
|
|
30
|
+
type: object
|
|
31
|
+
body:
|
|
32
|
+
type: string
|
|
33
|
+
truncated:
|
|
34
|
+
type: boolean
|
|
35
|
+
execute: fetch-url
|
|
36
|
+
---
|
|
37
|
+
|
|
38
|
+
执行受控网络请求,默认阻止私网地址并限制响应体大小。
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { ToolExecutor } from "../shared";
|
|
2
|
+
interface FetchUrlInput {
|
|
3
|
+
url: string;
|
|
4
|
+
method?: "GET" | "POST";
|
|
5
|
+
headers?: Record<string, string>;
|
|
6
|
+
body?: string;
|
|
7
|
+
timeoutMs?: number;
|
|
8
|
+
}
|
|
9
|
+
interface FetchUrlOutput {
|
|
10
|
+
status: number;
|
|
11
|
+
headers: Record<string, string>;
|
|
12
|
+
body: string;
|
|
13
|
+
truncated: boolean;
|
|
14
|
+
}
|
|
15
|
+
/**
|
|
16
|
+
* URL 抓取 Tool 执行器。
|
|
17
|
+
*/
|
|
18
|
+
export declare const fetchUrlExecutor: ToolExecutor<FetchUrlInput, FetchUrlOutput>;
|
|
19
|
+
export {};
|
|
20
|
+
//# sourceMappingURL=executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../../src/tools/fetch-url/executor.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAG9C,UAAU,aAAa;IACrB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,UAAU,cAAc;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;CACpB;AAID;;GAEG;AACH,eAAO,MAAM,gBAAgB,EAAE,YAAY,CAAC,aAAa,EAAE,cAAc,CAgCxE,CAAC"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import { assertPublicUrl, readResponseBodyWithLimit, withAbortTimeout } from "../../common/net";
|
|
2
|
+
import { assertNotAborted } from "../shared";
|
|
3
|
+
const MAX_BODY_BYTES = 5 * 1024 * 1024;
|
|
4
|
+
/**
|
|
5
|
+
* URL 抓取 Tool 执行器。
|
|
6
|
+
*/
|
|
7
|
+
export const fetchUrlExecutor = async (input, context) => {
|
|
8
|
+
assertNotAborted(context.abortSignal);
|
|
9
|
+
const parsed = await assertPublicUrl(input.url);
|
|
10
|
+
const timeout = withAbortTimeout(context.abortSignal, input.timeoutMs ?? 15_000, "TIMEOUT_FETCH_URL");
|
|
11
|
+
try {
|
|
12
|
+
const response = await fetch(parsed, {
|
|
13
|
+
method: input.method ?? "GET",
|
|
14
|
+
headers: input.headers,
|
|
15
|
+
body: input.body,
|
|
16
|
+
signal: timeout.signal,
|
|
17
|
+
});
|
|
18
|
+
const bodyResult = await readResponseBodyWithLimit(response, MAX_BODY_BYTES);
|
|
19
|
+
const headers = {};
|
|
20
|
+
response.headers.forEach((value, key) => {
|
|
21
|
+
headers[key] = value;
|
|
22
|
+
});
|
|
23
|
+
return {
|
|
24
|
+
status: response.status,
|
|
25
|
+
headers,
|
|
26
|
+
body: bodyResult.body,
|
|
27
|
+
truncated: bodyResult.truncated,
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
finally {
|
|
31
|
+
timeout.cleanup();
|
|
32
|
+
}
|
|
33
|
+
};
|
|
34
|
+
//# sourceMappingURL=executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../src/tools/fetch-url/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,yBAAyB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEhG,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAiB7C,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;AAEvC;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAgD,KAAK,EAChF,KAAK,EACL,OAAO,EACP,EAAE;IACF,gBAAgB,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACtC,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAChD,MAAM,OAAO,GAAG,gBAAgB,CAC9B,OAAO,CAAC,WAAW,EACnB,KAAK,CAAC,SAAS,IAAI,MAAM,EACzB,mBAAmB,CACpB,CAAC;IACF,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,EAAE;YACnC,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,KAAK;YAC7B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,yBAAyB,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;QAC7E,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACvB,CAAC,CAAC,CAAC;QACH,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,OAAO;YACP,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,SAAS,EAAE,UAAU,CAAC,SAAS;SAChC,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,OAAO,CAAC,OAAO,EAAE,CAAC;IACpB,CAAC;AACH,CAAC,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { ToolDescriptor } from "@tachu/core";
|
|
2
|
+
import type { ToolExecutor } from "./shared";
|
|
3
|
+
/**
|
|
4
|
+
* 内置工具描述符列表。
|
|
5
|
+
*/
|
|
6
|
+
export declare const toolDescriptors: ToolDescriptor[];
|
|
7
|
+
/**
|
|
8
|
+
* 工具执行函数注册表。
|
|
9
|
+
*/
|
|
10
|
+
export declare const toolExecutors: Record<string, ToolExecutor>;
|
|
11
|
+
export type { ToolExecutor } from "./shared";
|
|
12
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAS7C;;GAEG;AACH,eAAO,MAAM,eAAe,EAAE,cAAc,EAuK3C,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAQtD,CAAC;AAEF,YAAY,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC"}
|
|
@@ -0,0 +1,191 @@
|
|
|
1
|
+
import { readFileExecutor } from "./read-file/executor";
|
|
2
|
+
import { writeFileExecutor } from "./write-file/executor";
|
|
3
|
+
import { listDirExecutor } from "./list-dir/executor";
|
|
4
|
+
import { searchCodeExecutor } from "./search-code/executor";
|
|
5
|
+
import { fetchUrlExecutor } from "./fetch-url/executor";
|
|
6
|
+
import { runShellExecutor } from "./run-shell/executor";
|
|
7
|
+
import { applyPatchExecutor } from "./apply-patch/executor";
|
|
8
|
+
/**
|
|
9
|
+
* 内置工具描述符列表。
|
|
10
|
+
*/
|
|
11
|
+
export const toolDescriptors = [
|
|
12
|
+
{
|
|
13
|
+
kind: "tool",
|
|
14
|
+
name: "read-file",
|
|
15
|
+
description: "读取工作区内的文件内容",
|
|
16
|
+
sideEffect: "readonly",
|
|
17
|
+
idempotent: true,
|
|
18
|
+
requiresApproval: false,
|
|
19
|
+
timeout: 5000,
|
|
20
|
+
inputSchema: {
|
|
21
|
+
type: "object",
|
|
22
|
+
properties: {
|
|
23
|
+
path: { type: "string" },
|
|
24
|
+
encoding: { type: "string", enum: ["utf-8", "base64"] },
|
|
25
|
+
},
|
|
26
|
+
required: ["path"],
|
|
27
|
+
},
|
|
28
|
+
outputSchema: {
|
|
29
|
+
type: "object",
|
|
30
|
+
properties: { content: { type: "string" }, bytes: { type: "number" } },
|
|
31
|
+
},
|
|
32
|
+
execute: "read-file",
|
|
33
|
+
},
|
|
34
|
+
{
|
|
35
|
+
kind: "tool",
|
|
36
|
+
name: "write-file",
|
|
37
|
+
description: "写入工作区内文件内容",
|
|
38
|
+
sideEffect: "write",
|
|
39
|
+
idempotent: false,
|
|
40
|
+
requiresApproval: true,
|
|
41
|
+
timeout: 5000,
|
|
42
|
+
inputSchema: {
|
|
43
|
+
type: "object",
|
|
44
|
+
properties: {
|
|
45
|
+
path: { type: "string" },
|
|
46
|
+
content: { type: "string" },
|
|
47
|
+
encoding: { type: "string", enum: ["utf-8", "base64"] },
|
|
48
|
+
createDirs: { type: "boolean" },
|
|
49
|
+
},
|
|
50
|
+
required: ["path", "content"],
|
|
51
|
+
},
|
|
52
|
+
outputSchema: {
|
|
53
|
+
type: "object",
|
|
54
|
+
properties: { bytesWritten: { type: "number" } },
|
|
55
|
+
},
|
|
56
|
+
execute: "write-file",
|
|
57
|
+
},
|
|
58
|
+
{
|
|
59
|
+
kind: "tool",
|
|
60
|
+
name: "list-dir",
|
|
61
|
+
description: "列出工作区目录内容",
|
|
62
|
+
sideEffect: "readonly",
|
|
63
|
+
idempotent: true,
|
|
64
|
+
requiresApproval: false,
|
|
65
|
+
timeout: 3000,
|
|
66
|
+
inputSchema: {
|
|
67
|
+
type: "object",
|
|
68
|
+
properties: {
|
|
69
|
+
path: { type: "string" },
|
|
70
|
+
recursive: { type: "boolean" },
|
|
71
|
+
maxEntries: { type: "number" },
|
|
72
|
+
pattern: { type: "string" },
|
|
73
|
+
},
|
|
74
|
+
required: ["path"],
|
|
75
|
+
},
|
|
76
|
+
outputSchema: { type: "object", properties: { entries: { type: "array" } } },
|
|
77
|
+
execute: "list-dir",
|
|
78
|
+
},
|
|
79
|
+
{
|
|
80
|
+
kind: "tool",
|
|
81
|
+
name: "search-code",
|
|
82
|
+
description: "在工作区内按模式搜索代码",
|
|
83
|
+
sideEffect: "readonly",
|
|
84
|
+
idempotent: true,
|
|
85
|
+
requiresApproval: false,
|
|
86
|
+
timeout: 10000,
|
|
87
|
+
inputSchema: {
|
|
88
|
+
type: "object",
|
|
89
|
+
properties: {
|
|
90
|
+
pattern: { type: "string" },
|
|
91
|
+
path: { type: "string" },
|
|
92
|
+
fileGlob: { type: "string" },
|
|
93
|
+
maxResults: { type: "number" },
|
|
94
|
+
caseSensitive: { type: "boolean" },
|
|
95
|
+
},
|
|
96
|
+
required: ["pattern"],
|
|
97
|
+
},
|
|
98
|
+
outputSchema: { type: "object", properties: { matches: { type: "array" } } },
|
|
99
|
+
execute: "search-code",
|
|
100
|
+
},
|
|
101
|
+
{
|
|
102
|
+
kind: "tool",
|
|
103
|
+
name: "fetch-url",
|
|
104
|
+
description: "发送 HTTP 请求并返回响应内容",
|
|
105
|
+
sideEffect: "readonly",
|
|
106
|
+
idempotent: false,
|
|
107
|
+
requiresApproval: false,
|
|
108
|
+
timeout: 15000,
|
|
109
|
+
inputSchema: {
|
|
110
|
+
type: "object",
|
|
111
|
+
properties: {
|
|
112
|
+
url: { type: "string" },
|
|
113
|
+
method: { type: "string", enum: ["GET", "POST"] },
|
|
114
|
+
headers: { type: "object" },
|
|
115
|
+
body: { type: "string" },
|
|
116
|
+
timeoutMs: { type: "number" },
|
|
117
|
+
},
|
|
118
|
+
required: ["url"],
|
|
119
|
+
},
|
|
120
|
+
outputSchema: { type: "object", properties: { status: { type: "number" } } },
|
|
121
|
+
execute: "fetch-url",
|
|
122
|
+
},
|
|
123
|
+
{
|
|
124
|
+
kind: "tool",
|
|
125
|
+
name: "run-shell",
|
|
126
|
+
description: "在受控环境中执行 shell 命令",
|
|
127
|
+
sideEffect: "irreversible",
|
|
128
|
+
idempotent: false,
|
|
129
|
+
requiresApproval: true,
|
|
130
|
+
timeout: 30000,
|
|
131
|
+
inputSchema: {
|
|
132
|
+
type: "object",
|
|
133
|
+
properties: {
|
|
134
|
+
command: { type: "string" },
|
|
135
|
+
args: { type: "array", items: { type: "string" } },
|
|
136
|
+
cwd: { type: "string" },
|
|
137
|
+
env: { type: "object" },
|
|
138
|
+
timeoutMs: { type: "number" },
|
|
139
|
+
},
|
|
140
|
+
required: ["command"],
|
|
141
|
+
},
|
|
142
|
+
outputSchema: {
|
|
143
|
+
type: "object",
|
|
144
|
+
properties: {
|
|
145
|
+
stdout: { type: "string" },
|
|
146
|
+
stderr: { type: "string" },
|
|
147
|
+
exitCode: { type: "number" },
|
|
148
|
+
durationMs: { type: "number" },
|
|
149
|
+
},
|
|
150
|
+
},
|
|
151
|
+
execute: "run-shell",
|
|
152
|
+
},
|
|
153
|
+
{
|
|
154
|
+
kind: "tool",
|
|
155
|
+
name: "apply-patch",
|
|
156
|
+
description: "应用 unified diff 补丁并支持失败回滚",
|
|
157
|
+
sideEffect: "write",
|
|
158
|
+
idempotent: false,
|
|
159
|
+
requiresApproval: true,
|
|
160
|
+
timeout: 10000,
|
|
161
|
+
inputSchema: {
|
|
162
|
+
type: "object",
|
|
163
|
+
properties: {
|
|
164
|
+
patch: { type: "string" },
|
|
165
|
+
basePath: { type: "string" },
|
|
166
|
+
},
|
|
167
|
+
required: ["patch"],
|
|
168
|
+
},
|
|
169
|
+
outputSchema: {
|
|
170
|
+
type: "object",
|
|
171
|
+
properties: {
|
|
172
|
+
applied: { type: "array" },
|
|
173
|
+
success: { type: "boolean" },
|
|
174
|
+
},
|
|
175
|
+
},
|
|
176
|
+
execute: "apply-patch",
|
|
177
|
+
},
|
|
178
|
+
];
|
|
179
|
+
/**
|
|
180
|
+
* 工具执行函数注册表。
|
|
181
|
+
*/
|
|
182
|
+
export const toolExecutors = {
|
|
183
|
+
"read-file": readFileExecutor,
|
|
184
|
+
"write-file": writeFileExecutor,
|
|
185
|
+
"list-dir": listDirExecutor,
|
|
186
|
+
"search-code": searchCodeExecutor,
|
|
187
|
+
"fetch-url": fetchUrlExecutor,
|
|
188
|
+
"run-shell": runShellExecutor,
|
|
189
|
+
"apply-patch": applyPatchExecutor,
|
|
190
|
+
};
|
|
191
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAqB;IAC/C;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,aAAa;QAC1B,UAAU,EAAE,UAAU;QACtB,UAAU,EAAE,IAAI;QAChB,gBAAgB,EAAE,KAAK;QACvB,OAAO,EAAE,IAAI;QACb,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,EAAE;aACxD;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;SACvE;QACD,OAAO,EAAE,WAAW;KACrB;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,YAAY;QACzB,UAAU,EAAE,OAAO;QACnB,UAAU,EAAE,KAAK;QACjB,gBAAgB,EAAE,IAAI;QACtB,OAAO,EAAE,IAAI;QACb,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,EAAE;gBACvD,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;aAChC;YACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;SAC9B;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;SACjD;QACD,OAAO,EAAE,YAAY;KACtB;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,UAAU;QAChB,WAAW,EAAE,WAAW;QACxB,UAAU,EAAE,UAAU;QACtB,UAAU,EAAE,IAAI;QAChB,gBAAgB,EAAE,KAAK;QACvB,OAAO,EAAE,IAAI;QACb,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;gBAC9B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC9B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC5B;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;QACD,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE;QAC5E,OAAO,EAAE,UAAU;KACpB;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,cAAc;QAC3B,UAAU,EAAE,UAAU;QACtB,UAAU,EAAE,IAAI;QAChB,gBAAgB,EAAE,KAAK;QACvB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC5B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC9B,aAAa,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;aACnC;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;QACD,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE;QAC5E,OAAO,EAAE,aAAa;KACvB;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,mBAAmB;QAChC,UAAU,EAAE,UAAU;QACtB,UAAU,EAAE,KAAK;QACjB,gBAAgB,EAAE,KAAK;QACvB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvB,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE;gBACjD,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC9B;YACD,QAAQ,EAAE,CAAC,KAAK,CAAC;SAClB;QACD,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC5E,OAAO,EAAE,WAAW;KACrB;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,mBAAmB;QAChC,UAAU,EAAE,cAAc;QAC1B,UAAU,EAAE,KAAK;QACjB,gBAAgB,EAAE,IAAI;QACtB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBAClD,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvB,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC9B;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC1B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC1B,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC5B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC/B;SACF;QACD,OAAO,EAAE,WAAW;KACrB;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,2BAA2B;QACxC,UAAU,EAAE,OAAO;QACnB,UAAU,EAAE,KAAK;QACjB,gBAAgB,EAAE,IAAI;QACtB,OAAO,EAAE,KAAK;QACd,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC7B;YACD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;gBAC1B,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;aAC7B;SACF;QACD,OAAO,EAAE,aAAa;KACvB;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAiC;IACzD,WAAW,EAAE,gBAAgC;IAC7C,YAAY,EAAE,iBAAiC;IAC/C,UAAU,EAAE,eAA+B;IAC3C,aAAa,EAAE,kBAAkC;IACjD,WAAW,EAAE,gBAAgC;IAC7C,WAAW,EAAE,gBAAgC;IAC7C,aAAa,EAAE,kBAAkC;CAClD,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
---
|
|
2
|
+
kind: tool
|
|
3
|
+
name: list-dir
|
|
4
|
+
description: 列出工作区目录内容
|
|
5
|
+
sideEffect: readonly
|
|
6
|
+
idempotent: true
|
|
7
|
+
requiresApproval: false
|
|
8
|
+
timeout: 3000
|
|
9
|
+
inputSchema:
|
|
10
|
+
type: object
|
|
11
|
+
properties:
|
|
12
|
+
path:
|
|
13
|
+
type: string
|
|
14
|
+
recursive:
|
|
15
|
+
type: boolean
|
|
16
|
+
maxEntries:
|
|
17
|
+
type: number
|
|
18
|
+
pattern:
|
|
19
|
+
type: string
|
|
20
|
+
required: [path]
|
|
21
|
+
outputSchema:
|
|
22
|
+
type: object
|
|
23
|
+
properties:
|
|
24
|
+
entries:
|
|
25
|
+
type: array
|
|
26
|
+
execute: list-dir
|
|
27
|
+
---
|
|
28
|
+
|
|
29
|
+
列出目录下文件和子目录,支持递归和结果数量限制。
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { ToolExecutor } from "../shared";
|
|
2
|
+
interface ListDirInput {
|
|
3
|
+
path: string;
|
|
4
|
+
recursive?: boolean;
|
|
5
|
+
maxEntries?: number;
|
|
6
|
+
pattern?: string;
|
|
7
|
+
}
|
|
8
|
+
interface ListDirEntry {
|
|
9
|
+
name: string;
|
|
10
|
+
type: "file" | "directory";
|
|
11
|
+
size?: number;
|
|
12
|
+
}
|
|
13
|
+
interface ListDirOutput {
|
|
14
|
+
entries: ListDirEntry[];
|
|
15
|
+
truncated: boolean;
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* 列目录 Tool 执行器。
|
|
19
|
+
*/
|
|
20
|
+
export declare const listDirExecutor: ToolExecutor<ListDirInput, ListDirOutput>;
|
|
21
|
+
export {};
|
|
22
|
+
//# sourceMappingURL=executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../../src/tools/list-dir/executor.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAG9C,UAAU,YAAY;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,YAAY;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,aAAa;IACrB,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED;;GAEG;AACH,eAAO,MAAM,eAAe,EAAE,YAAY,CAAC,YAAY,EAAE,aAAa,CAyCrE,CAAC"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import { readdir, stat } from "node:fs/promises";
|
|
2
|
+
import { join } from "node:path";
|
|
3
|
+
import { resolveWorkspacePath, toWorkspaceRelativePath } from "../../common/path";
|
|
4
|
+
import { assertNotAborted } from "../shared";
|
|
5
|
+
/**
|
|
6
|
+
* 列目录 Tool 执行器。
|
|
7
|
+
*/
|
|
8
|
+
export const listDirExecutor = async (input, context) => {
|
|
9
|
+
const recursive = input.recursive ?? false;
|
|
10
|
+
const maxEntries = input.maxEntries ?? 1000;
|
|
11
|
+
const matcher = input.pattern ? new RegExp(input.pattern) : undefined;
|
|
12
|
+
const root = resolveWorkspacePath(context.workspaceRoot, input.path);
|
|
13
|
+
const entries = [];
|
|
14
|
+
let truncated = false;
|
|
15
|
+
const walk = async (dir) => {
|
|
16
|
+
assertNotAborted(context.abortSignal);
|
|
17
|
+
const children = await readdir(dir, { withFileTypes: true });
|
|
18
|
+
for (const child of children) {
|
|
19
|
+
if (entries.length >= maxEntries) {
|
|
20
|
+
truncated = true;
|
|
21
|
+
return;
|
|
22
|
+
}
|
|
23
|
+
const absolute = join(dir, child.name);
|
|
24
|
+
const relative = toWorkspaceRelativePath(context.workspaceRoot, absolute);
|
|
25
|
+
if (matcher && !matcher.test(relative)) {
|
|
26
|
+
if (recursive && child.isDirectory()) {
|
|
27
|
+
await walk(absolute);
|
|
28
|
+
}
|
|
29
|
+
continue;
|
|
30
|
+
}
|
|
31
|
+
if (child.isDirectory()) {
|
|
32
|
+
entries.push({ name: relative, type: "directory" });
|
|
33
|
+
if (recursive) {
|
|
34
|
+
await walk(absolute);
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
else if (child.isFile()) {
|
|
38
|
+
const fileStat = await stat(absolute);
|
|
39
|
+
entries.push({ name: relative, type: "file", size: fileStat.size });
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
};
|
|
43
|
+
await walk(root);
|
|
44
|
+
return { entries, truncated };
|
|
45
|
+
};
|
|
46
|
+
//# sourceMappingURL=executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../src/tools/list-dir/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAElF,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAoB7C;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAA8C,KAAK,EAC7E,KAAK,EACL,OAAO,EACP,EAAE;IACF,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC;IAC3C,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,IAAI,CAAC;IAC5C,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtE,MAAM,IAAI,GAAG,oBAAoB,CAAC,OAAO,CAAC,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACrE,MAAM,OAAO,GAAmB,EAAE,CAAC;IACnC,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,MAAM,IAAI,GAAG,KAAK,EAAE,GAAW,EAAiB,EAAE;QAChD,gBAAgB,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,IAAI,OAAO,CAAC,MAAM,IAAI,UAAU,EAAE,CAAC;gBACjC,SAAS,GAAG,IAAI,CAAC;gBACjB,OAAO;YACT,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,QAAQ,GAAG,uBAAuB,CAAC,OAAO,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YAC1E,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvC,IAAI,SAAS,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;oBACrC,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvB,CAAC;gBACD,SAAS;YACX,CAAC;YACD,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;gBACpD,IAAI,SAAS,EAAE,CAAC;oBACd,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACtC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC;IACjB,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AAChC,CAAC,CAAC"}
|