@rohal12/spindle-lsp 0.15.22 → 0.15.24

package/dist/bin.js

@@ -55,7 +55,7 @@ var VERSION;
 var init_version = __esm({
   "src/version.ts"() {
     "use strict";
-    VERSION = false ? pkg.version : "0.15.22";
+    VERSION = false ? pkg.version : "0.15.24";
   }
 });
 
@@ -88149,7 +88149,7 @@ function computeCodeLenses(uri, workspace) {
       command: { title }
     });
   };
-  const storyVarsPassage = workspace.passages.getStoryVariables();
+  const variableLines = new Map([...workspace.variables.getDeclared().keys()].flatMap((name) => workspace.variables.getDeclarations(name, "$").filter((declaration) => declaration.uri === uri).map((declaration) => [declaration.range.start.line, name])));
   const widgetLines = new Map(workspace.widgets.getAllWidgets().filter((w) => w.uri === uri).map((w) => [w.range.start.line, w.name]));
   for (let lineNum = 0; lineNum < lines.length; lineNum++) {
     const line = lines[lineNum];
@@ -88166,13 +88166,10 @@ function computeCodeLenses(uri, workspace) {
       const usageCount = findWidgetReferences(widgetName2, workspace, false, false).length;
       addLens(lineNum, `${usageCount} usage${usageCount !== 1 ? "s" : ""}`);
     }
-    if (storyVarsPassage && storyVarsPassage.uri === uri && lineNum > storyVarsPassage.range.start.line && lineNum <= storyVarsPassage.range.end.line) {
-      const varDeclMatch = line.match(/^\$(\w+)\s*=/);
-      if (varDeclMatch) {
-        const varName = varDeclMatch[1];
-        const usageCount = findVariableReferences(varName, workspace, false).length;
-        addLens(lineNum, `${usageCount} usage${usageCount !== 1 ? "s" : ""}`);
-      }
+    const varName = variableLines.get(lineNum);
+    if (varName !== void 0) {
+      const usageCount = findVariableReferences(varName, workspace, false).length;
+      addLens(lineNum, `${usageCount} usage${usageCount !== 1 ? "s" : ""}`);
     }
   }
   return lenses;