openAgent

1import { describe, expect, test } from "bun:test"
2import type { PermissionRequest } from "@opencode-ai/sdk/v2"
3import {
4  createPermissionBodyState,
5  permissionAlwaysLines,
6  permissionCancel,
7  permissionEscape,
8  permissionInfo,
9  permissionReject,
10  permissionRun,
11} from "@/cli/cmd/run/permission.shared"
12 
13function req(input: Partial<PermissionRequest> = {}): PermissionRequest {
14  return {
15    id: "perm-1",
16    sessionID: "session-1",
17    permission: "read",
18    patterns: [],
19    metadata: {},
20    always: [],
21    ...input,
22  }
23}
24 
25describe("run permission shared", () => {
26  test("replies immediately for allow once", () => {
27    const out = permissionRun(createPermissionBodyState("perm-1"), "perm-1", "once")
28 
29    expect(out.reply).toEqual({
30      requestID: "perm-1",
31      reply: "once",
32    })
33  })
34 
35  test("requires confirmation for allow always", () => {
36    const next = permissionRun(createPermissionBodyState("perm-1"), "perm-1", "always")
37    expect(next.state.stage).toBe("always")
38    expect(next.state.selected).toBe("confirm")
39    expect(next.reply).toBeUndefined()
40 
41    expect(permissionRun(next.state, "perm-1", "confirm").reply).toEqual({
42      requestID: "perm-1",
43      reply: "always",
44    })
45 
46    expect(permissionRun(next.state, "perm-1", "cancel").state).toMatchObject({
47      stage: "permission",
48      selected: "always",
49    })
50  })
51 
52  test("builds trimmed reject replies and stage transitions", () => {
53    const next = permissionRun(createPermissionBodyState("perm-1"), "perm-1", "reject")
54    expect(next.state.stage).toBe("reject")
55 
56    const out = permissionReject({ ...next.state, message: "  use rg  " }, "perm-1")
57    expect(out).toEqual({
58      requestID: "perm-1",
59      reply: "reject",
60      message: "use rg",
61    })
62 
63    expect(permissionCancel(next.state)).toMatchObject({
64      stage: "permission",
65      selected: "reject",
66    })
67 
68    expect(permissionEscape(createPermissionBodyState("perm-1"))).toMatchObject({
69      stage: "reject",
70      selected: "reject",
71    })
72 
73    expect(permissionEscape({ ...next.state, stage: "always", selected: "confirm" })).toMatchObject({
74      stage: "permission",
75      selected: "always",
76    })
77  })
78 
79  test("maps supported permission types into display info", () => {
80    expect(
81      permissionInfo(
82        req({
83          permission: "bash",
84          metadata: {
85            input: {
86              command: "git status --short",
87            },
88          },
89        }),
90      ),
91    ).toMatchObject({
92      title: "Shell command",
93      lines: ["$ git status --short"],
94    })
95 
96    expect(
97      permissionInfo(
98        req({
99          permission: "task",
100          metadata: {
101            description: "investigate stream",
102            subagent_type: "general",
103          },
104        }),
105      ),
106    ).toMatchObject({
107      title: "General Task",
108      lines: ["◉ investigate stream"],
109    })
110 
111    expect(
112      permissionInfo(
113        req({
114          permission: "external_directory",
115          patterns: ["/tmp/work/**/*.ts", "/tmp/work/**/*.tsx"],
116        }),
117      ),
118    ).toMatchObject({
119      title: "Access external directory /tmp/work",
120      lines: ["- /tmp/work/**/*.ts", "- /tmp/work/**/*.tsx"],
121    })
122 
123    expect(permissionInfo(req({ permission: "doom_loop" }))).toMatchObject({
124      title: "Continue after repeated failures",
125    })
126 
127    expect(permissionInfo(req({ permission: "custom_tool" }))).toMatchObject({
128      title: "Call tool custom_tool",
129      lines: ["Tool: custom_tool"],
130    })
131  })
132 
133  test("formats always-allow copy for wildcard and explicit patterns", () => {
134    expect(permissionAlwaysLines(req({ permission: "bash", always: ["*"] }))).toEqual([
135      "This will allow bash until OpenCode is restarted.",
136    ])
137 
138    expect(permissionAlwaysLines(req({ always: ["src/**/*.ts", "src/**/*.tsx"] }))).toEqual([
139      "This will allow the following patterns until OpenCode is restarted.",
140      "- src/**/*.ts",
141      "- src/**/*.tsx",
142    ])
143  })
144})
145