226 lines
7.3 KiB
Go
226 lines
7.3 KiB
Go
package library
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
// fakeRTK is an rtk on PATH that acts as rtk 0.50's installer does for
|
|
// Claude Code, OpenCode and Cursor: installing OpenCode or Cursor gives it to
|
|
// Claude Code as well. Its uninstaller isn't there: magpie doesn't use it.
|
|
const fakeRTK = `#!/bin/sh
|
|
c="${CLAUDE_CONFIG_DIR:-$HOME/.claude}"
|
|
claude() { [ -d "$c" ] && echo '{"hooks":{"PreToolUse":[{"hooks":[{"command":"rtk hook claude"}]}]}}' > "$c/settings.json"; }
|
|
case "$1" in
|
|
--version) echo "rtk 9.9.9"; exit 0 ;;
|
|
gain) echo '{"summary":{"total_commands":3,"total_input":100,"total_saved":60,"avg_savings_pct":60}}'; exit 0 ;;
|
|
esac
|
|
shift 2
|
|
case "$*" in
|
|
"--auto-patch") claude ;;
|
|
"--opencode --auto-patch") claude; /bin/mkdir -p "$XDG_CONFIG_HOME/opencode/plugins"; echo x > "$XDG_CONFIG_HOME/opencode/plugins/rtk.ts" ;;
|
|
"--agent cursor --auto-patch") claude; echo '{"hooks":{"preToolUse":[{"command":"rtk hook cursor"}]}}' > "$HOME/.cursor/hooks.json" ;;
|
|
*) echo "unknown: $*" >&2; exit 2 ;;
|
|
esac
|
|
`
|
|
|
|
func TestRTK(t *testing.T) {
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("the fake rtk is a shell script")
|
|
}
|
|
h := sandbox(t)
|
|
bin := filepath.Join(h, "bin")
|
|
write(t, filepath.Join(bin, "rtk"), fakeRTK)
|
|
if err := os.Chmod(filepath.Join(bin, "rtk"), 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("PATH", bin)
|
|
state := func() map[string]bool {
|
|
m := map[string]bool{}
|
|
for _, a := range ReadRTK().Agents {
|
|
m[a.ID] = a.On
|
|
}
|
|
return m
|
|
}
|
|
want := func(claude, opencode, cursor bool) {
|
|
t.Helper()
|
|
s := state()
|
|
if s["claude"] != claude || s["opencode"] != opencode || s["cursor"] != cursor {
|
|
t.Fatalf("claude, opencode, cursor = %v, %v, %v; want %v, %v, %v", s["claude"], s["opencode"], s["cursor"], claude, opencode, cursor)
|
|
}
|
|
}
|
|
set := func(id string, on bool) {
|
|
t.Helper()
|
|
if _, err := SetRTK(id, on); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
v := ReadRTK()
|
|
if v.Version != "9.9.9" || v.Gain == nil || v.Gain.Saved != 60 {
|
|
t.Fatalf("version %q, gain %+v", v.Version, v.Gain)
|
|
}
|
|
want(false, false, false)
|
|
// OpenCode's and Cursor's installers leave Claude Code as it was
|
|
set("opencode", true)
|
|
set("cursor", true)
|
|
want(false, true, true)
|
|
set("claude", true)
|
|
want(true, true, true)
|
|
// taking it out of Claude Code leaves the other two as they are
|
|
set("claude", false)
|
|
want(false, true, true)
|
|
set("claude", true)
|
|
set("opencode", false)
|
|
want(true, false, true)
|
|
set("cursor", false)
|
|
want(true, false, false)
|
|
|
|
if _, err := SetRTK("goose", true); err == nil {
|
|
t.Fatal("goose has no rtk hook, yet it was switched on")
|
|
}
|
|
t.Setenv("PATH", "")
|
|
if v := ReadRTK(); v.Path != "" {
|
|
t.Fatalf("rtk found at %s with nothing on PATH", v.Path)
|
|
}
|
|
if _, err := SetRTK("opencode", true); err == nil {
|
|
t.Fatal("switched on without rtk installed")
|
|
}
|
|
// the hook of an rtk since removed can still be taken out
|
|
set("claude", false)
|
|
want(false, false, false)
|
|
}
|
|
|
|
// TestRTKRemove: switching an agent off takes out just what rtk's installer
|
|
// put in, as rtk 0.50 writes it, with no rtk to do it.
|
|
func TestRTKRemove(t *testing.T) {
|
|
h := sandbox(t)
|
|
hook := func(key, matcher, cmd string) string {
|
|
return `"hooks": {
|
|
"` + key + `": [
|
|
{
|
|
"matcher": "` + matcher + `",
|
|
"hooks": [
|
|
{
|
|
"type": "command",
|
|
"command": "` + cmd + `"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}`
|
|
}
|
|
claude := filepath.Join(h, ".claude")
|
|
write(t, filepath.Join(claude, "settings.json"), `{
|
|
"model": "opus",
|
|
"hooks": {
|
|
"PreToolUse": [
|
|
{
|
|
"matcher": "Write",
|
|
"hooks": [{"type": "command", "command": "lint"}]
|
|
},
|
|
{
|
|
"matcher": "Bash",
|
|
"hooks": [
|
|
{
|
|
"type": "command",
|
|
"command": "rtk hook claude"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
}
|
|
`)
|
|
write(t, filepath.Join(claude, "CLAUDE.md"), "# mine\n\n@RTK.md\n")
|
|
write(t, filepath.Join(claude, "RTK.md"), "# RTK\n")
|
|
codex := filepath.Join(h, ".codex")
|
|
write(t, filepath.Join(codex, "hooks.json"), "{\n "+hook("PreToolUse", "Bash", "rtk hook codex")+"\n}\n")
|
|
write(t, filepath.Join(codex, "AGENTS.md"), "@"+filepath.Join(codex, "RTK.md")+"\n")
|
|
write(t, filepath.Join(codex, "RTK.md"), "# RTK\n")
|
|
gemini := filepath.Join(h, ".gemini")
|
|
sh := filepath.Join(gemini, "hooks", "rtk-hook-gemini.sh")
|
|
write(t, filepath.Join(gemini, "settings.json"), "{\n \"theme\": \"dark\",\n "+hook("BeforeTool", "run_shell_command", sh)+"\n}\n")
|
|
write(t, sh, "#!/bin/sh\n")
|
|
write(t, filepath.Join(gemini, "hooks", ".rtk-hook.sha256"), "x\n")
|
|
write(t, filepath.Join(gemini, "GEMINI.md"), "# g\n")
|
|
write(t, filepath.Join(h, ".cursor", "hooks.json"), `{"version":1,"hooks":{"preToolUse":[{"command":"rtk hook cursor","matcher":"Shell"}]}}`)
|
|
copilot := filepath.Join(h, ".copilot")
|
|
write(t, filepath.Join(copilot, "hooks", "rtk-rewrite.json"), "{}")
|
|
write(t, filepath.Join(copilot, "copilot-instructions.md"), "# c\n\n<!-- rtk-instructions v2 -->\nAlways prefix shell commands with rtk\n<!-- /rtk-instructions -->\n")
|
|
write(t, filepath.Join(h, ".pi", "agent", "extensions", "rtk.ts"), "x")
|
|
write(t, filepath.Join(h, ".config", "opencode", "plugins", "rtk.ts"), "x")
|
|
|
|
ids := []string{"claude", "codex", "gemini", "cursor", "copilot", "pi", "opencode"}
|
|
v := ReadRTK()
|
|
if v.Path != "" {
|
|
t.Fatalf("rtk found at %s", v.Path)
|
|
}
|
|
on := map[string]bool{}
|
|
for _, a := range v.Agents {
|
|
on[a.ID] = a.On
|
|
}
|
|
for _, id := range ids {
|
|
if !on[id] {
|
|
t.Fatalf("%s's hook isn't seen: %v", id, on)
|
|
}
|
|
if _, err := SetRTK(id, false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
for _, a := range ReadRTK().Agents {
|
|
if a.On {
|
|
t.Fatalf("%s still has rtk", a.ID)
|
|
}
|
|
}
|
|
jsonIs := func(p, want string) {
|
|
t.Helper()
|
|
var got, w any
|
|
if err := json.Unmarshal([]byte(read(t, p)), &got); err != nil {
|
|
t.Fatalf("%s: %v\n%s", p, err, read(t, p))
|
|
}
|
|
json.Unmarshal([]byte(want), &w)
|
|
if !reflect.DeepEqual(got, w) {
|
|
t.Fatalf("%s:\n%s\nwant %s", p, read(t, p), want)
|
|
}
|
|
}
|
|
jsonIs(filepath.Join(claude, "settings.json"), `{"model":"opus","hooks":{"PreToolUse":[{"matcher":"Write","hooks":[{"type":"command","command":"lint"}]}]}}`)
|
|
jsonIs(filepath.Join(codex, "hooks.json"), `{}`)
|
|
jsonIs(filepath.Join(gemini, "settings.json"), `{"theme":"dark"}`)
|
|
jsonIs(filepath.Join(h, ".cursor", "hooks.json"), `{"version":1}`)
|
|
for p, want := range map[string]string{
|
|
filepath.Join(claude, "CLAUDE.md"): "# mine\n",
|
|
filepath.Join(copilot, "copilot-instructions.md"): "# c\n",
|
|
filepath.Join(gemini, "GEMINI.md"): "# g\n",
|
|
filepath.Join(claude, "RTK.md"): "",
|
|
filepath.Join(codex, "AGENTS.md"): "",
|
|
filepath.Join(codex, "RTK.md"): "",
|
|
sh: "",
|
|
filepath.Join(copilot, "hooks", "rtk-rewrite.json"): "",
|
|
} {
|
|
if got := read(t, p); got != want {
|
|
t.Errorf("%s = %q, want %q", p, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDropHermesPlugin(t *testing.T) {
|
|
p := filepath.Join(t.TempDir(), "config.yaml")
|
|
for in, want := range map[string]string{
|
|
"model: x\nplugins:\n enabled:\n - rtk-rewrite\n": "model: x\n",
|
|
"model: x\nplugins:\n enabled:\n - a\n - rtk-rewrite\n": "model: x\nplugins:\n enabled:\n - a\n",
|
|
} {
|
|
write(t, p, in)
|
|
if err := dropHermesPlugin(p); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := read(t, p); got != want {
|
|
t.Errorf("%q → %q, want %q", in, got, want)
|
|
}
|
|
}
|
|
}
|