patch

+ error linting
+ clearer missing hblang message
+ misc

.gitignore

@@ -6,9 +6,12 @@
 !.vscode/extensions.json
 !.vscode/*.code-snippets
 
+node_modules/*
+
+out/*
+
 # Local History for Visual Studio Code
 .history/
 
 # Built Visual Studio Code Extensions
 *.vsix
-

.vscode/launch.json

@@ -0,0 +1,15 @@
+{
+	"version": "0.1.0",
+	"configurations": [
+		{
+			"name": "Launch Extension",
+			"type": "extensionHost",
+			"request": "launch",
+			"runtimeExecutable": "${execPath}",
+			"args": [
+				"--extensionDevelopmentPath=${workspaceRoot}"
+			],
+			"sourceMaps": true,
+		},
+	]
+}

.vscodeignore

@@ -0,0 +1,13 @@
+# ---> VisualStudioCode
+.vscode/*
+!.vscode/settings.json
+!.vscode/tasks.json
+!.vscode/launch.json
+!.vscode/extensions.json
+!.vscode/*.code-snippets
+
+# Local History for Visual Studio Code
+.history/
+
+# Built Visual Studio Code Extensions
+*.vsix

README.md

@@ -1,3 +1,12 @@
 # hblang-vscode
 
 syntax highlighting for hblang in vscode
+
+# Usage guide:
+
+1. cargo +nightly install hblang --git https://git.ablecorp.us/ableos/holey-bytes
+	> remember to keep this updated
+2. add `$HOME/.cargo/bin` (or wherever you installed hblang) to `$PATH`
+	> in bash, `export PATH=$PATH:$HOME/.cargo/bin`<br>
+	> in other shells, there may be other ways
+3. log out and log back in (or something, idk)

language-configuration.json

@@ -1,31 +1,79 @@
 {
 	"comments": {
 		"lineComment": "//",
-      "blockComment": ["/*", "*/"]
+		"blockComment": [
+			"/*",
+			"*/"
+		]
 	},
 	"brackets": [
-      ["{", "}"],
-      ["[", "]"],
-      ["(", ")"]
+		[
+			"{",
+			"}"
+		],
+		[
+			"[",
+			"]"
+		],
+		[
+			"(",
+			")"
+		]
 	],
 	"autoClosingPairs": [
-      { "open": "{", "close": "}" },
-      { "open": "[", "close": "]" },
-      { "open": "(", "close": ")" },
-      { "open": "\"", "close": "\"", "notIn": ["string"] },
-      { "open": "/*", "close": "*/", "notIn": ["string"] }
+		{
+			"open": "{",
+			"close": "}"
+		},
+		{
+			"open": "[",
+			"close": "]"
+		},
+		{
+			"open": "(",
+			"close": ")"
+		},
+		{
+			"open": "\"",
+			"close": "\"",
+			"notIn": [
+				"string"
+			]
+		},
+		{
+			"open": "/*",
+			"close": "*/",
+			"notIn": [
+				"string"
+			]
+		}
 	],
 	"surroundingPairs": [
-      { "open": "{", "close": "}" },
-      { "open": "[", "close": "]" },
-      { "open": "(", "close": ")" },
-      { "open": "\"", "close": "\"" }
+		{
+			"open": "{",
+			"close": "}"
+		},
+		{
+			"open": "[",
+			"close": "]"
+		},
+		{
+			"open": "(",
+			"close": ")"
+		},
+		{
+			"open": "\"",
+			"close": "\""
+		}
 	],
 	"folding": {
 		"markers": {
-        "start": "^\\s*//\\s*#region\\b",
-        "end": "^\\s*//\\s*#endregion\\b"
+			"start": "{\n",
+			"end": "}"
 		}
-    }
-  }
-  
+	},
+	"indentationRules": {
+		"increaseIndentPattern": "{",
+		"decreaseIndentPattern": "}"
+	},
+}

package.json

@@ -1,27 +1,79 @@
 {
 	"name": "hblang",
-  "repository": {"url": "https://git.ablecorp.us/aurlex/hblang-vscode"},
-  "publisher": "aurlex",
-  "displayName": "hblang",
-  "description": "Syntax highlighting for holey-bytes lang",
-  "version": "0.0.1",
-  "engines": {
-    "vscode": "^1.91.0"
+	"repository": {
+		"url": "https://git.ablecorp.us/koniifer/hblang-vscode"
 	},
+	"publisher": "koniifer",
+	"displayName": "hblang",
+	"description": "Syntax highlighter and formatter for holey-bytes lang",
+	"version": "0.2.2",
+	"engines": {
+		"vscode": "^1.75.0"
+	},
+	"main": "./out/main.js",
 	"categories": [
-    "Programming Languages"
+		"Programming Languages",
+		"Formatters"
+	],
+	"activationEvents": [
+		"onCommand:extension.formatDocument"
 	],
 	"contributes": {
-    "languages": [{
+		"languages": [
+			{
 				"id": "hblang",
-      "aliases": ["hblang", "hblang"],
-      "extensions": [".hb"],
+				"aliases": [
+					"hblang"
+				],
+				"extensions": [
+					".hb"
+				],
 				"configuration": "./language-configuration.json"
-    }],
-    "grammars": [{
+			}
+		],
+		"grammars": [
+			{
 				"language": "hblang",
 				"scopeName": "source.hblang",
 				"path": "./syntaxes/hblang.tmLanguage.json"
-    }]
+			}
+		],
+		"formatters": [
+			{
+				"language": "hblang",
+				"provider": {
+					"provideDocumentFormattingEdits": {
+						"command": "editor.action.formatDocument"
+					}
+				}
+			}
+		],
+		"configuration": {
+			"title": "hblang",
+			"properties": {
+				"hblang.compilerPath": {
+					"type": "string",
+					"default": "PATH",
+					"description": "Set the optional path for the hbc compiler. If set to 'PATH', it will rely on the system to provide hbc.",
+					"scope": "machine-overridable"
+				}
+			}
+		}
+	},
+	"scripts": {
+		"vscode:prepublish": "webpack",
+		"compile": "webpack --mode production",
+		"watch": "webpack --watch --mode development"
+	},
+	"devDependencies": {
+		"@types/vscode": "^1.75.0",
+		"os-browserify": "^0.3.0",
+		"path-browserify": "^1.0.1",
+		"stream-browserify": "^3.0.0",
+		"ts-loader": "^9.5.1",
+		"typescript": "^4.0.0",
+		"vscode": "^1.1.37",
+		"webpack": "^5.0.0",
+		"webpack-cli": "^4.10.0"
 	}
 }

src/main.ts

@@ -0,0 +1,120 @@
+import * as vscode from "vscode";
+import { execFile } from "child_process";
+import * as fs from "fs";
+import * as path from "path";
+
+const DEFAULT_EXECUTABLE = "hbc";
+let diagnosticCollection: vscode.DiagnosticCollection;
+
+function getExecutablePath(): string {
+  const config = vscode.workspace.getConfiguration("hblang");
+  return config.get<string>("compilerPath") === "PATH"
+    ? DEFAULT_EXECUTABLE
+    : config.get<string>("compilerPath")!;
+}
+
+async function runCommand(filePath: string, args: string[]): Promise<string> {
+  return new Promise((resolve, reject) => {
+    execFile(getExecutablePath(), args, (error, stdout, stderr) => {
+      if (error && error.code === "ENOENT") {
+        vscode.window.showErrorMessage(
+          "hblang compiler not found. Ensure 'hbc' is installed and available in PATH."
+        );
+        return reject(new Error(`Compiler not found: ${stderr || stdout}`));
+      }
+      return error
+        ? reject(new Error(stderr || `Exit code: ${error.code}`))
+        : resolve(stdout);
+    });
+  });
+}
+
+async function provideDocumentFormattingEdits(
+  document: vscode.TextDocument
+): Promise<vscode.TextEdit[]> {
+  const tempFilePath = path.join(
+    path.dirname(document.uri.fsPath),
+    `temp_${path.basename(document.uri.fsPath)}`
+  );
+  await fs.promises.writeFile(tempFilePath, document.getText());
+
+  try {
+    const stdout = await runCommand(tempFilePath, [
+      "--fmt-stdout",
+      tempFilePath,
+    ]);
+    const fullRange = new vscode.Range(
+      document.positionAt(0),
+      document.positionAt(document.getText().length)
+    );
+
+    diagnosticCollection.delete(document.uri);
+
+    return [vscode.TextEdit.replace(fullRange, stdout)];
+  } catch (error) {
+    return [];
+  } finally {
+    fs.promises.unlink(tempFilePath).catch(console.error);
+  }
+}
+
+async function lintDocument(document: vscode.TextDocument) {
+  try {
+    const stderr = await runCommand(document.uri.fsPath, [document.uri.fsPath]);
+    diagnosticCollection.set(document.uri, parseLintingErrors(stderr));
+  } catch (error) {
+    if (error instanceof Error) {
+      diagnosticCollection.set(document.uri, parseLintingErrors(error.message));
+    } else {
+      vscode.window.showErrorMessage("Error linting hblang document.");
+    }
+  }
+}
+
+function parseLintingErrors(stderr: string): vscode.Diagnostic[] {
+  return stderr
+    .split("\n")
+    .filter((line) => line && !line.includes("missing main function"))
+    .map((lineText) => {
+      const match = lineText.match(/^([^:]+):(\d+):(\d+):\s*(.+)$/);
+      if (!match) return null;
+
+      const [, , lineStr, columnStr, message] = match;
+      const lineNum = parseInt(lineStr, 10) - 1;
+      const columnNum = parseInt(columnStr, 10) - 1;
+
+      return new vscode.Diagnostic(
+        new vscode.Range(
+          lineNum,
+          columnNum,
+          lineNum,
+          columnNum + message.length
+        ),
+        message,
+        vscode.DiagnosticSeverity.Error
+      );
+    })
+    .filter(Boolean) as vscode.Diagnostic[];
+}
+
+export function activate(context: vscode.ExtensionContext) {
+  diagnosticCollection = vscode.languages.createDiagnosticCollection("hblang");
+  context.subscriptions.push(diagnosticCollection);
+
+  vscode.workspace.onDidSaveTextDocument(async (document) => {
+    if (document.languageId === "hblang") {
+      await lintDocument(document);
+
+      if (diagnosticCollection.get(document.uri)?.length === 0) {
+        vscode.commands.executeCommand(
+          "editor.action.formatDocument",
+          document
+        );
+      }
+    }
+  });
+
+  vscode.languages.registerDocumentFormattingEditProvider("hblang", {
+    provideDocumentFormattingEdits,
+  });
+}

syntaxes/hblang.tmLanguage.json

@@ -1,44 +1,73 @@
 {
 	"scopeName": "source.hblang",
 	"name": "hblang",
-  "fileTypes": ["hb"],
+	"fileTypes": [
+		"hb"
+	],
 	"patterns": [
 		{
-      "include": "#comments"
-    },
-    {
-      "include": "#strings"
-    },
-    {
-      "include": "#keywords"
-    },
-    {
-      "include": "#types"
-    },
-    {
-      "include": "#declarations"
-    },
-    {
-      "include": "#functions"
-    },
-    {
-      "include": "#macros"
-    },
-    {
-      "include": "#structs"
-    },
-    {
-      "include": "#arrays"
-    },
-    {
-      "include": "#pointers"
-    },
-    {
-      "include": "#operators"
+			"include": "#all"
 		}
 	],
 	"repository": {
-    "comments": {
+		"all": {
+			"patterns": [
+				{
+					"include": "#comment"
+				},
+				{
+					"include": "#string"
+				},
+				{
+					"include": "#struct"
+				},
+				{
+					"include": "#keyword"
+				},
+				{
+					"include": "#function"
+				},
+				{
+					"include": "#number"
+				},
+				{
+					"include": "#type"
+				},
+				{
+					"include": "#variable"
+				},
+				{
+					"include": "#operator"
+				},
+				{
+					"include": "#punctuation"
+				},
+				{
+					"include": "#array"
+				}
+			]
+		},
+		"parameter": {
+			"patterns": [
+				{
+					"name": "variable.parameter.hblang",
+					"match": "\\b\\w+\\b"
+				}
+			]
+		},
+		"number": {
+			"patterns": [
+				{
+					"name": "constant.numeric.hblang",
+					"match": "\\b\\d+(\\.\\d+)?\\b"
+				},
+				{
+					"name": "constant.numeric.hex.hblang",
+					"match": "0x[0-9A-Fa-f]+"
+				}
+			]
+		},
+		"comment": {
 			"patterns": [
 				{
 					"name": "comment.block.hblang",
@@ -52,106 +81,141 @@
 				}
 			]
 		},
-    "strings": {
+		"string": {
 			"patterns": [
 				{
 					"name": "string.quoted.double.hblang",
 					"begin": "\"",
-          "end": "\""
+					"end": "\"",
+					"beginCaptures": {
+						"0": {
+							"name": "punctuation.definition.string.begin.hblang"
+						}
+					},
+					"endCaptures": {
+						"0": {
+							"name": "punctuation.definition.string.end.hblang"
+						}
+					},
+					"patterns": [
+						{
+							"name": "constant.character.escape.hblang",
+							"match": "\\\\[\"'\\\\bfnrt]|\\\\(\\{[^}]+\\}|\\[[^]]+\\]|[a-zA-Z])"
+						},
+						{
+							"name": "invalid.illegal.escaped.hblang",
+							"match": "\\\\."
+						}
+					]
 				}
 			]
 		},
-    "keywords": {
+		"keyword": {
 			"patterns": [
 				{
 					"name": "keyword.control.hblang",
-          "match": "\\b(loop|break|if|else|return)\\b"
+					"match": "\\b(loop|break|if|else|return|packed|continue|true|false|struct|idk)\\b"
 				}
 			]
 		},
-    "types": {
+		"type": {
 			"patterns": [
 				{
 					"name": "storage.type.hblang",
-          "match": "\\b(uint|int|void|u\\d+|i\\d+|f\\d+)\\b"
+					"match": "\\b(uint|int|(u|i)(8|16|32)|void|bool|type|never|([A-Z]\\w+))\\b"
 				}
 			]
 		},
-    "declarations": {
+		"variable": {
 			"patterns": [
 				{
-          "name": "variable.other.declaration.hblang",
-          "match": "\\b[A-Z_][A-Z0-9_]*\\b"
+					"name": "variable.parameter.hblang",
+					"match": "\\b\\w+\\s*:=\\b"
 				},
 				{
-          "name": "variable.other.assignment.hblang",
-          "match": "\\b[a-z_][a-z0-9_]*\\b"
+					"name": "variable.parameter.hblang",
+					"match": "\\b\\w+\\s*=\\b"
+				},
+				{
+					"name": "variable.parameter.hblang",
+					"match": "(?<=\\w+\\.)(\\w+)\\b"
 				}
 			]
 		},
-    "operators": {
+		"operator": {
 			"patterns": [
 				{
 					"name": "keyword.operator.hblang",
-          "match": "\\b(\\^|\\*|&|<<|>>|>>>|<<=|>>=|>>>=|\\+|\\-|\\*|\\/|%|\\|\\||&&|!|==|!=|<|<=|>|>=|\\||&|\\^|~|<<|>>|\\+=|-=|\\*=|\\/=|%=|\\|=|&=|\\^=|~=|<<=|>>=|>>>|@\\w+)\\b"
+					"match": "(@|:=|<<=|>>=|>=|<=|^=|\\+=|-=|\\*=|\\/=|%=|\\|=|&=|\\^|\\*|&|&&|\\|\\||<<|>>|\\+|\\-|\\/|%|\\||!|==|!=|<|>|=)"
 				}
 			]
 		},
-    "functions": {
+		"struct": {
+			"patterns": [
+				{
+					"name": "entity.name.type.struct.hblang",
+					"match": "\\b[A-Z]\\w+\\b"
+				}
+			]
+		},
+		"punctuation": {
+			"patterns": [
+				{
+					"name": "punctuation.terminator.statement.hblang",
+					"match": ";|\\.|,|:"
+				}
+			]
+		},
+		"function": {
 			"patterns": [
 				{
 					"name": "entity.name.function.hblang",
-          "match": "\\b[a-z_][a-z0-9_]*\\b(?=\\()"
+					"begin": "(?<!\\w)fn\\s*\\(",
+					"end": "\\)",
+					"captures": {
+						"0": {
+							"name": "keyword.hblang"
 						}
-      ]
 					},
-    "macros": {
 					"patterns": [
 						{
-          "name": "entity.name.macro.hblang",
-          "match": "@\\b[a-z_][a-z0-9_]*\\b"
+							"include": "#all"
+						},
+						{
+							"name": "variable.parameter.hblang",
+							"match": "\\b\\w+\\b(\\s*,\\s*\\b\\w+\\b)*"
 						}
 					]
 				},
-    "structs": {
+				{
+					"name": "entity.name.function.hblang",
+					"begin": "(?:@)?\\b\\w+\\s*\\(",
+					"end": "\\)",
+					"captures": {
+						"0": {
+							"name": "entity.name.function.hblang"
+						}
+					},
 					"patterns": [
 						{
-          "name": "storage.type.struct.hblang",
-          "match": "\\b[a-zA-Z_][a-zA-Z0-9_]*\\s*:=\\s*struct\\b"
+							"include": "#all"
 						},
 						{
-          "name": "storage.type.struct.hblang",
-          "match": "\\b\\.[a-zA-Z_][a-zA-Z0-9_]*\\b"
+							"name": "variable.parameter.hblang",
+							"match": "\\b\\w+\\b(\\s*,\\s*\\b\\w+\\b)*"
 						}
 					]
+				}
+			]
+		}
 	},
-    "arrays": {
+	"array": {
 		"patterns": [
 			{
 				"name": "storage.type.array.hblang",
-          "match": "\\[\\b[a-zA-Z_][a-zA-Z0-9_]*\\b\\]"
-        },
-        {
-          "name": "storage.type.array.hblang",
-          "match": "\\[\\b[a-zA-Z_][a-zA-Z0-9_]*\\b;\\s*\\d+\\]"
+				"begin": "\\.\\(",
+				"end": "\\)"
 			}
 		]
-    },
-    "pointers": {
-      "patterns": [
-        {
-          "name": "storage.modifier.pointer.hblang",
-          "match": "\\^+"
-        },
-        {
-          "name": "keyword.operator.dereference.hblang",
-          "match": "\\*+"
-        },
-        {
-          "name": "keyword.operator.reference.hblang",
-          "match": "\\&+"
-        }
-      ]
-    }
 	}
 }

tests/dont-erase.hb

@@ -0,0 +1,3 @@
+main := fn(): void {
+    undefined_identifier
+}

tsconfig.json

@@ -0,0 +1,15 @@
+{
+	"compilerOptions": {
+		"target": "ES6",
+		"module": "commonjs",
+		"outDir": "./out",
+		"rootDir": "./src",
+		"strict": true,
+		"esModuleInterop": true,
+		"skipLibCheck": true,
+		"forceConsistentCasingInFileNames": true
+	},
+	"include": [
+		"src/**/*.ts"
+	]
+}

webpack.config.js

@@ -0,0 +1,34 @@
+const path = require('path');
+
+module.exports = {
+	mode: 'production',
+	entry: './src/main.ts',
+	output: {
+		path: path.resolve(__dirname, 'out'),
+		filename: 'main.js',
+		libraryTarget: 'commonjs2'  // This is correct for VS Code extensions
+	},
+	resolve: {
+		extensions: ['.ts', '.js'],
+		fallback: {
+			// For Node.js core modules not available in browser context, you set them to false
+			"child_process": false, // Not used in browser context
+			"fs": false,            // Not used in browser context
+			"os": require.resolve("os-browserify/browser"), // Polyfill for os
+			"path": require.resolve("path-browserify"),     // Polyfill for path
+		},
+	},
+	externals: {
+		vscode: 'commonjs vscode'  // Externalize the vscode module
+	},
+	module: {
+		rules: [
+			{
+				test: /\.ts$/,
+				use: 'ts-loader',
+				exclude: /node_modules/
+			}
+		]
+	},
+	target: 'node'  // Specify the target environment as Node.js
+};