@marimo-team/islands 0.25.3-dev18 → 0.25.3-dev181

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (618) hide show
  1. package/dist/{ConnectedDataExplorerComponent-CNMliVzw.js → ConnectedDataExplorerComponent-Le_OBfmt.js} +21 -19
  2. package/dist/{ErrorBoundary-DP07HKNC.js → ErrorBoundary-Do1bDnqE.js} +19 -19
  3. package/dist/{abnfDiagram-VCTEODGH-D-Ez91gV.js → abnfDiagram-VCTEODGH-Tf1ghyQQ.js} +2 -2
  4. package/dist/any-language-editor-BVFHDDQ3.js +75 -0
  5. package/dist/{architectureDiagram-5GKGNRK7-CTpcNtO8.js → architectureDiagram-5GKGNRK7-xlnbTG0T.js} +1 -1
  6. package/dist/{arrays-CN1djC4k.js → arrays-R6JsImiT.js} +8 -0
  7. package/dist/assets/__vite-browser-external-DVeFf2e7.js +1 -0
  8. package/dist/assets/worker-C1pfNBsx.js +113 -0
  9. package/dist/{button-CvjkwdSk.js → button-BQRdqQi1.js} +199 -192
  10. package/dist/{capabilities-CQ5IjhzL.js → capabilities-CI4xUPsp.js} +2 -2
  11. package/dist/{chat-ui-DaHynk8g.js → chat-ui-X-cWrL9u.js} +541 -540
  12. package/dist/{check-EDwSwP89.js → check-C9Mp9eCK.js} +1 -1
  13. package/dist/{chunk-IMKFNOWR-DGxi-zeo.js → chunk-IMKFNOWR-Cmd4a5jy.js} +1 -1
  14. package/dist/{chunk-SHT3W25Y-CV9J3rAB.js → chunk-SHT3W25Y-mSNp-WpO.js} +1 -1
  15. package/dist/{chunk-SVP7TREG-BL-9Fqgg.js → chunk-SVP7TREG-CWC49D8X.js} +1 -1
  16. package/dist/{chunk-TICWLB2K-Cu_XMjDM.js → chunk-TICWLB2K-De9FINtG.js} +1 -1
  17. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  18. package/dist/{classDiagram-ZZMXUADV-BXXV13Hq.js → classDiagram-ZZMXUADV-DweTkYdp.js} +2 -2
  19. package/dist/{classDiagram-v2-VYDZK3BY-B7FXJ1Cp.js → classDiagram-v2-VYDZK3BY-A6MR2TiE.js} +2 -2
  20. package/dist/{code-visibility-hTy956cB.js → code-visibility-CjilY2pq.js} +25 -25
  21. package/dist/{common-CiockEHa.js → common-CgbJYN1L.js} +6994 -6819
  22. package/dist/{copy-icon-Ft7t9bqz.js → copy-icon-De584pZ4.js} +3 -3
  23. package/dist/{cynefinDiagram-5FMLGOSQ-CH96rwba.js → cynefinDiagram-5FMLGOSQ-BSvo5le-.js} +1 -1
  24. package/dist/{dagre-GXQ25YYZ-BD3PmpFY.js → dagre-GXQ25YYZ-tXRQslO6.js} +1 -1
  25. package/dist/{data-grid-overlay-editor-XsGuhhu5.js → data-grid-overlay-editor-Bov1HRwt.js} +5 -5
  26. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  27. package/dist/{diagram-S7CK7UJ4-BsSJJldB.js → diagram-S7CK7UJ4-r3t5VtwI.js} +1 -1
  28. package/dist/{diagram-UQ7AKVKN-muQtNZc4.js → diagram-UQ7AKVKN-qgbC8YnT.js} +1 -1
  29. package/dist/{diagram-VX7I27RA-CdhuZAqR.js → diagram-VX7I27RA-CiHxEwtp.js} +1 -1
  30. package/dist/{diagram-Z3DM3KII-BD8mXjOj.js → diagram-Z3DM3KII-D95u15Ev.js} +1 -1
  31. package/dist/{dialog-CEBFZNyc.js → dialog-DVRfQuhD.js} +25 -25
  32. package/dist/{dist-F8Ep36rk.js → dist-B1M0bGO3.js} +1 -1
  33. package/dist/{dist-D6d86eNJ.js → dist-BA13RJj5.js} +1 -1
  34. package/dist/{dist-Bvs24R_E.js → dist-BJtmSgcE.js} +48 -48
  35. package/dist/{dist-CN_BeG2I.js → dist-Ba2VIn1K.js} +53 -53
  36. package/dist/{dist-PXYI_yuP.js → dist-Bj4bmwUm.js} +12 -12
  37. package/dist/{dist-DoE4XJBC.js → dist-BmS-ZGjl.js} +8626 -8626
  38. package/dist/{dist-BorMD-uX.js → dist-Bzogh7CW.js} +87 -87
  39. package/dist/{dist-CN8zI-XC.js → dist-CThPVKaP.js} +1385 -1382
  40. package/dist/{dist-DiDibOYr.js → dist-CTi5FTiV.js} +87 -87
  41. package/dist/{dist-diIx5qJW.js → dist-Ck_C32ye.js} +2 -2
  42. package/dist/{dist-BLfKmR-l.js → dist-D1fnFp1J.js} +1 -1
  43. package/dist/{dist-OR1OnN-y.js → dist-D3x9__Fu.js} +46 -46
  44. package/dist/{dist-BMQf5zIS.js → dist-D7mrh5sO.js} +36 -36
  45. package/dist/{dist-CSJ0MDRw.js → dist-D9Q3xjiP.js} +1 -1
  46. package/dist/{dist-BuaRHxOc.js → dist-DP60Ubh8.js} +2 -2
  47. package/dist/{dist-CSbh1PNP.js → dist-DTv9KkpW.js} +12 -12
  48. package/dist/{dist-D-wwqlzJ.js → dist-DVVwOfrD.js} +109 -109
  49. package/dist/{dist-C8fAKSH1.js → dist-DYXM73-k.js} +36 -36
  50. package/dist/{dist-DBtSl3sf.js → dist-DjeybPOb.js} +1 -1
  51. package/dist/{dist-CrSP_H1M.js → dist-DtGn9N4h.js} +197 -197
  52. package/dist/{dist-CLdpGQVr.js → dist-KN80qWCP.js} +2 -2
  53. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  54. package/dist/{dist-Cg7gzw-a.js → dist-O0qFpEJh.js} +56 -56
  55. package/dist/{dist-CkTTIdBe.js → dist-OUle2RT7.js} +207 -207
  56. package/dist/{dist-BK4Oj99o.js → dist-T5rh6JvL.js} +1 -1
  57. package/dist/{dropdown-menu-CP_qCYHC.js → dropdown-menu-DCMbbQ35.js} +1481 -1433
  58. package/dist/{ebnfDiagram-PWID7BFC-BIN1sweE.js → ebnfDiagram-PWID7BFC-P7e4bryK.js} +2 -2
  59. package/dist/{erDiagram-RLTQ6QDP-Ctp5cfoN.js → erDiagram-RLTQ6QDP-CUHZcmRj.js} +1 -1
  60. package/dist/{error-banner-D-WCIiMt.js → error-banner-DxJ3s8YO.js} +86 -86
  61. package/dist/{error-entries-Bv6bZPF3.js → error-entries-mpoD5t79.js} +4771 -4702
  62. package/dist/{esm-D2gTsXEV.js → esm-Bbw_Njjx.js} +575 -575
  63. package/dist/{esm-AMFssoNp.js → esm-mDQNEFe_.js} +354 -354
  64. package/dist/{feedback-button-CZNhgq7U.js → feedback-button-M3asLLYJ.js} +37 -37
  65. package/dist/{flowDiagram-HODETNUW-3EfLrlFb.js → flowDiagram-HODETNUW-BGZ95AzI.js} +2 -2
  66. package/dist/{formats-27CW_UP4.js → formats-DI7HEVv7.js} +3 -3
  67. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  68. package/dist/{glide-data-editor-DNPChdXf.js → glide-data-editor-BYx8nPvz.js} +370 -368
  69. package/dist/{glide-utils-DJyqh0EM.js → glide-utils-D98laidb.js} +16 -16
  70. package/dist/{globals-DcKi3Zmn.js → globals-Cha-b9Oo.js} +3 -3
  71. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  72. package/dist/{html-to-image-Bf8QMGi4.js → html-to-image-BoZthB4P.js} +668 -622
  73. package/dist/{infoDiagram-27XIBGKW-DipQoHxl.js → infoDiagram-27XIBGKW-BF0YX70D.js} +1 -1
  74. package/dist/{ishikawaDiagram-5VMMS53U-DgFMjSGz.js → ishikawaDiagram-5VMMS53U-_itsAAn-.js} +1 -1
  75. package/dist/{kanban-definition-UXKFOSKX-w1ACVjgl.js → kanban-definition-UXKFOSKX-VNW2Beqh.js} +1 -1
  76. package/dist/{label-BDKF3Afc.js → label-Qvw1E97W.js} +3 -3
  77. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  78. package/dist/{loader-BQOfjTIJ.js → loader-DOVB0Syl.js} +12 -12
  79. package/dist/main.js +9822 -7824
  80. package/dist/{mermaid-CHo49Yv-.js → mermaid-CZ6FrClS.js} +38 -38
  81. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  82. package/dist/{mindmap-definition-YA3MSWOX-CrT0KSfX.js → mindmap-definition-YA3MSWOX-DR4ripvd.js} +1 -1
  83. package/dist/{number-overlay-editor-GnpmNxbw.js → number-overlay-editor-9juAuy71.js} +3 -3
  84. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  85. package/dist/{once-BtE-wKRL.js → once-eG8Vmvjd.js} +1 -1
  86. package/dist/{pegDiagram-XKGWAZYB-BqvywNJA.js → pegDiagram-XKGWAZYB-CFr3XlGl.js} +2 -2
  87. package/dist/{pieDiagram-E7YTZNPT-BfxJ_XCc.js → pieDiagram-E7YTZNPT-j91XYSLY.js} +1 -1
  88. package/dist/precisionRound-DhAzeRcm.js +49 -0
  89. package/dist/{process-output-BZg56CCb.js → process-output-DLS4Mqg_.js} +4 -4
  90. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  91. package/dist/{railroadDiagram-O6MQD6OU-8vFLVTPu.js → railroadDiagram-O6MQD6OU-Cd47ZV9E.js} +2 -2
  92. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  93. package/dist/{requirementDiagram-BXWQKSXE-a_MxGvxe.js → requirementDiagram-BXWQKSXE-eJH6gLtX.js} +1 -1
  94. package/dist/{reveal-component-B1Wbl-hd.js → reveal-component-DPgpYGKs.js} +14 -14
  95. package/dist/{select-BButFi-d.js → select-2KOiReNY.js} +536 -457
  96. package/dist/{sequenceDiagram-WJ2MYXX4-CXFMnxGh.js → sequenceDiagram-WJ2MYXX4-DdfFEIrm.js} +1 -1
  97. package/dist/{spec-BwarBpUJ.js → spec-h69nC3SI.js} +41 -11
  98. package/dist/{stateDiagram-D77RDMKH-DQeA_CwI.js → stateDiagram-D77RDMKH-DkjDvip8.js} +2 -2
  99. package/dist/{stateDiagram-v2-MP3YSRHH-BxxQucm1.js → stateDiagram-v2-MP3YSRHH-DBHACnOo.js} +2 -2
  100. package/dist/style.css +1 -1
  101. package/dist/{swimlanes-42K2YHIH-Do6L-cYg.js → swimlanes-42K2YHIH-ogImlOmp.js} +1 -1
  102. package/dist/{swimlanesDiagram-VR7AAH4N-DUzPvSVU.js → swimlanesDiagram-VR7AAH4N-D6x-zsci.js} +2 -2
  103. package/dist/{swiper-component-DKntyn8p.js → swiper-component-BkCm-kLg.js} +2 -2
  104. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  105. package/dist/{timeline-definition-24CTP7MA-D3c46WM5.js → timeline-definition-24CTP7MA-DQpLtsHD.js} +1 -1
  106. package/dist/{toDate-UtiwmnH2.js → toDate-CDzoofJo.js} +37 -36
  107. package/dist/{toast-B9UMEi8u.js → toast-B5t9o6s_.js} +219 -201
  108. package/dist/{tooltip-BO1REbqs.js → tooltip-s4zZxwKe.js} +325 -301
  109. package/dist/{use-restore-focus-BEM7sFWl.js → use-restore-focus-BgDTd2tc.js} +1482 -1433
  110. package/dist/{useAsyncData-Deh5kszT.js → useAsyncData-Bn8SxUqM.js} +15 -14
  111. package/dist/{useDateFormatter-D8XruNGG.js → useDateFormatter-D3Tkfzej.js} +2 -2
  112. package/dist/{useDeepCompareMemoize-DXwzRINI.js → useDeepCompareMemoize-CC2tnPCk.js} +1 -1
  113. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-CSdWn6NP.js} +1 -1
  114. package/dist/{useLifecycle-j7D9uD1o.js → useLifecycle-BJlxHqWa.js} +28 -25
  115. package/dist/{useTheme-CRnjkku0.js → useTheme-DMfQYmlG.js} +301 -297
  116. package/dist/{vega-component-BeF9dfyO.js → vega-component-CovWHi60.js} +148 -148
  117. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  118. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  119. package/dist/{vennDiagram-4TSXK5OY-BSZD8JcZ.js → vennDiagram-4TSXK5OY-BD-SL_vq.js} +1 -1
  120. package/dist/{wardleyDiagram-VM6X3IG4-48cUWkfA.js → wardleyDiagram-VM6X3IG4-BDgJV4ah.js} +1 -1
  121. package/dist/{xychartDiagram-S5SC5T6Z-CswYgYeu.js → xychartDiagram-S5SC5T6Z-1hIHN6Rl.js} +2 -2
  122. package/package.json +16 -16
  123. package/src/__mocks__/document.tsx +54 -0
  124. package/src/__mocks__/notebook.ts +2 -1
  125. package/src/__mocks__/requests.ts +0 -3
  126. package/src/__tests__/module-entry.test.ts +44 -0
  127. package/src/__tests__/mount.test.ts +1 -21
  128. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  129. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  130. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  131. package/src/components/app-config/ai-config.tsx +7 -2
  132. package/src/components/app-config/user-config-form.tsx +35 -0
  133. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  134. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  135. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  136. package/src/components/chat/acp/blocks.tsx +3 -2
  137. package/src/components/chat/acp/session-tabs.tsx +8 -3
  138. package/src/components/chat/chat-components.tsx +2 -0
  139. package/src/components/chat/chat-panel.tsx +12 -11
  140. package/src/components/chat/chat-utils.ts +4 -8
  141. package/src/components/data-table/TableBottomBar.tsx +4 -1
  142. package/src/components/data-table/TableTopBar.tsx +2 -0
  143. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  144. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  145. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  146. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  147. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  148. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  149. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  150. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  151. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  152. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  153. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  154. package/src/components/data-table/cell-selection/feature.ts +2 -2
  155. package/src/components/data-table/cell-selection/types.ts +2 -2
  156. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  157. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  158. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  159. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  160. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  161. package/src/components/data-table/charts/charts.tsx +40 -30
  162. package/src/components/data-table/charts/storage.ts +16 -5
  163. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  164. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  165. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  166. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  167. package/src/components/data-table/columns.tsx +45 -46
  168. package/src/components/data-table/context-menu.tsx +36 -20
  169. package/src/components/data-table/data-table.tsx +17 -1
  170. package/src/components/data-table/date-popover.tsx +6 -4
  171. package/src/components/data-table/export-actions.tsx +76 -48
  172. package/src/components/data-table/pagination.tsx +44 -33
  173. package/src/components/data-table/range-focus/atoms.ts +4 -6
  174. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  175. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  176. package/src/components/data-table/renderers.tsx +16 -7
  177. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  178. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  179. package/src/components/data-table/search-highlight.tsx +79 -0
  180. package/src/components/data-table/types.ts +1 -0
  181. package/src/components/data-table/url-detector.tsx +49 -9
  182. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  183. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  184. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  185. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  186. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  187. package/src/components/datasources/column-preview.tsx +17 -12
  188. package/src/components/datasources/datasources.tsx +55 -42
  189. package/src/components/datasources/utils.ts +32 -23
  190. package/src/components/debugger/debugger-code.tsx +5 -0
  191. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  192. package/src/components/dependency-graph/elements.ts +24 -21
  193. package/src/components/dependency-graph/minimap-content.tsx +2 -6
  194. package/src/components/dependency-graph/panels.tsx +4 -2
  195. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  196. package/src/components/editor/SortableCell.tsx +1 -3
  197. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  198. package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
  199. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -13
  200. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  201. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -1
  202. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  203. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  204. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +34 -8
  205. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +2 -6
  206. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  207. package/src/components/editor/actions/pair-with-agent-commands.ts +2 -2
  208. package/src/components/editor/actions/pair-with-agent-modal.tsx +19 -16
  209. package/src/components/editor/actions/useCellActionButton.tsx +20 -26
  210. package/src/components/editor/actions/useNotebookActions.tsx +14 -19
  211. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  212. package/src/components/editor/actions/useSetCodeVisibility.ts +10 -11
  213. package/src/components/editor/actions/useSetOutputsExpanded.ts +13 -18
  214. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  215. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  216. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  217. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  218. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  219. package/src/components/editor/app-container.tsx +1 -1
  220. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  221. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  222. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  223. package/src/components/editor/cell/RunButton.tsx +2 -2
  224. package/src/components/editor/cell/StagedAICell.tsx +10 -2
  225. package/src/components/editor/cell/StopButton.tsx +2 -2
  226. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +3 -1
  227. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  228. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  229. package/src/components/editor/cell/code/cell-editor.tsx +22 -36
  230. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  231. package/src/components/editor/cell/collapse.tsx +5 -3
  232. package/src/components/editor/cell/useAddCell.ts +4 -8
  233. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  234. package/src/components/editor/cell/useRunCells.ts +2 -5
  235. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  236. package/src/components/editor/cell/useSplitCell.tsx +2 -1
  237. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  238. package/src/components/editor/chrome/components/feedback-button.tsx +1 -1
  239. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  240. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  241. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  242. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  243. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  244. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  245. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  246. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  247. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  248. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  249. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  250. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +9 -18
  251. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  252. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +61 -2
  253. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
  254. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  255. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  256. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  257. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  258. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  259. package/src/components/editor/chrome/wrapper/footer.tsx +2 -0
  260. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  261. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  262. package/src/components/editor/columns/sortable-column.tsx +1 -1
  263. package/src/components/editor/connections/components.tsx +5 -10
  264. package/src/components/editor/controls/Controls.tsx +3 -3
  265. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  266. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  267. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  268. package/src/components/editor/controls/command-palette.tsx +22 -10
  269. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  270. package/src/components/editor/documentation.css +34 -11
  271. package/src/components/editor/dynamic-favicon.tsx +21 -40
  272. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +9 -23
  273. package/src/components/editor/errors/auto-fix.tsx +4 -8
  274. package/src/components/editor/errors/send-error-report-button.tsx +3 -6
  275. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  276. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  277. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  278. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  279. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  280. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  281. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  282. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  283. package/src/components/editor/header/app-header.tsx +1 -1
  284. package/src/components/editor/header/status.tsx +40 -20
  285. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  286. package/src/components/editor/links/cell-link.tsx +16 -6
  287. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  288. package/src/components/editor/navigation/__tests__/navigation.test.ts +73 -64
  289. package/src/components/editor/navigation/clipboard.ts +12 -11
  290. package/src/components/editor/navigation/focus-utils.ts +4 -4
  291. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +11 -20
  292. package/src/components/editor/navigation/navigation.ts +20 -28
  293. package/src/components/editor/notebook-banner.tsx +1 -0
  294. package/src/components/editor/notebook-cell.tsx +19 -13
  295. package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
  296. package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
  297. package/src/components/editor/output/Outputs.css +8 -0
  298. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  299. package/src/components/editor/output/ansi-reduce.ts +33 -20
  300. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  301. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  302. package/src/components/editor/package-alert.tsx +14 -7
  303. package/src/components/editor/renderers/cell-array.tsx +10 -17
  304. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  305. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  306. package/src/components/home/components.tsx +17 -31
  307. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  308. package/src/components/icons/large-spinner.tsx +3 -14
  309. package/src/components/icons/spinner.tsx +1 -1
  310. package/src/components/markdown/markdown-renderer.tsx +4 -8
  311. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  312. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  313. package/src/components/pages/edit-page.tsx +32 -17
  314. package/src/components/pages/home-page.tsx +164 -91
  315. package/src/components/scratchpad/scratchpad.test.tsx +97 -0
  316. package/src/components/scratchpad/scratchpad.tsx +20 -10
  317. package/src/components/slides/__tests__/minimap-actions.test.tsx +9 -25
  318. package/src/components/slides/minimap.tsx +7 -3
  319. package/src/components/slides/reveal-component.tsx +4 -2
  320. package/src/components/slides/slide-cell-view.tsx +4 -4
  321. package/src/components/slides/slide-form.tsx +1 -0
  322. package/src/components/sort/SortableCellsProvider.tsx +2 -1
  323. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  324. package/src/components/tracing/tracing.tsx +45 -41
  325. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  326. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  327. package/src/components/ui/alert-dialog.tsx +1 -1
  328. package/src/components/ui/draggable-popover.tsx +24 -2
  329. package/src/components/ui/range-slider.tsx +22 -9
  330. package/src/core/ai/__tests__/staged-cells.test.ts +26 -25
  331. package/src/core/ai/context/providers/datasource.ts +6 -2
  332. package/src/core/ai/staged-cells.ts +15 -18
  333. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +5 -17
  334. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +2 -2
  335. package/src/core/ai/tools/base.ts +18 -9
  336. package/src/core/ai/tools/edit-notebook-tool.ts +74 -69
  337. package/src/core/ai/tools/registry.ts +13 -8
  338. package/src/core/ai/tools/sample-tool.ts +6 -7
  339. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -0
  340. package/src/core/cells/__tests__/actions.test.ts +474 -34
  341. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  342. package/src/core/cells/__tests__/cell.test.ts +39 -0
  343. package/src/core/cells/__tests__/cells.test.ts +260 -1165
  344. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  345. package/src/core/cells/__tests__/document.test.ts +414 -0
  346. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  347. package/src/core/cells/__tests__/notebook-outline.test.ts +114 -0
  348. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  349. package/src/core/cells/__tests__/session.test.ts +117 -36
  350. package/src/core/cells/__tests__/with-document-save.test.ts +67 -0
  351. package/src/core/cells/actions.ts +391 -3
  352. package/src/core/cells/add-missing-import.ts +46 -42
  353. package/src/core/cells/cell.ts +14 -3
  354. package/src/core/cells/cells.ts +99 -599
  355. package/src/core/cells/columns.ts +38 -0
  356. package/src/core/cells/document.ts +392 -0
  357. package/src/core/cells/outputs.ts +12 -28
  358. package/src/core/cells/pending-delete-service.ts +2 -6
  359. package/src/core/cells/session.ts +1 -1
  360. package/src/core/cells/types.ts +22 -1
  361. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  362. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  363. package/src/core/codemirror/ai/resources.ts +1 -1
  364. package/src/core/codemirror/cells/__tests__/extensions.test.ts +42 -1
  365. package/src/core/codemirror/cells/extensions.ts +27 -9
  366. package/src/core/codemirror/cells/state.ts +1 -1
  367. package/src/core/codemirror/format.ts +6 -5
  368. package/src/core/codemirror/keymaps/vim.ts +4 -8
  369. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  370. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  371. package/src/core/codemirror/language/cell-code.ts +114 -0
  372. package/src/core/codemirror/language/extension.ts +52 -35
  373. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  374. package/src/core/codemirror/language/languages/python.ts +16 -10
  375. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  376. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  377. package/src/core/codemirror/language/types.ts +7 -7
  378. package/src/core/codemirror/language/utils.ts +9 -39
  379. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  380. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  381. package/src/core/codemirror/readonly/extension.ts +2 -2
  382. package/src/core/codemirror/rtc/extension.ts +24 -13
  383. package/src/core/codemirror/rtc/loro/awareness.ts +34 -19
  384. package/src/core/codemirror/rtc/loro/sync.ts +1 -0
  385. package/src/core/codemirror/utils.ts +1 -0
  386. package/src/core/config/__tests__/config-schema.test.ts +5 -3
  387. package/src/core/config/config-schema.ts +3 -0
  388. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  389. package/src/core/datasets/data-source-discovery.ts +4 -6
  390. package/src/core/edit-app.tsx +19 -23
  391. package/src/core/errors/error-entries.ts +3 -0
  392. package/src/core/export/hooks.ts +1 -1
  393. package/src/core/export/layout.ts +2 -2
  394. package/src/core/hotkeys/actions.ts +2 -4
  395. package/src/core/islands/__tests__/bootstrap.test.ts +73 -43
  396. package/src/core/islands/__tests__/bridge.test.ts +72 -25
  397. package/src/core/islands/bootstrap.ts +46 -53
  398. package/src/core/islands/bridge.ts +74 -23
  399. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  400. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  401. package/src/core/islands/worker/controller.ts +4 -10
  402. package/src/core/islands/worker/worker.tsx +33 -33
  403. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  404. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  405. package/src/core/kernel/handlers.ts +18 -59
  406. package/src/core/network/__tests__/connection.test.ts +1 -1
  407. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  408. package/src/core/network/connection-notice.ts +1 -1
  409. package/src/core/network/connection.ts +1 -1
  410. package/src/core/network/requests-lazy.ts +1 -4
  411. package/src/core/network/requests-network.ts +0 -25
  412. package/src/core/network/requests-static.ts +0 -3
  413. package/src/core/network/requests-toasting.tsx +0 -3
  414. package/src/core/network/types.ts +0 -12
  415. package/src/core/network/useConnectionNotice.ts +8 -2
  416. package/src/core/notebook/attachments.ts +38 -0
  417. package/src/core/notebook/document/__tests__/codemirror.test.ts +552 -0
  418. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  419. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  420. package/src/core/notebook/document/__tests__/hold.test.ts +88 -0
  421. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  422. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  423. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  424. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  425. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  426. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  427. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  428. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  429. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  430. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  431. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  432. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  433. package/src/core/notebook/document/__tests__/user.ts +61 -0
  434. package/src/core/notebook/document/codemirror.ts +216 -0
  435. package/src/core/notebook/document/draft.ts +502 -0
  436. package/src/core/notebook/document/hooks.tsx +118 -0
  437. package/src/core/notebook/document/index.ts +24 -0
  438. package/src/core/notebook/document/model.ts +105 -0
  439. package/src/core/notebook/document/select.ts +41 -0
  440. package/src/core/notebook/document/state.ts +378 -0
  441. package/src/core/notebook/document/sync.ts +418 -0
  442. package/src/core/notebook/index.ts +18 -0
  443. package/src/core/notebook/replica/AGENTS.md +11 -0
  444. package/src/core/notebook/replica/__tests__/attachments.test.ts +56 -0
  445. package/src/core/notebook/replica/__tests__/document.test.ts +126 -0
  446. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  447. package/src/core/notebook/replica/__tests__/replica.test.ts +550 -0
  448. package/src/core/notebook/replica/backend.ts +19 -0
  449. package/src/core/notebook/replica/connection.ts +42 -0
  450. package/src/core/notebook/replica/index.ts +13 -0
  451. package/src/core/notebook/replica/replica.ts +286 -0
  452. package/src/core/notebook/replica/resources/attachments.ts +22 -0
  453. package/src/core/notebook/replica/resources/document.ts +101 -0
  454. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  455. package/src/core/packages/useSandboxController.ts +1 -1
  456. package/src/core/run-app.tsx +48 -51
  457. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  458. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  459. package/src/core/runtime/__tests__/runtime.test.ts +32 -58
  460. package/src/core/runtime/adapter.ts +1 -1
  461. package/src/core/runtime/config.ts +2 -2
  462. package/src/core/runtime/runtime.ts +38 -42
  463. package/src/core/runtime/types.ts +0 -2
  464. package/src/core/saving/filename.ts +1 -1
  465. package/src/core/saving/save-component.tsx +10 -4
  466. package/src/core/saving/useAutoSave.ts +1 -1
  467. package/src/core/session/__tests__/backend.test.ts +235 -0
  468. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +183 -0
  469. package/src/core/session/__tests__/useNotebookSession.hook.test.tsx +663 -0
  470. package/src/core/session/active-notebook.ts +11 -0
  471. package/src/core/session/backend.ts +125 -0
  472. package/src/core/{websocket → session}/types.ts +2 -2
  473. package/src/core/session/useNotebookReplica.ts +72 -0
  474. package/src/core/{websocket → session}/useNotebookSession.tsx +74 -68
  475. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  476. package/src/core/state/jotai.ts +1 -1
  477. package/src/core/static/__tests__/files.test.ts +38 -10
  478. package/src/core/static/files.ts +2 -0
  479. package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
  480. package/src/core/transports/__tests__/server.test.ts +408 -0
  481. package/src/core/transports/__tests__/static.test.ts +175 -0
  482. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  483. package/src/core/{websocket → transports}/connection-state.ts +9 -16
  484. package/src/core/transports/server.ts +152 -0
  485. package/src/core/transports/static.ts +84 -0
  486. package/src/core/transports/wasm.ts +97 -0
  487. package/src/core/wasm/__tests__/bridge.test.ts +213 -50
  488. package/src/core/wasm/__tests__/worker.test.ts +187 -0
  489. package/src/core/wasm/bridge.ts +75 -69
  490. package/src/core/wasm/rpc.ts +15 -3
  491. package/src/core/wasm/store.ts +3 -3
  492. package/src/core/wasm/worker/bootstrap.ts +55 -51
  493. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  494. package/src/core/wasm/worker/save-worker.ts +9 -10
  495. package/src/core/wasm/worker/types.ts +29 -18
  496. package/src/core/wasm/worker/worker.ts +88 -44
  497. package/src/css/globals.css +66 -22
  498. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  499. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  500. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  501. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  502. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  503. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  504. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  505. package/src/hooks/useAsyncData.ts +3 -3
  506. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  507. package/src/hooks/useDebounce.ts +6 -10
  508. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  509. package/src/hooks/useDelayElapsed.ts +14 -10
  510. package/src/hooks/useElapsedTime.ts +3 -3
  511. package/src/hooks/useEventListener.ts +1 -2
  512. package/src/hooks/useInputHistory.ts +4 -7
  513. package/src/hooks/useInternalStateWithSync.ts +6 -6
  514. package/src/hooks/useResizeObserver.ts +4 -7
  515. package/src/hooks/useScript.ts +67 -70
  516. package/src/hooks/useStateForKey.ts +93 -0
  517. package/src/mount.tsx +1 -5
  518. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  519. package/src/plugins/core/registerReactComponent.tsx +1 -1
  520. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  521. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  522. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  523. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  524. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  525. package/src/plugins/impl/FormPlugin.tsx +1 -1
  526. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  527. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  528. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  529. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
  530. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  531. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  532. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  533. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  534. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  535. package/src/plugins/impl/anywidget/model.ts +3 -1
  536. package/src/plugins/impl/anywidget/registry.ts +18 -14
  537. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  538. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  539. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  540. package/src/plugins/impl/chat/chat-ui.tsx +14 -11
  541. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  542. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  543. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  544. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  545. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  546. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  547. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  548. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  549. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  550. package/src/plugins/impl/data-frames/panel.tsx +43 -16
  551. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  552. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  553. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  554. package/src/plugins/impl/panel/utils.ts +17 -0
  555. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  556. package/src/plugins/impl/plotly/selection.ts +25 -19
  557. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  558. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  559. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  560. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  561. package/src/stories/button.stories.tsx +1 -0
  562. package/src/stories/cell.stories.tsx +1 -1
  563. package/src/stories/data-table.stories.tsx +31 -0
  564. package/src/stories/editor.stories.tsx +24 -17
  565. package/src/stories/form-wrapper.stories.tsx +2 -2
  566. package/src/stories/layout/vertical/one-column.stories.tsx +1 -1
  567. package/src/stories/runtime-story.tsx +23 -16
  568. package/src/stories/select.stories.tsx +2 -2
  569. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  570. package/src/utils/__tests__/arrays.test.ts +35 -0
  571. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  572. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
  573. package/src/utils/__tests__/numbers.test.ts +105 -1
  574. package/src/utils/__tests__/objects.test.ts +21 -0
  575. package/src/utils/arrays.ts +11 -2
  576. package/src/utils/blob.ts +1 -1
  577. package/src/utils/createReducer.ts +53 -26
  578. package/src/utils/maps.ts +0 -3
  579. package/src/utils/numbers.ts +79 -18
  580. package/src/utils/objects.ts +4 -9
  581. package/src/utils/once.ts +19 -19
  582. package/src/utils/tracer.ts +9 -5
  583. package/src/utils/zod-utils.ts +1 -3
  584. package/dist/any-language-editor-DZ_YTVus.js +0 -76
  585. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  586. package/dist/assets/worker-HPY9fbhz.js +0 -114
  587. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  588. package/dist/precisionRound-DMIVQrt2.js +0 -97
  589. package/src/core/attachments/state.ts +0 -21
  590. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  591. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  592. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  593. package/src/core/cells/document-changes.ts +0 -944
  594. package/src/core/host/AGENTS.md +0 -6
  595. package/src/core/host/__tests__/events.test.ts +0 -46
  596. package/src/core/host/__tests__/notebook-client.test.ts +0 -191
  597. package/src/core/host/__tests__/protocol.test.ts +0 -214
  598. package/src/core/host/connection.ts +0 -81
  599. package/src/core/host/events.ts +0 -60
  600. package/src/core/host/notebook-client.ts +0 -226
  601. package/src/core/host/protocol.ts +0 -1943
  602. package/src/core/state/__tests__/jotai.test.ts +0 -25
  603. package/src/core/wasm/worker/message-buffer.ts +0 -33
  604. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  605. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -140
  606. package/src/core/websocket/__tests__/useNotebookSession.hook.test.tsx +0 -883
  607. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  608. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  609. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  610. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  611. package/src/core/websocket/transports/basic.ts +0 -102
  612. package/src/core/websocket/transports/create.ts +0 -31
  613. package/src/core/websocket/transports/sse.ts +0 -308
  614. package/src/core/websocket/transports/transport.ts +0 -50
  615. package/src/core/websocket/transports/ws.ts +0 -98
  616. package/src/core/websocket/useNotebookClient.ts +0 -80
  617. /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
  618. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -1,6 +1,7 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
  /* oxlint-disable typescript/no-explicit-any */
3
3
 
4
+ import { createRPCRequestHandler } from "rpc-anywhere";
4
5
  import { toast } from "@/components/ui/use-toast";
5
6
  import { userConfigAtom } from "@/core/config/config";
6
7
  import { serializeBlob } from "@/utils/blob";
@@ -8,9 +9,12 @@ import { Deferred } from "@/utils/Deferred";
8
9
  import { throwNotImplemented } from "@/utils/functions";
9
10
  import { Logger } from "@/utils/Logger";
10
11
  import { reloadSafe } from "@/utils/reload-safe";
12
+ import { once } from "@/utils/once";
11
13
  import { generateUUID } from "@/utils/uuid";
12
14
  import { createModuleWorker } from "@/utils/worker";
13
15
  import { notebookIsRunningAtom } from "../cells/cells";
16
+ import { flushDocument } from "../cells/document";
17
+ import { createWasmNotebookBackend } from "../transports/wasm";
14
18
  import type { ClientCommand } from "../kernel/messages";
15
19
  import { getInitialAppMode } from "../mode";
16
20
  import { API } from "../network/api";
@@ -38,10 +42,9 @@ import type {
38
42
  } from "../network/types";
39
43
  import { filenameAtom } from "../saving/file-state";
40
44
  import { store } from "../state/jotai";
41
- import { BasicTransport } from "../websocket/transports/basic";
42
- import type { IConnectionTransport } from "../websocket/transports/transport";
45
+ import { activeNotebookReplicaAtom } from "../session/active-notebook";
43
46
  import { PyodideRouter } from "./router";
44
- import { getWorkerRPC } from "./rpc";
47
+ import { getWorkerRPC, type PersistNotebookRequest } from "./rpc";
45
48
  import { getWasmRuntimeConfig } from "./runtime-config";
46
49
  import { createShareableLink } from "./share";
47
50
  import { wasmInitStateAtom } from "./state";
@@ -68,11 +71,8 @@ export class PyodideBridge implements RunRequests, EditRequests {
68
71
 
69
72
  private rpc!: ReturnType<typeof getWorkerRPC<WorkerSchema>>;
70
73
  private saveRpc: SaveWorker | undefined;
71
- private pendingSessionSave: Promise<unknown> = Promise.resolve();
74
+ private notebookConsumers = new Map<string, (event: unknown) => void>();
72
75
  private interruptBuffer?: Uint8Array;
73
- private messageConsumer:
74
- | ((message: MessageEvent<string>) => void)
75
- | undefined;
76
76
 
77
77
  public initialized = new Deferred<void>();
78
78
 
@@ -82,7 +82,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
82
82
  return {
83
83
  readFile: throwNotImplemented,
84
84
  readNotebook: async () => (await notebookFileStore.readFile()) ?? "",
85
- saveNotebook: throwNotImplemented,
85
+ writeNotebook: throwNotImplemented,
86
86
  };
87
87
  }
88
88
 
@@ -97,7 +97,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
97
97
  },
98
98
  );
99
99
 
100
- const rpc = getWorkerRPC<SaveWorkerSchema>(saveWorker);
100
+ const rpc = getWorkerRPC<SaveWorkerSchema>(saveWorker, undefined, Infinity);
101
101
  rpc.send.bootstrap(runtimeConfig);
102
102
  return rpc.proxy.request;
103
103
  }
@@ -116,9 +116,28 @@ export class PyodideBridge implements RunRequests, EditRequests {
116
116
  });
117
117
 
118
118
  // Create the RPC
119
- this.rpc = getWorkerRPC<WorkerSchema>(worker);
119
+ this.rpc = getWorkerRPC<WorkerSchema>(
120
+ worker,
121
+ createRPCRequestHandler({
122
+ persistNotebook: async (snapshot: PersistNotebookRequest) => {
123
+ if (!this.saveRpc) {
124
+ throw new Error("Notebook storage is not initialized");
125
+ }
126
+ await this.saveRpc.writeNotebook(snapshot);
127
+ notebookFileStore.saveFile(snapshot.contents);
128
+ fallbackFileStore.saveFile(snapshot.contents);
129
+ return null;
130
+ },
131
+ }),
132
+ // Synchronous Python execution can block worker requests indefinitely.
133
+ Infinity,
134
+ );
120
135
  this.rpc.send.bootstrap(runtimeConfig);
121
136
 
137
+ this.rpc.addMessageListener("notebookEvent", ({ id, event }) => {
138
+ this.notebookConsumers.get(id)?.(JSON.parse(event));
139
+ });
140
+
122
141
  // Listeners
123
142
  this.rpc.addMessageListener("ready", () => {
124
143
  this.startSession();
@@ -154,9 +173,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
154
173
  store.set(wasmInitStateAtom, { kind: "error", message: error });
155
174
  this.initialized.reject(new Error(error));
156
175
  });
157
- this.rpc.addMessageListener("kernelMessage", ({ message }) => {
158
- this.messageConsumer?.(new MessageEvent("message", { data: message }));
159
- });
160
176
  }
161
177
 
162
178
  private async startSession() {
@@ -208,11 +224,31 @@ export class PyodideBridge implements RunRequests, EditRequests {
208
224
  }
209
225
  }
210
226
 
211
- public attachMessageConsumer(
212
- consumer: (message: MessageEvent<string>) => void,
213
- ) {
214
- this.messageConsumer = consumer;
215
- this.rpc.proxy.send.consumerReady({});
227
+ public async createNotebookBackend() {
228
+ await this.initialized.promise;
229
+ return createWasmNotebookBackend({
230
+ subscribe: async (cursor, emit, signal) => {
231
+ signal.throwIfAborted();
232
+ const id = generateUUID();
233
+ this.notebookConsumers.set(id, emit);
234
+ const close = once(() => {
235
+ signal.removeEventListener("abort", close);
236
+ this.notebookConsumers.delete(id);
237
+ void this.rpc.proxy.request
238
+ .notebookUnsubscribe(id)
239
+ .catch(Logger.error);
240
+ });
241
+ signal.addEventListener("abort", close, { once: true });
242
+ try {
243
+ await this.rpc.proxy.request.notebookSubscribe({ id, cursor });
244
+ return close;
245
+ } catch (error) {
246
+ close();
247
+ throw error;
248
+ }
249
+ },
250
+ call: (call) => this.rpc.proxy.request.notebookCall(call),
251
+ });
216
252
  }
217
253
 
218
254
  public sendRename: EditRequests["sendRename"] = async ({ filename }) => {
@@ -230,30 +266,23 @@ export class PyodideBridge implements RunRequests, EditRequests {
230
266
  return null;
231
267
  };
232
268
 
233
- public sendSave: EditRequests["sendSave"] = async (request) => {
234
- if (!this.saveRpc) {
235
- Logger.warn("Save RPC not initialized");
236
- return null;
269
+ private async saveHostNotebook() {
270
+ const replica = store.get(activeNotebookReplicaAtom);
271
+ const notebook = replica && store.get(replica.notebookAtom);
272
+ const backend = replica && store.get(replica.backendAtom);
273
+ if (!notebook || !backend) {
274
+ throw new Error("The notebook is not connected");
237
275
  }
238
-
239
- const durableSave = this.saveRpc.saveNotebook(request);
240
- // Generated exports read the session-owned app in the main worker.
241
- this.pendingSessionSave = this.pendingSessionSave
242
- .catch(() => undefined)
243
- .then(async () => {
244
- await durableSave;
245
- await this.rpc.proxy.request.saveNotebook(request);
246
- });
247
- void this.pendingSessionSave.catch((error) => {
248
- Logger.error(error);
249
- });
250
-
251
- await durableSave;
252
- const code = await this.readCode();
253
- if (code.contents) {
254
- notebookFileStore.saveFile(code.contents);
255
- fallbackFileStore.saveFile(code.contents);
276
+ const result = await backend.notebook(notebook.id).save();
277
+ if (!result.ok) {
278
+ throw new Error(result.problem.detail ?? result.problem.title);
256
279
  }
280
+ }
281
+
282
+ public sendSave: EditRequests["sendSave"] = async ({ layout }) => {
283
+ await flushDocument();
284
+ await this.rpc.proxy.request.setNotebookLayout(layout ?? null);
285
+ await this.saveHostNotebook();
257
286
  return null;
258
287
  };
259
288
 
@@ -316,14 +345,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
316
345
  return response as FormatResponse;
317
346
  };
318
347
 
319
- public sendDeleteCell: EditRequests["sendDeleteCell"] = async (request) => {
320
- await this.putControlRequest({
321
- type: "delete-cell",
322
- ...request,
323
- });
324
- return null;
325
- };
326
-
327
348
  public sendInstallMissingPackages: EditRequests["sendInstallMissingPackages"] =
328
349
  async (request) => {
329
350
  this.putControlRequest({
@@ -370,14 +391,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
370
391
  functionName: "save_app_config",
371
392
  payload: request,
372
393
  });
373
- return null;
374
- };
375
-
376
- public saveCellConfig: EditRequests["saveCellConfig"] = async (request) => {
377
- await this.putControlRequest({
378
- type: "update-cell-config",
379
- ...request,
380
- });
394
+ await this.saveHostNotebook();
381
395
  return null;
382
396
  };
383
397
 
@@ -409,7 +423,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
409
423
  return response as Snippets;
410
424
  };
411
425
 
412
- public openFile: EditRequests["openFile"] = async ({ path }) => {
426
+ public openFile: EditRequests["openFile"] = async () => {
413
427
  const url = createShareableLink({
414
428
  code: null,
415
429
  baseUrl: window.location.origin,
@@ -453,7 +467,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
453
467
  return null;
454
468
  };
455
469
 
456
- public sendInstantiate: RunRequests["sendInstantiate"] = async (request) => {
470
+ public sendInstantiate: RunRequests["sendInstantiate"] = async () => {
457
471
  return null;
458
472
  };
459
473
 
@@ -535,7 +549,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
535
549
  public exportAsHTML: EditRequests["exportAsHTML"] = async (
536
550
  request: ExportAsHTMLRequest,
537
551
  ) => {
538
- await this.pendingSessionSave;
552
+ await flushDocument();
539
553
  const response = await this.rpc.proxy.request.bridge({
540
554
  functionName: "export_html",
541
555
  payload: withDevAssetUrl(request),
@@ -546,7 +560,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
546
560
  public exportAsMarkdown: EditRequests["exportAsMarkdown"] = async (
547
561
  request: ExportAsMarkdownRequest,
548
562
  ) => {
549
- await this.pendingSessionSave;
563
+ await flushDocument();
550
564
  const response = await this.rpc.proxy.request.bridge({
551
565
  functionName: "export_markdown",
552
566
  payload: request,
@@ -557,7 +571,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
557
571
  public exportAsScript: EditRequests["exportAsScript"] = async (
558
572
  request: ExportAsScriptRequest,
559
573
  ) => {
560
- await this.pendingSessionSave;
574
+ await flushDocument();
561
575
  const response = await this.rpc.proxy.request.bridge({
562
576
  functionName: "export_script",
563
577
  payload: request,
@@ -628,8 +642,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
628
642
  return null;
629
643
  };
630
644
 
631
- public sendDocumentTransaction = () => Promise.resolve(null);
632
-
633
645
  public addPackage: EditRequests["addPackage"] = async (request) => {
634
646
  return this.rpc.proxy.request.addPackage(request);
635
647
  };
@@ -719,12 +731,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
719
731
  }
720
732
  }
721
733
 
722
- export function createPyodideConnection(): IConnectionTransport {
723
- return BasicTransport.withProducerCallback((callback) => {
724
- PyodideBridge.INSTANCE.attachMessageConsumer(callback);
725
- });
726
- }
727
-
728
734
  // Compose the worker name. The optional "::controller" suffix tells
729
735
  // getController.ts that the host page provides a custom /wasm/controller.js
730
736
  // and that the dynamic import should be attempted. Hosts opt in by setting
@@ -9,19 +9,31 @@ import { Logger } from "@/utils/Logger";
9
9
  import type { WasmRuntimeConfig } from "./runtime-config";
10
10
  import { TRANSPORT_ID } from "./worker/constants";
11
11
 
12
+ export interface PersistNotebookRequest {
13
+ filename: string;
14
+ contents: string;
15
+ }
16
+
12
17
  export type ParentSchema = RPCSchema<{
18
+ requests: {
19
+ persistNotebook: { params: PersistNotebookRequest; response: null };
20
+ };
13
21
  messages: {
14
- consumerReady: {};
15
22
  bootstrap: WasmRuntimeConfig;
16
23
  };
17
24
  }>;
18
25
 
19
- export function getWorkerRPC<WorkerSchema extends RPCSchema>(worker: Worker) {
26
+ export function getWorkerRPC<WorkerSchema extends RPCSchema>(
27
+ worker: Worker,
28
+ requestHandler?: RPCOptions<ParentSchema, WorkerSchema>["requestHandler"],
29
+ maxRequestTime = 20_000,
30
+ ) {
20
31
  return createRPC<ParentSchema, WorkerSchema>({
21
32
  transport: createWorkerTransport(worker, {
22
33
  transportId: TRANSPORT_ID,
23
34
  }),
24
- maxRequestTime: 20_000, // 20 seconds
35
+ requestHandler,
36
+ maxRequestTime,
25
37
  _debugHooks: {
26
38
  onSend: (message) => {
27
39
  Logger.debug("[rpc] Parent -> Worker", message);
@@ -41,7 +41,7 @@ const urlFileStore: FileStore = {
41
41
  };
42
42
 
43
43
  export const domElementFileStore: FileStore = {
44
- saveFile(contents: string) {
44
+ saveFile(_contents: string) {
45
45
  // Do nothing
46
46
  },
47
47
  readFile() {
@@ -54,7 +54,7 @@ export const domElementFileStore: FileStore = {
54
54
  };
55
55
 
56
56
  const remoteDefaultFileStore: FileStore = {
57
- saveFile(contents: string) {
57
+ saveFile(_contents: string) {
58
58
  // Do nothing
59
59
  },
60
60
  readFile() {
@@ -79,7 +79,7 @@ export const mountConfigFileStore: FileStore = {
79
79
  };
80
80
 
81
81
  const emptyFileStore: FileStore = {
82
- saveFile(contents: string) {
82
+ saveFile(_contents: string) {
83
83
  // Do nothing
84
84
  },
85
85
  readFile() {
@@ -1,15 +1,13 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
  import { loadPyodide, type PyodideInterface } from "pyodide";
3
3
  import type { PyCallable, PyProxy } from "pyodide/ffi";
4
- import type { UserConfig } from "@/core/config/config-schema";
5
- import type { NotificationPayload } from "@/core/kernel/messages";
6
- import type { JsonString } from "@/utils/json/base64";
7
4
  import { invariant } from "../../../utils/invariant";
5
+ import { once } from "../../../utils/once";
8
6
  import { Logger } from "../../../utils/Logger";
9
7
  import { WasmFileSystem } from "./fs";
10
8
  import { getMarimoWheel } from "./getMarimoWheel";
11
9
  import { t } from "./tracer";
12
- import type { SerializedBridge, WasmController } from "./types";
10
+ import type { SerializedBridge, WasmController, WasmSession } from "./types";
13
11
  import { shouldLoadDuckDBPackages } from "../utils";
14
12
 
15
13
  const MAKE_SNAPSHOT = false;
@@ -45,7 +43,7 @@ export class DefaultWasmController implements WasmController {
45
43
  protected pyodide: PyodideInterface | null = null;
46
44
  private packageLoadQueue = Promise.resolve();
47
45
  private sessionGeneration = 0;
48
- private activeSessionStops = new Set<PyCallable>();
46
+ private sessions = new Set<WasmSession>();
49
47
 
50
48
  public get requirePyodide() {
51
49
  invariant(this.pyodide, "Pyodide not loaded");
@@ -85,7 +83,7 @@ export class DefaultWasmController implements WasmController {
85
83
  packages: [
86
84
  "micropip",
87
85
  "msgspec",
88
- getMarimoWheel(opts.version),
86
+ getMarimoWheel(),
89
87
  "Markdown",
90
88
  "pymdown-extensions",
91
89
  "narwhals",
@@ -138,28 +136,15 @@ export class DefaultWasmController implements WasmController {
138
136
  });
139
137
  }
140
138
 
141
- public async startSession(opts: {
142
- queryParameters: Record<string, string | string[]>;
143
- code: string;
144
- filename: string | null;
145
- onMessage: (message: JsonString<NotificationPayload>) => void;
146
- userConfig: UserConfig;
147
- }): Promise<SerializedBridge> {
139
+ public async prepareSession(
140
+ opts: Parameters<WasmController["prepareSession"]>[0],
141
+ ): Promise<WasmSession> {
148
142
  const sessionGeneration = this.sessionGeneration;
149
- const { code, filename, onMessage, queryParameters, userConfig } = opts;
150
- // We pass down a messenger object to the code
151
- // This is used to have synchronous communication between the JS and Python code
152
- // Previously, we used a queue, but this would not properly flush the queue
153
- // during processing of a cell's code.
154
- //
155
- // This adds a messenger object to the global scope (import js; js.messenger.callback)
156
- self.messenger = {
157
- callback: onMessage,
158
- };
143
+ const { code, filename, queryParameters, userConfig } = opts;
159
144
  self.query_params = queryParameters;
160
145
  self.user_config = userConfig;
161
146
 
162
- const span = t.startSpan("startSession.runPython");
147
+ const span = t.startSpan("prepareSession.runPython");
163
148
  const nbFilename = filename || WasmFileSystem.NOTEBOOK_FILENAME;
164
149
  const sessionResources = this.requirePyodide.runPython(
165
150
  `
@@ -169,14 +154,12 @@ export class DefaultWasmController implements WasmController {
169
154
  import js
170
155
  from marimo._pyodide.bootstrap import create_session, instantiate
171
156
 
172
- assert js.messenger, "messenger is not defined"
173
157
  assert js.query_params, "query_params is not defined"
174
158
 
175
159
  def create_session_resources():
176
160
  session, bridge = create_session(
177
161
  filename="${nbFilename}",
178
162
  query_params=js.query_params.to_py(),
179
- message_callback=js.messenger.callback,
180
163
  user_config=js.user_config.to_py(),
181
164
  )
182
165
  session_task = None
@@ -198,6 +181,7 @@ export class DefaultWasmController implements WasmController {
198
181
  if task is not None:
199
182
  await asyncio.gather(task, return_exceptions=True)
200
183
  if bridge is not None:
184
+ bridge.close()
201
185
  bridge.session = None
202
186
  task = None
203
187
  kernel_task = None
@@ -235,45 +219,65 @@ export class DefaultWasmController implements WasmController {
235
219
  } finally {
236
220
  packagesProxy.destroy();
237
221
  }
238
- this.activeSessionStops.add(stopSession);
239
-
240
- // Fire and forget:
241
- // Load notebook dependencies and instantiate the session
242
- // We don't want to wait for this to finish,
243
- // so we can show the initial code immediately giving
244
- // a sense of responsiveness.
222
+ let stopped = false;
223
+ let stopping: Promise<void> | undefined;
224
+ const releaseInit = once(() => initSession.destroy());
225
+ const isStopped = () =>
226
+ stopped || sessionGeneration !== this.sessionGeneration;
227
+ // Load in parallel with subscription setup; only start() executes cells.
245
228
  const dependenciesReady = this.packageLoadQueue.then(() => {
246
- if (sessionGeneration !== this.sessionGeneration) {
229
+ if (isStopped()) {
247
230
  return;
248
231
  }
249
232
  return this.loadNotebookDeps(code, foundPackages);
250
233
  });
251
234
  this.packageLoadQueue = dependenciesReady.catch(() => undefined);
252
- void dependenciesReady
253
- .then(() => {
254
- if (sessionGeneration !== this.sessionGeneration) {
255
- return;
235
+ const session: WasmSession = {
236
+ bridge: bridgeProxy as unknown as SerializedBridge,
237
+ start: once(async () => {
238
+ try {
239
+ if (isStopped()) {
240
+ return;
241
+ }
242
+ await dependenciesReady;
243
+ if (!isStopped()) {
244
+ initSession(userConfig.runtime.auto_instantiate);
245
+ }
246
+ } catch (error) {
247
+ if (!isStopped()) {
248
+ throw error;
249
+ }
250
+ } finally {
251
+ releaseInit();
256
252
  }
257
- return initSession(userConfig.runtime.auto_instantiate);
258
- })
259
- .catch((error: unknown) => {
260
- Logger.error("Failed to load notebook dependencies", error);
261
- })
262
- .finally(() => initSession.destroy());
263
-
264
- return bridgeProxy as unknown as SerializedBridge;
253
+ }),
254
+ stop: () => {
255
+ stopped = true;
256
+ releaseInit();
257
+ stopping ??= (async () => {
258
+ await stopSession();
259
+ stopSession.destroy();
260
+ bridgeProxy.destroy();
261
+ this.sessions.delete(session);
262
+ })().catch((error: unknown) => {
263
+ stopping = undefined;
264
+ throw error;
265
+ });
266
+ return stopping;
267
+ },
268
+ };
269
+ this.sessions.add(session);
270
+ return session;
265
271
  }
266
272
 
267
273
  public async stopSession(): Promise<void> {
268
274
  this.sessionGeneration += 1;
269
- const stops = [...this.activeSessionStops];
275
+ const sessions = [...this.sessions];
270
276
  let hasFailure = false;
271
277
  let firstFailure: unknown;
272
- for (const stop of stops) {
278
+ for (const session of sessions) {
273
279
  try {
274
- await stop();
275
- this.activeSessionStops.delete(stop);
276
- stop.destroy();
280
+ await session.stop();
277
281
  } catch (error) {
278
282
  if (!hasFailure) {
279
283
  hasFailure = true;
@@ -1,6 +1,6 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
- export function getMarimoWheel(version: string) {
3
+ export function getMarimoWheel() {
4
4
  if (import.meta.env.DEV && import.meta.env.VITE_WASM_MARIMO_PREBUILT_WHEEL) {
5
5
  return "marimo-base";
6
6
  }
@@ -7,12 +7,11 @@ import {
7
7
  createWorkerParentTransport,
8
8
  type RPCSchema,
9
9
  } from "rpc-anywhere";
10
- import type { SaveNotebookRequest } from "@/core/network/types";
11
10
  import { decodeUtf8 } from "@/utils/strings";
12
11
  import { Deferred } from "../../../utils/Deferred";
13
12
  import { prettyError } from "../../../utils/errors";
14
13
  import { Logger } from "../../../utils/Logger";
15
- import type { ParentSchema } from "../rpc";
14
+ import type { ParentSchema, PersistNotebookRequest } from "../rpc";
16
15
  import type { WasmRuntimeConfig } from "../runtime-config";
17
16
  import { TRANSPORT_ID } from "./constants";
18
17
  import { WasmFileSystem } from "./fs";
@@ -70,14 +69,14 @@ const requestHandler = createRPCRequestHandler({
70
69
  await pyodideReadyPromise; // Make sure loading is done
71
70
  return WasmFileSystem.readNotebook(self.pyodide);
72
71
  },
73
- saveNotebook: async (opts: SaveNotebookRequest) => {
74
- await pyodideReadyPromise; // Make sure loading is done
75
- const saveFile = self.pyodide.runPython(`
76
- from marimo._pyodide.bootstrap import save_file
77
-
78
- save_file
79
- `);
80
- await saveFile(JSON.stringify(opts), WasmFileSystem.NOTEBOOK_FILENAME);
72
+ writeNotebook: async ({ filename, contents }: PersistNotebookRequest) => {
73
+ await pyodideReadyPromise;
74
+ const directory = filename.substring(0, filename.lastIndexOf("/"));
75
+ if (directory) {
76
+ self.pyodide.FS.mkdirTree(directory);
77
+ }
78
+ self.pyodide.FS.writeFile(filename, contents);
79
+ self.pyodide.FS.writeFile(WasmFileSystem.NOTEBOOK_FILENAME, contents);
81
80
  await WasmFileSystem.persistFilesToRemote(self.pyodide);
82
81
  },
83
82
  });
@@ -2,11 +2,7 @@
2
2
  import type { PyodideInterface } from "pyodide";
3
3
  import type { WasmRuntimeConfig } from "../runtime-config";
4
4
  import type { UserConfig } from "@/core/config/config-schema";
5
- import type {
6
- ClientCommand,
7
- NotificationPayload,
8
- } from "@/core/kernel/messages";
9
- import type { JsonString } from "@/utils/json/base64";
5
+ import type { ClientCommand } from "@/core/kernel/messages";
10
6
  import type {
11
7
  CodeCompletionRequest,
12
8
  CopyNotebookRequest,
@@ -34,7 +30,6 @@ import type {
34
30
  FormatCellsRequest,
35
31
  FormatResponse,
36
32
  SaveAppConfigurationRequest,
37
- SaveNotebookRequest,
38
33
  SaveUserConfigurationRequest,
39
34
  Snippets,
40
35
  } from "../../network/types";
@@ -58,19 +53,21 @@ export interface WasmController {
58
53
  code: string;
59
54
  filename: string | null;
60
55
  }): Promise<{ code: string; filename: string }>;
61
- /**
62
- * Start the session
63
- * @param opts.code - The code to start with
64
- * @param opts.fallbackCode - The code to fallback to
65
- * @param opts.filename - The filename to start with
66
- */
67
- startSession(opts: {
56
+ /** Prepare a session and load its dependencies without executing cells. */
57
+ prepareSession(opts: {
68
58
  queryParameters: Record<string, string | string[]>;
69
59
  code: string;
70
60
  filename: string | null;
71
61
  userConfig: UserConfig;
72
- onMessage: (message: JsonString<NotificationPayload>) => void;
73
- }): Promise<SerializedBridge>;
62
+ }): Promise<WasmSession>;
63
+ }
64
+
65
+ export interface WasmSession {
66
+ readonly bridge: SerializedBridge;
67
+ /** Start once dependencies are ready. Repeated calls share the same start. */
68
+ start(): Promise<void>;
69
+ /** Prevent a pending start and release the session and its bridge. */
70
+ stop(): Promise<void>;
74
71
  }
75
72
 
76
73
  export interface RawBridge {
@@ -81,7 +78,6 @@ export interface RawBridge {
81
78
  read_snippets(): Promise<Snippets>;
82
79
  get_environment_info(): Promise<EnvironmentInfo>;
83
80
  format(request: FormatCellsRequest): Promise<FormatResponse>;
84
- save(request: SaveNotebookRequest): Promise<string>;
85
81
  copy(request: CopyNotebookRequest): Promise<string>;
86
82
  save_app_config(request: SaveAppConfigurationRequest): Promise<string>;
87
83
  save_user_config(request: SaveUserConfigurationRequest): Promise<null>;
@@ -114,11 +110,26 @@ export type BridgePayload<T extends keyof RawBridge> = T extends keyof RawBridge
114
110
  : undefined;
115
111
 
116
112
  export type SerializedBridge = {
117
- [P in Exclude<keyof RawBridge, "save">]: RawBridge[P] extends (
113
+ [P in keyof RawBridge]: RawBridge[P] extends (
118
114
  payload: string,
119
115
  ) => Promise<unknown>
120
116
  ? (payload: string) => Promise<string>
121
117
  : RawBridge[P];
122
118
  } & {
123
- save(payload: string): Promise<void>;
119
+ notebook: {
120
+ readonly id: string;
121
+ subscribe(
122
+ id: string,
123
+ cursor: string | null,
124
+ emit: (event: string) => void,
125
+ ): Promise<void>;
126
+ unsubscribe(id: string): void;
127
+ call(call: string): Promise<string>;
128
+ transaction(request: string): Promise<void>;
129
+ set_layout(layout: string): Promise<void>;
130
+ set_persistence(
131
+ persist: (contents: string, filename: string) => Promise<void>,
132
+ ): void;
133
+ destroy(): void;
134
+ };
124
135
  };