@marimo-team/islands 0.25.3-dev273 → 0.25.3-dev279

package/dist/{useAsyncData-BYIAoMqn.js → useAsyncData-Ci4AvJff.js}

@@ -1,8 +1,8 @@
 import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
 import { t } from "./react-sHtZdjeP.js";
 import { t as n } from "./compiler-runtime-CldRdSsv.js";
-import { D as r } from "./utils-EBklsC5R.js";
-import { t as i } from "./invariant-DLmrl-hA.js";
+import { t as r } from "./invariant-DLmrl-hA.js";
+import { t as i } from "./useEvent-BRH-aCqi.js";
 //#region src/hooks/useAsyncData.ts
 var a = n(), o = /* @__PURE__ */ e(t(), 1), s = {
 	error(e, t) {
@@ -45,7 +45,7 @@ var a = n(), o = /* @__PURE__ */ e(t(), 1), s = {
 function c(e, t) {
 	let n = (0, a.c)(16), [c, d] = (0, o.useState)(0), [f, p] = (0, o.useState)(u), m;
 	n[0] === e ? m = n[1] : (m = typeof e == "function" ? { fetch: e } : e, n[0] = e, n[1] = m);
-	let h = r(m.fetch), g;
+	let h = i(m.fetch), g;
 	n[2] === h ? g = n[3] : (g = () => {
 		let e = new AbortController(), t = !1;
 		return p(l), h({ previous: () => {
@@ -67,7 +67,7 @@ function c(e, t) {
 	let v;
 	n[8] === f ? v = n[9] : (v = (e) => {
 		let t;
-		typeof e == "function" ? (i(f.status === "success" || f.status === "loading", "No previous state value."), t = e(f.data)) : t = e, p(s.success(t));
+		typeof e == "function" ? (r(f.status === "success" || f.status === "loading", "No previous state value."), t = e(f.data)) : t = e, p(s.success(t));
 	}, n[8] = f, n[9] = v);
 	let y;
 	n[10] === c ? y = n[11] : (y = () => d(c + 1), n[10] = c, n[11] = y);

package/dist/{useDateFormatter-ButWkdcK.js → useDateFormatter-CPlava52.js}

@@ -1,7 +1,7 @@
 import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
 import { t } from "./react-sHtZdjeP.js";
-import { h as n } from "./tooltip-BfcDQd2j.js";
-import { C as r } from "./select-OxgozOC8.js";
+import { h as n } from "./tooltip-DMMiOn2f.js";
+import { C as r } from "./select-DsoP1qe8.js";
 var i = n("chart-pie", [["path", {
 	d: "M21 12c.552 0 1.005-.449.95-.998a10 10 0 0 0-8.953-8.951c-.55-.055-.998.398-.998.95v8a1 1 0 0 0 1 1z",
 	key: "pzmjnu"

package/dist/{useDeepCompareMemoize-Bz0c5GN6.js → useDeepCompareMemoize-NaaWoFlX.js}

@@ -1,6 +1,6 @@
 import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
 import { t } from "./react-sHtZdjeP.js";
-import { l as n } from "./utils-EBklsC5R.js";
+import { Mt as n } from "./use-restore-focus-CosX7-gx.js";
 //#region src/hooks/useDeepCompareMemoize.ts
 var r = /* @__PURE__ */ e(t(), 1);
 function i(e) {

package/dist/useEvent-BRH-aCqi.js

@@ -0,0 +1,19 @@
+import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
+import { t } from "./react-sHtZdjeP.js";
+//#region ../node_modules/.pnpm/react-use-event-hook@0.9.6_react@19.3.0/node_modules/react-use-event-hook/dist/esm/useEvent.js
+var n = /* @__PURE__ */ e(t()), r = typeof window < "u" ? n.useInsertionEffect || n.useLayoutEffect : () => {};
+function i(e) {
+	let t = n.useRef(a);
+	r(() => {
+		t.current = e;
+	}, [e]);
+	let i = n.useRef(null);
+	return i.current ||= function() {
+		return t.current.apply(this, arguments);
+	}, i.current;
+}
+function a() {
+	throw Error("INVALID_USEEVENT_INVOCATION: the callback from useEvent cannot be invoked before the component has mounted.");
+}
+//#endregion
+export { i as t };

package/dist/{useIframeCapabilities-BIolGsF1.js → useIframeCapabilities-BXhDDoCG.js}

@@ -1,5 +1,5 @@
 import { t as e } from "./compiler-runtime-CldRdSsv.js";
-import { t } from "./capabilities-B6yy_CBN.js";
+import { t } from "./capabilities-D9FskwsK.js";
 //#region src/hooks/useIframeCapabilities.ts
 var n = e();
 function r() {

package/dist/{useLifecycle-JjxCC7Vu.js → useLifecycle-SniivYmC.js}

@@ -1,8 +1,8 @@
 import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
-import { h as t, s as n } from "./button-0-RqY1DW.js";
+import { h as t, r as n } from "./button-DZuNnIZO.js";
 import { t as r } from "./react-sHtZdjeP.js";
 import { t as i } from "./compiler-runtime-CldRdSsv.js";
-import { h as a } from "./tooltip-BfcDQd2j.js";
+import { h as a } from "./tooltip-DMMiOn2f.js";
 import { t as o } from "./jsx-runtime-3BD1kT0s.js";
 var s = a("calendar", [
 	["path", {

package/dist/{vega-component-rxIxAAh9.js → vega-component-DIxaDQNs.js}

@@ -1,23 +1,23 @@
 import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
-import { a as t, g as n, h as r, m as i } from "./button-0-RqY1DW.js";
+import { g as t, h as n, m as r, s as i } from "./button-DZuNnIZO.js";
 import { t as a } from "./react-sHtZdjeP.js";
 import { t as o } from "./compiler-runtime-CldRdSsv.js";
-import { t as s } from "./tooltip-BfcDQd2j.js";
-import { l as c, t as l } from "./config-BsPruWMR.js";
+import { t as s } from "./tooltip-DMMiOn2f.js";
+import { l as c, t as l } from "./config-C8MFHQ2E.js";
 import { t as u } from "./jsx-runtime-3BD1kT0s.js";
-import { n as d } from "./error-banner-DPc9eIw5.js";
-import { t as f } from "./debounce-DrhhziJv.js";
-import { n as p } from "./useTheme-DPLobSFs.js";
-import { D as ee } from "./utils-EBklsC5R.js";
-import { t as te } from "./objects-DCmMvZtn.js";
-import { a as ne, i as re, n as ie, t as m } from "./formats-DbSvAPY-.js";
+import { u as d } from "./use-restore-focus-CosX7-gx.js";
+import { n as f } from "./error-banner-CHCdHVFB.js";
+import { t as p } from "./debounce-DrhhziJv.js";
+import { t as ee } from "./objects-DCmMvZtn.js";
+import { t as te } from "./useEvent-BRH-aCqi.js";
+import { a as ne, i as re, n as ie, t as m } from "./formats-PzBQMUnr.js";
 import { a as h, t as g } from "./arrays-DoQ-AILx.js";
-import { t as ae } from "./useAsyncData-BYIAoMqn.js";
+import { t as ae } from "./useAsyncData-Ci4AvJff.js";
 import { t as _ } from "./semaphore-89jsoHtS.js";
 import { D as oe, n as se } from "./vega-loader.browser-B9Pz0EGP.js";
 import { t as ce } from "./react-vega-C4T9Rb_w.js";
-import { a as le, n as v } from "./loader-BVEtQ4F3.js";
-import { t as ue } from "./useDeepCompareMemoize-Bz0c5GN6.js";
+import { a as le, n as v } from "./loader-DZvtzQoD.js";
+import { t as ue } from "./useDeepCompareMemoize-NaaWoFlX.js";
 import { t as de } from "./use-vega-container-remeasure-Bz4PZMtt.js";
 //#region src/plugins/impl/vega/fix-relative-url.ts
 var fe = /* @__PURE__ */ e(a(), 1), pe = o();
@@ -103,7 +103,7 @@ var y = {
 	}
 };
 function C() {
-	return t() ? "metaKey" : "ctrlKey";
+	return i() ? "metaKey" : "ctrlKey";
 }
 var w = {
 	highlight() {
@@ -515,10 +515,10 @@ se("arrow", m);
 var K = (e) => {
 	let t = (0, pe.c)(12), { value: n, setValue: r, chartSelection: i, fieldSelection: a, spec: o, embedOptions: s } = e, c, l;
 	t[0] === o ? (c = t[1], l = t[2]) : (c = async () => W(o), l = [o], t[0] = o, t[1] = c, t[2] = l);
-	let { data: u, error: f } = ae(c, l);
-	if (f) {
+	let { data: u, error: d } = ae(c, l);
+	if (d) {
 		let e;
-		return t[3] === f ? e = t[4] : (e = /* @__PURE__ */ (0, G.jsx)(d, { error: f }), t[3] = f, t[4] = e), e;
+		return t[3] === d ? e = t[4] : (e = /* @__PURE__ */ (0, G.jsx)(f, { error: d }), t[3] = d, t[4] = e), e;
 	}
 	if (!u) return null;
 	let p;
@@ -531,7 +531,7 @@ var K = (e) => {
 		embedOptions: s
 	}), t[5] = i, t[6] = s, t[7] = a, t[8] = u, t[9] = r, t[10] = n, t[11] = p) : p = t[11], p;
 }, q = (e) => {
-	let a = (0, pe.c)(50), { value: o, setValue: l, chartSelection: u, fieldSelection: d, spec: ne, embedOptions: m } = e, { theme: h } = p(), g = (0, fe.useRef)(null), ae = (0, fe.useRef)(void 0), [_, se] = (0, fe.useState)(), v;
+	let a = (0, pe.c)(50), { value: o, setValue: l, chartSelection: u, fieldSelection: f, spec: ne, embedOptions: m } = e, { theme: h } = d(), g = (0, fe.useRef)(null), ae = (0, fe.useRef)(void 0), [_, se] = (0, fe.useState)(), v;
 	bb0: {
 		if (m && "actions" in m) {
 			v = m.actions;
@@ -544,10 +544,10 @@ var K = (e) => {
 		}, a[0] = e) : e = a[0], v = e;
 	}
 	let y = v, b = ue(ne), x;
-	a[1] !== u || a[2] !== d || a[3] !== b ? (x = he(me(b), {
+	a[1] !== u || a[2] !== f || a[3] !== b ? (x = he(me(b), {
 		chartSelection: u,
-		fieldSelection: d
-	}), a[1] = u, a[2] = d, a[3] = b, a[4] = x) : x = a[4];
+		fieldSelection: f
+	}), a[1] = u, a[2] = f, a[3] = b, a[4] = x) : x = a[4];
 	let C = x, w;
 	a[5] === C ? w = a[6] : (w = E(C), a[5] = C, a[6] = w);
 	let T = w, D;
@@ -557,11 +557,11 @@ var K = (e) => {
 			...e
 		});
 	}, a[7] = l, a[8] = o, a[9] = D) : D = a[9];
-	let O = ee(D), k;
-	a[10] === O ? k = a[11] : (k = f((e, t) => {
-		n.debug("[Vega signal]", e, t);
-		let r = te.mapValues(t, _e);
-		r = te.mapValues(r, ge), O({ [e]: r });
+	let O = te(D), k;
+	a[10] === O ? k = a[11] : (k = p((e, n) => {
+		t.debug("[Vega signal]", e, n);
+		let r = ee.mapValues(n, _e);
+		r = ee.mapValues(r, ge), O({ [e]: r });
 	}, 100), a[10] = O, a[11] = k);
 	let A = k, j = ue(T), M;
 	if (a[12] !== A || a[13] !== j) {
@@ -573,18 +573,18 @@ var K = (e) => {
 	} else M = a[14];
 	let N = M, P;
 	a[17] === C ? P = a[18] : (P = (e) => {
-		n.error(e), n.debug(C), se(e);
+		t.error(e), t.debug(C), se(e);
 	}, a[17] = C, a[18] = P);
-	let F = ee(P), I;
+	let F = te(P), I;
 	a[19] === Symbol.for("react.memo_cache_sentinel") ? (I = (e) => {
-		n.debug("[Vega view] created", e), ae.current = e, se(void 0);
+		t.debug("[Vega view] created", e), ae.current = e, se(void 0);
 	}, a[19] = I) : I = a[19];
-	let L = ee(I), R;
+	let L = te(I), R;
 	a[20] === T ? R = a[21] : (R = () => {
 		let e = [];
 		if (S.hasPoint(T) && e.push(["Point selection", "click to select a point; hold shift for multi-select"]), S.hasInterval(T) && e.push(["Interval selection", "click and drag to select an interval"]), S.hasLegend(T) && e.push(["Legend selection", "click to select a legend item; hold shift for multi-select"]), S.hasPanZoom(T)) {
-			let n = t() ? "⌘" : "ctrl";
-			e.push(["Pan", `hold the ${n} key and drag`], ["Zoom", `hold the ${n} key and scroll`]);
+			let t = i() ? "⌘" : "ctrl";
+			e.push(["Pan", `hold the ${t} key and drag`], ["Zoom", `hold the ${t} key and scroll`]);
 		}
 		return e.length === 0 ? null : /* @__PURE__ */ (0, G.jsx)(s, {
 			delayDuration: 300,
@@ -618,21 +618,21 @@ var K = (e) => {
 	de(g, K === "container");
 	let q, J;
 	a[32] !== U || a[33] !== N ? (q = () => (N.forEach((e) => {
-		let { signalName: t, handler: r } = e;
+		let { signalName: n, handler: r } = e;
 		try {
-			U?.view.addSignalListener(t, r);
+			U?.view.addSignalListener(n, r);
 		} catch (e) {
-			let t = e;
-			n.error(t);
+			let n = e;
+			t.error(n);
 		}
 	}), () => {
 		N.forEach((e) => {
-			let { signalName: t, handler: r } = e;
+			let { signalName: n, handler: r } = e;
 			try {
-				U?.view.removeSignalListener(t, r);
+				U?.view.removeSignalListener(n, r);
 			} catch (e) {
-				let t = e;
-				n.error(t);
+				let n = e;
+				t.error(n);
 			}
 		});
 	}), J = [U, N], a[32] = U, a[33] = N, a[34] = q, a[35] = J) : (q = a[34], J = a[35]), (0, fe.useEffect)(q, J);
@@ -645,7 +645,7 @@ var K = (e) => {
 		})]
 	}), a[36] = _, a[37] = Y);
 	let X, ye;
-	a[38] === Symbol.for("react.memo_cache_sentinel") ? (X = r("relative"), ye = i.stopPropagation(), a[38] = X, a[39] = ye) : (X = a[38], ye = a[39]);
+	a[38] === Symbol.for("react.memo_cache_sentinel") ? (X = n("relative"), ye = r.stopPropagation(), a[38] = X, a[39] = ye) : (X = a[38], ye = a[39]);
 	let Z;
 	a[40] === K ? Z = a[41] : (Z = /* @__PURE__ */ (0, G.jsx)("div", {
 		ref: g,

package/dist/{vennDiagram-4TSXK5OY-Db8GtH1G.js → vennDiagram-4TSXK5OY-C-hQ4iNJ.js}

@@ -3,7 +3,7 @@ import { t } from "./src-CA_m4ONl.js";
 import { n } from "./chunk-Y2CYZVJY-DKKS1VyV.js";
 import { $ as r, H as i, K as a, U as o, a as s, b as c, c as l, et as u, f as d, nt as f, tt as p, v as m, w as h, y as g } from "./chunk-DU6HZSFF-BwiyaNpg.js";
 import { t as _ } from "./rough.esm-CtV9AnV1.js";
-import { p as v } from "./mermaid-rd09e1JL.js";
+import { p as v } from "./mermaid-A36unKXJ.js";
 //#region ../node_modules/.pnpm/khroma@2.1.0/node_modules/khroma/dist/methods/transparentize.js
 var y = (e, t) => p(e, "a", -t), b = 1e-10;
 function x(e, t) {

package/dist/{wardleyDiagram-VM6X3IG4-DmIiC3FU.js → wardleyDiagram-VM6X3IG4-C2MQ_Umv.js}

@@ -2,7 +2,7 @@ import { i as e } from "./chunk-75Z2AOVW-D7gxIs2Q.js";
 import { n as t } from "./src-CA_m4ONl.js";
 import { n } from "./chunk-Y2CYZVJY-DKKS1VyV.js";
 import { D as r, H as i, K as a, U as o, a as s, b as c, c as l, v as u, w as d, x as f, y as p } from "./chunk-DU6HZSFF-BwiyaNpg.js";
-import { p as m } from "./mermaid-rd09e1JL.js";
+import { p as m } from "./mermaid-A36unKXJ.js";
 import { n as h } from "./mermaid-parser.core-DxuSfYwJ.js";
 import { t as g } from "./chunk-JWPE2WC7-DJvv1t8U.js";
 //#region ../node_modules/.pnpm/mermaid@11.17.2/node_modules/mermaid/dist/chunks/mermaid.core/wardleyDiagram-VM6X3IG4.mjs

package/dist/{xychartDiagram-S5SC5T6Z-DS9OPrNq.js → xychartDiagram-S5SC5T6Z-BtTTBn7Q.js}

@@ -9,7 +9,7 @@ import { n as s } from "./chunk-Y2CYZVJY-DKKS1VyV.js";
 import { D as c, H as l, K as u, U as d, a as f, b as p, c as m, f as h, v as g, w as _, y as v, z as y } from "./chunk-DU6HZSFF-BwiyaNpg.js";
 import "./chunk-PWAF6VOD-CQw5qATZ.js";
 import { t as b } from "./chunk-GMAD6QVW-BhaJx0X3.js";
-import { p as x } from "./mermaid-rd09e1JL.js";
+import { p as x } from "./mermaid-A36unKXJ.js";
 //#region ../node_modules/.pnpm/d3-scale@4.0.2/node_modules/d3-scale/src/band.js
 function S() {
 	var e = t().unknown(void 0), r = e.domain, i = e.range, o = 0, s = 1, c, l, u = !1, d = 0, f = 0, p = .5;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.25.3-dev273",
+  "version": "0.25.3-dev279",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/__mocks__/executions.ts

@@ -8,6 +8,7 @@ import type { CellExecutionInputs } from "@/core/executions";
 import {
   ConnectionState,
   type ConnectionStatus,
+  type Document,
   type Execution,
   type Executions,
   type NotebookReplica,
@@ -162,16 +163,16 @@ export const MockExecutions = {
     };
   },
 
-  /** Make the store's active replica report `records` and `runtime`. */
+  /** Make the store's active replica report `records`, `runtime`, and `document`. */
   seed(
     store: Store,
     records: Protocol.Execution[],
     runtime: Protocol.Runtime | null = null,
+    document: Document | null = null,
   ) {
     const current = store.get(activeNotebookReplicaAtom);
     const replica = {
       connectionAtom: atom<ConnectionStatus>({ state: ConnectionState.OPEN }),
-      documentAtom: atom(null),
       backendAtom: atom(null),
       attachmentsAtom: atom([]),
       connect: () => () => undefined,
@@ -185,6 +186,7 @@ export const MockExecutions = {
         operations: [],
         updated_at: null,
       }),
+      documentAtom: atom<Document | null>(document),
       runtimeAtom: atom(runtime),
       executionsAtom: atom(mirror(records)),
       resultsAtom: atom(results(records)),

package/src/__mocks__/notebook.ts

@@ -39,8 +39,6 @@ export const MockNotebook = {
           expand_output: false,
           ...data.config,
         },
-        edited: false,
-        lastCodeRun: null,
         serializedEditorState: null,
         ...data,
       })),

package/src/__tests__/CellStatus.test.tsx

@@ -3,6 +3,7 @@
 import { render } from "@testing-library/react";
 import { afterEach, beforeEach, describe, expect, test, vi } from "vitest";
 import { TooltipProvider } from "@/components/ui/tooltip";
+import type { CellFreshness } from "@/core/executions";
 import { formatElapsedTime, type Milliseconds } from "@/utils/time";
 import {
   CellStatusComponent,
@@ -70,9 +71,8 @@ describe("CellStatusComponent", () => {
   // Base props that will be modified for different test cases
   const baseProps = {
     editing: true,
-    edited: false,
     disabled: false,
-    staleInputs: false,
+    freshness: { sourceChanged: false, inputsChanged: false } as CellFreshness,
     status: "idle" as const,
     interrupted: false,
     elapsedTime: null,
@@ -94,7 +94,7 @@ describe("CellStatusComponent", () => {
     const props = {
       ...baseProps,
       disabled: true,
-      staleInputs: true,
+      freshness: { sourceChanged: false, inputsChanged: true } as CellFreshness,
       startedAt: 1_704_096_000_000 as Milliseconds, // Jan 1, 2024 12:00:00
     };
 
@@ -140,7 +140,7 @@ describe("CellStatusComponent", () => {
     const props = {
       ...baseProps,
       blocked: true,
-      staleInputs: true,
+      freshness: { sourceChanged: false, inputsChanged: true } as CellFreshness,
       startedAt: 1_704_096_000_000 as Milliseconds,
     };
 
@@ -215,7 +215,7 @@ describe("CellStatusComponent", () => {
   test("renders edited state", () => {
     const props = {
       ...baseProps,
-      edited: true,
+      freshness: { sourceChanged: true, inputsChanged: false } as CellFreshness,
       elapsedTime: 1500 as Milliseconds,
       startedAt: 1_704_096_000_000 as Milliseconds,
     };
@@ -231,7 +231,7 @@ describe("CellStatusComponent", () => {
   test("renders stale inputs state", () => {
     const props = {
       ...baseProps,
-      staleInputs: true,
+      freshness: { sourceChanged: false, inputsChanged: true } as CellFreshness,
       elapsedTime: 1500 as Milliseconds,
       startedAt: 1_704_096_000_000 as Milliseconds,
     };

package/src/components/editor/__tests__/data-attributes.test.tsx

@@ -93,7 +93,6 @@ describe("Cell data attributes", () => {
           [cid]: {
             code: "",
             name: cellName,
-            edited: false,
             config: {
               disabled: false,
               hide_code: false,
@@ -106,7 +105,6 @@ describe("Cell data attributes", () => {
       notebook.cellRuntime[cid] = createCellRuntimeState({
         output: null,
         consoleOutputs: [],
-        staleInputs: false,
         debuggerActive: false,
         outline: null,
       });

package/src/components/editor/__tests__/output-persistence.test.tsx

@@ -11,6 +11,11 @@ import { cellId } from "@/__tests__/branded";
 import { TooltipProvider } from "@/components/ui/tooltip";
 import { notebookAtom } from "@/core/cells/cells";
 import { DocumentProvider } from "@/core/notebook/document";
+import {
+  cell,
+  document as documentOf,
+  withCells,
+} from "@/core/notebook/document/__tests__/fixtures";
 import type { CellRuntimeState } from "@/core/cells/types";
 import { createCellRuntimeState } from "@/core/cells/types";
 import { defaultUserConfig } from "@/core/config/config-schema";
@@ -41,7 +46,6 @@ function seedNotebook(
       [cid]: {
         code: "1 + 1",
         name: "test_cell",
-        edited: false,
         config: {
           disabled: false,
           hide_code: false,
@@ -188,11 +192,22 @@ describe("output persistence across mode changes", () => {
 
   it("does not dim edited-but-not-run outputs as stale while presenting", () => {
     const { store, wrapper } = createTestWrapper();
+    // The host holds newer source than the cell last ran with
+    MockExecutions.seed(
+      store,
+      [MockExecutions.completed(cid, { code_fingerprint: "ran" })],
+      null,
+      withCells(
+        documentOf(),
+        {
+          c1: null,
+          c2: null,
+          [cid]: cell(cid, "1 + 1", { code_fingerprint: "edited" }),
+        },
+        [cid],
+      ),
+    );
     seedNotebook(store);
-    // Mark the cell as edited without re-running it
-    const notebook = store.get(notebookAtom);
-    notebook.cellData[cid] = { ...notebook.cellData[cid], edited: true };
-    store.set(notebookAtom, { ...notebook });
 
     const { container, rerender } = render(renderCell("edit"), { wrapper });
     expect(container.querySelector(".marimo-output-stale")).toBeTruthy();

package/src/components/editor/cell/CellStatus.tsx

@@ -7,7 +7,7 @@ import {
 } from "lucide-react";
 import { useDateFormatter } from "react-aria";
 import { MultiIcon } from "@/components/icons/multi-icon";
-import type { CellRunStatus } from "@/core/executions";
+import type { CellFreshness, CellRunStatus } from "@/core/executions";
 import { Logger } from "@/utils/Logger";
 import { useElapsedTime } from "../../../hooks/useElapsedTime";
 import { Tooltip } from "../../ui/tooltip";
@@ -18,11 +18,11 @@ import { formatElapsedTime, type Milliseconds, Time } from "@/utils/time";
 
 export interface CellStatusComponentProps {
   editing: boolean;
-  edited: boolean;
   disabled: boolean;
   /** Whether a cell it depends on is disabled. */
   blocked: boolean;
-  staleInputs: boolean;
+  /** Whether the cell's result is current. */
+  freshness: CellFreshness;
   status: CellRunStatus;
   /** Whether the cell's last run was interrupted. */
   interrupted: boolean;
@@ -38,8 +38,7 @@ export const CellStatusComponent: React.FC<CellStatusComponentProps> = ({
   status,
   disabled,
   blocked,
-  staleInputs,
-  edited,
+  freshness,
   interrupted,
   elapsedTime,
   startedAt,
@@ -48,6 +47,8 @@ export const CellStatusComponent: React.FC<CellStatusComponentProps> = ({
   if (!editing) {
     return null;
   }
+  const edited = freshness.sourceChanged;
+  const staleInputs = freshness.inputsChanged;
 
   const lastRanTime = startedAt ? (
     <LastRanTime lastRanTime={startedAt} />

package/src/components/editor/cell/__tests__/useRunCells.test.tsx

@@ -0,0 +1,169 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { act, renderHook } from "@testing-library/react";
+import { Provider } from "jotai";
+import type { ReactNode } from "react";
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { MockExecutions } from "@/__mocks__/executions";
+import { MockNotebook } from "@/__mocks__/notebook";
+import { MockRequestClient } from "@/__mocks__/requests";
+import { cellId } from "@/__tests__/branded";
+import { notebookAtom } from "@/core/cells/cells";
+import {
+  cellExecutionInputsAtom,
+  cellExecutionView,
+  notebookQueuedOrRunningCountAtom,
+} from "@/core/executions";
+import { requestClientAtom } from "@/core/network/requests";
+import type { DocumentModel } from "@/core/notebook/document";
+import {
+  cell,
+  document as documentOf,
+  withCells,
+} from "@/core/notebook/document/__tests__/fixtures";
+import { activeDocumentModelAtom } from "@/core/session/active-notebook";
+import { store } from "@/core/state/jotai";
+import { runCells, useRunStaleCells } from "../useRunCells";
+
+const ENABLED = cellId("enabled");
+const DISABLED = cellId("disabled");
+
+describe("runCells", () => {
+  it("sends a disabled cell to the kernel without reading it as queued", async () => {
+    MockExecutions.seed(store, []);
+    const notebook = MockNotebook.notebookState({
+      cellData: {
+        [ENABLED]: { code: "x = 1" },
+        [DISABLED]: { code: "y = 2", config: { disabled: true } },
+      },
+    });
+    const sendRun = vi.fn(async () => null);
+
+    await runCells({
+      cellIds: [ENABLED, DISABLED],
+      sendRun,
+      prepareForRun: vi.fn(),
+      notebook,
+    });
+
+    expect(sendRun).toHaveBeenCalledWith({
+      cellIds: [ENABLED, DISABLED],
+      codes: ["x = 1", "y = 2"],
+    });
+    const inputs = store.get(cellExecutionInputsAtom);
+    expect(cellExecutionView(inputs, ENABLED).status).toBe("queued");
+    expect(cellExecutionView(inputs, DISABLED).status).toBe("idle");
+    expect(store.get(notebookQueuedOrRunningCountAtom)).toBe(1);
+  });
+});
+
+describe("useRunStaleCells", () => {
+  const C1 = cellId("c1");
+  const C2 = cellId("c2");
+
+  afterEach(() => {
+    store.set(activeDocumentModelAtom, null);
+  });
+
+  it("flushes pending edits before choosing the stale cells", async () => {
+    const document = documentOf();
+    const replica = MockExecutions.seed(
+      store,
+      [
+        MockExecutions.completed(C1, { code_fingerprint: "c1-hash" }),
+        MockExecutions.completed(C2, { code_fingerprint: "c2-hash" }),
+      ],
+      null,
+      document,
+    );
+    store.set(
+      notebookAtom,
+      MockNotebook.notebookState({
+        cellData: { [C1]: { code: "x = 2" }, [C2]: { code: "y = 2" } },
+      }),
+    );
+    const client = MockRequestClient.create();
+    store.set(requestClientAtom, client);
+    // The edit to c1 is still in flight: the host confirms it on flush.
+    const flush = vi.fn(async () => {
+      await new Promise((resolve) => setTimeout(resolve, 0));
+      store.set(
+        replica.documentAtom,
+        withCells(document, {
+          [C1]: cell(C1, "x = 2", { code_fingerprint: "c1-hash-2" }),
+        }),
+      );
+    });
+    store.set(activeDocumentModelAtom, { flush } as unknown as DocumentModel);
+    const wrapper = ({ children }: { children: ReactNode }) => (
+      <Provider store={store}>{children}</Provider>
+    );
+    const { result } = renderHook(useRunStaleCells, { wrapper });
+
+    await act(() => result.current());
+
+    expect(flush).toHaveBeenCalledOnce();
+    expect(client.sendRun).toHaveBeenCalledWith({
+      cellIds: [C1],
+      codes: ["x = 2"],
+    });
+  });
+
+  it("does nothing when the notebook changed while the flush was pending", async () => {
+    MockExecutions.seed(
+      store,
+      [MockExecutions.completed(C1, { code_fingerprint: "stale" })],
+      null,
+      documentOf(),
+    );
+    store.set(
+      notebookAtom,
+      MockNotebook.notebookState({ cellData: { [C1]: { code: "x = 1" } } }),
+    );
+    const client = MockRequestClient.create();
+    store.set(requestClientAtom, client);
+    const flush = vi.fn(async () => {
+      store.set(activeDocumentModelAtom, null);
+      throw new Error("aborted");
+    });
+    store.set(activeDocumentModelAtom, { flush } as unknown as DocumentModel);
+    const wrapper = ({ children }: { children: ReactNode }) => (
+      <Provider store={store}>{children}</Provider>
+    );
+    const { result } = renderHook(useRunStaleCells, { wrapper });
+
+    await act(() => result.current());
+
+    expect(client.sendRun).not.toHaveBeenCalled();
+  });
+
+  it("still runs what is stale when the flush fails", async () => {
+    MockExecutions.seed(
+      store,
+      [MockExecutions.completed(C1, { code_fingerprint: "stale" })],
+      null,
+      documentOf(),
+    );
+    store.set(
+      notebookAtom,
+      MockNotebook.notebookState({ cellData: { [C1]: { code: "x = 1" } } }),
+    );
+    const client = MockRequestClient.create();
+    store.set(requestClientAtom, client);
+    const flush = vi.fn(async () => {
+      throw new Error("offline");
+    });
+    store.set(activeDocumentModelAtom, { flush } as unknown as DocumentModel);
+    const wrapper = ({ children }: { children: ReactNode }) => (
+      <Provider store={store}>{children}</Provider>
+    );
+    const { result } = renderHook(useRunStaleCells, { wrapper });
+
+    await act(() => result.current());
+
+    expect(client.sendRun).toHaveBeenCalledWith({
+      cellIds: [C1],
+      codes: ["x = 1"],
+    });
+  });
+});