Configure CodeMirror to treat dataviewjs as javascript (#2329)

* Configure CodeMirror to treat dataviewjs as javascript and revert on unload * Run prettier * Handle custom dataviewjs block name
blacksmithgu · May 26, 2024 · 61291ae · 61291ae
1 parent b0c36b7
commit 61291ae
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -161,6 +161,21 @@ export default class DataviewPlugin extends Plugin {
                 });
             })
         );
+
+        this.registerDataviewjsCodeHighlighting();
+        this.register(() => this.unregisterDataviewjsCodeHighlighting());
+    }
+
+    public registerDataviewjsCodeHighlighting(): void {
+        window.CodeMirror.defineMode(this.settings.dataviewJsKeyword, config =>
+            window.CodeMirror.getMode(config, "javascript")
+        );
+    }
+
+    public unregisterDataviewjsCodeHighlighting(): void {
+        window.CodeMirror.defineMode(this.settings.dataviewJsKeyword, config =>
+            window.CodeMirror.getMode(config, "null")
+        );
     }
 
     private debouncedRefresh: () => void = () => null;
@@ -378,7 +393,9 @@ class GeneralSettingsTab extends PluginSettingTab {
                     .setValue(this.plugin.settings.dataviewJsKeyword)
                     .onChange(async value => {
                         if (value.length == 0) return;
+                        this.plugin.unregisterDataviewjsCodeHighlighting();
                         await this.plugin.updateSettings({ dataviewJsKeyword: value });
+                        this.plugin.registerDataviewjsCodeHighlighting();
                     })
             );