@marimo-team/islands 0.25.3-dev23 → 0.25.3-dev230

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 (745) hide show
  1. package/dist/{ConnectedDataExplorerComponent-DfG3PRaJ.js → ConnectedDataExplorerComponent-BBihxHrr.js} +120 -116
  2. package/dist/{ErrorBoundary-h0Ea26dF.js → ErrorBoundary-DZwEkAT6.js} +16 -16
  3. package/dist/_baseSet-BfI5FjBM.js +328 -0
  4. package/dist/{abnfDiagram-VCTEODGH-B3w-XWX8.js → abnfDiagram-VCTEODGH-DBcJVxYP.js} +2 -2
  5. package/dist/any-language-editor-jGaR0EAo.js +75 -0
  6. package/dist/{architectureDiagram-5GKGNRK7-C18U9A6k.js → architectureDiagram-5GKGNRK7-BcHZXF4S.js} +1 -1
  7. package/dist/{arrays-CN1djC4k.js → arrays-DoQ-AILx.js} +15 -27
  8. package/dist/assets/__vite-browser-external-BKBpimif.js +1 -0
  9. package/dist/assets/worker-CgoMtD-_.js +113 -0
  10. package/dist/{blockDiagram-I7D4REHJ-Cg25UrxT.js → blockDiagram-I7D4REHJ-Dkhe6_PP.js} +1 -1
  11. package/dist/{button-CvjkwdSk.js → button-B6RjDKJx.js} +227 -217
  12. package/dist/{capabilities-CQ5IjhzL.js → capabilities-CkVzWJEX.js} +2 -2
  13. package/dist/{chat-ui-w8SVynkn.js → chat-ui-DljFsfWf.js} +1277 -1275
  14. package/dist/{check-BMKBBrdc.js → check-Z_vtI3OL.js} +1 -1
  15. package/dist/{chunk-IMKFNOWR-CrS5bzqW.js → chunk-IMKFNOWR-CTZlwBAA.js} +1 -1
  16. package/dist/{chunk-SHT3W25Y-StpvPvUG.js → chunk-SHT3W25Y-53g_RZHm.js} +1 -1
  17. package/dist/{chunk-SVP7TREG-CBXjbYZC.js → chunk-SVP7TREG-P6abNGV_.js} +1 -1
  18. package/dist/{chunk-TICWLB2K-Dp6bIb-q.js → chunk-TICWLB2K-BTJJqkyF.js} +1 -1
  19. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  20. package/dist/{classDiagram-ZZMXUADV-BFYuxV5g.js → classDiagram-ZZMXUADV-BUGqOrTm.js} +2 -2
  21. package/dist/{classDiagram-v2-VYDZK3BY-D_tjXVMl.js → classDiagram-v2-VYDZK3BY-CXpQvO-s.js} +2 -2
  22. package/dist/code-visibility-BOmgV0Fh.js +134 -0
  23. package/dist/{common-DNPxRLm_.js → common-DdHcuklD.js} +9837 -9658
  24. package/dist/config-DbjzjwxX.js +636 -0
  25. package/dist/{copy-icon-BgUO8jsi.js → copy-icon-CG57cxwb.js} +3 -3
  26. package/dist/createReducer-DePYJ0F3.js +63 -0
  27. package/dist/{cynefinDiagram-5FMLGOSQ-DSgYrHB2.js → cynefinDiagram-5FMLGOSQ-DOcnl8wW.js} +1 -1
  28. package/dist/{dagre-GXQ25YYZ-GYlruCBf.js → dagre-GXQ25YYZ-CuHR9QfG.js} +3 -3
  29. package/dist/{dagre-BGSfXNEb.js → dagre-mqiMAsFY.js} +531 -680
  30. package/dist/{data-grid-overlay-editor-B9QxYH-I.js → data-grid-overlay-editor-Dxdsrndd.js} +5 -5
  31. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  32. package/dist/{debounce-C3jyATpc.js → debounce-DrhhziJv.js} +2 -2
  33. package/dist/{diagram-S7CK7UJ4-DRu5LX5_.js → diagram-S7CK7UJ4-BPiG4kso.js} +1 -1
  34. package/dist/{diagram-UQ7AKVKN-DD7uiejP.js → diagram-UQ7AKVKN-_piBOvOC.js} +1 -1
  35. package/dist/{diagram-VX7I27RA-ByH9KAJJ.js → diagram-VX7I27RA-DCzVeJ8M.js} +1 -1
  36. package/dist/{diagram-Z3DM3KII-BsQHPRke.js → diagram-Z3DM3KII-D64CF1jQ.js} +1 -1
  37. package/dist/{dialog-Cc479ZEH.js → dialog-DsZ49mmU.js} +26 -26
  38. package/dist/{dist-Bvs24R_E.js → dist-11T5wq3_.js} +31 -30
  39. package/dist/{dist-F8Ep36rk.js → dist-17ugwGZc.js} +59 -58
  40. package/dist/{dist-BMQf5zIS.js → dist-7cRjf--y.js} +123 -122
  41. package/dist/{dist-diIx5qJW.js → dist-B8pkfkzc.js} +31 -30
  42. package/dist/{dist-CN_BeG2I.js → dist-B9UoM_wq.js} +58 -57
  43. package/dist/{dist-CLdpGQVr.js → dist-BGuMTs1i.js} +118 -117
  44. package/dist/{dist-BK4Oj99o.js → dist-BHW11fni.js} +18 -17
  45. package/dist/dist-BLY4m1Wi.js +850 -0
  46. package/dist/{dist-PXYI_yuP.js → dist-BNlcVoxs.js} +24 -23
  47. package/dist/{dist-D6d86eNJ.js → dist-BQYe5Bju.js} +47 -46
  48. package/dist/{dist-CkTTIdBe.js → dist-BX1mn8na.js} +337 -337
  49. package/dist/{dist-DiDibOYr.js → dist-Bh0PTKUW.js} +151 -150
  50. package/dist/{dist-CSJ0MDRw.js → dist-CKaXK92p.js} +56 -55
  51. package/dist/{dist-C8fAKSH1.js → dist-CLhwfhpJ.js} +53 -52
  52. package/dist/{dist-DoE4XJBC.js → dist-CYeqK4Xw.js} +9801 -10649
  53. package/dist/{dist-BLfKmR-l.js → dist-Cxlp3WH6.js} +16 -15
  54. package/dist/{dist-D-wwqlzJ.js → dist-DSJPGj42.js} +158 -157
  55. package/dist/{dist-BorMD-uX.js → dist-DU9ICsWZ.js} +75 -74
  56. package/dist/{dist-DBtSl3sf.js → dist-DWv0ALBi.js} +47 -46
  57. package/dist/{dist-OR1OnN-y.js → dist-DiBpWEsB.js} +31 -30
  58. package/dist/{dist-CrSP_H1M.js → dist-DiLFTlnl.js} +242 -242
  59. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  60. package/dist/{dist-CSbh1PNP.js → dist-cVDV5IuI.js} +23 -22
  61. package/dist/{dist-BuaRHxOc.js → dist-fRfaOoq8.js} +44 -43
  62. package/dist/{dist-Cg7gzw-a.js → dist-gTTI-9Px.js} +78 -77
  63. package/dist/dropdown-menu-BRiE7Vnp.js +6083 -0
  64. package/dist/{ebnfDiagram-PWID7BFC-CQRK5_eT.js → ebnfDiagram-PWID7BFC-BapQPML-.js} +2 -2
  65. package/dist/{erDiagram-RLTQ6QDP-BKtWrDVh.js → erDiagram-RLTQ6QDP-CNQvC2_Q.js} +1 -1
  66. package/dist/{error-banner-CFPHFw4h.js → error-banner-IsonnvA0.js} +84 -83
  67. package/dist/error-entries-CRPfxMqm.js +2505 -0
  68. package/dist/{esm-D2gTsXEV.js → esm-BgKjMvWw.js} +530 -529
  69. package/dist/{esm-AMFssoNp.js → esm-CHjVSexr.js} +328 -329
  70. package/dist/feedback-button-bkhTNCeV.js +374 -0
  71. package/dist/{flowDiagram-HODETNUW-1IN-ih2E.js → flowDiagram-HODETNUW-BBt6S6oe.js} +2 -2
  72. package/dist/{formats-X4mMQWzX.js → formats-BK4VefeF.js} +6 -6
  73. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  74. package/dist/{glide-data-editor-BIYjd9hQ.js → glide-data-editor-Cd5G4gkf.js} +417 -415
  75. package/dist/{glide-utils-DM9zBO4W.js → glide-utils-CQddGpa-.js} +16 -16
  76. package/dist/{globals-BX22lYk2.js → globals-Cm5hDd9R.js} +3 -3
  77. package/dist/{graphlib-BmY_CCV9.js → graphlib-nlwT3Wvy.js} +177 -220
  78. package/dist/hasIn-C20JsQ2m.js +28 -0
  79. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  80. package/dist/{error-entries-Damuuclv.js → html-to-image-0FooaKop.js} +26911 -26769
  81. package/dist/{infoDiagram-27XIBGKW-CkB-zr4p.js → infoDiagram-27XIBGKW-Hd8Ifz82.js} +1 -1
  82. package/dist/isArrayLikeObject-B-HXCsex.js +626 -0
  83. package/dist/isEmpty-CdX7MTE1.js +14 -0
  84. package/dist/{ishikawaDiagram-5VMMS53U-wlyLjBmM.js → ishikawaDiagram-5VMMS53U-CP7hnHWx.js} +1 -1
  85. package/dist/{kanban-definition-UXKFOSKX-C2usY0i-.js → kanban-definition-UXKFOSKX-CX91sa9b.js} +1 -1
  86. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  87. package/dist/{loader-BQOfjTIJ.js → loader-BisjOLiW.js} +12 -12
  88. package/dist/main.js +10253 -8126
  89. package/dist/{mermaid-BA-Z6wGL.js → mermaid-Djb3Ni9t.js} +54 -54
  90. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  91. package/dist/{mindmap-definition-YA3MSWOX-nKOWS8RR.js → mindmap-definition-YA3MSWOX-DgCNo07O.js} +1 -1
  92. package/dist/{now-jM8-M7IR.js → now-BKZCLE5P.js} +12 -13
  93. package/dist/{number-overlay-editor-DK-HbLsV.js → number-overlay-editor-BLCXOIUf.js} +3 -3
  94. package/dist/{dist-CN8zI-XC.js → objectWithoutPropertiesLoose-ai5NUqGa.js} +262 -240
  95. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  96. package/dist/{once-BtE-wKRL.js → once-NnpX_HnX.js} +1 -1
  97. package/dist/{pegDiagram-XKGWAZYB-Dac9EIKD.js → pegDiagram-XKGWAZYB-SGJhnIKo.js} +2 -2
  98. package/dist/pick-tCszMUbm.js +35 -0
  99. package/dist/{pieDiagram-E7YTZNPT-B7S-DsW_.js → pieDiagram-E7YTZNPT-1stXIWCQ.js} +1 -1
  100. package/dist/precisionRound-DhAzeRcm.js +49 -0
  101. package/dist/{process-output-CTPTtxXB.js → process-output-BOvkTEDt.js} +4 -4
  102. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  103. package/dist/{railroadDiagram-O6MQD6OU-Cfa5Q-hr.js → railroadDiagram-O6MQD6OU-DY1mojzB.js} +2 -2
  104. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  105. package/dist/{requirementDiagram-BXWQKSXE-d8zXCJIk.js → requirementDiagram-BXWQKSXE-gHzhUp0D.js} +1 -1
  106. package/dist/{reveal-component-Dl9ZugRb.js → reveal-component-BcLwP52P.js} +426 -403
  107. package/dist/schemas-CdscWNy_.js +4747 -0
  108. package/dist/select-BSxlITBo.js +1200 -0
  109. package/dist/{sequenceDiagram-WJ2MYXX4-b6pei3lJ.js → sequenceDiagram-WJ2MYXX4-rjPtD9fB.js} +1 -1
  110. package/dist/{spec-BJzgmvVa.js → spec-Dq8RNplp.js} +41 -11
  111. package/dist/{stateDiagram-D77RDMKH-Be-xSpaZ.js → stateDiagram-D77RDMKH-DMNxjUxV.js} +4 -4
  112. package/dist/{stateDiagram-v2-MP3YSRHH-5Y-FxesJ.js → stateDiagram-v2-MP3YSRHH-CumAaqGo.js} +2 -2
  113. package/dist/static-state-BldGd99T.js +72 -0
  114. package/dist/strings-BGW-fXVJ.js +507 -0
  115. package/dist/style.css +1 -1
  116. package/dist/{swimlanes-42K2YHIH-nRRl4NU0.js → swimlanes-42K2YHIH-BU14_Tej.js} +1 -1
  117. package/dist/{swimlanesDiagram-VR7AAH4N-BS79510V.js → swimlanesDiagram-VR7AAH4N-CieAnPqJ.js} +2 -2
  118. package/dist/{swiper-component-DKntyn8p.js → swiper-component-Dir1o8ml.js} +2 -2
  119. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  120. package/dist/{timeline-definition-24CTP7MA-Cfzf8WPf.js → timeline-definition-24CTP7MA-CrVj4E1K.js} +1 -1
  121. package/dist/toast-BWIPi3aF.js +491 -0
  122. package/dist/{tooltip-CtBZ5q8p.js → tooltip-C3UnvsRe.js} +329 -305
  123. package/dist/use-restore-focus-BWO0z1lo.js +1672 -0
  124. package/dist/{useAsyncData-BD534XmW.js → useAsyncData-DsZMvNXn.js} +15 -14
  125. package/dist/{useDateFormatter-DkajGwwd.js → useDateFormatter-ChB4eyim.js} +2 -2
  126. package/dist/{useDeepCompareMemoize-C72_RNsu.js → useDeepCompareMemoize-KMBBux8C.js} +1 -1
  127. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-cMNQ80CK.js} +1 -1
  128. package/dist/useLifecycle-BquH9ET-.js +112 -0
  129. package/dist/useTheme-DgOLVISn.js +337 -0
  130. package/dist/utils-DXkj82sK.js +813 -0
  131. package/dist/{vega-component-DfcYgj-Y.js → vega-component-Bwg_Kuh-.js} +171 -170
  132. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  133. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  134. package/dist/{vennDiagram-4TSXK5OY-CeY6Z1xZ.js → vennDiagram-4TSXK5OY-ZLxQ8yhm.js} +1 -1
  135. package/dist/{wardleyDiagram-VM6X3IG4-Cxs9NtVi.js → wardleyDiagram-VM6X3IG4-D88pcLY1.js} +1 -1
  136. package/dist/{xychartDiagram-S5SC5T6Z-DNWbvPK7.js → xychartDiagram-S5SC5T6Z-_13d4g8g.js} +2 -2
  137. package/package.json +16 -20
  138. package/src/__mocks__/document.tsx +63 -0
  139. package/src/__mocks__/notebook.ts +3 -4
  140. package/src/__mocks__/partial.ts +26 -0
  141. package/src/__mocks__/requests.ts +1 -4
  142. package/src/__tests__/main.test.tsx +24 -1
  143. package/src/__tests__/module-entry.test.ts +44 -0
  144. package/src/__tests__/mount.test.ts +1 -21
  145. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  146. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  147. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  148. package/src/components/app-config/ai-config.tsx +155 -149
  149. package/src/components/app-config/app-config-button.tsx +6 -5
  150. package/src/components/app-config/app-config-form.tsx +5 -9
  151. package/src/components/app-config/common.tsx +7 -9
  152. package/src/components/app-config/data-form.tsx +112 -113
  153. package/src/components/app-config/mcp-config.tsx +30 -53
  154. package/src/components/app-config/user-config-form.css +91 -0
  155. package/src/components/app-config/user-config-form.tsx +266 -349
  156. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  157. package/src/components/chat/__tests__/notebook-access-selector.test.tsx +64 -0
  158. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  159. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  160. package/src/components/chat/acp/blocks.tsx +3 -2
  161. package/src/components/chat/acp/session-tabs.tsx +8 -3
  162. package/src/components/chat/capabilities-popover.tsx +1 -5
  163. package/src/components/chat/chat-components.tsx +2 -0
  164. package/src/components/chat/chat-panel.tsx +23 -104
  165. package/src/components/chat/chat-utils.ts +4 -8
  166. package/src/components/chat/notebook-access-selector.stories.tsx +22 -0
  167. package/src/components/chat/notebook-access-selector.tsx +100 -0
  168. package/src/components/data-table/TableBottomBar.tsx +4 -1
  169. package/src/components/data-table/TableTopBar.tsx +2 -0
  170. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  171. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  172. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  173. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  174. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  175. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  176. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  177. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  178. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  179. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  180. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  181. package/src/components/data-table/cell-selection/feature.ts +2 -2
  182. package/src/components/data-table/cell-selection/types.ts +2 -2
  183. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  184. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  185. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  186. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  187. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  188. package/src/components/data-table/charts/charts.tsx +40 -30
  189. package/src/components/data-table/charts/storage.ts +16 -5
  190. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  191. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  192. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  193. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  194. package/src/components/data-table/columns.tsx +45 -46
  195. package/src/components/data-table/context-menu.tsx +36 -20
  196. package/src/components/data-table/data-table.tsx +17 -1
  197. package/src/components/data-table/date-popover.tsx +6 -4
  198. package/src/components/data-table/export-actions.tsx +76 -48
  199. package/src/components/data-table/pagination.tsx +44 -33
  200. package/src/components/data-table/range-focus/atoms.ts +4 -6
  201. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  202. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  203. package/src/components/data-table/renderers.tsx +16 -7
  204. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  205. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  206. package/src/components/data-table/search-highlight.tsx +79 -0
  207. package/src/components/data-table/types.ts +1 -0
  208. package/src/components/data-table/url-detector.tsx +49 -9
  209. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  210. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  211. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  212. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  213. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  214. package/src/components/datasources/column-preview.tsx +17 -12
  215. package/src/components/datasources/datasources.tsx +56 -43
  216. package/src/components/datasources/utils.ts +32 -23
  217. package/src/components/debugger/debugger-code.tsx +5 -0
  218. package/src/components/dependency-graph/custom-node.tsx +1 -1
  219. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  220. package/src/components/dependency-graph/elements.ts +24 -21
  221. package/src/components/dependency-graph/minimap-content.tsx +13 -19
  222. package/src/components/dependency-graph/panels.tsx +4 -2
  223. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  224. package/src/components/editor/SortableCell.tsx +1 -3
  225. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  226. package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
  227. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -14
  228. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  229. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -2
  230. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  231. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  232. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +40 -10
  233. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +6 -10
  234. package/src/components/editor/actions/__tests__/useCellActionButton.test.tsx +113 -0
  235. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  236. package/src/components/editor/actions/pair-with-agent-commands.ts +2 -2
  237. package/src/components/editor/actions/pair-with-agent-modal.tsx +24 -21
  238. package/src/components/editor/actions/pair-with-agent.tsx +15 -10
  239. package/src/components/editor/actions/useCellActionButton.tsx +36 -31
  240. package/src/components/editor/actions/useConfigActions.tsx +3 -4
  241. package/src/components/editor/actions/useNotebookActions.tsx +19 -23
  242. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  243. package/src/components/editor/actions/useSetCodeVisibility.ts +11 -12
  244. package/src/components/editor/actions/useSetOutputsExpanded.ts +14 -19
  245. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  246. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  247. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  248. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  249. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  250. package/src/components/editor/alerts/stdin-blocking-alert.tsx +1 -1
  251. package/src/components/editor/app-container.tsx +1 -1
  252. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  253. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  254. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  255. package/src/components/editor/cell/RunButton.tsx +2 -2
  256. package/src/components/editor/cell/StagedAICell.tsx +10 -2
  257. package/src/components/editor/cell/StopButton.tsx +2 -2
  258. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +4 -2
  259. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  260. package/src/components/editor/cell/cell-actions.tsx +12 -1
  261. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  262. package/src/components/editor/cell/code/cell-editor.tsx +46 -165
  263. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  264. package/src/components/editor/cell/collapse.tsx +11 -4
  265. package/src/components/editor/cell/useAddCell.ts +4 -8
  266. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  267. package/src/components/editor/cell/useRunCells.ts +2 -5
  268. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  269. package/src/components/editor/cell/useSplitCell.tsx +10 -6
  270. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  271. package/src/components/editor/chrome/components/feedback-button.tsx +2 -8
  272. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  273. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  274. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  275. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  276. package/src/components/editor/chrome/panels/dependency-graph-panel.tsx +1 -1
  277. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  278. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  279. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  280. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  281. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  282. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  283. package/src/components/editor/chrome/panels/session-panel.tsx +1 -1
  284. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  285. package/src/components/editor/chrome/panels/variable-panel.tsx +1 -5
  286. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +6 -9
  287. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  288. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +62 -3
  289. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
  290. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  291. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  292. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  293. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  294. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  295. package/src/components/editor/chrome/wrapper/footer.tsx +2 -2
  296. package/src/components/editor/chrome/wrapper/minimap-state.ts +1 -1
  297. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  298. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  299. package/src/components/editor/columns/__tests__/cell-column.test.tsx +1 -1
  300. package/src/components/editor/columns/cell-column.tsx +1 -1
  301. package/src/components/editor/columns/sortable-column.tsx +2 -2
  302. package/src/components/editor/connections/components.tsx +5 -10
  303. package/src/components/editor/controls/Controls.tsx +4 -4
  304. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  305. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  306. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  307. package/src/components/editor/controls/command-palette.tsx +22 -10
  308. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  309. package/src/components/editor/documentation.css +34 -11
  310. package/src/components/editor/dynamic-favicon.tsx +21 -40
  311. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +25 -20
  312. package/src/components/editor/errors/auto-fix.tsx +4 -8
  313. package/src/components/editor/errors/send-error-report-button.tsx +3 -6
  314. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  315. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  316. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  317. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  318. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  319. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  320. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  321. package/src/components/editor/header/__tests__/filename-form.test.tsx +1 -4
  322. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  323. package/src/components/editor/header/app-header.tsx +1 -1
  324. package/src/components/editor/header/filename-form.tsx +1 -1
  325. package/src/components/editor/header/status.tsx +40 -20
  326. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  327. package/src/components/editor/links/cell-link-list.tsx +1 -1
  328. package/src/components/editor/links/cell-link.tsx +17 -7
  329. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  330. package/src/components/editor/navigation/__tests__/navigation.test.ts +108 -85
  331. package/src/components/editor/navigation/__tests__/selection.test.ts +43 -21
  332. package/src/components/editor/navigation/clipboard.ts +17 -14
  333. package/src/components/editor/navigation/focus-utils.ts +23 -4
  334. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +17 -33
  335. package/src/components/editor/navigation/navigation.ts +71 -70
  336. package/src/components/editor/navigation/selection.ts +30 -29
  337. package/src/components/editor/notebook-banner.tsx +1 -0
  338. package/src/components/editor/notebook-cell.tsx +29 -32
  339. package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
  340. package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
  341. package/src/components/editor/output/Outputs.css +8 -0
  342. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  343. package/src/components/editor/output/ansi-reduce.ts +33 -20
  344. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  345. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  346. package/src/components/editor/package-alert.tsx +14 -7
  347. package/src/components/editor/renderers/cell-array.tsx +22 -33
  348. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  349. package/src/components/editor/renderers/slides-layout/__tests__/plugin.test.ts +0 -1
  350. package/src/components/editor/renderers/slides-layout/slides-layout.tsx +1 -1
  351. package/src/components/editor/renderers/vertical-layout/__tests__/useFocusFirstEditor.test.ts +4 -4
  352. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  353. package/src/components/editor/renderers/vertical-layout/useFocusFirstEditor.ts +1 -1
  354. package/src/components/editor/renderers/vertical-layout/vertical-layout.tsx +1 -1
  355. package/src/components/editor/stdin-blocking-alert.tsx +1 -1
  356. package/src/components/home/components.tsx +17 -31
  357. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  358. package/src/components/icons/large-spinner.tsx +3 -14
  359. package/src/components/icons/spinner.tsx +1 -1
  360. package/src/components/markdown/markdown-renderer.tsx +4 -8
  361. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  362. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  363. package/src/components/pages/edit-page.tsx +19 -17
  364. package/src/components/pages/home-page.tsx +164 -91
  365. package/src/components/scratchpad/scratchpad.test.tsx +89 -0
  366. package/src/components/scratchpad/scratchpad.tsx +43 -32
  367. package/src/components/scratchpad/state.ts +7 -0
  368. package/src/components/slides/__tests__/minimap-actions.test.tsx +22 -29
  369. package/src/components/slides/__tests__/minimap.test.ts +38 -19
  370. package/src/components/slides/__tests__/slide-scroll-hint.test.tsx +37 -0
  371. package/src/components/slides/minimap.tsx +17 -12
  372. package/src/components/slides/reveal-component.tsx +58 -26
  373. package/src/components/slides/reveal-slides.css +5 -4
  374. package/src/components/slides/slide-cell-view.tsx +4 -5
  375. package/src/components/slides/slide-form.tsx +1 -0
  376. package/src/components/slides/slide-scroll-hint.tsx +3 -1
  377. package/src/components/sort/SortableCellsProvider.tsx +8 -25
  378. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  379. package/src/components/tracing/tracing.tsx +46 -42
  380. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  381. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  382. package/src/components/ui/alert-dialog.tsx +1 -1
  383. package/src/components/ui/draggable-popover.tsx +24 -2
  384. package/src/components/ui/range-slider.tsx +22 -9
  385. package/src/core/MarimoApp.tsx +1 -1
  386. package/src/core/ai/__tests__/staged-cells.test.ts +32 -33
  387. package/src/core/ai/context/providers/__tests__/cell-output.test.ts +8 -12
  388. package/src/core/ai/context/providers/cell-output.ts +2 -2
  389. package/src/core/ai/context/providers/datasource.ts +6 -2
  390. package/src/core/ai/staged-cells.ts +18 -19
  391. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +40 -88
  392. package/src/core/ai/tools/__tests__/registry.test.ts +17 -0
  393. package/src/core/ai/tools/base.ts +18 -15
  394. package/src/core/ai/tools/edit-notebook-tool.ts +86 -80
  395. package/src/core/ai/tools/registry.ts +13 -10
  396. package/src/core/ai/tools/sample-tool.ts +6 -7
  397. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -37
  398. package/src/core/cells/__tests__/actions.test.ts +624 -35
  399. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  400. package/src/core/cells/__tests__/arrange.test.ts +432 -0
  401. package/src/core/cells/__tests__/cell.test.ts +39 -0
  402. package/src/core/cells/__tests__/cells.test.ts +251 -2540
  403. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  404. package/src/core/cells/__tests__/document.test.ts +164 -0
  405. package/src/core/cells/__tests__/editor.test.ts +57 -0
  406. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  407. package/src/core/cells/__tests__/harness.ts +106 -0
  408. package/src/core/cells/__tests__/layout.test.ts +177 -0
  409. package/src/core/cells/__tests__/notebook-outline.test.ts +122 -0
  410. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  411. package/src/core/cells/__tests__/session.test.ts +445 -211
  412. package/src/core/cells/__tests__/utils.test.ts +9 -108
  413. package/src/core/cells/actions.ts +675 -6
  414. package/src/core/cells/add-missing-import.ts +47 -43
  415. package/src/core/cells/arrange.ts +529 -0
  416. package/src/core/cells/cell.ts +14 -3
  417. package/src/core/cells/cells.ts +126 -1363
  418. package/src/core/cells/columns.ts +38 -0
  419. package/src/core/cells/document.ts +178 -0
  420. package/src/core/cells/edit.ts +30 -0
  421. package/src/core/cells/editor.ts +82 -0
  422. package/src/core/cells/layout.ts +221 -0
  423. package/src/core/cells/outputs.ts +12 -28
  424. package/src/core/cells/pending-delete-service.ts +3 -7
  425. package/src/core/cells/session.ts +76 -53
  426. package/src/core/cells/types.ts +22 -6
  427. package/src/core/cells/utils.ts +10 -8
  428. package/src/core/codemirror/__tests__/format.test.ts +0 -34
  429. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  430. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  431. package/src/core/codemirror/ai/resources.ts +1 -1
  432. package/src/core/codemirror/cells/__tests__/extensions.test.ts +2 -6
  433. package/src/core/codemirror/cells/extensions.ts +30 -28
  434. package/src/core/codemirror/cells/state.ts +1 -6
  435. package/src/core/codemirror/copilot/__tests__/getCodes.test.ts +6 -7
  436. package/src/core/codemirror/copilot/getCodes.ts +2 -2
  437. package/src/core/codemirror/format.ts +6 -23
  438. package/src/core/codemirror/keymaps/vim.ts +4 -8
  439. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  440. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  441. package/src/core/codemirror/language/cell-code.ts +114 -0
  442. package/src/core/codemirror/language/extension.ts +52 -35
  443. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  444. package/src/core/codemirror/language/languages/python.ts +16 -10
  445. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  446. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  447. package/src/core/codemirror/language/types.ts +7 -7
  448. package/src/core/codemirror/language/utils.ts +9 -39
  449. package/src/core/codemirror/lsp/__tests__/notebook-lsp.test.ts +1 -2
  450. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  451. package/src/core/codemirror/lsp/utils.ts +1 -1
  452. package/src/core/codemirror/misc/__tests__/paste.test.ts +3 -1
  453. package/src/core/codemirror/misc/paste.ts +1 -1
  454. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  455. package/src/core/codemirror/readonly/extension.ts +2 -2
  456. package/src/core/codemirror/utils.ts +1 -0
  457. package/src/core/config/__tests__/app-config.test.ts +64 -0
  458. package/src/core/config/__tests__/config-schema.test.ts +56 -6
  459. package/src/core/config/config-schema.ts +17 -4
  460. package/src/core/config/config.ts +65 -20
  461. package/src/core/config/feature-flag.tsx +0 -2
  462. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  463. package/src/core/datasets/data-source-discovery.ts +4 -6
  464. package/src/core/documentation/doc-lookup.ts +1 -1
  465. package/src/core/edit-app.tsx +21 -28
  466. package/src/core/errors/error-entries.ts +4 -1
  467. package/src/core/export/hooks.ts +1 -1
  468. package/src/core/export/layout.ts +2 -2
  469. package/src/core/hotkeys/actions.ts +2 -4
  470. package/src/core/islands/__tests__/bootstrap.test.ts +69 -17
  471. package/src/core/islands/__tests__/bridge.test.ts +85 -27
  472. package/src/core/islands/bootstrap.ts +46 -53
  473. package/src/core/islands/bridge.ts +84 -25
  474. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  475. package/src/core/islands/components/web-components.tsx +1 -1
  476. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  477. package/src/core/islands/worker/controller.ts +4 -10
  478. package/src/core/islands/worker/worker.tsx +33 -33
  479. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  480. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  481. package/src/core/kernel/handlers.ts +19 -72
  482. package/src/core/meta/__tests__/code-visibility.test.tsx +78 -70
  483. package/src/core/meta/code-visibility.ts +18 -9
  484. package/src/core/network/__tests__/connection.test.ts +1 -1
  485. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  486. package/src/core/network/connection-notice.ts +1 -1
  487. package/src/core/network/connection.ts +1 -1
  488. package/src/core/network/requests-lazy.ts +2 -5
  489. package/src/core/network/requests-network.ts +5 -31
  490. package/src/core/network/requests-static.ts +1 -4
  491. package/src/core/network/requests-toasting.tsx +1 -4
  492. package/src/core/network/types.ts +6 -14
  493. package/src/core/network/useConnectionNotice.ts +8 -2
  494. package/src/core/notebook/__tests__/operations.test.ts +51 -0
  495. package/src/core/notebook/attachments.ts +38 -0
  496. package/src/core/notebook/document/__tests__/codemirror.test.ts +758 -0
  497. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  498. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  499. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  500. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  501. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  502. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  503. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  504. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  505. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  506. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  507. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  508. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  509. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  510. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  511. package/src/core/notebook/document/__tests__/user.ts +61 -0
  512. package/src/core/notebook/document/codemirror.ts +216 -0
  513. package/src/core/notebook/document/draft.ts +502 -0
  514. package/src/core/notebook/document/hooks.tsx +118 -0
  515. package/src/core/notebook/document/index.ts +24 -0
  516. package/src/core/notebook/document/model.ts +102 -0
  517. package/src/core/notebook/document/select.ts +41 -0
  518. package/src/core/notebook/document/state.ts +378 -0
  519. package/src/core/notebook/document/sync.ts +369 -0
  520. package/src/core/notebook/index.ts +19 -0
  521. package/src/core/notebook/operations.ts +30 -0
  522. package/src/core/notebook/replica/AGENTS.md +11 -0
  523. package/src/core/{host → notebook/replica}/__tests__/attachments.test.ts +9 -10
  524. package/src/core/{host → notebook/replica}/__tests__/document.test.ts +18 -16
  525. package/src/core/notebook/replica/__tests__/executions.test.ts +201 -0
  526. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  527. package/src/core/notebook/replica/__tests__/replica.test.ts +559 -0
  528. package/src/core/notebook/replica/__tests__/results.test.ts +193 -0
  529. package/src/core/notebook/replica/__tests__/runtime.test.ts +78 -0
  530. package/src/core/notebook/replica/backend.ts +19 -0
  531. package/src/core/notebook/replica/connection.ts +42 -0
  532. package/src/core/notebook/replica/index.ts +20 -0
  533. package/src/core/notebook/replica/replica.ts +336 -0
  534. package/src/core/{host → notebook/replica}/resources/attachments.ts +6 -3
  535. package/src/core/{host → notebook/replica}/resources/document.ts +4 -4
  536. package/src/core/notebook/replica/resources/executions.ts +157 -0
  537. package/src/core/notebook/replica/resources/results.ts +48 -0
  538. package/src/core/notebook/replica/resources/runtime.ts +14 -0
  539. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  540. package/src/core/packages/useSandboxController.ts +1 -1
  541. package/src/core/run-app.tsx +48 -51
  542. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  543. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  544. package/src/core/runtime/__tests__/runtime.test.ts +32 -74
  545. package/src/core/runtime/adapter.ts +1 -1
  546. package/src/core/runtime/config.ts +2 -2
  547. package/src/core/runtime/runtime.ts +33 -46
  548. package/src/core/runtime/types.ts +0 -2
  549. package/src/core/saving/__tests__/document.test.tsx +123 -0
  550. package/src/core/saving/__tests__/state.test.ts +111 -0
  551. package/src/core/saving/__tests__/useAutoSave.test.tsx +67 -0
  552. package/src/core/saving/document.ts +33 -0
  553. package/src/core/saving/filename.ts +1 -1
  554. package/src/core/saving/save-component.tsx +39 -60
  555. package/src/core/saving/state.ts +38 -39
  556. package/src/core/saving/useAutoSave.ts +17 -53
  557. package/src/core/session/__tests__/backend.test.ts +236 -0
  558. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +184 -0
  559. package/src/core/{websocket → session}/__tests__/useNotebookSession.hook.test.tsx +213 -262
  560. package/src/core/session/active-notebook.ts +28 -0
  561. package/src/core/session/backend.ts +123 -0
  562. package/src/core/session/embedded.ts +7 -0
  563. package/src/core/{websocket → session}/types.ts +2 -2
  564. package/src/core/session/useNotebookReplica.ts +74 -0
  565. package/src/core/{websocket → session}/useNotebookSession.tsx +77 -64
  566. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  567. package/src/core/state/jotai.ts +1 -1
  568. package/src/core/static/__tests__/files.test.ts +38 -10
  569. package/src/core/static/files.ts +2 -0
  570. package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
  571. package/src/core/transports/__tests__/server.test.ts +408 -0
  572. package/src/core/transports/__tests__/static.test.ts +306 -0
  573. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  574. package/src/core/{websocket → transports}/connection-state.ts +9 -16
  575. package/src/core/transports/server.ts +152 -0
  576. package/src/core/transports/static.ts +166 -0
  577. package/src/core/transports/wasm.ts +97 -0
  578. package/src/core/wasm/__tests__/bridge.test.ts +105 -65
  579. package/src/core/wasm/__tests__/worker.test.ts +187 -0
  580. package/src/core/wasm/bridge.ts +69 -72
  581. package/src/core/wasm/rpc.ts +15 -3
  582. package/src/core/wasm/store.ts +3 -3
  583. package/src/core/wasm/worker/bootstrap.ts +55 -51
  584. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  585. package/src/core/wasm/worker/save-worker.ts +9 -10
  586. package/src/core/wasm/worker/types.ts +29 -18
  587. package/src/core/wasm/worker/worker.ts +89 -45
  588. package/src/css/app/codemirror.css +1 -1
  589. package/src/css/globals.css +66 -22
  590. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  591. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  592. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  593. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  594. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  595. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  596. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  597. package/src/hooks/useAsyncData.ts +3 -3
  598. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  599. package/src/hooks/useDebounce.ts +6 -10
  600. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  601. package/src/hooks/useDelayElapsed.ts +14 -10
  602. package/src/hooks/useElapsedTime.ts +3 -3
  603. package/src/hooks/useEventListener.ts +1 -2
  604. package/src/hooks/useInputHistory.ts +4 -7
  605. package/src/hooks/useInternalStateWithSync.ts +6 -6
  606. package/src/hooks/useResizeObserver.ts +4 -7
  607. package/src/hooks/useScript.ts +67 -70
  608. package/src/hooks/useStateForKey.ts +93 -0
  609. package/src/mount.tsx +20 -15
  610. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  611. package/src/plugins/core/registerReactComponent.tsx +1 -1
  612. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  613. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  614. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  615. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  616. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  617. package/src/plugins/impl/FormPlugin.tsx +1 -1
  618. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  619. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  620. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  621. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
  622. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  623. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  624. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  625. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  626. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  627. package/src/plugins/impl/anywidget/model.ts +3 -1
  628. package/src/plugins/impl/anywidget/registry.ts +18 -14
  629. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  630. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  631. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  632. package/src/plugins/impl/chat/chat-ui.tsx +14 -11
  633. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  634. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  635. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  636. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  637. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  638. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  639. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  640. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  641. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  642. package/src/plugins/impl/data-frames/panel.tsx +43 -16
  643. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  644. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  645. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  646. package/src/plugins/impl/panel/utils.ts +17 -0
  647. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  648. package/src/plugins/impl/plotly/selection.ts +25 -19
  649. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  650. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  651. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  652. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  653. package/src/stories/button.stories.tsx +1 -0
  654. package/src/stories/cell.stories.tsx +2 -5
  655. package/src/stories/data-table.stories.tsx +31 -0
  656. package/src/stories/editor.stories.tsx +24 -17
  657. package/src/stories/form-wrapper.stories.tsx +2 -2
  658. package/src/stories/layout/vertical/one-column.stories.tsx +28 -23
  659. package/src/stories/runtime-story.tsx +23 -17
  660. package/src/stories/select.stories.tsx +2 -2
  661. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  662. package/src/utils/__tests__/arrays.test.ts +35 -0
  663. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  664. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
  665. package/src/utils/__tests__/numbers.test.ts +105 -1
  666. package/src/utils/__tests__/objects.test.ts +21 -0
  667. package/src/utils/arrays.ts +11 -2
  668. package/src/utils/blob.ts +1 -1
  669. package/src/utils/createReducer.ts +53 -26
  670. package/src/utils/maps.ts +0 -3
  671. package/src/utils/numbers.ts +79 -18
  672. package/src/utils/objects.ts +4 -9
  673. package/src/utils/tracer.ts +9 -5
  674. package/src/utils/zod-utils.ts +1 -3
  675. package/dist/_baseMerge-BiGRwuqv.js +0 -165
  676. package/dist/any-language-editor-D4q-IFcM.js +0 -76
  677. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  678. package/dist/assets/worker-HPY9fbhz.js +0 -114
  679. package/dist/code-visibility-vqPDeo1_.js +0 -138
  680. package/dist/dropdown-menu-DHP9P4VE.js +0 -4678
  681. package/dist/feedback-button-BPcyKSZd.js +0 -379
  682. package/dist/hasIn-o2Us0uVx.js +0 -132
  683. package/dist/html-to-image-BdA59IUD.js +0 -3251
  684. package/dist/isArrayLikeObject-BZl-15ir.js +0 -450
  685. package/dist/isEmpty-BSZ1LaUo.js +0 -35
  686. package/dist/isSymbol-kQcfYbNX.js +0 -8
  687. package/dist/label-DoysmLMW.js +0 -281
  688. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  689. package/dist/objectWithoutPropertiesLoose-BRDcTsfB.js +0 -23
  690. package/dist/pick-CJeoxYOS.js +0 -51
  691. package/dist/precisionRound-DMIVQrt2.js +0 -97
  692. package/dist/select-DemoKT4G.js +0 -1344
  693. package/dist/toDate-B5mIgxwZ.js +0 -1185
  694. package/dist/toast-PkdZraM8.js +0 -1830
  695. package/dist/use-restore-focus-CKjfFjLM.js +0 -6369
  696. package/dist/useLifecycle-BGWzTwHK.js +0 -169
  697. package/dist/useTheme-BhQvec43.js +0 -904
  698. package/src/components/editor/cell/code/__tests__/rtc-editor-config.test.ts +0 -50
  699. package/src/components/editor/chrome/wrapper/footer-items/rtc-status.tsx +0 -65
  700. package/src/components/editor/navigation/__tests__/state.test.ts +0 -81
  701. package/src/components/editor/navigation/state.ts +0 -65
  702. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +0 -656
  703. package/src/core/ai/tools/run-cells-tool.ts +0 -303
  704. package/src/core/attachments/state.ts +0 -18
  705. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  706. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  707. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  708. package/src/core/cells/document-changes.ts +0 -944
  709. package/src/core/codemirror/rtc/extension.ts +0 -453
  710. package/src/core/codemirror/rtc/loro/__tests__/sync.test.ts +0 -52
  711. package/src/core/codemirror/rtc/loro/awareness.ts +0 -531
  712. package/src/core/codemirror/rtc/loro/colors.ts +0 -38
  713. package/src/core/codemirror/rtc/loro/sync.ts +0 -168
  714. package/src/core/host/AGENTS.md +0 -11
  715. package/src/core/host/__tests__/events.test.ts +0 -46
  716. package/src/core/host/__tests__/fixtures.ts +0 -39
  717. package/src/core/host/__tests__/notebook-client.test.ts +0 -433
  718. package/src/core/host/__tests__/protocol.test.ts +0 -214
  719. package/src/core/host/__tests__/stream.test.ts +0 -211
  720. package/src/core/host/connection.ts +0 -81
  721. package/src/core/host/events.ts +0 -61
  722. package/src/core/host/notebook-client.ts +0 -290
  723. package/src/core/host/protocol.ts +0 -1943
  724. package/src/core/host/resources/resource.ts +0 -59
  725. package/src/core/host/stream.ts +0 -213
  726. package/src/core/rtc/state.ts +0 -34
  727. package/src/core/state/__tests__/jotai.test.ts +0 -25
  728. package/src/core/wasm/worker/message-buffer.ts +0 -33
  729. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  730. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -170
  731. package/src/core/websocket/state.ts +0 -7
  732. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  733. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  734. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  735. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  736. package/src/core/websocket/transports/basic.ts +0 -102
  737. package/src/core/websocket/transports/create.ts +0 -31
  738. package/src/core/websocket/transports/sse.ts +0 -308
  739. package/src/core/websocket/transports/transport.ts +0 -50
  740. package/src/core/websocket/transports/ws.ts +0 -98
  741. package/src/core/websocket/useNotebookClient.ts +0 -88
  742. package/src/utils/__tests__/id-tree.test.ts +0 -2145
  743. package/src/utils/id-tree.tsx +0 -1134
  744. /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
  745. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -1,16 +1,21 @@
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
+ import { Arrays } from "@/utils/arrays";
7
9
  import { Deferred } from "@/utils/Deferred";
8
10
  import { throwNotImplemented } from "@/utils/functions";
9
11
  import { Logger } from "@/utils/Logger";
10
12
  import { reloadSafe } from "@/utils/reload-safe";
13
+ import { once } from "@/utils/once";
11
14
  import { generateUUID } from "@/utils/uuid";
12
15
  import { createModuleWorker } from "@/utils/worker";
13
16
  import { notebookIsRunningAtom } from "../cells/cells";
17
+ import { flushDocument } from "../cells/document";
18
+ import { createWasmNotebookBackend } from "../transports/wasm";
14
19
  import type { ClientCommand } from "../kernel/messages";
15
20
  import { getInitialAppMode } from "../mode";
16
21
  import { API } from "../network/api";
@@ -38,10 +43,8 @@ import type {
38
43
  } from "../network/types";
39
44
  import { filenameAtom } from "../saving/file-state";
40
45
  import { store } from "../state/jotai";
41
- import { BasicTransport } from "../websocket/transports/basic";
42
- import type { IConnectionTransport } from "../websocket/transports/transport";
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,8 @@ 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;
237
- }
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);
256
- }
269
+ public saveLayout: EditRequests["saveLayout"] = async ({ layout }) => {
270
+ await this.rpc.proxy.request.setNotebookLayout(layout ?? null);
257
271
  return null;
258
272
  };
259
273
 
@@ -282,7 +296,14 @@ export class PyodideBridge implements RunRequests, EditRequests {
282
296
 
283
297
  await this.putControlRequest({
284
298
  type: "execute-cells",
285
- ...request,
299
+ request: request.request,
300
+ executionRequests: Arrays.zip(request.cellIds, request.codes).map(
301
+ ([cellId, code]) => ({
302
+ type: "execute-cell",
303
+ cellId,
304
+ code,
305
+ }),
306
+ ),
286
307
  });
287
308
  return null;
288
309
  };
@@ -316,14 +337,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
316
337
  return response as FormatResponse;
317
338
  };
318
339
 
319
- public sendDeleteCell: EditRequests["sendDeleteCell"] = async (request) => {
320
- await this.putControlRequest({
321
- type: "delete-cell",
322
- ...request,
323
- });
324
- return null;
325
- };
326
-
327
340
  public sendInstallMissingPackages: EditRequests["sendInstallMissingPackages"] =
328
341
  async (request) => {
329
342
  this.putControlRequest({
@@ -373,14 +386,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
373
386
  return null;
374
387
  };
375
388
 
376
- public saveCellConfig: EditRequests["saveCellConfig"] = async (request) => {
377
- await this.putControlRequest({
378
- type: "update-cell-config",
379
- ...request,
380
- });
381
- return null;
382
- };
383
-
384
389
  public sendRestart = async (): Promise<null> => {
385
390
  // Save first
386
391
  const code = await this.readCode();
@@ -409,7 +414,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
409
414
  return response as Snippets;
410
415
  };
411
416
 
412
- public openFile: EditRequests["openFile"] = async ({ path }) => {
417
+ public openFile: EditRequests["openFile"] = async () => {
413
418
  const url = createShareableLink({
414
419
  code: null,
415
420
  baseUrl: window.location.origin,
@@ -453,7 +458,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
453
458
  return null;
454
459
  };
455
460
 
456
- public sendInstantiate: RunRequests["sendInstantiate"] = async (request) => {
461
+ public sendInstantiate: RunRequests["sendInstantiate"] = async () => {
457
462
  return null;
458
463
  };
459
464
 
@@ -535,7 +540,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
535
540
  public exportAsHTML: EditRequests["exportAsHTML"] = async (
536
541
  request: ExportAsHTMLRequest,
537
542
  ) => {
538
- await this.pendingSessionSave;
543
+ await flushDocument();
539
544
  const response = await this.rpc.proxy.request.bridge({
540
545
  functionName: "export_html",
541
546
  payload: withDevAssetUrl(request),
@@ -546,7 +551,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
546
551
  public exportAsMarkdown: EditRequests["exportAsMarkdown"] = async (
547
552
  request: ExportAsMarkdownRequest,
548
553
  ) => {
549
- await this.pendingSessionSave;
554
+ await flushDocument();
550
555
  const response = await this.rpc.proxy.request.bridge({
551
556
  functionName: "export_markdown",
552
557
  payload: request,
@@ -557,7 +562,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
557
562
  public exportAsScript: EditRequests["exportAsScript"] = async (
558
563
  request: ExportAsScriptRequest,
559
564
  ) => {
560
- await this.pendingSessionSave;
565
+ await flushDocument();
561
566
  const response = await this.rpc.proxy.request.bridge({
562
567
  functionName: "export_script",
563
568
  payload: request,
@@ -628,8 +633,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
628
633
  return null;
629
634
  };
630
635
 
631
- public sendDocumentTransaction = () => Promise.resolve(null);
632
-
633
636
  public addPackage: EditRequests["addPackage"] = async (request) => {
634
637
  return this.rpc.proxy.request.addPackage(request);
635
638
  };
@@ -719,12 +722,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
719
722
  }
720
723
  }
721
724
 
722
- export function createPyodideConnection(): IConnectionTransport {
723
- return BasicTransport.withProducerCallback((callback) => {
724
- PyodideBridge.INSTANCE.attachMessageConsumer(callback);
725
- });
726
- }
727
-
728
725
  // Compose the worker name. The optional "::controller" suffix tells
729
726
  // getController.ts that the host page provides a custom /wasm/controller.js
730
727
  // 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
  };