render submit result to the bottom of notebook

Sorosliu1029 · Sorosliu1029 · commit a848642a4e2e · 2025-06-09T13:11:29.000+08:00
diff --git a/jupyterlab_leetcode/handlers/leetcode_handler.py b/jupyterlab_leetcode/handlers/leetcode_handler.py
@@ -482,9 +482,6 @@ async def check(self, cnt=0):
             return
 
         self.check_result = json.loads(resp.body)
-        print(
-            f"Checking submission {self.submission_id}, attempts: {cnt}, result: {self.check_result}"
-        )
         self.write_message(
             {
                 "type": "submissionResult",
diff --git a/src/components/LeetCode.tsx b/src/components/LeetCode.tsx
@@ -43,6 +43,7 @@ const LeetCode: React.FC<{
   }, []);
 
   const openNoteBook = (path: string) => {
+    // TODO: notebook.NotebookActions.runCells of pre-code cell
     docManager.openOrReveal(path);
   };
 
diff --git a/src/components/LeetCodeNotebookHeader.tsx b/src/components/LeetCodeNotebookHeader.tsx
@@ -1,5 +1,6 @@
 import React, { useEffect, useState } from 'react';
-import { NotebookPanel } from '@jupyterlab/notebook';
+import { NotebookPanel, NotebookActions } from '@jupyterlab/notebook';
+import { ICellModel } from '@jupyterlab/cells';
 import { submitNotebook } from '../services/notebook';
 import { makeWebSocket } from '../services/handler';
 import {
@@ -48,6 +49,61 @@ const LeetCodeNotebookHeader: React.FC<{ notebook: NotebookPanel }> = ({
     return ws;
   };
 
+  const getResultCell = () => {
+    let resultCellModel: ICellModel | null = null;
+    const cells = notebook.content.model?.cells ?? [];
+    for (const cell of cells) {
+      if (cell.metadata['id'] === 'result') {
+        resultCellModel = cell;
+      }
+    }
+    if (!resultCellModel) {
+      const activeCellIdx = cells.length ? cells.length - 1 : 0;
+      notebook.content.activeCellIndex = activeCellIdx;
+      NotebookActions.insertBelow(notebook.content);
+      const activeCell = notebook.content.activeCell;
+      if (activeCell) {
+        resultCellModel = activeCell.model;
+        resultCellModel.setMetadata('id', 'result');
+      }
+    }
+    return resultCellModel;
+  };
+
+  const populateResultCell = (
+    cellModel: ICellModel,
+    result: Extract<LeetCodeSubmissionResult, { state: 'SUCCESS' }>
+  ) => {
+    let source = '';
+    switch (result.status_msg) {
+      case 'Accepted': {
+        source = `# Accepted\n\nRuntime: ${result.display_runtime}\nMemory: ${result.status_memory}\n\nOutput:\n\`\`\`\n${result.code_output}\n\`\`\``;
+        break;
+      }
+      case 'Wrong Answer': {
+        source = `# Wrong Answer\n\nExpected: ${result.expected_output}\nGot: ${result.code_output}\n\nInput:\n\`\`\`\n${result.input_formatted}\n\`\`\``;
+        break;
+      }
+      case 'Time Limit Exceeded': {
+        source = `# Time Limit Exceeded\n\nRuntime: ${result.display_runtime}\nMemory: ${result.status_memory}\n\nInput:\n\`\`\`\n${result.input_formatted}\n\`\`\``;
+        break;
+      }
+      case 'Memory Limit Exceeded': {
+        source = `# Memory Limit Exceeded\n\nRuntime: ${result.display_runtime}\nMemory: ${result.status_memory}\n\nInput:\n\`\`\`\n${result.input_formatted}\n\`\`\``;
+        break;
+      }
+      case 'Runtime Error': {
+        source = `# Runtime Error\n\nRuntime: ${result.display_runtime}\nMemory: ${result.status_memory}\n\nOutput:\n\`\`\`\n${result.code_output}\n\`\`\`\n\nInput:\n\`\`\`\n${result.input_formatted}\n\`\`\``;
+        break;
+      }
+      case 'Internal Error': {
+        source = `# Internal Error\n\nRuntime: ${result.display_runtime}\nMemory: ${result.status_memory}\n\nOutput:\n\`\`\`\n${result.code_output}\n\`\`\`\n\nInput:\n\`\`\`\n${result.input_formatted}\n\`\`\``;
+        break;
+      }
+    }
+    cellModel.sharedModel.setSource(source);
+  };
+
   useEffect(() => {
     if (!submissionId) {
       return;
@@ -70,6 +126,18 @@ const LeetCodeNotebookHeader: React.FC<{ notebook: NotebookPanel }> = ({
     }
   }, [ws, ws?.readyState]);
 
+  useEffect(() => {
+    if (result?.state !== 'SUCCESS') {
+      return;
+    }
+    const resultCellModel = getResultCell();
+    if (resultCellModel) {
+      populateResultCell(resultCellModel, result);
+      NotebookActions.changeCellType(notebook.content, 'markdown');
+      NotebookActions.run(notebook.content);
+    }
+  }, [result?.state]);
+
   return (
     <div>
       <button onClick={submit}>Submit</button>
diff --git a/src/types/leetcode.d.ts b/src/types/leetcode.d.ts
@@ -108,5 +108,5 @@ export type LeetCodeSubmissionResult =
       memory_percentile: number | null;
       input_formatted: string;
       input: string;
-      status_msg: string;
+      status_msg: 'Accepted' | 'Wrong Answer' | string;
     };

Original file line number	Diff line number	Diff line change
`@@ -482,9 +482,6 @@ async def check(self, cnt=0):`
`482`	`482`	`return`
`483`	`483`
`484`	`484`	`self.check_result = json.loads(resp.body)`
`485`		`- print(`
`486`		`- f"Checking submission {self.submission_id}, attempts: {cnt}, result: {self.check_result}"`
`487`		`- )`
`488`	`485`	`self.write_message(`
`489`	`486`	`{`
`490`	`487`	`"type": "submissionResult",`