expose context for cell language

quarto-dev · Nov 21, 2024 · 530aa48 · 530aa48
1 parent 69fce35
commit 530aa48
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/apps/vscode/src/vdoc/vdoc.ts b/apps/vscode/src/vdoc/vdoc.ts
@@ -13,7 +13,7 @@
  *
  */
 
-import { Position, TextDocument, Uri, Range } from "vscode";
+import { Position, TextDocument, Uri, Range, commands } from "vscode";
 import { Token, isExecutableLanguageBlock, languageBlockAtPosition, languageNameFromBlock } from "quarto-core";
 
 import { isQuartoDoc } from "../core/doc";
@@ -157,8 +157,12 @@ export async function virtualDocUri(
 export function languageAtPosition(tokens: Token[], position: Position) {
   const block = languageBlockAtPosition(tokens, position);
   if (block) {
-    return languageFromBlock(block);
+    const language = languageFromBlock(block);
+    // expose cell language for use in keybindings, etc
+    commands.executeCommand('setContext', 'quarto.cellLang', language?.extension);
+    return language;
   } else {
+    commands.executeCommand('setContext', 'quarto.cellLang', undefined);
     return undefined;
   }
 }