@astralyn/pi 1.4.5 → 1.4.7

package/dist/modes/interactive/components/config-selector.js

@@ -5,11 +5,20 @@ import { homedir } from "node:os";
 import { basename, dirname, join, relative } from "node:path";
 import { Container, getKeybindings, Input, matchesKey, Spacer, truncateToWidth, visibleWidth, } from "@earendil-works/pi-tui";
 import { CONFIG_DIR_NAME } from "../../../config.js";
+import { readSkillDisableModelInvocation, resolveSkillEffectiveState, } from "../../../core/skills.js";
 import { BUILTIN_PATH_PREFIX } from "../../../core/source-info.js";
 import { canonicalizePath, isLocalPath, resolvePath } from "../../../utils/paths.js";
 import { theme } from "../theme/theme.js";
 import { DynamicBorder } from "./dynamic-border.js";
 import { keyHint, rawKeyHint } from "./keybinding-hints.js";
+function hasResourcePattern(patterns, pattern) {
+    const normalized = pattern.replace(/\\/g, "/");
+    for (const candidate of patterns) {
+        if (candidate.replace(/\\/g, "/") === normalized)
+            return true;
+    }
+    return false;
+}
 const RESOURCE_TYPES = ["extensions", "skills", "prompts", "themes"];
 const RESOURCE_TYPE_LABELS = {
     extensions: "Extensions",
@@ -101,6 +110,8 @@ function buildGroups(resolved, agentDir) {
                 displayName,
                 groupKey,
                 subgroupKey,
+                skillInvocationOverride: res.skillInvocationOverride,
+                frontmatterDisableModelInvocation: resourceType === "skills" ? readSkillDisableModelInvocation(path) : undefined,
             });
         }
     };
@@ -142,7 +153,7 @@ class ConfigSelectorHeader {
         const title = theme.bold(this.writeScope === "project" ? "Project Local Resources" : "Global Resources");
         const sep = theme.fg("muted", " · ");
         const switchHint = this.projectModeAvailable ? keyHint("tui.input.tab", "switch mode") + sep : "";
-        const actionHint = this.writeScope === "project" ? rawKeyHint("space", "cycle inherit/+/-") : rawKeyHint("space", "toggle");
+        const actionHint = rawKeyHint("space", "change");
         const hint = switchHint + actionHint + sep + rawKeyHint("esc", "close");
         const spacing = Math.max(1, width - visibleWidth(title) - visibleWidth(hint));
         const scopeHint = this.writeScope === "project"
@@ -173,6 +184,7 @@ class ResourceList {
         this.agentDir = agentDir;
         this.writeScope = writeScope;
         this.inheritedEnabledByKey = this.buildInheritedEnabledMap(groupsByScope.global);
+        this.inheritedSkillStateByKey = this.buildInheritedSkillStateMap(groupsByScope.global);
         this.searchInput = new Input();
         // 8 lines of chrome: top spacer + top border + spacer + header (2 lines) + spacer + bottom spacer + bottom border
         const chrome = 8;
@@ -199,6 +211,19 @@ class ResourceList {
         }
         return result;
     }
+    buildInheritedSkillStateMap(groups) {
+        const result = new Map();
+        for (const group of groups) {
+            for (const subgroup of group.subgroups) {
+                if (subgroup.type !== "skills")
+                    continue;
+                for (const item of subgroup.items) {
+                    result.set(this.getResourceItemKey(item), this.getSkillEffectiveState(item));
+                }
+            }
+        }
+        return result;
+    }
     buildFlatList() {
         this.flatItems = [];
         for (const group of this.groups) {
@@ -276,6 +301,12 @@ class ResourceList {
     }
     updateItem(item, enabled) {
         item.enabled = enabled;
+        if (this.writeScope === "global") {
+            this.inheritedEnabledByKey.set(this.getResourceItemKey(item), enabled);
+            if (item.resourceType === "skills") {
+                this.inheritedSkillStateByKey.set(this.getResourceItemKey(item), this.getSkillEffectiveState(item));
+            }
+        }
         // Update in groups too
         for (const group of this.groups) {
             for (const subgroup of group.subgroups) {
@@ -323,7 +354,11 @@ class ResourceList {
                 const dimmed = this.isDimmedItem(item);
                 const nameText = isSelected && !dimmed ? theme.bold(item.displayName) : item.displayName;
                 const name = dimmed ? theme.fg("dim", nameText) : nameText;
-                lines.push(truncateToWidth(`${cursor}    ${this.renderCheckbox(item)} ${name}${this.getItemSuffix(item)}`, width, "..."));
+                const prefix = `${cursor}    ${this.renderCheckbox(item)} `;
+                const suffix = this.getItemSuffix(item);
+                const nameWidth = Math.max(1, width - visibleWidth(prefix) - visibleWidth(suffix));
+                const renderedName = truncateToWidth(name, nameWidth, "...");
+                lines.push(truncateToWidth(`${prefix}${renderedName}${suffix}`, width, "..."));
             }
         }
         // Scroll indicator
@@ -394,6 +429,22 @@ class ResourceList {
         this.filterItems(this.searchInput.getValue());
     }
     toggleResource(item) {
+        if (item.resourceType === "skills") {
+            if (this.writeScope === "project") {
+                const state = this.getNextProjectSkillOverrideState(item);
+                if (!this.setProjectSkillOverride(item, state))
+                    return undefined;
+                item.skillInvocationOverride =
+                    state === "on" ? "model-and-user" : state === "user-only" ? "user-only" : undefined;
+                return state === "inherit" ? this.getInheritedSkillEffectiveState(item) !== "off" : state !== "off";
+            }
+            const currentState = this.getSkillEffectiveState(item);
+            const nextState = currentState === "on" ? "user-only" : currentState === "user-only" ? "off" : "on";
+            this.setGlobalSkillEffectiveState(item, nextState);
+            item.skillInvocationOverride =
+                nextState === "on" ? "model-and-user" : nextState === "user-only" ? "user-only" : undefined;
+            return nextState !== "off";
+        }
         if (this.writeScope === "project") {
             const state = this.getNextOverrideState(item);
             if (!this.setProjectResourceOverride(item, state))
@@ -505,7 +556,104 @@ class ResourceList {
             this.settingsManager.setPackages(packages);
         }
     }
+    setGlobalSkillEffectiveState(item, state) {
+        const enabled = state !== "off";
+        if (item.metadata.origin === "top-level")
+            this.toggleTopLevelResource(item, enabled);
+        else
+            this.togglePackageResource(item, enabled);
+        this.setSkillInvocationOverride(item, "user", state === "on" ? "model-and-user" : state === "user-only" ? "user-only" : undefined);
+    }
+    setProjectSkillOverride(item, state) {
+        const loadState = state === "inherit" ? "inherit" : state === "off" ? "unload" : "load";
+        if (!this.setProjectResourceOverride(item, loadState))
+            return false;
+        this.setSkillInvocationOverride(item, "project", state === "on" ? "model-and-user" : state === "user-only" ? "user-only" : undefined);
+        return true;
+    }
+    setSkillInvocationOverride(item, scope, override) {
+        if (item.metadata.origin === "top-level") {
+            const settings = scope === "project" ? this.settingsManager.getProjectSettings() : this.settingsManager.getGlobalSettings();
+            const updated = { ...settings.skillInvocationOverrides };
+            const patterns = this.getTopLevelOverridePatterns(item, scope);
+            for (const pattern of Object.keys(updated)) {
+                if (hasResourcePattern(patterns, pattern))
+                    delete updated[pattern];
+            }
+            if (override) {
+                const pattern = scope === "project" && this.isInheritedGlobalItem(item)
+                    ? item.path
+                    : this.getResourcePatternForScope(item, scope);
+                updated[pattern] = override;
+            }
+            if (scope === "project")
+                this.settingsManager.setProjectSkillInvocationOverrides(updated);
+            else
+                this.settingsManager.setSkillInvocationOverrides(updated);
+            return;
+        }
+        const settings = scope === "project" ? this.settingsManager.getProjectSettings() : this.settingsManager.getGlobalSettings();
+        const packages = [...(settings.packages ?? [])];
+        const pkgIndex = packages.findIndex((pkg) => this.packageSourceStringMatches(item.metadata.source, this.getItemScope(item), typeof pkg === "string" ? pkg : pkg.source, scope));
+        if (pkgIndex === -1)
+            return;
+        let pkg = packages[pkgIndex];
+        if (typeof pkg === "string") {
+            if (!override)
+                return;
+            pkg = { source: pkg };
+            packages[pkgIndex] = pkg;
+        }
+        const updated = { ...pkg.skillInvocationOverrides };
+        const patterns = this.getPackageSkillInvocationOverridePatterns(item);
+        for (const pattern of Object.keys(updated)) {
+            if (hasResourcePattern(patterns, pattern))
+                delete updated[pattern];
+        }
+        const pattern = this.getPackageResourcePattern(item);
+        if (override)
+            updated[pattern] = override;
+        pkg.skillInvocationOverrides = Object.keys(updated).length > 0 ? updated : undefined;
+        const hasConfiguration = RESOURCE_TYPES.some((key) => pkg[key] !== undefined) || pkg.skillInvocationOverrides !== undefined;
+        if (!hasConfiguration) {
+            if (scope === "project" && pkg.autoload === false)
+                packages.splice(pkgIndex, 1);
+            else
+                packages[pkgIndex] = pkg.source;
+        }
+        if (scope === "project")
+            this.settingsManager.setProjectPackages(packages);
+        else
+            this.settingsManager.setPackages(packages);
+    }
     renderCheckbox(item) {
+        if (item.resourceType === "skills") {
+            if (this.writeScope === "project") {
+                const state = this.getProjectSkillOverrideState(item);
+                if (state === "on")
+                    return theme.fg("success", "[+]");
+                if (state === "user-only")
+                    return theme.fg("accent", "[u]");
+                if (state === "off")
+                    return theme.fg("warning", "[-]");
+                if (state === "load") {
+                    const effectiveState = this.getSkillEffectiveState(item);
+                    if (effectiveState === "user-only")
+                        return theme.fg("accent", "[u]");
+                    if (effectiveState === "off")
+                        return theme.fg("warning", "[-]");
+                    return theme.fg("success", "[+]");
+                }
+                const inheritedState = this.getInheritedSkillEffectiveState(item);
+                return theme.fg("dim", inheritedState === "on" ? "[x]" : inheritedState === "user-only" ? "[u]" : "[ ]");
+            }
+            const state = this.getSkillEffectiveState(item);
+            if (state === "on")
+                return theme.fg("success", "[x]");
+            if (state === "user-only")
+                return theme.fg("accent", "[u]");
+            return theme.fg("dim", "[ ]");
+        }
         if (this.writeScope === "project") {
             const state = this.getProjectOverrideState(item);
             if (state === "load")
@@ -517,6 +665,21 @@ class ResourceList {
         return item.enabled ? theme.fg("success", "[x]") : theme.fg("dim", "[ ]");
     }
     getItemSuffix(item) {
+        if (item.resourceType === "skills") {
+            if (this.writeScope === "project") {
+                const state = this.getProjectSkillOverrideState(item);
+                if (state === "on")
+                    return theme.fg("muted", "  project on");
+                if (state === "user-only")
+                    return theme.fg("muted", "  project user only");
+                if (state === "off")
+                    return theme.fg("muted", "  project off");
+                if (state === "load")
+                    return theme.fg("muted", "  project load");
+                return this.isInheritedGlobalItem(item) ? theme.fg("dim", "  inherited global") : "";
+            }
+            return this.getSkillEffectiveState(item) === "user-only" ? theme.fg("accent", "  user only") : "";
+        }
         if (this.writeScope !== "project")
             return "";
         const state = this.getProjectOverrideState(item);
@@ -527,9 +690,80 @@ class ResourceList {
         return this.isInheritedGlobalItem(item) ? theme.fg("dim", "  inherited global") : "";
     }
     isDimmedItem(item) {
-        return (this.writeScope === "project" &&
-            this.isInheritedGlobalItem(item) &&
-            this.getProjectOverrideState(item) === "inherit");
+        if (this.writeScope !== "project" || !this.isInheritedGlobalItem(item))
+            return false;
+        return item.resourceType === "skills"
+            ? this.getProjectSkillOverrideState(item) === "inherit"
+            : this.getProjectOverrideState(item) === "inherit";
+    }
+    getSkillEffectiveState(item) {
+        return resolveSkillEffectiveState({
+            enabled: item.enabled,
+            invocationOverride: item.skillInvocationOverride,
+            frontmatterDisableModelInvocation: item.frontmatterDisableModelInvocation,
+        });
+    }
+    getInheritedSkillEffectiveState(item) {
+        const inherited = this.inheritedSkillStateByKey.get(this.getResourceItemKey(item));
+        if (inherited)
+            return inherited;
+        const userScoped = this.getItemScope(item) === "user";
+        return resolveSkillEffectiveState({
+            enabled: userScoped ? item.enabled : true,
+            invocationOverride: userScoped ? item.skillInvocationOverride : undefined,
+            frontmatterDisableModelInvocation: item.frontmatterDisableModelInvocation,
+        });
+    }
+    getNextProjectSkillOverrideState(item) {
+        const states = ["on", "user-only", "off"];
+        const inheritedIndex = states.indexOf(this.getInheritedSkillEffectiveState(item));
+        const sequence = [...states.slice(inheritedIndex + 1), ...states.slice(0, inheritedIndex + 1)];
+        const current = this.getProjectSkillOverrideState(item);
+        if (current === "inherit")
+            return sequence[0];
+        if (current === "load")
+            return "inherit";
+        const currentIndex = sequence.indexOf(current);
+        return sequence[currentIndex + 1] ?? "inherit";
+    }
+    getProjectSkillOverrideState(item) {
+        if (this.writeScope !== "project")
+            return "inherit";
+        let loadState;
+        let invocationOverrides;
+        let patterns;
+        if (item.metadata.origin === "top-level") {
+            const settings = this.settingsManager.getProjectSettings();
+            loadState = this.getOverrideStateFromEntries((settings.skills ?? []), this.getTopLevelOverridePatterns(item, "project"), false);
+            invocationOverrides = settings.skillInvocationOverrides;
+            patterns = this.getTopLevelOverridePatterns(item, "project");
+        }
+        else {
+            const pkg = this.findMatchingPackageSource(item, "project");
+            if (typeof pkg !== "object")
+                return "inherit";
+            loadState =
+                pkg.skills === undefined
+                    ? "inherit"
+                    : this.getOverrideStateFromEntries(pkg.skills, this.getPackageSkillInvocationOverridePatterns(item), pkg.autoload !== false);
+            invocationOverrides = pkg.skillInvocationOverrides;
+            patterns = this.getPackageSkillInvocationOverridePatterns(item);
+        }
+        if (loadState === "unload")
+            return "off";
+        const invocation = this.getSkillInvocationOverrideFromMap(invocationOverrides, patterns);
+        if (invocation)
+            return invocation === "user-only" ? "user-only" : "on";
+        return loadState;
+    }
+    getSkillInvocationOverrideFromMap(overrides, patterns) {
+        for (const [pattern, override] of Object.entries(overrides ?? {})) {
+            if (!hasResourcePattern(patterns, pattern))
+                continue;
+            if (override === "model-and-user" || override === "user-only")
+                return override;
+        }
+        return undefined;
     }
     setProjectResourceOverride(item, state) {
         return item.metadata.origin === "top-level"
@@ -586,7 +820,8 @@ class ResourceList {
         if (state !== "inherit")
             updated.push(`${state === "load" ? "+" : "-"}${pattern}`);
         pkg[item.resourceType] = updated.length > 0 ? updated : undefined;
-        if (!RESOURCE_TYPES.some((key) => pkg[key] !== undefined)) {
+        if (!RESOURCE_TYPES.some((key) => pkg[key] !== undefined) &&
+            pkg.skillInvocationOverrides === undefined) {
             if (pkg.autoload === false)
                 packages.splice(pkgIndex, 1);
             else
@@ -623,7 +858,7 @@ class ResourceList {
             return "unload";
         let state = "inherit";
         for (const entry of entries) {
-            if (!patterns.has(this.getPatternEntryTarget(entry)))
+            if (!hasResourcePattern(patterns, this.getPatternEntryTarget(entry)))
                 continue;
             if (entry.startsWith("!") || entry.startsWith("-"))
                 state = "unload";
@@ -648,6 +883,13 @@ class ResourceList {
         ]);
         if (item.metadata.baseDir)
             patterns.add(relative(item.metadata.baseDir, item.path));
+        if (item.resourceType === "skills" && basename(item.path) === "SKILL.md") {
+            const parentDir = dirname(item.path);
+            patterns.add(parentDir);
+            patterns.add(relative(baseDir, parentDir));
+            if (item.metadata.baseDir)
+                patterns.add(relative(item.metadata.baseDir, parentDir));
+        }
         return patterns;
     }
     getResourcePatternForScope(item, scope) {
@@ -702,6 +944,15 @@ class ResourceList {
         const baseDir = item.metadata.baseDir ?? dirname(item.path);
         return relative(baseDir, item.path);
     }
+    getPackageSkillInvocationOverridePatterns(item) {
+        const baseDir = item.metadata.baseDir ?? dirname(item.path);
+        const patterns = new Set([this.getPackageResourcePattern(item), item.path]);
+        if (basename(item.path) === "SKILL.md") {
+            patterns.add(relative(baseDir, dirname(item.path)));
+            patterns.add(dirname(item.path));
+        }
+        return patterns;
+    }
 }
 export class ConfigSelectorComponent extends Container {
     get focused() {