release: WRNexusJS 0.2.36

editors/vscode/src/extension.js

@@ -271,10 +271,227 @@ function activate(context) {
   registerCompletionProvider(context);
   registerDefinitionProvider(context);
   registerFormatter(context);
+
+  context.subscriptions.push(
+    vscode.languages.registerDocumentSemanticTokensProvider(
+      {
+        language: "wrn",
+      },
+      wrnSemanticTokensProvider,
+      semanticTokenLegend,
+    ),
+  );
 }
 
 function deactivate() {}
 
+const semanticTokenLegend = new vscode.SemanticTokensLegend(
+  ["variable"],
+  ["declaration", "modification"],
+);
+
+/**
+ * @param {string} value
+ * @returns {string}
+ */
+function escapeRegExp(value) {
+  return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+}
+
+/**
+ * @param {string} text
+ * @returns {Map<string, Set<number>>}
+ */
+function collectStateVariables(text) {
+  const states = new Map();
+  const statePattern = /\bstate\s+([A-Za-z_$][A-Za-z0-9_$]*)\s*(?==|;|\r?$)/gm;
+
+  let match;
+
+  while ((match = statePattern.exec(text)) !== null) {
+    const name = match[1];
+
+    if (!name) {
+      continue;
+    }
+
+    const offset = match.index + match[0].lastIndexOf(name);
+    const declarations = states.get(name) || new Set();
+
+    declarations.add(offset);
+    states.set(name, declarations);
+  }
+
+  return states;
+}
+
+/**
+ * Comments are excluded, but quoted attribute expressions are deliberately
+ * included because WRN state references commonly appear in values such as
+ * class:flex="centered" and @click="enabled = !enabled".
+ *
+ * @param {string} text
+ * @returns {Array<{start: number, end: number}>}
+ */
+function collectCommentRanges(text) {
+  const ranges = [];
+  let index = 0;
+  let quote = null;
+  let escaped = false;
+
+  while (index < text.length) {
+    const current = text[index];
+    const next = text[index + 1];
+
+    if (quote) {
+      if (escaped) {
+        escaped = false;
+        index++;
+        continue;
+      }
+
+      if (current === "\\") {
+        escaped = true;
+        index++;
+        continue;
+      }
+
+      if (current === quote) {
+        quote = null;
+      }
+
+      index++;
+      continue;
+    }
+
+    if (current === '"' || current === "'" || current === "`") {
+      quote = current;
+      index++;
+      continue;
+    }
+
+    if (current === "/" && next === "/") {
+      const start = index;
+      index += 2;
+
+      while (index < text.length && text[index] !== "\n") {
+        index++;
+      }
+
+      ranges.push({ start, end: index });
+      continue;
+    }
+
+    if (current === "/" && next === "*") {
+      const start = index;
+      index += 2;
+
+      while (index < text.length && !(text[index] === "*" && text[index + 1] === "/")) {
+        index++;
+      }
+
+      index = Math.min(text.length, index + 2);
+      ranges.push({ start, end: index });
+      continue;
+    }
+
+    index++;
+  }
+
+  return ranges;
+}
+
+/**
+ * @param {number} offset
+ * @param {Array<{start: number, end: number}>} ranges
+ * @returns {boolean}
+ */
+function isInsideComment(offset, ranges) {
+  let low = 0;
+  let high = ranges.length - 1;
+
+  while (low <= high) {
+    const middle = Math.floor((low + high) / 2);
+    const range = ranges[middle];
+
+    if (!range) {
+      return false;
+    }
+
+    if (offset < range.start) {
+      high = middle - 1;
+    } else if (offset >= range.end) {
+      low = middle + 1;
+    } else {
+      return true;
+    }
+  }
+
+  return false;
+}
+
+/**
+ * @param {string} text
+ * @param {number} offset
+ * @param {number} length
+ * @returns {boolean}
+ */
+function isStateModification(text, offset, length) {
+  const after = text.slice(offset + length).match(/^\s*(=|\+=|-=|\*=|\/=|%=|\+\+|--)/);
+
+  if (after) {
+    return true;
+  }
+
+  const before = text.slice(Math.max(0, offset - 8), offset).match(/(\+\+|--)\s*$/);
+
+  return Boolean(before);
+}
+
+const wrnSemanticTokensProvider = {
+  /**
+   * @param {vscode.TextDocument} document
+   * @param {vscode.CancellationToken} token
+   */
+  provideDocumentSemanticTokens(document, token) {
+    const builder = new vscode.SemanticTokensBuilder(semanticTokenLegend);
+    const text = document.getText();
+    const states = collectStateVariables(text);
+    const commentRanges = collectCommentRanges(text);
+
+    for (const [name, declarationOffsets] of states) {
+      if (token.isCancellationRequested) {
+        return builder.build();
+      }
+
+      const pattern = new RegExp(`(?<![A-Za-z0-9_$])${escapeRegExp(name)}(?![A-Za-z0-9_$])`, "g");
+
+      let match;
+
+      while ((match = pattern.exec(text)) !== null) {
+        const offset = match.index;
+
+        if (isInsideComment(offset, commentRanges)) {
+          continue;
+        }
+
+        const position = document.positionAt(offset);
+        let modifiers = [];
+
+        if (declarationOffsets.has(offset)) {
+          modifiers = ["declaration"];
+        } else if (isStateModification(text, offset, name.length)) {
+          modifiers = ["modification"];
+        }
+
+        builder.push(position.line, position.character, name.length, "variable", modifiers);
+      }
+    }
+
+    return builder.build();
+  },
+};
+
 module.exports = {
   activate,
   deactivate,