@marimo-team/islands 0.25.3-dev2 → 0.25.3-dev200

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 (690) hide show
  1. package/dist/{ConnectedDataExplorerComponent-CNSxdKyJ.js → ConnectedDataExplorerComponent-BsWiIPFk.js} +22 -20
  2. package/dist/{ErrorBoundary-DWEpirf3.js → ErrorBoundary-UWARECs-.js} +19 -19
  3. package/dist/{abnfDiagram-VCTEODGH-DQDjNXt3.js → abnfDiagram-VCTEODGH-I2x4Yu25.js} +2 -2
  4. package/dist/any-language-editor-BVFHDDQ3.js +75 -0
  5. package/dist/{architectureDiagram-5GKGNRK7-Nmls7Wxn.js → architectureDiagram-5GKGNRK7-1bSQMKzK.js} +1 -1
  6. package/dist/{arrays-CN1djC4k.js → arrays-DoQ-AILx.js} +15 -27
  7. package/dist/assets/__vite-browser-external-BCJsx8QS.js +1 -0
  8. package/dist/assets/worker-DodjWT_Z.js +113 -0
  9. package/dist/{button-CvjkwdSk.js → button-BQRdqQi1.js} +199 -192
  10. package/dist/{capabilities-CQ5IjhzL.js → capabilities-C2QUNRXi.js} +2 -2
  11. package/dist/{chat-ui-BDpymeoe.js → chat-ui-DXDoSuS4.js} +667 -667
  12. package/dist/{check-EDwSwP89.js → check-C9Mp9eCK.js} +1 -1
  13. package/dist/{chunk-IMKFNOWR-CoIFEo-1.js → chunk-IMKFNOWR-fsCGAJ9G.js} +1 -1
  14. package/dist/{chunk-SHT3W25Y-lHiGx3ZC.js → chunk-SHT3W25Y-D2STW4hj.js} +1 -1
  15. package/dist/{chunk-SVP7TREG-BvsHIrY_.js → chunk-SVP7TREG-CiY7cQG-.js} +1 -1
  16. package/dist/{chunk-TICWLB2K-BC8YIm_n.js → chunk-TICWLB2K-C_t5RBp9.js} +1 -1
  17. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  18. package/dist/{classDiagram-ZZMXUADV-Bt98lkkv.js → classDiagram-ZZMXUADV-B2JvZAP4.js} +2 -2
  19. package/dist/{classDiagram-v2-VYDZK3BY-D9bfbcC3.js → classDiagram-v2-VYDZK3BY-L8VsZG5A.js} +2 -2
  20. package/dist/{code-visibility-CLNfgCjd.js → code-visibility-33fxfX1p.js} +45 -49
  21. package/dist/{common-fyD_6mT2.js → common-D09oJJ8Z.js} +7354 -6884
  22. package/dist/{copy-icon-Ft7t9bqz.js → copy-icon-De584pZ4.js} +3 -3
  23. package/dist/{cynefinDiagram-5FMLGOSQ-CgWra7wt.js → cynefinDiagram-5FMLGOSQ-DdllC264.js} +1 -1
  24. package/dist/{dagre-GXQ25YYZ-CFZ8-35c.js → dagre-GXQ25YYZ-BaWHw4_f.js} +1 -1
  25. package/dist/{data-grid-overlay-editor-BmqASKep.js → data-grid-overlay-editor-lGsInXfR.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-lCNQrW7d.js → diagram-S7CK7UJ4-pFt-tUDe.js} +1 -1
  28. package/dist/{diagram-UQ7AKVKN-CQPQ6pP_.js → diagram-UQ7AKVKN-DqGJiBp4.js} +1 -1
  29. package/dist/{diagram-VX7I27RA-Cf3b0llz.js → diagram-VX7I27RA-Dr-yzMjf.js} +1 -1
  30. package/dist/{diagram-Z3DM3KII-Bt4QLrpn.js → diagram-Z3DM3KII-8xzJzv-w.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-NlnsmNA6.js} +1481 -1433
  58. package/dist/{ebnfDiagram-PWID7BFC-D3NAyPFY.js → ebnfDiagram-PWID7BFC-Br04Wy-E.js} +2 -2
  59. package/dist/{erDiagram-RLTQ6QDP-Tlu09_Nm.js → erDiagram-RLTQ6QDP-D5-XVgvv.js} +1 -1
  60. package/dist/{error-banner-Da600cwz.js → error-banner-DxJ3s8YO.js} +89 -84
  61. package/dist/{cells-C8iCYvLU.js → error-entries-C2Ux0Fjt.js} +7970 -8605
  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-B0CMhVv7.js +374 -0
  65. package/dist/{flowDiagram-HODETNUW-CAYbpJmV.js → flowDiagram-HODETNUW-CVl2kREF.js} +2 -2
  66. package/dist/{formats-BGh2thRc.js → formats-BCYthwyc.js} +3 -3
  67. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  68. package/dist/{glide-data-editor-BkjP14aq.js → glide-data-editor-CL535A5q.js} +370 -368
  69. package/dist/{glide-utils-DJyqh0EM.js → glide-utils-D98laidb.js} +16 -16
  70. package/dist/{globals-BN2O0F8N.js → globals-Bsr6T2yD.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-CZY1pIZH.js → html-to-image-CkH7vdBZ.js} +478 -381
  73. package/dist/{infoDiagram-27XIBGKW-XrDxhr0x.js → infoDiagram-27XIBGKW-BJ4vNcZE.js} +1 -1
  74. package/dist/{ishikawaDiagram-5VMMS53U-CD1muzov.js → ishikawaDiagram-5VMMS53U-B_Jy1Dsm.js} +1 -1
  75. package/dist/{kanban-definition-UXKFOSKX-Dg9W142_.js → kanban-definition-UXKFOSKX-C6MkS9vw.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 +9706 -7708
  80. package/dist/{mermaid-CiAXjLpA.js → mermaid-Bh918qrq.js} +52 -52
  81. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  82. package/dist/{mindmap-definition-YA3MSWOX-DgCidgYJ.js → mindmap-definition-YA3MSWOX-BP3ZTuir.js} +1 -1
  83. package/dist/{number-overlay-editor-BEBWJnbL.js → number-overlay-editor-CzbkG4E2.js} +3 -3
  84. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  85. package/dist/{once-BtE-wKRL.js → once-NnpX_HnX.js} +1 -1
  86. package/dist/{pegDiagram-XKGWAZYB-coOYYKB6.js → pegDiagram-XKGWAZYB-nNGOQtnt.js} +2 -2
  87. package/dist/{pieDiagram-E7YTZNPT-BnXTc3bj.js → pieDiagram-E7YTZNPT-CpRz5BZz.js} +1 -1
  88. package/dist/precisionRound-DhAzeRcm.js +49 -0
  89. package/dist/{process-output-e4jj9Ijt.js → process-output-OujAtFW4.js} +4 -4
  90. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  91. package/dist/{railroadDiagram-O6MQD6OU-B28-u83n.js → railroadDiagram-O6MQD6OU-BqOEN4I0.js} +2 -2
  92. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  93. package/dist/{requirementDiagram-BXWQKSXE-CdYCTMu0.js → requirementDiagram-BXWQKSXE-CEfg6vu9.js} +1 -1
  94. package/dist/{reveal-component-DK69bOki.js → reveal-component-CbEK0sNX.js} +733 -710
  95. package/dist/{select-BButFi-d.js → select-2KOiReNY.js} +536 -457
  96. package/dist/{sequenceDiagram-WJ2MYXX4-CM3aRAe_.js → sequenceDiagram-WJ2MYXX4-IOmpQcOi.js} +1 -1
  97. package/dist/{spec-BwarBpUJ.js → spec-Dkp64vVQ.js} +41 -11
  98. package/dist/{stateDiagram-D77RDMKH-DOZUMVj8.js → stateDiagram-D77RDMKH-tgU-EY9_.js} +2 -2
  99. package/dist/{stateDiagram-v2-MP3YSRHH-CISYxma6.js → stateDiagram-v2-MP3YSRHH-BofhG6q1.js} +2 -2
  100. package/dist/style.css +1 -1
  101. package/dist/{swimlanes-42K2YHIH-BkSppxp8.js → swimlanes-42K2YHIH-DW4vWDBd.js} +1 -1
  102. package/dist/{swimlanesDiagram-VR7AAH4N-c8G5pGxC.js → swimlanesDiagram-VR7AAH4N-BJcrqUJr.js} +2 -2
  103. package/dist/{swiper-component-DKntyn8p.js → swiper-component-BVfnhbnt.js} +2 -2
  104. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  105. package/dist/{timeline-definition-24CTP7MA-CaMQayqK.js → timeline-definition-24CTP7MA-D_QU3NGs.js} +1 -1
  106. package/dist/{toDate-BNJf01u1.js → toDate-BNDEXbfV.js} +77 -74
  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-ASDni9vH.js} +15 -14
  111. package/dist/{useDateFormatter-D8XruNGG.js → useDateFormatter-D3Tkfzej.js} +2 -2
  112. package/dist/{useDeepCompareMemoize-DXwzRINI.js → useDeepCompareMemoize-B1Ogyf53.js} +1 -1
  113. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-6a_yDNXp.js} +1 -1
  114. package/dist/{useLifecycle-j7D9uD1o.js → useLifecycle-Bfpa2m7k.js} +28 -25
  115. package/dist/{useTheme-CRnjkku0.js → useTheme-5TwyOh3B.js} +522 -314
  116. package/dist/{vega-component-DG29TAB7.js → vega-component-BHS9UOtC.js} +158 -158
  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-CF8pfaK3.js → vennDiagram-4TSXK5OY-YFYsNIAh.js} +1 -1
  120. package/dist/{wardleyDiagram-VM6X3IG4-dGPu39d-.js → wardleyDiagram-VM6X3IG4-i_n8mAWC.js} +1 -1
  121. package/dist/{xychartDiagram-S5SC5T6Z-CP6-9q56.js → xychartDiagram-S5SC5T6Z-BYGN95wW.js} +2 -2
  122. package/package.json +16 -16
  123. package/src/__mocks__/document.tsx +63 -0
  124. package/src/__mocks__/notebook.ts +3 -4
  125. package/src/__mocks__/partial.ts +26 -0
  126. package/src/__mocks__/requests.ts +0 -3
  127. package/src/__tests__/fixtures/{pair-preview.ts → pair-config.ts} +3 -2
  128. package/src/__tests__/main.test.tsx +24 -1
  129. package/src/__tests__/module-entry.test.ts +44 -0
  130. package/src/__tests__/mount.test.ts +9 -9
  131. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  132. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  133. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  134. package/src/components/app-config/ai-config.tsx +7 -2
  135. package/src/components/app-config/app-config-form.tsx +5 -9
  136. package/src/components/app-config/user-config-form.tsx +35 -0
  137. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  138. package/src/components/chat/__tests__/notebook-access-selector.test.tsx +64 -0
  139. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  140. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  141. package/src/components/chat/acp/blocks.tsx +3 -2
  142. package/src/components/chat/acp/session-tabs.tsx +8 -3
  143. package/src/components/chat/chat-components.tsx +2 -0
  144. package/src/components/chat/chat-panel.tsx +20 -101
  145. package/src/components/chat/chat-utils.ts +4 -8
  146. package/src/components/chat/notebook-access-selector.stories.tsx +22 -0
  147. package/src/components/chat/notebook-access-selector.tsx +100 -0
  148. package/src/components/data-table/TableBottomBar.tsx +4 -1
  149. package/src/components/data-table/TableTopBar.tsx +2 -0
  150. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  151. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  152. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  153. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  154. package/src/components/data-table/__tests__/context-menu.test.ts +176 -0
  155. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  156. package/src/components/data-table/__tests__/date-filter-inputs.test.tsx +24 -1
  157. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  158. package/src/components/data-table/__tests__/export-actions.test.tsx +434 -6
  159. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  160. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  161. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  162. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  163. package/src/components/data-table/cell-selection/feature.ts +2 -2
  164. package/src/components/data-table/cell-selection/types.ts +2 -2
  165. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  166. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  167. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  168. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  169. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  170. package/src/components/data-table/charts/charts.tsx +40 -30
  171. package/src/components/data-table/charts/storage.ts +16 -5
  172. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  173. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  174. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  175. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  176. package/src/components/data-table/columns.tsx +45 -46
  177. package/src/components/data-table/context-menu.tsx +161 -46
  178. package/src/components/data-table/data-table.tsx +17 -1
  179. package/src/components/data-table/date-filter-inputs.tsx +15 -42
  180. package/src/components/data-table/date-popover.tsx +6 -4
  181. package/src/components/data-table/export-actions.tsx +267 -163
  182. package/src/components/data-table/pagination.tsx +44 -33
  183. package/src/components/data-table/range-focus/atoms.ts +4 -6
  184. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  185. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  186. package/src/components/data-table/renderers.tsx +16 -7
  187. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  188. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  189. package/src/components/data-table/schemas.ts +14 -36
  190. package/src/components/data-table/search-highlight.tsx +79 -0
  191. package/src/components/data-table/types.ts +1 -0
  192. package/src/components/data-table/url-detector.tsx +49 -9
  193. package/src/components/data-table/utils.ts +18 -0
  194. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  195. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  196. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  197. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  198. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  199. package/src/components/datasources/column-preview.tsx +17 -12
  200. package/src/components/datasources/datasources.tsx +56 -43
  201. package/src/components/datasources/utils.ts +32 -23
  202. package/src/components/debugger/debugger-code.tsx +5 -0
  203. package/src/components/dependency-graph/custom-node.tsx +1 -1
  204. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  205. package/src/components/dependency-graph/elements.ts +24 -21
  206. package/src/components/dependency-graph/minimap-content.tsx +13 -19
  207. package/src/components/dependency-graph/panels.tsx +4 -2
  208. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  209. package/src/components/editor/Output.tsx +9 -2
  210. package/src/components/editor/SortableCell.tsx +1 -3
  211. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  212. package/src/components/editor/__tests__/agent-toast.test.tsx +58 -0
  213. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -14
  214. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  215. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -2
  216. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  217. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  218. package/src/components/editor/actions/__tests__/pair-with-agent-commands.test.ts +80 -252
  219. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +218 -40
  220. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +2 -6
  221. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  222. package/src/components/editor/actions/pair-with-agent-commands.ts +44 -53
  223. package/src/components/editor/actions/pair-with-agent-modal.tsx +101 -38
  224. package/src/components/editor/actions/useCellActionButton.tsx +26 -31
  225. package/src/components/editor/actions/useConfigActions.tsx +3 -4
  226. package/src/components/editor/actions/useNotebookActions.tsx +19 -23
  227. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  228. package/src/components/editor/actions/useSetCodeVisibility.ts +11 -12
  229. package/src/components/editor/actions/useSetOutputsExpanded.ts +14 -19
  230. package/src/components/editor/agent-toast.ts +15 -0
  231. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  232. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  233. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  234. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  235. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  236. package/src/components/editor/alerts/stdin-blocking-alert.tsx +1 -1
  237. package/src/components/editor/app-container.tsx +1 -1
  238. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  239. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  240. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  241. package/src/components/editor/cell/RunButton.tsx +2 -2
  242. package/src/components/editor/cell/StagedAICell.tsx +10 -2
  243. package/src/components/editor/cell/StopButton.tsx +2 -2
  244. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +3 -1
  245. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  246. package/src/components/editor/cell/cell-actions.tsx +12 -1
  247. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  248. package/src/components/editor/cell/code/cell-editor.tsx +36 -43
  249. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  250. package/src/components/editor/cell/collapse.tsx +11 -4
  251. package/src/components/editor/cell/useAddCell.ts +4 -8
  252. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  253. package/src/components/editor/cell/useRunCells.ts +2 -5
  254. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  255. package/src/components/editor/cell/useSplitCell.tsx +10 -6
  256. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  257. package/src/components/editor/chrome/components/feedback-button.tsx +2 -8
  258. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  259. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  260. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  261. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  262. package/src/components/editor/chrome/panels/dependency-graph-panel.tsx +1 -1
  263. package/src/components/editor/chrome/panels/error-panel.tsx +1 -0
  264. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  265. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  266. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  267. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  268. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  269. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  270. package/src/components/editor/chrome/panels/session-panel.tsx +1 -1
  271. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  272. package/src/components/editor/chrome/panels/variable-panel.tsx +1 -5
  273. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +363 -0
  274. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  275. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +61 -2
  276. package/src/components/editor/chrome/wrapper/footer-item.tsx +32 -24
  277. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +237 -0
  278. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  279. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  280. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  281. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  282. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  283. package/src/components/editor/chrome/wrapper/footer.tsx +4 -0
  284. package/src/components/editor/chrome/wrapper/minimap-state.ts +1 -1
  285. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  286. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  287. package/src/components/editor/columns/__tests__/cell-column.test.tsx +1 -1
  288. package/src/components/editor/columns/cell-column.tsx +1 -1
  289. package/src/components/editor/columns/sortable-column.tsx +2 -2
  290. package/src/components/editor/connections/components.tsx +5 -10
  291. package/src/components/editor/controls/Controls.tsx +4 -4
  292. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  293. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  294. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  295. package/src/components/editor/controls/command-palette.tsx +22 -10
  296. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  297. package/src/components/editor/documentation.css +34 -11
  298. package/src/components/editor/dynamic-favicon.tsx +21 -40
  299. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +536 -0
  300. package/src/components/editor/errors/auto-fix.tsx +4 -8
  301. package/src/components/editor/errors/send-error-report-button.tsx +66 -0
  302. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  303. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  304. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  305. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  306. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  307. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  308. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  309. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  310. package/src/components/editor/header/app-header.tsx +1 -1
  311. package/src/components/editor/header/status.tsx +40 -20
  312. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  313. package/src/components/editor/links/cell-link-list.tsx +1 -1
  314. package/src/components/editor/links/cell-link.tsx +17 -7
  315. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  316. package/src/components/editor/navigation/__tests__/navigation.test.ts +108 -85
  317. package/src/components/editor/navigation/__tests__/selection.test.ts +43 -21
  318. package/src/components/editor/navigation/clipboard.ts +17 -14
  319. package/src/components/editor/navigation/focus-utils.ts +23 -4
  320. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +17 -33
  321. package/src/components/editor/navigation/navigation.ts +71 -70
  322. package/src/components/editor/navigation/selection.ts +30 -29
  323. package/src/components/editor/notebook-banner.tsx +1 -0
  324. package/src/components/editor/notebook-cell.tsx +29 -31
  325. package/src/components/editor/output/MarimoErrorOutput.tsx +23 -6
  326. package/src/components/editor/output/MarimoTracebackOutput.tsx +54 -37
  327. package/src/components/editor/output/Outputs.css +8 -0
  328. package/src/components/editor/output/__tests__/ansi-reduce.test.ts +18 -0
  329. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  330. package/src/components/editor/output/ansi-reduce.ts +36 -17
  331. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  332. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  333. package/src/components/editor/package-alert.tsx +14 -7
  334. package/src/components/editor/renderers/cell-array.tsx +22 -33
  335. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  336. package/src/components/editor/renderers/slides-layout/__tests__/plugin.test.ts +0 -1
  337. package/src/components/editor/renderers/slides-layout/slides-layout.tsx +1 -1
  338. package/src/components/editor/renderers/vertical-layout/__tests__/useFocusFirstEditor.test.ts +4 -4
  339. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  340. package/src/components/editor/renderers/vertical-layout/useFocusFirstEditor.ts +1 -1
  341. package/src/components/editor/renderers/vertical-layout/vertical-layout.tsx +1 -1
  342. package/src/components/editor/stdin-blocking-alert.tsx +1 -1
  343. package/src/components/home/components.tsx +17 -31
  344. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  345. package/src/components/icons/large-spinner.tsx +3 -14
  346. package/src/components/icons/spinner.tsx +1 -1
  347. package/src/components/markdown/markdown-renderer.tsx +4 -8
  348. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  349. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  350. package/src/components/pages/edit-page.tsx +19 -17
  351. package/src/components/pages/home-page.tsx +164 -91
  352. package/src/components/scratchpad/scratchpad.test.tsx +97 -0
  353. package/src/components/scratchpad/scratchpad.tsx +20 -11
  354. package/src/components/slides/__tests__/minimap-actions.test.tsx +22 -29
  355. package/src/components/slides/__tests__/minimap.test.ts +38 -19
  356. package/src/components/slides/__tests__/slide-scroll-hint.test.tsx +37 -0
  357. package/src/components/slides/minimap.tsx +17 -12
  358. package/src/components/slides/reveal-component.tsx +58 -26
  359. package/src/components/slides/reveal-slides.css +5 -4
  360. package/src/components/slides/slide-cell-view.tsx +4 -5
  361. package/src/components/slides/slide-form.tsx +1 -0
  362. package/src/components/slides/slide-scroll-hint.tsx +3 -1
  363. package/src/components/sort/SortableCellsProvider.tsx +8 -25
  364. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  365. package/src/components/tracing/tracing.tsx +46 -42
  366. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  367. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  368. package/src/components/ui/alert-dialog.tsx +1 -1
  369. package/src/components/ui/draggable-popover.tsx +24 -2
  370. package/src/components/ui/range-slider.tsx +22 -9
  371. package/src/core/MarimoApp.tsx +1 -1
  372. package/src/core/ai/__tests__/staged-cells.test.ts +27 -26
  373. package/src/core/ai/context/providers/__tests__/cell-output.test.ts +8 -12
  374. package/src/core/ai/context/providers/cell-output.ts +2 -2
  375. package/src/core/ai/context/providers/datasource.ts +6 -2
  376. package/src/core/ai/staged-cells.ts +15 -18
  377. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +40 -88
  378. package/src/core/ai/tools/__tests__/registry.test.ts +17 -0
  379. package/src/core/ai/tools/base.ts +18 -15
  380. package/src/core/ai/tools/edit-notebook-tool.ts +86 -80
  381. package/src/core/ai/tools/registry.ts +13 -10
  382. package/src/core/ai/tools/sample-tool.ts +6 -7
  383. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -37
  384. package/src/core/cells/__tests__/actions.test.ts +624 -35
  385. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  386. package/src/core/cells/__tests__/arrange.test.ts +432 -0
  387. package/src/core/cells/__tests__/cell.test.ts +39 -0
  388. package/src/core/cells/__tests__/cells.test.ts +256 -2477
  389. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  390. package/src/core/cells/__tests__/document.test.ts +341 -0
  391. package/src/core/cells/__tests__/editor.test.ts +57 -0
  392. package/src/core/cells/__tests__/error-report.test.ts +139 -0
  393. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  394. package/src/core/cells/__tests__/harness.ts +106 -0
  395. package/src/core/cells/__tests__/layout.test.ts +177 -0
  396. package/src/core/cells/__tests__/notebook-outline.test.ts +120 -0
  397. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  398. package/src/core/cells/__tests__/session.test.ts +493 -210
  399. package/src/core/cells/__tests__/utils.test.ts +9 -108
  400. package/src/core/cells/__tests__/with-document-save.test.ts +67 -0
  401. package/src/core/cells/actions.ts +636 -6
  402. package/src/core/cells/add-missing-import.ts +47 -43
  403. package/src/core/cells/arrange.ts +529 -0
  404. package/src/core/cells/cell.ts +14 -3
  405. package/src/core/cells/cells.ts +120 -1337
  406. package/src/core/cells/columns.ts +38 -0
  407. package/src/core/cells/document.ts +289 -0
  408. package/src/core/cells/edit.ts +30 -0
  409. package/src/core/cells/editor.ts +82 -0
  410. package/src/core/cells/error-report.ts +63 -0
  411. package/src/core/cells/layout.ts +221 -0
  412. package/src/core/cells/outputs.ts +12 -28
  413. package/src/core/cells/pending-delete-service.ts +3 -7
  414. package/src/core/cells/session.ts +104 -54
  415. package/src/core/cells/types.ts +22 -6
  416. package/src/core/cells/utils.ts +10 -8
  417. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  418. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  419. package/src/core/codemirror/ai/resources.ts +1 -1
  420. package/src/core/codemirror/cells/__tests__/extensions.test.ts +42 -1
  421. package/src/core/codemirror/cells/extensions.ts +27 -9
  422. package/src/core/codemirror/cells/state.ts +1 -1
  423. package/src/core/codemirror/copilot/__tests__/getCodes.test.ts +6 -7
  424. package/src/core/codemirror/copilot/getCodes.ts +2 -2
  425. package/src/core/codemirror/format.ts +6 -5
  426. package/src/core/codemirror/keymaps/vim.ts +4 -8
  427. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  428. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  429. package/src/core/codemirror/language/cell-code.ts +114 -0
  430. package/src/core/codemirror/language/extension.ts +52 -35
  431. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  432. package/src/core/codemirror/language/languages/python.ts +16 -10
  433. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  434. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  435. package/src/core/codemirror/language/types.ts +7 -7
  436. package/src/core/codemirror/language/utils.ts +9 -39
  437. package/src/core/codemirror/lsp/__tests__/notebook-lsp.test.ts +1 -2
  438. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  439. package/src/core/codemirror/lsp/utils.ts +1 -1
  440. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  441. package/src/core/codemirror/readonly/extension.ts +2 -2
  442. package/src/core/codemirror/rtc/extension.ts +24 -13
  443. package/src/core/codemirror/rtc/loro/awareness.ts +34 -19
  444. package/src/core/codemirror/rtc/loro/sync.ts +1 -0
  445. package/src/core/codemirror/utils.ts +1 -0
  446. package/src/core/config/__tests__/app-config.test.ts +64 -0
  447. package/src/core/config/__tests__/config-schema.test.ts +13 -6
  448. package/src/core/config/config-schema.ts +13 -3
  449. package/src/core/config/config.ts +26 -12
  450. package/src/core/config/pair.ts +5 -3
  451. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  452. package/src/core/datasets/data-source-discovery.ts +4 -6
  453. package/src/core/documentation/doc-lookup.ts +1 -1
  454. package/src/core/edit-app.tsx +21 -28
  455. package/src/core/errors/error-entries.ts +4 -1
  456. package/src/core/export/hooks.ts +1 -1
  457. package/src/core/export/layout.ts +2 -2
  458. package/src/core/hotkeys/actions.ts +2 -4
  459. package/src/core/islands/__tests__/bootstrap.test.ts +69 -17
  460. package/src/core/islands/__tests__/bridge.test.ts +72 -25
  461. package/src/core/islands/bootstrap.ts +46 -53
  462. package/src/core/islands/bridge.ts +74 -23
  463. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  464. package/src/core/islands/components/web-components.tsx +1 -1
  465. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  466. package/src/core/islands/worker/controller.ts +4 -10
  467. package/src/core/islands/worker/worker.tsx +33 -33
  468. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  469. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  470. package/src/core/kernel/handlers.ts +19 -72
  471. package/src/core/meta/__tests__/code-visibility.test.tsx +78 -70
  472. package/src/core/meta/code-visibility.ts +18 -9
  473. package/src/core/network/__tests__/connection.test.ts +1 -1
  474. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  475. package/src/core/network/connection-notice.ts +1 -1
  476. package/src/core/network/connection.ts +5 -3
  477. package/src/core/network/requests-lazy.ts +1 -4
  478. package/src/core/network/requests-network.ts +0 -25
  479. package/src/core/network/requests-static.ts +0 -3
  480. package/src/core/network/requests-toasting.tsx +0 -3
  481. package/src/core/network/types.ts +0 -12
  482. package/src/core/network/useConnectionNotice.ts +8 -2
  483. package/src/core/notebook/__tests__/operations.test.ts +51 -0
  484. package/src/core/notebook/attachments.ts +38 -0
  485. package/src/core/notebook/document/__tests__/codemirror.test.ts +586 -0
  486. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  487. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  488. package/src/core/notebook/document/__tests__/hold.test.ts +88 -0
  489. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  490. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  491. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  492. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  493. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  494. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  495. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  496. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  497. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  498. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  499. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  500. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  501. package/src/core/notebook/document/__tests__/user.ts +61 -0
  502. package/src/core/notebook/document/codemirror.ts +216 -0
  503. package/src/core/notebook/document/draft.ts +502 -0
  504. package/src/core/notebook/document/hooks.tsx +118 -0
  505. package/src/core/notebook/document/index.ts +24 -0
  506. package/src/core/notebook/document/model.ts +105 -0
  507. package/src/core/notebook/document/select.ts +41 -0
  508. package/src/core/notebook/document/state.ts +378 -0
  509. package/src/core/notebook/document/sync.ts +418 -0
  510. package/src/core/notebook/index.ts +19 -0
  511. package/src/core/notebook/operations.ts +30 -0
  512. package/src/core/notebook/replica/AGENTS.md +11 -0
  513. package/src/core/notebook/replica/__tests__/attachments.test.ts +56 -0
  514. package/src/core/notebook/replica/__tests__/document.test.ts +126 -0
  515. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  516. package/src/core/notebook/replica/__tests__/replica.test.ts +550 -0
  517. package/src/core/notebook/replica/backend.ts +19 -0
  518. package/src/core/{websocket/types.ts → notebook/replica/connection.ts} +10 -15
  519. package/src/core/notebook/replica/index.ts +13 -0
  520. package/src/core/notebook/replica/replica.ts +286 -0
  521. package/src/core/notebook/replica/resources/attachments.ts +22 -0
  522. package/src/core/notebook/replica/resources/document.ts +101 -0
  523. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  524. package/src/core/packages/useSandboxController.ts +1 -1
  525. package/src/core/run-app.tsx +48 -51
  526. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  527. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  528. package/src/core/runtime/__tests__/runtime.test.ts +33 -15
  529. package/src/core/runtime/adapter.ts +1 -1
  530. package/src/core/runtime/config.ts +2 -2
  531. package/src/core/runtime/runtime.ts +39 -30
  532. package/src/core/saving/__tests__/state.test.ts +111 -0
  533. package/src/core/saving/__tests__/useAutoSave.test.tsx +67 -0
  534. package/src/core/saving/filename.ts +1 -1
  535. package/src/core/saving/save-component.tsx +28 -36
  536. package/src/core/saving/state.ts +38 -39
  537. package/src/core/saving/useAutoSave.ts +17 -53
  538. package/src/core/session/__tests__/backend.test.ts +236 -0
  539. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +184 -0
  540. package/src/core/{websocket/__tests__/useMarimoKernelConnection.hook.test.tsx → session/__tests__/useNotebookSession.hook.test.tsx} +252 -185
  541. package/src/core/session/active-notebook.ts +22 -0
  542. package/src/core/session/backend.ts +123 -0
  543. package/src/core/session/embedded.ts +7 -0
  544. package/src/core/session/types.ts +17 -0
  545. package/src/core/session/useNotebookReplica.ts +74 -0
  546. package/src/core/{websocket/useMarimoKernelConnection.tsx → session/useNotebookSession.tsx} +120 -273
  547. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  548. package/src/core/state/jotai.ts +1 -1
  549. package/src/core/static/__tests__/files.test.ts +38 -10
  550. package/src/core/static/files.ts +2 -0
  551. package/src/core/{websocket/__tests__/useMarimoKernelConnection.test.ts → transports/__tests__/connection-state.test.ts} +16 -30
  552. package/src/core/transports/__tests__/server.test.ts +408 -0
  553. package/src/core/transports/__tests__/static.test.ts +306 -0
  554. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  555. package/src/core/transports/connection-state.ts +63 -0
  556. package/src/core/transports/server.ts +152 -0
  557. package/src/core/transports/static.ts +166 -0
  558. package/src/core/transports/wasm.ts +97 -0
  559. package/src/core/wasm/__tests__/bridge.test.ts +214 -50
  560. package/src/core/wasm/__tests__/worker.test.ts +187 -0
  561. package/src/core/wasm/bridge.ts +75 -69
  562. package/src/core/wasm/rpc.ts +15 -3
  563. package/src/core/wasm/store.ts +3 -3
  564. package/src/core/wasm/worker/bootstrap.ts +55 -51
  565. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  566. package/src/core/wasm/worker/save-worker.ts +9 -10
  567. package/src/core/wasm/worker/types.ts +29 -18
  568. package/src/core/wasm/worker/worker.ts +88 -44
  569. package/src/css/globals.css +66 -22
  570. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  571. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  572. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  573. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  574. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  575. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  576. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  577. package/src/hooks/useAsyncData.ts +3 -3
  578. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  579. package/src/hooks/useDebounce.ts +6 -10
  580. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  581. package/src/hooks/useDelayElapsed.ts +14 -10
  582. package/src/hooks/useElapsedTime.ts +3 -3
  583. package/src/hooks/useEventListener.ts +1 -2
  584. package/src/hooks/useInputHistory.ts +4 -7
  585. package/src/hooks/useInternalStateWithSync.ts +6 -6
  586. package/src/hooks/useResizeObserver.ts +4 -7
  587. package/src/hooks/useScript.ts +67 -70
  588. package/src/hooks/useStateForKey.ts +93 -0
  589. package/src/mount.tsx +23 -14
  590. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  591. package/src/plugins/core/registerReactComponent.tsx +1 -1
  592. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  593. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  594. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  595. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  596. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  597. package/src/plugins/impl/FormPlugin.tsx +1 -1
  598. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  599. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  600. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  601. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +471 -1
  602. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  603. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  604. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  605. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  606. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  607. package/src/plugins/impl/anywidget/model.ts +3 -1
  608. package/src/plugins/impl/anywidget/registry.ts +18 -14
  609. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  610. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  611. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  612. package/src/plugins/impl/chat/chat-ui.tsx +14 -11
  613. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  614. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  615. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  616. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  617. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  618. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  619. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  620. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  621. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  622. package/src/plugins/impl/data-frames/panel.tsx +43 -16
  623. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  624. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  625. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  626. package/src/plugins/impl/panel/utils.ts +17 -0
  627. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  628. package/src/plugins/impl/plotly/selection.ts +25 -19
  629. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  630. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  631. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  632. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  633. package/src/stories/button.stories.tsx +1 -0
  634. package/src/stories/cell.stories.tsx +2 -5
  635. package/src/stories/data-table.stories.tsx +31 -0
  636. package/src/stories/editor.stories.tsx +24 -17
  637. package/src/stories/form-wrapper.stories.tsx +2 -2
  638. package/src/stories/layout/vertical/one-column.stories.tsx +2 -5
  639. package/src/stories/runtime-story.tsx +23 -17
  640. package/src/stories/select.stories.tsx +2 -2
  641. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  642. package/src/utils/__tests__/arrays.test.ts +35 -0
  643. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  644. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
  645. package/src/utils/__tests__/dates.test.ts +116 -0
  646. package/src/utils/__tests__/numbers.test.ts +105 -1
  647. package/src/utils/__tests__/objects.test.ts +21 -0
  648. package/src/utils/arrays.ts +11 -2
  649. package/src/utils/blob.ts +1 -1
  650. package/src/utils/createReducer.ts +53 -26
  651. package/src/utils/dates.ts +94 -0
  652. package/src/utils/maps.ts +0 -3
  653. package/src/utils/numbers.ts +79 -18
  654. package/src/utils/objects.ts +4 -9
  655. package/src/utils/once.ts +19 -19
  656. package/src/utils/tracer.ts +9 -5
  657. package/src/utils/zod-utils.ts +1 -3
  658. package/dist/any-language-editor-B9REGlD9.js +0 -76
  659. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  660. package/dist/assets/worker-HPY9fbhz.js +0 -114
  661. package/dist/error-entries-CzoYkIQZ.js +0 -56
  662. package/dist/feedback-button-fND8YT59.js +0 -380
  663. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  664. package/dist/precisionRound-DMIVQrt2.js +0 -97
  665. package/src/components/editor/navigation/__tests__/state.test.ts +0 -81
  666. package/src/components/editor/navigation/state.ts +0 -65
  667. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +0 -656
  668. package/src/core/ai/tools/run-cells-tool.ts +0 -303
  669. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  670. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  671. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  672. package/src/core/cells/document-changes.ts +0 -944
  673. package/src/core/host/AGENTS.md +0 -6
  674. package/src/core/host/__tests__/protocol.test.ts +0 -214
  675. package/src/core/host/protocol.ts +0 -1943
  676. package/src/core/state/__tests__/jotai.test.ts +0 -25
  677. package/src/core/wasm/worker/message-buffer.ts +0 -33
  678. package/src/core/websocket/__tests__/useWebSocket.test.ts +0 -49
  679. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  680. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -338
  681. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  682. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  683. package/src/core/websocket/transports/basic.ts +0 -102
  684. package/src/core/websocket/transports/sse.ts +0 -308
  685. package/src/core/websocket/transports/transport.ts +0 -61
  686. package/src/core/websocket/transports/ws.ts +0 -98
  687. package/src/core/websocket/useWebSocket.tsx +0 -88
  688. package/src/utils/__tests__/id-tree.test.ts +0 -2145
  689. package/src/utils/id-tree.tsx +0 -1134
  690. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -0,0 +1,79 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import NodeFs from "node:fs";
4
+ import NodeUrl from "node:url";
5
+ import type { NotebookTransport } from "@marimo-team/client";
6
+ import * as Protocol from "@marimo-team/client/protocol";
7
+ import { vi } from "vitest";
8
+ import type { NotebookBackend } from "../backend";
9
+
10
+ type Commands = ReturnType<NotebookBackend["notebook"]>;
11
+
12
+ /** A backend command, with the notebook it was sent to. */
13
+ export type FakeEdit = (
14
+ notebookId: Protocol.NotebookId,
15
+ ...args: Parameters<Commands["edit"]>
16
+ ) => ReturnType<Commands["edit"]>;
17
+
18
+ export type FakeSave = (
19
+ notebookId: Protocol.NotebookId,
20
+ ...args: Parameters<Commands["save"]>
21
+ ) => ReturnType<Commands["save"]>;
22
+
23
+ /** A backend over `transport` whose commands go to `edit` and `save`. */
24
+ export function fakeBackend({
25
+ transport = async () => undefined,
26
+ edit = vi.fn<FakeEdit>(),
27
+ save = vi.fn<FakeSave>(),
28
+ checkHealth,
29
+ }: {
30
+ transport?: NotebookTransport;
31
+ edit?: FakeEdit;
32
+ save?: FakeSave;
33
+ checkHealth?: () => Promise<boolean>;
34
+ } = {}): NotebookBackend {
35
+ return {
36
+ transport,
37
+ notebook: (notebookId) => ({
38
+ edit: (...args) => edit(notebookId, ...args),
39
+ save: (...args) => save(notebookId, ...args),
40
+ }),
41
+ checkHealth,
42
+ };
43
+ }
44
+
45
+ /** Read the protocol's portable examples as transport messages. */
46
+ export function fixtureMessages(name: string) {
47
+ const path = new NodeUrl.URL(
48
+ `../../../../../../protocols/host/fixtures/${name}`,
49
+ import.meta.url,
50
+ );
51
+ return NodeFs.readFileSync(path, "utf8")
52
+ .split(/\r?\n\r?\n/)
53
+ .flatMap((frame) => {
54
+ const event = /^event: (.+)$/m.exec(frame)?.[1];
55
+ if (!event) {
56
+ return [];
57
+ }
58
+ const data = frame
59
+ .split(/\r?\n/)
60
+ .filter((line) => line.startsWith("data:"))
61
+ .map((line) => line.slice(5).trimStart())
62
+ .join("\n");
63
+ const payload: unknown = JSON.parse(data);
64
+ return [
65
+ {
66
+ event,
67
+ id: /^id: (.+)$/m.exec(frame)?.[1],
68
+ data: payload,
69
+ },
70
+ ];
71
+ });
72
+ }
73
+
74
+ /** Read the protocol's portable examples as decoded protocol events. */
75
+ export function fixtureEvents(name: string): Protocol.EventOutput[] {
76
+ return fixtureMessages(name).map(({ id, ...message }) =>
77
+ Protocol.Event.parse(id === undefined ? message : { ...message, id }),
78
+ );
79
+ }
@@ -0,0 +1,550 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+ // @vitest-environment jsdom
3
+
4
+ import { type NotebookTransport, StreamStopped } from "@marimo-team/client";
5
+ import { createStore } from "jotai";
6
+ import { afterEach, describe, expect, it, onTestFinished, vi } from "vitest";
7
+ import { Deferred } from "@/utils/Deferred";
8
+ import { ConnectionState } from "../../../session/types";
9
+ import type { ConnectionStatus } from "../connection";
10
+ import { createNotebookReplica } from "../replica";
11
+ import { fakeBackend, fixtureEvents, fixtureMessages } from "./fixtures";
12
+
13
+ /** One connection the fake transport opened. */
14
+ interface Attempt {
15
+ readonly cursor: string | null;
16
+ readonly signal: AbortSignal;
17
+ open(): void;
18
+ receive(message: unknown): void;
19
+ /** End the connection so the stream reconnects. */
20
+ drop(): void;
21
+ /** End the stream, with the status the editor shows if the transport has one. */
22
+ stop(status?: ConnectionStatus): void;
23
+ }
24
+
25
+ function setup() {
26
+ const attempts: Attempt[] = [];
27
+ const transport: NotebookTransport = ({ cursor, signal }, handlers) =>
28
+ new Promise((resolve, reject) => {
29
+ attempts.push({
30
+ cursor,
31
+ signal,
32
+ open: handlers.open,
33
+ receive: handlers.message,
34
+ drop: resolve,
35
+ stop: (status) =>
36
+ reject(new StreamStopped("stopped", { cause: status })),
37
+ });
38
+ signal.addEventListener("abort", () => resolve(), { once: true });
39
+ });
40
+ const onNotification = vi.fn();
41
+ const checkHealth = vi.fn(async () => true);
42
+ const store = createStore();
43
+ const backend = fakeBackend({ transport, checkHealth });
44
+ const client = createNotebookReplica(store);
45
+ /** Follow the backend until `signal` aborts. */
46
+ const connect = (signal: AbortSignal) => {
47
+ const disconnect = client.connect(backend, {
48
+ onNotification: (notification) => onNotification({ data: notification }),
49
+ });
50
+ signal.addEventListener("abort", disconnect, { once: true });
51
+ };
52
+ let next = 0;
53
+ /** Wait for the next connection, and open it. */
54
+ const opened = async () => {
55
+ await vi.waitFor(() => expect(attempts.length).toBeGreaterThan(next));
56
+ const attempt = attempts[next++];
57
+ attempt.open();
58
+ return attempt;
59
+ };
60
+ return {
61
+ attempts,
62
+ backend,
63
+ client,
64
+ connect,
65
+ onNotification,
66
+ store,
67
+ checkHealth,
68
+ opened,
69
+ };
70
+ }
71
+
72
+ const lifetimes: AbortController[] = [];
73
+ function lifetime() {
74
+ const controller = new AbortController();
75
+ lifetimes.push(controller);
76
+ return controller;
77
+ }
78
+
79
+ afterEach(() => {
80
+ lifetimes.splice(0).forEach((controller) => controller.abort());
81
+ vi.useRealTimers();
82
+ });
83
+
84
+ /** Wait out the stream's backoff before it reconnects. */
85
+ const backoff = () => vi.advanceTimersByTimeAsync(2000);
86
+
87
+ const kernelReady = {
88
+ event: "marimo",
89
+ data: { marimo: { op: "kernel-ready", resumed: false } },
90
+ };
91
+
92
+ const notebook = {
93
+ id: "nb/with space",
94
+ project_id: "p1",
95
+ path: null,
96
+ name: "Notebook",
97
+ operations: ["notebook.edit"],
98
+ updated_at: null,
99
+ };
100
+
101
+ describe("NotebookReplica lifetime", () => {
102
+ it("ends the previous connection when it connects again", async () => {
103
+ const { attempts, backend, client, opened, store } = setup();
104
+ const first = client.connect(backend, { onNotification: vi.fn() });
105
+ const { receive } = await opened();
106
+ fixtureMessages("notebook-events.sse").forEach(receive);
107
+ const second = client.connect(backend, { onNotification: vi.fn() });
108
+ onTestFinished(second);
109
+ expect(attempts[0].signal.aborted).toBe(true);
110
+ expect(store.get(client.documentAtom)).toBeNull();
111
+ await opened();
112
+ first();
113
+ expect(attempts[1].signal.aborted).toBe(false);
114
+ expect(store.get(client.connectionAtom)).toEqual({
115
+ state: ConnectionState.OPEN,
116
+ });
117
+ });
118
+
119
+ it("exposes the backend it follows until that connection ends", async () => {
120
+ const { backend, client, opened, store } = setup();
121
+ expect(store.get(client.backendAtom)).toBeNull();
122
+ const disconnect = client.connect(backend, { onNotification: vi.fn() });
123
+ expect(store.get(client.backendAtom)).toBe(backend);
124
+ const { receive, drop } = await opened();
125
+ fixtureMessages("notebook-events.sse").forEach(receive);
126
+ receive({ event: "reset", data: {} });
127
+ drop();
128
+ expect(store.get(client.backendAtom)).toBe(backend);
129
+
130
+ const other = fakeBackend();
131
+ onTestFinished(client.connect(other, { onNotification: vi.fn() }));
132
+ expect(store.get(client.backendAtom)).toBe(other);
133
+ disconnect();
134
+ expect(store.get(client.backendAtom)).toBe(other);
135
+ });
136
+
137
+ it("exposes identity and capabilities only after the notebook is ready", async () => {
138
+ const { connect, opened, client, store } = setup();
139
+ connect(lifetime().signal);
140
+ const { receive } = await opened();
141
+ receive({ event: "notebook", data: { notebook } });
142
+ expect(store.get(client.notebookAtom)).toBeNull();
143
+ receive({ event: "ready", id: "1", data: {} });
144
+ expect(store.get(client.notebookAtom)).toEqual(notebook);
145
+ receive({
146
+ event: "notebook",
147
+ id: "2",
148
+ data: { notebook: { ...notebook, operations: [] } },
149
+ });
150
+ expect(store.get(client.notebookAtom)?.operations).toEqual([]);
151
+ receive({ event: "notebook.removed", id: "3", data: { id: notebook.id } });
152
+ expect(store.get(client.notebookAtom)).toBeNull();
153
+ });
154
+
155
+ it("rebuilds the replica after a reconnect reset selects a new notebook", async () => {
156
+ vi.useFakeTimers();
157
+ const { connect, opened, client, store } = setup();
158
+ connect(lifetime().signal);
159
+ const first = await opened();
160
+ fixtureMessages("notebook-events.sse").forEach(first.receive);
161
+ expect(store.get(client.documentAtom)?.revision.notebook_id).toBe("n1");
162
+
163
+ first.drop();
164
+ await backoff();
165
+ expect(store.get(client.documentAtom)?.revision.notebook_id).toBe("n1");
166
+ expect(store.get(client.notebookAtom)).toBeNull();
167
+ const second = await opened();
168
+ expect(second.cursor).toBe("97");
169
+ second.receive({ event: "reset", data: {} });
170
+ expect(store.get(client.documentAtom)).toBeNull();
171
+ for (const event of fixtureMessages("notebook-events.sse")) {
172
+ second.receive(
173
+ JSON.parse(JSON.stringify(event).replaceAll('"n1"', '"n2"')),
174
+ );
175
+ }
176
+ expect(store.get(client.documentAtom)?.revision.notebook_id).toBe("n2");
177
+ expect(store.get(client.connectionAtom).state).toBe(ConnectionState.OPEN);
178
+ expect(store.get(client.notebookAtom)?.id).toBe("n2");
179
+ });
180
+
181
+ it.each([
182
+ ["a message that is not an event", "{broken"],
183
+ ["an event from another notebook", "identity"],
184
+ ])("stops consuming a stream on %s", async (_name, message) => {
185
+ const { connect, opened, client, store, onNotification } = setup();
186
+ connect(lifetime().signal);
187
+ const attempt = await opened();
188
+ if (message === "identity") {
189
+ attempt.receive({ event: "notebook", data: { notebook } });
190
+ attempt.receive({
191
+ event: "notebook",
192
+ data: { notebook: { ...notebook, id: "n2" } },
193
+ });
194
+ } else {
195
+ attempt.receive(message);
196
+ }
197
+ attempt.receive({
198
+ event: "marimo",
199
+ data: { marimo: { op: "reconnected" } },
200
+ });
201
+ await vi.waitFor(() =>
202
+ expect(store.get(client.connectionAtom).state).toBe(
203
+ ConnectionState.CLOSED,
204
+ ),
205
+ );
206
+ expect(store.get(client.connectionAtom)).toMatchInlineSnapshot(`
207
+ {
208
+ "code": "KERNEL_DISCONNECTED",
209
+ "reason": "Failed to decode notebook message",
210
+ "state": "CLOSED",
211
+ }
212
+ `);
213
+ expect(attempt.signal.aborted).toBe(true);
214
+ expect(onNotification).not.toHaveBeenCalled();
215
+ });
216
+
217
+ it("exposes the confirmed document until the lifetime ends", async () => {
218
+ vi.useFakeTimers();
219
+ const { connect, opened, client, store } = setup();
220
+ const current = lifetime();
221
+ connect(current.signal);
222
+ const attempt = await opened();
223
+ const snapshot = fixtureMessages("notebook-events.sse");
224
+ snapshot.slice(0, -3).forEach(attempt.receive);
225
+ expect(store.get(client.documentAtom)).toBeNull();
226
+ snapshot.slice(-3).forEach(attempt.receive);
227
+ expect(store.get(client.documentAtom)?.revision.order).toEqual([
228
+ "c1",
229
+ "c3",
230
+ "k7f2",
231
+ ]);
232
+ attempt.drop();
233
+ await backoff();
234
+ expect(store.get(client.documentAtom)?.revision.order).toEqual([
235
+ "c1",
236
+ "c3",
237
+ "k7f2",
238
+ ]);
239
+ current.abort();
240
+ expect(store.get(client.documentAtom)).toBeNull();
241
+ });
242
+
243
+ it.each([
244
+ ["a malformed native envelope", { event: "marimo", id: "98", data: {} }],
245
+ ["null native event data", { event: "marimo", id: "98", data: null }],
246
+ ["an invalid protocol record", { event: "cell", id: "98", data: {} }],
247
+ ["an undecodable message", "{broken"],
248
+ ["a changed notebook", { event: "notebook", id: "98", data: { notebook } }],
249
+ ])(
250
+ "rebuilds from a fresh snapshot after rejecting %s",
251
+ async (_name, message) => {
252
+ const { connect, opened, client, store, onNotification } = setup();
253
+ connect(lifetime().signal);
254
+ const first = await opened();
255
+ const snapshot = fixtureMessages("notebook-events.sse");
256
+ snapshot.forEach(first.receive);
257
+ expect(store.get(client.documentAtom)).not.toBeNull();
258
+ onNotification.mockClear();
259
+
260
+ first.receive(message);
261
+ first.receive(kernelReady);
262
+ expect({
263
+ connection: store.get(client.connectionAtom),
264
+ document: store.get(client.documentAtom),
265
+ attachments: store.get(client.attachmentsAtom),
266
+ aborted: first.signal.aborted,
267
+ notifications: onNotification.mock.calls,
268
+ }).toEqual({
269
+ connection: {
270
+ state: ConnectionState.CLOSED,
271
+ code: "KERNEL_DISCONNECTED",
272
+ reason: "Failed to decode notebook message",
273
+ },
274
+ document: null,
275
+ attachments: [],
276
+ aborted: true,
277
+ notifications: [],
278
+ });
279
+
280
+ await client.reconnect();
281
+ const second = await opened();
282
+ expect(second.cursor).toBeNull();
283
+ snapshot.forEach(second.receive);
284
+ expect(store.get(client.documentAtom)?.revision.notebook_id).toBe("n1");
285
+ expect(store.get(client.connectionAtom)).toEqual({
286
+ state: ConnectionState.OPEN,
287
+ });
288
+ },
289
+ );
290
+
291
+ it.each([
292
+ [
293
+ "a final close",
294
+ {
295
+ state: ConnectionState.CLOSED,
296
+ code: "KERNEL_DISCONNECTED",
297
+ reason: "not authorized",
298
+ },
299
+ ],
300
+ ["the host giving up", undefined],
301
+ ] as const)(
302
+ "shows the stop after %s and resumes on an explicit reconnect",
303
+ async (_name, status) => {
304
+ const { connect, opened, attempts, client, store, checkHealth } = setup();
305
+ connect(lifetime().signal);
306
+ const first = await opened();
307
+ first.receive({ event: "ready", id: "5", data: {} });
308
+ first.stop(status);
309
+ await vi.waitFor(() =>
310
+ expect(store.get(client.connectionAtom)).toEqual(
311
+ status ?? {
312
+ state: ConnectionState.CLOSED,
313
+ code: "KERNEL_DISCONNECTED",
314
+ reason: "kernel not found",
315
+ },
316
+ ),
317
+ );
318
+ expect(attempts).toHaveLength(1);
319
+ await client.reconnect();
320
+ expect(checkHealth).toHaveBeenCalledOnce();
321
+ expect((await opened()).cursor).toBe("5");
322
+ },
323
+ );
324
+
325
+ it("does not revive an aborted lifetime when a manual health check finishes", async () => {
326
+ const { connect, opened, attempts, client, checkHealth } = setup();
327
+ const health = new Deferred<boolean>();
328
+ checkHealth.mockReturnValue(health.promise);
329
+ const current = lifetime();
330
+ connect(current.signal);
331
+ const first = await opened();
332
+ first.stop({
333
+ state: ConnectionState.CLOSED,
334
+ code: "KERNEL_DISCONNECTED",
335
+ reason: "",
336
+ });
337
+ await vi.waitFor(() => expect(checkHealth).not.toHaveBeenCalled());
338
+ const attempt = client.reconnect();
339
+ current.abort();
340
+ health.resolve(true);
341
+ await attempt;
342
+ expect(attempts).toHaveLength(1);
343
+ });
344
+
345
+ it("discards an interrupted snapshot before a manual reconnect", async () => {
346
+ const { connect, opened, client, store } = setup();
347
+ connect(lifetime().signal);
348
+ const first = await opened();
349
+ const snapshot = fixtureEvents("notebook-events.sse");
350
+ for (const event of snapshot) {
351
+ if (event.event === "ready") {
352
+ break;
353
+ }
354
+ first.receive(event);
355
+ }
356
+ first.stop();
357
+ await vi.waitFor(() =>
358
+ expect(store.get(client.connectionAtom).state).toBe(
359
+ ConnectionState.CLOSED,
360
+ ),
361
+ );
362
+
363
+ await client.reconnect();
364
+ const second = await opened();
365
+ expect(second.cursor).toBeNull();
366
+ for (const event of snapshot) {
367
+ if (event.event === "notebook") {
368
+ second.receive(event);
369
+ }
370
+ if (event.event === "document") {
371
+ second.receive({
372
+ ...event,
373
+ data: { document: { ...event.data.document, order: [] } },
374
+ });
375
+ }
376
+ }
377
+ second.receive({ event: "ready", id: "100", data: {} });
378
+ expect({
379
+ connection: store.get(client.connectionAtom),
380
+ order: store.get(client.documentAtom)?.revision.order,
381
+ attachments: store.get(client.attachmentsAtom),
382
+ }).toEqual({
383
+ connection: { state: ConnectionState.OPEN },
384
+ order: [],
385
+ attachments: [],
386
+ });
387
+ });
388
+
389
+ it("does not connect after abort", async () => {
390
+ const { connect, attempts, client, store } = setup();
391
+ const current = lifetime();
392
+ connect(current.signal);
393
+ current.abort();
394
+ await client.reconnect();
395
+ await new Promise((resolve) => setTimeout(resolve, 10));
396
+ expect(attempts).toHaveLength(0);
397
+ expect(store.get(client.connectionAtom)).toEqual({
398
+ state: ConnectionState.NOT_STARTED,
399
+ });
400
+ });
401
+
402
+ it("forwards notifications for the current lifetime only", async () => {
403
+ const { connect, opened, onNotification } = setup();
404
+ const first = lifetime();
405
+ connect(first.signal);
406
+ const earlier = await opened();
407
+ first.abort();
408
+ connect(lifetime().signal);
409
+ const current = await opened();
410
+ earlier.receive(kernelReady);
411
+ current.receive({
412
+ event: "marimo",
413
+ data: { marimo: { op: "reconnected" } },
414
+ });
415
+ expect(onNotification).toHaveBeenCalledExactlyOnceWith({
416
+ data: { op: "reconnected" },
417
+ });
418
+ });
419
+ });
420
+
421
+ describe("NotebookReplica attachments", () => {
422
+ const [browser, agent] = fixtureEvents("notebook-events.sse").flatMap(
423
+ (event) => (event.event === "attachment" ? [event.data.attachment] : []),
424
+ );
425
+
426
+ async function open() {
427
+ const result = setup();
428
+ result.connect(lifetime().signal);
429
+ return { ...result, attempt: await result.opened() };
430
+ }
431
+
432
+ it("publishes a complete snapshot and applies live changes independently of kernel readiness", async () => {
433
+ const { client, store, attempt } = await open();
434
+ const snapshot = fixtureMessages("notebook-events.sse");
435
+ const readyIndex = snapshot.findIndex((event) => event.event === "ready");
436
+ snapshot.slice(0, readyIndex).forEach(attempt.receive);
437
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
438
+ snapshot.slice(readyIndex).forEach(attempt.receive);
439
+ expect(store.get(client.attachmentsAtom)).toEqual([browser, agent]);
440
+ const renamed = { ...agent, name: "Renamed agent" };
441
+ attempt.receive({
442
+ event: "attachment",
443
+ id: "98",
444
+ data: { attachment: renamed },
445
+ });
446
+ attempt.receive({
447
+ event: "attachment.removed",
448
+ id: "99",
449
+ data: { id: browser.id },
450
+ });
451
+ expect(store.get(client.attachmentsAtom)).toEqual([renamed]);
452
+ attempt.receive(kernelReady);
453
+ expect(store.get(client.attachmentsAtom)).toEqual([renamed]);
454
+ attempt.receive({
455
+ event: "attachment.removed",
456
+ id: "100",
457
+ data: { id: agent.id },
458
+ });
459
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
460
+ });
461
+
462
+ it.each(["automatic", "manual"])(
463
+ "preserves an incomplete live document across %s reconnect and waits for replay ready",
464
+ async (recovery) => {
465
+ vi.useFakeTimers();
466
+ const { client, store, attempt, opened } = await open();
467
+ expect(attempt.cursor).toBeNull();
468
+ fixtureMessages("notebook-events.sse").forEach(attempt.receive);
469
+ const confirmed = store.get(client.documentAtom);
470
+ const edits = fixtureMessages("edits.sse");
471
+ edits.slice(0, 3).forEach(attempt.receive);
472
+ const renamed = { ...agent, name: "Renamed agent" };
473
+ attempt.receive({
474
+ event: "attachment",
475
+ id: "112a",
476
+ data: { attachment: renamed },
477
+ });
478
+ expect(store.get(client.attachmentsAtom)).toEqual([browser, renamed]);
479
+ if (recovery === "automatic") {
480
+ attempt.drop();
481
+ await backoff();
482
+ } else {
483
+ attempt.stop();
484
+ await vi.advanceTimersByTimeAsync(0);
485
+ await client.reconnect();
486
+ }
487
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
488
+ const replay = await opened();
489
+ expect(replay.cursor).toBe("112a");
490
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
491
+ expect(store.get(client.documentAtom)).toBe(confirmed);
492
+ edits.slice(3).forEach(replay.receive);
493
+ replay.receive({
494
+ event: "attachment.removed",
495
+ id: "115",
496
+ data: { id: browser.id },
497
+ });
498
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
499
+ replay.receive({ event: "ready", id: "116", data: {} });
500
+ expect(store.get(client.attachmentsAtom)).toEqual([renamed]);
501
+ const revisions = fixtureEvents("edits.sse").flatMap((event) =>
502
+ event.event === "document" ? [event.data.document] : [],
503
+ );
504
+ expect(store.get(client.documentAtom)?.revision).toEqual(
505
+ revisions.at(-1),
506
+ );
507
+ },
508
+ );
509
+
510
+ it.each(["drop", "stop", "removed", "abort"])(
511
+ "clears presence on %s and ignores messages after disposal",
512
+ async (reason) => {
513
+ vi.useFakeTimers();
514
+ const current = lifetime();
515
+ const result = setup();
516
+ result.connect(current.signal);
517
+ const { client, store } = result;
518
+ const attempt = await result.opened();
519
+ fixtureMessages("notebook-events.sse").forEach(attempt.receive);
520
+ expect(store.get(client.attachmentsAtom)).toEqual([browser, agent]);
521
+ switch (reason) {
522
+ case "drop":
523
+ attempt.drop();
524
+ await backoff();
525
+ break;
526
+ case "stop":
527
+ attempt.stop({
528
+ state: ConnectionState.CLOSED,
529
+ code: "KERNEL_DISCONNECTED",
530
+ reason: "not authorized",
531
+ });
532
+ await vi.advanceTimersByTimeAsync(0);
533
+ break;
534
+ case "removed":
535
+ attempt.receive({
536
+ event: "notebook.removed",
537
+ id: "98",
538
+ data: { id: "n1" },
539
+ });
540
+ break;
541
+ default:
542
+ current.abort();
543
+ }
544
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
545
+ current.abort();
546
+ fixtureMessages("notebook-events.sse").forEach(attempt.receive);
547
+ expect(store.get(client.attachmentsAtom)).toEqual([]);
548
+ },
549
+ );
550
+ });
@@ -0,0 +1,19 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ /** A notebook's events and commands, from a host or a stand-in for one. */
4
+
5
+ import type { NotebookClient, NotebookTransport } from "@marimo-team/client";
6
+ import type { NotebookId } from "@marimo-team/client/protocol";
7
+
8
+ /** Events and commands for a notebook, independent of React or a replica. */
9
+ export interface NotebookBackend {
10
+ readonly transport: NotebookTransport;
11
+ /**
12
+ * Commands for a notebook the transport reported.
13
+ *
14
+ * Method presence is not permission. The notebook advertises its operations.
15
+ */
16
+ notebook(id: NotebookId): Pick<NotebookClient, "edit" | "save">;
17
+ /** Whether the host is reachable before an explicit reconnect. Absent means always. */
18
+ checkHealth?(): Promise<boolean>;
19
+ }
@@ -1,11 +1,5 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
- import type { NotificationMessageData } from "../kernel/messages";
4
-
5
- export type ConnectionPhase =
6
- | NotificationMessageData<"startup-progress">["phase"]
7
- | "reconnecting";
8
-
9
3
  export const ConnectionState = {
10
4
  NOT_STARTED: "NOT_STARTED",
11
5
  CONNECTING: "CONNECTING",
@@ -17,27 +11,22 @@ export const ConnectionState = {
17
11
  export type ConnectionState =
18
12
  (typeof ConnectionState)[keyof typeof ConnectionState];
19
13
 
20
- export const WebSocketClosedReason = {
14
+ export const ConnectionClosedReason = {
21
15
  KERNEL_DISCONNECTED: "KERNEL_DISCONNECTED",
22
16
  KERNEL_STARTUP_ERROR: "KERNEL_STARTUP_ERROR",
23
17
  } as const;
24
18
 
25
- export type WebSocketClosedReason =
26
- (typeof WebSocketClosedReason)[keyof typeof WebSocketClosedReason];
19
+ export type ConnectionClosedReason =
20
+ (typeof ConnectionClosedReason)[keyof typeof ConnectionClosedReason];
27
21
 
28
22
  export type ConnectionStatus =
29
23
  | {
30
24
  state: typeof ConnectionState.CLOSED;
31
- code: WebSocketClosedReason;
32
- /**
33
- * Human-readable reason for closing the connection.
34
- */
25
+ code: ConnectionClosedReason;
35
26
  reason: string;
36
- phase?: ConnectionPhase;
37
27
  }
38
28
  | {
39
29
  state: typeof ConnectionState.CONNECTING;
40
- phase?: ConnectionPhase;
41
30
  }
42
31
  | {
43
32
  state:
@@ -45,3 +34,9 @@ export type ConnectionStatus =
45
34
  | typeof ConnectionState.CLOSING
46
35
  | typeof ConnectionState.NOT_STARTED;
47
36
  };
37
+
38
+ /** The recovery policy for a closed connection. */
39
+ export interface CloseDecision {
40
+ kind: "terminal" | "retry";
41
+ status: ConnectionStatus;
42
+ }