prettier-plugin-wolfram 0.7.2 → 0.7.4

package/README.md

@@ -210,6 +210,8 @@ The package also exposes a small rule runner:
 npx prettier-wolfram lint "src/**/*.wl"
 ```
 
+Matched directories and non-Wolfram file extensions are skipped.
+
 It prints diagnostics as:
 
 ```text
@@ -263,12 +265,12 @@ npm run publish:vscode:pre-release
 The extension README with editor setup, settings, diagnostics, and file
 association behavior lives at `vscode-extension/README.md`.
 
-## Publishing To Verdaccio
+## Publishing To npm
 
 Log in:
 
 ```bash
-npm login --registry http://localhost:4873
+npm login
 ```
 
 Preview package contents:
@@ -280,11 +282,11 @@ npm pack --dry-run
 Publish:
 
 ```bash
-npm publish --registry http://localhost:4873
+npm publish
 ```
 
 Verify:
 
 ```bash
-npm view prettier-plugin-wolfram --registry http://localhost:4873
+npm view prettier-plugin-wolfram
 ```

package/bin/prettier-wolfram-lsp.js

@@ -0,0 +1,128 @@
+#!/usr/bin/env node
+// bin/prettier-wolfram-lsp.js
+// Minimal LSP server that publishes prettier-wolfram lint diagnostics over stdio.
+
+import { WolframParser } from "../src/parser/index.js";
+import { runRules } from "../src/rules/index.js";
+import { buildOffsetTable, addOffsets } from "../src/utils/offsets.js";
+
+const parser = new WolframParser();
+
+// ── LSP framing ──────────────────────────────────────────────────────────────
+
+let inputBuf = "";
+
+process.stdin.setEncoding("utf8");
+process.stdin.on("data", (chunk) => {
+	inputBuf += chunk;
+	drainBuffer();
+});
+
+function drainBuffer() {
+	while (true) {
+		const sep = inputBuf.indexOf("\r\n\r\n");
+		if (sep === -1) break;
+		const header = inputBuf.slice(0, sep);
+		const m = header.match(/Content-Length:\s*(\d+)/i);
+		if (!m) { inputBuf = inputBuf.slice(sep + 4); continue; }
+		const len = parseInt(m[1], 10);
+		const bodyStart = sep + 4;
+		if (inputBuf.length < bodyStart + len) break;
+		const body = inputBuf.slice(bodyStart, bodyStart + len);
+		inputBuf = inputBuf.slice(bodyStart + len);
+		try { dispatch(JSON.parse(body)); } catch {}
+	}
+}
+
+function send(obj) {
+	const body = JSON.stringify(obj);
+	process.stdout.write(`Content-Length: ${Buffer.byteLength(body, "utf8")}\r\n\r\n${body}`);
+}
+
+const reply  = (id, result) => send({ jsonrpc: "2.0", id, result });
+const notify = (method, params) => send({ jsonrpc: "2.0", method, params });
+
+// ── Lint ─────────────────────────────────────────────────────────────────────
+
+const docs   = new Map(); // uri → text
+const timers = new Map(); // uri → debounce timer
+
+function scheduleLint(uri) {
+	if (timers.has(uri)) clearTimeout(timers.get(uri));
+	timers.set(uri, setTimeout(async () => {
+		timers.delete(uri);
+		const source = docs.get(uri);
+		if (source == null) return;
+		try {
+			const cst  = await parser.getCST(source);
+			const table = buildOffsetTable(source);
+			addOffsets(cst, table);
+			const issues = await runRules(cst, {});
+
+			const diagnostics = issues.map((d) => {
+				const [start, end] = d.node?.source ?? [[1, 1], [1, 1]];
+				return {
+					range: {
+						start: { line: start[0] - 1, character: start[1] - 1 },
+						end:   { line: end[0]   - 1, character: end[1]   - 1 },
+					},
+					severity: d.level === "error" ? 1 : 2,
+					source:   "prettier-wolfram",
+					code:     d.rule,
+					message:  d.message,
+				};
+			});
+
+			notify("textDocument/publishDiagnostics", { uri, diagnostics });
+		} catch {
+			notify("textDocument/publishDiagnostics", { uri, diagnostics: [] });
+		}
+	}, 200));
+}
+
+// ── Dispatch ─────────────────────────────────────────────────────────────────
+
+function dispatch(msg) {
+	const { method, id, params } = msg;
+
+	switch (method) {
+		case "initialize":
+			reply(id, {
+				capabilities: {
+					textDocumentSync: { openClose: true, change: 1 },
+				},
+				serverInfo: { name: "prettier-wolfram-ls", version: "0.1.0" },
+			});
+			break;
+
+		case "initialized":
+			break;
+
+		case "textDocument/didOpen":
+			docs.set(params.textDocument.uri, params.textDocument.text);
+			scheduleLint(params.textDocument.uri);
+			break;
+
+		case "textDocument/didChange":
+			docs.set(params.textDocument.uri, params.contentChanges[0].text);
+			scheduleLint(params.textDocument.uri);
+			break;
+
+		case "textDocument/didClose":
+			docs.delete(params.textDocument.uri);
+			notify("textDocument/publishDiagnostics", { uri: params.textDocument.uri, diagnostics: [] });
+			break;
+
+		case "shutdown":
+			reply(id, null);
+			break;
+
+		case "exit":
+			process.exit(0);
+			break;
+
+		default:
+			if (id != null)
+				send({ jsonrpc: "2.0", id, error: { code: -32601, message: "Method not found" } });
+	}
+}

package/bin/prettier-wolfram.js

@@ -2,12 +2,23 @@
 // bin/prettier-wolfram.js
 // Usage: prettier-wolfram lint [options] <glob...>
 
-import { readFileSync } from "fs";
+import { readFileSync, statSync } from "fs";
 import { globSync } from "fs";
+import { extname } from "path";
 import { WolframParser } from "../src/parser/index.js";
 import { runRules } from "../src/rules/index.js";
 import { buildOffsetTable, addOffsets } from "../src/utils/offsets.js";
 
+const WOLFRAM_EXTENSIONS = new Set([
+	".wl",
+	".wls",
+	".wlt",
+	".mt",
+	".m",
+	".vsnb",
+	".nb",
+]);
+
 const [, , command, ...args] = process.argv;
 
 if (command !== "lint") {
@@ -31,8 +42,14 @@ let totalDiagnostics = 0;
 for (const pattern of args) {
 	const files = globSync(pattern, { absolute: true });
 	for (const file of files) {
-		const source = readFileSync(file, "utf8");
 		try {
+			if (
+				!statSync(file).isFile() ||
+				!WOLFRAM_EXTENSIONS.has(extname(file).toLowerCase())
+			) {
+				continue;
+			}
+			const source = readFileSync(file, "utf8");
 			const cst = await parser.getCST(source);
 			const table = buildOffsetTable(source);
 			addOffsets(cst, table);

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "prettier-plugin-wolfram",
-	"version": "0.7.2",
+	"version": "0.7.4",
 	"description": "Prettier plugin for Wolfram Language using tree-sitter",
 	"license": "MIT",
 	"repository": {
@@ -21,7 +21,8 @@
 		".": "./src/index.js"
 	},
 	"bin": {
-		"prettier-wolfram": "bin/prettier-wolfram.js"
+		"prettier-wolfram": "bin/prettier-wolfram.js",
+		"prettier-wolfram-lsp": "bin/prettier-wolfram-lsp.js"
 	},
 	"files": [
 		"bin/",

package/src/parser/adapter.js

@@ -7,12 +7,18 @@ const GROUP_KIND = { "{": "List", "(": "GroupParen", "[": "Group", "<|": "Associ
 const GROUP_OPEN_LEAF = { "{": "Token`OpenCurly", "(": "Token`OpenParen", "[": "Token`OpenSquare", "<|": "Token`LessBar" };
 const GROUP_CLOSE_LEAF = { "}": "Token`CloseCurly", ")": "Token`CloseParen", "]": "Token`CloseSquare", "|>": "Token`BarGreater" };
 
-export function adapt(tree, source) {
-	const lineIndex = makeLineIndex(source);
-	const ctx = { source, lineIndex };
+// preprocessedSource is the version passed to tree-sitter (may have ⁢ for InvisibleTimes);
+// source is the original — used only for the unformattable fallback.
+export function adapt(tree, source, preprocessedSource) {
+	const ps = preprocessedSource ?? source;
+	const lineIndex = makeLineIndex(ps);
+	const ctx = { source: ps, lineIndex };
 	const root = tree.rootNode;
 	if (subtreeHasError(root)) {
-		const src = nodeSource(root, lineIndex);
+		// Use original source positions for the error fallback so the printer can
+		// emit the unmodified source text verbatim.
+		const errLineIndex = makeLineIndex(source);
+		const src = nodeSource(root, errLineIndex);
 		return { type: "ContainerNode", kind: "String", children: [{ type: "Unknown", kind: "SyntaxErrorNode[]", source: src }], source: src };
 	}
 	// Hoist top-level semicolon chains that end with a trailing ";" (MISSING rhs) into separate
@@ -289,17 +295,34 @@ function delimLeaf(node, kind, value, ctx) {
 	return { type: "LeafNode", kind, value, source: nodeSource(node, ctx.lineIndex) };
 }
 
-// Produce trivia (whitespace/newline) LeafNodes for the gap between two character offsets.
-// Splits on newlines: each run of spaces produces Token`Whitespace, each "\n" produces Token`Newline.
+// Produce trivia LeafNodes for the gap between two character offsets.
+// Splits on newlines and (*...*) comments (which may appear between tokens in right-associative
+// binary nodes like :=). Each whitespace run → Token`Whitespace, each "\n" → Token`Newline,
+// each (*...*) comment → Token`Comment.
 function triviaLeaves(fromIdx, toIdx, ctx) {
 	if (fromIdx >= toIdx) return [];
 	const gap = ctx.source.slice(fromIdx, toIdx);
-	if (!/[\s]/.test(gap)) return [];
+	if (gap.length === 0) return [];
 	const leaves = [];
 	let i = 0;
 	while (i < gap.length) {
 		const ch = gap[i];
-		if (ch === "\n") {
+		// Nested WL comment (*...*)
+		if (ch === "(" && gap[i + 1] === "*") {
+			const start = i;
+			i += 2;
+			let depth = 1;
+			while (i < gap.length && depth > 0) {
+				if (gap[i] === "(" && gap[i + 1] === "*") { depth++; i += 2; }
+				else if (gap[i] === "*" && gap[i + 1] === ")") { depth--; i += 2; }
+				else i++;
+			}
+			const commentText = gap.slice(start, i);
+			const startChar = fromIdx + start;
+			const endChar = fromIdx + i;
+			const src = [offsetToLineCol(ctx.lineIndex, startChar), offsetToLineCol(ctx.lineIndex, endChar)];
+			leaves.push({ type: "LeafNode", kind: "Token`Comment", value: commentText, source: src });
+		} else if (ch === "\n") {
 			const endChar = fromIdx + i + 1;
 			const src = [offsetToLineCol(ctx.lineIndex, fromIdx + i), offsetToLineCol(ctx.lineIndex, endChar)];
 			leaves.push({ type: "LeafNode", kind: "Token`Newline", value: "\n", source: src });
@@ -311,13 +334,19 @@ function triviaLeaves(fromIdx, toIdx, ctx) {
 			leaves.push({ type: "LeafNode", kind: "Token`Newline", value: nl, source: src });
 			i += nl.length;
 		} else {
-			// Collect run of whitespace (non-newline)
+			// Collect run of non-comment, non-newline chars
 			let j = i;
-			while (j < gap.length && gap[j] !== "\n" && gap[j] !== "\r") j++;
+			while (j < gap.length && gap[j] !== "\n" && gap[j] !== "\r" && !(gap[j] === "(" && gap[j + 1] === "*")) j++;
+			if (j === i) { i++; continue; } // safety: skip single unknown char
 			const ws = gap.slice(i, j);
 			const startChar = fromIdx + i;
 			const endChar = fromIdx + j;
 			const src = [offsetToLineCol(ctx.lineIndex, startChar), offsetToLineCol(ctx.lineIndex, endChar)];
+			if (/\S/.test(ws)) {
+				// Non-whitespace that isn't a comment — shouldn't happen in valid WL, skip it
+				i = j;
+				continue;
+			}
 			leaves.push({ type: "LeafNode", kind: "Token`Whitespace", value: ws, source: src });
 			i = j;
 		}
@@ -361,6 +390,8 @@ const TOKEN_KIND_NAME = {
 	// additional infix operators missing from original table
 	"**": "StarStar", "|": "Bar", "||": "BarBar", "&&": "AmpAmp",
 	"@*": "AtStar", "/*": "SlashStar",
+	// InvisibleTimes: WL space-multiplication, encoded as U+2062 during preprocessing
+	"⁢": "InvisibleTimes",
 };
 
 const INEQUALITY_OPS = new Set(["<", "<=", ">", ">=", "==", "!=", "===", "=!="]);

package/src/parser/index.js

@@ -18,12 +18,64 @@ async function getLanguage() {
 	return _langPromise;
 }
 
+// Replace space-based implicit multiplication (a  b) with U+2062 (InvisibleTimes)
+// so the grammar can parse it. Skip content inside strings and nested comments.
+export function preprocessInvisibleTimes(src) {
+	let result = "";
+	let i = 0;
+	const n = src.length;
+	while (i < n) {
+		// Skip quoted string
+		if (src[i] === '"') {
+			const start = i++;
+			while (i < n && src[i] !== '"') {
+				if (src[i] === "\\") i++;
+				i++;
+			}
+			if (i < n) i++;
+			result += src.slice(start, i);
+			continue;
+		}
+		// Skip nested WL comment (* ... *)
+		if (src[i] === "(" && src[i + 1] === "*") {
+			const start = i;
+			i += 2;
+			let depth = 1;
+			while (i < n && depth > 0) {
+				if (src[i] === "(" && src[i + 1] === "*") { depth++; i += 2; }
+				else if (src[i] === "*" && src[i + 1] === ")") { depth--; i += 2; }
+				else i++;
+			}
+			result += src.slice(start, i);
+			continue;
+		}
+		// Two or more spaces between word chars on same line → InvisibleTimes
+		if (src[i] === " " && src[i + 1] === " ") {
+			// Check previous meaningful char is a word char
+			const prevChar = result.length > 0 ? result[result.length - 1] : "";
+			if (/\w/.test(prevChar)) {
+				// Consume all spaces and peek at next non-space char
+				let j = i;
+				while (j < n && src[j] === " ") j++;
+				if (j < n && /\w/.test(src[j])) {
+					result += "⁢"; // InvisibleTimes, spaces stripped (they're extras)
+					i = j;
+					continue;
+				}
+			}
+		}
+		result += src[i++];
+	}
+	return result;
+}
+
 export class WolframParser {
 	async getCST(sourceText) {
 		const lang = await getLanguage();
 		const parser = new Parser();
 		parser.setLanguage(lang);
-		const tree = parser.parse(sourceText);
-		return adapt(tree, sourceText);
+		const preprocessed = preprocessInvisibleTimes(sourceText);
+		const tree = parser.parse(preprocessed);
+		return adapt(tree, sourceText, preprocessed);
 	}
 }

package/src/parser/operators.js

@@ -8,6 +8,8 @@ export const INFIX_OPS = {
 	"==": "Equal", "!=": "Unequal", "<": "Less", "<=": "LessEqual",
 	">": "Greater", ">=": "GreaterEqual",
 	"@*": "Composition", "/*": "RightComposition",
+	// U+2062: WL InvisibleTimes (space-multiplication), inserted by preprocessor
+	"⁢": "InvisibleTimes",
 };
 export const BINARY_OPS = {
 	"=": "Set", ":=": "SetDelayed", "^=": "UpSet", "^:=": "UpSetDelayed",

package/src/translator/nodes/binary.js

@@ -163,7 +163,7 @@ export function printBinary(node, options, print) {
 	const rhsWillBreak =
 		isMultilineStringLeaf(rhs, rhsDoc) || isMultilineStringJoin(rhs);
 
-	if (node.op === "BinaryAt" || node.op === "BinarySlashSlash") {
+	if (node.op === "BinaryAt") {
 		return group([
 			lhsDoc,
 			`${gap}${opStr}`,
@@ -172,6 +172,14 @@ export function printBinary(node, options, print) {
 		]);
 	}
 
+	if (node.op === "BinarySlashSlash") {
+		return group([
+			lhsDoc,
+			`${gap}${opStr}`,
+			indent([space ? line : softline, rhsDoc]),
+		]);
+	}
+
 	if (!space) {
 		if (rhsWillBreak) {
 			return group([lhsDoc, opStr, indent([line, rhsDoc])]);

package/src/translator/nodes/infix.js

@@ -331,6 +331,15 @@ export function printInfix(node, options, print) {
 		return printOriginalSource(node, options);
 	}
 
+	if (node.op === "InvisibleTimes") {
+		const terms = operands(node);
+		const parts = [print(terms[0])];
+		for (let i = 1; i < terms.length; i++) {
+			parts.push(" ", print(terms[i]));
+		}
+		return group(parts);
+	}
+
 	if (node.op === "MessageName") {
 		const parts = operands(node);
 		return group(