| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127 |
- import { describe, expect, test } from "bun:test"
- import path from "path"
- import type { Tool } from "../../src/tool/tool"
- import { Instance } from "../../src/project/instance"
- import { assertExternalDirectory } from "../../src/tool/external-directory"
- import type { PermissionNext } from "../../src/permission/next"
- const baseCtx: Omit<Tool.Context, "ask"> = {
- sessionID: "test",
- messageID: "",
- callID: "",
- agent: "build",
- abort: AbortSignal.any([]),
- messages: [],
- metadata: () => {},
- }
- describe("tool.assertExternalDirectory", () => {
- test("no-ops for empty target", async () => {
- const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
- const ctx: Tool.Context = {
- ...baseCtx,
- ask: async (req) => {
- requests.push(req)
- },
- }
- await Instance.provide({
- directory: "/tmp",
- fn: async () => {
- await assertExternalDirectory(ctx)
- },
- })
- expect(requests.length).toBe(0)
- })
- test("no-ops for paths inside Instance.directory", async () => {
- const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
- const ctx: Tool.Context = {
- ...baseCtx,
- ask: async (req) => {
- requests.push(req)
- },
- }
- await Instance.provide({
- directory: "/tmp/project",
- fn: async () => {
- await assertExternalDirectory(ctx, path.join("/tmp/project", "file.txt"))
- },
- })
- expect(requests.length).toBe(0)
- })
- test("asks with a single canonical glob", async () => {
- const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
- const ctx: Tool.Context = {
- ...baseCtx,
- ask: async (req) => {
- requests.push(req)
- },
- }
- const directory = "/tmp/project"
- const target = "/tmp/outside/file.txt"
- const expected = path.join(path.dirname(target), "*")
- await Instance.provide({
- directory,
- fn: async () => {
- await assertExternalDirectory(ctx, target)
- },
- })
- const req = requests.find((r) => r.permission === "external_directory")
- expect(req).toBeDefined()
- expect(req!.patterns).toEqual([expected])
- expect(req!.always).toEqual([expected])
- })
- test("uses target directory when kind=directory", async () => {
- const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
- const ctx: Tool.Context = {
- ...baseCtx,
- ask: async (req) => {
- requests.push(req)
- },
- }
- const directory = "/tmp/project"
- const target = "/tmp/outside"
- const expected = path.join(target, "*")
- await Instance.provide({
- directory,
- fn: async () => {
- await assertExternalDirectory(ctx, target, { kind: "directory" })
- },
- })
- const req = requests.find((r) => r.permission === "external_directory")
- expect(req).toBeDefined()
- expect(req!.patterns).toEqual([expected])
- expect(req!.always).toEqual([expected])
- })
- test("skips prompting when bypass=true", async () => {
- const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
- const ctx: Tool.Context = {
- ...baseCtx,
- ask: async (req) => {
- requests.push(req)
- },
- }
- await Instance.provide({
- directory: "/tmp/project",
- fn: async () => {
- await assertExternalDirectory(ctx, "/tmp/outside/file.txt", { bypass: true })
- },
- })
- expect(requests.length).toBe(0)
- })
- })
|