agent_engine/skills/

registry.rs

//! Slash command registry: built-ins + dynamically registered skills.

use std::collections::HashMap;
use std::sync::{Arc, RwLock};
use crate::skills::{LoadedSkill, Plugin};
use crate::help::HelpEntry;

#[derive(Debug, Clone)]
pub struct PluginSummary {
    pub name: String,
    pub skill_count: usize,
}

#[derive(Clone, Debug)]
pub enum RegisteredPluginCommandBackend {
    Shell { command: String, args: Vec<String> },
    ExtensionTool { tool: String, input: serde_json::Value },
    SkillPrompt { skill: String, prompt: String },
    Interactive { plugin_extension_id: String },
}

/// Editor kind for a plugin-declared settings field, normalised from the
/// manifest into a form the settings UI can consume directly.
#[derive(Clone, Debug, PartialEq, Eq)]
pub enum PluginSettingsEditor {
    Text { numeric: bool },
    Cycler { options: Vec<String> },
    Picker,
    /// Editor body rendered by the plugin via the
    /// `settings.editor.*` JSON-RPC contract.
    Custom,
}

#[derive(Clone, Debug)]
pub struct PluginSettingsField {
    pub key: String,
    pub label: String,
    pub editor: PluginSettingsEditor,
    pub help: Option<String>,
    pub default: Option<serde_json::Value>,
}

/// A plugin-declared category contributed to the `/settings` modal.
///
/// The owning plugin is recorded in `plugin` so that the settings UI can
/// scope reads/writes to that plugin's config namespace and dispatch
/// custom-editor JSON-RPC events to the right extension.
#[derive(Clone, Debug)]
pub struct PluginSettingsCategory {
    pub plugin: String,
    pub id: String,
    pub label: String,
    pub fields: Vec<PluginSettingsField>,
}

/// Resolution outcome for a typed slash command.
#[derive(Clone, Debug)]
pub struct RegisteredPluginCommand {
    pub plugin: String,
    pub name: String,
    pub description: Option<String>,
    pub backend: RegisteredPluginCommandBackend,
    pub plugin_root: std::path::PathBuf,
}

/// A plugin's claim on a top-level lifecycle command namespace, derived
/// from `provides.sidecar.lifecycle` in plugin.json. Phase 8 slice 8A.
///
/// When a plugin declares a lifecycle claim, the command registry
/// auto-registers `/<command> toggle` and `/<command> status` (and any
/// future lifecycle verbs) so the plugin's own UX namespace —
/// e.g. a plugin-owned command — drives sidecar lifecycle instead
/// of the modality-laden `/sidecar` builtin.
///
/// Multiple plugins may register lifecycle claims; in Phase 8 slice 8A
/// the host still hosts at most one active sidecar so all claims share
/// a single backing `App.sidecar` slot. Slice 8B widens this.
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct LifecycleClaim {
    /// Plugin (manifest) name that owns this claim.
    pub plugin: String,
    /// Top-level command word the plugin claims.
    pub command: String,
    /// Optional settings-category id that the host should inject
    /// per-plugin lifecycle keys (e.g. toggle keybind) into.
    pub settings_category: Option<String>,
    /// Human-readable label for pills, error messages, status lines.
    pub display_name: String,
    /// Sort key for pill ordering (higher = earlier). Range `-100..=100`,
    /// already clamped during manifest deserialisation.
    pub importance: i32,
}

pub enum Resolution {
    /// A built-in command (dispatched via existing handle_command).
    Builtin,
    /// A single unambiguous skill.
    Skill(Arc<LoadedSkill>),
    /// A plugin command from plugin.json `commands`.
    PluginCommand(Arc<RegisteredPluginCommand>),
    /// Multiple skills share this unqualified name; user must qualify.
    Ambiguous(Vec<String>), // list of plugin-qualified names
    /// No such command.
    Unknown,
}

struct Inner {
    skills: HashMap<String, Vec<Arc<LoadedSkill>>>, // unqualified name -> all matches
    qualified: HashMap<String, Arc<LoadedSkill>>,   // "plugin:skill" -> single
    plugin_commands: HashMap<String, Arc<RegisteredPluginCommand>>, // "plugin:cmd" -> single
    plugin_help_entries: Vec<HelpEntry>,
    /// Plugin-declared settings categories, in plugin discovery order.
    /// Path B Phase 4. The settings UI snapshots this on each open.
    plugin_settings_categories: Vec<PluginSettingsCategory>,
    /// Plugin-declared lifecycle claims (Phase 8 slice 8A). Indexed by
    /// the claimed command word. At most one claim per
    /// command word; first-loaded plugin wins on collision.
    lifecycle_claims: HashMap<String, LifecycleClaim>,
    /// Plugins that lost a lifecycle-claim collision, recorded so the
    /// `/extensions` view can surface a warning. Each entry is
    /// `(losing_plugin, command, winning_plugin)`.
    lifecycle_claim_collisions: Vec<(String, String, String)>,
}

pub struct CommandRegistry {
    builtins: Vec<&'static str>,
    inner: RwLock<Inner>,
}

impl CommandRegistry {
    pub fn new(builtins: &[&'static str], skills: Vec<LoadedSkill>) -> Self {
        Self::new_with_plugins(builtins, skills, vec![])
    }

    pub fn new_with_plugins(builtins: &[&'static str], skills: Vec<LoadedSkill>, plugins: Vec<Plugin>) -> Self {
        let r = CommandRegistry {
            builtins: builtins.to_vec(),
            inner: RwLock::new(Inner {
                skills: HashMap::new(),
                qualified: HashMap::new(),
                plugin_commands: HashMap::new(),
                plugin_help_entries: Vec::new(),
                plugin_settings_categories: Vec::new(),
                lifecycle_claims: HashMap::new(),
                lifecycle_claim_collisions: Vec::new(),
            }),
        };
        r.rebuild_with_plugins(skills, plugins);
        r
    }

    /// Atomically replace the skill set. Built-ins are unchanged.
    pub fn rebuild_with(&self, skills: Vec<LoadedSkill>) {
        self.rebuild_with_plugins(skills, vec![]);
    }

    /// Atomically replace the skill and plugin-command set. Built-ins are unchanged.
    pub fn rebuild_with_plugins(&self, skills: Vec<LoadedSkill>, plugins: Vec<Plugin>) {
        let builtins_set: std::collections::HashSet<&str> =
            self.builtins.iter().copied().collect();
        let mut new_skills: HashMap<String, Vec<Arc<LoadedSkill>>> = HashMap::new();
        let mut new_qualified: HashMap<String, Arc<LoadedSkill>> = HashMap::new();
        let mut new_plugin_commands: HashMap<String, Arc<RegisteredPluginCommand>> = HashMap::new();
        let mut new_plugin_help_entries: Vec<HelpEntry> = Vec::new();
        let mut new_plugin_settings_categories: Vec<PluginSettingsCategory> = Vec::new();
        let mut new_lifecycle_claims: HashMap<String, LifecycleClaim> = HashMap::new();
        let mut new_lifecycle_collisions: Vec<(String, String, String)> = Vec::new();
        for plugin in plugins {
            if let Some(manifest) = plugin.manifest {
                new_plugin_help_entries.extend(manifest.help_entries.iter().cloned().map(|mut entry| {
                    entry.source = Some(manifest.name.clone());
                    entry
                }));
                // Phase 8 slice 8A: harvest lifecycle claims from
                // provides.sidecar.lifecycle. First-loaded wins on
                // collision; the loser is recorded for surfacing in
                // /extensions and the chat log.
                if let Some(ref provides) = manifest.provides {
                    if let Some(ref sidecar) = provides.sidecar {
                        if let Some(ref lc) = sidecar.lifecycle {
                            let claim = LifecycleClaim {
                                plugin: manifest.name.clone(),
                                command: lc.command.clone(),
                                settings_category: lc.settings_category.clone(),
                                display_name: lc.effective_display_name().to_string(),
                                importance: lc.importance,
                            };
                            // Block plugins from hijacking built-in commands
                            if builtins_set.contains(claim.command.as_str()) {
                                tracing::warn!(
                                    "plugin '{}' attempted to claim builtin command '{}'; rejected",
                                    claim.plugin, claim.command,
                                );
                            } else if let Some(existing) = new_lifecycle_claims.get(&claim.command) {
                                new_lifecycle_collisions.push((
                                    claim.plugin.clone(),
                                    claim.command.clone(),
                                    existing.plugin.clone(),
                                ));
                                tracing::warn!(
                                    "lifecycle command '{}' claimed by both '{}' and '{}'; first-loaded wins",
                                    claim.command, existing.plugin, claim.plugin,
                                );
                            } else {
                                new_lifecycle_claims.insert(claim.command.clone(), claim);
                            }
                        }
                    }
                }
                if let Some(ref settings) = manifest.settings {
                    for cat in &settings.categories {
                        let fields = cat
                            .fields
                            .iter()
                            .map(|f| PluginSettingsField {
                                key: f.key.clone(),
                                label: f.label.clone(),
                                editor: match f.editor {
                                    crate::skills::manifest::ManifestEditorKind::Text => {
                                        PluginSettingsEditor::Text { numeric: f.numeric }
                                    }
                                    crate::skills::manifest::ManifestEditorKind::Cycler => {
                                        PluginSettingsEditor::Cycler {
                                            options: f.options.clone(),
                                        }
                                    }
                                    crate::skills::manifest::ManifestEditorKind::Picker => {
                                        PluginSettingsEditor::Picker
                                    }
                                    crate::skills::manifest::ManifestEditorKind::Custom => {
                                        PluginSettingsEditor::Custom
                                    }
                                },
                                help: f.help.clone(),
                                default: f.default.clone(),
                            })
                            .collect();
                        new_plugin_settings_categories.push(PluginSettingsCategory {
                            plugin: manifest.name.clone(),
                            id: cat.id.clone(),
                            label: cat.label.clone(),
                            fields,
                        });
                    }
                }
                for cmd in manifest.commands {
                    let (name, description, backend) = match cmd {
                        crate::skills::manifest::ManifestCommand::Shell(cmd) => (
                            cmd.name,
                            cmd.description,
                            RegisteredPluginCommandBackend::Shell { command: cmd.command, args: cmd.args },
                        ),
                        crate::skills::manifest::ManifestCommand::ExtensionTool(cmd) => (
                            cmd.name,
                            cmd.description,
                            RegisteredPluginCommandBackend::ExtensionTool { tool: cmd.tool, input: cmd.input },
                        ),
                        crate::skills::manifest::ManifestCommand::SkillPrompt(cmd) => (
                            cmd.name,
                            cmd.description,
                            RegisteredPluginCommandBackend::SkillPrompt { skill: cmd.skill, prompt: cmd.prompt },
                        ),
                        crate::skills::manifest::ManifestCommand::Interactive(cmd) => {
                            if !cmd.interactive {
                                continue;
                            }
                            (
                                cmd.name,
                                cmd.description,
                                RegisteredPluginCommandBackend::Interactive {
                                    plugin_extension_id: manifest
                                        .extension
                                        .as_ref()
                                        .map(|_| plugin.name.clone())
                                        .unwrap_or_else(|| plugin.name.clone()),
                                },
                            )
                        },
                    };
                    let q = format!("{}:{}", manifest.name, name);
                    // Block plugins from registering commands that match builtin names
                    if builtins_set.contains(name.as_str()) {
                        tracing::warn!(
                            "plugin '{}' command '{}' shadows builtin; skipping",
                            manifest.name, name,
                        );
                        continue;
                    }
                    new_plugin_commands.insert(q, Arc::new(RegisteredPluginCommand {
                        plugin: manifest.name.clone(),
                        name,
                        description,
                        backend,
                        plugin_root: plugin.root.clone(),
                    }));
                }
            }
        }
        for s in skills {
            let arc = Arc::new(s);
            // Unqualified entry
            if builtins_set.contains(arc.name.as_str()) {
                tracing::warn!(
                    "skill '{}' shadowed by built-in; reachable only via qualified form '{}:{}'",
                    arc.name,
                    arc.plugin.as_deref().unwrap_or("?"),
                    arc.name
                );
            } else {
                new_skills.entry(arc.name.clone()).or_default().push(arc.clone());
            }
            // Qualified entry
            if let Some(ref p) = arc.plugin {
                let q = format!("{}:{}", p, arc.name);
                new_qualified.insert(q, arc.clone());
            }
        }
        // Phase 8 slice 8A.3: for each lifecycle claim with a
        // settings_category, inject a synthetic `_lifecycle_toggle_key`
        // field at the front of the matching plugin-declared category.
        // No-op (with a warn) if the named category doesn't exist.
        for claim in new_lifecycle_claims.values() {
            let Some(ref cat_id) = claim.settings_category else {
                continue;
            };
            let pos = new_plugin_settings_categories
                .iter()
                .position(|c| c.plugin == claim.plugin && &c.id == cat_id);
            match pos {
                Some(idx) => {
                    let injected = PluginSettingsField {
                        key: "_lifecycle_toggle_key".to_string(),
                        label: "Toggle key".to_string(),
                        editor: PluginSettingsEditor::Cycler {
                            options: ["F8", "F2", "F12", "C-V", "C-G"]
                                .iter()
                                .map(|s| s.to_string())
                                .collect(),
                        },
                        help: Some("Keybind that toggles this sidecar.".to_string()),
                        default: None,
                    };
                    new_plugin_settings_categories[idx]
                        .fields
                        .insert(0, injected);
                }
                None => {
                    tracing::warn!(
                        "lifecycle claim for plugin '{}' references settings_category '{}' but no such category was declared; skipping toggle-key injection",
                        claim.plugin,
                        cat_id,
                    );
                }
            }
        }

        let mut w = self.inner.write().unwrap();
        w.skills = new_skills;
        w.qualified = new_qualified;
        w.plugin_commands = new_plugin_commands;
        w.plugin_help_entries = new_plugin_help_entries;
        w.plugin_settings_categories = new_plugin_settings_categories;
        w.lifecycle_claims = new_lifecycle_claims;
        w.lifecycle_claim_collisions = new_lifecycle_collisions;
    }

    pub fn resolve(&self, cmd: &str) -> Resolution {
        let r = self.inner.read().unwrap();
        if cmd.contains(':') {
            if let Some(c) = r.plugin_commands.get(cmd) {
                return Resolution::PluginCommand(c.clone());
            }
            return match r.qualified.get(cmd) {
                Some(s) => Resolution::Skill(s.clone()),
                None => Resolution::Unknown,
            };
        }
        if self.builtins.contains(&cmd) {
            return Resolution::Builtin;
        }
        match r.skills.get(cmd) {
            Some(v) if v.len() == 1 => Resolution::Skill(v[0].clone()),
            Some(v) => Resolution::Ambiguous(
                v.iter()
                    .map(|s| format!("{}:{}", s.plugin.as_deref().unwrap_or("?"), s.name))
                    .collect(),
            ),
            None => {
                // Fall back to an unambiguous plugin command claiming
                // this bare name. `find_plugin_command_unqualified`
                // returns Some only when exactly one plugin claims it;
                // ambiguous (>=2) and missing both fall through to
                // Unknown so the user must qualify.
                drop(r);
                if let Some(c) = self.find_plugin_command_unqualified(cmd) {
                    return Resolution::PluginCommand(c);
                }
                Resolution::Unknown
            }
        }
    }

    /// Look up a plugin command by its unqualified name. Returns the
    /// only match, or None if zero or multiple plugins claim the name.
    /// Used when a builtin command has the same identifier as a
    /// plugin-owned command that also has extension subcommands
    /// builtin and a plugin command name) and we need to bypass the
    /// builtin check to reach the plugin.
    pub fn find_plugin_command_unqualified(&self, name: &str) -> Option<Arc<RegisteredPluginCommand>> {
        let r = self.inner.read().unwrap();
        let mut matches = r.plugin_commands.values().filter(|c| c.name == name);
        let first = matches.next()?.clone();
        if matches.next().is_some() {
            return None; // ambiguous
        }
        Some(first)
    }

    /// All commands for autocomplete/help: builtins + unique unqualified skill names, sorted.
    pub fn all_commands(&self) -> Vec<String> {
        let r = self.inner.read().unwrap();
        let mut v: Vec<String> = self.builtins.iter().map(|s| s.to_string()).collect();
        v.extend(r.skills.keys().cloned());
        // Plugin commands: group by unqualified name. If exactly one
        // plugin claims a name, surface it bare so users can type
        // `/crush` instead of `/crush:crush`. If two or more plugins
        // share the name, surface every qualified key so the user can
        // disambiguate. Mirrors `resolve()`'s bare-resolution logic.
        let mut by_name: HashMap<&str, Vec<&str>> = HashMap::new();
        for (qkey, c) in r.plugin_commands.iter() {
            by_name.entry(c.name.as_str()).or_default().push(qkey.as_str());
        }
        for (name, qkeys) in by_name {
            if qkeys.len() == 1 {
                v.push(name.to_string());
            } else {
                v.extend(qkeys.iter().map(|s| s.to_string()));
            }
        }
        // Phase 8 slice 8A: plugin-claimed lifecycle commands surface
        // as top-level commands too.
        v.extend(r.lifecycle_claims.keys().cloned());
        v.sort();
        v.dedup();
        v
    }

    /// Look up the lifecycle claim for a top-level command word, if any.
    /// Phase 8 slice 8A. Returns the claim regardless of arg/subcommand —
    /// callers (the dispatcher) decide what to do with `toggle` / `status`.
    pub fn lifecycle_for_command(&self, cmd: &str) -> Option<LifecycleClaim> {
        let r = self.inner.read().unwrap();
        r.lifecycle_claims.get(cmd).cloned()
    }

    /// Snapshot of every claimed lifecycle command, in arbitrary order.
    /// Used by `/extensions` to render the active claims and by the
    /// pill renderer to enumerate sidecars.
    pub fn lifecycle_claims(&self) -> Vec<LifecycleClaim> {
        self.inner.read().unwrap().lifecycle_claims.values().cloned().collect()
    }

    /// Plugins that lost a lifecycle-claim collision during the most
    /// recent rebuild. Each entry is `(losing_plugin, command,
    /// winning_plugin)`. Surfaced in `/extensions`.
    pub fn lifecycle_claim_collisions(&self) -> Vec<(String, String, String)> {
        self.inner.read().unwrap().lifecycle_claim_collisions.clone()
    }

    pub fn plugins(&self) -> Vec<PluginSummary> {
        let r = self.inner.read().unwrap();
        let mut counts: std::collections::HashMap<String, usize> = std::collections::HashMap::new();
        let mut seen: std::collections::HashSet<(String, String)> = std::collections::HashSet::new();
        for c in r.plugin_commands.values() {
            let key = (c.plugin.clone(), c.name.clone());
            if seen.insert(key) {
                *counts.entry(c.plugin.clone()).or_insert(0) += 0;
            }
        }
        for s in r.qualified.values() {
            if let Some(ref p) = s.plugin {
                let key = (p.clone(), s.name.clone());
                if seen.insert(key) {
                    *counts.entry(p.clone()).or_insert(0) += 1;
                }
            }
        }
        counts.into_iter()
            .map(|(name, skill_count)| PluginSummary { name, skill_count })
            .collect()
    }

    pub fn plugin_help_entries(&self) -> Vec<HelpEntry> {
        self.inner.read().unwrap().plugin_help_entries.clone()
    }

    /// Snapshot of every plugin-declared settings category, preserving
    /// plugin discovery order. The settings UI calls this on each open
    /// to merge plugin categories with the built-in ones.
    pub fn plugin_settings_categories(&self) -> Vec<PluginSettingsCategory> {
        self.inner.read().unwrap().plugin_settings_categories.clone()
    }

    pub fn all_skills(&self) -> Vec<Arc<LoadedSkill>> {
        let r = self.inner.read().unwrap();
        let mut seen: std::collections::HashSet<(Option<String>, String)> =
            std::collections::HashSet::new();
        let mut out = Vec::new();
        for list in r.skills.values() {
            for s in list {
                let key = (s.plugin.clone(), s.name.clone());
                if seen.insert(key) { out.push(s.clone()); }
            }
        }
        for s in r.qualified.values() {
            let key = (s.plugin.clone(), s.name.clone());
            if seen.insert(key) { out.push(s.clone()); }
        }
        out
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::path::PathBuf;
    use crate::skills::manifest::{ManifestCommand, ManifestShellCommand};

    fn mk_cmd(plugin: &str, name: &str, root: PathBuf) -> Plugin {
        Plugin {
            name: plugin.to_string(),
            root,
            marketplace: None,
            version: None,
            description: None,
            extension: None,
            manifest: Some(crate::skills::manifest::PluginManifest {
                name: plugin.to_string(),
                version: None,
                description: None,
                keybinds: vec![],
                compatibility: None,
                commands: vec![ManifestCommand::Shell(ManifestShellCommand {
                    name: name.to_string(),
                    description: Some("desc".to_string()),
                    command: "printf".to_string(),
                    args: vec!["hi".to_string()],
                })],
                extension: None,
                help_entries: vec![],
                provides: None,
                settings: None,
            }),
        }
    }


    fn mk_interactive_cmd(plugin: &str, name: &str, root: PathBuf) -> Plugin {
        Plugin {
            name: plugin.to_string(),
            root,
            marketplace: None,
            version: None,
            description: None,
            extension: None,
            manifest: Some(crate::skills::manifest::PluginManifest {
                name: plugin.to_string(),
                version: None,
                description: None,
                keybinds: vec![],
                compatibility: None,
                commands: vec![ManifestCommand::Interactive(crate::skills::manifest::ManifestInteractiveCommand {
                    name: name.to_string(),
                    description: Some("interactive desc".to_string()),
                    interactive: true,
                    subcommands: vec!["help".to_string()],
                })],
                extension: None,
                help_entries: vec![],
                provides: None,
                settings: None,
            }),
        }
    }

    #[test]
    fn registers_interactive_plugin_command_backend() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_interactive_cmd("demo-plugin", "demo", PathBuf::from("/tmp/demo"))],
        );

        match reg.resolve("demo-plugin:demo") {
            Resolution::PluginCommand(cmd) => match &cmd.backend {
                RegisteredPluginCommandBackend::Interactive { plugin_extension_id } => {
                    assert_eq!(plugin_extension_id, "demo-plugin");
                    assert_eq!(cmd.name, "demo");
                }
                other => panic!("expected interactive backend, got {other:?}"),
            },
            _ => panic!("expected plugin command resolution"),
        }
    }

    fn mk(name: &str, plugin: Option<&str>) -> LoadedSkill {
        LoadedSkill {
            name: name.to_string(),
            description: String::new(),
            body: String::new(),
            plugin: plugin.map(str::to_string),
            base_dir: PathBuf::from("/"),
            source_path: PathBuf::from("/SKILL.md"),
        }
    }


    #[test]
    fn plugin_help_entries_are_tagged_with_manifest_name() {
        let root = PathBuf::from("/tmp/plugin-root");
        let mut plugin = mk_cmd("acme-tools", "sync", root);
        plugin.manifest.as_mut().unwrap().help_entries.push(HelpEntry {
            id: "acme-sync".to_string(),
            command: "/acme:sync".to_string(),
            title: "Acme Sync".to_string(),
            summary: "Sync Acme workspace state.".to_string(),
            category: "Plugin".to_string(),
            topic: crate::help::HelpTopicKind::Command,
            protected: false,
            common: false,
            aliases: vec![],
            keywords: vec![],
            lines: vec![],
            usage: Some("/acme:sync [workspace]".to_string()),
            examples: vec![crate::help::HelpExample {
                command: "/acme:sync docs".to_string(),
                description: "Sync docs.".to_string(),
            }],
            related: vec![],
            source: None,
        });

        let registry = CommandRegistry::new_with_plugins(&[], vec![], vec![plugin]);
        let entries = registry.plugin_help_entries();

        assert_eq!(entries.len(), 1);
        assert_eq!(entries[0].source.as_deref(), Some("acme-tools"));
        assert_eq!(entries[0].usage.as_deref(), Some("/acme:sync [workspace]"));
        assert_eq!(entries[0].examples[0].command, "/acme:sync docs");
    }

    #[test]
    fn resolve_builtin() {
        let r = CommandRegistry::new(&["clear"], vec![]);
        assert!(matches!(r.resolve("clear"), Resolution::Builtin));
    }

    #[test]
    fn resolve_unknown() {
        let r = CommandRegistry::new(&["clear"], vec![]);
        assert!(matches!(r.resolve("xyz"), Resolution::Unknown));
    }

    #[test]
    fn resolve_unique_skill() {
        let r = CommandRegistry::new(&[], vec![mk("search", Some("p"))]);
        match r.resolve("search") {
            Resolution::Skill(s) => assert_eq!(s.name, "search"),
            _ => panic!(),
        }
    }

    #[test]
    fn resolve_ambiguous() {
        let r = CommandRegistry::new(&[], vec![
            mk("search", Some("p1")),
            mk("search", Some("p2")),
        ]);
        match r.resolve("search") {
            Resolution::Ambiguous(v) => {
                assert_eq!(v.len(), 2);
                assert!(v.iter().any(|s| s == "p1:search"));
                assert!(v.iter().any(|s| s == "p2:search"));
            }
            _ => panic!(),
        }
    }

    #[test]
    fn resolve_qualified() {
        let r = CommandRegistry::new(&[], vec![
            mk("search", Some("p1")),
            mk("search", Some("p2")),
        ]);
        match r.resolve("p1:search") {
            Resolution::Skill(s) => assert_eq!(s.plugin.as_deref(), Some("p1")),
            _ => panic!(),
        }
    }

    #[test]
    fn builtin_shadows_skill_unqualified() {
        // Skill named "clear" should not win over the built-in.
        let r = CommandRegistry::new(&["clear"], vec![mk("clear", Some("p"))]);
        assert!(matches!(r.resolve("clear"), Resolution::Builtin));
        // Qualified form still works.
        match r.resolve("p:clear") {
            Resolution::Skill(s) => assert_eq!(s.name, "clear"),
            _ => panic!(),
        }
    }

    #[test]
    fn all_commands_sorted_and_deduped() {
        let r = CommandRegistry::new(&["clear", "model"], vec![
            mk("search", Some("p")),
            mk("help-me", None),
        ]);
        let cmds = r.all_commands();
        assert_eq!(cmds, vec!["clear", "help-me", "model", "search"]);
    }

    #[test]
    fn all_skills_dedups_plugin_skill() {
        let r = CommandRegistry::new(&[], vec![mk("search", Some("p"))]);
        let all = r.all_skills();
        assert_eq!(all.len(), 1);
        assert_eq!(all[0].name, "search");
        assert_eq!(all[0].plugin.as_deref(), Some("p"));
    }

    #[test]
    fn all_skills_includes_shadowed_skill() {
        let r = CommandRegistry::new(&["clear"], vec![mk("clear", Some("p"))]);
        let all = r.all_skills();
        assert_eq!(all.len(), 1);
        assert_eq!(all[0].name, "clear");
        assert_eq!(all[0].plugin.as_deref(), Some("p"));
    }

    #[test]
    fn resolve_qualified_unknown_returns_unknown() {
        let r = CommandRegistry::new(&[], vec![mk("search", Some("p1"))]);
        assert!(matches!(r.resolve("p1:nosuch"), Resolution::Unknown));
        assert!(matches!(r.resolve("nosuch:search"), Resolution::Unknown));
    }

    #[test]
    fn rebuild_replaces_skills() {
        let r = CommandRegistry::new(&["clear"], vec![mk("old", None)]);
        assert!(matches!(r.resolve("old"), Resolution::Skill(_)));
        assert!(matches!(r.resolve("new"), Resolution::Unknown));
        r.rebuild_with(vec![mk("new", None)]);
        assert!(matches!(r.resolve("old"), Resolution::Unknown));
        assert!(matches!(r.resolve("new"), Resolution::Skill(_)));
    }

    #[test]
    fn rebuild_visible_through_shared_arc() {
        let r = std::sync::Arc::new(CommandRegistry::new(&[], vec![mk("a", None)]));
        let r2 = r.clone();
        r.rebuild_with(vec![mk("b", None)]);
        assert!(matches!(r2.resolve("b"), Resolution::Skill(_)));
        assert!(matches!(r2.resolve("a"), Resolution::Unknown));
    }

    #[test]
    fn resolve_qualified_plugin_command() {
        let r = CommandRegistry::new_with_plugins(&[], vec![], vec![mk_cmd("p", "hello", PathBuf::from("/tmp/p"))]);
        match r.resolve("p:hello") {
            Resolution::PluginCommand(cmd) => {
                assert_eq!(cmd.plugin, "p");
                assert_eq!(cmd.name, "hello");
                assert!(matches!(
                    &cmd.backend,
                    RegisteredPluginCommandBackend::Shell { command, .. } if command == "printf"
                ));
                assert_eq!(cmd.plugin_root, PathBuf::from("/tmp/p"));
            }
            _ => panic!(),
        }
    }

    #[test]
    fn all_commands_includes_unambiguous_plugin_commands_bare() {
        // Updated from `all_commands_includes_qualified_plugin_commands`:
        // a sole plugin command is now surfaced bare ("hello"), not
        // qualified ("p:hello"). Ambiguous names stay qualified — see
        // `all_commands_ambiguous_plugin_commands_stay_qualified`.
        let r = CommandRegistry::new_with_plugins(&["help"], vec![], vec![mk_cmd("p", "hello", PathBuf::from("/tmp/p"))]);
        let cmds = r.all_commands();
        assert!(cmds.contains(&"help".to_string()));
        assert!(cmds.contains(&"hello".to_string()));
        assert!(!cmds.contains(&"p:hello".to_string()));
    }

    #[test]
    fn resolve_bare_unique_plugin_command() {
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_cmd("crush", "crush", PathBuf::from("/tmp/crush"))],
        );
        match r.resolve("crush") {
            Resolution::PluginCommand(cmd) => {
                assert_eq!(cmd.plugin, "crush");
                assert_eq!(cmd.name, "crush");
            }
            _ => panic!("expected PluginCommand"),
        }
        // Qualified form still works (no regression).
        match r.resolve("crush:crush") {
            Resolution::PluginCommand(cmd) => assert_eq!(cmd.plugin, "crush"),
            _ => panic!("expected PluginCommand"),
        }
    }

    #[test]
    fn resolve_bare_ambiguous_plugin_command_unknown() {
        // Two plugins both expose "foo". Bare resolution must NOT pick
        // one — `find_plugin_command_unqualified` returns None for >1
        // match, so we fall through to Unknown. Qualified forms still
        // resolve.
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![
                mk_cmd("a", "foo", PathBuf::from("/tmp/a")),
                mk_cmd("b", "foo", PathBuf::from("/tmp/b")),
            ],
        );
        assert!(matches!(r.resolve("foo"), Resolution::Unknown));
        assert!(matches!(r.resolve("a:foo"), Resolution::PluginCommand(_)));
        assert!(matches!(r.resolve("b:foo"), Resolution::PluginCommand(_)));
    }

    #[test]
    fn all_commands_ambiguous_plugin_commands_stay_qualified() {
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![
                mk_cmd("a", "foo", PathBuf::from("/tmp/a")),
                mk_cmd("b", "foo", PathBuf::from("/tmp/b")),
            ],
        );
        let cmds = r.all_commands();
        assert!(cmds.contains(&"a:foo".to_string()));
        assert!(cmds.contains(&"b:foo".to_string()));
        assert!(!cmds.contains(&"foo".to_string()));
    }

    #[test]
    fn plugins_summary_groups_by_plugin_name() {
        let r = CommandRegistry::new(&[], vec![
            mk("a", Some("p1")),
            mk("b", Some("p1")),
            mk("c", Some("p2")),
            mk("loose", None),
        ]);
        let mut plugins = r.plugins();
        plugins.sort_by(|a, b| a.name.cmp(&b.name));
        assert_eq!(plugins.len(), 2);
        assert_eq!(plugins[0].name, "p1");
        assert_eq!(plugins[0].skill_count, 2);
        assert_eq!(plugins[1].name, "p2");
        assert_eq!(plugins[1].skill_count, 1);
    }

    fn mk_plugin_with_settings(plugin: &str, root: PathBuf) -> Plugin {
        use crate::skills::manifest::{
            ManifestEditorKind, ManifestSettings, ManifestSettingsCategory,
            ManifestSettingsField,
        };
        Plugin {
            name: plugin.to_string(),
            root,
            marketplace: None,
            version: None,
            description: None,
            extension: None,
            manifest: Some(crate::skills::manifest::PluginManifest {
                name: plugin.to_string(),
                version: None,
                description: None,
                keybinds: vec![],
                compatibility: None,
                commands: vec![],
                extension: None,
                help_entries: vec![],
                provides: None,
                settings: Some(ManifestSettings {
                    categories: vec![ManifestSettingsCategory {
                        id: "demo".to_string(),
                        label: "Demo".to_string(),
                        fields: vec![
                            ManifestSettingsField {
                                key: "backend".to_string(),
                                label: "Backend".to_string(),
                                editor: ManifestEditorKind::Cycler,
                                options: vec!["a".to_string(), "b".to_string()],
                                help: None,
                                default: None,
                                numeric: false,
                            },
                            ManifestSettingsField {
                                key: "endpoint".to_string(),
                                label: "Endpoint".to_string(),
                                editor: ManifestEditorKind::Text,
                                options: vec![],
                                help: Some("URL".to_string()),
                                default: None,
                                numeric: false,
                            },
                        ],
                    }],
                }),
            }),
        }
    }

    #[test]
    fn plugin_settings_categories_exposed_after_rebuild() {
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_settings("demo-plugin", PathBuf::from("/tmp/demo"))],
        );
        let cats = r.plugin_settings_categories();
        assert_eq!(cats.len(), 1, "expected one plugin settings category");
        let cat = &cats[0];
        assert_eq!(cat.plugin, "demo-plugin");
        assert_eq!(cat.id, "demo");
        assert_eq!(cat.label, "Demo");
        assert_eq!(cat.fields.len(), 2);

        match &cat.fields[0].editor {
            PluginSettingsEditor::Cycler { options } => {
                assert_eq!(options, &vec!["a".to_string(), "b".to_string()]);
            }
            other => panic!("expected cycler, got {other:?}"),
        }
        assert!(matches!(
            cat.fields[1].editor,
            PluginSettingsEditor::Text { numeric: false }
        ));
        assert_eq!(cat.fields[1].help.as_deref(), Some("URL"));
    }

    #[test]
    fn plugin_settings_categories_empty_without_settings_block() {
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_cmd("p", "hello", PathBuf::from("/tmp/p"))],
        );
        assert!(r.plugin_settings_categories().is_empty());
    }

    #[test]
    fn plugin_settings_categories_replaced_on_rebuild() {
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_settings("demo-plugin", PathBuf::from("/tmp/demo"))],
        );
        assert_eq!(r.plugin_settings_categories().len(), 1);
        r.rebuild_with_plugins(vec![], vec![]);
        assert!(r.plugin_settings_categories().is_empty());
    }

    #[test]
    fn plugin_settings_categories_does_not_hardcode_capture() {
        // Acceptance: declarative cycler/text fields are represented in
        // core data without plugin-specific knowledge.
        let r = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_settings("totally-unrelated", PathBuf::from("/tmp/x"))],
        );
        let cats = r.plugin_settings_categories();
        assert_eq!(cats[0].plugin, "totally-unrelated");
        assert_eq!(cats[0].id, "demo");
        assert!(cats[0].fields.iter().any(|f| matches!(
            f.editor,
            PluginSettingsEditor::Cycler { .. }
        )));
        assert!(cats[0].fields.iter().any(|f| matches!(
            f.editor,
            PluginSettingsEditor::Text { .. }
        )));
    }

    // ---- Phase 8 slice 8A: lifecycle claims ----

    fn mk_plugin_with_lifecycle(
        plugin: &str,
        command: &str,
        display: Option<&str>,
        importance: i32,
        settings_category: Option<&str>,
    ) -> Plugin {
        use crate::skills::manifest::{
            PluginManifest, PluginProvides, SidecarLifecycle, SidecarManifest,
        };
        Plugin {
            name: plugin.to_string(),
            root: PathBuf::from(format!("/tmp/{plugin}")),
            marketplace: None,
            version: None,
            description: None,
            extension: None,
            manifest: Some(PluginManifest {
                name: plugin.to_string(),
                version: None,
                description: None,
                keybinds: vec![],
                compatibility: None,
                commands: vec![],
                extension: None,
                help_entries: vec![],
                provides: Some(PluginProvides {
                    sidecar: Some(SidecarManifest {
                        command: "bin/run".to_string(),
                        setup: None,
                        protocol_version: 1,
                        model: None,
                        lifecycle: Some(SidecarLifecycle {
                            command: command.to_string(),
                            settings_category: settings_category.map(str::to_string),
                            display_name: display.map(str::to_string),
                            importance,
                        }),
                    }),
                }),
                settings: None,
            }),
        }
    }

    #[test]
    fn lifecycle_claim_registers_under_command_word() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_lifecycle(
                "sample-sidecar",
                "capture",
                Some("Sample"),
                50,
                Some("capture"),
            )],
        );
        let claim = reg
            .lifecycle_for_command("capture")
            .expect("sample lifecycle claim should be registered");
        assert_eq!(claim.plugin, "sample-sidecar");
        assert_eq!(claim.command, "capture");
        assert_eq!(claim.display_name, "Sample");
        assert_eq!(claim.importance, 50);
        assert_eq!(claim.settings_category.as_deref(), Some("capture"));
    }

    #[test]
    fn lifecycle_claim_display_name_falls_back_to_command() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_lifecycle("p", "ocr", None, 0, None)],
        );
        let claim = reg.lifecycle_for_command("ocr").unwrap();
        assert_eq!(claim.display_name, "ocr");
    }

    #[test]
    fn lifecycle_claim_surfaces_in_all_commands() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_lifecycle("sample-sidecar", "capture", None, 0, None)],
        );
        assert!(reg.all_commands().contains(&"capture".to_string()));
    }

    #[test]
    fn lifecycle_claim_collision_first_loaded_wins() {
        // Two plugins both claim "capture"; first in the discovery
        // order (the vec we pass) should win.
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![
                mk_plugin_with_lifecycle("alpha-sidecar", "capture", Some("Alpha"), 10, None),
                mk_plugin_with_lifecycle("beta-sidecar", "capture", Some("Beta"), 90, None),
            ],
        );
        let claim = reg.lifecycle_for_command("capture").unwrap();
        assert_eq!(claim.plugin, "alpha-sidecar");
        let collisions = reg.lifecycle_claim_collisions();
        assert_eq!(collisions.len(), 1);
        assert_eq!(collisions[0], (
            "beta-sidecar".to_string(),
            "capture".to_string(),
            "alpha-sidecar".to_string(),
        ));
    }

    #[test]
    fn lifecycle_claims_returns_all_unique_command_words() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![
                mk_plugin_with_lifecycle("sample-sidecar", "capture", None, 50, None),
                mk_plugin_with_lifecycle("ocr-plugin", "ocr", None, 30, None),
            ],
        );
        let claims = reg.lifecycle_claims();
        let mut names: Vec<_> = claims.iter().map(|c| c.command.as_str()).collect();
        names.sort();
        assert_eq!(names, vec!["capture", "ocr"]);
    }

    #[test]
    fn lifecycle_for_command_returns_none_when_no_claim() {
        let reg = CommandRegistry::new_with_plugins(&[], vec![], vec![]);
        assert!(reg.lifecycle_for_command("capture").is_none());
    }

    #[test]
    fn rebuild_replaces_lifecycle_claims_atomically() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_with_lifecycle("sample-sidecar", "capture", None, 0, None)],
        );
        assert!(reg.lifecycle_for_command("capture").is_some());
        // Rebuild without the plugin: the claim must vanish.
        reg.rebuild_with_plugins(vec![], vec![]);
        assert!(reg.lifecycle_for_command("capture").is_none());
        assert!(reg.lifecycle_claim_collisions().is_empty());
    }

    // ---- Phase 8 slice 8A.3: virtual toggle-key field injection ----

    fn mk_plugin_lifecycle_plus_settings(
        plugin: &str,
        command: &str,
        lifecycle_settings_category: Option<&str>,
        category_ids: &[&str],
    ) -> Plugin {
        use crate::skills::manifest::{
            ManifestEditorKind, ManifestSettings, ManifestSettingsCategory,
            ManifestSettingsField, PluginManifest, PluginProvides, SidecarLifecycle,
            SidecarManifest,
        };
        Plugin {
            name: plugin.to_string(),
            root: PathBuf::from(format!("/tmp/{plugin}")),
            marketplace: None,
            version: None,
            description: None,
            extension: None,
            manifest: Some(PluginManifest {
                name: plugin.to_string(),
                version: None,
                description: None,
                keybinds: vec![],
                compatibility: None,
                commands: vec![],
                extension: None,
                help_entries: vec![],
                provides: Some(PluginProvides {
                    sidecar: Some(SidecarManifest {
                        command: "bin/run".to_string(),
                        setup: None,
                        protocol_version: 1,
                        model: None,
                        lifecycle: Some(SidecarLifecycle {
                            command: command.to_string(),
                            settings_category: lifecycle_settings_category.map(str::to_string),
                            display_name: None,
                            importance: 0,
                        }),
                    }),
                }),
                settings: Some(ManifestSettings {
                    categories: category_ids
                        .iter()
                        .map(|id| ManifestSettingsCategory {
                            id: id.to_string(),
                            label: id.to_string(),
                            fields: vec![ManifestSettingsField {
                                key: "existing".to_string(),
                                label: "Existing".to_string(),
                                editor: ManifestEditorKind::Text,
                                options: vec![],
                                help: None,
                                default: None,
                                numeric: false,
                            }],
                        })
                        .collect(),
                }),
            }),
        }
    }

    #[test]
    fn lifecycle_injects_virtual_toggle_key_into_matching_category() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_lifecycle_plus_settings(
                "sample-sidecar",
                "capture",
                Some("capture"),
                &["capture"],
            )],
        );
        let cats = reg.plugin_settings_categories();
        let capture = cats
            .iter()
            .find(|c| c.id == "capture" && c.plugin == "sample-sidecar")
            .expect("sample category present");
        assert!(!capture.fields.is_empty());
        let first = &capture.fields[0];
        assert_eq!(first.key, "_lifecycle_toggle_key");
        assert_eq!(first.label, "Toggle key");
        match &first.editor {
            PluginSettingsEditor::Cycler { options } => {
                assert_eq!(
                    options,
                    &vec![
                        "F8".to_string(),
                        "F2".to_string(),
                        "F12".to_string(),
                        "C-V".to_string(),
                        "C-G".to_string(),
                    ]
                );
            }
            other => panic!("expected cycler, got {other:?}"),
        }
        assert_eq!(capture.fields[1].key, "existing");
    }

    #[test]
    fn lifecycle_no_injection_when_settings_category_is_none() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_lifecycle_plus_settings("p", "ocr", None, &["capture"])],
        );
        let cats = reg.plugin_settings_categories();
        let capture = cats.iter().find(|c| c.id == "capture").expect("category");
        assert!(capture.fields.iter().all(|f| f.key != "_lifecycle_toggle_key"));
    }

    #[test]
    fn lifecycle_no_injection_when_category_does_not_exist() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![mk_plugin_lifecycle_plus_settings(
                "p",
                "capture",
                Some("nonexistent"),
                &["capture"],
            )],
        );
        let cats = reg.plugin_settings_categories();
        for c in &cats {
            assert!(c.fields.iter().all(|f| f.key != "_lifecycle_toggle_key"));
        }
    }

    #[test]
    fn lifecycle_two_plugins_each_get_injection_in_their_own_category() {
        let reg = CommandRegistry::new_with_plugins(
            &[],
            vec![],
            vec![
                mk_plugin_lifecycle_plus_settings(
                    "sidecar-plugin",
                    "capture",
                    Some("capture"),
                    &["capture"],
                ),
                mk_plugin_lifecycle_plus_settings(
                    "ocr-plugin",
                    "ocr",
                    Some("ocr"),
                    &["ocr"],
                ),
            ],
        );
        let cats = reg.plugin_settings_categories();
        let capture = cats.iter().find(|c| c.plugin == "sidecar-plugin").unwrap();
        let ocr = cats.iter().find(|c| c.plugin == "ocr-plugin").unwrap();
        assert_eq!(capture.fields[0].key, "_lifecycle_toggle_key");
        assert_eq!(ocr.fields[0].key, "_lifecycle_toggle_key");
    }
}