@bubblebrain-ai/bubble 0.0.1

package/dist/lsp/service.js

@@ -0,0 +1,695 @@
+import { execFile as execFileCallback, spawn } from "node:child_process";
+import { EventEmitter } from "node:events";
+import { access, readFile } from "node:fs/promises";
+import { createRequire } from "node:module";
+import { basename, dirname, extname, isAbsolute, join, normalize, relative, resolve } from "node:path";
+import { promisify } from "node:util";
+import { pathToFileURL, fileURLToPath } from "node:url";
+import { createMessageConnection, StreamMessageReader, StreamMessageWriter, } from "vscode-jsonrpc/node.js";
+import { customLspServerEntries, isLspEnabled, isLspServerEnabled } from "./config.js";
+const execFile = promisify(execFileCallback);
+const services = new Map();
+const TS_EXTENSIONS = [".ts", ".tsx", ".js", ".jsx", ".mjs", ".cjs", ".mts", ".cts"];
+const NODE_ROOT_MARKERS = ["package-lock.json", "bun.lockb", "bun.lock", "pnpm-lock.yaml", "yarn.lock", "package.json"];
+const DENO_ROOT_MARKERS = ["deno.json", "deno.jsonc"];
+export class ProjectLspService {
+    cwd;
+    emitter = new EventEmitter();
+    clients = new Map();
+    spawning = new Map();
+    starting = new Map();
+    broken = new Map();
+    unavailable = new Set();
+    diagnosticsByFile = new Map();
+    waiters = new Set();
+    disposed = false;
+    config;
+    constructor(cwd, config) {
+        this.cwd = cwd;
+        this.config = config;
+    }
+    updateConfig(config) {
+        const wasDisabled = this.isDisabled();
+        this.config = config;
+        this.unavailable.clear();
+        this.broken.clear();
+        if (!wasDisabled && this.isDisabled()) {
+            void this.shutdownClients();
+        }
+        else {
+            void this.shutdownDisabledClients();
+        }
+        this.emitStatus();
+    }
+    async restart() {
+        this.broken.clear();
+        this.unavailable.clear();
+        await this.shutdownClients();
+        this.emitStatus();
+    }
+    onStatusChange(listener) {
+        this.emitter.on("status", listener);
+        return () => this.emitter.off("status", listener);
+    }
+    isDisabled() {
+        return !isLspEnabled(this.config);
+    }
+    status() {
+        const connected = [...this.clients.values()].map((client) => ({
+            id: client.id,
+            name: client.name,
+            root: relative(this.cwd, client.root) || ".",
+            status: "connected",
+        }));
+        const starting = [...this.starting.entries()]
+            .filter(([key]) => !this.clients.has(key) && !this.broken.has(key))
+            .map(([, status]) => status);
+        return [...connected, ...starting, ...this.broken.values()];
+    }
+    async hasClients(filePath) {
+        const file = this.resolveInsideCwd(filePath);
+        if (!file)
+            return false;
+        return (await this.matchingServers(file)).length > 0;
+    }
+    async touchFile(filePath, diagnostics) {
+        const file = this.resolveInsideCwd(filePath);
+        if (!file)
+            return;
+        const clients = await this.getClients(file);
+        if (!clients.length)
+            return;
+        await Promise.all(clients.map((client) => this.openOrChange(client, file)));
+        if (diagnostics) {
+            await this.waitForDiagnostics(file, diagnostics === "full" ? 8000 : 3500);
+        }
+    }
+    diagnostics() {
+        const result = {};
+        for (const [file, byServer] of this.diagnosticsByFile.entries()) {
+            result[file] = [...byServer.values()].flat();
+        }
+        return result;
+    }
+    async hover(input) {
+        return this.runRequest(input, "textDocument/hover", {
+            textDocument: { uri: pathToFileURL(input.file).href },
+            position: { line: input.line, character: input.character },
+        });
+    }
+    async definition(input) {
+        return this.runRequest(input, "textDocument/definition", {
+            textDocument: { uri: pathToFileURL(input.file).href },
+            position: { line: input.line, character: input.character },
+        });
+    }
+    async references(input) {
+        return this.runRequest(input, "textDocument/references", {
+            textDocument: { uri: pathToFileURL(input.file).href },
+            position: { line: input.line, character: input.character },
+            context: { includeDeclaration: true },
+        });
+    }
+    async implementation(input) {
+        return this.runRequest(input, "textDocument/implementation", {
+            textDocument: { uri: pathToFileURL(input.file).href },
+            position: { line: input.line, character: input.character },
+        });
+    }
+    async documentSymbol(filePath) {
+        const file = this.resolveInsideCwd(filePath);
+        if (!file)
+            return [];
+        const clients = await this.getClients(file);
+        await Promise.all(clients.map((client) => this.openOrChange(client, file)));
+        const results = await Promise.all(clients.map((client) => client.connection
+            .sendRequest("textDocument/documentSymbol", { textDocument: { uri: pathToFileURL(file).href } })
+            .catch(() => [])));
+        return results.flatMap(normalizeLspResult);
+    }
+    async workspaceSymbol(query) {
+        const results = await Promise.all([...this.clients.values()].map((client) => client.connection.sendRequest("workspace/symbol", { query }).catch(() => [])));
+        return results.flatMap(normalizeLspResult).slice(0, 50);
+    }
+    async prepareCallHierarchy(input) {
+        return this.runRequest(input, "textDocument/prepareCallHierarchy", {
+            textDocument: { uri: pathToFileURL(input.file).href },
+            position: { line: input.line, character: input.character },
+        });
+    }
+    async incomingCalls(input) {
+        return this.callHierarchyRequest(input, "callHierarchy/incomingCalls");
+    }
+    async outgoingCalls(input) {
+        return this.callHierarchyRequest(input, "callHierarchy/outgoingCalls");
+    }
+    async shutdown() {
+        if (this.disposed)
+            return;
+        this.disposed = true;
+        for (const waiter of this.waiters) {
+            waiter.resolve();
+        }
+        this.waiters.clear();
+        await this.shutdownClients();
+    }
+    shutdownNow() {
+        for (const waiter of this.waiters)
+            waiter.resolve();
+        this.waiters.clear();
+        for (const client of this.clients.values()) {
+            client.stopping = true;
+            client.connection.dispose();
+            client.process.kill();
+        }
+        this.clients.clear();
+    }
+    async runRequest(input, method, params) {
+        const file = this.resolveInsideCwd(input.file);
+        if (!file)
+            return [];
+        const clients = await this.getClients(file);
+        await Promise.all(clients.map((client) => this.openOrChange(client, file)));
+        const results = await Promise.all(clients.map((client) => client.connection.sendRequest(method, params).catch(() => [])));
+        return results.flatMap(normalizeLspResult);
+    }
+    async callHierarchyRequest(input, method) {
+        const file = this.resolveInsideCwd(input.file);
+        if (!file)
+            return [];
+        const clients = await this.getClients(file);
+        const results = await Promise.all(clients.map(async (client) => {
+            await this.openOrChange(client, file);
+            const items = await client.connection
+                .sendRequest("textDocument/prepareCallHierarchy", {
+                textDocument: { uri: pathToFileURL(file).href },
+                position: { line: input.line, character: input.character },
+            })
+                .catch(() => []);
+            const first = normalizeLspResult(items)[0];
+            if (!first)
+                return [];
+            return client.connection.sendRequest(method, { item: first }).catch(() => []);
+        }));
+        return results.flatMap(normalizeLspResult);
+    }
+    async getClients(file) {
+        const matches = await this.matchingServers(file);
+        const clients = await Promise.all(matches.map(({ server, root }) => this.getClient(server, root)));
+        return clients.filter((client) => !!client);
+    }
+    async getClient(server, root) {
+        const key = `${root}:${server.id}`;
+        const existing = this.clients.get(key);
+        if (existing)
+            return existing;
+        if (this.broken.has(key))
+            return undefined;
+        if (this.unavailable.has(key))
+            return undefined;
+        const inflight = this.spawning.get(key);
+        if (inflight)
+            return inflight;
+        this.starting.set(key, {
+            id: server.id,
+            name: server.name,
+            root: relative(this.cwd, root) || ".",
+            status: "starting",
+            message: "starting",
+        });
+        this.emitStatus();
+        const task = this.spawnClient(server, root, key);
+        this.spawning.set(key, task);
+        task.finally(() => {
+            if (this.spawning.get(key) === task)
+                this.spawning.delete(key);
+            if (this.starting.delete(key))
+                this.emitStatus();
+        });
+        return task;
+    }
+    async spawnClient(server, root, key) {
+        try {
+            const handle = await server.spawn(root, { cwd: this.cwd });
+            if (!handle) {
+                this.unavailable.add(key);
+                return undefined;
+            }
+            const connection = createMessageConnection(new StreamMessageReader(handle.process.stdout), new StreamMessageWriter(handle.process.stdin), {
+                error: () => { },
+                warn: () => { },
+                info: () => { },
+                log: () => { },
+            });
+            const client = {
+                key,
+                id: server.id,
+                name: server.name,
+                root,
+                languageId: server.languageId,
+                process: handle.process,
+                connection,
+                documents: new Map(),
+                stopping: false,
+            };
+            connection.onNotification("textDocument/publishDiagnostics", (params) => {
+                if (!params?.uri)
+                    return;
+                const file = normalize(fileURLToPath(params.uri));
+                const byServer = this.diagnosticsByFile.get(file) ?? new Map();
+                byServer.set(server.id, params.diagnostics ?? []);
+                this.diagnosticsByFile.set(file, byServer);
+                this.resolveDiagnosticWaiters(file);
+            });
+            connection.onRequest("workspace/configuration", (params) => server.configuration?.(root, Array.isArray(params?.items) ? params.items : []) ?? []);
+            connection.onRequest("workspace/workspaceFolders", () => [{ uri: pathToFileURL(root).href, name: basename(root) }]);
+            connection.onRequest("client/registerCapability", () => null);
+            connection.onRequest("client/unregisterCapability", () => null);
+            handle.process.once("exit", (code, signal) => {
+                this.clients.delete(key);
+                if (!this.disposed && !client.stopping) {
+                    this.markBroken(key, server, root, `server exited${code === null ? "" : ` with code ${code}`}${signal ? ` (${signal})` : ""}`);
+                }
+            });
+            connection.listen();
+            await connection.sendRequest("initialize", {
+                processId: process.pid,
+                rootPath: root,
+                rootUri: pathToFileURL(root).href,
+                workspaceFolders: [{ uri: pathToFileURL(root).href, name: basename(root) }],
+                initializationOptions: handle.initializationOptions ?? {},
+                capabilities: {
+                    textDocument: {
+                        synchronization: { dynamicRegistration: true, didSave: true },
+                        hover: { dynamicRegistration: true, contentFormat: ["markdown", "plaintext"] },
+                        definition: { dynamicRegistration: true },
+                        references: { dynamicRegistration: true },
+                        implementation: { dynamicRegistration: true },
+                        documentSymbol: { dynamicRegistration: true, hierarchicalDocumentSymbolSupport: true },
+                        callHierarchy: { dynamicRegistration: true },
+                        publishDiagnostics: { relatedInformation: true, versionSupport: true },
+                    },
+                    workspace: {
+                        configuration: true,
+                        workspaceFolders: true,
+                        symbol: { dynamicRegistration: true },
+                    },
+                    window: { workDoneProgress: false },
+                },
+            });
+            connection.sendNotification("initialized", {});
+            this.clients.set(key, client);
+            this.emitStatus();
+            return client;
+        }
+        catch (error) {
+            this.markBroken(key, server, root, error instanceof Error ? error.message : String(error));
+            return undefined;
+        }
+    }
+    async openOrChange(client, file) {
+        const text = await readFile(file, "utf-8");
+        const languageId = client.languageId(file);
+        const uri = pathToFileURL(file).href;
+        const existing = client.documents.get(file);
+        if (!existing) {
+            client.documents.set(file, { languageId, version: 1 });
+            client.connection.sendNotification("textDocument/didOpen", {
+                textDocument: { uri, languageId, version: 1, text },
+            });
+            return;
+        }
+        existing.version += 1;
+        client.connection.sendNotification("textDocument/didChange", {
+            textDocument: { uri, version: existing.version },
+            contentChanges: [{ text }],
+        });
+    }
+    waitForDiagnostics(file, timeoutMs) {
+        return new Promise((resolve) => {
+            let settled = false;
+            const timeout = setTimeout(() => done(), timeoutMs);
+            const debounce = setTimeout(() => {
+                if (this.diagnosticsByFile.has(file))
+                    done();
+            }, 150);
+            const done = () => {
+                if (settled)
+                    return;
+                settled = true;
+                clearTimeout(timeout);
+                clearTimeout(debounce);
+                this.waiters.delete(waiter);
+                resolve();
+            };
+            const waiter = { file, resolve: done, timeout };
+            this.waiters.add(waiter);
+        });
+    }
+    resolveDiagnosticWaiters(file) {
+        for (const waiter of [...this.waiters]) {
+            if (waiter.file !== file)
+                continue;
+            waiter.resolve();
+        }
+    }
+    async matchingServers(file) {
+        if (!isLspEnabled(this.config))
+            return [];
+        const extension = extname(file) || file;
+        const result = [];
+        for (const server of this.servers()) {
+            if (!isLspServerEnabled(this.config, server.id))
+                continue;
+            if (server.extensions.length && !server.extensions.includes(extension))
+                continue;
+            const root = await server.root(file, { cwd: this.cwd });
+            if (!root)
+                continue;
+            if (this.broken.has(`${root}:${server.id}`))
+                continue;
+            result.push({ server, root });
+        }
+        return result;
+    }
+    servers() {
+        return [...BUILTIN_SERVERS, ...customServers(this.config)];
+    }
+    markBroken(key, server, root, message) {
+        this.starting.delete(key);
+        this.broken.set(key, {
+            id: server.id,
+            name: server.name,
+            root: relative(this.cwd, root) || ".",
+            status: "error",
+            message,
+        });
+        this.emitStatus();
+    }
+    emitStatus() {
+        this.emitter.emit("status");
+    }
+    resolveInsideCwd(filePath) {
+        const file = normalize(isAbsolute(filePath) ? filePath : resolve(this.cwd, filePath));
+        const rel = relative(this.cwd, file);
+        if (rel.startsWith("..") || isAbsolute(rel))
+            return undefined;
+        return file;
+    }
+    async shutdownClient(client) {
+        client.stopping = true;
+        await client.connection.sendRequest("shutdown").catch(() => undefined);
+        client.connection.sendNotification("exit");
+        client.connection.dispose();
+        client.process.kill();
+    }
+    async shutdownClients() {
+        await Promise.all([...this.clients.values()].map((client) => this.shutdownClient(client)));
+        this.clients.clear();
+        this.starting.clear();
+        this.diagnosticsByFile.clear();
+    }
+    async shutdownDisabledClients() {
+        const disabled = [...this.clients.values()].filter((client) => !isLspServerEnabled(this.config, client.id));
+        await Promise.all(disabled.map((client) => this.shutdownClient(client)));
+        for (const client of disabled) {
+            this.clients.delete(client.key);
+        }
+    }
+}
+export const TypeScriptLspService = ProjectLspService;
+export function getLspService(cwd, config) {
+    const key = resolve(cwd);
+    const existing = services.get(key);
+    if (existing) {
+        existing.updateConfig(config);
+        return existing;
+    }
+    const service = new ProjectLspService(key, config);
+    services.set(key, service);
+    process.once("exit", () => service.shutdownNow());
+    return service;
+}
+const TypeScriptServer = {
+    id: "typescript",
+    name: "typescript",
+    extensions: TS_EXTENSIONS,
+    root: nearestRoot(NODE_ROOT_MARKERS, DENO_ROOT_MARKERS),
+    languageId: languageIdFor,
+    async spawn(root, ctx) {
+        const requireFromRoot = createRequire(join(root, "package.json"));
+        const requireFromSelf = createRequire(import.meta.url);
+        const tsserverPath = resolveModule(requireFromRoot, "typescript/lib/tsserver.js")
+            ?? resolveModule(requireFromSelf, "typescript/lib/tsserver.js");
+        const serverPath = resolveModule(requireFromRoot, "typescript-language-server/lib/cli.mjs")
+            ?? resolveModule(requireFromSelf, "typescript-language-server/lib/cli.mjs");
+        if (!tsserverPath || !serverPath)
+            return undefined;
+        return {
+            process: spawn(process.execPath, [serverPath, "--stdio"], { cwd: root, env: process.env, stdio: "pipe" }),
+            initializationOptions: { tsserver: { path: tsserverPath } },
+        };
+    },
+};
+const DenoServer = {
+    id: "deno",
+    name: "deno",
+    extensions: [".ts", ".tsx", ".js", ".jsx", ".mjs"],
+    root: denoRoot,
+    languageId: languageIdFor,
+    async spawn(root) {
+        const deno = await findExecutable("deno", [root]);
+        if (!deno)
+            return undefined;
+        return {
+            process: spawn(deno, ["lsp"], { cwd: root, env: process.env, stdio: "pipe" }),
+            initializationOptions: { enable: true, lint: true },
+        };
+    },
+};
+const VueServer = {
+    id: "vue",
+    name: "vue",
+    extensions: [".vue"],
+    root: nearestRoot(NODE_ROOT_MARKERS),
+    languageId: () => "vue",
+    async spawn(root) {
+        const requireFromRoot = createRequire(join(root, "package.json"));
+        const requireFromSelf = createRequire(import.meta.url);
+        const serverPath = resolveModule(requireFromRoot, "@vue/language-server/bin/vue-language-server.js")
+            ?? resolveModule(requireFromSelf, "@vue/language-server/bin/vue-language-server.js");
+        const tsdk = dirname(resolveModule(requireFromRoot, "typescript/lib/tsserverlibrary.js")
+            ?? resolveModule(requireFromSelf, "typescript/lib/tsserverlibrary.js")
+            ?? "");
+        if (!serverPath)
+            return undefined;
+        const args = tsdk ? [serverPath, "--stdio", "--tsdk", tsdk] : [serverPath, "--stdio"];
+        return {
+            process: spawn(process.execPath, args, { cwd: root, env: process.env, stdio: "pipe" }),
+        };
+    },
+};
+const ESLintServer = {
+    id: "eslint",
+    name: "eslint",
+    extensions: [...TS_EXTENSIONS, ".vue"],
+    root: eslintRoot,
+    languageId: languageIdFor,
+    async spawn(root, ctx) {
+        const requireFromRoot = createRequire(join(root, "package.json"));
+        const requireFromSelf = createRequire(import.meta.url);
+        const eslint = resolveModule(requireFromRoot, "eslint");
+        if (!eslint)
+            return undefined;
+        const serverPath = resolveModule(requireFromRoot, "vscode-langservers-extracted/bin/vscode-eslint-language-server")
+            ?? resolveModule(requireFromSelf, "vscode-langservers-extracted/bin/vscode-eslint-language-server");
+        if (!serverPath)
+            return undefined;
+        return {
+            process: spawn(process.execPath, [serverPath, "--stdio"], { cwd: root, env: process.env, stdio: "pipe" }),
+        };
+    },
+    configuration: (root, items) => items.map(() => eslintConfiguration(root)),
+};
+const OxlintServer = {
+    id: "oxlint",
+    name: "oxlint",
+    extensions: [...TS_EXTENSIONS, ".vue", ".astro", ".svelte"],
+    root: nearestRoot([".oxlintrc.json", ...NODE_ROOT_MARKERS]),
+    languageId: languageIdFor,
+    async spawn(root) {
+        const oxlint = await findExecutable("oxlint", [root]);
+        if (!oxlint || !(await supportsArg(oxlint, "--lsp")))
+            return undefined;
+        return {
+            process: spawn(oxlint, ["--lsp"], { cwd: root, env: process.env, stdio: "pipe" }),
+        };
+    },
+};
+const BUILTIN_SERVERS = [DenoServer, TypeScriptServer, VueServer, ESLintServer, OxlintServer];
+function customServers(config) {
+    return customLspServerEntries(config).map(([id, server]) => ({
+        id,
+        name: id,
+        extensions: server.extensions ?? [],
+        root: customRoot(server.rootMarkers),
+        languageId: () => server.languageId ?? id,
+        async spawn(root) {
+            const command = server.command;
+            if (!command?.length)
+                return undefined;
+            return {
+                process: spawn(command[0], command.slice(1), {
+                    cwd: root,
+                    env: { ...process.env, ...(server.env ?? {}) },
+                    stdio: "pipe",
+                }),
+                initializationOptions: server.initializationOptions,
+            };
+        },
+    }));
+}
+function customRoot(rootMarkers) {
+    return async (file, ctx) => {
+        if (!rootMarkers?.length)
+            return ctx.cwd;
+        const marker = await findUp(dirname(file), ctx.cwd, rootMarkers);
+        return marker ? dirname(marker) : undefined;
+    };
+}
+function nearestRoot(includeMarkers, excludeMarkers = []) {
+    return async (file, ctx) => {
+        if (excludeMarkers.length && await findUp(dirname(file), ctx.cwd, excludeMarkers))
+            return undefined;
+        const marker = await findUp(dirname(file), ctx.cwd, includeMarkers);
+        return marker ? dirname(marker) : ctx.cwd;
+    };
+}
+async function denoRoot(file, ctx) {
+    const marker = await findUp(dirname(file), ctx.cwd, DENO_ROOT_MARKERS);
+    return marker ? dirname(marker) : undefined;
+}
+async function eslintRoot(file, ctx) {
+    const root = await nearestRoot(NODE_ROOT_MARKERS)(file, ctx);
+    if (!root)
+        return undefined;
+    const config = await findUp(dirname(file), root, [
+        "eslint.config.js",
+        "eslint.config.mjs",
+        "eslint.config.cjs",
+        ".eslintrc",
+        ".eslintrc.json",
+        ".eslintrc.js",
+        ".eslintrc.cjs",
+        ".eslintrc.yaml",
+        ".eslintrc.yml",
+    ]);
+    return config ? root : undefined;
+}
+async function findUp(start, stop, markers) {
+    let dir = start;
+    while (true) {
+        for (const marker of markers) {
+            const target = join(dir, marker);
+            try {
+                await access(target);
+                return target;
+            }
+            catch {
+                // keep searching
+            }
+        }
+        if (dir === stop || dir === dirname(dir))
+            return undefined;
+        dir = dirname(dir);
+    }
+}
+function resolveModule(requireFn, modulePath) {
+    try {
+        return requireFn.resolve(modulePath);
+    }
+    catch {
+        return undefined;
+    }
+}
+async function findExecutable(command, roots) {
+    const ext = process.platform === "win32" ? ".cmd" : "";
+    for (const root of roots) {
+        const local = join(root, "node_modules", ".bin", command + ext);
+        try {
+            await access(local);
+            return local;
+        }
+        catch {
+            // try next location
+        }
+    }
+    for (const dir of (process.env.PATH ?? "").split(":")) {
+        if (!dir)
+            continue;
+        const target = join(dir, command + ext);
+        try {
+            await access(target);
+            return target;
+        }
+        catch {
+            // try next PATH entry
+        }
+    }
+    return undefined;
+}
+async function supportsArg(binary, arg) {
+    try {
+        const { stdout, stderr } = await execFile(binary, ["--help"], { timeout: 3000 });
+        return `${stdout}\n${stderr}`.includes(arg);
+    }
+    catch {
+        return false;
+    }
+}
+function languageIdFor(file) {
+    switch (extname(file)) {
+        case ".ts":
+        case ".mts":
+        case ".cts":
+            return "typescript";
+        case ".tsx":
+            return "typescriptreact";
+        case ".jsx":
+            return "javascriptreact";
+        case ".vue":
+            return "vue";
+        default:
+            return "javascript";
+    }
+}
+function normalizeLspResult(result) {
+    if (!result)
+        return [];
+    return Array.isArray(result) ? result.filter(Boolean) : [result];
+}
+function eslintConfiguration(root) {
+    return {
+        validate: "on",
+        packageManager: packageManagerFor(root),
+        useESLintClass: false,
+        experimental: { useFlatConfig: false },
+        codeAction: {
+            disableRuleComment: { enable: true, location: "separateLine" },
+            showDocumentation: { enable: true },
+        },
+        codeActionOnSave: { enable: false, mode: "all" },
+        format: false,
+        nodePath: null,
+        onIgnoredFiles: "off",
+        options: {},
+        problems: { shortenToSingleLine: false },
+        quiet: false,
+        rulesCustomizations: [],
+        run: "onType",
+        workingDirectory: { mode: "location" },
+    };
+}
+function packageManagerFor(root) {
+    return root.includes("pnpm") ? "pnpm" : "npm";
+}

package/dist/main.d.ts

@@ -0,0 +1,5 @@
+#!/usr/bin/env bun
+/**
+ * Main entry point - assembles all layers and runs the agent.
+ */
+export {};