aleff-github · webbrain-one · Aug 5, 2026 · Sep 9, 2026
diff --git a/out/extension.js b/out/extension.js
diff --git a/out/extension.js.map b/out/extension.js.map
diff --git a/package.json b/package.json
@@ -111,6 +111,38 @@
         "language": "duckyscript",
         "path": "./snippets/snippets.json"
       }
+    ],
+    "configuration": {
+      "type": "object",
+      "title": "DuckyScript",
+      "properties": {
+        "duckyscript.customCommands": {
+          "type": "array",
+          "items": {
+            "type": "string"
+          },
+          "default": [],
+          "scope": "resource",
+          "markdownDescription": "Extra DuckyScript commands to highlight, for firmware that extends the language (e.g. [USB Army Knife](https://github.com/i-am-shodan/USBArmyKnife/wiki/Commands), Flipper Zero). Matching is case-insensitive and applies to the first word of a line."
+        }
+      }
+    },
+    "semanticTokenTypes": [
+      {
+        "id": "duckyscriptCustomCommand",
+        "superType": "function",
+        "description": "A user-defined DuckyScript command from duckyscript.customCommands."
+      }
+    ],
+    "semanticTokenScopes": [
+      {
+        "language": "duckyscript",
+        "scopes": {
+          "duckyscriptCustomCommand": [
+            "support.function.builtin.duckyscript"
+          ]
+        }
+      }
     ]
   },
   "scripts": {

diff --git a/src/extension.ts b/src/extension.ts
@@ -54,6 +54,71 @@ const updatedCommandGroups = commandGroups.map(group =>
 
 const ALL_COMMANDS = commandGroups.flat();
 
+const CUSTOM_COMMAND_TOKEN = 'duckyscriptCustomCommand';
+
+const semanticTokenLegend = new vscode.SemanticTokensLegend([CUSTOM_COMMAND_TOKEN], []);
+
+function escapeRegExp(text: string): string {
+    return text.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+
+class CustomCommandSemanticTokensProvider implements vscode.DocumentSemanticTokensProvider, vscode.Disposable {
+    private commandPattern: RegExp | undefined;
+    private readonly _emitter = new vscode.EventEmitter<void>();
+    private readonly _configListener: vscode.Disposable;
+    onDidChangeSemanticTokens = this._emitter.event;
+
+    constructor() {
+        this.updateCustomCommands();
+        this._configListener = vscode.workspace.onDidChangeConfiguration(e => {
+            if (e.affectsConfiguration('duckyscript.customCommands')) {
+                this.updateCustomCommands();
+                this._emitter.fire();
+            }
+        });
+    }
+
+    private updateCustomCommands() {
+        const config = vscode.workspace.getConfiguration('duckyscript');
+        const commands = config.get<string[]>('customCommands', [])
+            .filter(c => typeof c === 'string' && c.trim().length > 0)
+            .map(c => c.trim())
+            // Longest first, so PREFIX_LONG wins over PREFIX.
+            .sort((a, b) => b.length - a.length);
+
+        this.commandPattern = commands.length
+            ? new RegExp(`^\\s*(${commands.map(escapeRegExp).join('|')})(?![\\w-])`, 'i')
+            : undefined;
+    }
+
+    provideDocumentSemanticTokens(
+        document: vscode.TextDocument,
+        token: vscode.CancellationToken
+    ): vscode.ProviderResult<vscode.SemanticTokens> {
+        const builder = new vscode.SemanticTokensBuilder(semanticTokenLegend);
+        if (!this.commandPattern) {
+            return builder.build();
+        }
+
+        for (let i = 0; i < document.lineCount; i++) {
+            if (token.isCancellationRequested) {
+                return builder.build();
+            }
+            const match = this.commandPattern.exec(document.lineAt(i).text);
+            if (match) {
+                const startColumn = match[0].length - match[1].length;
+                builder.push(i, startColumn, match[1].length, 0, 0);
+            }
+        }
+        return builder.build();
+    }
+
+    dispose() {
+        this._configListener.dispose();
+        this._emitter.dispose();
+    }
+}
+
 // We implement a CompletionItemProvider for our language
 class MyLanguageCompletionItemProvider implements vscode.CompletionItemProvider {
     // This method is called when the user activates the suggestions (e.g., Ctrl+Space)
@@ -98,6 +163,17 @@ export function activate(context: vscode.ExtensionContext) {
     );
 
     context.subscriptions.push(providerDisposable);
+
+    // Custom commands syntax highlighting
+    const customCommandProvider = new CustomCommandSemanticTokensProvider();
+    context.subscriptions.push(
+        customCommandProvider,
+        vscode.languages.registerDocumentSemanticTokensProvider(
+            { language: 'duckyscript' },
+            customCommandProvider,
+            semanticTokenLegend
+        )
+    );
 }
 
 function deactivate() { }

diff --git a/themes/DarkDuckyScript-color-theme.json b/themes/DarkDuckyScript-color-theme.json
@@ -1,7 +1,7 @@
 {
     "name": "DuckyScript - Night Owl based",
     "type": "dark",
-    "semanticHighlighting": false,
+    "semanticHighlighting": true,
     "colors": {
       "contrastBorder": "#122d42",
       "focusBorder": "#122d42",

diff --git a/themes/LightDuckyScript-color-theme.json b/themes/LightDuckyScript-color-theme.json
@@ -1,7 +1,7 @@
 {
 	"name": "DuckyScript - Night Owl Light based",
 	"type": "light",
-	"semanticHighlighting": false,
+	"semanticHighlighting": true,
 	"colors": {
 		// Base
 		"foreground": "#403f53",