jupyterlab-ipyflow 0.0.151 → 0.0.153

package/lib/index.js

@@ -2,6 +2,31 @@ import { ICommandPalette } from '@jupyterlab/apputils';
 import { CodeCell } from '@jupyterlab/cells';
 import { INotebookTracker } from '@jupyterlab/notebook';
 const IPYFLOW_KERNEL_NAME = 'ipyflow';
+const waitingClass = 'waiting-cell';
+const readyClass = 'ready-cell';
+const readyMakingClass = 'ready-making-cell';
+const readyMakingInputClass = 'ready-making-input-cell';
+const linkedWaitingClass = 'linked-waiting';
+const linkedReadyMakerClass = 'linked-ready-maker';
+// ipyflow frontend state
+let dirtyCells = new Set();
+let waitingCells = new Set();
+let readyCells = new Set();
+let waiterLinks = {};
+let readyMakerLinks = {};
+let activeCell = null;
+let activeCellId = null;
+let cellsById = {};
+let cellModelsById = {};
+let orderIdxById = {};
+let cellPendingExecution = null;
+let lastExecutionMode = null;
+let isReactivelyExecuting = false;
+let lastExecutionHighlights = null;
+let executedReactiveReadyCells = new Set();
+let newReadyCells = new Set();
+let forcedReactiveCells = new Set();
+const cleanup = new Event('cleanup');
 /**
  * Initialization data for the jupyterlab-ipyflow extension.
  */
@@ -79,35 +104,6 @@ const extension = {
         });
     }
 };
-const waitingClass = 'waiting-cell';
-const readyClass = 'ready-cell';
-const readyMakingClass = 'ready-making-cell';
-const readyMakingInputClass = 'ready-making-input-cell';
-const linkedWaitingClass = 'linked-waiting';
-const linkedReadyMakerClass = 'linked-ready-maker';
-let dirtyCells = new Set();
-let waitingCells = new Set();
-let readyCells = new Set();
-let waiterLinks = {};
-let readyMakerLinks = {};
-let activeCell = null;
-let activeCellId = null;
-let cellsById = {};
-let cellModelsById = {};
-let orderIdxById = {};
-let cellPendingExecution = null;
-let lastExecutionMode = null;
-let isReactivelyExecuting = false;
-let lastExecutionHighlights = null;
-let executedReactiveReadyCells = new Set();
-let newReadyCells = new Set();
-let forcedReactiveCells = new Set();
-const cleanup = new Event('cleanup');
-const resetReactiveState = () => {
-    newReadyCells = new Set();
-    forcedReactiveCells = new Set();
-    executedReactiveReadyCells = new Set();
-};
 const getJpInputCollapser = (elem) => {
     if (elem === null || elem === undefined) {
         return null;
@@ -353,27 +349,36 @@ const connectToComm = (session, notebook) => {
         else if (msg.content.data['type'] === 'compute_exec_schedule') {
             waitingCells = new Set(msg.content.data['waiting_cells']);
             readyCells = new Set(msg.content.data['ready_cells']);
-            newReadyCells = new Set([...newReadyCells, ...msg.content.data['new_ready_cells']]);
-            forcedReactiveCells = new Set([...forcedReactiveCells, ...msg.content.data['forced_reactive_cells']]);
+            newReadyCells = new Set([
+                ...newReadyCells,
+                ...msg.content.data['new_ready_cells'],
+            ]);
+            forcedReactiveCells = new Set([
+                ...forcedReactiveCells,
+                ...msg.content.data['forced_reactive_cells'],
+            ]);
             waiterLinks = msg.content.data['waiter_links'];
             readyMakerLinks = msg.content.data['ready_maker_links'];
             cellPendingExecution = null;
             const exec_mode = msg.content.data['exec_mode'];
+            isReactivelyExecuting = isReactivelyExecuting || (exec_mode === 'reactive');
             const flow_order = msg.content.data['flow_order'];
             const exec_schedule = msg.content.data['exec_schedule'];
             lastExecutionMode = exec_mode;
             lastExecutionHighlights = msg.content.data['highlights'];
-            executedReactiveReadyCells.add(msg.content.data['last_executed_cell_id']);
+            const lastExecutedCellId = msg.content.data['last_executed_cell_id'];
+            executedReactiveReadyCells.add(lastExecutedCellId);
             const last_execution_was_error = msg.content.data['last_execution_was_error'];
             if (!last_execution_was_error) {
                 for (const cell of notebook.widgets) {
                     if (cell.model.type !== 'code' || executedReactiveReadyCells.has(cell.model.id)) {
                         continue;
                     }
-                    if (!forcedReactiveCells.has(cell.model.id)) {
-                        if (exec_mode !== 'reactive' || !newReadyCells.has(cell.model.id)) {
-                            continue;
-                        }
+                    if (!newReadyCells.has(cell.model.id)) {
+                        continue;
+                    }
+                    if (!forcedReactiveCells.has(cell.model.id) && exec_mode !== 'reactive') {
+                        continue;
                     }
                     const codeCell = cell;
                     if (cellPendingExecution === null) {
@@ -397,14 +402,13 @@ const connectToComm = (session, notebook) => {
                     if (lastExecutionHighlights === 'reactive') {
                         readyCells = executedReactiveReadyCells;
                     }
-                    resetReactiveState();
                     comm.send({
                         type: 'reactivity_cleanup',
                     });
                 }
-                else {
-                    executedReactiveReadyCells = new Set();
-                }
+                forcedReactiveCells = new Set();
+                newReadyCells = new Set();
+                executedReactiveReadyCells = new Set();
                 updateUI(notebook);
                 isReactivelyExecuting = false;
             }

package/package.json

@@ -63,5 +63,5 @@
     "extension": true,
     "outputDir": "../../core/ipyflow/resources/labextension/"
   },
-  "version": "0.0.151"
+  "version": "0.0.153"
 }