@dot-agent/language-server 0.2.0

package/package.json

@@ -0,0 +1,32 @@
+{
+  "name": "@dot-agent/language-server",
+  "version": "0.2.0",
+  "description": "LSP server for .agent DSL files",
+  "license": "Apache-2.0",
+  "author": "Danilo Borges <contato@daniloborg.es>",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/dot-agent-spec/language-server.git"
+  },
+  "homepage": "https://github.com/dot-agent-spec/language-server#readme",
+  "bugs": {
+    "url": "https://github.com/dot-agent-spec/language-server/issues"
+  },
+  "keywords": ["lsp", "language-server", "agent", "dsl", "flow"],
+  "main": "server.js",
+  "engines": {
+    "node": ">=18"
+  },
+  "scripts": {
+    "start": "node server.js --stdio"
+  },
+  "dependencies": {
+    "vscode-languageserver": "^9.0.1",
+    "vscode-languageserver-textdocument": "^1.0.12",
+    "web-tree-sitter": "^0.25.0",
+    "@dot-agent/tree-sitter": "^0.2.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/parser.js

@@ -0,0 +1,126 @@
+/*
+ * Copyright (c) 2026 Danilo Borges (https://github.com/daniloborges)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+'use strict';
+
+const path = require('path');
+const { Parser, Language } = require('web-tree-sitter');
+const grammar = require('@dot-agent/tree-sitter');
+
+// path.resolve ensures absolute paths survive vsix packaging and cwd changes
+const AGENT_WASM = path.resolve(grammar.agentWasmPath);
+const BEHAVIOR_WASM  = path.resolve(grammar.behaviorWasmPath);
+
+let agentParser, behaviorParser;
+
+async function initParsers() {
+    await Parser.init();
+    const Agent = await Language.load(AGENT_WASM);
+    const Behavior  = await Language.load(BEHAVIOR_WASM);
+    agentParser = new Parser();
+    agentParser.setLanguage(Agent);
+    behaviorParser = new Parser();
+    behaviorParser.setLanguage(Behavior);
+}
+
+// Cache: uri → { version, tree }
+const cache = new Map();
+
+function parse(uri, langId, text, version) {
+    const prev = cache.get(uri);
+    if (prev?.version === version) return prev.tree;
+    const parser = langId === 'behavior' ? behaviorParser : agentParser;
+    if (!parser) return null;
+    const tree = parser.parse(text, prev?.tree);   // incremental reuse when possible
+    cache.set(uri, { version, tree });
+    return tree;
+}
+
+function evict(uri) {
+    cache.delete(uri);
+}
+
+// ── AST helpers ──────────────────────────────────────────────────────────────
+
+function nodesOfType(tree, type) {
+    if (!tree) return [];
+    return tree.rootNode.descendantsOfType(type);
+}
+
+function nodeAtOffset(tree, offset) {
+    if (!tree) return null;
+    return tree.rootNode.descendantForIndex(offset);
+}
+
+// Convert a tree-sitter {row, column} position to an LSP Range
+function nodeToRange(node) {
+    return {
+        start: { line: node.startPosition.row, character: node.startPosition.column },
+        end:   { line: node.endPosition.row,   character: node.endPosition.column },
+    };
+}
+
+// Convert LSP position (line, character) to a byte offset in text
+function positionToOffset(text, line, character) {
+    let offset = 0;
+    for (let i = 0; i < line; i++) {
+        const nl = text.indexOf('\n', offset);
+        offset = nl === -1 ? text.length : nl + 1;
+    }
+    return offset + character;
+}
+
+// Extract the word (identifier chars including dots) around position
+function wordAtPosition(text, line, character) {
+    const lines = text.split('\n');
+    const lineText = lines[line] || '';
+    let start = character, end = character;
+    while (start > 0 && /[a-zA-Z0-9_.]/.test(lineText[start - 1])) start--;
+    while (end < lineText.length && /[a-zA-Z0-9_.]/.test(lineText[end])) end++;
+    return { word: lineText.slice(start, end), start, end };
+}
+
+// Walk up the tree from offset, skipping ERROR/MISSING nodes
+function getContextNode(tree, offset) {
+    let node = nodeAtOffset(tree, offset);
+    while (node && (node.isError || node.isMissing)) {
+        node = node.parent;
+    }
+    // If still on an error, try the previous sibling's last descendant
+    if (!node || node.type === 'ERROR') {
+        const raw = nodeAtOffset(tree, offset);
+        const parent = raw?.parent;
+        if (parent) {
+            const siblings = parent.children;
+            const idx = siblings.findIndex(c => c.startIndex > offset);
+            const prev = idx > 0 ? siblings[idx - 1] : siblings[siblings.length - 1];
+            if (prev && !prev.isError) node = prev;
+        }
+    }
+    return node ?? tree?.rootNode;
+}
+
+module.exports = {
+    initParsers,
+    parse,
+    evict,
+    nodesOfType,
+    nodeAtOffset,
+    nodeToRange,
+    positionToOffset,
+    wordAtPosition,
+    getContextNode,
+};

package/server.js

@@ -0,0 +1,169 @@
+/*
+ * Copyright (c) 2026 Danilo Borges (https://github.com/daniloborges)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+'use strict';
+
+const {
+    createConnection,
+    TextDocuments,
+    ProposedFeatures,
+    TextDocumentSyncKind,
+    SymbolKind,
+} = require('vscode-languageserver/node');
+const { TextDocument } = require('vscode-languageserver-textdocument');
+
+const { initParsers, parse, evict } = require('./parser');
+
+const { provideHover }          = require('./features/hover');
+const { provideCompletions }    = require('./features/completions');
+const { diagnose }              = require('./features/diagnostics');
+const { provideDocumentSymbols }= require('./features/symbols');
+const { provideDefinition }     = require('./features/definition');
+const { provideReferences }     = require('./features/references');
+const { provideRenameEdits }    = require('./features/rename');
+const { format }                = require('./features/formatting');
+const { provideDocumentLinks }  = require('./features/links');
+
+const connection = createConnection(ProposedFeatures.all);
+const documents  = new TextDocuments(TextDocument);
+
+// ── Initialization ───────────────────────────────────────────────────────────
+
+// web-tree-sitter is strictly async — await before advertising capabilities
+// so no feature handler fires before parsers are ready.
+connection.onInitialize(async () => {
+    await initParsers();
+    return {
+        capabilities: {
+            textDocumentSync: TextDocumentSyncKind.Incremental,
+            hoverProvider: true,
+            completionProvider: { triggerCharacters: ['.', ' '] },
+            definitionProvider: true,
+            referencesProvider: true,
+            renameProvider: { prepareProvider: false },
+            documentSymbolProvider: true,
+            documentFormattingProvider: true,
+            documentLinkProvider: { resolveProvider: false },
+        },
+    };
+});
+
+// ── Helpers ───────────────────────────────────────────────────────────────────
+
+function getTree(doc) {
+    return parse(doc.uri, doc.languageId, doc.getText(), doc.version);
+}
+
+// ── Diagnostics (push on change) ─────────────────────────────────────────────
+
+function validate(doc) {
+    const langId = doc.languageId;
+    if (langId !== 'agent' && langId !== 'behavior') return;
+    const tree = getTree(doc);
+    const diagnostics = diagnose(langId, tree, doc.getText());
+    connection.sendDiagnostics({ uri: doc.uri, diagnostics });
+}
+
+documents.onDidChangeContent(e => validate(e.document));
+documents.onDidOpen(e => validate(e.document));
+documents.onDidClose(e => {
+    evict(e.document.uri);
+    connection.sendDiagnostics({ uri: e.document.uri, diagnostics: [] });
+});
+
+// ── Hover ────────────────────────────────────────────────────────────────────
+
+connection.onHover(({ textDocument, position }) => {
+    const doc = documents.get(textDocument.uri);
+    if (!doc) return null;
+    return provideHover(doc.languageId, doc.getText(), position);
+});
+
+// ── Completion ───────────────────────────────────────────────────────────────
+
+connection.onCompletion(({ textDocument, position }) => {
+    const doc = documents.get(textDocument.uri);
+    if (!doc) return [];
+    return provideCompletions(doc.languageId, getTree(doc), doc.getText(), position);
+});
+
+// ── Definition ───────────────────────────────────────────────────────────────
+
+connection.onDefinition(({ textDocument, position }) => {
+    const doc = documents.get(textDocument.uri);
+    if (!doc) return null;
+    return provideDefinition(doc.languageId, getTree(doc), doc.getText(), doc.uri, position);
+});
+
+// ── References ───────────────────────────────────────────────────────────────
+
+connection.onReferences(({ textDocument, position }) => {
+    const doc = documents.get(textDocument.uri);
+    if (!doc) return [];
+    return provideReferences(doc.languageId, getTree(doc), doc.getText(), doc.uri, position);
+});
+
+// ── Rename ───────────────────────────────────────────────────────────────────
+
+connection.onRenameRequest(({ textDocument, position, newName }) => {
+    const doc = documents.get(textDocument.uri);
+    if (!doc) return null;
+    return provideRenameEdits(doc.languageId, getTree(doc), doc.getText(), doc.uri, position, newName);
+});
+
+// ── Document Symbols ─────────────────────────────────────────────────────────
+
+connection.onDocumentSymbol(({ textDocument }) => {
+    try {
+        const doc = documents.get(textDocument.uri);
+        if (!doc) {
+            connection.console.log('[symbols] doc not found for ' + textDocument.uri);
+            return [];
+        }
+        const result = provideDocumentSymbols(doc.languageId, getTree(doc));
+        connection.console.log(`[symbols] langId=${doc.languageId} count=${result.length} sample=${JSON.stringify(result[0]).slice(0,120)}`);
+        return result;
+    } catch (e) {
+        connection.console.error('[symbols] threw: ' + e.message + '\n' + e.stack);
+        return [];
+    }
+});
+
+// ── Document Links ───────────────────────────────────────────────────────────
+
+connection.onDocumentLinks(({ textDocument }) => {
+    try {
+        const doc = documents.get(textDocument.uri);
+        if (!doc) return [];
+        return provideDocumentLinks(doc.languageId, getTree(doc), doc.uri);
+    } catch (e) {
+        connection.console.error(`documentLinks error: ${e.message}`);
+        return [];
+    }
+});
+
+// ── Formatting ───────────────────────────────────────────────────────────────
+
+connection.onDocumentFormatting(({ textDocument }) => {
+    const doc = documents.get(textDocument.uri);
+    if (!doc) return [];
+    return format(doc.languageId, doc.getText());
+});
+
+// ── Start ────────────────────────────────────────────────────────────────────
+
+documents.listen(connection);
+connection.listen();