@marimo-team/frontend 0.23.5-dev4 → 0.23.5-dev6

package/src/core/meta/__tests__/code-visibility.test.tsx

@@ -0,0 +1,141 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { renderHook } from "@testing-library/react";
+import { createStore, Provider } from "jotai";
+import type { ReactNode } from "react";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { showCodeInRunModeAtom } from "@/core/meta/state";
+import { type AppMode, kioskModeAtom, viewStateAtom } from "@/core/mode";
+import { useNotebookCodeAvailable } from "../code-visibility";
+
+interface StoreOpts {
+  mode?: AppMode;
+  kiosk?: boolean;
+  showInRunMode?: boolean;
+}
+
+function makeStore({
+  mode = "read",
+  kiosk = false,
+  showInRunMode = true,
+}: StoreOpts = {}) {
+  const store = createStore();
+  store.set(viewStateAtom, { mode, cellAnchor: null });
+  store.set(kioskModeAtom, kiosk);
+  store.set(showCodeInRunModeAtom, showInRunMode);
+  return store;
+}
+
+function wrap(store: ReturnType<typeof createStore>) {
+  return ({ children }: { children: ReactNode }) => (
+    <Provider store={store}>{children}</Provider>
+  );
+}
+
+const cellsWithCode = [{ code: "x = 1" }, { code: "" }];
+const cellsWithoutCode = [{ code: "" }, { code: "" }];
+
+const originalHref = window.location.href;
+
+function setSearch(search: string) {
+  window.history.replaceState(null, "", `/${search}`);
+}
+
+describe("useNotebookCodeAvailable", () => {
+  beforeEach(() => {
+    setSearch("");
+  });
+
+  afterEach(() => {
+    window.history.replaceState(null, "", originalHref);
+  });
+
+  it("returns true in edit mode regardless of cells", () => {
+    const store = makeStore({ mode: "edit" });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithoutCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(true);
+  });
+
+  it("returns true in present mode (only reachable from edit)", () => {
+    const store = makeStore({ mode: "present" });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithoutCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(true);
+  });
+
+  it("returns true in kiosk mode even when read mode and no code", () => {
+    const store = makeStore({
+      mode: "read",
+      kiosk: true,
+      showInRunMode: false,
+    });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithoutCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(true);
+  });
+
+  it("returns true in read mode when at least one cell has code", () => {
+    const store = makeStore({ mode: "read" });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(true);
+  });
+
+  it("returns false in read mode when every cell.code is empty (server stripped)", () => {
+    const store = makeStore({ mode: "read" });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithoutCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(false);
+  });
+
+  it("returns false in read mode when host opts out via showAppCode", () => {
+    const store = makeStore({ mode: "read", showInRunMode: false });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(false);
+  });
+
+  it("returns false in read mode when ?include-code=false", () => {
+    setSearch("?include-code=false");
+    const store = makeStore({ mode: "read" });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(false);
+  });
+
+  it("ignores ?include-code=false in kiosk mode", () => {
+    setSearch("?include-code=false");
+    const store = makeStore({ mode: "read", kiosk: true });
+    const { result } = renderHook(
+      () => useNotebookCodeAvailable(cellsWithoutCode),
+      { wrapper: wrap(store) },
+    );
+    expect(result.current).toBe(true);
+  });
+
+  it("returns false for non-edit/non-read modes (home, gallery)", () => {
+    for (const mode of ["home", "gallery"] as const) {
+      const store = makeStore({ mode });
+      const { result } = renderHook(
+        () => useNotebookCodeAvailable(cellsWithCode),
+        { wrapper: wrap(store) },
+      );
+      expect(result.current).toBe(false);
+    }
+  });
+});

package/src/core/meta/code-visibility.ts

@@ -0,0 +1,48 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { useAtomValue } from "jotai";
+import { KnownQueryParams } from "@/core/constants";
+import { showCodeInRunModeAtom } from "@/core/meta/state";
+import { kioskModeAtom, viewStateAtom } from "@/core/mode";
+import { logNever } from "@/utils/assertNever";
+
+/**
+ * Whether the notebook source code reached the frontend and can be rendered.
+ *
+ * In `marimo run` the server omits cell sources unless `--include-code` is
+ * set (every `cell.code` arrives as `""`). Use this to gate any "show code"
+ * / "copy code" / "download .py" affordance.
+ */
+export function useNotebookCodeAvailable(
+  cells: ReadonlyArray<{ code: string }>,
+): boolean {
+  const kioskMode = useAtomValue(kioskModeAtom);
+  const { mode } = useAtomValue(viewStateAtom);
+  const showInRunMode = useAtomValue(showCodeInRunModeAtom);
+
+  if (kioskMode) {
+    return true;
+  }
+
+  switch (mode) {
+    case "edit":
+    case "present":
+      return true;
+    case "home":
+    case "gallery":
+      return false;
+    case "read": {
+      if (!showInRunMode) {
+        return false;
+      }
+      const params = new URLSearchParams(window.location.search);
+      if (params.get(KnownQueryParams.includeCode) === "false") {
+        return false;
+      }
+      return cells.some((cell) => Boolean(cell.code));
+    }
+    default:
+      logNever(mode);
+      return false;
+  }
+}

package/dist/assets/RenderHTML-DR6lqJGS.js

@@ -1 +0,0 @@
-import{s as y}from"./chunk-LvLJmgfZ.js";import{i as g,p as C,u as M}from"./useEvent-D91BmmQi.js";import{t as S}from"./react-Bj1aDYRI.js";import{_n as H,ai as I,it as j,m as L,oi as c,xn as N,yt as k}from"./cells-A-ljPpuj.js";import{t as E}from"./compiler-runtime-B3qBwwSJ.js";import{_}from"./useEventListener-BR0C1MaI.js";import{o as w}from"./utils-DIGrmLDO.js";import{t as A}from"./jsx-runtime-BqBOg78p.js";import{t as F}from"./tooltip-Gcwqb_SK.js";import{t as O}from"./copy-icon-C67c9EwB.js";import{t as W}from"./usePress-CtfZXGno.js";import{n as q}from"./useDebounce-BDIglWmG.js";import{t as R}from"./useRunCells-DU1tawR-.js";var z=E(),m=y(S(),1),l=y(A(),1);const V=e=>{let t=(0,z.c)(16),r,n,i;t[0]===e?(r=t[1],n=t[2],i=t[3]):({href:n,children:r,...i}=e,t[0]=e,t[1]=r,t[2]=n,t[3]=i);let a=(0,m.useRef)(null),o;t[4]===n?o=t[5]:(o=()=>{let p=new URL(globalThis.location.href);p.hash=n,globalThis.history.pushState({},"",p.toString()),globalThis.dispatchEvent(new HashChangeEvent("hashchange"));let T=n.slice(1),x=document.getElementById(T);x&&x.scrollIntoView({behavior:"smooth",block:"start"})},t[4]=n,t[5]=o);let s=o,f;t[6]===s?f=t[7]:(f={onPress:()=>{s()}},t[6]=s,t[7]=f);let{pressProps:h}=W(f),u;t[8]===s?u=t[9]:(u=p=>{p.preventDefault(),s()},t[8]=s,t[9]=u);let b=u,d;return t[10]!==r||t[11]!==b||t[12]!==n||t[13]!==h||t[14]!==i?(d=(0,l.jsx)("a",{ref:a,href:n,...h,onClick:b,...i,children:r}),t[10]=r,t[11]=b,t[12]=n,t[13]=h,t[14]=i,t[15]=d):d=t[15],d};async function P(e){let t=L().inOrderIds.at(0);if(t)try{let r=await H.request({document:e,cellId:t});if(!r||r.options.length===0)return;let n=e.split(".").pop()??e,i=r.options[0],a=r.options.find(o=>o.name===n)??i;a!=null&&a.completion_info&&g.set(N,{documentation:a.completion_info})}catch(r){_.debug(`Doc lookup failed for "${e}"`,r)}}var D=E();const $=e=>{let t=(0,D.c)(8),{qualifiedName:r,children:n}=e,i;t[0]===r?i=t[1]:(i=()=>{P(r)},t[0]=r,t[1]=i);let a=q(i,100),o;t[2]===a?o=t[3]:(o=()=>a.cancel(),t[2]=a,t[3]=o);let s;return t[4]!==n||t[5]!==a||t[6]!==o?(s=(0,l.jsx)("span",{onMouseEnter:a,onMouseLeave:o,children:n}),t[4]=n,t[5]=a,t[6]=o,t[7]=s):s=t[7],s};function B(e){if(typeof e!="object"||!e)return!1;let t=e;return!(t.trusted!==!0||t.notebookCode!==void 0&&typeof t.notebookCode!="string")}function U(){let e=window==null?void 0:window.__MARIMO_EXPORT_CONTEXT__;return B(e)?e:void 0}function v(){var e;return((e=U())==null?void 0:e.trusted)===!0}function X(){if(g.get(R)||v()||g.get(w))return!0;try{if(j()==="read")return!0}catch{}return!1}var G=C(e=>{let t=e(R),r=e(w);if(t||r||v())return!1;let n=!0;try{n=j()==="read"}catch{return!0}return!n});function K(){return M(G)}var Z=E(),J=e=>{if(e instanceof c.Element&&!/^[A-Za-z][\w-]*$/.test(e.name))return m.createElement(m.Fragment)},Q=(e,t)=>{if(t instanceof c.Element&&t.name==="body"){if((0,m.isValidElement)(e)&&"props"in e){let r=e.props.children;return(0,l.jsx)(l.Fragment,{children:r})}return}},Y=(e,t)=>{if(t instanceof c.Element&&t.name==="html"){if((0,m.isValidElement)(e)&&"props"in e){let r=e.props.children;return(0,l.jsx)(l.Fragment,{children:r})}return}},tt=e=>{if(e instanceof c.Element&&e.attribs&&e.name==="iframe"){let t=document.createElement("iframe");return Object.entries(e.attribs).forEach(([r,n])=>{r.startsWith('"')&&r.endsWith('"')&&(r=r.slice(1,-1)),t.setAttribute(r,n)}),(0,l.jsx)("div",{dangerouslySetInnerHTML:{__html:t.outerHTML}})}},et=e=>{if(e instanceof c.Element&&e.name==="script"){let t=e.attribs.src;if(!t)return;if(!X())return _.warn(`[RenderHTML] refusing <script src> in untrusted context: ${t}`),(0,l.jsx)(l.Fragment,{});if(![...document.querySelectorAll("script[src]")].some(r=>r.getAttribute("src")===t)){let r=document.createElement("script");r.src=t,document.head.append(r)}return(0,l.jsx)(l.Fragment,{})}},rt=(e,t)=>{if(t instanceof c.Element&&t.name==="a"){let r=t.attribs.href;if(r!=null&&r.startsWith("#")&&!r.startsWith("#code/")){let n=null;return(0,m.isValidElement)(e)&&"props"in e&&(n=e.props.children),(0,l.jsx)(V,{href:r,...t.attribs,children:n})}}},nt=(e,t,r)=>{var n,i;if(t instanceof c.Element&&t.name==="div"&&((i=(n=t.attribs)==null?void 0:n.class)!=null&&i.includes("codehilite")))return(0,l.jsx)(ot,{children:e},r)},it=(e,t)=>{var r;if(t instanceof c.Element&&((r=t.attribs)!=null&&r["data-marimo-doc"])){let n=t.attribs["data-marimo-doc"];return(0,l.jsx)($,{qualifiedName:n,children:e})}},at=(e,t)=>{var r,n;if(t instanceof c.Element&&((r=t.attribs)!=null&&r["data-tooltip"])){if((((n=t.name)==null?void 0:n.toLowerCase())??"").startsWith("marimo-"))return;let i=t.attribs["data-tooltip"];return(0,l.jsx)(F,{content:i,children:e})}},ot=e=>{let t=(0,Z.c)(3),{children:r}=e,n=(0,m.useRef)(null),i;t[0]===Symbol.for("react.memo_cache_sentinel")?(i=(0,l.jsx)("div",{className:"absolute top-2 right-2 opacity-0 group-hover:opacity-100 transition-opacity",children:(0,l.jsx)(O,{tooltip:!1,className:"p-1",value:()=>{var s;let o=(s=n.current)==null?void 0:s.firstChild;return o&&o.textContent||""}})}),t[0]=i):i=t[0];let a;return t[1]===r?a=t[2]:(a=(0,l.jsxs)("div",{className:"relative group codehilite-wrapper",ref:n,children:[r,i]}),t[1]=r,t[2]=a),a};const st=({html:e,additionalReplacements:t=[],alwaysSanitizeHtml:r=!0})=>(0,l.jsx)(lt,{html:e,alwaysSanitizeHtml:r,additionalReplacements:t});var lt=({html:e,additionalReplacements:t=[],alwaysSanitizeHtml:r})=>{let n=K();return ct({html:(0,m.useMemo)(()=>r||n?k(e):e,[e,r,n]),additionalReplacements:t})};function ct({html:e,additionalReplacements:t=[]}){let r=[J,tt,et,...t],n=[nt,rt,it,at,Q,Y];return I(e,{replace:(i,a)=>{for(let o of r){let s=o(i,a);if(s)return s}return i},transform:(i,a,o)=>{for(let s of n){let f=s(i,a,o);if(f)return f}return i}})}export{v as n,st as t};

package/dist/assets/channel-BMMN1KqZ.js

@@ -1 +0,0 @@
-import{et as r,tt as e}from"./chunk-ABZYJK2D-X_7JfTbo.js";var o=(t,a)=>e.lang.round(r.parse(t)[a]);export{o as t};

package/dist/assets/classDiagram-2ON5EDUG-DDa_MJVI.js

@@ -1 +0,0 @@
-import"./purify.es-DC4RGS9t.js";import"./src-nKy6a3tt.js";import"./chunk-S3R3BYOJ-GS_DuFun.js";import{n as t}from"./src-CmKhyEBC.js";import"./chunk-ABZYJK2D-X_7JfTbo.js";import"./chunk-HN2XXSSU-BBxJMYrA.js";import"./chunk-CVBHYZKI-Brsoob-W.js";import"./chunk-ATLVNIR6-BvOe4C7U.js";import"./dist-DxxvVPQH.js";import"./chunk-JA3XYJ7Z-EueUUcXO.js";import"./chunk-JZLCHNYA-Bx4SqpPS.js";import"./chunk-QXUST7PY-COkAFJCN.js";import"./chunk-N4CR4FBY-DlrBiMsl.js";import"./chunk-FMBD7UC4-Bt9icXTy.js";import"./chunk-55IACEB6-DD051KDZ.js";import"./chunk-QN33PNHL-DQrYk9Fj.js";import{i,n as o,r as m,t as p}from"./chunk-B4BG7PRW-DuJr4Vxg.js";var a={parser:o,get db(){return new p},renderer:m,styles:i,init:t(r=>{r.class||(r.class={}),r.class.arrowMarkerAbsolute=r.arrowMarkerAbsolute},"init")};export{a as diagram};

package/dist/assets/classDiagram-v2-WZHVMYZB-dZzbS5Uq.js

@@ -1 +0,0 @@
-import"./purify.es-DC4RGS9t.js";import"./src-nKy6a3tt.js";import"./chunk-S3R3BYOJ-GS_DuFun.js";import{n as t}from"./src-CmKhyEBC.js";import"./chunk-ABZYJK2D-X_7JfTbo.js";import"./chunk-HN2XXSSU-BBxJMYrA.js";import"./chunk-CVBHYZKI-Brsoob-W.js";import"./chunk-ATLVNIR6-BvOe4C7U.js";import"./dist-DxxvVPQH.js";import"./chunk-JA3XYJ7Z-EueUUcXO.js";import"./chunk-JZLCHNYA-Bx4SqpPS.js";import"./chunk-QXUST7PY-COkAFJCN.js";import"./chunk-N4CR4FBY-DlrBiMsl.js";import"./chunk-FMBD7UC4-Bt9icXTy.js";import"./chunk-55IACEB6-DD051KDZ.js";import"./chunk-QN33PNHL-DQrYk9Fj.js";import{i,n as o,r as m,t as p}from"./chunk-B4BG7PRW-DuJr4Vxg.js";var a={parser:o,get db(){return new p},renderer:m,styles:i,init:t(r=>{r.class||(r.class={}),r.class.arrowMarkerAbsolute=r.arrowMarkerAbsolute},"init")};export{a as diagram};

package/dist/assets/dist-BN_qisx5.js

@@ -1 +0,0 @@
-import"./dist-BrR4M-k3.js";import"./dist-DGxytjtv.js";import{n as s,r as a,t as o}from"./dist-OWcDq-O3.js";export{o as sass,s as sassCompletionSource,a as sassLanguage};

package/dist/assets/dist-Bzlk7AB1.js

@@ -1 +0,0 @@
-import"./dist-BrR4M-k3.js";import"./dist-CEysMnMQ.js";import"./dist-DGxytjtv.js";import"./dist-DZ2qbK2N.js";import{n as p,t as o}from"./dist-DmZscTVZ.js";export{o as php,p as phpLanguage};

package/dist/assets/dist-CPCp_PIz.js

@@ -1 +0,0 @@
-import"./dist-BrR4M-k3.js";import"./dist-CEysMnMQ.js";import"./dist-DGxytjtv.js";import"./dist-DZ2qbK2N.js";import{i,n as r,r as a,t as o}from"./dist-C46P-i95.js";export{o as closePercentBrace,r as liquid,a as liquidCompletionSource,i as liquidLanguage};

package/dist/assets/dist-CPWl1X8t.js

@@ -1 +0,0 @@
-import"./dist-BrR4M-k3.js";import"./dist-CEysMnMQ.js";import"./dist-DGxytjtv.js";import"./dist-DZ2qbK2N.js";import{n as o,t as r}from"./dist-BtG3xAGt.js";export{r as vue,o as vueLanguage};

package/dist/assets/dist-Cc_qFbl6.js

@@ -1 +0,0 @@
-import"./dist-BrR4M-k3.js";import"./dist-DGxytjtv.js";import{n as s,r as a,t as e}from"./dist-0xasIeT1.js";export{e as less,s as lessCompletionSource,a as lessLanguage};

package/dist/assets/dist-Crt3qZze.js

@@ -1 +0,0 @@
-import"./dist-BrR4M-k3.js";import"./dist-CEysMnMQ.js";import"./dist-DGxytjtv.js";import"./dist-DZ2qbK2N.js";import{i as a,n as r,r as o,t as i}from"./dist-COpce3-f.js";export{i as closePercentBrace,r as jinja,o as jinjaCompletionSource,a as jinjaLanguage};