oldschoola
715eb0792c
Previously only 4 commands (/plan, /goal, /mcp, /ssh) stored their text in history via per-handler addToHistory calls. All other built-in slash commands were silently skipped because executeBuiltinSlashCommand returned true before the input controller's addToHistory was reached. - Centralize history recording in the input controller after successful slash command dispatch, for both Enter and Ctrl+Enter submit paths. - Remove all 10 per-command addToHistory calls from slash command handlers to prevent duplicates. - Add shouldSkipHistory() security filter to exclude commands that may carry secrets: /login <url> (OAuth callback with code=/state= params) and /mcp add --token <token> (bearer token). - Add regression tests for the security filter (8 cases). - Update 7 existing test files to remove handler-level addToHistory assertions (now the input controller's responsibility).
55 lines
1.7 KiB
TypeScript
55 lines
1.7 KiB
TypeScript
import { describe, expect, it, vi } from "bun:test";
|
|
import type { InteractiveModeContext } from "@oh-my-pi/pi-coding-agent/modes/types";
|
|
import { executeBuiltinSlashCommand } from "@oh-my-pi/pi-coding-agent/slash-commands/builtin-registry";
|
|
|
|
function createRuntime() {
|
|
const handleBtwCommand = vi.fn(async () => {});
|
|
const setText = vi.fn();
|
|
const addToHistory = vi.fn();
|
|
return {
|
|
handleBtwCommand,
|
|
setText,
|
|
addToHistory,
|
|
runtime: {
|
|
ctx: {
|
|
editor: { setText, addToHistory } as unknown as InteractiveModeContext["editor"],
|
|
handleBtwCommand,
|
|
} as unknown as InteractiveModeContext,
|
|
},
|
|
};
|
|
}
|
|
|
|
describe("/btw slash command", () => {
|
|
it("routes the full question through the interactive btw handler", async () => {
|
|
const harness = createRuntime();
|
|
|
|
const handled = await executeBuiltinSlashCommand("/btw why is it doing that?", harness.runtime);
|
|
|
|
expect(handled).toBe(true);
|
|
expect(harness.setText).toHaveBeenCalledWith("");
|
|
expect(harness.handleBtwCommand).toHaveBeenCalledWith("why is it doing that?");
|
|
});
|
|
|
|
it("preserves the raw multi-word suffix after /btw", async () => {
|
|
const harness = createRuntime();
|
|
|
|
const handled = await executeBuiltinSlashCommand(
|
|
"/btw explain why the cache reuse matters here",
|
|
harness.runtime,
|
|
);
|
|
|
|
expect(handled).toBe(true);
|
|
expect(harness.handleBtwCommand).toHaveBeenCalledWith("explain why the cache reuse matters here");
|
|
});
|
|
|
|
it("handles a blank /btw invocation without error", async () => {
|
|
const harness = createRuntime();
|
|
|
|
const handled = await executeBuiltinSlashCommand("/btw ", harness.runtime);
|
|
|
|
expect(handled).toBe(true);
|
|
expect(harness.setText).toHaveBeenCalledWith("");
|
|
expect(harness.handleBtwCommand).toHaveBeenCalledWith("");
|
|
});
|
|
});
|