Merge pull request #17 from RealZST/fix/extension-management

fix: plugin toggle, delete cleanup, and Gemini native extension support

Author: RealZST

.gitignore

@@ -1,6 +1,7 @@
 /target
 /dist
 /node_modules
+/social
 *.swp
 *.swo
 .DS_Store

crates/hk-core/src/adapter/claude.rs

@@ -325,6 +325,7 @@ impl AgentAdapter for ClaudeAdapter {
                 source: source.clone(),
                 enabled: enabled_set.contains(key),
                 path: install_path,
+                uri: None,
                 installed_at,
                 updated_at,
             });

crates/hk-core/src/adapter/codex.rs

@@ -203,6 +203,23 @@ impl AgentAdapter for CodexAdapter {
     }
 
     fn read_plugins(&self) -> Vec<PluginEntry> {
+        // Read disabled plugins from config.toml [plugins."name@source"] enabled = false
+        let disabled_plugins: std::collections::HashSet<String> = std::fs::read_to_string(self.mcp_config_path()).ok()
+            .and_then(|content| content.parse::<toml::Table>().ok())
+            .and_then(|doc| doc.get("plugins").and_then(|v| v.as_table()).cloned())
+            .map(|plugins| {
+                plugins.into_iter()
+                    .filter(|(_, v)| {
+                        v.as_table()
+                            .and_then(|t| t.get("enabled"))
+                            .and_then(|e| e.as_bool())
+                            == Some(false)
+                    })
+                    .map(|(k, _)| k)
+                    .collect()
+            })
+            .unwrap_or_default();
+
         // Codex plugins are cached at ~/.codex/plugins/cache/{marketplace}/{plugin}/{version}/
         // Each has .codex-plugin/plugin.json manifest
         let cache_dir = self.base_dir().join("plugins").join("cache");
@@ -261,10 +278,11 @@ impl AgentAdapter for CodexAdapter {
                         plugin_name.clone()
                     };
                     entries.push(PluginEntry {
-                        name,
+                        name: name.clone(),
                         source: marketplace_name.clone(),
-                        enabled: true,
+                        enabled: !disabled_plugins.contains(&format!("{}@{}", name, &marketplace_name)),
                         path: Some(version_dir.path().to_path_buf()), // version level — matches manifest location
+                        uri: None,
                         installed_at: None,
                         updated_at: None,
                     });
@@ -408,6 +426,33 @@ mod tests {
         assert!(plugins.is_empty());
     }
 
+    #[test]
+    fn read_plugins_respects_config_toml_enabled() {
+        let tmp = tempfile::tempdir().unwrap();
+        let adapter = CodexAdapter::with_home(tmp.path().to_path_buf());
+
+        // Set up a plugin in cache
+        let plugin_dir = tmp.path().join(".codex/plugins/cache/test-marketplace/my-plugin/1.0.0/.codex-plugin");
+        fs::create_dir_all(&plugin_dir).unwrap();
+        fs::write(plugin_dir.join("plugin.json"), r#"{"name":"my-plugin"}"#).unwrap();
+
+        // No config.toml → plugin should be enabled (default)
+        let entries = adapter.read_plugins();
+        assert_eq!(entries.len(), 1);
+        assert!(entries[0].enabled);
+
+        // config.toml with plugin disabled
+        let config_path = tmp.path().join(".codex/config.toml");
+        fs::write(&config_path, r#"
+[plugins."my-plugin@test-marketplace"]
+enabled = false
+"#).unwrap();
+
+        let entries = adapter.read_plugins();
+        assert_eq!(entries.len(), 1);
+        assert!(!entries[0].enabled, "Plugin should be disabled per config.toml");
+    }
+
     #[test]
     fn read_hooks_object_format() {
         let tmp = tempfile::tempdir().unwrap();

crates/hk-core/src/adapter/copilot.rs

@@ -16,7 +16,40 @@
 // Events: SessionStart, UserPromptSubmit, PreToolUse, PostToolUse, PreCompact, SubagentStart, SubagentStop, Stop
 
 use super::{AgentAdapter, HookEntry, HookFormat, McpFormat, McpServerEntry, PluginEntry};
-use std::path::PathBuf;
+use std::path::{Path, PathBuf};
+
+/// Read VS Code agent plugin enablement from state.vscdb.
+/// Returns the set of plugin URIs that are disabled (enabled = false).
+/// Gracefully returns empty set on any error (DB not found, locked, schema change).
+fn read_vscode_disabled_plugins(vscode_user_dir: &Path) -> std::collections::HashSet<String> {
+    let db_path = vscode_user_dir
+        .join("globalStorage")
+        .join("state.vscdb");
+    let mut disabled = std::collections::HashSet::new();
+    let Ok(conn) = rusqlite::Connection::open_with_flags(
+        &db_path,
+        rusqlite::OpenFlags::SQLITE_OPEN_READ_ONLY | rusqlite::OpenFlags::SQLITE_OPEN_NO_MUTEX,
+    ) else {
+        return disabled;
+    };
+    let Ok(value): Result<String, _> = conn.query_row(
+        "SELECT value FROM ItemTable WHERE key = 'agentPlugins.enablement'",
+        [],
+        |row| row.get(0),
+    ) else {
+        return disabled;
+    };
+    // Format: [["file:///path/to/plugin", true/false], ...]
+    let Ok(entries) = serde_json::from_str::<Vec<(String, bool)>>(&value) else {
+        return disabled;
+    };
+    for (uri, enabled) in entries {
+        if !enabled {
+            disabled.insert(uri);
+        }
+    }
+    disabled
+}
 
 pub struct CopilotAdapter {
     home: PathBuf,
@@ -67,6 +100,9 @@ impl AgentAdapter for CopilotAdapter {
     fn detect(&self) -> bool {
         self.base_dir().exists()
     }
+    fn vscode_user_dir(&self) -> Option<PathBuf> {
+        Some(self.vscode_user_dir())
+    }
     fn skill_dirs(&self) -> Vec<PathBuf> {
         vec![
             self.base_dir().join("skills"),
@@ -146,6 +182,7 @@ impl AgentAdapter for CopilotAdapter {
         let mut entries = Vec::new();
 
         // 1. Copilot CLI plugins: ~/.copilot/installed-plugins/{marketplace}/{plugin}/
+        //    CLI has no native enable/disable — always enabled if present.
         let cli_base = self.base_dir().join("installed-plugins");
         if let Ok(marketplaces) = std::fs::read_dir(&cli_base) {
             for marketplace in marketplaces.flatten() {
@@ -169,6 +206,7 @@ impl AgentAdapter for CopilotAdapter {
                         source: mp_name.clone(),
                         enabled: true,
                         path: Some(plugin.path()),
+                        uri: None,
                         installed_at: None,
                         updated_at: None,
                     });
@@ -178,8 +216,14 @@ impl AgentAdapter for CopilotAdapter {
 
         // 2. VS Code agent plugins: ~/.vscode/agent-plugins/installed.json
         //    Each entry has pluginUri pointing to plugins/{name}/ with .github/plugin/plugin.json
+        //    Enable/disable state stored in VS Code's state.vscdb under "agentPlugins.enablement".
         let vscode_base = self.home.join(".vscode").join("agent-plugins");
         let installed_json = vscode_base.join("installed.json");
+
+        // Read VS Code agent plugin enablement state from state.vscdb
+        // Format: [["file:///path/to/plugin", true/false], ...]
+        let disabled_uris = read_vscode_disabled_plugins(&self.vscode_user_dir());
+
         if let Ok(content) = std::fs::read_to_string(&installed_json)
             && let Ok(registry) = serde_json::from_str::<serde_json::Value>(&content)
             && let Some(installed) = registry.get("installed").and_then(|v| v.as_array())
@@ -206,11 +250,13 @@ impl AgentAdapter for CopilotAdapter {
                             .map(|n| n.to_string_lossy().to_string())
                             .unwrap_or_else(|| marketplace.to_string())
                     });
+                let enabled = !disabled_uris.contains(plugin_uri);
                 entries.push(PluginEntry {
                     name,
                     source: marketplace.to_string(),
-                    enabled: true,
+                    enabled,
                     path: Some(plugin_path),
+                    uri: Some(plugin_uri.to_string()),
                     installed_at: None,
                     updated_at: None,
                 });

crates/hk-core/src/adapter/cursor.rs

@@ -206,6 +206,7 @@ impl AgentAdapter for CursorAdapter {
                     source: "local".into(),
                     enabled: true,
                     path: Some(dir.path()),
+                    uri: None,
                     installed_at: None,
                     updated_at: None,
                 });
@@ -242,6 +243,7 @@ impl AgentAdapter for CursorAdapter {
                             source: mp_name.clone(),
                             enabled: true,
                             path: Some(plugin.path()),
+                            uri: None,
                             installed_at: None,
                             updated_at: None,
                         });