external-directory.test.ts 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. import { describe, expect, test } from "bun:test"
  2. import path from "path"
  3. import type { Tool } from "../../src/tool/tool"
  4. import { Instance } from "../../src/project/instance"
  5. import { assertExternalDirectory } from "../../src/tool/external-directory"
  6. import type { PermissionNext } from "../../src/permission/next"
  7. const baseCtx: Omit<Tool.Context, "ask"> = {
  8. sessionID: "test",
  9. messageID: "",
  10. callID: "",
  11. agent: "build",
  12. abort: AbortSignal.any([]),
  13. messages: [],
  14. metadata: () => {},
  15. }
  16. describe("tool.assertExternalDirectory", () => {
  17. test("no-ops for empty target", async () => {
  18. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  19. const ctx: Tool.Context = {
  20. ...baseCtx,
  21. ask: async (req) => {
  22. requests.push(req)
  23. },
  24. }
  25. await Instance.provide({
  26. directory: "/tmp",
  27. fn: async () => {
  28. await assertExternalDirectory(ctx)
  29. },
  30. })
  31. expect(requests.length).toBe(0)
  32. })
  33. test("no-ops for paths inside Instance.directory", async () => {
  34. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  35. const ctx: Tool.Context = {
  36. ...baseCtx,
  37. ask: async (req) => {
  38. requests.push(req)
  39. },
  40. }
  41. await Instance.provide({
  42. directory: "/tmp/project",
  43. fn: async () => {
  44. await assertExternalDirectory(ctx, path.join("/tmp/project", "file.txt"))
  45. },
  46. })
  47. expect(requests.length).toBe(0)
  48. })
  49. test("asks with a single canonical glob", async () => {
  50. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  51. const ctx: Tool.Context = {
  52. ...baseCtx,
  53. ask: async (req) => {
  54. requests.push(req)
  55. },
  56. }
  57. const directory = "/tmp/project"
  58. const target = "/tmp/outside/file.txt"
  59. const expected = path.join(path.dirname(target), "*")
  60. await Instance.provide({
  61. directory,
  62. fn: async () => {
  63. await assertExternalDirectory(ctx, target)
  64. },
  65. })
  66. const req = requests.find((r) => r.permission === "external_directory")
  67. expect(req).toBeDefined()
  68. expect(req!.patterns).toEqual([expected])
  69. expect(req!.always).toEqual([expected])
  70. })
  71. test("uses target directory when kind=directory", async () => {
  72. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  73. const ctx: Tool.Context = {
  74. ...baseCtx,
  75. ask: async (req) => {
  76. requests.push(req)
  77. },
  78. }
  79. const directory = "/tmp/project"
  80. const target = "/tmp/outside"
  81. const expected = path.join(target, "*")
  82. await Instance.provide({
  83. directory,
  84. fn: async () => {
  85. await assertExternalDirectory(ctx, target, { kind: "directory" })
  86. },
  87. })
  88. const req = requests.find((r) => r.permission === "external_directory")
  89. expect(req).toBeDefined()
  90. expect(req!.patterns).toEqual([expected])
  91. expect(req!.always).toEqual([expected])
  92. })
  93. test("skips prompting when bypass=true", async () => {
  94. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  95. const ctx: Tool.Context = {
  96. ...baseCtx,
  97. ask: async (req) => {
  98. requests.push(req)
  99. },
  100. }
  101. await Instance.provide({
  102. directory: "/tmp/project",
  103. fn: async () => {
  104. await assertExternalDirectory(ctx, "/tmp/outside/file.txt", { bypass: true })
  105. },
  106. })
  107. expect(requests.length).toBe(0)
  108. })
  109. })