@tinfoilsh/tinfoil-terminal 0.1.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/AGENTS.md +27 -0
- package/Dockerfile +18 -0
- package/README.md +15 -0
- package/bin/opencode +88 -0
- package/bunfig.toml +7 -0
- package/package.json +134 -0
- package/parsers-config.ts +253 -0
- package/script/build.ts +176 -0
- package/script/postinstall.mjs +122 -0
- package/script/publish-registries.ts +187 -0
- package/script/publish.ts +70 -0
- package/script/schema.ts +47 -0
- package/src/acp/README.md +164 -0
- package/src/acp/agent.ts +1124 -0
- package/src/acp/session.ts +101 -0
- package/src/acp/types.ts +22 -0
- package/src/agent/agent.ts +284 -0
- package/src/agent/generate.txt +75 -0
- package/src/agent/prompt/compaction.txt +12 -0
- package/src/agent/prompt/explore.txt +18 -0
- package/src/agent/prompt/summary.txt +11 -0
- package/src/agent/prompt/title.txt +43 -0
- package/src/auth/index.ts +73 -0
- package/src/bun/index.ts +130 -0
- package/src/bus/bus-event.ts +43 -0
- package/src/bus/global.ts +10 -0
- package/src/bus/index.ts +105 -0
- package/src/cli/bootstrap.ts +17 -0
- package/src/cli/cmd/acp.ts +69 -0
- package/src/cli/cmd/agent.ts +257 -0
- package/src/cli/cmd/auth.ts +400 -0
- package/src/cli/cmd/cmd.ts +7 -0
- package/src/cli/cmd/debug/agent.ts +29 -0
- package/src/cli/cmd/debug/config.ts +16 -0
- package/src/cli/cmd/debug/file.ts +97 -0
- package/src/cli/cmd/debug/index.ts +48 -0
- package/src/cli/cmd/debug/lsp.ts +52 -0
- package/src/cli/cmd/debug/ripgrep.ts +87 -0
- package/src/cli/cmd/debug/scrap.ts +16 -0
- package/src/cli/cmd/debug/skill.ts +16 -0
- package/src/cli/cmd/debug/snapshot.ts +52 -0
- package/src/cli/cmd/export.ts +88 -0
- package/src/cli/cmd/generate.ts +38 -0
- package/src/cli/cmd/github.ts +1546 -0
- package/src/cli/cmd/import.ts +98 -0
- package/src/cli/cmd/mcp.ts +671 -0
- package/src/cli/cmd/models.ts +77 -0
- package/src/cli/cmd/pr.ts +112 -0
- package/src/cli/cmd/run.ts +395 -0
- package/src/cli/cmd/serve.ts +16 -0
- package/src/cli/cmd/session.ts +135 -0
- package/src/cli/cmd/stats.ts +402 -0
- package/src/cli/cmd/tui/app.tsx +743 -0
- package/src/cli/cmd/tui/attach.ts +31 -0
- package/src/cli/cmd/tui/component/border.tsx +21 -0
- package/src/cli/cmd/tui/component/dialog-agent.tsx +31 -0
- package/src/cli/cmd/tui/component/dialog-command.tsx +124 -0
- package/src/cli/cmd/tui/component/dialog-mcp.tsx +86 -0
- package/src/cli/cmd/tui/component/dialog-model.tsx +168 -0
- package/src/cli/cmd/tui/component/dialog-provider.tsx +47 -0
- package/src/cli/cmd/tui/component/dialog-session-list.tsx +115 -0
- package/src/cli/cmd/tui/component/dialog-session-rename.tsx +31 -0
- package/src/cli/cmd/tui/component/dialog-stash.tsx +86 -0
- package/src/cli/cmd/tui/component/dialog-status.tsx +162 -0
- package/src/cli/cmd/tui/component/dialog-tag.tsx +44 -0
- package/src/cli/cmd/tui/component/dialog-theme-list.tsx +50 -0
- package/src/cli/cmd/tui/component/did-you-know.tsx +85 -0
- package/src/cli/cmd/tui/component/logo.tsx +33 -0
- package/src/cli/cmd/tui/component/prompt/autocomplete.tsx +718 -0
- package/src/cli/cmd/tui/component/prompt/frecency.tsx +89 -0
- package/src/cli/cmd/tui/component/prompt/history.tsx +108 -0
- package/src/cli/cmd/tui/component/prompt/index.tsx +1083 -0
- package/src/cli/cmd/tui/component/prompt/stash.tsx +101 -0
- package/src/cli/cmd/tui/component/textarea-keybindings.ts +73 -0
- package/src/cli/cmd/tui/component/tips.ts +103 -0
- package/src/cli/cmd/tui/component/todo-item.tsx +32 -0
- package/src/cli/cmd/tui/context/args.tsx +14 -0
- package/src/cli/cmd/tui/context/directory.ts +13 -0
- package/src/cli/cmd/tui/context/exit.tsx +23 -0
- package/src/cli/cmd/tui/context/helper.tsx +25 -0
- package/src/cli/cmd/tui/context/keybind.tsx +101 -0
- package/src/cli/cmd/tui/context/kv.tsx +49 -0
- package/src/cli/cmd/tui/context/local.tsx +393 -0
- package/src/cli/cmd/tui/context/prompt.tsx +18 -0
- package/src/cli/cmd/tui/context/route.tsx +46 -0
- package/src/cli/cmd/tui/context/sdk.tsx +94 -0
- package/src/cli/cmd/tui/context/sync.tsx +427 -0
- package/src/cli/cmd/tui/context/theme/aura.json +69 -0
- package/src/cli/cmd/tui/context/theme/ayu.json +80 -0
- package/src/cli/cmd/tui/context/theme/catppuccin-frappe.json +233 -0
- package/src/cli/cmd/tui/context/theme/catppuccin-macchiato.json +233 -0
- package/src/cli/cmd/tui/context/theme/catppuccin.json +112 -0
- package/src/cli/cmd/tui/context/theme/cobalt2.json +228 -0
- package/src/cli/cmd/tui/context/theme/cursor.json +249 -0
- package/src/cli/cmd/tui/context/theme/dracula.json +219 -0
- package/src/cli/cmd/tui/context/theme/everforest.json +241 -0
- package/src/cli/cmd/tui/context/theme/flexoki.json +237 -0
- package/src/cli/cmd/tui/context/theme/github.json +233 -0
- package/src/cli/cmd/tui/context/theme/gruvbox.json +95 -0
- package/src/cli/cmd/tui/context/theme/kanagawa.json +77 -0
- package/src/cli/cmd/tui/context/theme/lucent-orng.json +237 -0
- package/src/cli/cmd/tui/context/theme/material.json +235 -0
- package/src/cli/cmd/tui/context/theme/matrix.json +77 -0
- package/src/cli/cmd/tui/context/theme/mercury.json +252 -0
- package/src/cli/cmd/tui/context/theme/monokai.json +221 -0
- package/src/cli/cmd/tui/context/theme/nightowl.json +221 -0
- package/src/cli/cmd/tui/context/theme/nord.json +223 -0
- package/src/cli/cmd/tui/context/theme/one-dark.json +84 -0
- package/src/cli/cmd/tui/context/theme/orng.json +249 -0
- package/src/cli/cmd/tui/context/theme/osaka-jade.json +93 -0
- package/src/cli/cmd/tui/context/theme/palenight.json +222 -0
- package/src/cli/cmd/tui/context/theme/rosepine.json +234 -0
- package/src/cli/cmd/tui/context/theme/solarized.json +223 -0
- package/src/cli/cmd/tui/context/theme/synthwave84.json +226 -0
- package/src/cli/cmd/tui/context/theme/tokyonight.json +243 -0
- package/src/cli/cmd/tui/context/theme/vercel.json +245 -0
- package/src/cli/cmd/tui/context/theme/vesper.json +218 -0
- package/src/cli/cmd/tui/context/theme/zenburn.json +223 -0
- package/src/cli/cmd/tui/context/theme.tsx +1150 -0
- package/src/cli/cmd/tui/event.ts +46 -0
- package/src/cli/cmd/tui/routes/home.tsx +138 -0
- package/src/cli/cmd/tui/routes/session/dialog-fork-from-timeline.tsx +64 -0
- package/src/cli/cmd/tui/routes/session/dialog-message.tsx +109 -0
- package/src/cli/cmd/tui/routes/session/dialog-subagent.tsx +26 -0
- package/src/cli/cmd/tui/routes/session/dialog-timeline.tsx +47 -0
- package/src/cli/cmd/tui/routes/session/footer.tsx +88 -0
- package/src/cli/cmd/tui/routes/session/header.tsx +105 -0
- package/src/cli/cmd/tui/routes/session/index.tsx +1897 -0
- package/src/cli/cmd/tui/routes/session/permission.tsx +416 -0
- package/src/cli/cmd/tui/routes/session/question.tsx +368 -0
- package/src/cli/cmd/tui/routes/session/sidebar.tsx +279 -0
- package/src/cli/cmd/tui/spawn.ts +46 -0
- package/src/cli/cmd/tui/thread.ts +174 -0
- package/src/cli/cmd/tui/ui/dialog-alert.tsx +57 -0
- package/src/cli/cmd/tui/ui/dialog-confirm.tsx +83 -0
- package/src/cli/cmd/tui/ui/dialog-export-options.tsx +204 -0
- package/src/cli/cmd/tui/ui/dialog-help.tsx +38 -0
- package/src/cli/cmd/tui/ui/dialog-prompt.tsx +77 -0
- package/src/cli/cmd/tui/ui/dialog-select.tsx +345 -0
- package/src/cli/cmd/tui/ui/dialog.tsx +172 -0
- package/src/cli/cmd/tui/ui/link.tsx +28 -0
- package/src/cli/cmd/tui/ui/spinner.ts +368 -0
- package/src/cli/cmd/tui/ui/toast.tsx +100 -0
- package/src/cli/cmd/tui/util/clipboard.ts +128 -0
- package/src/cli/cmd/tui/util/editor.ts +32 -0
- package/src/cli/cmd/tui/util/signal.ts +7 -0
- package/src/cli/cmd/tui/util/terminal.ts +114 -0
- package/src/cli/cmd/tui/util/transcript.ts +98 -0
- package/src/cli/cmd/tui/worker.ts +95 -0
- package/src/cli/cmd/uninstall.ts +344 -0
- package/src/cli/cmd/upgrade.ts +67 -0
- package/src/cli/cmd/web.ts +73 -0
- package/src/cli/error.ts +57 -0
- package/src/cli/network.ts +53 -0
- package/src/cli/ui.ts +87 -0
- package/src/cli/upgrade.ts +25 -0
- package/src/command/index.ts +131 -0
- package/src/command/template/initialize.txt +10 -0
- package/src/command/template/review.txt +97 -0
- package/src/config/config.ts +1226 -0
- package/src/config/markdown.ts +41 -0
- package/src/env/index.ts +26 -0
- package/src/file/ignore.ts +83 -0
- package/src/file/index.ts +411 -0
- package/src/file/ripgrep.ts +409 -0
- package/src/file/time.ts +64 -0
- package/src/file/watcher.ts +118 -0
- package/src/flag/flag.ts +51 -0
- package/src/format/formatter.ts +359 -0
- package/src/format/index.ts +137 -0
- package/src/global/index.ts +55 -0
- package/src/id/id.ts +83 -0
- package/src/ide/index.ts +76 -0
- package/src/index.ts +161 -0
- package/src/installation/index.ts +205 -0
- package/src/lsp/client.ts +252 -0
- package/src/lsp/index.ts +485 -0
- package/src/lsp/language.ts +119 -0
- package/src/lsp/server.ts +2032 -0
- package/src/mcp/auth.ts +135 -0
- package/src/mcp/index.ts +874 -0
- package/src/mcp/oauth-callback.ts +200 -0
- package/src/mcp/oauth-provider.ts +154 -0
- package/src/patch/index.ts +622 -0
- package/src/permission/arity.ts +163 -0
- package/src/permission/index.ts +210 -0
- package/src/permission/next.ts +269 -0
- package/src/plugin/codex.ts +524 -0
- package/src/plugin/index.ts +118 -0
- package/src/project/bootstrap.ts +31 -0
- package/src/project/instance.ts +78 -0
- package/src/project/project.ts +316 -0
- package/src/project/state.ts +65 -0
- package/src/project/vcs.ts +76 -0
- package/src/provider/auth.ts +147 -0
- package/src/provider/models-macro.ts +11 -0
- package/src/provider/models.ts +86 -0
- package/src/provider/provider.ts +429 -0
- package/src/provider/sdk/openai-compatible/src/README.md +5 -0
- package/src/provider/sdk/openai-compatible/src/index.ts +2 -0
- package/src/provider/sdk/openai-compatible/src/openai-compatible-provider.ts +100 -0
- package/src/provider/sdk/openai-compatible/src/responses/convert-to-openai-responses-input.ts +303 -0
- package/src/provider/sdk/openai-compatible/src/responses/map-openai-responses-finish-reason.ts +22 -0
- package/src/provider/sdk/openai-compatible/src/responses/openai-config.ts +18 -0
- package/src/provider/sdk/openai-compatible/src/responses/openai-error.ts +22 -0
- package/src/provider/sdk/openai-compatible/src/responses/openai-responses-api-types.ts +207 -0
- package/src/provider/sdk/openai-compatible/src/responses/openai-responses-language-model.ts +1713 -0
- package/src/provider/sdk/openai-compatible/src/responses/openai-responses-prepare-tools.ts +177 -0
- package/src/provider/sdk/openai-compatible/src/responses/openai-responses-settings.ts +1 -0
- package/src/provider/sdk/openai-compatible/src/responses/tool/code-interpreter.ts +88 -0
- package/src/provider/sdk/openai-compatible/src/responses/tool/file-search.ts +128 -0
- package/src/provider/sdk/openai-compatible/src/responses/tool/image-generation.ts +115 -0
- package/src/provider/sdk/openai-compatible/src/responses/tool/local-shell.ts +65 -0
- package/src/provider/sdk/openai-compatible/src/responses/tool/web-search-preview.ts +104 -0
- package/src/provider/sdk/openai-compatible/src/responses/tool/web-search.ts +103 -0
- package/src/provider/transform.ts +654 -0
- package/src/pty/index.ts +229 -0
- package/src/question/index.ts +170 -0
- package/src/server/error.ts +36 -0
- package/src/server/mdns.ts +57 -0
- package/src/server/project.ts +79 -0
- package/src/server/question.ts +95 -0
- package/src/server/server.ts +2878 -0
- package/src/server/tui.ts +71 -0
- package/src/session/compaction.ts +225 -0
- package/src/session/index.ts +476 -0
- package/src/session/llm.ts +235 -0
- package/src/session/message-v2.ts +683 -0
- package/src/session/message.ts +189 -0
- package/src/session/processor.ts +406 -0
- package/src/session/prompt/anthropic-20250930.txt +166 -0
- package/src/session/prompt/anthropic.txt +105 -0
- package/src/session/prompt/anthropic_spoof.txt +1 -0
- package/src/session/prompt/beast.txt +147 -0
- package/src/session/prompt/build-switch.txt +5 -0
- package/src/session/prompt/codex.txt +318 -0
- package/src/session/prompt/codex_header.txt +318 -0
- package/src/session/prompt/copilot-gpt-5.txt +143 -0
- package/src/session/prompt/gemini.txt +155 -0
- package/src/session/prompt/max-steps.txt +16 -0
- package/src/session/prompt/plan-reminder-anthropic.txt +67 -0
- package/src/session/prompt/plan.txt +26 -0
- package/src/session/prompt/qwen.txt +109 -0
- package/src/session/prompt.ts +1652 -0
- package/src/session/retry.ts +90 -0
- package/src/session/revert.ts +108 -0
- package/src/session/status.ts +76 -0
- package/src/session/summary.ts +194 -0
- package/src/session/system.ts +138 -0
- package/src/session/todo.ts +37 -0
- package/src/share/share-next.ts +194 -0
- package/src/share/share.ts +87 -0
- package/src/shell/shell.ts +67 -0
- package/src/skill/index.ts +1 -0
- package/src/skill/skill.ts +127 -0
- package/src/snapshot/index.ts +199 -0
- package/src/storage/storage.ts +226 -0
- package/src/tool/bash.ts +258 -0
- package/src/tool/bash.txt +115 -0
- package/src/tool/batch.ts +175 -0
- package/src/tool/batch.txt +24 -0
- package/src/tool/codesearch.ts +132 -0
- package/src/tool/codesearch.txt +12 -0
- package/src/tool/edit.ts +645 -0
- package/src/tool/edit.txt +10 -0
- package/src/tool/external-directory.ts +33 -0
- package/src/tool/glob.ts +77 -0
- package/src/tool/glob.txt +6 -0
- package/src/tool/grep.ts +136 -0
- package/src/tool/grep.txt +8 -0
- package/src/tool/invalid.ts +17 -0
- package/src/tool/ls.ts +121 -0
- package/src/tool/ls.txt +1 -0
- package/src/tool/lsp.ts +96 -0
- package/src/tool/lsp.txt +19 -0
- package/src/tool/multiedit.ts +46 -0
- package/src/tool/multiedit.txt +41 -0
- package/src/tool/patch.ts +201 -0
- package/src/tool/patch.txt +1 -0
- package/src/tool/question.ts +33 -0
- package/src/tool/question.txt +10 -0
- package/src/tool/read.ts +200 -0
- package/src/tool/read.txt +12 -0
- package/src/tool/registry.ts +141 -0
- package/src/tool/skill.ts +75 -0
- package/src/tool/task.ts +181 -0
- package/src/tool/task.txt +60 -0
- package/src/tool/todo.ts +53 -0
- package/src/tool/todoread.txt +14 -0
- package/src/tool/todowrite.txt +167 -0
- package/src/tool/tool.ts +88 -0
- package/src/tool/truncation.ts +98 -0
- package/src/tool/webfetch.ts +182 -0
- package/src/tool/webfetch.txt +13 -0
- package/src/tool/websearch.ts +144 -0
- package/src/tool/websearch.txt +11 -0
- package/src/tool/write.ts +80 -0
- package/src/tool/write.txt +8 -0
- package/src/util/archive.ts +16 -0
- package/src/util/color.ts +19 -0
- package/src/util/context.ts +25 -0
- package/src/util/defer.ts +12 -0
- package/src/util/eventloop.ts +20 -0
- package/src/util/filesystem.ts +83 -0
- package/src/util/fn.ts +11 -0
- package/src/util/iife.ts +3 -0
- package/src/util/keybind.ts +102 -0
- package/src/util/lazy.ts +18 -0
- package/src/util/locale.ts +81 -0
- package/src/util/lock.ts +98 -0
- package/src/util/log.ts +180 -0
- package/src/util/queue.ts +32 -0
- package/src/util/rpc.ts +66 -0
- package/src/util/scrap.ts +10 -0
- package/src/util/signal.ts +12 -0
- package/src/util/timeout.ts +14 -0
- package/src/util/token.ts +7 -0
- package/src/util/wildcard.ts +54 -0
- package/src/worktree/index.ts +217 -0
- package/sst-env.d.ts +9 -0
- package/test/agent/agent.test.ts +511 -0
- package/test/bun.test.ts +53 -0
- package/test/cli/github-action.test.ts +129 -0
- package/test/cli/github-remote.test.ts +80 -0
- package/test/cli/tui/transcript.test.ts +297 -0
- package/test/config/agent-color.test.ts +66 -0
- package/test/config/config.test.ts +1235 -0
- package/test/config/markdown.test.ts +89 -0
- package/test/file/ignore.test.ts +10 -0
- package/test/file/path-traversal.test.ts +115 -0
- package/test/fixture/fixture.ts +45 -0
- package/test/fixture/lsp/fake-lsp-server.js +77 -0
- package/test/ide/ide.test.ts +82 -0
- package/test/keybind.test.ts +421 -0
- package/test/lsp/client.test.ts +95 -0
- package/test/mcp/headers.test.ts +153 -0
- package/test/patch/patch.test.ts +348 -0
- package/test/permission/arity.test.ts +33 -0
- package/test/permission/next.test.ts +652 -0
- package/test/permission-task.test.ts +319 -0
- package/test/plugin/codex.test.ts +123 -0
- package/test/preload.ts +65 -0
- package/test/project/project.test.ts +120 -0
- package/test/provider/amazon-bedrock.test.ts +205 -0
- package/test/provider/provider.test.ts +2127 -0
- package/test/provider/transform.test.ts +1155 -0
- package/test/question/question.test.ts +300 -0
- package/test/server/session-select.test.ts +78 -0
- package/test/session/compaction.test.ts +251 -0
- package/test/session/message-v2.test.ts +570 -0
- package/test/session/retry.test.ts +131 -0
- package/test/session/revert-compact.test.ts +285 -0
- package/test/session/session.test.ts +71 -0
- package/test/skill/skill.test.ts +185 -0
- package/test/snapshot/snapshot.test.ts +939 -0
- package/test/tool/__snapshots__/tool.test.ts.snap +9 -0
- package/test/tool/bash.test.ts +320 -0
- package/test/tool/external-directory.test.ts +126 -0
- package/test/tool/fixtures/large-image.png +0 -0
- package/test/tool/fixtures/models-api.json +33453 -0
- package/test/tool/grep.test.ts +109 -0
- package/test/tool/patch.test.ts +261 -0
- package/test/tool/read.test.ts +303 -0
- package/test/tool/truncation.test.ts +159 -0
- package/test/util/iife.test.ts +36 -0
- package/test/util/lazy.test.ts +50 -0
- package/test/util/timeout.test.ts +21 -0
- package/test/util/wildcard.test.ts +55 -0
- package/tsconfig.json +16 -0
|
@@ -0,0 +1,109 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test"
|
|
2
|
+
import path from "path"
|
|
3
|
+
import { GrepTool } from "../../src/tool/grep"
|
|
4
|
+
import { Instance } from "../../src/project/instance"
|
|
5
|
+
import { tmpdir } from "../fixture/fixture"
|
|
6
|
+
|
|
7
|
+
const ctx = {
|
|
8
|
+
sessionID: "test",
|
|
9
|
+
messageID: "",
|
|
10
|
+
callID: "",
|
|
11
|
+
agent: "build",
|
|
12
|
+
abort: AbortSignal.any([]),
|
|
13
|
+
metadata: () => {},
|
|
14
|
+
ask: async () => {},
|
|
15
|
+
}
|
|
16
|
+
|
|
17
|
+
const projectRoot = path.join(__dirname, "../..")
|
|
18
|
+
|
|
19
|
+
describe("tool.grep", () => {
|
|
20
|
+
test("basic search", async () => {
|
|
21
|
+
await Instance.provide({
|
|
22
|
+
directory: projectRoot,
|
|
23
|
+
fn: async () => {
|
|
24
|
+
const grep = await GrepTool.init()
|
|
25
|
+
const result = await grep.execute(
|
|
26
|
+
{
|
|
27
|
+
pattern: "export",
|
|
28
|
+
path: path.join(projectRoot, "src/tool"),
|
|
29
|
+
include: "*.ts",
|
|
30
|
+
},
|
|
31
|
+
ctx,
|
|
32
|
+
)
|
|
33
|
+
expect(result.metadata.matches).toBeGreaterThan(0)
|
|
34
|
+
expect(result.output).toContain("Found")
|
|
35
|
+
},
|
|
36
|
+
})
|
|
37
|
+
})
|
|
38
|
+
|
|
39
|
+
test("no matches returns correct output", async () => {
|
|
40
|
+
await using tmp = await tmpdir({
|
|
41
|
+
init: async (dir) => {
|
|
42
|
+
await Bun.write(path.join(dir, "test.txt"), "hello world")
|
|
43
|
+
},
|
|
44
|
+
})
|
|
45
|
+
await Instance.provide({
|
|
46
|
+
directory: tmp.path,
|
|
47
|
+
fn: async () => {
|
|
48
|
+
const grep = await GrepTool.init()
|
|
49
|
+
const result = await grep.execute(
|
|
50
|
+
{
|
|
51
|
+
pattern: "xyznonexistentpatternxyz123",
|
|
52
|
+
path: tmp.path,
|
|
53
|
+
},
|
|
54
|
+
ctx,
|
|
55
|
+
)
|
|
56
|
+
expect(result.metadata.matches).toBe(0)
|
|
57
|
+
expect(result.output).toBe("No files found")
|
|
58
|
+
},
|
|
59
|
+
})
|
|
60
|
+
})
|
|
61
|
+
|
|
62
|
+
test("handles CRLF line endings in output", async () => {
|
|
63
|
+
// This test verifies the regex split handles both \n and \r\n
|
|
64
|
+
await using tmp = await tmpdir({
|
|
65
|
+
init: async (dir) => {
|
|
66
|
+
// Create a test file with content
|
|
67
|
+
await Bun.write(path.join(dir, "test.txt"), "line1\nline2\nline3")
|
|
68
|
+
},
|
|
69
|
+
})
|
|
70
|
+
await Instance.provide({
|
|
71
|
+
directory: tmp.path,
|
|
72
|
+
fn: async () => {
|
|
73
|
+
const grep = await GrepTool.init()
|
|
74
|
+
const result = await grep.execute(
|
|
75
|
+
{
|
|
76
|
+
pattern: "line",
|
|
77
|
+
path: tmp.path,
|
|
78
|
+
},
|
|
79
|
+
ctx,
|
|
80
|
+
)
|
|
81
|
+
expect(result.metadata.matches).toBeGreaterThan(0)
|
|
82
|
+
},
|
|
83
|
+
})
|
|
84
|
+
})
|
|
85
|
+
})
|
|
86
|
+
|
|
87
|
+
describe("CRLF regex handling", () => {
|
|
88
|
+
test("regex correctly splits Unix line endings", () => {
|
|
89
|
+
const unixOutput = "file1.txt|1|content1\nfile2.txt|2|content2\nfile3.txt|3|content3"
|
|
90
|
+
const lines = unixOutput.trim().split(/\r?\n/)
|
|
91
|
+
expect(lines.length).toBe(3)
|
|
92
|
+
expect(lines[0]).toBe("file1.txt|1|content1")
|
|
93
|
+
expect(lines[2]).toBe("file3.txt|3|content3")
|
|
94
|
+
})
|
|
95
|
+
|
|
96
|
+
test("regex correctly splits Windows CRLF line endings", () => {
|
|
97
|
+
const windowsOutput = "file1.txt|1|content1\r\nfile2.txt|2|content2\r\nfile3.txt|3|content3"
|
|
98
|
+
const lines = windowsOutput.trim().split(/\r?\n/)
|
|
99
|
+
expect(lines.length).toBe(3)
|
|
100
|
+
expect(lines[0]).toBe("file1.txt|1|content1")
|
|
101
|
+
expect(lines[2]).toBe("file3.txt|3|content3")
|
|
102
|
+
})
|
|
103
|
+
|
|
104
|
+
test("regex handles mixed line endings", () => {
|
|
105
|
+
const mixedOutput = "file1.txt|1|content1\nfile2.txt|2|content2\r\nfile3.txt|3|content3"
|
|
106
|
+
const lines = mixedOutput.trim().split(/\r?\n/)
|
|
107
|
+
expect(lines.length).toBe(3)
|
|
108
|
+
})
|
|
109
|
+
})
|
|
@@ -0,0 +1,261 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test"
|
|
2
|
+
import path from "path"
|
|
3
|
+
import { PatchTool } from "../../src/tool/patch"
|
|
4
|
+
import { Instance } from "../../src/project/instance"
|
|
5
|
+
import { tmpdir } from "../fixture/fixture"
|
|
6
|
+
import { PermissionNext } from "../../src/permission/next"
|
|
7
|
+
import * as fs from "fs/promises"
|
|
8
|
+
|
|
9
|
+
const ctx = {
|
|
10
|
+
sessionID: "test",
|
|
11
|
+
messageID: "",
|
|
12
|
+
callID: "",
|
|
13
|
+
agent: "build",
|
|
14
|
+
abort: AbortSignal.any([]),
|
|
15
|
+
metadata: () => {},
|
|
16
|
+
ask: async () => {},
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
const patchTool = await PatchTool.init()
|
|
20
|
+
|
|
21
|
+
describe("tool.patch", () => {
|
|
22
|
+
test("should validate required parameters", async () => {
|
|
23
|
+
await Instance.provide({
|
|
24
|
+
directory: "/tmp",
|
|
25
|
+
fn: async () => {
|
|
26
|
+
expect(patchTool.execute({ patchText: "" }, ctx)).rejects.toThrow("patchText is required")
|
|
27
|
+
},
|
|
28
|
+
})
|
|
29
|
+
})
|
|
30
|
+
|
|
31
|
+
test("should validate patch format", async () => {
|
|
32
|
+
await Instance.provide({
|
|
33
|
+
directory: "/tmp",
|
|
34
|
+
fn: async () => {
|
|
35
|
+
expect(patchTool.execute({ patchText: "invalid patch" }, ctx)).rejects.toThrow("Failed to parse patch")
|
|
36
|
+
},
|
|
37
|
+
})
|
|
38
|
+
})
|
|
39
|
+
|
|
40
|
+
test("should handle empty patch", async () => {
|
|
41
|
+
await Instance.provide({
|
|
42
|
+
directory: "/tmp",
|
|
43
|
+
fn: async () => {
|
|
44
|
+
const emptyPatch = `*** Begin Patch
|
|
45
|
+
*** End Patch`
|
|
46
|
+
|
|
47
|
+
expect(patchTool.execute({ patchText: emptyPatch }, ctx)).rejects.toThrow("No file changes found in patch")
|
|
48
|
+
},
|
|
49
|
+
})
|
|
50
|
+
})
|
|
51
|
+
|
|
52
|
+
test.skip("should ask permission for files outside working directory", async () => {
|
|
53
|
+
await Instance.provide({
|
|
54
|
+
directory: "/tmp",
|
|
55
|
+
fn: async () => {
|
|
56
|
+
const maliciousPatch = `*** Begin Patch
|
|
57
|
+
*** Add File: /etc/passwd
|
|
58
|
+
+malicious content
|
|
59
|
+
*** End Patch`
|
|
60
|
+
patchTool.execute({ patchText: maliciousPatch }, ctx)
|
|
61
|
+
// TODO: this sucks
|
|
62
|
+
await new Promise((resolve) => setTimeout(resolve, 1000))
|
|
63
|
+
const pending = await PermissionNext.list()
|
|
64
|
+
expect(pending.find((p) => p.sessionID === ctx.sessionID)).toBeDefined()
|
|
65
|
+
},
|
|
66
|
+
})
|
|
67
|
+
})
|
|
68
|
+
|
|
69
|
+
test("should handle simple add file operation", async () => {
|
|
70
|
+
await using fixture = await tmpdir()
|
|
71
|
+
|
|
72
|
+
await Instance.provide({
|
|
73
|
+
directory: fixture.path,
|
|
74
|
+
fn: async () => {
|
|
75
|
+
const patchText = `*** Begin Patch
|
|
76
|
+
*** Add File: test-file.txt
|
|
77
|
+
+Hello World
|
|
78
|
+
+This is a test file
|
|
79
|
+
*** End Patch`
|
|
80
|
+
|
|
81
|
+
const result = await patchTool.execute({ patchText }, ctx)
|
|
82
|
+
|
|
83
|
+
expect(result.title).toContain("files changed")
|
|
84
|
+
expect(result.metadata.diff).toBeDefined()
|
|
85
|
+
expect(result.output).toContain("Patch applied successfully")
|
|
86
|
+
|
|
87
|
+
// Verify file was created
|
|
88
|
+
const filePath = path.join(fixture.path, "test-file.txt")
|
|
89
|
+
const content = await fs.readFile(filePath, "utf-8")
|
|
90
|
+
expect(content).toBe("Hello World\nThis is a test file")
|
|
91
|
+
},
|
|
92
|
+
})
|
|
93
|
+
})
|
|
94
|
+
|
|
95
|
+
test("should handle file with context update", async () => {
|
|
96
|
+
await using fixture = await tmpdir()
|
|
97
|
+
|
|
98
|
+
await Instance.provide({
|
|
99
|
+
directory: fixture.path,
|
|
100
|
+
fn: async () => {
|
|
101
|
+
const patchText = `*** Begin Patch
|
|
102
|
+
*** Add File: config.js
|
|
103
|
+
+const API_KEY = "test-key"
|
|
104
|
+
+const DEBUG = false
|
|
105
|
+
+const VERSION = "1.0"
|
|
106
|
+
*** End Patch`
|
|
107
|
+
|
|
108
|
+
const result = await patchTool.execute({ patchText }, ctx)
|
|
109
|
+
|
|
110
|
+
expect(result.title).toContain("files changed")
|
|
111
|
+
expect(result.metadata.diff).toBeDefined()
|
|
112
|
+
expect(result.output).toContain("Patch applied successfully")
|
|
113
|
+
|
|
114
|
+
// Verify file was created with correct content
|
|
115
|
+
const filePath = path.join(fixture.path, "config.js")
|
|
116
|
+
const content = await fs.readFile(filePath, "utf-8")
|
|
117
|
+
expect(content).toBe('const API_KEY = "test-key"\nconst DEBUG = false\nconst VERSION = "1.0"')
|
|
118
|
+
},
|
|
119
|
+
})
|
|
120
|
+
})
|
|
121
|
+
|
|
122
|
+
test("should handle multiple file operations", async () => {
|
|
123
|
+
await using fixture = await tmpdir()
|
|
124
|
+
|
|
125
|
+
await Instance.provide({
|
|
126
|
+
directory: fixture.path,
|
|
127
|
+
fn: async () => {
|
|
128
|
+
const patchText = `*** Begin Patch
|
|
129
|
+
*** Add File: file1.txt
|
|
130
|
+
+Content of file 1
|
|
131
|
+
*** Add File: file2.txt
|
|
132
|
+
+Content of file 2
|
|
133
|
+
*** Add File: file3.txt
|
|
134
|
+
+Content of file 3
|
|
135
|
+
*** End Patch`
|
|
136
|
+
|
|
137
|
+
const result = await patchTool.execute({ patchText }, ctx)
|
|
138
|
+
|
|
139
|
+
expect(result.title).toContain("3 files changed")
|
|
140
|
+
expect(result.metadata.diff).toBeDefined()
|
|
141
|
+
expect(result.output).toContain("Patch applied successfully")
|
|
142
|
+
|
|
143
|
+
// Verify all files were created
|
|
144
|
+
for (let i = 1; i <= 3; i++) {
|
|
145
|
+
const filePath = path.join(fixture.path, `file${i}.txt`)
|
|
146
|
+
const content = await fs.readFile(filePath, "utf-8")
|
|
147
|
+
expect(content).toBe(`Content of file ${i}`)
|
|
148
|
+
}
|
|
149
|
+
},
|
|
150
|
+
})
|
|
151
|
+
})
|
|
152
|
+
|
|
153
|
+
test("should create parent directories when adding nested files", async () => {
|
|
154
|
+
await using fixture = await tmpdir()
|
|
155
|
+
|
|
156
|
+
await Instance.provide({
|
|
157
|
+
directory: fixture.path,
|
|
158
|
+
fn: async () => {
|
|
159
|
+
const patchText = `*** Begin Patch
|
|
160
|
+
*** Add File: deep/nested/file.txt
|
|
161
|
+
+Deep nested content
|
|
162
|
+
*** End Patch`
|
|
163
|
+
|
|
164
|
+
const result = await patchTool.execute({ patchText }, ctx)
|
|
165
|
+
|
|
166
|
+
expect(result.title).toContain("files changed")
|
|
167
|
+
expect(result.output).toContain("Patch applied successfully")
|
|
168
|
+
|
|
169
|
+
// Verify nested file was created
|
|
170
|
+
const nestedPath = path.join(fixture.path, "deep", "nested", "file.txt")
|
|
171
|
+
const exists = await fs
|
|
172
|
+
.access(nestedPath)
|
|
173
|
+
.then(() => true)
|
|
174
|
+
.catch(() => false)
|
|
175
|
+
expect(exists).toBe(true)
|
|
176
|
+
|
|
177
|
+
const content = await fs.readFile(nestedPath, "utf-8")
|
|
178
|
+
expect(content).toBe("Deep nested content")
|
|
179
|
+
},
|
|
180
|
+
})
|
|
181
|
+
})
|
|
182
|
+
|
|
183
|
+
test("should generate proper unified diff in metadata", async () => {
|
|
184
|
+
await using fixture = await tmpdir()
|
|
185
|
+
|
|
186
|
+
await Instance.provide({
|
|
187
|
+
directory: fixture.path,
|
|
188
|
+
fn: async () => {
|
|
189
|
+
// First create a file with simple content
|
|
190
|
+
const patchText1 = `*** Begin Patch
|
|
191
|
+
*** Add File: test.txt
|
|
192
|
+
+line 1
|
|
193
|
+
+line 2
|
|
194
|
+
+line 3
|
|
195
|
+
*** End Patch`
|
|
196
|
+
|
|
197
|
+
await patchTool.execute({ patchText: patchText1 }, ctx)
|
|
198
|
+
|
|
199
|
+
// Now create an update patch
|
|
200
|
+
const patchText2 = `*** Begin Patch
|
|
201
|
+
*** Update File: test.txt
|
|
202
|
+
@@
|
|
203
|
+
line 1
|
|
204
|
+
-line 2
|
|
205
|
+
+line 2 updated
|
|
206
|
+
line 3
|
|
207
|
+
*** End Patch`
|
|
208
|
+
|
|
209
|
+
const result = await patchTool.execute({ patchText: patchText2 }, ctx)
|
|
210
|
+
|
|
211
|
+
expect(result.metadata.diff).toBeDefined()
|
|
212
|
+
expect(result.metadata.diff).toContain("@@")
|
|
213
|
+
expect(result.metadata.diff).toContain("-line 2")
|
|
214
|
+
expect(result.metadata.diff).toContain("+line 2 updated")
|
|
215
|
+
},
|
|
216
|
+
})
|
|
217
|
+
})
|
|
218
|
+
|
|
219
|
+
test("should handle complex patch with multiple operations", async () => {
|
|
220
|
+
await using fixture = await tmpdir()
|
|
221
|
+
|
|
222
|
+
await Instance.provide({
|
|
223
|
+
directory: fixture.path,
|
|
224
|
+
fn: async () => {
|
|
225
|
+
const patchText = `*** Begin Patch
|
|
226
|
+
*** Add File: new.txt
|
|
227
|
+
+This is a new file
|
|
228
|
+
+with multiple lines
|
|
229
|
+
*** Add File: existing.txt
|
|
230
|
+
+old content
|
|
231
|
+
+new line
|
|
232
|
+
+more content
|
|
233
|
+
*** Add File: config.json
|
|
234
|
+
+{
|
|
235
|
+
+ "version": "1.0",
|
|
236
|
+
+ "debug": true
|
|
237
|
+
+}
|
|
238
|
+
*** End Patch`
|
|
239
|
+
|
|
240
|
+
const result = await patchTool.execute({ patchText }, ctx)
|
|
241
|
+
|
|
242
|
+
expect(result.title).toContain("3 files changed")
|
|
243
|
+
expect(result.metadata.diff).toBeDefined()
|
|
244
|
+
expect(result.output).toContain("Patch applied successfully")
|
|
245
|
+
|
|
246
|
+
// Verify all files were created
|
|
247
|
+
const newPath = path.join(fixture.path, "new.txt")
|
|
248
|
+
const newContent = await fs.readFile(newPath, "utf-8")
|
|
249
|
+
expect(newContent).toBe("This is a new file\nwith multiple lines")
|
|
250
|
+
|
|
251
|
+
const existingPath = path.join(fixture.path, "existing.txt")
|
|
252
|
+
const existingContent = await fs.readFile(existingPath, "utf-8")
|
|
253
|
+
expect(existingContent).toBe("old content\nnew line\nmore content")
|
|
254
|
+
|
|
255
|
+
const configPath = path.join(fixture.path, "config.json")
|
|
256
|
+
const configContent = await fs.readFile(configPath, "utf-8")
|
|
257
|
+
expect(configContent).toBe('{\n "version": "1.0",\n "debug": true\n}')
|
|
258
|
+
},
|
|
259
|
+
})
|
|
260
|
+
})
|
|
261
|
+
})
|
|
@@ -0,0 +1,303 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test"
|
|
2
|
+
import path from "path"
|
|
3
|
+
import { ReadTool } from "../../src/tool/read"
|
|
4
|
+
import { Instance } from "../../src/project/instance"
|
|
5
|
+
import { tmpdir } from "../fixture/fixture"
|
|
6
|
+
import { PermissionNext } from "../../src/permission/next"
|
|
7
|
+
import { Agent } from "../../src/agent/agent"
|
|
8
|
+
|
|
9
|
+
const FIXTURES_DIR = path.join(import.meta.dir, "fixtures")
|
|
10
|
+
|
|
11
|
+
const ctx = {
|
|
12
|
+
sessionID: "test",
|
|
13
|
+
messageID: "",
|
|
14
|
+
callID: "",
|
|
15
|
+
agent: "build",
|
|
16
|
+
abort: AbortSignal.any([]),
|
|
17
|
+
metadata: () => {},
|
|
18
|
+
ask: async () => {},
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
describe("tool.read external_directory permission", () => {
|
|
22
|
+
test("allows reading absolute path inside project directory", async () => {
|
|
23
|
+
await using tmp = await tmpdir({
|
|
24
|
+
init: async (dir) => {
|
|
25
|
+
await Bun.write(path.join(dir, "test.txt"), "hello world")
|
|
26
|
+
},
|
|
27
|
+
})
|
|
28
|
+
await Instance.provide({
|
|
29
|
+
directory: tmp.path,
|
|
30
|
+
fn: async () => {
|
|
31
|
+
const read = await ReadTool.init()
|
|
32
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "test.txt") }, ctx)
|
|
33
|
+
expect(result.output).toContain("hello world")
|
|
34
|
+
},
|
|
35
|
+
})
|
|
36
|
+
})
|
|
37
|
+
|
|
38
|
+
test("allows reading file in subdirectory inside project directory", async () => {
|
|
39
|
+
await using tmp = await tmpdir({
|
|
40
|
+
init: async (dir) => {
|
|
41
|
+
await Bun.write(path.join(dir, "subdir", "test.txt"), "nested content")
|
|
42
|
+
},
|
|
43
|
+
})
|
|
44
|
+
await Instance.provide({
|
|
45
|
+
directory: tmp.path,
|
|
46
|
+
fn: async () => {
|
|
47
|
+
const read = await ReadTool.init()
|
|
48
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "subdir", "test.txt") }, ctx)
|
|
49
|
+
expect(result.output).toContain("nested content")
|
|
50
|
+
},
|
|
51
|
+
})
|
|
52
|
+
})
|
|
53
|
+
|
|
54
|
+
test("asks for external_directory permission when reading absolute path outside project", async () => {
|
|
55
|
+
await using outerTmp = await tmpdir({
|
|
56
|
+
init: async (dir) => {
|
|
57
|
+
await Bun.write(path.join(dir, "secret.txt"), "secret data")
|
|
58
|
+
},
|
|
59
|
+
})
|
|
60
|
+
await using tmp = await tmpdir({ git: true })
|
|
61
|
+
await Instance.provide({
|
|
62
|
+
directory: tmp.path,
|
|
63
|
+
fn: async () => {
|
|
64
|
+
const read = await ReadTool.init()
|
|
65
|
+
const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
|
|
66
|
+
const testCtx = {
|
|
67
|
+
...ctx,
|
|
68
|
+
ask: async (req: Omit<PermissionNext.Request, "id" | "sessionID" | "tool">) => {
|
|
69
|
+
requests.push(req)
|
|
70
|
+
},
|
|
71
|
+
}
|
|
72
|
+
await read.execute({ filePath: path.join(outerTmp.path, "secret.txt") }, testCtx)
|
|
73
|
+
const extDirReq = requests.find((r) => r.permission === "external_directory")
|
|
74
|
+
expect(extDirReq).toBeDefined()
|
|
75
|
+
expect(extDirReq!.patterns.some((p) => p.includes(outerTmp.path))).toBe(true)
|
|
76
|
+
},
|
|
77
|
+
})
|
|
78
|
+
})
|
|
79
|
+
|
|
80
|
+
test("asks for external_directory permission when reading relative path outside project", async () => {
|
|
81
|
+
await using tmp = await tmpdir({ git: true })
|
|
82
|
+
await Instance.provide({
|
|
83
|
+
directory: tmp.path,
|
|
84
|
+
fn: async () => {
|
|
85
|
+
const read = await ReadTool.init()
|
|
86
|
+
const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
|
|
87
|
+
const testCtx = {
|
|
88
|
+
...ctx,
|
|
89
|
+
ask: async (req: Omit<PermissionNext.Request, "id" | "sessionID" | "tool">) => {
|
|
90
|
+
requests.push(req)
|
|
91
|
+
},
|
|
92
|
+
}
|
|
93
|
+
// This will fail because file doesn't exist, but we can check if permission was asked
|
|
94
|
+
await read.execute({ filePath: "../outside.txt" }, testCtx).catch(() => {})
|
|
95
|
+
const extDirReq = requests.find((r) => r.permission === "external_directory")
|
|
96
|
+
expect(extDirReq).toBeDefined()
|
|
97
|
+
},
|
|
98
|
+
})
|
|
99
|
+
})
|
|
100
|
+
|
|
101
|
+
test("does not ask for external_directory permission when reading inside project", async () => {
|
|
102
|
+
await using tmp = await tmpdir({
|
|
103
|
+
git: true,
|
|
104
|
+
init: async (dir) => {
|
|
105
|
+
await Bun.write(path.join(dir, "internal.txt"), "internal content")
|
|
106
|
+
},
|
|
107
|
+
})
|
|
108
|
+
await Instance.provide({
|
|
109
|
+
directory: tmp.path,
|
|
110
|
+
fn: async () => {
|
|
111
|
+
const read = await ReadTool.init()
|
|
112
|
+
const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
|
|
113
|
+
const testCtx = {
|
|
114
|
+
...ctx,
|
|
115
|
+
ask: async (req: Omit<PermissionNext.Request, "id" | "sessionID" | "tool">) => {
|
|
116
|
+
requests.push(req)
|
|
117
|
+
},
|
|
118
|
+
}
|
|
119
|
+
await read.execute({ filePath: path.join(tmp.path, "internal.txt") }, testCtx)
|
|
120
|
+
const extDirReq = requests.find((r) => r.permission === "external_directory")
|
|
121
|
+
expect(extDirReq).toBeUndefined()
|
|
122
|
+
},
|
|
123
|
+
})
|
|
124
|
+
})
|
|
125
|
+
})
|
|
126
|
+
|
|
127
|
+
describe("tool.read env file blocking", () => {
|
|
128
|
+
const cases: [string, boolean][] = [
|
|
129
|
+
[".env", true],
|
|
130
|
+
[".env.local", true],
|
|
131
|
+
[".env.production", true],
|
|
132
|
+
[".env.development.local", true],
|
|
133
|
+
[".env.example", false],
|
|
134
|
+
[".envrc", false],
|
|
135
|
+
["environment.ts", false],
|
|
136
|
+
]
|
|
137
|
+
|
|
138
|
+
describe.each(["build", "plan"])("agent=%s", (agentName) => {
|
|
139
|
+
test.each(cases)("%s blocked=%s", async (filename, blocked) => {
|
|
140
|
+
await using tmp = await tmpdir({
|
|
141
|
+
init: (dir) => Bun.write(path.join(dir, filename), "content"),
|
|
142
|
+
})
|
|
143
|
+
await Instance.provide({
|
|
144
|
+
directory: tmp.path,
|
|
145
|
+
fn: async () => {
|
|
146
|
+
const agent = await Agent.get(agentName)
|
|
147
|
+
const ctxWithPermissions = {
|
|
148
|
+
...ctx,
|
|
149
|
+
ask: async (req: Omit<PermissionNext.Request, "id" | "sessionID" | "tool">) => {
|
|
150
|
+
for (const pattern of req.patterns) {
|
|
151
|
+
const rule = PermissionNext.evaluate(req.permission, pattern, agent.permission)
|
|
152
|
+
if (rule.action === "deny") {
|
|
153
|
+
throw new PermissionNext.DeniedError(agent.permission)
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
},
|
|
157
|
+
}
|
|
158
|
+
const read = await ReadTool.init()
|
|
159
|
+
const promise = read.execute({ filePath: path.join(tmp.path, filename) }, ctxWithPermissions)
|
|
160
|
+
if (blocked) {
|
|
161
|
+
await expect(promise).rejects.toThrow(PermissionNext.DeniedError)
|
|
162
|
+
} else {
|
|
163
|
+
expect((await promise).output).toContain("content")
|
|
164
|
+
}
|
|
165
|
+
},
|
|
166
|
+
})
|
|
167
|
+
})
|
|
168
|
+
})
|
|
169
|
+
})
|
|
170
|
+
|
|
171
|
+
describe("tool.read truncation", () => {
|
|
172
|
+
test("truncates large file by bytes and sets truncated metadata", async () => {
|
|
173
|
+
await using tmp = await tmpdir({
|
|
174
|
+
init: async (dir) => {
|
|
175
|
+
const content = await Bun.file(path.join(FIXTURES_DIR, "models-api.json")).text()
|
|
176
|
+
await Bun.write(path.join(dir, "large.json"), content)
|
|
177
|
+
},
|
|
178
|
+
})
|
|
179
|
+
await Instance.provide({
|
|
180
|
+
directory: tmp.path,
|
|
181
|
+
fn: async () => {
|
|
182
|
+
const read = await ReadTool.init()
|
|
183
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "large.json") }, ctx)
|
|
184
|
+
expect(result.metadata.truncated).toBe(true)
|
|
185
|
+
expect(result.output).toContain("Output truncated at")
|
|
186
|
+
expect(result.output).toContain("bytes")
|
|
187
|
+
},
|
|
188
|
+
})
|
|
189
|
+
})
|
|
190
|
+
|
|
191
|
+
test("truncates by line count when limit is specified", async () => {
|
|
192
|
+
await using tmp = await tmpdir({
|
|
193
|
+
init: async (dir) => {
|
|
194
|
+
const lines = Array.from({ length: 100 }, (_, i) => `line${i}`).join("\n")
|
|
195
|
+
await Bun.write(path.join(dir, "many-lines.txt"), lines)
|
|
196
|
+
},
|
|
197
|
+
})
|
|
198
|
+
await Instance.provide({
|
|
199
|
+
directory: tmp.path,
|
|
200
|
+
fn: async () => {
|
|
201
|
+
const read = await ReadTool.init()
|
|
202
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "many-lines.txt"), limit: 10 }, ctx)
|
|
203
|
+
expect(result.metadata.truncated).toBe(true)
|
|
204
|
+
expect(result.output).toContain("File has more lines")
|
|
205
|
+
expect(result.output).toContain("line0")
|
|
206
|
+
expect(result.output).toContain("line9")
|
|
207
|
+
expect(result.output).not.toContain("line10")
|
|
208
|
+
},
|
|
209
|
+
})
|
|
210
|
+
})
|
|
211
|
+
|
|
212
|
+
test("does not truncate small file", async () => {
|
|
213
|
+
await using tmp = await tmpdir({
|
|
214
|
+
init: async (dir) => {
|
|
215
|
+
await Bun.write(path.join(dir, "small.txt"), "hello world")
|
|
216
|
+
},
|
|
217
|
+
})
|
|
218
|
+
await Instance.provide({
|
|
219
|
+
directory: tmp.path,
|
|
220
|
+
fn: async () => {
|
|
221
|
+
const read = await ReadTool.init()
|
|
222
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "small.txt") }, ctx)
|
|
223
|
+
expect(result.metadata.truncated).toBe(false)
|
|
224
|
+
expect(result.output).toContain("End of file")
|
|
225
|
+
},
|
|
226
|
+
})
|
|
227
|
+
})
|
|
228
|
+
|
|
229
|
+
test("respects offset parameter", async () => {
|
|
230
|
+
await using tmp = await tmpdir({
|
|
231
|
+
init: async (dir) => {
|
|
232
|
+
const lines = Array.from({ length: 20 }, (_, i) => `line${i}`).join("\n")
|
|
233
|
+
await Bun.write(path.join(dir, "offset.txt"), lines)
|
|
234
|
+
},
|
|
235
|
+
})
|
|
236
|
+
await Instance.provide({
|
|
237
|
+
directory: tmp.path,
|
|
238
|
+
fn: async () => {
|
|
239
|
+
const read = await ReadTool.init()
|
|
240
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "offset.txt"), offset: 10, limit: 5 }, ctx)
|
|
241
|
+
expect(result.output).toContain("line10")
|
|
242
|
+
expect(result.output).toContain("line14")
|
|
243
|
+
expect(result.output).not.toContain("line0")
|
|
244
|
+
expect(result.output).not.toContain("line15")
|
|
245
|
+
},
|
|
246
|
+
})
|
|
247
|
+
})
|
|
248
|
+
|
|
249
|
+
test("truncates long lines", async () => {
|
|
250
|
+
await using tmp = await tmpdir({
|
|
251
|
+
init: async (dir) => {
|
|
252
|
+
const longLine = "x".repeat(3000)
|
|
253
|
+
await Bun.write(path.join(dir, "long-line.txt"), longLine)
|
|
254
|
+
},
|
|
255
|
+
})
|
|
256
|
+
await Instance.provide({
|
|
257
|
+
directory: tmp.path,
|
|
258
|
+
fn: async () => {
|
|
259
|
+
const read = await ReadTool.init()
|
|
260
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "long-line.txt") }, ctx)
|
|
261
|
+
expect(result.output).toContain("...")
|
|
262
|
+
expect(result.output.length).toBeLessThan(3000)
|
|
263
|
+
},
|
|
264
|
+
})
|
|
265
|
+
})
|
|
266
|
+
|
|
267
|
+
test("image files set truncated to false", async () => {
|
|
268
|
+
await using tmp = await tmpdir({
|
|
269
|
+
init: async (dir) => {
|
|
270
|
+
// 1x1 red PNG
|
|
271
|
+
const png = Buffer.from(
|
|
272
|
+
"iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8DwHwAFBQIAX8jx0gAAAABJRU5ErkJggg==",
|
|
273
|
+
"base64",
|
|
274
|
+
)
|
|
275
|
+
await Bun.write(path.join(dir, "image.png"), png)
|
|
276
|
+
},
|
|
277
|
+
})
|
|
278
|
+
await Instance.provide({
|
|
279
|
+
directory: tmp.path,
|
|
280
|
+
fn: async () => {
|
|
281
|
+
const read = await ReadTool.init()
|
|
282
|
+
const result = await read.execute({ filePath: path.join(tmp.path, "image.png") }, ctx)
|
|
283
|
+
expect(result.metadata.truncated).toBe(false)
|
|
284
|
+
expect(result.attachments).toBeDefined()
|
|
285
|
+
expect(result.attachments?.length).toBe(1)
|
|
286
|
+
},
|
|
287
|
+
})
|
|
288
|
+
})
|
|
289
|
+
|
|
290
|
+
test("large image files are properly attached without error", async () => {
|
|
291
|
+
await Instance.provide({
|
|
292
|
+
directory: FIXTURES_DIR,
|
|
293
|
+
fn: async () => {
|
|
294
|
+
const read = await ReadTool.init()
|
|
295
|
+
const result = await read.execute({ filePath: path.join(FIXTURES_DIR, "large-image.png") }, ctx)
|
|
296
|
+
expect(result.metadata.truncated).toBe(false)
|
|
297
|
+
expect(result.attachments).toBeDefined()
|
|
298
|
+
expect(result.attachments?.length).toBe(1)
|
|
299
|
+
expect(result.attachments?.[0].type).toBe("file")
|
|
300
|
+
},
|
|
301
|
+
})
|
|
302
|
+
})
|
|
303
|
+
})
|