Step 143: index Emacs Elisp functions

2026-02-09 18:02:57 -07:00
parent beaec58073
commit 568fa38f46
10 changed files with 301 additions and 19 deletions
--- a/editor/src/EmacsFunctionDiscovery.h
+++ b/editor/src/EmacsFunctionDiscovery.h
@@ -0,0 +1,139 @@
+#pragma once
+#include "EmacsIntegration.h"
+#include "EmacsPackageBrowser.h"
+#include "LibraryIndexer.h"
+#include "ast/Module.h"
+#include "ast/ExternalModule.h"
+#include "ast/TypeSignature.h"
+#include <string>
+#include <vector>
+#include <unordered_map>
+#include <algorithm>
+
+struct EmacsFunctionDoc {
+    std::string name;
+    std::string signature;
+    std::string doc;
+};
+
+struct EmacsFunctionIndex {
+    std::unordered_map<std::string, std::vector<EmacsFunctionDoc>> functionsByPackage;
+};
+
+static std::vector<std::string> parseLines(const std::string& text) {
+    std::vector<std::string> out;
+    size_t start = 0;
+    while (start < text.size()) {
+        size_t end = text.find('\n', start);
+        if (end == std::string::npos) end = text.size();
+        std::string line = trimCopy(text.substr(start, end - start));
+        if (!line.empty()) out.push_back(std::move(line));
+        start = end + 1;
+    }
+    return out;
+}
+
+static std::vector<std::string> queryEmacsFunctions(EmacsConnection& emacs,
+                                                    const std::string& prefix,
+                                                    std::string& logOut) {
+    std::string cmd = ElispCommandBuilder::aproposFunctions(prefix);
+    std::string resp = emacs.sendCommand(cmd);
+    if (resp.empty() && !emacs.getLastError().empty()) {
+        logOut += "[emacs] " + emacs.getLastError() + "\n";
+        return {};
+    }
+    return parseLines(resp);
+}
+
+static EmacsFunctionDoc queryEmacsFunctionDoc(EmacsConnection& emacs,
+                                              const std::string& name,
+                                              std::string& logOut) {
+    EmacsFunctionDoc info;
+    info.name = name;
+    std::string cmd = ElispCommandBuilder::describeFunction(name);
+    std::string resp = emacs.sendCommand(cmd);
+    if (resp.empty() && !emacs.getLastError().empty()) {
+        logOut += "[emacs] " + emacs.getLastError() + "\n";
+        return info;
+    }
+    auto pos = resp.find('\n');
+    if (pos == std::string::npos) {
+        info.signature = trimCopy(resp);
+    } else {
+        info.signature = trimCopy(resp.substr(0, pos));
+        info.doc = trimCopy(resp.substr(pos + 1));
+    }
+    return info;
+}
+
+static std::vector<std::string> buildPackageFunctionPrefixes(const std::string& pkgName) {
+    std::vector<std::string> prefixes;
+    if (pkgName.empty()) return prefixes;
+    prefixes.push_back(pkgName + "-");
+    prefixes.push_back(pkgName);
+    auto dash = pkgName.find('-');
+    if (dash != std::string::npos) {
+        prefixes.push_back(pkgName.substr(0, dash) + "-");
+    }
+    return prefixes;
+}
+
+static void refreshEmacsFunctionIndex(EmacsFunctionIndex& index,
+                                      EmacsConnection& emacs,
+                                      const std::vector<EmacsPackageEntry>& packages,
+                                      std::string& logOut) {
+    index.functionsByPackage.clear();
+    for (const auto& pkg : packages) {
+        if (pkg.status != "loaded") continue;
+        std::vector<std::string> functions;
+        for (const auto& prefix : buildPackageFunctionPrefixes(pkg.name)) {
+            functions = queryEmacsFunctions(emacs, "^" + prefix, logOut);
+            if (!functions.empty()) break;
+        }
+        if (functions.empty()) continue;
+        std::vector<EmacsFunctionDoc> docs;
+        docs.reserve(functions.size());
+        for (const auto& fn : functions) {
+            docs.push_back(queryEmacsFunctionDoc(emacs, fn, logOut));
+        }
+        index.functionsByPackage[pkg.name] = std::move(docs);
+    }
+}
+
+static void appendEmacsExternalModules(Module* module,
+                                       const EmacsFunctionIndex& index,
+                                       int& signatureId) {
+    if (!module) return;
+    for (const auto& kv : index.functionsByPackage) {
+        const std::string& pkgName = kv.first;
+        auto* ext = new ExternalModule("emacs_" + pkgName, pkgName, "elisp");
+        for (const auto& fn : kv.second) {
+            auto* sig = new TypeSignature("emacs_sig_" + std::to_string(signatureId++), fn.name);
+            ext->addChild("signatures", sig);
+        }
+        module->addChild("externalModules", ext);
+    }
+}
+
+static void addEmacsSymbolsToLibraryIndex(LibraryIndexData& libIndex,
+                                          const EmacsFunctionIndex& index) {
+    for (const auto& kv : index.functionsByPackage) {
+        const std::string& pkgName = kv.first;
+        std::vector<LSPClient::WorkspaceSymbol> symbols;
+        symbols.reserve(kv.second.size());
+        for (const auto& fn : kv.second) {
+            LSPClient::WorkspaceSymbol sym;
+            sym.name = fn.name;
+            sym.containerName = pkgName;
+            if (!fn.signature.empty()) {
+                sym.detail = fn.signature;
+                if (!fn.doc.empty()) sym.detail += "\n" + fn.doc;
+            } else {
+                sym.detail = fn.doc;
+            }
+            sym.kind = 12; // Function
+            symbols.push_back(std::move(sym));
+        }
+        if (!symbols.empty()) libIndex.symbolsByLibrary[pkgName] = std::move(symbols);
+    }
+}