276 lines
9.3 KiB
TypeScript
276 lines
9.3 KiB
TypeScript
|
import { describe, expect, it, beforeEach, mock } from "bun:test"
|
||
|
|
import type {
|
||
|
|
AutoSlashCommandHookInput,
|
||
|
|
AutoSlashCommandHookOutput,
|
||
|
|
} from "./types"
|
||
|
|
|
||
|
|
const logMock = mock(() => {})
|
||
|
|
|
||
|
|
mock.module("../../shared", () => ({
|
||
|
|
log: logMock,
|
||
|
|
parseFrontmatter: (content: string) => ({ data: {}, body: content }),
|
||
|
|
resolveCommandsInText: async (text: string) => text,
|
||
|
|
resolveFileReferencesInText: async (text: string) => text,
|
||
|
|
sanitizeModelField: (model: unknown) => model,
|
||
|
|
getClaudeConfigDir: () => "/mock/.claude",
|
||
|
|
}))
|
||
|
|
|
||
|
|
mock.module("../../shared/file-utils", () => ({
|
||
|
|
isMarkdownFile: () => false,
|
||
|
|
}))
|
||
|
|
|
||
|
|
mock.module("../../features/opencode-skill-loader", () => ({
|
||
|
|
discoverAllSkills: () => [],
|
||
|
|
}))
|
||
|
|
|
||
|
|
mock.module("fs", () => ({
|
||
|
|
existsSync: () => false,
|
||
|
|
readdirSync: () => [],
|
||
|
|
readFileSync: () => "",
|
||
|
|
}))
|
||
|
|
|
||
|
|
const { createAutoSlashCommandHook } = await import("./index")
|
||
|
|
|
||
|
|
function createMockInput(sessionID: string, messageID?: string): AutoSlashCommandHookInput {
|
||
|
|
return {
|
||
|
|
sessionID,
|
||
|
|
messageID: messageID ?? `msg-${Date.now()}-${Math.random()}`,
|
||
|
|
agent: "test-agent",
|
||
|
|
model: { providerID: "anthropic", modelID: "claude-sonnet-4-5" },
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function createMockOutput(text: string): AutoSlashCommandHookOutput {
|
||
|
|
return {
|
||
|
|
message: {
|
||
|
|
agent: "test-agent",
|
||
|
|
model: { providerID: "anthropic", modelID: "claude-sonnet-4-5" },
|
||
|
|
path: { cwd: "/test", root: "/test" },
|
||
|
|
tools: {},
|
||
|
|
},
|
||
|
|
parts: [{ type: "text", text }],
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
describe("createAutoSlashCommandHook", () => {
|
||
|
|
beforeEach(() => {
|
||
|
|
logMock.mockClear()
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("slash command replacement", () => {
|
||
|
|
it("should replace message with error when command not found", async () => {
|
||
|
|
// #given a slash command that doesn't exist
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-notfound-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("/nonexistent-command args")
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should replace with error message
|
||
|
|
const textPart = output.parts.find((p) => p.type === "text")
|
||
|
|
expect(textPart?.text).toContain("<auto-slash-command>")
|
||
|
|
expect(textPart?.text).toContain("not found")
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should wrap replacement in auto-slash-command tags", async () => {
|
||
|
|
// #given any slash command
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-tags-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("/some-command")
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should wrap in tags
|
||
|
|
const textPart = output.parts.find((p) => p.type === "text")
|
||
|
|
expect(textPart?.text).toContain("<auto-slash-command>")
|
||
|
|
expect(textPart?.text).toContain("</auto-slash-command>")
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should completely replace original message text", async () => {
|
||
|
|
// #given slash command
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-replace-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("/test-cmd some args")
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then original text should be replaced, not prepended
|
||
|
|
const textPart = output.parts.find((p) => p.type === "text")
|
||
|
|
expect(textPart?.text).not.toContain("/test-cmd some args\n<auto-slash-command>")
|
||
|
|
expect(textPart?.text?.startsWith("<auto-slash-command>")).toBe(true)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("no slash command", () => {
|
||
|
|
it("should do nothing for regular text", async () => {
|
||
|
|
// #given regular text without slash
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-regular-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("Just regular text")
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not modify
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should do nothing for slash in middle of text", async () => {
|
||
|
|
// #given slash in middle
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-middle-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("Please run /commit later")
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not detect (not at start)
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("excluded commands", () => {
|
||
|
|
it("should NOT trigger for ralph-loop command", async () => {
|
||
|
|
// #given ralph-loop command
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-ralph-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("/ralph-loop do something")
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not modify (excluded command)
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should NOT trigger for cancel-ralph command", async () => {
|
||
|
|
// #given cancel-ralph command
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-cancel-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("/cancel-ralph")
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not modify
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("already processed", () => {
|
||
|
|
it("should skip if auto-slash-command tags already present", async () => {
|
||
|
|
// #given text with existing tags
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-existing-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput(
|
||
|
|
"<auto-slash-command>/commit</auto-slash-command>"
|
||
|
|
)
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not modify
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("code blocks", () => {
|
||
|
|
it("should NOT detect command inside code block", async () => {
|
||
|
|
// #given command inside code block
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-codeblock-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("```\n/commit\n```")
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not detect
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("edge cases", () => {
|
||
|
|
it("should handle empty text", async () => {
|
||
|
|
// #given empty text
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-empty-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("")
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
// #then should not throw
|
||
|
|
await expect(hook["chat.message"](input, output)).resolves.toBeUndefined()
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should handle just slash", async () => {
|
||
|
|
// #given just slash
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-slash-only-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput("/")
|
||
|
|
const originalText = output.parts[0].text
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should not modify
|
||
|
|
expect(output.parts[0].text).toBe(originalText)
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should handle command with special characters in args", async () => {
|
||
|
|
// #given command with special characters
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-special-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output = createMockOutput('/execute "test & stuff <tag>"')
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should handle gracefully (not found, but processed)
|
||
|
|
const textPart = output.parts.find((p) => p.type === "text")
|
||
|
|
expect(textPart?.text).toContain("<auto-slash-command>")
|
||
|
|
expect(textPart?.text).toContain("/execute")
|
||
|
|
})
|
||
|
|
|
||
|
|
it("should handle multiple text parts", async () => {
|
||
|
|
// #given multiple text parts
|
||
|
|
const hook = createAutoSlashCommandHook()
|
||
|
|
const sessionID = `test-session-multi-${Date.now()}`
|
||
|
|
const input = createMockInput(sessionID)
|
||
|
|
const output: AutoSlashCommandHookOutput = {
|
||
|
|
message: {},
|
||
|
|
parts: [
|
||
|
|
{ type: "text", text: "/commit " },
|
||
|
|
{ type: "text", text: "fix bug" },
|
||
|
|
],
|
||
|
|
}
|
||
|
|
|
||
|
|
// #when hook is called
|
||
|
|
await hook["chat.message"](input, output)
|
||
|
|
|
||
|
|
// #then should detect from combined text and modify first text part
|
||
|
|
const firstTextPart = output.parts.find((p) => p.type === "text")
|
||
|
|
expect(firstTextPart?.text).toContain("<auto-slash-command>")
|
||
|
|
})
|
||
|
|
})
|
||
|
|
})
|