@cryptiklemur/lattice 4.0.2 → 5.0.1

package/dist/server/handlers/plugins.js

@@ -0,0 +1,593 @@
+import { readFileSync, existsSync, readdirSync } from "node:fs";
+import { join } from "node:path";
+import { homedir } from "node:os";
+import { execSync, spawn } from "node:child_process";
+import { registerHandler } from "../ws/router.js";
+import { sendTo } from "../ws/broadcast.js";
+var PLUGINS_DIR = join(homedir(), ".claude", "plugins");
+var INSTALLED_FILE = join(PLUGINS_DIR, "installed_plugins.json");
+var MARKETPLACES_FILE = join(PLUGINS_DIR, "known_marketplaces.json");
+var INSTALL_COUNTS_FILE = join(PLUGINS_DIR, "install-counts-cache.json");
+function readJsonFile(path) {
+    try {
+        if (!existsSync(path))
+            return null;
+        return JSON.parse(readFileSync(path, "utf-8"));
+    }
+    catch {
+        return null;
+    }
+}
+function getInstallCounts() {
+    var data = readJsonFile(INSTALL_COUNTS_FILE);
+    var map = new Map();
+    if (!data || !data.counts)
+        return map;
+    for (var i = 0; i < data.counts.length; i++) {
+        map.set(data.counts[i].plugin, data.counts[i].unique_installs);
+    }
+    return map;
+}
+function readPluginJson(installPath) {
+    var pluginJsonPath = join(installPath, ".claude-plugin", "plugin.json");
+    return readJsonFile(pluginJsonPath);
+}
+function countSkills(installPath) {
+    var skillsDir = join(installPath, "skills");
+    if (!existsSync(skillsDir))
+        return 0;
+    try {
+        var entries = readdirSync(skillsDir, { withFileTypes: true });
+        var count = 0;
+        for (var i = 0; i < entries.length; i++) {
+            if (entries[i].isDirectory())
+                count++;
+        }
+        return count;
+    }
+    catch {
+        return 0;
+    }
+}
+function countHooks(installPath) {
+    var hooksPath = join(installPath, "hooks", "hooks.json");
+    var data = readJsonFile(hooksPath);
+    if (!data || !data.hooks)
+        return 0;
+    return Object.keys(data.hooks).length;
+}
+function countRules(installPath) {
+    var rulesDir = join(installPath, "rules");
+    if (!existsSync(rulesDir))
+        return 0;
+    try {
+        var entries = readdirSync(rulesDir);
+        var count = 0;
+        for (var i = 0; i < entries.length; i++) {
+            if (entries[i].endsWith(".md"))
+                count++;
+        }
+        return count;
+    }
+    catch {
+        return 0;
+    }
+}
+function getInstalledPlugins() {
+    var data = readJsonFile(INSTALLED_FILE);
+    if (!data || !data.plugins)
+        return [];
+    var installCounts = getInstallCounts();
+    var plugins = [];
+    var keys = Object.keys(data.plugins);
+    for (var k = 0; k < keys.length; k++) {
+        var key = keys[k];
+        var entries = data.plugins[key];
+        var atIdx = key.lastIndexOf("@");
+        var pluginName = atIdx > 0 ? key.slice(0, atIdx) : key;
+        var marketplace = atIdx > 0 ? key.slice(atIdx + 1) : "";
+        for (var e = 0; e < entries.length; e++) {
+            var entry = entries[e];
+            var meta = readPluginJson(entry.installPath);
+            plugins.push({
+                name: pluginName,
+                marketplace: marketplace,
+                key: key,
+                version: entry.version,
+                scope: entry.scope,
+                installPath: entry.installPath,
+                installedAt: entry.installedAt,
+                lastUpdated: entry.lastUpdated,
+                gitCommitSha: entry.gitCommitSha,
+                description: meta?.description ?? "",
+                skillCount: countSkills(entry.installPath),
+                hookCount: countHooks(entry.installPath),
+                ruleCount: countRules(entry.installPath),
+                installs: installCounts.get(key),
+            });
+        }
+    }
+    plugins.sort(function (a, b) { return a.name.localeCompare(b.name); });
+    return plugins;
+}
+function getMarketplaces() {
+    var data = readJsonFile(MARKETPLACES_FILE);
+    if (!data)
+        return [];
+    var result = [];
+    var keys = Object.keys(data);
+    for (var i = 0; i < keys.length; i++) {
+        var entry = data[keys[i]];
+        result.push({
+            name: keys[i],
+            source: entry.source,
+            installLocation: entry.installLocation,
+            lastUpdated: entry.lastUpdated,
+        });
+    }
+    return result;
+}
+function searchMarketplacePlugins(query, marketplaceFilter) {
+    var marketplaces = readJsonFile(MARKETPLACES_FILE);
+    if (!marketplaces)
+        return [];
+    var installCounts = getInstallCounts();
+    var installedData = readJsonFile(INSTALLED_FILE);
+    var installedKeys = new Set();
+    var installedVersions = new Map();
+    if (installedData && installedData.plugins) {
+        var iKeys = Object.keys(installedData.plugins);
+        for (var ik = 0; ik < iKeys.length; ik++) {
+            installedKeys.add(iKeys[ik]);
+            var versions = installedData.plugins[iKeys[ik]];
+            if (versions.length > 0) {
+                installedVersions.set(iKeys[ik], versions[0].version);
+            }
+        }
+    }
+    var results = [];
+    var lowerQuery = query.toLowerCase();
+    var mKeys = Object.keys(marketplaces);
+    for (var m = 0; m < mKeys.length; m++) {
+        var mName = mKeys[m];
+        if (marketplaceFilter && mName !== marketplaceFilter)
+            continue;
+        var mkt = marketplaces[mName];
+        var pluginsDir = join(mkt.installLocation, "plugins");
+        if (!existsSync(pluginsDir))
+            continue;
+        try {
+            var pluginDirs = readdirSync(pluginsDir, { withFileTypes: true });
+            for (var p = 0; p < pluginDirs.length; p++) {
+                if (!pluginDirs[p].isDirectory())
+                    continue;
+                var dirName = pluginDirs[p].name;
+                if (dirName.toLowerCase().indexOf(lowerQuery) === -1) {
+                    var meta = readPluginJson(join(pluginsDir, dirName));
+                    if (!meta || meta.description.toLowerCase().indexOf(lowerQuery) === -1) {
+                        continue;
+                    }
+                }
+                var pluginMeta = readPluginJson(join(pluginsDir, dirName));
+                var key = dirName + "@" + mName;
+                results.push({
+                    name: dirName,
+                    marketplace: mName,
+                    description: pluginMeta?.description ?? "",
+                    author: pluginMeta?.author,
+                    installed: installedKeys.has(key),
+                    installedVersion: installedVersions.get(key),
+                    installs: installCounts.get(key),
+                });
+            }
+        }
+        catch { }
+    }
+    results.sort(function (a, b) {
+        var ai = a.installs ?? 0;
+        var bi = b.installs ?? 0;
+        return bi - ai;
+    });
+    return results;
+}
+function parseFrontmatter(content) {
+    var match = content.match(/^---\r?\n([\s\S]*?)\r?\n---/);
+    if (!match)
+        return { name: "", description: "" };
+    var yaml = match[1];
+    var name = "";
+    var desc = "";
+    var lines = yaml.split(/\r?\n/);
+    for (var i = 0; i < lines.length; i++) {
+        var line = lines[i];
+        var nameMatch = line.match(/^name:\s*(.+)/);
+        if (nameMatch)
+            name = nameMatch[1].trim().replace(/^["']|["']$/g, "");
+        var descMatch = line.match(/^description:\s*(.+)/);
+        if (descMatch)
+            desc = descMatch[1].trim().replace(/^["']|["']$/g, "");
+    }
+    return { name, description: desc };
+}
+function getPluginDetails(pluginName, marketplace) {
+    var data = readJsonFile(INSTALLED_FILE);
+    if (!data || !data.plugins)
+        return null;
+    var key = pluginName + "@" + marketplace;
+    var entries = data.plugins[key];
+    if (!entries || entries.length === 0)
+        return null;
+    var entry = entries[0];
+    var meta = readPluginJson(entry.installPath);
+    var skills = [];
+    var skillsDir = join(entry.installPath, "skills");
+    if (existsSync(skillsDir)) {
+        try {
+            var skillDirs = readdirSync(skillsDir, { withFileTypes: true });
+            for (var s = 0; s < skillDirs.length; s++) {
+                if (!skillDirs[s].isDirectory())
+                    continue;
+                var skillFile = join(skillsDir, skillDirs[s].name, "SKILL.md");
+                if (existsSync(skillFile)) {
+                    var content = readFileSync(skillFile, "utf-8");
+                    var fm = parseFrontmatter(content);
+                    skills.push({ name: fm.name || skillDirs[s].name, description: fm.description });
+                }
+            }
+        }
+        catch { }
+    }
+    var hooks = {};
+    var hooksPath = join(entry.installPath, "hooks", "hooks.json");
+    var hooksData = readJsonFile(hooksPath);
+    if (hooksData && hooksData.hooks) {
+        hooks = hooksData.hooks;
+    }
+    var rules = [];
+    var rulesDir = join(entry.installPath, "rules");
+    if (existsSync(rulesDir)) {
+        try {
+            var ruleFiles = readdirSync(rulesDir);
+            for (var r = 0; r < ruleFiles.length; r++) {
+                if (ruleFiles[r].endsWith(".md")) {
+                    rules.push(ruleFiles[r]);
+                }
+            }
+        }
+        catch { }
+    }
+    return {
+        name: pluginName,
+        marketplace: marketplace,
+        version: entry.version,
+        description: meta?.description ?? "",
+        author: meta?.author,
+        homepage: meta?.homepage,
+        license: meta?.license,
+        keywords: meta?.keywords,
+        skills: skills,
+        hooks: hooks,
+        rules: rules,
+        installPath: entry.installPath,
+        installedAt: entry.installedAt,
+        lastUpdated: entry.lastUpdated,
+        gitCommitSha: entry.gitCommitSha,
+    };
+}
+export function getPluginMcpServers() {
+    var data = readJsonFile(INSTALLED_FILE);
+    if (!data || !data.plugins)
+        return {};
+    var servers = {};
+    var keys = Object.keys(data.plugins);
+    for (var k = 0; k < keys.length; k++) {
+        var entries = data.plugins[keys[k]];
+        for (var e = 0; e < entries.length; e++) {
+            var mcpPath = join(entries[e].installPath, ".mcp.json");
+            var mcpData = readJsonFile(mcpPath);
+            if (mcpData && mcpData.mcpServers) {
+                var sKeys = Object.keys(mcpData.mcpServers);
+                for (var s = 0; s < sKeys.length; s++) {
+                    servers[sKeys[s]] = mcpData.mcpServers[sKeys[s]];
+                }
+            }
+        }
+    }
+    return servers;
+}
+export function getInstalledPluginCount() {
+    var data = readJsonFile(INSTALLED_FILE);
+    if (!data || !data.plugins)
+        return 0;
+    return Object.keys(data.plugins).length;
+}
+export function getPluginSkillRuleTokenEstimate() {
+    var data = readJsonFile(INSTALLED_FILE);
+    if (!data || !data.plugins)
+        return 0;
+    var totalChars = 0;
+    var keys = Object.keys(data.plugins);
+    for (var k = 0; k < keys.length; k++) {
+        var entries = data.plugins[keys[k]];
+        for (var e = 0; e < entries.length; e++) {
+            var skillsDir = join(entries[e].installPath, "skills");
+            if (existsSync(skillsDir)) {
+                try {
+                    var dirs = readdirSync(skillsDir, { withFileTypes: true });
+                    for (var d = 0; d < dirs.length; d++) {
+                        if (!dirs[d].isDirectory())
+                            continue;
+                        var skillFile = join(skillsDir, dirs[d].name, "SKILL.md");
+                        if (existsSync(skillFile)) {
+                            totalChars += readFileSync(skillFile, "utf-8").length;
+                        }
+                    }
+                }
+                catch { }
+            }
+            var rulesDir = join(entries[e].installPath, "rules");
+            if (existsSync(rulesDir)) {
+                try {
+                    var ruleFiles = readdirSync(rulesDir);
+                    for (var r = 0; r < ruleFiles.length; r++) {
+                        if (ruleFiles[r].endsWith(".md")) {
+                            totalChars += readFileSync(join(rulesDir, ruleFiles[r]), "utf-8").length;
+                        }
+                    }
+                }
+                catch { }
+            }
+        }
+    }
+    return Math.round(totalChars / 4);
+}
+function whichBinary(name) {
+    try {
+        execSync("which " + JSON.stringify(name), { encoding: "utf-8", timeout: 3000, stdio: ["pipe", "pipe", "pipe"] });
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+var LSP_BINARY_MAP = {
+    "typescript-lsp": "typescript-language-server",
+    "pyright-lsp": "pyright-langserver",
+    "gopls-lsp": "gopls",
+    "rust-analyzer-lsp": "rust-analyzer",
+    "clangd-lsp": "clangd",
+    "csharp-lsp": "OmniSharp",
+    "ruby-lsp": "ruby-lsp",
+    "swift-lsp": "sourcekit-lsp",
+    "kotlin-lsp": "kotlin-language-server",
+    "lua-lsp": "lua-language-server",
+    "php-lsp": "phpactor",
+    "jdtls-lsp": "jdtls",
+};
+function getPluginErrors() {
+    var data = readJsonFile(INSTALLED_FILE);
+    if (!data || !data.plugins)
+        return [];
+    var errors = [];
+    var keys = Object.keys(data.plugins);
+    for (var k = 0; k < keys.length; k++) {
+        var key = keys[k];
+        var entries = data.plugins[key];
+        var atIdx = key.lastIndexOf("@");
+        var pluginName = atIdx > 0 ? key.slice(0, atIdx) : key;
+        var marketplace = atIdx > 0 ? key.slice(atIdx + 1) : "";
+        for (var e = 0; e < entries.length; e++) {
+            var entry = entries[e];
+            var errs = [];
+            if (!existsSync(entry.installPath)) {
+                errs.push("Install path does not exist: " + entry.installPath);
+            }
+            else {
+                try {
+                    var output = execSync("claude plugin validate " + JSON.stringify(entry.installPath) + " 2>&1", { encoding: "utf-8", timeout: 10000 });
+                    var errorLines = output.split("\n").filter(function (l) { return l.trim().startsWith("❯") || l.trim().startsWith("✘"); });
+                    for (var el = 0; el < errorLines.length; el++) {
+                        var line = errorLines[el].trim();
+                        if (line.startsWith("❯")) {
+                            errs.push(line.slice(1).trim());
+                        }
+                    }
+                }
+                catch (validateErr) {
+                    var stderr = String(validateErr);
+                    var stderrLines = stderr.split("\n").filter(function (l) { return l.trim().startsWith("❯"); });
+                    for (var sl = 0; sl < stderrLines.length; sl++) {
+                        errs.push(stderrLines[sl].trim().slice(1).trim());
+                    }
+                }
+                var lspBinary = LSP_BINARY_MAP[pluginName];
+                if (lspBinary && !whichBinary(lspBinary)) {
+                    errs.push("Executable not found in $PATH: \"" + lspBinary + "\"");
+                }
+            }
+            if (errs.length > 0) {
+                errors.push({ key: key, name: pluginName, marketplace: marketplace, errors: errs });
+            }
+        }
+    }
+    return errors;
+}
+function discoverPlugins() {
+    var marketplaces = readJsonFile(MARKETPLACES_FILE);
+    if (!marketplaces)
+        return [];
+    var installCounts = getInstallCounts();
+    var installedData = readJsonFile(INSTALLED_FILE);
+    var installedKeys = new Set();
+    var installedVersions = new Map();
+    if (installedData && installedData.plugins) {
+        var iKeys = Object.keys(installedData.plugins);
+        for (var ik = 0; ik < iKeys.length; ik++) {
+            installedKeys.add(iKeys[ik]);
+            var versions = installedData.plugins[iKeys[ik]];
+            if (versions.length > 0) {
+                installedVersions.set(iKeys[ik], versions[0].version);
+            }
+        }
+    }
+    var results = [];
+    var mKeys = Object.keys(marketplaces);
+    for (var m = 0; m < mKeys.length; m++) {
+        var mName = mKeys[m];
+        var mkt = marketplaces[mName];
+        var pluginsDir = join(mkt.installLocation, "plugins");
+        if (!existsSync(pluginsDir))
+            continue;
+        try {
+            var pluginDirs = readdirSync(pluginsDir, { withFileTypes: true });
+            for (var p = 0; p < pluginDirs.length; p++) {
+                if (!pluginDirs[p].isDirectory())
+                    continue;
+                var dirName = pluginDirs[p].name;
+                var meta = readPluginJson(join(pluginsDir, dirName));
+                var key = dirName + "@" + mName;
+                results.push({
+                    name: dirName,
+                    marketplace: mName,
+                    description: meta?.description ?? "",
+                    author: meta?.author,
+                    installed: installedKeys.has(key),
+                    installedVersion: installedVersions.get(key),
+                    installs: installCounts.get(key),
+                });
+            }
+        }
+        catch { }
+    }
+    results.sort(function (a, b) {
+        var ai = a.installs ?? 0;
+        var bi = b.installs ?? 0;
+        return bi - ai;
+    });
+    return results;
+}
+registerHandler("plugin", function (clientId, message) {
+    if (message.type === "plugin:list") {
+        var plugins = getInstalledPlugins();
+        sendTo(clientId, { type: "plugin:list_result", plugins: plugins });
+        return;
+    }
+    if (message.type === "plugin:marketplaces") {
+        var marketplaces = getMarketplaces();
+        sendTo(clientId, { type: "plugin:marketplaces_result", marketplaces: marketplaces });
+        return;
+    }
+    if (message.type === "plugin:search") {
+        var searchMsg = message;
+        var query = searchMsg.query.trim();
+        if (!query) {
+            sendTo(clientId, { type: "plugin:search_result", query: query, plugins: [], count: 0 });
+            return;
+        }
+        var found = searchMarketplacePlugins(query, searchMsg.marketplace);
+        sendTo(clientId, { type: "plugin:search_result", query: query, plugins: found, count: found.length });
+        return;
+    }
+    if (message.type === "plugin:install") {
+        var installMsg = message;
+        var installArg = installMsg.name + "@" + installMsg.marketplace;
+        try {
+            var proc = spawn("claude", ["plugin", "install", installArg], {
+                cwd: homedir(),
+                stdio: ["ignore", "pipe", "pipe"],
+            });
+            var timeout = setTimeout(function () {
+                proc.kill();
+            }, 120000);
+            proc.on("close", function (code) {
+                clearTimeout(timeout);
+                if (code === 0) {
+                    sendTo(clientId, { type: "plugin:install_result", success: true, message: "Installed " + installMsg.name + " successfully" });
+                    sendTo(clientId, { type: "plugin:list_result", plugins: getInstalledPlugins() });
+                }
+                else {
+                    sendTo(clientId, { type: "plugin:install_result", success: false, message: "Install failed (exit code " + code + ")" });
+                }
+            });
+        }
+        catch (err) {
+            sendTo(clientId, { type: "plugin:install_result", success: false, message: "Failed to start install: " + String(err) });
+        }
+        return;
+    }
+    if (message.type === "plugin:uninstall") {
+        var uninstallMsg = message;
+        var uninstallArg = uninstallMsg.name + "@" + uninstallMsg.marketplace;
+        try {
+            var uninstallProc = spawn("claude", ["plugin", "uninstall", uninstallArg], {
+                cwd: homedir(),
+                stdio: ["ignore", "pipe", "pipe"],
+            });
+            var uninstallTimeout = setTimeout(function () {
+                uninstallProc.kill();
+            }, 60000);
+            uninstallProc.on("close", function (code) {
+                clearTimeout(uninstallTimeout);
+                if (code === 0) {
+                    sendTo(clientId, { type: "plugin:uninstall_result", success: true, message: "Uninstalled " + uninstallMsg.name + " successfully" });
+                    sendTo(clientId, { type: "plugin:list_result", plugins: getInstalledPlugins() });
+                }
+                else {
+                    sendTo(clientId, { type: "plugin:uninstall_result", success: false, message: "Uninstall failed (exit code " + code + ")" });
+                }
+            });
+        }
+        catch (err) {
+            sendTo(clientId, { type: "plugin:uninstall_result", success: false, message: "Failed to start uninstall: " + String(err) });
+        }
+        return;
+    }
+    if (message.type === "plugin:update") {
+        var updateMsg = message;
+        var updateArg = updateMsg.name + "@" + updateMsg.marketplace;
+        try {
+            var updateProc = spawn("claude", ["plugin", "update", updateArg], {
+                cwd: homedir(),
+                stdio: ["ignore", "pipe", "pipe"],
+            });
+            var updateTimeout = setTimeout(function () {
+                updateProc.kill();
+            }, 120000);
+            updateProc.on("close", function (code) {
+                clearTimeout(updateTimeout);
+                if (code === 0) {
+                    sendTo(clientId, { type: "plugin:update_result", success: true, message: "Updated " + updateMsg.name + " successfully" });
+                    sendTo(clientId, { type: "plugin:list_result", plugins: getInstalledPlugins() });
+                }
+                else {
+                    sendTo(clientId, { type: "plugin:update_result", success: false, message: "Update failed (exit code " + code + ")" });
+                }
+            });
+        }
+        catch (err) {
+            sendTo(clientId, { type: "plugin:update_result", success: false, message: "Failed to start update: " + String(err) });
+        }
+        return;
+    }
+    if (message.type === "plugin:details") {
+        var detailsMsg = message;
+        var details = getPluginDetails(detailsMsg.name, detailsMsg.marketplace);
+        if (details) {
+            sendTo(clientId, { type: "plugin:details_result", plugin: details });
+        }
+        else {
+            sendTo(clientId, { type: "plugin:details_result", plugin: null, error: "Plugin not found" });
+        }
+        return;
+    }
+    if (message.type === "plugin:discover") {
+        var allPlugins = discoverPlugins();
+        sendTo(clientId, { type: "plugin:discover_result", plugins: allPlugins });
+        return;
+    }
+    if (message.type === "plugin:errors") {
+        var pluginErrors = getPluginErrors();
+        sendTo(clientId, { type: "plugin:errors_result", errors: pluginErrors });
+        return;
+    }
+});