@marko/language-server 0.12.8 → 0.12.11

package/dist/index.mjs

@@ -571,7 +571,7 @@ function loadCompilerInfo(dir) {
   const rootDir = lassoPackageRoot.getRootDir(dir);
   const pkgPath = rootDir && resolveFrom.silent(rootDir, "@marko/compiler/package.json");
   const pkg = pkgPath && __require(pkgPath);
-  const cache2 = /* @__PURE__ */ new Map();
+  const cache3 = /* @__PURE__ */ new Map();
   let translator = builtinTranslator;
   let compiler = builtinCompiler;
   if (pkg && /^5\./.test(pkg.version)) {
@@ -588,16 +588,16 @@ function loadCompilerInfo(dir) {
     }
   }
   return {
-    cache: cache2,
+    cache: cache3,
     get lookup() {
-      let lookup = cache2.get(lookupKey);
+      let lookup = cache3.get(lookupKey);
       if (lookup === void 0) {
         try {
           lookup = compiler.taglib.buildLookup(dir, translator);
         } catch {
           lookup = builtinInfo.lookup;
         }
-        cache2.set(lookupKey, lookup);
+        cache3.set(lookupKey, lookup);
       }
       return lookup;
     },
@@ -856,11 +856,147 @@ function AttrName({
   return completions;
 }
 
+// src/service/marko/complete/AttrValue.ts
+import path3 from "path";
+import {
+  CompletionItemKind as CompletionItemKind4,
+  Range as Range2,
+  TextEdit as TextEdit4
+} from "vscode-languageserver";
+
+// src/service/marko/util/is-document-link-attr.ts
+var linkedAttrs = /* @__PURE__ */ new Map([
+  [
+    "src",
+    /* @__PURE__ */ new Set([
+      "audio",
+      "embed",
+      "iframe",
+      "img",
+      "input",
+      "script",
+      "source",
+      "track",
+      "video"
+    ])
+  ],
+  ["href", /* @__PURE__ */ new Set(["a", "area", "link"])],
+  ["data", /* @__PURE__ */ new Set(["object"])],
+  ["poster", /* @__PURE__ */ new Set(["video"])]
+]);
+function isDocumentLinkAttr(doc, tag, attr) {
+  var _a, _b;
+  return tag.nameText && attr.type === 8 /* AttrNamed */ && ((_a = attr.value) == null ? void 0 : _a.type) === 11 /* AttrValue */ && /^['"]$/.test(doc.getText()[attr.value.value.start]) && ((_b = linkedAttrs.get(doc.getText().slice(attr.name.start, attr.name.end))) == null ? void 0 : _b.has(tag.nameText)) || false;
+}
+
+// src/utils/file-system.ts
+import fs from "fs/promises";
+import { FileType } from "vscode-css-languageservice";
+import { fileURLToPath } from "url";
+var file_system_default = {
+  stat,
+  readDirectory
+};
+async function stat(uri) {
+  let type = FileType.Unknown;
+  let ctime = -1;
+  let mtime = -1;
+  let size = -1;
+  try {
+    const stat2 = await fs.stat(fileURLToPath(uri));
+    if (stat2.isDirectory())
+      type = FileType.Directory;
+    else if (stat2.isFile())
+      type = FileType.File;
+    ctime = stat2.ctimeMs;
+    mtime = stat2.mtimeMs;
+    size = stat2.size;
+  } catch {
+  }
+  return {
+    type,
+    ctime,
+    mtime,
+    size
+  };
+}
+async function readDirectory(uri) {
+  try {
+    const entries = await fs.readdir(fileURLToPath(uri));
+    const base = uri.at(-1) === "/" ? uri : `${uri}/`;
+    return (await Promise.all(entries.map(async (entry) => [entry, (await stat(new URL(entry, base).toString())).type]))).filter(([, type]) => type !== FileType.Unknown);
+  } catch {
+    return [];
+  }
+}
+
+// src/utils/resolve-url.ts
+function resolveUrl(to, base) {
+  try {
+    const url = new URL(to, base);
+    if (url.protocol === "file:")
+      return url.toString();
+  } catch {
+    return void 0;
+  }
+}
+
+// src/service/marko/complete/AttrValue.ts
+async function AttrValue({
+  document,
+  offset,
+  node,
+  parsed,
+  code
+}) {
+  const attr = node.parent;
+  if (isDocumentLinkAttr(document, attr.parent, attr)) {
+    const start = node.value.start + 1;
+    if (code[start] !== ".")
+      return;
+    const end = node.value.end - 1;
+    const relativeOffset = offset - start;
+    const rawValue = parsed.read({
+      start,
+      end
+    });
+    const segmentStart = rawValue.lastIndexOf("/", relativeOffset);
+    if (segmentStart === -1)
+      return;
+    const req = rawValue.slice(0, segmentStart);
+    const uri = resolveUrl(req, document.uri);
+    if (uri) {
+      const result = [];
+      const curFile = req === "." ? path3.basename(document.uri) : void 0;
+      const replaceRange = Range2.create(document.positionAt(start + segmentStart + 1), document.positionAt(start + rawValue.length));
+      for (const [entry, type] of await file_system_default.readDirectory(uri)) {
+        if (entry[0] !== "." && entry !== curFile) {
+          result.push(type === FileType.Directory ? {
+            label: `${entry}/`,
+            kind: CompletionItemKind4.Folder,
+            textEdit: TextEdit4.replace(replaceRange, `${entry}/`),
+            command: {
+              title: "Suggest",
+              command: "editor.action.triggerSuggest"
+            }
+          } : {
+            label: entry,
+            kind: CompletionItemKind4.File,
+            textEdit: TextEdit4.replace(replaceRange, entry)
+          });
+        }
+      }
+      return result;
+    }
+  }
+}
+
 // src/service/marko/complete/index.ts
 var handlers = {
   Tag,
   OpenTagName,
-  AttrName
+  AttrName,
+  AttrValue
 };
 var doComplete = async (doc, params) => {
   var _a;
@@ -879,15 +1015,15 @@ var doComplete = async (doc, params) => {
 };
 
 // src/service/marko/validate.ts
-import { Diagnostic, DiagnosticSeverity, Range as Range2 } from "vscode-languageserver";
+import { Diagnostic, DiagnosticSeverity, Range as Range3 } from "vscode-languageserver";
 var markoErrorRegExp = /^(.+?)(?:\((\d+)(?:\s*,\s*(\d+))?\))?: (.*)$/gm;
 var doValidate = (doc) => {
   const fsPath = getDocFile(doc);
   const diagnostics = [];
-  const { compiler, translator, cache: cache2 } = getCompilerInfo(doc);
+  const { compiler, translator, cache: cache3 } = getCompilerInfo(doc);
   try {
     compiler.compileSync(doc.getText(), fsPath || "untitled.marko", {
-      cache: cache2,
+      cache: cache3,
       translator,
       code: false,
       output: "source",
@@ -899,16 +1035,16 @@ var doValidate = (doc) => {
       const [, fileName, rawLine, rawCol, msg] = match;
       const line = (parseInt(rawLine, 10) || 1) - 1;
       const col = (parseInt(rawCol, 10) || 1) - 1;
-      diagnostics.push(Diagnostic.create(Range2.create(line, col, line, col), msg, DiagnosticSeverity.Error, void 0, fileName));
+      diagnostics.push(Diagnostic.create(Range3.create(line, col, line, col), msg, DiagnosticSeverity.Error, void 0, fileName));
     }
   }
   return diagnostics;
 };
 
 // src/service/marko/definition/OpenTagName.ts
-import path3 from "path";
+import path4 from "path";
 import { URI as URI4 } from "vscode-uri";
-import { Range as Range4, LocationLink } from "vscode-languageserver";
+import { Range as Range5, LocationLink } from "vscode-languageserver";
 
 // src/utils/regexp-builder.ts
 function RegExpBuilder(strings, ...expressions) {
@@ -935,14 +1071,14 @@ function escape(val) {
 }
 
 // src/utils/utils.ts
-import fs from "fs";
+import fs2 from "fs";
 import { URI as URI3 } from "vscode-uri";
-import { Position, Range as Range3 } from "vscode-languageserver";
+import { Position, Range as Range4 } from "vscode-languageserver";
 import { TextDocument } from "vscode-languageserver-textdocument";
-var START_OF_FILE = Range3.create(Position.create(0, 0), Position.create(0, 0));
+var START_OF_FILE = Range4.create(Position.create(0, 0), Position.create(0, 0));
 function createTextDocument(filename) {
   const uri = URI3.file(filename).toString();
-  const content = fs.readFileSync(filename, "utf-8");
+  const content = fs2.readFileSync(filename, "utf-8");
   return TextDocument.create(uri, "plaintext", 0, content);
 }
 
@@ -967,14 +1103,14 @@ function OpenTagName2({
     return;
   }
   const tagEntryFile = tagDef.template || tagDef.renderer || tagDef.filePath;
-  if (!path3.isAbsolute(tagEntryFile)) {
+  if (!path4.isAbsolute(tagEntryFile)) {
     return;
   }
   if (/\/marko(?:-tag)?\.json$/.test(tagEntryFile)) {
     const tagDefDoc = createTextDocument(tagEntryFile);
     const match = RegExpBuilder`/"(?:<${tag.nameText}>|${tag.nameText})"\s*:\s*[^\r\n,]+/g`.exec(tagDefDoc.getText());
     if (match && match.index) {
-      range = Range4.create(tagDefDoc.positionAt(match.index), tagDefDoc.positionAt(match.index + match[0].length));
+      range = Range5.create(tagDefDoc.positionAt(match.index), tagDefDoc.positionAt(match.index + match[0].length));
     }
   }
   return [
@@ -984,7 +1120,7 @@ function OpenTagName2({
 
 // src/service/marko/definition/AttrName.ts
 import { URI as URI5 } from "vscode-uri";
-import { Range as Range5, LocationLink as LocationLink2 } from "vscode-languageserver";
+import { Range as Range6, LocationLink as LocationLink2 } from "vscode-languageserver";
 function AttrName2({
   lookup,
   parsed,
@@ -1008,7 +1144,7 @@ function AttrName2({
     const tagDefDoc = createTextDocument(attrEntryFile);
     const match = RegExpBuilder`/"@${attrName}"\s*:\s*[^\r\n,]+/g`.exec(tagDefDoc.getText());
     if (match && match.index) {
-      range = Range5.create(tagDefDoc.positionAt(match.index), tagDefDoc.positionAt(match.index + match[0].length));
+      range = Range6.create(tagDefDoc.positionAt(match.index), tagDefDoc.positionAt(match.index + match[0].length));
     }
   }
   return [
@@ -1037,14 +1173,83 @@ var findDefinition = async (doc, params) => {
   })) || [];
 };
 
-// src/service/marko/format.ts
-import { Range as Range6, TextEdit as TextEdit4 } from "vscode-languageserver";
+// src/service/marko/document-links/extract.ts
+import { DocumentLink } from "vscode-languageserver";
 import { URI as URI6 } from "vscode-uri";
+var importTagReg = /(['"])<((?:[^\1\\>]+|\\.)*)>?\1/g;
+function extractDocumentLinks(doc, parsed, lookup) {
+  if (URI6.parse(doc.uri).scheme === "untitled") {
+    return [];
+  }
+  const links = [];
+  const { program } = parsed;
+  const code = doc.getText();
+  const read = (range) => code.slice(range.start, range.end);
+  const visit = (node) => {
+    switch (node.type) {
+      case 1 /* Tag */:
+        if (node.attrs && node.nameText) {
+          for (const attr of node.attrs) {
+            if (isDocumentLinkAttr(doc, node, attr)) {
+              links.push(DocumentLink.create({
+                start: parsed.positionAt(attr.value.value.start),
+                end: parsed.positionAt(attr.value.value.end)
+              }, resolveUrl(read(attr.value.value).slice(1, -1), doc.uri)));
+            }
+          }
+        }
+        if (node.body) {
+          for (const child of node.body) {
+            visit(child);
+          }
+        }
+        break;
+    }
+  };
+  for (const item of program.static) {
+    if (item.type === 20 /* Statement */ && code[item.start] === "i") {
+      importTagReg.lastIndex = 0;
+      const value = parsed.read(item);
+      const match = importTagReg.exec(value);
+      if (match) {
+        const [{ length }, , tagName] = match;
+        const tagDef = lookup.getTag(tagName);
+        const fileForTag = tagDef && (tagDef.template || tagDef.renderer);
+        if (fileForTag) {
+          links.push(DocumentLink.create({
+            start: parsed.positionAt(item.start + match.index),
+            end: parsed.positionAt(item.start + match.index + length)
+          }, fileForTag));
+        }
+      }
+    }
+  }
+  for (const item of program.body) {
+    visit(item);
+  }
+  return links;
+}
+
+// src/service/marko/document-links/index.ts
+var cache = /* @__PURE__ */ new WeakMap();
+var findDocumentLinks = async (doc) => {
+  const parsed = parse2(doc);
+  let result = cache.get(parsed);
+  if (!result) {
+    result = extractDocumentLinks(doc, parsed, getCompilerInfo(doc).lookup);
+    cache.set(parsed, result);
+  }
+  return result;
+};
+
+// src/service/marko/format.ts
+import { Range as Range7, TextEdit as TextEdit5 } from "vscode-languageserver";
+import { URI as URI7 } from "vscode-uri";
 import * as prettier from "prettier";
 import * as markoPrettier from "prettier-plugin-marko";
 var format2 = async (doc, params, cancel) => {
   try {
-    const { fsPath, scheme } = URI6.parse(doc.uri);
+    const { fsPath, scheme } = URI7.parse(doc.uri);
     const text = doc.getText();
     const options = {
       parser: "marko",
@@ -1059,7 +1264,7 @@ var format2 = async (doc, params, cancel) => {
     if (cancel.isCancellationRequested)
       return;
     return [
-      TextEdit4.replace(Range6.create(doc.positionAt(0), doc.positionAt(text.length)), prettier.format(text, options))
+      TextEdit5.replace(Range7.create(doc.positionAt(0), doc.positionAt(text.length)), prettier.format(text, options))
     ];
   } catch (e) {
     displayError(e);
@@ -1071,13 +1276,14 @@ var marko_default = {
   doComplete,
   doValidate,
   findDefinition,
+  findDocumentLinks,
   format: format2
 };
 
 // src/service/stylesheet/index.ts
 import {
   CompletionList as CompletionList2,
-  Range as Range8,
+  Range as Range9,
   TextDocumentEdit
 } from "vscode-languageserver";
 import {
@@ -1223,7 +1429,7 @@ function extractStyleSheets(code, program, lookup) {
           for (const attr of node.attrs) {
             if (attr.type === 8 /* AttrNamed */ && ((_a = attr.value) == null ? void 0 : _a.type) === 11 /* AttrValue */ && /^['"]$/.test(code[attr.value.value.start])) {
               const name = read(attr.name);
-              if (name === "#style" || name === "style" && lookup && node.nameText && name === "style" && ((_b = lookup.getTag(node.nameText)) == null ? void 0 : _b.html)) {
+              if (name === "#style" || name === "style" && node.nameText && name === "style" && ((_b = lookup.getTag(node.nameText)) == null ? void 0 : _b.html)) {
                 getExtractor("css").write`:root{${{
                   start: attr.value.value.start + 1,
                   end: attr.value.value.end - 1
@@ -1245,13 +1451,17 @@ function extractStyleSheets(code, program, lookup) {
 }
 
 // src/service/stylesheet/index.ts
-var cache = /* @__PURE__ */ new WeakMap();
+var cache2 = /* @__PURE__ */ new WeakMap();
 var services = {
   css: getCSSLanguageService,
   less: getLESSLanguageService,
   scss: getSCSSLanguageService
 };
+var clientCapabilities;
 var StyleSheetService = {
+  initialize(params) {
+    clientCapabilities = params.capabilities;
+  },
   async doComplete(doc, params) {
     const infoByExt = getStyleSheetInfo(doc);
     const sourceOffset = doc.offsetAt(params.position);
@@ -1261,28 +1471,31 @@ var StyleSheetService = {
       if (generatedOffset === void 0)
         continue;
       const { service: service2, virtualDoc } = info;
-      const result = service2.doComplete(virtualDoc, virtualDoc.positionAt(generatedOffset), info.parsed);
-      for (const item of result.items) {
-        if (item.additionalTextEdits) {
-          for (const textEdit2 of item.additionalTextEdits) {
-            updateTextEdit(doc, info, textEdit2);
+      const result = await service2.doComplete2(virtualDoc, virtualDoc.positionAt(generatedOffset), info.parsed, { resolveReference: resolveUrl });
+      if (result.itemDefaults) {
+        const { editRange } = result.itemDefaults;
+        if (editRange) {
+          if ("start" in editRange) {
+            result.itemDefaults.editRange = getSourceRange(doc, info, editRange);
+          } else {
+            editRange.insert = getSourceRange(doc, info, editRange.insert);
+            editRange.replace = getSourceRange(doc, info, editRange.replace);
           }
         }
-        const { textEdit } = item;
-        if (textEdit) {
-          if (textEdit.range) {
-            updateTextEdit(doc, info, textEdit);
-          }
-          if (textEdit.insert) {
-            updateInsertReplaceEdit(doc, info, textEdit);
-          }
+      }
+      for (const item of result.items) {
+        if (item.textEdit) {
+          item.textEdit = getSourceInsertReplaceEdit(doc, info, item.textEdit);
+        }
+        if (item.additionalTextEdits) {
+          item.additionalTextEdits = getSourceEdits(doc, info, item.additionalTextEdits);
         }
       }
       return result;
     }
     return CompletionList2.create([], true);
   },
-  async findDefinition(doc, params) {
+  findDefinition(doc, params) {
     const infoByExt = getStyleSheetInfo(doc);
     const sourceOffset = doc.offsetAt(params.position);
     for (const ext in infoByExt) {
@@ -1292,29 +1505,101 @@ var StyleSheetService = {
         continue;
       const { service: service2, virtualDoc } = info;
       const result = service2.findDefinition(virtualDoc, virtualDoc.positionAt(generatedOffset), info.parsed);
-      if (result && updateRange(doc, info, result.range)) {
-        return result;
+      if (result) {
+        const range = getSourceRange(doc, info, result.range);
+        if (range) {
+          return {
+            range,
+            uri: doc.uri
+          };
+        }
       }
       break;
     }
   },
-  async findDocumentColors(doc) {
+  findReferences(doc, params) {
+    const infoByExt = getStyleSheetInfo(doc);
+    const sourceOffset = doc.offsetAt(params.position);
+    for (const ext in infoByExt) {
+      const info = infoByExt[ext];
+      const generatedOffset = info.generatedOffsetAt(sourceOffset);
+      if (generatedOffset === void 0)
+        continue;
+      const { service: service2, virtualDoc } = info;
+      const result = [];
+      for (const location of service2.findReferences(virtualDoc, virtualDoc.positionAt(generatedOffset), info.parsed)) {
+        const range = getSourceRange(doc, info, location.range);
+        if (range) {
+          result.push({
+            range,
+            uri: location.uri
+          });
+        }
+      }
+      return result.length ? result : void 0;
+    }
+  },
+  async findDocumentLinks(doc) {
     const infoByExt = getStyleSheetInfo(doc);
     const result = [];
     for (const ext in infoByExt) {
       const info = infoByExt[ext];
       const { service: service2, virtualDoc } = info;
-      for (const colorInfo of service2.findDocumentColors(virtualDoc, info.parsed)) {
-        if (updateRange(doc, info, colorInfo.range)) {
-          result.push(colorInfo);
+      for (const link of await service2.findDocumentLinks2(virtualDoc, info.parsed, { resolveReference: resolveUrl })) {
+        const range = getSourceRange(doc, info, link.range);
+        if (range) {
+          result.push({
+            range,
+            target: link.target,
+            tooltip: link.tooltip,
+            data: link.data
+          });
         }
       }
     }
-    if (result.length) {
-      return result;
+    return result.length ? result : void 0;
+  },
+  findDocumentHighlights(doc, params) {
+    const infoByExt = getStyleSheetInfo(doc);
+    const sourceOffset = doc.offsetAt(params.position);
+    for (const ext in infoByExt) {
+      const info = infoByExt[ext];
+      const generatedOffset = info.generatedOffsetAt(sourceOffset);
+      if (generatedOffset === void 0)
+        continue;
+      const { service: service2, virtualDoc } = info;
+      const result = [];
+      for (const highlight of service2.findDocumentHighlights(virtualDoc, virtualDoc.positionAt(generatedOffset), info.parsed)) {
+        const range = getSourceRange(doc, info, highlight.range);
+        if (range) {
+          result.push({
+            range,
+            kind: highlight.kind
+          });
+        }
+      }
+      return result.length ? result : void 0;
+    }
+  },
+  findDocumentColors(doc) {
+    const infoByExt = getStyleSheetInfo(doc);
+    const result = [];
+    for (const ext in infoByExt) {
+      const info = infoByExt[ext];
+      const { service: service2, virtualDoc } = info;
+      for (const colorInfo of service2.findDocumentColors(virtualDoc, info.parsed)) {
+        const range = getSourceRange(doc, info, colorInfo.range);
+        if (range) {
+          result.push({
+            range,
+            color: colorInfo.color
+          });
+        }
+      }
     }
+    return result.length ? result : void 0;
   },
-  async getColorPresentations(doc, params) {
+  getColorPresentations(doc, params) {
     const infoByExt = getStyleSheetInfo(doc);
     const sourceOffset = doc.offsetAt(params.range.start);
     for (const ext in infoByExt) {
@@ -1326,21 +1611,22 @@ var StyleSheetService = {
       if (generatedOffsetEnd === void 0)
         continue;
       const { service: service2, virtualDoc } = info;
-      const result = service2.getColorPresentations(virtualDoc, info.parsed, params.color, Range8.create(virtualDoc.positionAt(generatedOffsetStart), virtualDoc.positionAt(generatedOffsetEnd)));
-      for (const colorPresentation of result) {
-        if (colorPresentation.textEdit) {
-          updateTextEdit(doc, info, colorPresentation.textEdit);
-        }
-        if (colorPresentation.additionalTextEdits) {
-          for (const textEdit of colorPresentation.additionalTextEdits) {
-            updateTextEdit(doc, info, textEdit);
-          }
+      const result = [];
+      for (const colorPresentation of service2.getColorPresentations(virtualDoc, info.parsed, params.color, Range9.create(virtualDoc.positionAt(generatedOffsetStart), virtualDoc.positionAt(generatedOffsetEnd)))) {
+        const textEdit = colorPresentation.textEdit && getSourceEdit(doc, info, colorPresentation.textEdit);
+        const additionalTextEdits = colorPresentation.additionalTextEdits && getSourceEdits(doc, info, colorPresentation.additionalTextEdits);
+        if (textEdit || additionalTextEdits) {
+          result.push({
+            label: colorPresentation.label,
+            textEdit,
+            additionalTextEdits
+          });
         }
       }
-      return result;
+      return result.length ? result : void 0;
     }
   },
-  async doHover(doc, params) {
+  doHover(doc, params) {
     const infoByExt = getStyleSheetInfo(doc);
     const sourceOffset = doc.offsetAt(params.position);
     for (const ext in infoByExt) {
@@ -1350,8 +1636,18 @@ var StyleSheetService = {
         continue;
       const { service: service2, virtualDoc } = info;
       const result = service2.doHover(virtualDoc, virtualDoc.positionAt(generatedOffset), info.parsed);
-      if (result && (!result.range || updateRange(doc, info, result.range))) {
-        return result;
+      if (result) {
+        if (result.range) {
+          const range = getSourceRange(doc, info, result.range);
+          if (range) {
+            return {
+              range,
+              contents: result.contents
+            };
+          }
+        } else {
+          return result;
+        }
       }
     }
   },
@@ -1368,9 +1664,7 @@ var StyleSheetService = {
       if (result.changes) {
         for (const uri in result.changes) {
           if (uri === doc.uri) {
-            for (const textEdit of result.changes[uri]) {
-              updateTextEdit(doc, info, textEdit);
-            }
+            result.changes[uri] = getSourceEdits(doc, info, result.changes[uri]) || [];
           }
         }
       }
@@ -1378,9 +1672,7 @@ var StyleSheetService = {
         for (const change of result.documentChanges) {
           if (TextDocumentEdit.is(change)) {
             if (change.textDocument.uri === doc.uri) {
-              for (const textEdit of change.edits) {
-                updateTextEdit(doc, info, textEdit);
-              }
+              change.edits = getSourceEdits(doc, info, change.edits) || [];
             }
           }
         }
@@ -1388,7 +1680,7 @@ var StyleSheetService = {
       return result;
     }
   },
-  async doCodeActions(doc, params) {
+  doCodeActions(doc, params) {
     var _a;
     const infoByExt = getStyleSheetInfo(doc);
     const sourceOffset = doc.offsetAt(params.range.start);
@@ -1401,65 +1693,100 @@ var StyleSheetService = {
       if (generatedOffsetEnd === void 0)
         continue;
       const { service: service2, virtualDoc } = info;
-      const result = service2.doCodeActions(virtualDoc, Range8.create(virtualDoc.positionAt(generatedOffsetStart), virtualDoc.positionAt(generatedOffsetEnd)), params.context, info.parsed);
-      if (result) {
-        for (const command of result) {
-          const edits = (_a = command.arguments) == null ? void 0 : _a[2];
-          if (edits) {
-            for (const textEdit of edits) {
-              updateTextEdit(doc, info, textEdit);
-            }
-          }
+      const result = service2.doCodeActions(virtualDoc, Range9.create(virtualDoc.positionAt(generatedOffsetStart), virtualDoc.positionAt(generatedOffsetEnd)), params.context, info.parsed);
+      for (const command of result) {
+        const edits = (_a = command.arguments) == null ? void 0 : _a[2];
+        if (edits && Array.isArray(edits) && isTextEdit(edits[0])) {
+          command.arguments[2] = getSourceEdits(doc, info, edits);
         }
-        return result;
       }
+      return result;
     }
   },
-  async doValidate(doc) {
+  doValidate(doc) {
     const infoByExt = getStyleSheetInfo(doc);
     const result = [];
     for (const ext in infoByExt) {
       const info = infoByExt[ext];
       for (const diag of info.service.doValidation(info.virtualDoc, info.parsed)) {
-        if (updateRange(doc, info, diag.range)) {
+        const range = getSourceRange(doc, info, diag.range);
+        if (range) {
+          diag.range = range;
           result.push(diag);
         }
       }
     }
-    return result;
+    return result.length ? result : void 0;
   }
 };
-function updateTextEdit(doc, info, textEdit) {
-  if (!updateRange(doc, info, textEdit.range)) {
-    textEdit.newText = "";
-    textEdit.range = START_OF_FILE;
+function getSourceEdits(doc, info, edits) {
+  const result = [];
+  for (const edit of edits) {
+    const sourceEdit = getSourceEdit(doc, info, edit);
+    if (sourceEdit) {
+      result.push(sourceEdit);
+    }
   }
+  return result.length ? result : void 0;
 }
-function updateInsertReplaceEdit(doc, info, insertReplaceEdit) {
-  if (!updateRange(doc, info, insertReplaceEdit.insert)) {
-    insertReplaceEdit.newText = "";
-    insertReplaceEdit.insert = START_OF_FILE;
+function getSourceEdit(doc, info, textEdit) {
+  const range = getSourceRange(doc, info, textEdit.range);
+  if (range) {
+    return {
+      newText: textEdit.newText,
+      range
+    };
   }
 }
-function updateRange(doc, info, range) {
+function getSourceInsertReplaceEdit(doc, info, textEdit) {
+  if (isTextEdit(textEdit)) {
+    return getSourceEdit(doc, info, textEdit);
+  } else if (textEdit.replace) {
+    const range = getSourceRange(doc, info, textEdit.replace);
+    if (range) {
+      return {
+        newText: textEdit.newText,
+        replace: range
+      };
+    }
+  } else {
+    const range = getSourceRange(doc, info, textEdit.insert);
+    if (range) {
+      return {
+        newText: textEdit.newText,
+        insert: range
+      };
+    }
+  }
+}
+function getSourceRange(doc, info, range) {
   const start = info.sourceOffsetAt(info.virtualDoc.offsetAt(range.start));
-  const end = info.sourceOffsetAt(info.virtualDoc.offsetAt(range.end));
-  if (start !== void 0 || end !== void 0) {
-    range.start = doc.positionAt(start ?? end);
-    range.end = doc.positionAt(end ?? start);
-    return true;
+  if (start === void 0)
+    return;
+  let end = start;
+  if (range.start.line !== range.end.line || range.start.character !== range.end.character) {
+    end = info.sourceOffsetAt(info.virtualDoc.offsetAt(range.end));
+    if (end === void 0)
+      return;
   }
-  return false;
+  const pos = doc.positionAt(start);
+  return {
+    start: pos,
+    end: start === end ? pos : doc.positionAt(end)
+  };
 }
 function getStyleSheetInfo(doc) {
   var _a;
   const parsed = parse2(doc);
-  let cached = cache.get(parsed);
+  let cached = cache2.get(parsed);
   if (!cached) {
     const results = extractStyleSheets(doc.getText(), parsed.program, getCompilerInfo(doc).lookup);
     cached = {};
     for (const ext in results) {
-      const service2 = (_a = services[ext]) == null ? void 0 : _a.call(services);
+      const service2 = (_a = services[ext]) == null ? void 0 : _a.call(services, {
+        fileSystemProvider: file_system_default,
+        clientCapabilities
+      });
       if (!service2)
         continue;
       const { generated, sourceOffsetAt, generatedOffsetAt } = results[ext];
@@ -1472,14 +1799,23 @@ function getStyleSheetInfo(doc) {
         parsed: service2.parseStylesheet(virtualDoc)
       };
     }
-    cache.set(parsed, cached);
+    cache2.set(parsed, cached);
   }
   return cached;
 }
+function isTextEdit(edit) {
+  return edit.range !== void 0;
+}
 
 // src/service/index.ts
 var plugins = [marko_default, StyleSheetService];
 var service = {
+  async initialize(params) {
+    await Promise.all(plugins.map((plugin) => {
+      var _a;
+      return (_a = plugin.initialize) == null ? void 0 : _a.call(plugin, params);
+    }));
+  },
   async doComplete(doc, params, cancel) {
     const result = CompletionList3.create([], false);
     try {
@@ -1532,6 +1868,78 @@ var service = {
     }
     return result;
   },
+  async findReferences(doc, params, cancel) {
+    let result;
+    try {
+      const requests = plugins.map((plugin) => {
+        var _a;
+        return (_a = plugin.findReferences) == null ? void 0 : _a.call(plugin, doc, params, cancel);
+      });
+      for (const pending of requests) {
+        const cur = await pending;
+        if (cancel.isCancellationRequested)
+          return;
+        if (cur) {
+          if (result) {
+            result.push(...cur);
+          } else {
+            result = cur;
+          }
+        }
+      }
+    } catch (err) {
+      displayError(err);
+    }
+    return result;
+  },
+  async findDocumentLinks(doc, params, cancel) {
+    let result;
+    try {
+      const requests = plugins.map((plugin) => {
+        var _a;
+        return (_a = plugin.findDocumentLinks) == null ? void 0 : _a.call(plugin, doc, params, cancel);
+      });
+      for (const pending of requests) {
+        const cur = await pending;
+        if (cancel.isCancellationRequested)
+          return;
+        if (cur) {
+          if (result) {
+            result.push(...cur);
+          } else {
+            result = cur;
+          }
+        }
+      }
+    } catch (err) {
+      displayError(err);
+    }
+    return result;
+  },
+  async findDocumentHighlights(doc, params, cancel) {
+    let result;
+    try {
+      const requests = plugins.map((plugin) => {
+        var _a;
+        return (_a = plugin.findDocumentHighlights) == null ? void 0 : _a.call(plugin, doc, params, cancel);
+      });
+      for (const pending of requests) {
+        const cur = await pending;
+        if (cancel.isCancellationRequested)
+          return;
+        if (cur) {
+          if (result) {
+            result.push(...cur);
+          } else {
+            result = cur;
+          }
+        }
+      }
+    } catch (err) {
+      displayError(err);
+    }
+    return result;
+  },
   async findDocumentColors(doc, params, cancel) {
     let result;
     try {
@@ -1705,8 +2113,9 @@ console.error = (...args) => {
 };
 process.on("uncaughtException", console.error);
 process.on("unhandledRejection", console.error);
-connection2.onInitialize(() => {
+connection2.onInitialize(async (params) => {
   setup(connection2);
+  await service.initialize(params);
   return {
     capabilities: {
       textDocumentSync: TextDocumentSyncKind.Incremental,
@@ -1715,7 +2124,10 @@ connection2.onInitialize(() => {
       hoverProvider: true,
       renameProvider: true,
       codeActionProvider: true,
+      referencesProvider: true,
+      documentLinkProvider: { resolveProvider: false },
       colorProvider: true,
+      documentHighlightProvider: true,
       completionProvider: {
         triggerCharacters: [
           ".",
@@ -1755,6 +2167,15 @@ connection2.onCompletion(async (params, cancel) => {
 connection2.onDefinition(async (params, cancel) => {
   return await service.findDefinition(documents.get(params.textDocument.uri), params, cancel) || null;
 });
+connection2.onReferences(async (params, cancel) => {
+  return await service.findReferences(documents.get(params.textDocument.uri), params, cancel) || null;
+});
+connection2.onDocumentLinks(async (params, cancel) => {
+  return await service.findDocumentLinks(documents.get(params.textDocument.uri), params, cancel) || null;
+});
+connection2.onDocumentHighlight(async (params, cancel) => {
+  return await service.findDocumentHighlights(documents.get(params.textDocument.uri), params, cancel) || null;
+});
 connection2.onDocumentColor(async (params, cancel) => {
   return await service.findDocumentColors(documents.get(params.textDocument.uri), params, cancel) || null;
 });