import { mkdirSync, rmSync, writeFileSync } from "node:fs"; import { tmpdir } from "node:os"; import { join } from "node:path"; import { afterEach, beforeEach, describe, expect, it } from "vitest"; import { DefaultPackageManager, type ProgressEvent, type ResolvedResource } from "../src/core/package-manager.js"; import { SettingsManager } from "../src/core/settings-manager.js"; // Helper to check if a resource is enabled const isEnabled = (r: ResolvedResource, pathMatch: string, matchFn: "endsWith" | "includes" = "endsWith") => matchFn === "endsWith" ? r.path.endsWith(pathMatch) && r.enabled : r.path.includes(pathMatch) && r.enabled; const isDisabled = (r: ResolvedResource, pathMatch: string, matchFn: "endsWith" | "includes" = "endsWith") => matchFn === "endsWith" ? r.path.endsWith(pathMatch) && !r.enabled : r.path.includes(pathMatch) && !r.enabled; describe("DefaultPackageManager", () => { let tempDir: string; let agentDir: string; let settingsManager: SettingsManager; let packageManager: DefaultPackageManager; beforeEach(() => { tempDir = join(tmpdir(), `pm-test-${Date.now()}-${Math.random().toString(36).slice(2)}`); mkdirSync(tempDir, { recursive: true }); agentDir = join(tempDir, "agent"); mkdirSync(agentDir, { recursive: true }); settingsManager = SettingsManager.inMemory(); packageManager = new DefaultPackageManager({ cwd: tempDir, agentDir, settingsManager, }); }); afterEach(() => { rmSync(tempDir, { recursive: true, force: true }); }); describe("resolve", () => { it("should return empty paths when no sources configured", async () => { const result = await packageManager.resolve(); expect(result.extensions).toEqual([]); expect(result.skills).toEqual([]); expect(result.prompts).toEqual([]); expect(result.themes).toEqual([]); }); it("should resolve local extension paths from settings", async () => { const extDir = join(agentDir, "extensions"); mkdirSync(extDir, { recursive: true }); const extPath = join(extDir, "my-extension.ts"); writeFileSync(extPath, "export default function() {}"); settingsManager.setExtensionPaths(["extensions/my-extension.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => r.path === extPath && r.enabled)).toBe(true); }); it("should resolve skill paths from settings", async () => { const skillDir = join(agentDir, "skills", "my-skill"); mkdirSync(skillDir, { recursive: true }); const skillFile = join(skillDir, "SKILL.md"); writeFileSync( skillFile, `--- name: test-skill description: A test skill --- Content`, ); settingsManager.setSkillPaths(["skills"]); const result = await packageManager.resolve(); // Skills with SKILL.md are returned as file paths expect(result.skills.some((r) => r.path === skillFile && r.enabled)).toBe(true); }); it("should resolve project paths relative to .pi", async () => { const extDir = join(tempDir, ".pi", "extensions"); mkdirSync(extDir, { recursive: true }); const extPath = join(extDir, "project-ext.ts"); writeFileSync(extPath, "export default function() {}"); settingsManager.setProjectExtensionPaths(["extensions/project-ext.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => r.path === extPath && r.enabled)).toBe(true); }); it("should auto-discover user prompts with overrides", async () => { const promptsDir = join(agentDir, "prompts"); mkdirSync(promptsDir, { recursive: true }); const promptPath = join(promptsDir, "auto.md"); writeFileSync(promptPath, "Auto prompt"); settingsManager.setPromptTemplatePaths(["!prompts/auto.md"]); const result = await packageManager.resolve(); expect(result.prompts.some((r) => r.path === promptPath && !r.enabled)).toBe(true); }); it("should auto-discover project prompts with overrides", async () => { const promptsDir = join(tempDir, ".pi", "prompts"); mkdirSync(promptsDir, { recursive: true }); const promptPath = join(promptsDir, "is.md"); writeFileSync(promptPath, "Is prompt"); settingsManager.setProjectPromptTemplatePaths(["!prompts/is.md"]); const result = await packageManager.resolve(); expect(result.prompts.some((r) => r.path === promptPath && !r.enabled)).toBe(true); }); }); describe("ignore files", () => { it("should respect .gitignore in skill directories", async () => { const skillsDir = join(agentDir, "skills"); mkdirSync(skillsDir, { recursive: true }); writeFileSync(join(skillsDir, ".gitignore"), "venv\n__pycache__\n"); const goodSkillDir = join(skillsDir, "good-skill"); mkdirSync(goodSkillDir, { recursive: true }); writeFileSync(join(goodSkillDir, "SKILL.md"), "---\nname: good-skill\ndescription: Good\n---\nContent"); const ignoredSkillDir = join(skillsDir, "venv", "bad-skill"); mkdirSync(ignoredSkillDir, { recursive: true }); writeFileSync(join(ignoredSkillDir, "SKILL.md"), "---\nname: bad-skill\ndescription: Bad\n---\nContent"); settingsManager.setSkillPaths(["skills"]); const result = await packageManager.resolve(); expect(result.skills.some((r) => r.path.includes("good-skill") && r.enabled)).toBe(true); expect(result.skills.some((r) => r.path.includes("venv") && r.enabled)).toBe(false); }); it("should not apply parent .gitignore to .pi auto-discovery", async () => { writeFileSync(join(tempDir, ".gitignore"), ".pi\n"); const skillDir = join(tempDir, ".pi", "skills", "auto-skill"); mkdirSync(skillDir, { recursive: true }); const skillPath = join(skillDir, "SKILL.md"); writeFileSync(skillPath, "---\nname: auto-skill\ndescription: Auto\n---\nContent"); const result = await packageManager.resolve(); expect(result.skills.some((r) => r.path === skillPath && r.enabled)).toBe(true); }); }); describe("resolveExtensionSources", () => { it("should resolve local paths", async () => { const extPath = join(tempDir, "ext.ts"); writeFileSync(extPath, "export default function() {}"); const result = await packageManager.resolveExtensionSources([extPath]); expect(result.extensions.some((r) => r.path === extPath && r.enabled)).toBe(true); }); it("should handle directories with pi manifest", async () => { const pkgDir = join(tempDir, "my-package"); mkdirSync(pkgDir, { recursive: true }); writeFileSync( join(pkgDir, "package.json"), JSON.stringify({ name: "my-package", pi: { extensions: ["./src/index.ts"], skills: ["./skills"], }, }), ); mkdirSync(join(pkgDir, "src"), { recursive: true }); writeFileSync(join(pkgDir, "src", "index.ts"), "export default function() {}"); mkdirSync(join(pkgDir, "skills", "my-skill"), { recursive: true }); writeFileSync( join(pkgDir, "skills", "my-skill", "SKILL.md"), "---\nname: my-skill\ndescription: Test\n---\nContent", ); const result = await packageManager.resolveExtensionSources([pkgDir]); expect(result.extensions.some((r) => r.path === join(pkgDir, "src", "index.ts") && r.enabled)).toBe(true); // Skills with SKILL.md are returned as file paths expect(result.skills.some((r) => r.path === join(pkgDir, "skills", "my-skill", "SKILL.md") && r.enabled)).toBe( true, ); }); it("should handle directories with auto-discovery layout", async () => { const pkgDir = join(tempDir, "auto-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); mkdirSync(join(pkgDir, "themes"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "main.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "themes", "dark.json"), "{}"); const result = await packageManager.resolveExtensionSources([pkgDir]); expect(result.extensions.some((r) => r.path.endsWith("main.ts") && r.enabled)).toBe(true); expect(result.themes.some((r) => r.path.endsWith("dark.json") && r.enabled)).toBe(true); }); }); describe("progress callback", () => { it("should emit progress events", async () => { const events: ProgressEvent[] = []; packageManager.setProgressCallback((event) => events.push(event)); const extPath = join(tempDir, "ext.ts"); writeFileSync(extPath, "export default function() {}"); // Local paths don't trigger install progress, but we can verify the callback is set await packageManager.resolveExtensionSources([extPath]); // For now just verify no errors - npm/git would trigger actual events expect(events.length).toBe(0); }); }); describe("source parsing", () => { it("should emit progress events on install attempt", async () => { const events: ProgressEvent[] = []; packageManager.setProgressCallback((event) => events.push(event)); // Use public install method which emits progress events try { await packageManager.install("npm:nonexistent-package@1.0.0"); } catch { // Expected to fail - package doesn't exist } // Should have emitted start event before failure expect(events.some((e) => e.type === "start" && e.action === "install")).toBe(true); // Should have emitted error event expect(events.some((e) => e.type === "error")).toBe(true); }); it("should recognize github URLs without git: prefix", async () => { const events: ProgressEvent[] = []; packageManager.setProgressCallback((event) => events.push(event)); // This should be parsed as a git source, not throw "unsupported" try { await packageManager.install("https://github.com/nonexistent/repo"); } catch { // Expected to fail - repo doesn't exist } // Should have attempted clone, not thrown unsupported error expect(events.some((e) => e.type === "start" && e.action === "install")).toBe(true); }); }); describe("HTTPS git URL parsing (old behavior)", () => { it("should parse HTTPS GitHub URLs correctly", async () => { const parsed = (packageManager as any).parseSource("https://github.com/user/repo"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("github.com"); expect(parsed.path).toBe("user/repo"); expect(parsed.pinned).toBe(false); }); it("should parse HTTPS URLs with git: prefix", async () => { const parsed = (packageManager as any).parseSource("git:https://github.com/user/repo"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("github.com"); expect(parsed.path).toBe("user/repo"); }); it("should parse HTTPS URLs with ref", async () => { const parsed = (packageManager as any).parseSource("https://github.com/user/repo@v1.2.3"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("github.com"); expect(parsed.path).toBe("user/repo"); expect(parsed.ref).toBe("v1.2.3"); expect(parsed.pinned).toBe(true); }); it("should parse HTTPS URLs without protocol", async () => { const parsed = (packageManager as any).parseSource("github.com/user/repo"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("github.com"); expect(parsed.path).toBe("user/repo"); }); it("should parse HTTPS URLs with .git suffix", async () => { const parsed = (packageManager as any).parseSource("https://github.com/user/repo.git"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("github.com"); expect(parsed.path).toBe("user/repo"); }); it("should parse GitLab HTTPS URLs", async () => { const parsed = (packageManager as any).parseSource("https://gitlab.com/user/repo"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("gitlab.com"); expect(parsed.path).toBe("user/repo"); }); it("should parse Bitbucket HTTPS URLs", async () => { const parsed = (packageManager as any).parseSource("https://bitbucket.org/user/repo"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("bitbucket.org"); expect(parsed.path).toBe("user/repo"); }); it("should parse Codeberg HTTPS URLs", async () => { const parsed = (packageManager as any).parseSource("https://codeberg.org/user/repo"); expect(parsed.type).toBe("git"); expect(parsed.host).toBe("codeberg.org"); expect(parsed.path).toBe("user/repo"); }); it("should generate correct package identity for HTTPS URLs", async () => { const identity1 = (packageManager as any).getPackageIdentity("https://github.com/user/repo"); const identity2 = (packageManager as any).getPackageIdentity("https://github.com/user/repo@v1.0.0"); const identity3 = (packageManager as any).getPackageIdentity("github.com/user/repo"); const identity4 = (packageManager as any).getPackageIdentity("https://github.com/user/repo.git"); // All should have the same identity (normalized) expect(identity1).toBe("git:github.com/user/repo"); expect(identity2).toBe("git:github.com/user/repo"); expect(identity3).toBe("git:github.com/user/repo"); expect(identity4).toBe("git:github.com/user/repo"); }); it("should deduplicate HTTPS URLs with different formats", async () => { const pkgDir = join(tempDir, "https-dedup-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "test.ts"), "export default function() {}"); // Mock the package as if it were cloned from different URL formats // In reality, these would all point to the same local dir after install settingsManager.setPackages([ "https://github.com/user/repo", "github.com/user/repo", "https://github.com/user/repo.git", ]); // Since these URLs don't actually exist and we can't clone them, // we verify they produce the same identity const id1 = (packageManager as any).getPackageIdentity("https://github.com/user/repo"); const id2 = (packageManager as any).getPackageIdentity("github.com/user/repo"); const id3 = (packageManager as any).getPackageIdentity("https://github.com/user/repo.git"); expect(id1).toBe(id2); expect(id2).toBe(id3); }); it("should handle HTTPS URLs with refs in resolve", async () => { // This tests that the ref is properly extracted and stored const parsed = (packageManager as any).parseSource("https://github.com/user/repo@main"); expect(parsed.ref).toBe("main"); expect(parsed.pinned).toBe(true); const parsed2 = (packageManager as any).parseSource("https://github.com/user/repo@feature/branch"); expect(parsed2.ref).toBe("feature/branch"); }); }); describe("pattern filtering in top-level arrays", () => { it("should exclude extensions with ! pattern", async () => { const extDir = join(agentDir, "extensions"); mkdirSync(extDir, { recursive: true }); writeFileSync(join(extDir, "keep.ts"), "export default function() {}"); writeFileSync(join(extDir, "remove.ts"), "export default function() {}"); settingsManager.setExtensionPaths(["extensions", "!**/remove.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isEnabled(r, "keep.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "remove.ts"))).toBe(true); }); it("should filter themes with glob patterns", async () => { const themesDir = join(agentDir, "themes"); mkdirSync(themesDir, { recursive: true }); writeFileSync(join(themesDir, "dark.json"), "{}"); writeFileSync(join(themesDir, "light.json"), "{}"); writeFileSync(join(themesDir, "funky.json"), "{}"); settingsManager.setThemePaths(["themes", "!funky.json"]); const result = await packageManager.resolve(); expect(result.themes.some((r) => isEnabled(r, "dark.json"))).toBe(true); expect(result.themes.some((r) => isEnabled(r, "light.json"))).toBe(true); expect(result.themes.some((r) => isDisabled(r, "funky.json"))).toBe(true); }); it("should filter prompts with exclusion pattern", async () => { const promptsDir = join(agentDir, "prompts"); mkdirSync(promptsDir, { recursive: true }); writeFileSync(join(promptsDir, "review.md"), "Review code"); writeFileSync(join(promptsDir, "explain.md"), "Explain code"); settingsManager.setPromptTemplatePaths(["prompts", "!explain.md"]); const result = await packageManager.resolve(); expect(result.prompts.some((r) => isEnabled(r, "review.md"))).toBe(true); expect(result.prompts.some((r) => isDisabled(r, "explain.md"))).toBe(true); }); it("should filter skills with exclusion pattern", async () => { const skillsDir = join(agentDir, "skills"); mkdirSync(join(skillsDir, "good-skill"), { recursive: true }); mkdirSync(join(skillsDir, "bad-skill"), { recursive: true }); writeFileSync( join(skillsDir, "good-skill", "SKILL.md"), "---\nname: good-skill\ndescription: Good\n---\nContent", ); writeFileSync( join(skillsDir, "bad-skill", "SKILL.md"), "---\nname: bad-skill\ndescription: Bad\n---\nContent", ); settingsManager.setSkillPaths(["skills", "!**/bad-skill"]); const result = await packageManager.resolve(); expect(result.skills.some((r) => isEnabled(r, "good-skill", "includes"))).toBe(true); expect(result.skills.some((r) => isDisabled(r, "bad-skill", "includes"))).toBe(true); }); it("should work without patterns (backward compatible)", async () => { const extDir = join(agentDir, "extensions"); mkdirSync(extDir, { recursive: true }); const extPath = join(extDir, "my-ext.ts"); writeFileSync(extPath, "export default function() {}"); settingsManager.setExtensionPaths(["extensions/my-ext.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => r.path === extPath && r.enabled)).toBe(true); }); }); describe("pattern filtering in pi manifest", () => { it("should support glob patterns in manifest extensions", async () => { const pkgDir = join(tempDir, "manifest-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); mkdirSync(join(pkgDir, "node_modules/dep/extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "local.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "node_modules/dep/extensions", "remote.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "node_modules/dep/extensions", "skip.ts"), "export default function() {}"); writeFileSync( join(pkgDir, "package.json"), JSON.stringify({ name: "manifest-pkg", pi: { extensions: ["extensions", "node_modules/dep/extensions", "!**/skip.ts"], }, }), ); const result = await packageManager.resolveExtensionSources([pkgDir]); expect(result.extensions.some((r) => isEnabled(r, "local.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "remote.ts"))).toBe(true); expect(result.extensions.some((r) => r.path.endsWith("skip.ts"))).toBe(false); }); it("should support glob patterns in manifest skills", async () => { const pkgDir = join(tempDir, "skill-manifest-pkg"); mkdirSync(join(pkgDir, "skills/good-skill"), { recursive: true }); mkdirSync(join(pkgDir, "skills/bad-skill"), { recursive: true }); writeFileSync( join(pkgDir, "skills/good-skill", "SKILL.md"), "---\nname: good-skill\ndescription: Good\n---\nContent", ); writeFileSync( join(pkgDir, "skills/bad-skill", "SKILL.md"), "---\nname: bad-skill\ndescription: Bad\n---\nContent", ); writeFileSync( join(pkgDir, "package.json"), JSON.stringify({ name: "skill-manifest-pkg", pi: { skills: ["skills", "!**/bad-skill"], }, }), ); const result = await packageManager.resolveExtensionSources([pkgDir]); expect(result.skills.some((r) => isEnabled(r, "good-skill", "includes"))).toBe(true); expect(result.skills.some((r) => r.path.includes("bad-skill"))).toBe(false); }); }); describe("pattern filtering in package filters", () => { it("should apply user filters on top of manifest filters (not replace)", async () => { // Manifest excludes baz.ts, user excludes bar.ts // Result should exclude BOTH const pkgDir = join(tempDir, "layered-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "foo.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "bar.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "baz.ts"), "export default function() {}"); writeFileSync( join(pkgDir, "package.json"), JSON.stringify({ name: "layered-pkg", pi: { extensions: ["extensions", "!**/baz.ts"], }, }), ); // User filter adds exclusion for bar.ts settingsManager.setPackages([ { source: pkgDir, extensions: ["!**/bar.ts"], skills: [], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); // foo.ts should be included (not excluded by anyone) expect(result.extensions.some((r) => isEnabled(r, "foo.ts"))).toBe(true); // bar.ts should be excluded (by user) expect(result.extensions.some((r) => isDisabled(r, "bar.ts"))).toBe(true); // baz.ts should be excluded (by manifest) expect(result.extensions.some((r) => r.path.endsWith("baz.ts"))).toBe(false); }); it("should exclude extensions from package with ! pattern", async () => { const pkgDir = join(tempDir, "pattern-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "foo.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "bar.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "baz.ts"), "export default function() {}"); settingsManager.setPackages([ { source: pkgDir, extensions: ["!**/baz.ts"], skills: [], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isEnabled(r, "foo.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "bar.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "baz.ts"))).toBe(true); }); it("should filter themes from package", async () => { const pkgDir = join(tempDir, "theme-pkg"); mkdirSync(join(pkgDir, "themes"), { recursive: true }); writeFileSync(join(pkgDir, "themes", "nice.json"), "{}"); writeFileSync(join(pkgDir, "themes", "ugly.json"), "{}"); settingsManager.setPackages([ { source: pkgDir, extensions: [], skills: [], prompts: [], themes: ["!ugly.json"], }, ]); const result = await packageManager.resolve(); expect(result.themes.some((r) => isEnabled(r, "nice.json"))).toBe(true); expect(result.themes.some((r) => isDisabled(r, "ugly.json"))).toBe(true); }); it("should combine include and exclude patterns", async () => { const pkgDir = join(tempDir, "combo-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "alpha.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "beta.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "gamma.ts"), "export default function() {}"); settingsManager.setPackages([ { source: pkgDir, extensions: ["**/alpha.ts", "**/beta.ts", "!**/beta.ts"], skills: [], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isEnabled(r, "alpha.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "beta.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "gamma.ts"))).toBe(true); }); it("should work with direct paths (no patterns)", async () => { const pkgDir = join(tempDir, "direct-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "one.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "two.ts"), "export default function() {}"); settingsManager.setPackages([ { source: pkgDir, extensions: ["extensions/one.ts"], skills: [], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isEnabled(r, "one.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "two.ts"))).toBe(true); }); }); describe("force-include patterns", () => { it("should force-include extensions with + pattern after exclusion", async () => { const extDir = join(agentDir, "extensions"); mkdirSync(extDir, { recursive: true }); writeFileSync(join(extDir, "keep.ts"), "export default function() {}"); writeFileSync(join(extDir, "excluded.ts"), "export default function() {}"); writeFileSync(join(extDir, "force-back.ts"), "export default function() {}"); // Exclude all, then force-include one back settingsManager.setExtensionPaths(["extensions", "!extensions/*.ts", "+extensions/force-back.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isDisabled(r, "keep.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "excluded.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "force-back.ts"))).toBe(true); }); it("should force-include overrides exclude in package filters", async () => { const pkgDir = join(tempDir, "force-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "alpha.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "beta.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "gamma.ts"), "export default function() {}"); settingsManager.setPackages([ { source: pkgDir, extensions: ["!**/*.ts", "+extensions/beta.ts"], skills: [], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isDisabled(r, "alpha.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "beta.ts"))).toBe(true); expect(result.extensions.some((r) => isDisabled(r, "gamma.ts"))).toBe(true); }); it("should force-include multiple resources", async () => { const pkgDir = join(tempDir, "multi-force-pkg"); mkdirSync(join(pkgDir, "skills/skill-a"), { recursive: true }); mkdirSync(join(pkgDir, "skills/skill-b"), { recursive: true }); mkdirSync(join(pkgDir, "skills/skill-c"), { recursive: true }); writeFileSync(join(pkgDir, "skills/skill-a", "SKILL.md"), "---\nname: skill-a\ndescription: A\n---\nContent"); writeFileSync(join(pkgDir, "skills/skill-b", "SKILL.md"), "---\nname: skill-b\ndescription: B\n---\nContent"); writeFileSync(join(pkgDir, "skills/skill-c", "SKILL.md"), "---\nname: skill-c\ndescription: C\n---\nContent"); settingsManager.setPackages([ { source: pkgDir, extensions: [], skills: ["!**/*", "+skills/skill-a", "+skills/skill-c"], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); expect(result.skills.some((r) => isEnabled(r, "skill-a", "includes"))).toBe(true); expect(result.skills.some((r) => isDisabled(r, "skill-b", "includes"))).toBe(true); expect(result.skills.some((r) => isEnabled(r, "skill-c", "includes"))).toBe(true); }); it("should force-include after specific exclusion", async () => { const extDir = join(agentDir, "extensions"); mkdirSync(extDir, { recursive: true }); writeFileSync(join(extDir, "a.ts"), "export default function() {}"); writeFileSync(join(extDir, "b.ts"), "export default function() {}"); // Specifically exclude b.ts, then force it back settingsManager.setExtensionPaths(["extensions", "!extensions/b.ts", "+extensions/b.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isEnabled(r, "a.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "b.ts"))).toBe(true); }); it("should handle force-include in manifest patterns", async () => { const pkgDir = join(tempDir, "manifest-force-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "one.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "two.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "three.ts"), "export default function() {}"); writeFileSync( join(pkgDir, "package.json"), JSON.stringify({ name: "manifest-force-pkg", pi: { extensions: ["extensions", "!**/two.ts", "+extensions/two.ts"], }, }), ); const result = await packageManager.resolveExtensionSources([pkgDir]); expect(result.extensions.some((r) => isEnabled(r, "one.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "two.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "three.ts"))).toBe(true); }); it("should force-include themes", async () => { const themesDir = join(agentDir, "themes"); mkdirSync(themesDir, { recursive: true }); writeFileSync(join(themesDir, "dark.json"), "{}"); writeFileSync(join(themesDir, "light.json"), "{}"); writeFileSync(join(themesDir, "special.json"), "{}"); settingsManager.setThemePaths(["themes", "!themes/*.json", "+themes/special.json"]); const result = await packageManager.resolve(); expect(result.themes.some((r) => isDisabled(r, "dark.json"))).toBe(true); expect(result.themes.some((r) => isDisabled(r, "light.json"))).toBe(true); expect(result.themes.some((r) => isEnabled(r, "special.json"))).toBe(true); }); it("should force-include prompts", async () => { const promptsDir = join(agentDir, "prompts"); mkdirSync(promptsDir, { recursive: true }); writeFileSync(join(promptsDir, "review.md"), "Review"); writeFileSync(join(promptsDir, "explain.md"), "Explain"); writeFileSync(join(promptsDir, "debug.md"), "Debug"); settingsManager.setPromptTemplatePaths(["prompts", "!prompts/*.md", "+prompts/debug.md"]); const result = await packageManager.resolve(); expect(result.prompts.some((r) => isDisabled(r, "review.md"))).toBe(true); expect(result.prompts.some((r) => isDisabled(r, "explain.md"))).toBe(true); expect(result.prompts.some((r) => isEnabled(r, "debug.md"))).toBe(true); }); }); describe("force-exclude patterns", () => { it("should force-exclude top-level resources", async () => { const extDir = join(agentDir, "extensions"); mkdirSync(extDir, { recursive: true }); writeFileSync(join(extDir, "alpha.ts"), "export default function() {}"); writeFileSync(join(extDir, "beta.ts"), "export default function() {}"); settingsManager.setExtensionPaths(["extensions", "+extensions/alpha.ts", "-extensions/alpha.ts"]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isDisabled(r, "alpha.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "beta.ts"))).toBe(true); }); it("should force-exclude in package filters", async () => { const pkgDir = join(tempDir, "force-exclude-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "alpha.ts"), "export default function() {}"); writeFileSync(join(pkgDir, "extensions", "beta.ts"), "export default function() {}"); settingsManager.setPackages([ { source: pkgDir, extensions: ["extensions/*.ts", "+extensions/alpha.ts", "-extensions/alpha.ts"], skills: [], prompts: [], themes: [], }, ]); const result = await packageManager.resolve(); expect(result.extensions.some((r) => isDisabled(r, "alpha.ts"))).toBe(true); expect(result.extensions.some((r) => isEnabled(r, "beta.ts"))).toBe(true); }); }); describe("package deduplication", () => { it("should dedupe same local package in global and project (project wins)", async () => { const pkgDir = join(tempDir, "shared-pkg"); mkdirSync(join(pkgDir, "extensions"), { recursive: true }); writeFileSync(join(pkgDir, "extensions", "shared.ts"), "export default function() {}"); // Same package in both global and project settingsManager.setPackages([pkgDir]); // global settingsManager.setProjectPackages([pkgDir]); // project // Debug: verify settings are stored correctly const globalSettings = settingsManager.getGlobalSettings(); const projectSettings = settingsManager.getProjectSettings(); expect(globalSettings.packages).toEqual([pkgDir]); expect(projectSettings.packages).toEqual([pkgDir]); const result = await packageManager.resolve(); // Should only appear once (deduped), with project scope const sharedPaths = result.extensions.filter((r) => r.path.includes("shared-pkg")); expect(sharedPaths.length).toBe(1); expect(sharedPaths[0].metadata.scope).toBe("project"); }); it("should keep both if different packages", async () => { const pkg1Dir = join(tempDir, "pkg1"); const pkg2Dir = join(tempDir, "pkg2"); mkdirSync(join(pkg1Dir, "extensions"), { recursive: true }); mkdirSync(join(pkg2Dir, "extensions"), { recursive: true }); writeFileSync(join(pkg1Dir, "extensions", "from-pkg1.ts"), "export default function() {}"); writeFileSync(join(pkg2Dir, "extensions", "from-pkg2.ts"), "export default function() {}"); settingsManager.setPackages([pkg1Dir]); // global settingsManager.setProjectPackages([pkg2Dir]); // project const result = await packageManager.resolve(); expect(result.extensions.some((r) => r.path.includes("pkg1"))).toBe(true); expect(result.extensions.some((r) => r.path.includes("pkg2"))).toBe(true); }); it("should dedupe SSH and HTTPS URLs for same repo", async () => { // Same repository, different URL formats const httpsUrl = "https://github.com/user/repo"; const sshUrl = "git@github.com:user/repo"; const httpsIdentity = (packageManager as any).getPackageIdentity(httpsUrl); const sshIdentity = (packageManager as any).getPackageIdentity(sshUrl); // Both should resolve to the same identity expect(httpsIdentity).toBe("git:github.com/user/repo"); expect(sshIdentity).toBe("git:github.com/user/repo"); expect(httpsIdentity).toBe(sshIdentity); }); it("should dedupe SSH and HTTPS with refs", async () => { const httpsUrl = "https://github.com/user/repo@v1.0.0"; const sshUrl = "git@github.com:user/repo@v1.0.0"; const httpsIdentity = (packageManager as any).getPackageIdentity(httpsUrl); const sshIdentity = (packageManager as any).getPackageIdentity(sshUrl); // Identity should ignore ref (version) expect(httpsIdentity).toBe("git:github.com/user/repo"); expect(sshIdentity).toBe("git:github.com/user/repo"); expect(httpsIdentity).toBe(sshIdentity); }); it("should dedupe SSH URL with ssh:// protocol and git@ format", async () => { const sshProtocol = "ssh://git@github.com/user/repo"; const gitAt = "git@github.com:user/repo"; const sshProtocolIdentity = (packageManager as any).getPackageIdentity(sshProtocol); const gitAtIdentity = (packageManager as any).getPackageIdentity(gitAt); // Both SSH formats should resolve to same identity expect(sshProtocolIdentity).toBe("git:github.com/user/repo"); expect(gitAtIdentity).toBe("git:github.com/user/repo"); expect(sshProtocolIdentity).toBe(gitAtIdentity); }); it("should dedupe all URL formats for same repo (HTTPS, SSH, git@)", async () => { const urls = [ "https://github.com/user/repo", "github.com/user/repo", "https://github.com/user/repo.git", "git@github.com:user/repo", "git@github.com:user/repo.git", "ssh://git@github.com/user/repo", "git:https://github.com/user/repo", ]; const identities = urls.map((url) => (packageManager as any).getPackageIdentity(url)); // All should produce the same identity const uniqueIdentities = [...new Set(identities)]; expect(uniqueIdentities.length).toBe(1); expect(uniqueIdentities[0]).toBe("git:github.com/user/repo"); }); it("should keep different repos separate (HTTPS vs SSH)", async () => { const repo1Https = "https://github.com/user/repo1"; const repo2Ssh = "git@github.com:user/repo2"; const id1 = (packageManager as any).getPackageIdentity(repo1Https); const id2 = (packageManager as any).getPackageIdentity(repo2Ssh); // Different repos should have different identities expect(id1).toBe("git:github.com/user/repo1"); expect(id2).toBe("git:github.com/user/repo2"); expect(id1).not.toBe(id2); }); }); describe("multi-file extension discovery (issue #1102)", () => { it("should only load index.ts from subdirectories, not helper modules", async () => { // Regression test: packages with multi-file extensions in subdirectories // should only load the index.ts entry point, not helper modules like agents.ts const pkgDir = join(tempDir, "multifile-pkg"); mkdirSync(join(pkgDir, "extensions", "subagent"), { recursive: true }); // Main entry point writeFileSync( join(pkgDir, "extensions", "subagent", "index.ts"), `import { helper } from "./agents.js"; export default function(api) { api.registerTool({ name: "test", description: "test", execute: async () => helper() }); }`, ); // Helper module (should NOT be loaded as standalone extension) writeFileSync( join(pkgDir, "extensions", "subagent", "agents.ts"), `export function helper() { return "helper"; }`, ); // Top-level extension file (should be loaded) writeFileSync(join(pkgDir, "extensions", "standalone.ts"), "export default function(api) {}"); const result = await packageManager.resolveExtensionSources([pkgDir]); // Should find the index.ts and standalone.ts expect(result.extensions.some((r) => r.path.endsWith("subagent/index.ts") && r.enabled)).toBe(true); expect(result.extensions.some((r) => r.path.endsWith("standalone.ts") && r.enabled)).toBe(true); // Should NOT find agents.ts as a standalone extension expect(result.extensions.some((r) => r.path.endsWith("agents.ts"))).toBe(false); }); it("should respect package.json pi.extensions manifest in subdirectories", async () => { const pkgDir = join(tempDir, "manifest-subdir-pkg"); mkdirSync(join(pkgDir, "extensions", "custom"), { recursive: true }); // Subdirectory with its own manifest writeFileSync( join(pkgDir, "extensions", "custom", "package.json"), JSON.stringify({ pi: { extensions: ["./main.ts"], }, }), ); writeFileSync(join(pkgDir, "extensions", "custom", "main.ts"), "export default function(api) {}"); writeFileSync(join(pkgDir, "extensions", "custom", "utils.ts"), "export const util = 1;"); const result = await packageManager.resolveExtensionSources([pkgDir]); // Should find main.ts declared in manifest expect(result.extensions.some((r) => r.path.endsWith("custom/main.ts") && r.enabled)).toBe(true); // Should NOT find utils.ts (not declared in manifest) expect(result.extensions.some((r) => r.path.endsWith("utils.ts"))).toBe(false); }); it("should handle mixed top-level files and subdirectories", async () => { const pkgDir = join(tempDir, "mixed-pkg"); mkdirSync(join(pkgDir, "extensions", "complex"), { recursive: true }); // Top-level extension writeFileSync(join(pkgDir, "extensions", "simple.ts"), "export default function(api) {}"); // Subdirectory with index.ts + helpers writeFileSync( join(pkgDir, "extensions", "complex", "index.ts"), "import { a } from './a.js'; export default function(api) {}", ); writeFileSync(join(pkgDir, "extensions", "complex", "a.ts"), "export const a = 1;"); writeFileSync(join(pkgDir, "extensions", "complex", "b.ts"), "export const b = 2;"); const result = await packageManager.resolveExtensionSources([pkgDir]); // Should find simple.ts and complex/index.ts expect(result.extensions.some((r) => r.path.endsWith("simple.ts") && r.enabled)).toBe(true); expect(result.extensions.some((r) => r.path.endsWith("complex/index.ts") && r.enabled)).toBe(true); // Should NOT find helper modules expect(result.extensions.some((r) => r.path.endsWith("complex/a.ts"))).toBe(false); expect(result.extensions.some((r) => r.path.endsWith("complex/b.ts"))).toBe(false); // Total should be exactly 2 expect(result.extensions.filter((r) => r.enabled).length).toBe(2); }); it("should skip subdirectories without index.ts or manifest", async () => { const pkgDir = join(tempDir, "no-entry-pkg"); mkdirSync(join(pkgDir, "extensions", "broken"), { recursive: true }); // Subdirectory with no index.ts and no manifest writeFileSync(join(pkgDir, "extensions", "broken", "helper.ts"), "export const x = 1;"); writeFileSync(join(pkgDir, "extensions", "broken", "another.ts"), "export const y = 2;"); // Valid top-level extension writeFileSync(join(pkgDir, "extensions", "valid.ts"), "export default function(api) {}"); const result = await packageManager.resolveExtensionSources([pkgDir]); // Should only find the valid top-level extension expect(result.extensions.some((r) => r.path.endsWith("valid.ts") && r.enabled)).toBe(true); expect(result.extensions.filter((r) => r.enabled).length).toBe(1); }); }); });