Addressed review comments and file formatting

Author: diningPhilosopher64

src/jupyter_matlab_kernel/kernels/base_kernel.py

@@ -27,7 +27,7 @@
 )
 from jupyter_matlab_kernel.mwi_exceptions import MATLABConnectionError
 
-from jupyter_matlab_kernel.kernels.labextension_comm import LabExtensionCommunication
+from jupyter_matlab_kernel.kernels.comms import LabExtensionCommunication
 
 
 _MATLAB_STARTUP_TIMEOUT = mwi_settings.get_process_startup_timeout()
@@ -145,6 +145,9 @@ def __init__(self, *args, **kwargs):
 
         self.labext_comm = LabExtensionCommunication(self)
 
+        # Custom handling of comm messages for jupyterlab extension communication.
+        # https://jupyter-client.readthedocs.io/en/latest/messaging.html#custom-messages
+
         # Override only comm handlers to keep implementation clean by separating
         # JupyterLab extension communication logic from core kernel functionality.
         # Other handlers (interrupt_request, execute_request, etc.) remain in base class.

src/jupyter_matlab_kernel/kernels/comms/__init__.py

@@ -0,0 +1,3 @@
+# Copyright 2025 The MathWorks, Inc.
+
+from .labextension import LabExtensionCommunication

…rnels/labextension_comm/communication.py …lab_kernel/kernels/comms/labextension.pysrc/jupyter_matlab_kernel/kernels/labextension_comm/communication.py renamed to src/jupyter_matlab_kernel/kernels/comms/labextension.py src/jupyter_matlab_kernel/kernels/labextension_comm/communication.py renamed to src/jupyter_matlab_kernel/kernels/comms/labextension.py

@@ -1,7 +1,6 @@
 # Copyright 2025 The MathWorks, Inc.
 
 from ipykernel.comm import Comm
-import time
 
 
 class LabExtensionCommunication:

src/jupyter_matlab_kernel/kernels/labextension_comm/__init__.py

@@ -36,6 +36,7 @@
     "import/no-webpack-loader-syntax": "warn",
     "node/no-callback-literal": ["off", "warn"],
     "node/handle-callback-err": ["off", "warn"],
-    "node/no-deprecated-api": ["off", "warn"]
+    "node/no-deprecated-api": ["off", "warn"],
+    "no-multiple-empty-lines": "warn"
   }
 }

src/jupyter_matlab_labextension/.eslintrc.json

@@ -25,10 +25,10 @@ export function lineIndent (
         currentLineText.match(/^(?:\s*)(case)\b$/) &&
     previousLineText.match(/^(?:\s*)(switch)\b/)
     ) {
-    // First case in a switch statement.
+        // First case in a switch statement.
         return prevLineIndent + indentUnit;
     } else if (currentLineText.match(/^(?:\s*)(end)\b$/)) {
-    // Treat "end" separately to avoid mistakenly correcting the end of a switch statement.
+        // Treat "end" separately to avoid mistakenly correcting the end of a switch statement.
         const currentLeadingWhitespace = currentLineText.match(
             leadingWhitespacePattern
         );
@@ -44,7 +44,7 @@ export function lineIndent (
             return 0;
         }
     } else {
-    // Other cases
+        // Other cases
         let lineIndent = prevLineIndent;
         const indentMatch = previousLineText.match(indentPattern);
         if (indentMatch !== null) {

src/jupyter_matlab_labextension/src/codemirror-lang-matlab/indent-matlab.ts

@@ -163,22 +163,20 @@ implements
 }
 
 // A unique token for the comm service
-export const CommunicationService = new Token<MatlabCommunicationExtension>(
-    '@mathworks/MatlabCommPlugin'
-);
+export const IMatlabCommunication = new Token<ICommunicationService>('@mathworks/matlab-comm:IMatlabCommunication');
 
 export const matlabCommPlugin: JupyterFrontEndPlugin<MatlabCommunicationExtension> =
   {
       id: '@mathworks/matlabCommPlugin',
       autoStart: true,
       requires: [INotebookTracker],
-      provides: CommunicationService,
+      provides: IMatlabCommunication,
       activate: (app: JupyterFrontEnd): MatlabCommunicationExtension => {
           const matlabCommExtension = new MatlabCommunicationExtension();
           app.docRegistry.addWidgetExtension('Notebook', matlabCommExtension);
 
           // Dispose resources created by this plugin when the page unloads.
-          // Need to add this seperately if the jupyterlab tab is closed directly.
+          // Need to handle this separately for the case when jupyterlab tab is closed directly
           window.addEventListener('beforeunload', () => {
               matlabCommExtension.deleteComms();
           });

src/jupyter_matlab_labextension/src/plugins/matlabCommunication.ts

@@ -27,8 +27,8 @@ function registerMFiles (
 ) {
     const { commands } = app;
     const createNewMatlabFile = async (args: ReadonlyPartialJSONObject) => {
-    /** Get the directory in which the MATLAB file must be created;
-     * otherwise take the current filebrowser directory. */
+        /** Get the directory in which the MATLAB file must be created;
+         * otherwise take the current filebrowser directory. */
         const cwd = args.cwd;
 
         /** Create a new untitled MATLAB file. */

src/jupyter_matlab_labextension/src/plugins/matlabFiles.ts

@@ -33,7 +33,7 @@ implements DocumentRegistry.IWidgetExtension<NotebookPanel, INotebookModel> {
         panel: NotebookPanel,
         context: DocumentRegistry.IContext<INotebookModel>
     ): IDisposable {
-    /**  Create the toolbar button to open MATLAB in a browser. */
+        /**  Create the toolbar button to open MATLAB in a browser. */
         const matlabToolbarButton = new ToolbarButton({
             className: 'openMATLABButton',
             icon: matlabIcon,

src/jupyter_matlab_labextension/src/plugins/matlabToolbarButton.ts

@@ -70,9 +70,9 @@ export class NotebookInfo {
                             resolve();
                         }
                     };
-          this._panel!.sessionContext.session?.kernel?.statusChanged.connect(
-              onStatusChanged
-          );
+                    this._panel!.sessionContext.session?.kernel?.statusChanged.connect(
+                        onStatusChanged
+                    );
                 }
             });
         }