Add initial save magic command

Author: malcolm-sparkfun

kernel/src/kernel.ts

@@ -3,6 +3,10 @@ import { KernelMessage } from '@jupyterlab/services';
 import { ServiceContainer } from './services/ServiceContainer';
 
 const reconnectString: string = "%connect%"
+const saveString: string = "%save%";
+const bootFileName: string = "boot.py";
+const bootSavePrefix: string = `with open(${bootFileName}, 'w') as f:\n    f.write('''\n`;
+const bootSaveSuffix: string = "\n''')\n";
 
 export class EmbeddedKernel extends BaseKernel {
 
@@ -76,6 +80,56 @@ export class EmbeddedKernel extends BaseKernel {
       await this.serviceContainer.deviceService.disconnect();
       await this.serviceContainer.deviceService.connect(this.outputResponse.bind(this));
     }
+    if (code.includes(saveString)) {
+      // Burn the code after the save string into the boot.py file
+      this.outputResponse("Save command detected, saving code to boot.py...");
+      const codeToSave = code.split(saveString)[1].trim();
+      if (!codeToSave) {
+        return {
+          status: 'error',
+          execution_count: this.executionCount,
+          ename: 'ValueError',
+          evalue: 'No code provided to save',
+          traceback: ['Please provide code to save after the %save% command']
+        };
+      }
+      // Burn the code by adding it to the boot.py file. Can call executeRequest with the code to save 
+      // but append the code to write it to the boot.py file at the start of the code.
+      const bootCode = `# This code is automatically generated by the Embedded Kernel\n${codeToSave}`;
+      const bootCodeWithSave = `${bootSavePrefix}${bootCode}${bootSaveSuffix}`;
+
+      // Execute the command to save the boot.py file
+      if (!this.serviceContainer.deviceService.transport) {
+        console.log("[Kernel] executeRequest - No transport available for saving boot.py");
+        return {
+          status: 'error',
+          execution_count: this.executionCount,
+          ename: 'TransportError',
+          evalue: 'No transport available to save boot.py',
+          traceback: ['Please connect a device first']
+        };
+      }
+      const result = await this.serviceContainer.consoleService.executeCommand(bootCodeWithSave, (content) => {
+        console.log("[Kernel] executeRequest - Streaming output:", content);
+        this.stream(content);
+      });
+      if (!result.success) {
+        console.log("[Kernel] executeRequest - Command execution failed:", result.error);
+        return {
+          status: 'error',
+          execution_count: this.executionCount,
+          ename: 'ExecutionError',
+          evalue: result.error || 'Unknown error',
+          traceback: [result.error || 'Unknown error']
+        };
+      }
+      console.log("[Kernel] executeRequest - Code saved to boot.py successfully");
+      return {
+        status: 'ok',
+        execution_count: this.executionCount,
+        user_expressions: {},
+      };
+    }
 
     try {
       console.log("[Kernel] executeRequest - Checking transport");
@@ -97,7 +151,7 @@ export class EmbeddedKernel extends BaseKernel {
       console.log("[Kernel] executeRequest - Executing command via ConsoleService");
       // Execute the command and handle the output
       const result = await this.serviceContainer.consoleService.executeCommand(code, (content) => {
-        console.log("[Kernel] executeRequest - Streaming output:", content.text.substring(0, 50) + (content.text.length > 50 ? '...' : ''));
+        console.log("[Kernel] executeRequest - Streaming output:", content);
         this.stream(content);
       });