51dde4d43f
This PR ports the hashline edit tool from oh-my-pi to oh-my-opencode as an experimental feature. ## Features - New experimental.hashline_edit config flag - hashline_edit tool with 4 operations: set_line, replace_lines, insert_after, replace - Hash-based line anchors for safe concurrent editing - Edit tool disabler for non-OpenAI providers - Read output enhancer with LINE:HASH prefixes - Provider state tracking module ## Technical Details - xxHash32-based 2-char hex hashes - Bottom-up edit application to prevent index shifting - OpenAI provider exemption (uses native apply_patch) - 90 tests covering all operations and edge cases - All files under 200 LOC limit ## Files Added/Modified - src/tools/hashline-edit/ (7 files, ~400 LOC) - src/hooks/hashline-edit-disabler/ (4 files, ~200 LOC) - src/hooks/hashline-read-enhancer/ (3 files, ~400 LOC) - src/features/hashline-provider-state.ts (13 LOC) - src/config/schema/experimental.ts (hashline_edit flag) - src/config/schema/hooks.ts (2 new hook names) - src/plugin/tool-registry.ts (conditional registration) - src/plugin/chat-params.ts (provider state tracking) - src/tools/index.ts (export) - src/hooks/index.ts (exports)
87 lines
2.7 KiB
TypeScript
87 lines
2.7 KiB
TypeScript
import { describe, expect, it, afterAll, mock } from "bun:test"
|
|
import type { PluginInput } from "@opencode-ai/plugin"
|
|
import { createOpencodeClient } from "@opencode-ai/sdk"
|
|
import type { Todo } from "@opencode-ai/sdk"
|
|
import { createCompactionTodoPreserverHook } from "./index"
|
|
|
|
const updateMock = mock(async () => {})
|
|
|
|
mock.module("opencode/session/todo", () => ({
|
|
Todo: {
|
|
update: updateMock,
|
|
},
|
|
}))
|
|
|
|
afterAll(() => {
|
|
mock.module("opencode/session/todo", () => ({
|
|
Todo: {
|
|
update: async () => {},
|
|
},
|
|
}))
|
|
})
|
|
|
|
function createMockContext(todoResponses: Array<Todo>[]): PluginInput {
|
|
let callIndex = 0
|
|
|
|
const client = createOpencodeClient({ directory: "/tmp/test" })
|
|
type SessionTodoOptions = Parameters<typeof client.session.todo>[0]
|
|
type SessionTodoResult = ReturnType<typeof client.session.todo>
|
|
|
|
const request = new Request("http://localhost")
|
|
const response = new Response()
|
|
client.session.todo = mock((_: SessionTodoOptions): SessionTodoResult => {
|
|
const current = todoResponses[Math.min(callIndex, todoResponses.length - 1)] ?? []
|
|
callIndex += 1
|
|
return Promise.resolve({ data: current, error: undefined, request, response })
|
|
})
|
|
|
|
return {
|
|
client,
|
|
project: { id: "test-project", worktree: "/tmp/test", time: { created: Date.now() } },
|
|
directory: "/tmp/test",
|
|
worktree: "/tmp/test",
|
|
serverUrl: new URL("http://localhost"),
|
|
$: Bun.$,
|
|
}
|
|
}
|
|
|
|
describe("compaction-todo-preserver", () => {
|
|
it("restores todos after compaction when missing", async () => {
|
|
//#given
|
|
updateMock.mockClear()
|
|
const sessionID = "session-compaction-missing"
|
|
const todos: Todo[] = [
|
|
{ id: "1", content: "Task 1", status: "pending", priority: "high" },
|
|
{ id: "2", content: "Task 2", status: "in_progress", priority: "medium" },
|
|
]
|
|
const ctx = createMockContext([todos, []])
|
|
const hook = createCompactionTodoPreserverHook(ctx)
|
|
|
|
//#when
|
|
await hook.capture(sessionID)
|
|
await hook.event({ event: { type: "session.compacted", properties: { sessionID } } })
|
|
|
|
//#then
|
|
expect(updateMock).toHaveBeenCalledTimes(1)
|
|
expect(updateMock).toHaveBeenCalledWith({ sessionID, todos })
|
|
})
|
|
|
|
it("skips restore when todos already present", async () => {
|
|
//#given
|
|
updateMock.mockClear()
|
|
const sessionID = "session-compaction-present"
|
|
const todos: Todo[] = [
|
|
{ id: "1", content: "Task 1", status: "pending", priority: "high" },
|
|
]
|
|
const ctx = createMockContext([todos, todos])
|
|
const hook = createCompactionTodoPreserverHook(ctx)
|
|
|
|
//#when
|
|
await hook.capture(sessionID)
|
|
await hook.event({ event: { type: "session.compacted", properties: { sessionID } } })
|
|
|
|
//#then
|
|
expect(updateMock).not.toHaveBeenCalled()
|
|
})
|
|
})
|