@marimo-team/frontend 0.23.12-dev19 → 0.23.12-dev20

package/dist/index.html

@@ -66,7 +66,7 @@
     <marimo-server-token data-token="{{ server_token }}" hidden></marimo-server-token>
     <!-- /TODO -->
     <title>{{ title }}</title>
-    <script type="module" crossorigin src="./assets/index-BUZYcPA6.js"></script>
+    <script type="module" crossorigin src="./assets/index-D5xY00XH.js"></script>
     <link rel="modulepreload" crossorigin href="./assets/preload-helper-y72bE5iF.js">
     <link rel="modulepreload" crossorigin href="./assets/chunk-LvLJmgfZ.js">
     <link rel="modulepreload" crossorigin href="./assets/react-Bj1aDYRI.js">
@@ -133,7 +133,7 @@
     <link rel="modulepreload" crossorigin href="./assets/debounce-OiXznToA.js">
     <link rel="modulepreload" crossorigin href="./assets/database-zap-DIlSqcdp.js">
     <link rel="modulepreload" crossorigin href="./assets/main-DmhaVp6Y.js">
-    <link rel="modulepreload" crossorigin href="./assets/cells-BidbgSuB.js">
+    <link rel="modulepreload" crossorigin href="./assets/cells-06TDmJ8V.js">
     <link rel="modulepreload" crossorigin href="./assets/ErrorBoundary-Bo313Z9X.js">
     <link rel="modulepreload" crossorigin href="./assets/kbd-BCXa3Ocm.js">
     <link rel="modulepreload" crossorigin href="./assets/useInstallPackage-DEHK-C20.js">
@@ -147,38 +147,38 @@
     <link rel="modulepreload" crossorigin href="./assets/usePress-mjvipKvz.js">
     <link rel="modulepreload" crossorigin href="./assets/input-b2KTOk0X.js">
     <link rel="modulepreload" crossorigin href="./assets/ImperativeModal-MQnbMIO2.js">
-    <link rel="modulepreload" crossorigin href="./assets/cell-link-BWqpbaiv.js">
+    <link rel="modulepreload" crossorigin href="./assets/cell-link-DgWTiCtG.js">
     <link rel="modulepreload" crossorigin href="./assets/multi-map-9ogswH46.js">
     <link rel="modulepreload" crossorigin href="./assets/alert-Bjh7v4lb.js">
     <link rel="modulepreload" crossorigin href="./assets/chevron-right-C7RH0Nc7.js">
     <link rel="modulepreload" crossorigin href="./assets/dropdown-menu-DfGLyCbP.js">
     <link rel="modulepreload" crossorigin href="./assets/links-DK7te6sJ.js">
-    <link rel="modulepreload" crossorigin href="./assets/useRunCells-DFXV3Rm5.js">
+    <link rel="modulepreload" crossorigin href="./assets/useRunCells-DNaOKKqW.js">
     <link rel="modulepreload" crossorigin href="./assets/copy-CyasABTK.js">
     <link rel="modulepreload" crossorigin href="./assets/copy-IBNGDypD.js">
     <link rel="modulepreload" crossorigin href="./assets/copy-icon-BCIGtBo0.js">
-    <link rel="modulepreload" crossorigin href="./assets/RenderHTML-DXJCvdz4.js">
-    <link rel="modulepreload" crossorigin href="./assets/datasource-DI-YUWev.js">
-    <link rel="modulepreload" crossorigin href="./assets/state-ClDZE-YW.js">
+    <link rel="modulepreload" crossorigin href="./assets/RenderHTML-Cq4mwWYz.js">
+    <link rel="modulepreload" crossorigin href="./assets/datasource-mVfTJz3V.js">
+    <link rel="modulepreload" crossorigin href="./assets/state-B3ypjIx-.js">
     <link rel="modulepreload" crossorigin href="./assets/package-BoMcg6sy.js">
     <link rel="modulepreload" crossorigin href="./assets/sparkles-DctOjjEz.js">
-    <link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-G8Q8ZYoi.js">
+    <link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-BQt_FXJL.js">
     <link rel="modulepreload" crossorigin href="./assets/eye-off-DI0EQXw0.js">
     <link rel="modulepreload" crossorigin href="./assets/refresh-cw-DcEasnos.js">
-    <link rel="modulepreload" crossorigin href="./assets/tree-actions-CesD4Rcl.js">
+    <link rel="modulepreload" crossorigin href="./assets/tree-actions-lVyJAYy7.js">
     <link rel="modulepreload" crossorigin href="./assets/spinner-DTBVbtl-.js">
-    <link rel="modulepreload" crossorigin href="./assets/html-to-image-TCtIotr9.js">
-    <link rel="modulepreload" crossorigin href="./assets/focus-CocfMydm.js">
+    <link rel="modulepreload" crossorigin href="./assets/html-to-image-CbQeOjEJ.js">
+    <link rel="modulepreload" crossorigin href="./assets/focus-DqhVePQT.js">
     <link rel="modulepreload" crossorigin href="./assets/useAsyncData-CT1LpOBD.js">
     <link rel="modulepreload" crossorigin href="./assets/LazyAnyLanguageCodeMirror-CUQODgGJ.js">
     <link rel="modulepreload" crossorigin href="./assets/micromark-factory-space-P--XWZhg.js">
     <link rel="modulepreload" crossorigin href="./assets/chunk-5FQGJX7Z-C6VN0dQj.js">
-    <link rel="modulepreload" crossorigin href="./assets/markdown-renderer-CGZTB2Dx.js">
+    <link rel="modulepreload" crossorigin href="./assets/markdown-renderer-C1yypOYF.js">
     <link rel="modulepreload" crossorigin href="./assets/command-CsbsBLBw.js">
     <link rel="modulepreload" crossorigin href="./assets/field-CA9HCE2Y.js">
     <link rel="modulepreload" crossorigin href="./assets/popover-DMNO6Poc.js">
     <link rel="modulepreload" crossorigin href="./assets/errors-48KFPvCp.js">
-    <link rel="modulepreload" crossorigin href="./assets/download-DgEarSps.js">
+    <link rel="modulepreload" crossorigin href="./assets/download-VxqV5OSK.js">
     <link rel="modulepreload" crossorigin href="./assets/table-Fk_msRcD.js">
     <link rel="modulepreload" crossorigin href="./assets/useIframeCapabilities-DuGyzPVb.js">
     <link rel="modulepreload" crossorigin href="./assets/error-banner-C4DZksWV.js">
@@ -198,21 +198,21 @@
     <link rel="modulepreload" crossorigin href="./assets/plus-CExKUszm.js">
     <link rel="modulepreload" crossorigin href="./assets/trash-2-CjPoUZCP.js">
     <link rel="modulepreload" crossorigin href="./assets/react-resizable-panels.browser.esm-CgWOEYeG.js">
-    <link rel="modulepreload" crossorigin href="./assets/JsonOutput-BOGU8eqb.js">
+    <link rel="modulepreload" crossorigin href="./assets/JsonOutput-B1X71OMV.js">
     <link rel="modulepreload" crossorigin href="./assets/chart-no-axes-column-DVEV2j1H.js">
     <link rel="modulepreload" crossorigin href="./assets/square-function-CPbVtjKW.js">
     <link rel="modulepreload" crossorigin href="./assets/spec-BzxoPflZ.js">
-    <link rel="modulepreload" crossorigin href="./assets/column-preview-rFcgfgNL.js">
+    <link rel="modulepreload" crossorigin href="./assets/column-preview-CchF5iQz.js">
     <link rel="modulepreload" crossorigin href="./assets/icons-D3YFXNmD.js">
-    <link rel="modulepreload" crossorigin href="./assets/radio-group-Czn6pjhm.js">
-    <link rel="modulepreload" crossorigin href="./assets/floating-outline-BDx00oB1.js">
+    <link rel="modulepreload" crossorigin href="./assets/radio-group-BaOG2bSA.js">
+    <link rel="modulepreload" crossorigin href="./assets/floating-outline-3UBj62S1.js">
     <link rel="modulepreload" crossorigin href="./assets/objectWithoutPropertiesLoose-Msf1n32r.js">
     <link rel="modulepreload" crossorigin href="./assets/esm-DSv-aO_T.js">
-    <link rel="modulepreload" crossorigin href="./assets/readonly-python-code-DtUWOxnZ.js">
+    <link rel="modulepreload" crossorigin href="./assets/readonly-python-code-CNyK8G1T.js">
     <link rel="modulepreload" crossorigin href="./assets/file-headphone-CUN2DC78.js">
     <link rel="modulepreload" crossorigin href="./assets/file-BvoBtS_V.js">
     <link rel="modulepreload" crossorigin href="./assets/image-jhXeQEgs.js">
-    <link rel="modulepreload" crossorigin href="./assets/file-icons-BwFX01F6.js">
+    <link rel="modulepreload" crossorigin href="./assets/file-icons-BeJWUbBH.js">
     <link rel="modulepreload" crossorigin href="./assets/switch-DHhZkGEv.js">
     <link rel="modulepreload" crossorigin href="./assets/events-BRbNg7Li.js">
     <link rel="modulepreload" crossorigin href="./assets/globals-BDWM3c_z.js">
@@ -221,11 +221,11 @@
     <link rel="modulepreload" crossorigin href="./assets/memoize-Tp7rARFe.js">
     <link rel="modulepreload" crossorigin href="./assets/get-C-qh_et5.js">
     <link rel="modulepreload" crossorigin href="./assets/_baseSet-CsW0tABf.js">
-    <link rel="modulepreload" crossorigin href="./assets/state-BZgmxb7a.js">
+    <link rel="modulepreload" crossorigin href="./assets/state-DD6m9X2h.js">
     <link rel="modulepreload" crossorigin href="./assets/label-CFvCMYuj.js">
-    <link rel="modulepreload" crossorigin href="./assets/textarea-Dz09niIv.js">
+    <link rel="modulepreload" crossorigin href="./assets/textarea-BM7hAwR3.js">
     <link rel="modulepreload" crossorigin href="./assets/refresh-ccw-CUkO1Vlo.js">
-    <link rel="modulepreload" crossorigin href="./assets/form-CgvTS8au.js">
+    <link rel="modulepreload" crossorigin href="./assets/form-BAB18E37.js">
     <link rel="modulepreload" crossorigin href="./assets/renderShortcut-CKbqJyj4.js">
     <link rel="modulepreload" crossorigin href="./assets/useBoolean-48bgGoSa.js">
     <link rel="modulepreload" crossorigin href="./assets/useDeepCompareMemoize-zUHU--0D.js">

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/frontend",
-  "version": "0.23.12-dev19",
+  "version": "0.23.12-dev20",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/core/codemirror/go-to-definition/__tests__/commands.test.ts

@@ -253,6 +253,73 @@ a = 10`;
     `);
   });
 
+  test("from-import alias is the binding, not the imported name", async () => {
+    const code = `\
+from math import sin as my_sin
+print(my_sin)`;
+    view = createEditor(code);
+    const usagePosition = code.lastIndexOf("my_sin");
+    const result = goToVariableDefinition(view, "my_sin", usagePosition);
+
+    expect(result).toBe(true);
+    await tick();
+    // The alias `my_sin` (after `as`) is the real binding.
+    expect(renderEditorView(view)).toMatchInlineSnapshot(`
+      "
+      from math import sin as my_sin
+                              ^
+      print(my_sin)
+      "
+    `);
+  });
+
+  test("module path in from-import is not a local definition", async () => {
+    const code = `\
+from math import sin
+print(math)`;
+    view = createEditor(code);
+    const usagePosition = code.lastIndexOf("math");
+    // `math` is a module reference in the from-clause, not a binding in this
+    // cell, so the scoped resolver should return false and let the caller fall
+    // through to cross-cell resolution.
+    const result = goToVariableDefinition(view, "math", usagePosition);
+
+    expect(result).toBe(false);
+    expect(view.state.selection.main.head).toBe(0);
+  });
+
+  test("imported name without `as` is a local definition", async () => {
+    const code = `\
+from math import sin
+print(sin)`;
+    view = createEditor(code);
+    const usagePosition = code.lastIndexOf("sin");
+    const result = goToVariableDefinition(view, "sin", usagePosition);
+
+    expect(result).toBe(true);
+    await tick();
+    expect(renderEditorView(view)).toMatchInlineSnapshot(`
+      "
+      from math import sin
+                       ^
+      print(sin)
+      "
+    `);
+  });
+
+  test("imported name shadowed by `as` is not a binding", async () => {
+    const code = `\
+from math import sin as my_sin
+print(sin)`;
+    view = createEditor(code);
+    const usagePosition = code.lastIndexOf("sin");
+    // `sin` here refers to nothing in this cell (it was renamed to `my_sin`),
+    // so the scoped resolver should return false.
+    const result = goToVariableDefinition(view, "sin", usagePosition);
+
+    expect(result).toBe(false);
+  });
+
   test("selects outer-scope function declaration", async () => {
     view = createEditor(`\
 def x():

package/src/core/codemirror/go-to-definition/__tests__/utils.test.ts

@@ -133,4 +133,51 @@ output = _x + 10`;
     await tick();
     expect(view.state.selection.main.head).toBe(code.indexOf("_x = 10"));
   });
+
+  test("falls through to cross-cell when in-cell occurrence is only a module path in a from-import", async () => {
+    // Regression: ImportStatement used to register every VariableName child
+    // (the module path and pre-`as` names) as in-cell declarations, so the
+    // local-first short-circuit would steal F12 from cross-cell resolution.
+    const moduleCell = cellId("module-cell");
+    const usageCell = cellId("usage-cell");
+    const moduleCode = `mymodule = 100`;
+    const usageCode = `\
+from mymodule import something
+print(mymodule)`;
+
+    const moduleView = createEditor(moduleCode, moduleCode.length);
+    const usageView = createEditor(
+      usageCode,
+      usageCode.lastIndexOf("mymodule"),
+    );
+    views.push(moduleView, usageView);
+
+    const notebook = initialNotebookState();
+    notebook.cellHandles[moduleCell] = {
+      current: { editorView: moduleView, editorViewOrNull: moduleView },
+    } as never;
+    notebook.cellHandles[usageCell] = {
+      current: { editorView: usageView, editorViewOrNull: usageView },
+    } as never;
+
+    store.set(notebookAtom, notebook);
+    store.set(variablesAtom, {
+      [variableName("mymodule")]: {
+        dataType: "int",
+        declaredBy: [moduleCell],
+        name: variableName("mymodule"),
+        usedBy: [usageCell],
+        value: "100",
+      },
+    });
+
+    const result = goToDefinitionAtCursorPosition(usageView);
+
+    expect(result).toBe(true);
+    await tick();
+    // Cross-cell jump: moduleView's cursor should land on `mymodule = 100`.
+    expect(moduleView.state.selection.main.head).toBe(
+      moduleCode.indexOf("mymodule"),
+    );
+  });
 });

package/src/core/codemirror/go-to-definition/commands.ts

@@ -307,33 +307,52 @@ function collectMatchingDeclarations(
       break;
     }
     case "ImportStatement": {
+      // The grammar emits one ImportStatement for both `import x [as y]` and
+      // `from m import x [as y], ...`. Direct children include the keywords
+      // (`from`/`import`/`as`), commas, dots, and every VariableName from the
+      // module path AND the import list. We only want the names that actually
+      // bind in the current scope: the post-`as` alias if present, otherwise
+      // the imported name itself. Names before `import` (the from-path) and
+      // the original name when an alias follows it are NOT bindings.
       const subCursor = node.cursor();
       subCursor.firstChild();
-      do {
-        if (
-          subCursor.name === "VariableName" &&
-          state.doc.sliceString(subCursor.from, subCursor.to) === variableName
-        ) {
-          addDeclaration(declarations, currentScope, subCursor.from);
+      let pastImport = false;
+      // Buffer the most recent post-`import` VariableName so we can defer
+      // committing it until we know whether `as` follows.
+      let pending: { from: number; matches: boolean } | null = null;
+      const commit = () => {
+        if (pending?.matches) {
+          addDeclaration(declarations, currentScope, pending.from);
         }
-      } while (subCursor.nextSibling());
-      break;
-    }
-    case "ImportFromStatement": {
-      const subCursor = node.cursor();
-      subCursor.firstChild();
-      let foundImport = false;
+        pending = null;
+      };
       do {
         if (subCursor.name === "import") {
-          foundImport = true;
-        } else if (
-          foundImport &&
-          subCursor.name === "VariableName" &&
-          state.doc.sliceString(subCursor.from, subCursor.to) === variableName
-        ) {
-          addDeclaration(declarations, currentScope, subCursor.from);
+          pastImport = true;
+          continue;
+        }
+        if (!pastImport) {
+          continue;
+        }
+        if (subCursor.name === "as") {
+          // Next VariableName is the alias and replaces `pending`.
+          pending = null;
+          continue;
+        }
+        if (subCursor.name === "VariableName") {
+          // Flush any previous pending name (no `as` followed it).
+          commit();
+          pending = {
+            from: subCursor.from,
+            matches:
+              state.doc.sliceString(subCursor.from, subCursor.to) ===
+              variableName,
+          };
+        } else if (subCursor.name === ",") {
+          commit();
         }
       } while (subCursor.nextSibling());
+      commit();
       break;
     }
     case "TryStatement":
@@ -410,23 +429,21 @@ function findScopedDefinitionPosition(
  * @param view The editor view which contains the variable name.
  * @param variableName The name of the variable to select, if found in the editor.
  * @param usagePosition The position of the variable usage, if available.
- * @param fallbackToFirstMatch Whether to fall back to the first matching
- * variable name when no scoped definition is found. Defaults to true.
  */
 export function goToVariableDefinition(
   view: EditorView,
   variableName: string,
   usagePosition?: number,
-  fallbackToFirstMatch = true,
 ): boolean {
   const { state } = view;
-  let from: number | null = null;
-  if (usagePosition !== undefined) {
-    from = findScopedDefinitionPosition(state, variableName, usagePosition);
-  }
-  if (from === null && fallbackToFirstMatch) {
-    from = findFirstMatchingVariable(state, variableName);
-  }
+  // When the caller knows the usage position, trust the scoped lookup. Falling
+  // back to first-match would defeat the local-vs-cross-cell decision in
+  // goToDefinition: if the symbol only appears as a module path in an import,
+  // scoped resolution returns null and we want the caller to try other cells.
+  const from =
+    usagePosition !== undefined
+      ? findScopedDefinitionPosition(state, variableName, usagePosition)
+      : findFirstMatchingVariable(state, variableName);
 
   if (from === null) {
     return false;

package/src/core/codemirror/go-to-definition/utils.ts

@@ -82,7 +82,6 @@ export function goToDefinition(
       view,
       variableName,
       usagePosition,
-      false,
     );
     if (foundLocally) {
       return true;

package/src/core/codemirror/reactive-references/__tests__/analyzer.test.ts

@@ -640,6 +640,60 @@ def run(polars):
     `);
   });
 
+  test("set comprehension target shadows outer global", () => {
+    // Regression: SCOPE_CREATING_NODES used "SetComprehension" instead of the
+    // grammar's "SetComprehensionExpression", so set comprehensions never
+    // created a scope and their for-target was treated as reactive.
+    expect(runHighlight(["x"], "result = {x for x in range(5)}"))
+      .toMatchInlineSnapshot(`
+      "
+      result = {x for x in range(5)}
+      "
+    `);
+  });
+
+  test("from-import module path stays reactive", () => {
+    // Regression: ImportStatement collected every VariableName child, so the
+    // module name in `from m import y` was wrongly treated as a local binding.
+    expect(
+      runHighlight(
+        ["math"],
+        `
+def f():
+    from math import sin as my_sin
+    return math + my_sin(1)`,
+      ),
+    ).toMatchInlineSnapshot(`
+      "
+      def f():
+          from math import sin as my_sin
+          return math + my_sin(1)
+                 ^^^^
+      "
+    `);
+  });
+
+  test("from-import: aliased imported name is not a binding", () => {
+    // Regression: `sin` in `from math import sin as my_sin` was incorrectly
+    // registered as a local binding, hiding genuine reactive uses of `sin`.
+    expect(
+      runHighlight(
+        ["sin"],
+        `
+def f():
+    from math import sin as my_sin
+    return sin + my_sin(1)`,
+      ),
+    ).toMatchInlineSnapshot(`
+      "
+      def f():
+          from math import sin as my_sin
+          return sin + my_sin(1)
+                 ^^^
+      "
+    `);
+  });
+
   test("lambda inside function with outer global", () => {
     expect(
       runHighlight(

package/src/core/codemirror/reactive-references/analyzer.ts

@@ -16,7 +16,7 @@ const SCOPE_CREATING_NODES = new Set([
   "FunctionDefinition",
   "LambdaExpression",
   "ArrayComprehensionExpression",
-  "SetComprehension",
+  "SetComprehensionExpression",
   "DictionaryComprehensionExpression",
   "ComprehensionExpression",
   "ClassDefinition",
@@ -152,7 +152,7 @@ export function findReactiveVariables(options: {
       }
       case "ArrayComprehensionExpression":
       case "DictionaryComprehensionExpression":
-      case "SetComprehension":
+      case "SetComprehensionExpression":
       case "ComprehensionExpression": {
         // Domprehension variables - look for VariableName or TupleExpression after 'for'
         const subCursor = node.cursor();
@@ -276,49 +276,52 @@ export function findReactiveVariables(options: {
         break;
       }
       case "ImportStatement": {
-        // Handle import x
+        // The grammar emits a single ImportStatement for both `import x [as y]`
+        // and `from m import x [as y], ...`. Direct children mix keywords,
+        // module-path names (before `import`), imported names, and aliases.
+        // Only post-`import` names that aren't shadowed by a following `as`
+        // (and the alias itself when `as` is present) bind in the current
+        // scope.
         const subCursor = node.cursor();
         subCursor.firstChild();
-        do {
-          if (subCursor.name === "VariableName") {
-            const varName = options.state.doc.sliceString(
-              subCursor.from,
-              subCursor.to,
-            );
-
-            const currentScope =
-              currentScopeStack[currentScopeStack.length - 1] ?? -1;
-            if (!allDeclarations.has(currentScope)) {
-              allDeclarations.set(currentScope, new Set());
-            }
-            allDeclarations.get(currentScope)?.add(varName);
+        const currentScope =
+          currentScopeStack[currentScopeStack.length - 1] ?? -1;
+        if (!allDeclarations.has(currentScope)) {
+          allDeclarations.set(currentScope, new Set());
+        }
+        const scope = allDeclarations.get(currentScope);
+        let pastImport = false;
+        let pending: string | null = null;
+        const commit = () => {
+          if (pending !== null) {
+            scope?.add(pending);
           }
-        } while (subCursor.nextSibling());
-
-        break;
-      }
-      case "ImportFromStatement": {
-        // Handle from x import y as z
-        const subCursor = node.cursor();
-        subCursor.firstChild();
-        let foundImport = false;
+          pending = null;
+        };
         do {
           if (subCursor.name === "import") {
-            foundImport = true;
-          } else if (foundImport && subCursor.name === "VariableName") {
-            const varName = options.state.doc.sliceString(
+            pastImport = true;
+            continue;
+          }
+          if (!pastImport) {
+            continue;
+          }
+          if (subCursor.name === "as") {
+            // Drop the imported name; the next VariableName is the alias.
+            pending = null;
+            continue;
+          }
+          if (subCursor.name === "VariableName") {
+            commit();
+            pending = options.state.doc.sliceString(
               subCursor.from,
               subCursor.to,
             );
-            // Add to the current innermost scope
-            const currentScope =
-              currentScopeStack[currentScopeStack.length - 1] ?? -1;
-            if (!allDeclarations.has(currentScope)) {
-              allDeclarations.set(currentScope, new Set());
-            }
-            allDeclarations.get(currentScope)?.add(varName);
+          } else if (subCursor.name === ",") {
+            commit();
           }
         } while (subCursor.nextSibling());
+        commit();
 
         break;
       }
@@ -424,6 +427,12 @@ export function findReactiveVariables(options: {
     const nodeName = cursor.name;
     const nodeStart = cursor.from;
 
+    // Names inside an import statement are module paths, imported names, or
+    // aliases — none of them are reactive *uses* of an outer-cell global.
+    if (nodeName === "ImportStatement") {
+      return;
+    }
+
     const isNewScope = SCOPE_CREATING_NODES.has(nodeName);
 
     let currentScopeStack = scopeStack;