From f644370f5d8424ca0bf07dd6b582a8b0552b8252 Mon Sep 17 00:00:00 2001
From: Duc Trung Le <leductrungxf@gmail.com>
Date: Mon, 25 Nov 2024 20:34:13 +0100
Subject: [PATCH] Create cell widgets

---
 .../base/src/suggestionsPanel/cellWidget.ts   | 47 +++++++++++++++++++
 packages/base/src/suggestionsPanel/widget.ts  | 16 +++----
 2 files changed, 55 insertions(+), 8 deletions(-)
 create mode 100644 packages/base/src/suggestionsPanel/cellWidget.ts

diff --git a/packages/base/src/suggestionsPanel/cellWidget.ts b/packages/base/src/suggestionsPanel/cellWidget.ts
new file mode 100644
index 0000000..ac35ea9
--- /dev/null
+++ b/packages/base/src/suggestionsPanel/cellWidget.ts
@@ -0,0 +1,47 @@
+import { Cell, CodeCell, CodeCellModel } from '@jupyterlab/cells';
+import {
+  CodeMirrorEditorFactory,
+  EditorLanguageRegistry
+} from '@jupyterlab/codemirror';
+import { ICell } from '@jupyterlab/nbformat';
+import {
+  RenderMimeRegistry,
+  standardRendererFactories as initialFactories
+} from '@jupyterlab/rendermime';
+import { Panel } from '@lumino/widgets';
+import { suggestionCellStyle } from './style';
+
+export class CellWidget extends Panel {
+  constructor(options: CellWidget.IOptions) {
+    super(options);
+    this.addClass(suggestionCellStyle);
+    const rendermime = new RenderMimeRegistry({ initialFactories });
+    const languages = new EditorLanguageRegistry();
+    EditorLanguageRegistry.getDefaultLanguages()
+      .filter(language =>
+        ['ipython', 'julia', 'python'].includes(language.name.toLowerCase())
+      )
+      .forEach(language => {
+        languages.addLanguage(language);
+      });
+    const factoryService = new CodeMirrorEditorFactory({
+      languages
+    });
+    const model = new CodeCellModel();
+    model.sharedModel.setSource(options.cellModel.source as string);
+    const cellWidget = new CodeCell({
+      contentFactory: new Cell.ContentFactory({
+        editorFactory: factoryService.newInlineEditor.bind(factoryService)
+      }),
+      rendermime,
+      model
+    });
+    this.addWidget(cellWidget);
+  }
+}
+
+export namespace CellWidget {
+  export interface IOptions extends Panel.IOptions {
+    cellModel: ICell;
+  }
+}
diff --git a/packages/base/src/suggestionsPanel/widget.ts b/packages/base/src/suggestionsPanel/widget.ts
index 99014cc..55f43e5 100644
--- a/packages/base/src/suggestionsPanel/widget.ts
+++ b/packages/base/src/suggestionsPanel/widget.ts
@@ -1,7 +1,9 @@
 import { PanelWithToolbar } from '@jupyterlab/ui-components';
+import { Panel } from '@lumino/widgets';
+
 import { ISuggestionChange, ISuggestionsModel } from '../types';
-import { Panel, Widget } from '@lumino/widgets';
-import { suggestionCellStyle, suggestionsWidgetAreaStyle } from './style';
+import { CellWidget } from './cellWidget';
+import { suggestionsWidgetAreaStyle } from './style';
 
 export class SuggestionsWidget extends PanelWithToolbar {
   constructor(options: SuggestionsWidget.IOptions) {
@@ -30,10 +32,9 @@ export class SuggestionsWidget extends PanelWithToolbar {
       case 'added': {
         const suggestion = this._model.getSuggestion({ cellId, suggestionId });
         if (suggestion) {
-          const w = new Widget();
-          w.addClass(suggestionCellStyle);
+          const w = new CellWidget({ cellModel: suggestion.content });
           w.id = suggestionId;
-          w.node.innerHTML = `<pre>${JSON.stringify(suggestion, null, 2)}</pre>`;
+          // w.node.innerHTML = `<pre>${JSON.stringify(suggestion, null, 2)}</pre>`;
           this._suggestionsArea.addWidget(w);
         }
         break;
@@ -56,10 +57,9 @@ export class SuggestionsWidget extends PanelWithToolbar {
     if (allSuggestions) {
       for (const [_, val] of allSuggestions.entries()) {
         Object.entries(val).forEach(([suggestionId, suggestionDef]) => {
-          const w = new Widget();
-          w.addClass(suggestionCellStyle);
+          const w = new CellWidget({ cellModel: suggestionDef.content });
           w.id = suggestionId;
-          w.node.innerHTML = `<pre>${JSON.stringify(suggestionDef, null, 2)}</pre>`;
+          // w.node.innerHTML = `<pre>${JSON.stringify(suggestionDef, null, 2)}</pre>`;
           this._suggestionsArea.addWidget(w);
         });
       }