214 lines
7.4 KiB
Go
214 lines
7.4 KiB
Go
package library
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"slices"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func isLink(t *testing.T, p string) bool {
|
|
t.Helper()
|
|
fi, err := os.Lstat(p)
|
|
return err == nil && fi.Mode()&os.ModeSymlink != 0
|
|
}
|
|
|
|
func gone(t *testing.T, p string) {
|
|
t.Helper()
|
|
if _, err := os.Lstat(p); !os.IsNotExist(err) {
|
|
t.Errorf("%s is still there", p)
|
|
}
|
|
}
|
|
|
|
// projectLib is a sandbox with a library skill pdf (from a folder) and a
|
|
// project folder with a .gitignore of the user's.
|
|
func projectLib(t *testing.T) (h, proj string) {
|
|
h = sandbox(t)
|
|
src := filepath.Join(h, "src")
|
|
skill(t, filepath.Join(src, "pdf"), "pdf", "Read PDFs")
|
|
skill(t, filepath.Join(src, "docx"), "docx", "Word")
|
|
ok(t)(InstallSkills(src, []string{"pdf", "docx"}, nil))
|
|
proj = filepath.Join(t.TempDir(), "app")
|
|
write(t, filepath.Join(proj, ".gitignore"), "node_modules/\n")
|
|
ok(t)(AddProject(proj))
|
|
return h, proj
|
|
}
|
|
|
|
func TestProjectSkillsLinked(t *testing.T) {
|
|
_, proj := projectLib(t)
|
|
ok(t)(ProjectSkill(proj, "pdf", []string{"claude", "codex", "gemini"}))
|
|
for _, e := range []string{".claude/skills/pdf", ".agents/skills/pdf"} {
|
|
p := filepath.Join(proj, e)
|
|
if !isLink(t, p) || !ours(p, "pdf") {
|
|
t.Errorf("%s isn't a link to the library's", e)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(p, "SKILL.md")); err != nil {
|
|
t.Errorf("%s: %v", e, err)
|
|
}
|
|
}
|
|
want := "node_modules/\n\n" + ignoreBegin + "\n/.agents/skills/pdf\n/.claude/skills/pdf\n" + ignoreEnd + "\n"
|
|
if g := read(t, filepath.Join(proj, ".gitignore")); g != want {
|
|
t.Errorf(".gitignore:\n%s", g)
|
|
}
|
|
v, _ := Read(nil)
|
|
if len(v.Projects) != 1 || !slices.Equal(v.Projects[0].Skills["pdf"], []string{"claude", "codex", "gemini"}) {
|
|
t.Errorf("projects: %+v", v.Projects)
|
|
}
|
|
|
|
// one agent of two sharing .agents/skills taken away: it stays for the other
|
|
ok(t)(ProjectSkill(proj, "pdf", []string{"claude", "codex"}))
|
|
if !isLink(t, filepath.Join(proj, ".agents/skills/pdf")) {
|
|
t.Error(".agents/skills/pdf went while codex still has it")
|
|
}
|
|
ok(t)(ProjectSkill(proj, "pdf", []string{"codex"}))
|
|
gone(t, filepath.Join(proj, ".claude/skills/pdf"))
|
|
gone(t, filepath.Join(proj, ".claude")) // magpie made it, and it's empty
|
|
if g := read(t, filepath.Join(proj, ".gitignore")); strings.Contains(g, ".claude") || !strings.Contains(g, "/.agents/skills/pdf") {
|
|
t.Errorf(".gitignore:\n%s", g)
|
|
}
|
|
|
|
// removing the project takes away all it placed, and its lines
|
|
ok(t)(RemoveProject(proj))
|
|
gone(t, filepath.Join(proj, ".agents"))
|
|
if g := read(t, filepath.Join(proj, ".gitignore")); g != "node_modules/\n" {
|
|
t.Errorf(".gitignore after: %q", g)
|
|
}
|
|
if v, _ := Read(nil); len(v.Projects) != 0 {
|
|
t.Errorf("projects: %+v", v.Projects)
|
|
}
|
|
}
|
|
|
|
func TestProjectGitignoreMadeAndTakenAway(t *testing.T) {
|
|
_, proj := projectLib(t)
|
|
os.Remove(filepath.Join(proj, ".gitignore"))
|
|
ok(t)(ProjectSkill(proj, "docx", []string{"claude"}))
|
|
if g := read(t, filepath.Join(proj, ".gitignore")); g != ignoreBegin+"\n/.claude/skills/docx\n"+ignoreEnd+"\n" {
|
|
t.Errorf(".gitignore: %q", g)
|
|
}
|
|
ok(t)(ProjectSkill(proj, "docx", nil))
|
|
gone(t, filepath.Join(proj, ".gitignore"))
|
|
gone(t, filepath.Join(proj, ".claude/skills/docx"))
|
|
}
|
|
|
|
func TestProjectSkillsCopied(t *testing.T) {
|
|
h, proj := projectLib(t)
|
|
ok(t)(ProjectCopy(proj, true))
|
|
ok(t)(ProjectSkill(proj, "pdf", []string{"claude"}))
|
|
p := filepath.Join(proj, ".claude/skills/pdf")
|
|
if isLink(t, p) || !ours(p, "pdf") {
|
|
t.Fatal("not magpie's copy")
|
|
}
|
|
if s := read(t, filepath.Join(p, "scripts/run.sh")); s != "echo hi\n" {
|
|
t.Errorf("run.sh: %q", s)
|
|
}
|
|
// the library's changes: the next change to the library copies it again
|
|
write(t, filepath.Join(h, "src/pdf/SKILL.md"), "---\nname: pdf\ndescription: Changed\n---\n")
|
|
ok(t)(Sync())
|
|
if !strings.Contains(read(t, filepath.Join(p, "SKILL.md")), "Changed") {
|
|
t.Error("the copy wasn't made again")
|
|
}
|
|
// back to links: the copy becomes a link
|
|
ok(t)(ProjectCopy(proj, false))
|
|
if !isLink(t, p) {
|
|
t.Error("still a copy")
|
|
}
|
|
}
|
|
|
|
func TestProjectRefusesForeignFolder(t *testing.T) {
|
|
_, proj := projectLib(t)
|
|
skill(t, filepath.Join(proj, ".claude/skills/pdf"), "pdf", "The project's own")
|
|
r, err := ProjectSkill(proj, "pdf", []string{"claude", "codex"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(r.Problems) != 1 || r.Problems[0].Agent != proj || !strings.Contains(r.Problems[0].Error, "isn't magpie's") {
|
|
t.Fatalf("problems: %+v", r.Problems)
|
|
}
|
|
if !strings.Contains(read(t, filepath.Join(proj, ".claude/skills/pdf/SKILL.md")), "The project's own") {
|
|
t.Error("the project's own was overwritten")
|
|
}
|
|
if !isLink(t, filepath.Join(proj, ".agents/skills/pdf")) {
|
|
t.Error("codex's went with it")
|
|
}
|
|
if g := read(t, filepath.Join(proj, ".gitignore")); strings.Contains(g, ".claude/skills/pdf") {
|
|
t.Errorf("the project's own is in .gitignore:\n%s", g)
|
|
}
|
|
// removing the project leaves the folder that isn't magpie's
|
|
ok(t)(ProjectSkill(proj, "pdf", nil))
|
|
ok(t)(RemoveProject(proj))
|
|
if _, err := os.Stat(filepath.Join(proj, ".claude/skills/pdf/SKILL.md")); err != nil {
|
|
t.Error("the project's own went")
|
|
}
|
|
}
|
|
|
|
func TestProjectRefusesHomeAndUnknownAgents(t *testing.T) {
|
|
h, proj := projectLib(t)
|
|
if _, err := AddProject(h); err == nil {
|
|
t.Error("home added as a project")
|
|
}
|
|
if _, err := AddProject(proj); err == nil {
|
|
t.Error("added twice")
|
|
}
|
|
if _, err := ProjectSkill(proj, "pdf", []string{"goose"}); err == nil {
|
|
t.Error("goose given a project skill")
|
|
}
|
|
}
|
|
|
|
func TestRemoveSkillLeavesProjects(t *testing.T) {
|
|
_, proj := projectLib(t)
|
|
ok(t)(ProjectSkill(proj, "pdf", []string{"claude"}))
|
|
ok(t)(RemoveSkill("pdf"))
|
|
gone(t, filepath.Join(proj, ".claude/skills/pdf"))
|
|
if v, _ := Read(nil); len(v.Projects[0].Skills) != 0 {
|
|
t.Errorf("skills: %+v", v.Projects[0].Skills)
|
|
}
|
|
}
|
|
|
|
func TestUpdateSkillRefreshesProjects(t *testing.T) {
|
|
sandbox(t)
|
|
version := "one"
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write(tarball(t, map[string]string{
|
|
"skills/pdf/SKILL.md": "---\nname: pdf\ndescription: PDFs " + version + "\n---\n",
|
|
"skills/docx/SKILL.md": "---\nname: docx\ndescription: Word " + version + "\n---\n",
|
|
}))
|
|
}))
|
|
defer srv.Close()
|
|
old := tarballURL
|
|
tarballURL = func(repo, ref string) string { return srv.URL + "/" + repo + "/" + ref }
|
|
defer func() { tarballURL = old }()
|
|
|
|
ok(t)(InstallSkills("owner/repo", []string{"skills/pdf", "skills/docx"}, nil))
|
|
linked, copied := filepath.Join(t.TempDir(), "linked"), filepath.Join(t.TempDir(), "copied")
|
|
os.MkdirAll(linked, 0o755)
|
|
os.MkdirAll(copied, 0o755)
|
|
ok(t)(AddProject(linked))
|
|
ok(t)(AddProject(copied))
|
|
ok(t)(ProjectCopy(copied, true))
|
|
for _, p := range []string{linked, copied} {
|
|
ok(t)(ProjectSkill(p, "pdf", []string{"claude"}))
|
|
ok(t)(ProjectSkill(p, "docx", []string{"codex"}))
|
|
}
|
|
version = "two"
|
|
ok(t)(UpdateSkill("pdf"))
|
|
ok(t)(UpdateSomeSkills([]string{"docx"}))
|
|
for _, p := range []string{linked, copied} {
|
|
for _, e := range []string{".claude/skills/pdf", ".agents/skills/docx"} {
|
|
if s := read(t, filepath.Join(p, e, "SKILL.md")); !strings.Contains(s, " two") {
|
|
t.Errorf("%s/%s after the update: %q", filepath.Base(p), e, s)
|
|
}
|
|
}
|
|
}
|
|
if !isLink(t, filepath.Join(linked, ".claude/skills/pdf")) || isLink(t, filepath.Join(copied, ".claude/skills/pdf")) {
|
|
t.Error("links and copies changed places")
|
|
}
|
|
version = "three"
|
|
ok(t)(UpdateSkills())
|
|
if s := read(t, filepath.Join(copied, ".agents/skills/docx/SKILL.md")); !strings.Contains(s, "three") {
|
|
t.Errorf("copy after update all: %q", s)
|
|
}
|
|
}
|