2 files changed, 49 insertions, 0 deletions
diff --git a/plugin/src/brackets.js b/plugin/src/brackets.js
new file mode 100644
index 0000000..ad79ea5
--- /dev/null
+++ b/plugin/src/brackets.js
@@ -0,0 +1,25 @@
+'use strict';
+
+const { codeOnly } = require('./lexical');
+const pairs = { ')': '(', ']': '[' };
+
+function unmatchedBrackets(text) {
+	const code = codeOnly(text);
+	const stack = [];
+	const unmatched = [];
+	for (let offset = 0; offset < code.length; offset++) {
+		const char = code[offset];
+		if (char === '(' || char === '[') {
+			stack.push({ char, offset });
+		} else if (pairs[char]) {
+			const opening = stack.pop();
+			if (!opening || opening.char !== pairs[char]) {
+				if (opening) unmatched.push(opening.offset);
+				unmatched.push(offset);
+			}
+		}
+	}
+	return unmatched.concat(stack.map(opening => opening.offset)).sort((a, b) => a - b);
+}
+
+module.exports = { unmatchedBrackets };
diff --git a/plugin/src/extension.js b/plugin/src/extension.js
index a7b4834..68bb7fd 100644
--- a/plugin/src/extension.js
+++ b/plugin/src/extension.js
@@ -3,6 +3,7 @@
 const vscode = require('vscode');
 const { indentationEdits } = require('./indentation');
 const { codeOnly } = require('./lexical');
+const { unmatchedBrackets } = require('./brackets');
 const languageId = 'cie-pseudocode';
 
 function formatLines(document, startLine, endLine, keepBlankIndent = false) {
@@ -12,7 +13,29 @@ function formatLines(document, startLine, endLine, keepBlankIndent = false) {
 }
 
 function activate(context) {
+	const invalidBracket = vscode.window.createTextEditorDecorationType({
+		dark: { color: '#F44747' }
+	});
+	function refreshBrackets(document) {
+		const rangesByDocument = new Map();
+		for (const editor of vscode.window.visibleTextEditors) {
+			if (document && editor.document !== document) continue;
+			if (editor.document.languageId !== languageId) {
+				editor.setDecorations(invalidBracket, []);
+				continue;
+			}
+			if (!rangesByDocument.has(editor.document)) {
+				const ranges = unmatchedBrackets(editor.document.getText()).map(offset =>
+					new vscode.Range(editor.document.positionAt(offset), editor.document.positionAt(offset + 1)));
+				rangesByDocument.set(editor.document, ranges);
+			}
+			editor.setDecorations(invalidBracket, rangesByDocument.get(editor.document));
+		}
+	}
 	context.subscriptions.push(
+		invalidBracket,
+		vscode.workspace.onDidChangeTextDocument(event => refreshBrackets(event.document)),
+		vscode.window.onDidChangeVisibleTextEditors(() => refreshBrackets()),
 		vscode.languages.registerDocumentRangeFormattingEditProvider(languageId, {
 			provideDocumentRangeFormattingEdits(document, range) {
 				const endLine = range.end.character === 0 && range.end.line > range.start.line
@@ -28,6 +51,7 @@ function activate(context) {
 			}
 		}, '\n', 'E', 'F', 'L', 'T', 'N')
 	);
+	refreshBrackets();
 }
 
 module.exports = { activate };