Get language used in kernel

Author: jahn96

src/CodeSnippetWidget.tsx

@@ -31,6 +31,7 @@ import { CodeSnippetDisplay } from './CodeSnippetDisplay';
 import { CodeSnippetInputDialog } from './CodeSnippetInputDialog';
 
 import React from 'react';
+import { Notebook } from '@jupyterlab/notebook';
 
 /**
  * A class used to indicate a snippet item.
@@ -245,6 +246,13 @@ export class CodeSnippetWidget extends ReactWidget {
    * Handle the `'lm-drop'` event for the widget.
    */
   private async _evtDrop(event: IDragEvent): Promise<void> {
+    // TODO: get language from kernel
+    const notebook: Notebook = event.mimeData.getData('internal:cells')[0]
+      .parent;
+
+    const language = notebook.model.defaultKernelLanguage;
+    console.log(language);
+
     const data = this.findCellData(event.mimeData);
     if (data === undefined) {
       return;
@@ -300,7 +308,8 @@ export class CodeSnippetWidget extends ReactWidget {
     } else {
       // Handle the case where we are copying cells
       event.dropAction = 'copy';
-      CodeSnippetInputDialog(this, data, idx);
+
+      CodeSnippetInputDialog(this, data, language, idx);
     }
 
     // Reorder snippet just to make sure id's are in order.

src/index.ts

@@ -40,6 +40,11 @@ import {
   ICodeSnippetEditorMetadata,
 } from './CodeSnippetEditor';
 import { CodeSnippetService } from './CodeSnippetService';
+import { NotebookPanel } from '@jupyterlab/notebook';
+import { DocumentWidget } from '@jupyterlab/docregistry';
+// import { NotebookPanel } from '@jupyterlab/notebook';
+// import { ServerConnection, SettingManager } from '@jupyterlab/services';
+// import { URLExt } from '@jupyterlab/coreutils';
 
 const CODE_SNIPPET_EXTENSION_ID = 'code-snippet-extension';
 
@@ -178,6 +183,16 @@ function activateCodeSnippet(
     isToggled: () => toggled,
     iconClass: 'some-css-icon-class',
     execute: () => {
+      let language = '';
+      // get the language of document or notebook
+      if (app.shell.currentWidget instanceof NotebookPanel) {
+        language = (app.shell.currentWidget as NotebookPanel).sessionContext
+          .kernelPreference.language;
+      } else if (app.shell.currentWidget instanceof DocumentWidget) {
+        language = (app.shell.currentWidget as DocumentWidget).context.model
+          .defaultKernelLanguage;
+      }
+
       const highlightedCode = getSelectedText();
       if (highlightedCode === '') {
         //if user just right-clicks cell(s) to save
@@ -202,12 +217,14 @@ function activateCodeSnippet(
         CodeSnippetInputDialog(
           codeSnippetWidget,
           resultArray,
+          language,
           codeSnippetWidget.codeSnippetManager.snippets.length
         );
       } else {
         CodeSnippetInputDialog(
           codeSnippetWidget,
           highlightedCode.split('\n'),
+          language
           codeSnippetWidget.codeSnippetManager.snippets.length
         );
       }