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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (618) hide show
  1. package/dist/{ConnectedDataExplorerComponent-CNMliVzw.js → ConnectedDataExplorerComponent-Le_OBfmt.js} +21 -19
  2. package/dist/{ErrorBoundary-DP07HKNC.js → ErrorBoundary-CcSbeila.js} +19 -19
  3. package/dist/{abnfDiagram-VCTEODGH-D-Ez91gV.js → abnfDiagram-VCTEODGH-Tf1ghyQQ.js} +2 -2
  4. package/dist/any-language-editor-BVFHDDQ3.js +75 -0
  5. package/dist/{architectureDiagram-5GKGNRK7-CTpcNtO8.js → architectureDiagram-5GKGNRK7-xlnbTG0T.js} +1 -1
  6. package/dist/{arrays-CN1djC4k.js → arrays-R6JsImiT.js} +8 -0
  7. package/dist/assets/__vite-browser-external-DVeFf2e7.js +1 -0
  8. package/dist/assets/worker-C1pfNBsx.js +113 -0
  9. package/dist/{button-CvjkwdSk.js → button-BQRdqQi1.js} +199 -192
  10. package/dist/{capabilities-CQ5IjhzL.js → capabilities-CI4xUPsp.js} +2 -2
  11. package/dist/{chat-ui-DaHynk8g.js → chat-ui-X-cWrL9u.js} +541 -540
  12. package/dist/{check-EDwSwP89.js → check-C9Mp9eCK.js} +1 -1
  13. package/dist/{chunk-IMKFNOWR-DGxi-zeo.js → chunk-IMKFNOWR-Cmd4a5jy.js} +1 -1
  14. package/dist/{chunk-SHT3W25Y-CV9J3rAB.js → chunk-SHT3W25Y-mSNp-WpO.js} +1 -1
  15. package/dist/{chunk-SVP7TREG-BL-9Fqgg.js → chunk-SVP7TREG-CWC49D8X.js} +1 -1
  16. package/dist/{chunk-TICWLB2K-Cu_XMjDM.js → chunk-TICWLB2K-De9FINtG.js} +1 -1
  17. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  18. package/dist/{classDiagram-ZZMXUADV-BXXV13Hq.js → classDiagram-ZZMXUADV-DweTkYdp.js} +2 -2
  19. package/dist/{classDiagram-v2-VYDZK3BY-B7FXJ1Cp.js → classDiagram-v2-VYDZK3BY-A6MR2TiE.js} +2 -2
  20. package/dist/{code-visibility-hTy956cB.js → code-visibility-BbBLg2nW.js} +25 -25
  21. package/dist/{common-CiockEHa.js → common-BuwBk7_9.js} +6994 -6819
  22. package/dist/{copy-icon-Ft7t9bqz.js → copy-icon-De584pZ4.js} +3 -3
  23. package/dist/{cynefinDiagram-5FMLGOSQ-CH96rwba.js → cynefinDiagram-5FMLGOSQ-BSvo5le-.js} +1 -1
  24. package/dist/{dagre-GXQ25YYZ-BD3PmpFY.js → dagre-GXQ25YYZ-tXRQslO6.js} +1 -1
  25. package/dist/{data-grid-overlay-editor-XsGuhhu5.js → data-grid-overlay-editor-Bt_qy-zy.js} +5 -5
  26. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  27. package/dist/{diagram-S7CK7UJ4-BsSJJldB.js → diagram-S7CK7UJ4-r3t5VtwI.js} +1 -1
  28. package/dist/{diagram-UQ7AKVKN-muQtNZc4.js → diagram-UQ7AKVKN-qgbC8YnT.js} +1 -1
  29. package/dist/{diagram-VX7I27RA-CdhuZAqR.js → diagram-VX7I27RA-CiHxEwtp.js} +1 -1
  30. package/dist/{diagram-Z3DM3KII-BD8mXjOj.js → diagram-Z3DM3KII-D95u15Ev.js} +1 -1
  31. package/dist/{dialog-CEBFZNyc.js → dialog-DVRfQuhD.js} +25 -25
  32. package/dist/{dist-F8Ep36rk.js → dist-B1M0bGO3.js} +1 -1
  33. package/dist/{dist-D6d86eNJ.js → dist-BA13RJj5.js} +1 -1
  34. package/dist/{dist-Bvs24R_E.js → dist-BJtmSgcE.js} +48 -48
  35. package/dist/{dist-CN_BeG2I.js → dist-Ba2VIn1K.js} +53 -53
  36. package/dist/{dist-PXYI_yuP.js → dist-Bj4bmwUm.js} +12 -12
  37. package/dist/{dist-DoE4XJBC.js → dist-BmS-ZGjl.js} +8626 -8626
  38. package/dist/{dist-BorMD-uX.js → dist-Bzogh7CW.js} +87 -87
  39. package/dist/{dist-CN8zI-XC.js → dist-CThPVKaP.js} +1385 -1382
  40. package/dist/{dist-DiDibOYr.js → dist-CTi5FTiV.js} +87 -87
  41. package/dist/{dist-diIx5qJW.js → dist-Ck_C32ye.js} +2 -2
  42. package/dist/{dist-BLfKmR-l.js → dist-D1fnFp1J.js} +1 -1
  43. package/dist/{dist-OR1OnN-y.js → dist-D3x9__Fu.js} +46 -46
  44. package/dist/{dist-BMQf5zIS.js → dist-D7mrh5sO.js} +36 -36
  45. package/dist/{dist-CSJ0MDRw.js → dist-D9Q3xjiP.js} +1 -1
  46. package/dist/{dist-BuaRHxOc.js → dist-DP60Ubh8.js} +2 -2
  47. package/dist/{dist-CSbh1PNP.js → dist-DTv9KkpW.js} +12 -12
  48. package/dist/{dist-D-wwqlzJ.js → dist-DVVwOfrD.js} +109 -109
  49. package/dist/{dist-C8fAKSH1.js → dist-DYXM73-k.js} +36 -36
  50. package/dist/{dist-DBtSl3sf.js → dist-DjeybPOb.js} +1 -1
  51. package/dist/{dist-CrSP_H1M.js → dist-DtGn9N4h.js} +197 -197
  52. package/dist/{dist-CLdpGQVr.js → dist-KN80qWCP.js} +2 -2
  53. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  54. package/dist/{dist-Cg7gzw-a.js → dist-O0qFpEJh.js} +56 -56
  55. package/dist/{dist-CkTTIdBe.js → dist-OUle2RT7.js} +207 -207
  56. package/dist/{dist-BK4Oj99o.js → dist-T5rh6JvL.js} +1 -1
  57. package/dist/{dropdown-menu-CP_qCYHC.js → dropdown-menu-DCMbbQ35.js} +1481 -1433
  58. package/dist/{ebnfDiagram-PWID7BFC-BIN1sweE.js → ebnfDiagram-PWID7BFC-P7e4bryK.js} +2 -2
  59. package/dist/{erDiagram-RLTQ6QDP-Ctp5cfoN.js → erDiagram-RLTQ6QDP-CUHZcmRj.js} +1 -1
  60. package/dist/{error-banner-D-WCIiMt.js → error-banner-DxJ3s8YO.js} +86 -86
  61. package/dist/{error-entries-Bv6bZPF3.js → error-entries-mpoD5t79.js} +4771 -4702
  62. package/dist/{esm-D2gTsXEV.js → esm-Bbw_Njjx.js} +575 -575
  63. package/dist/{esm-AMFssoNp.js → esm-mDQNEFe_.js} +354 -354
  64. package/dist/{feedback-button-CZNhgq7U.js → feedback-button-CuRYTzzX.js} +37 -37
  65. package/dist/{flowDiagram-HODETNUW-3EfLrlFb.js → flowDiagram-HODETNUW-BGZ95AzI.js} +2 -2
  66. package/dist/{formats-27CW_UP4.js → formats-DI7HEVv7.js} +3 -3
  67. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  68. package/dist/{glide-data-editor-DNPChdXf.js → glide-data-editor-vvFmzxva.js} +370 -368
  69. package/dist/{glide-utils-DJyqh0EM.js → glide-utils-D98laidb.js} +16 -16
  70. package/dist/{globals-DcKi3Zmn.js → globals-CIofsJK9.js} +3 -3
  71. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  72. package/dist/{html-to-image-Bf8QMGi4.js → html-to-image-BoZthB4P.js} +668 -622
  73. package/dist/{infoDiagram-27XIBGKW-DipQoHxl.js → infoDiagram-27XIBGKW-BF0YX70D.js} +1 -1
  74. package/dist/{ishikawaDiagram-5VMMS53U-DgFMjSGz.js → ishikawaDiagram-5VMMS53U-_itsAAn-.js} +1 -1
  75. package/dist/{kanban-definition-UXKFOSKX-w1ACVjgl.js → kanban-definition-UXKFOSKX-VNW2Beqh.js} +1 -1
  76. package/dist/{label-BDKF3Afc.js → label-Qvw1E97W.js} +3 -3
  77. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  78. package/dist/{loader-BQOfjTIJ.js → loader-DOVB0Syl.js} +12 -12
  79. package/dist/main.js +9822 -7824
  80. package/dist/{mermaid-CHo49Yv-.js → mermaid-CZ6FrClS.js} +38 -38
  81. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  82. package/dist/{mindmap-definition-YA3MSWOX-CrT0KSfX.js → mindmap-definition-YA3MSWOX-DR4ripvd.js} +1 -1
  83. package/dist/{number-overlay-editor-GnpmNxbw.js → number-overlay-editor-BiEkIBfV.js} +3 -3
  84. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  85. package/dist/{once-BtE-wKRL.js → once-eG8Vmvjd.js} +1 -1
  86. package/dist/{pegDiagram-XKGWAZYB-BqvywNJA.js → pegDiagram-XKGWAZYB-CFr3XlGl.js} +2 -2
  87. package/dist/{pieDiagram-E7YTZNPT-BfxJ_XCc.js → pieDiagram-E7YTZNPT-j91XYSLY.js} +1 -1
  88. package/dist/precisionRound-DhAzeRcm.js +49 -0
  89. package/dist/{process-output-BZg56CCb.js → process-output-DLS4Mqg_.js} +4 -4
  90. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  91. package/dist/{railroadDiagram-O6MQD6OU-8vFLVTPu.js → railroadDiagram-O6MQD6OU-Cd47ZV9E.js} +2 -2
  92. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  93. package/dist/{requirementDiagram-BXWQKSXE-a_MxGvxe.js → requirementDiagram-BXWQKSXE-eJH6gLtX.js} +1 -1
  94. package/dist/{reveal-component-B1Wbl-hd.js → reveal-component-C0zVfY6v.js} +14 -14
  95. package/dist/{select-BButFi-d.js → select-2KOiReNY.js} +536 -457
  96. package/dist/{sequenceDiagram-WJ2MYXX4-CXFMnxGh.js → sequenceDiagram-WJ2MYXX4-DdfFEIrm.js} +1 -1
  97. package/dist/{spec-BwarBpUJ.js → spec-h69nC3SI.js} +41 -11
  98. package/dist/{stateDiagram-D77RDMKH-DQeA_CwI.js → stateDiagram-D77RDMKH-DkjDvip8.js} +2 -2
  99. package/dist/{stateDiagram-v2-MP3YSRHH-BxxQucm1.js → stateDiagram-v2-MP3YSRHH-DBHACnOo.js} +2 -2
  100. package/dist/style.css +1 -1
  101. package/dist/{swimlanes-42K2YHIH-Do6L-cYg.js → swimlanes-42K2YHIH-ogImlOmp.js} +1 -1
  102. package/dist/{swimlanesDiagram-VR7AAH4N-DUzPvSVU.js → swimlanesDiagram-VR7AAH4N-D6x-zsci.js} +2 -2
  103. package/dist/{swiper-component-DKntyn8p.js → swiper-component-BkCm-kLg.js} +2 -2
  104. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  105. package/dist/{timeline-definition-24CTP7MA-D3c46WM5.js → timeline-definition-24CTP7MA-DQpLtsHD.js} +1 -1
  106. package/dist/{toDate-UtiwmnH2.js → toDate-CDzoofJo.js} +37 -36
  107. package/dist/{toast-B9UMEi8u.js → toast-B5t9o6s_.js} +219 -201
  108. package/dist/{tooltip-BO1REbqs.js → tooltip-s4zZxwKe.js} +325 -301
  109. package/dist/{use-restore-focus-BEM7sFWl.js → use-restore-focus-BgDTd2tc.js} +1482 -1433
  110. package/dist/{useAsyncData-Deh5kszT.js → useAsyncData-Bn8SxUqM.js} +15 -14
  111. package/dist/{useDateFormatter-D8XruNGG.js → useDateFormatter-D3Tkfzej.js} +2 -2
  112. package/dist/{useDeepCompareMemoize-DXwzRINI.js → useDeepCompareMemoize-CC2tnPCk.js} +1 -1
  113. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-CSdWn6NP.js} +1 -1
  114. package/dist/{useLifecycle-j7D9uD1o.js → useLifecycle-BJlxHqWa.js} +28 -25
  115. package/dist/{useTheme-CRnjkku0.js → useTheme-DMfQYmlG.js} +301 -297
  116. package/dist/{vega-component-BeF9dfyO.js → vega-component-CovWHi60.js} +148 -148
  117. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  118. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  119. package/dist/{vennDiagram-4TSXK5OY-BSZD8JcZ.js → vennDiagram-4TSXK5OY-BD-SL_vq.js} +1 -1
  120. package/dist/{wardleyDiagram-VM6X3IG4-48cUWkfA.js → wardleyDiagram-VM6X3IG4-BDgJV4ah.js} +1 -1
  121. package/dist/{xychartDiagram-S5SC5T6Z-CswYgYeu.js → xychartDiagram-S5SC5T6Z-1hIHN6Rl.js} +2 -2
  122. package/package.json +16 -16
  123. package/src/__mocks__/document.tsx +54 -0
  124. package/src/__mocks__/notebook.ts +2 -1
  125. package/src/__mocks__/requests.ts +0 -3
  126. package/src/__tests__/module-entry.test.ts +44 -0
  127. package/src/__tests__/mount.test.ts +1 -21
  128. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  129. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  130. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  131. package/src/components/app-config/ai-config.tsx +7 -2
  132. package/src/components/app-config/user-config-form.tsx +35 -0
  133. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  134. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  135. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  136. package/src/components/chat/acp/blocks.tsx +3 -2
  137. package/src/components/chat/acp/session-tabs.tsx +8 -3
  138. package/src/components/chat/chat-components.tsx +2 -0
  139. package/src/components/chat/chat-panel.tsx +12 -11
  140. package/src/components/chat/chat-utils.ts +4 -8
  141. package/src/components/data-table/TableBottomBar.tsx +4 -1
  142. package/src/components/data-table/TableTopBar.tsx +2 -0
  143. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  144. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  145. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  146. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  147. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  148. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  149. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  150. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  151. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  152. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  153. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  154. package/src/components/data-table/cell-selection/feature.ts +2 -2
  155. package/src/components/data-table/cell-selection/types.ts +2 -2
  156. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  157. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  158. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  159. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  160. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  161. package/src/components/data-table/charts/charts.tsx +40 -30
  162. package/src/components/data-table/charts/storage.ts +16 -5
  163. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  164. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  165. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  166. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  167. package/src/components/data-table/columns.tsx +45 -46
  168. package/src/components/data-table/context-menu.tsx +36 -20
  169. package/src/components/data-table/data-table.tsx +17 -1
  170. package/src/components/data-table/date-popover.tsx +6 -4
  171. package/src/components/data-table/export-actions.tsx +76 -48
  172. package/src/components/data-table/pagination.tsx +44 -33
  173. package/src/components/data-table/range-focus/atoms.ts +4 -6
  174. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  175. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  176. package/src/components/data-table/renderers.tsx +16 -7
  177. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  178. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  179. package/src/components/data-table/search-highlight.tsx +79 -0
  180. package/src/components/data-table/types.ts +1 -0
  181. package/src/components/data-table/url-detector.tsx +49 -9
  182. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  183. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  184. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  185. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  186. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  187. package/src/components/datasources/column-preview.tsx +17 -12
  188. package/src/components/datasources/datasources.tsx +55 -42
  189. package/src/components/datasources/utils.ts +32 -23
  190. package/src/components/debugger/debugger-code.tsx +5 -0
  191. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  192. package/src/components/dependency-graph/elements.ts +24 -21
  193. package/src/components/dependency-graph/minimap-content.tsx +2 -6
  194. package/src/components/dependency-graph/panels.tsx +4 -2
  195. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  196. package/src/components/editor/SortableCell.tsx +1 -3
  197. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  198. package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
  199. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -13
  200. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  201. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -1
  202. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  203. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  204. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +34 -8
  205. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +2 -6
  206. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  207. package/src/components/editor/actions/pair-with-agent-commands.ts +2 -2
  208. package/src/components/editor/actions/pair-with-agent-modal.tsx +19 -16
  209. package/src/components/editor/actions/useCellActionButton.tsx +20 -26
  210. package/src/components/editor/actions/useNotebookActions.tsx +14 -19
  211. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  212. package/src/components/editor/actions/useSetCodeVisibility.ts +10 -11
  213. package/src/components/editor/actions/useSetOutputsExpanded.ts +13 -18
  214. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  215. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  216. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  217. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  218. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  219. package/src/components/editor/app-container.tsx +1 -1
  220. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  221. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  222. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  223. package/src/components/editor/cell/RunButton.tsx +2 -2
  224. package/src/components/editor/cell/StagedAICell.tsx +10 -2
  225. package/src/components/editor/cell/StopButton.tsx +2 -2
  226. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +3 -1
  227. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  228. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  229. package/src/components/editor/cell/code/cell-editor.tsx +22 -36
  230. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  231. package/src/components/editor/cell/collapse.tsx +5 -3
  232. package/src/components/editor/cell/useAddCell.ts +4 -8
  233. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  234. package/src/components/editor/cell/useRunCells.ts +2 -5
  235. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  236. package/src/components/editor/cell/useSplitCell.tsx +2 -1
  237. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  238. package/src/components/editor/chrome/components/feedback-button.tsx +1 -1
  239. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  240. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  241. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  242. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  243. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  244. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  245. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  246. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  247. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  248. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  249. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  250. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +9 -18
  251. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  252. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +61 -2
  253. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
  254. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  255. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  256. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  257. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  258. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  259. package/src/components/editor/chrome/wrapper/footer.tsx +2 -0
  260. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  261. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  262. package/src/components/editor/columns/sortable-column.tsx +1 -1
  263. package/src/components/editor/connections/components.tsx +5 -10
  264. package/src/components/editor/controls/Controls.tsx +3 -3
  265. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  266. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  267. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  268. package/src/components/editor/controls/command-palette.tsx +22 -10
  269. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  270. package/src/components/editor/documentation.css +34 -11
  271. package/src/components/editor/dynamic-favicon.tsx +21 -40
  272. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +9 -23
  273. package/src/components/editor/errors/auto-fix.tsx +4 -8
  274. package/src/components/editor/errors/send-error-report-button.tsx +3 -6
  275. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  276. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  277. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  278. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  279. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  280. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  281. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  282. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  283. package/src/components/editor/header/app-header.tsx +1 -1
  284. package/src/components/editor/header/status.tsx +40 -20
  285. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  286. package/src/components/editor/links/cell-link.tsx +16 -6
  287. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  288. package/src/components/editor/navigation/__tests__/navigation.test.ts +73 -64
  289. package/src/components/editor/navigation/clipboard.ts +12 -11
  290. package/src/components/editor/navigation/focus-utils.ts +4 -4
  291. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +11 -20
  292. package/src/components/editor/navigation/navigation.ts +20 -28
  293. package/src/components/editor/notebook-banner.tsx +1 -0
  294. package/src/components/editor/notebook-cell.tsx +19 -13
  295. package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
  296. package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
  297. package/src/components/editor/output/Outputs.css +8 -0
  298. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  299. package/src/components/editor/output/ansi-reduce.ts +33 -20
  300. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  301. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  302. package/src/components/editor/package-alert.tsx +14 -7
  303. package/src/components/editor/renderers/cell-array.tsx +10 -17
  304. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  305. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  306. package/src/components/home/components.tsx +17 -31
  307. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  308. package/src/components/icons/large-spinner.tsx +3 -14
  309. package/src/components/icons/spinner.tsx +1 -1
  310. package/src/components/markdown/markdown-renderer.tsx +4 -8
  311. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  312. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  313. package/src/components/pages/edit-page.tsx +32 -17
  314. package/src/components/pages/home-page.tsx +164 -91
  315. package/src/components/scratchpad/scratchpad.test.tsx +97 -0
  316. package/src/components/scratchpad/scratchpad.tsx +20 -10
  317. package/src/components/slides/__tests__/minimap-actions.test.tsx +9 -25
  318. package/src/components/slides/minimap.tsx +7 -3
  319. package/src/components/slides/reveal-component.tsx +4 -2
  320. package/src/components/slides/slide-cell-view.tsx +4 -4
  321. package/src/components/slides/slide-form.tsx +1 -0
  322. package/src/components/sort/SortableCellsProvider.tsx +2 -1
  323. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  324. package/src/components/tracing/tracing.tsx +45 -41
  325. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  326. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  327. package/src/components/ui/alert-dialog.tsx +1 -1
  328. package/src/components/ui/draggable-popover.tsx +24 -2
  329. package/src/components/ui/range-slider.tsx +22 -9
  330. package/src/core/ai/__tests__/staged-cells.test.ts +26 -25
  331. package/src/core/ai/context/providers/datasource.ts +6 -2
  332. package/src/core/ai/staged-cells.ts +15 -18
  333. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +5 -17
  334. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +2 -2
  335. package/src/core/ai/tools/base.ts +18 -9
  336. package/src/core/ai/tools/edit-notebook-tool.ts +74 -69
  337. package/src/core/ai/tools/registry.ts +13 -8
  338. package/src/core/ai/tools/sample-tool.ts +6 -7
  339. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -0
  340. package/src/core/cells/__tests__/actions.test.ts +474 -34
  341. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  342. package/src/core/cells/__tests__/cell.test.ts +39 -0
  343. package/src/core/cells/__tests__/cells.test.ts +260 -1165
  344. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  345. package/src/core/cells/__tests__/document.test.ts +414 -0
  346. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  347. package/src/core/cells/__tests__/notebook-outline.test.ts +114 -0
  348. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  349. package/src/core/cells/__tests__/session.test.ts +117 -36
  350. package/src/core/cells/__tests__/with-document-save.test.ts +67 -0
  351. package/src/core/cells/actions.ts +391 -3
  352. package/src/core/cells/add-missing-import.ts +46 -42
  353. package/src/core/cells/cell.ts +14 -3
  354. package/src/core/cells/cells.ts +99 -599
  355. package/src/core/cells/columns.ts +38 -0
  356. package/src/core/cells/document.ts +392 -0
  357. package/src/core/cells/outputs.ts +12 -28
  358. package/src/core/cells/pending-delete-service.ts +2 -6
  359. package/src/core/cells/session.ts +1 -1
  360. package/src/core/cells/types.ts +22 -1
  361. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  362. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  363. package/src/core/codemirror/ai/resources.ts +1 -1
  364. package/src/core/codemirror/cells/__tests__/extensions.test.ts +42 -1
  365. package/src/core/codemirror/cells/extensions.ts +27 -9
  366. package/src/core/codemirror/cells/state.ts +1 -1
  367. package/src/core/codemirror/format.ts +6 -5
  368. package/src/core/codemirror/keymaps/vim.ts +4 -8
  369. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  370. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  371. package/src/core/codemirror/language/cell-code.ts +114 -0
  372. package/src/core/codemirror/language/extension.ts +52 -35
  373. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  374. package/src/core/codemirror/language/languages/python.ts +16 -10
  375. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  376. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  377. package/src/core/codemirror/language/types.ts +7 -7
  378. package/src/core/codemirror/language/utils.ts +9 -39
  379. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  380. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  381. package/src/core/codemirror/readonly/extension.ts +2 -2
  382. package/src/core/codemirror/rtc/extension.ts +24 -13
  383. package/src/core/codemirror/rtc/loro/awareness.ts +34 -19
  384. package/src/core/codemirror/rtc/loro/sync.ts +1 -0
  385. package/src/core/codemirror/utils.ts +1 -0
  386. package/src/core/config/__tests__/config-schema.test.ts +5 -3
  387. package/src/core/config/config-schema.ts +3 -0
  388. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  389. package/src/core/datasets/data-source-discovery.ts +4 -6
  390. package/src/core/edit-app.tsx +19 -23
  391. package/src/core/errors/error-entries.ts +3 -0
  392. package/src/core/export/hooks.ts +1 -1
  393. package/src/core/export/layout.ts +2 -2
  394. package/src/core/hotkeys/actions.ts +2 -4
  395. package/src/core/islands/__tests__/bootstrap.test.ts +73 -43
  396. package/src/core/islands/__tests__/bridge.test.ts +72 -25
  397. package/src/core/islands/bootstrap.ts +46 -53
  398. package/src/core/islands/bridge.ts +74 -23
  399. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  400. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  401. package/src/core/islands/worker/controller.ts +4 -10
  402. package/src/core/islands/worker/worker.tsx +33 -33
  403. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  404. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  405. package/src/core/kernel/handlers.ts +18 -59
  406. package/src/core/network/__tests__/connection.test.ts +1 -1
  407. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  408. package/src/core/network/connection-notice.ts +1 -1
  409. package/src/core/network/connection.ts +1 -1
  410. package/src/core/network/requests-lazy.ts +1 -4
  411. package/src/core/network/requests-network.ts +0 -25
  412. package/src/core/network/requests-static.ts +0 -3
  413. package/src/core/network/requests-toasting.tsx +0 -3
  414. package/src/core/network/types.ts +0 -12
  415. package/src/core/network/useConnectionNotice.ts +8 -2
  416. package/src/core/notebook/attachments.ts +38 -0
  417. package/src/core/notebook/document/__tests__/codemirror.test.ts +552 -0
  418. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  419. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  420. package/src/core/notebook/document/__tests__/hold.test.ts +88 -0
  421. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  422. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  423. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  424. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  425. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  426. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  427. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  428. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  429. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  430. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  431. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  432. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  433. package/src/core/notebook/document/__tests__/user.ts +61 -0
  434. package/src/core/notebook/document/codemirror.ts +216 -0
  435. package/src/core/notebook/document/draft.ts +502 -0
  436. package/src/core/notebook/document/hooks.tsx +118 -0
  437. package/src/core/notebook/document/index.ts +24 -0
  438. package/src/core/notebook/document/model.ts +105 -0
  439. package/src/core/notebook/document/select.ts +41 -0
  440. package/src/core/notebook/document/state.ts +378 -0
  441. package/src/core/notebook/document/sync.ts +418 -0
  442. package/src/core/notebook/index.ts +18 -0
  443. package/src/core/notebook/replica/AGENTS.md +11 -0
  444. package/src/core/notebook/replica/__tests__/attachments.test.ts +56 -0
  445. package/src/core/notebook/replica/__tests__/document.test.ts +126 -0
  446. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  447. package/src/core/notebook/replica/__tests__/replica.test.ts +550 -0
  448. package/src/core/notebook/replica/backend.ts +19 -0
  449. package/src/core/notebook/replica/connection.ts +42 -0
  450. package/src/core/notebook/replica/index.ts +13 -0
  451. package/src/core/notebook/replica/replica.ts +286 -0
  452. package/src/core/notebook/replica/resources/attachments.ts +22 -0
  453. package/src/core/notebook/replica/resources/document.ts +101 -0
  454. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  455. package/src/core/packages/useSandboxController.ts +1 -1
  456. package/src/core/run-app.tsx +48 -51
  457. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  458. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  459. package/src/core/runtime/__tests__/runtime.test.ts +32 -58
  460. package/src/core/runtime/adapter.ts +1 -1
  461. package/src/core/runtime/config.ts +2 -2
  462. package/src/core/runtime/runtime.ts +38 -42
  463. package/src/core/runtime/types.ts +0 -2
  464. package/src/core/saving/filename.ts +1 -1
  465. package/src/core/saving/save-component.tsx +10 -4
  466. package/src/core/saving/useAutoSave.ts +1 -1
  467. package/src/core/session/__tests__/backend.test.ts +235 -0
  468. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +183 -0
  469. package/src/core/session/__tests__/useNotebookSession.hook.test.tsx +663 -0
  470. package/src/core/session/active-notebook.ts +11 -0
  471. package/src/core/session/backend.ts +125 -0
  472. package/src/core/{websocket → session}/types.ts +2 -2
  473. package/src/core/session/useNotebookReplica.ts +72 -0
  474. package/src/core/{websocket → session}/useNotebookSession.tsx +74 -68
  475. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  476. package/src/core/state/jotai.ts +1 -1
  477. package/src/core/static/__tests__/files.test.ts +38 -10
  478. package/src/core/static/files.ts +2 -0
  479. package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
  480. package/src/core/transports/__tests__/server.test.ts +408 -0
  481. package/src/core/transports/__tests__/static.test.ts +175 -0
  482. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  483. package/src/core/{websocket → transports}/connection-state.ts +9 -16
  484. package/src/core/transports/server.ts +152 -0
  485. package/src/core/transports/static.ts +84 -0
  486. package/src/core/transports/wasm.ts +97 -0
  487. package/src/core/wasm/__tests__/bridge.test.ts +213 -50
  488. package/src/core/wasm/__tests__/worker.test.ts +187 -0
  489. package/src/core/wasm/bridge.ts +75 -69
  490. package/src/core/wasm/rpc.ts +15 -3
  491. package/src/core/wasm/store.ts +3 -3
  492. package/src/core/wasm/worker/bootstrap.ts +55 -51
  493. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  494. package/src/core/wasm/worker/save-worker.ts +9 -10
  495. package/src/core/wasm/worker/types.ts +29 -18
  496. package/src/core/wasm/worker/worker.ts +88 -44
  497. package/src/css/globals.css +66 -22
  498. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  499. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  500. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  501. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  502. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  503. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  504. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  505. package/src/hooks/useAsyncData.ts +3 -3
  506. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  507. package/src/hooks/useDebounce.ts +6 -10
  508. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  509. package/src/hooks/useDelayElapsed.ts +14 -10
  510. package/src/hooks/useElapsedTime.ts +3 -3
  511. package/src/hooks/useEventListener.ts +1 -2
  512. package/src/hooks/useInputHistory.ts +4 -7
  513. package/src/hooks/useInternalStateWithSync.ts +6 -6
  514. package/src/hooks/useResizeObserver.ts +4 -7
  515. package/src/hooks/useScript.ts +67 -70
  516. package/src/hooks/useStateForKey.ts +93 -0
  517. package/src/mount.tsx +1 -5
  518. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  519. package/src/plugins/core/registerReactComponent.tsx +1 -1
  520. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  521. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  522. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  523. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  524. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  525. package/src/plugins/impl/FormPlugin.tsx +1 -1
  526. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  527. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  528. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  529. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
  530. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  531. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  532. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  533. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  534. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  535. package/src/plugins/impl/anywidget/model.ts +3 -1
  536. package/src/plugins/impl/anywidget/registry.ts +18 -14
  537. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  538. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  539. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  540. package/src/plugins/impl/chat/chat-ui.tsx +14 -11
  541. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  542. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  543. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  544. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  545. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  546. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  547. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  548. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  549. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  550. package/src/plugins/impl/data-frames/panel.tsx +43 -16
  551. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  552. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  553. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  554. package/src/plugins/impl/panel/utils.ts +17 -0
  555. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  556. package/src/plugins/impl/plotly/selection.ts +25 -19
  557. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  558. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  559. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  560. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  561. package/src/stories/button.stories.tsx +1 -0
  562. package/src/stories/cell.stories.tsx +1 -1
  563. package/src/stories/data-table.stories.tsx +31 -0
  564. package/src/stories/editor.stories.tsx +24 -17
  565. package/src/stories/form-wrapper.stories.tsx +2 -2
  566. package/src/stories/layout/vertical/one-column.stories.tsx +1 -1
  567. package/src/stories/runtime-story.tsx +23 -16
  568. package/src/stories/select.stories.tsx +2 -2
  569. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  570. package/src/utils/__tests__/arrays.test.ts +35 -0
  571. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  572. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
  573. package/src/utils/__tests__/numbers.test.ts +105 -1
  574. package/src/utils/__tests__/objects.test.ts +21 -0
  575. package/src/utils/arrays.ts +11 -2
  576. package/src/utils/blob.ts +1 -1
  577. package/src/utils/createReducer.ts +53 -26
  578. package/src/utils/maps.ts +0 -3
  579. package/src/utils/numbers.ts +79 -18
  580. package/src/utils/objects.ts +4 -9
  581. package/src/utils/once.ts +19 -19
  582. package/src/utils/tracer.ts +9 -5
  583. package/src/utils/zod-utils.ts +1 -3
  584. package/dist/any-language-editor-DZ_YTVus.js +0 -76
  585. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  586. package/dist/assets/worker-HPY9fbhz.js +0 -114
  587. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  588. package/dist/precisionRound-DMIVQrt2.js +0 -97
  589. package/src/core/attachments/state.ts +0 -21
  590. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  591. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  592. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  593. package/src/core/cells/document-changes.ts +0 -944
  594. package/src/core/host/AGENTS.md +0 -6
  595. package/src/core/host/__tests__/events.test.ts +0 -46
  596. package/src/core/host/__tests__/notebook-client.test.ts +0 -191
  597. package/src/core/host/__tests__/protocol.test.ts +0 -214
  598. package/src/core/host/connection.ts +0 -81
  599. package/src/core/host/events.ts +0 -60
  600. package/src/core/host/notebook-client.ts +0 -226
  601. package/src/core/host/protocol.ts +0 -1943
  602. package/src/core/state/__tests__/jotai.test.ts +0 -25
  603. package/src/core/wasm/worker/message-buffer.ts +0 -33
  604. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  605. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -140
  606. package/src/core/websocket/__tests__/useNotebookSession.hook.test.tsx +0 -883
  607. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  608. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  609. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  610. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  611. package/src/core/websocket/transports/basic.ts +0 -102
  612. package/src/core/websocket/transports/create.ts +0 -31
  613. package/src/core/websocket/transports/sse.ts +0 -308
  614. package/src/core/websocket/transports/transport.ts +0 -50
  615. package/src/core/websocket/transports/ws.ts +0 -98
  616. package/src/core/websocket/useNotebookClient.ts +0 -80
  617. /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
  618. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -104,6 +104,7 @@ const CachePanel = () => {
104
104
  {/* Header with Refresh Button */}
105
105
  <div className="flex items-center justify-end">
106
106
  <Button
107
+ aria-label="Refresh cache"
107
108
  variant="ghost"
108
109
  size="icon"
109
110
  className="h-6 w-6"
@@ -3,7 +3,8 @@
3
3
  import { FileTextIcon } from "lucide-react";
4
4
  import React from "react";
5
5
  import { ClearButton } from "@/components/buttons/clear-button";
6
- import { useCellActions, useCellLogs } from "@/core/cells/cells";
6
+ import { useCellLogs } from "@/core/cells/cells";
7
+ import { useCellActions } from "@/core/cells/actions";
7
8
  import {
8
9
  type CellLog,
9
10
  formatLogTimestamp,
@@ -0,0 +1,84 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { fireEvent, render, screen } from "@testing-library/react";
4
+ import { describe, expect, it, vi } from "vitest";
5
+ import type { OutlineItem } from "@/core/cells/outline";
6
+ import { MiniMap, OutlineList } from "../floating-outline";
7
+ import { scrollToOutlineItem } from "../useActiveOutline";
8
+
9
+ vi.mock("../useActiveOutline", () => ({ scrollToOutlineItem: vi.fn() }));
10
+
11
+ const items: OutlineItem[] = [
12
+ { name: "First", level: 1, by: { id: "first" } },
13
+ { name: "Second", html: "<em>Second</em>", level: 2, by: { id: "second" } },
14
+ ];
15
+
16
+ describe("outline keyboard navigation", () => {
17
+ it.each([
18
+ { Component: MiniMap, role: "button" },
19
+ { Component: OutlineList, role: "link" },
20
+ ])(
21
+ "keeps the focused $role when other headings change",
22
+ ({ Component, role }) => {
23
+ const duplicate: OutlineItem = {
24
+ name: "Repeated second",
25
+ level: 2,
26
+ by: { id: "second" },
27
+ };
28
+ const initial = [...items, duplicate];
29
+ const props = {
30
+ activeHeaderId: undefined,
31
+ activeOccurrences: undefined,
32
+ };
33
+ const { rerender } = render(<Component {...props} items={initial} />);
34
+ const focused = screen.getByRole(role, { name: "Repeated second" });
35
+ focused.focus();
36
+
37
+ for (const by of [{ id: "inserted" }, { path: "second" }]) {
38
+ const inserted: OutlineItem = { name: "Inserted", level: 1, by };
39
+ rerender(<Component {...props} items={[inserted, ...initial]} />);
40
+ expect(screen.getByRole(role, { name: "Repeated second" })).toBe(
41
+ focused,
42
+ );
43
+ expect(focused).toHaveFocus();
44
+ }
45
+
46
+ rerender(
47
+ <Component {...props} items={[items[1], duplicate, items[0]]} />,
48
+ );
49
+ expect(screen.getByRole(role, { name: "Repeated second" })).toBe(focused);
50
+ expect(focused).toHaveFocus();
51
+ fireEvent.click(focused);
52
+ expect(scrollToOutlineItem).toHaveBeenLastCalledWith(duplicate, 1);
53
+ },
54
+ );
55
+
56
+ it("activates plain and rich headings with Enter", () => {
57
+ render(
58
+ <OutlineList
59
+ items={items}
60
+ activeHeaderId={undefined}
61
+ activeOccurrences={undefined}
62
+ />,
63
+ );
64
+ items.forEach((item) => {
65
+ const link = screen.getByRole("link", { name: item.name });
66
+ expect(link).toHaveAttribute("tabindex", "0");
67
+ fireEvent.keyDown(link, { key: "Enter" });
68
+ expect(scrollToOutlineItem).toHaveBeenLastCalledWith(item, 0);
69
+ });
70
+ });
71
+
72
+ it("names each minimap button", () => {
73
+ render(
74
+ <MiniMap
75
+ items={items}
76
+ activeHeaderId={undefined}
77
+ activeOccurrences={undefined}
78
+ />,
79
+ );
80
+ const button = screen.getByRole("button", { name: "Second" });
81
+ fireEvent.click(button);
82
+ expect(scrollToOutlineItem).toHaveBeenLastCalledWith(items[1], 0);
83
+ });
84
+ });
@@ -17,6 +17,8 @@ export const FloatingOutline: React.FC = () => {
17
17
  findOutlineElements(items),
18
18
  );
19
19
  const [isHovered, setIsHovered] = React.useState(false);
20
+ const [isFocused, setIsFocused] = React.useState(false);
21
+ const isVisible = isHovered || isFocused;
20
22
 
21
23
  // Hide if < 2 items
22
24
  // It's kinda useless to have an outline with only one item
@@ -29,6 +31,12 @@ export const FloatingOutline: React.FC = () => {
29
31
  <div
30
32
  onMouseEnter={() => setIsHovered(true)}
31
33
  onMouseLeave={() => setIsHovered(false)}
34
+ onFocus={() => setIsFocused(true)}
35
+ onBlur={(e) => {
36
+ if (!e.currentTarget.contains(e.relatedTarget)) {
37
+ setIsFocused(false);
38
+ }
39
+ }}
32
40
  className={cn(
33
41
  "fixed top-[25vh] right-8 z-10000 print:hidden",
34
42
  // Hide on small screens
@@ -36,9 +44,10 @@ export const FloatingOutline: React.FC = () => {
36
44
  )}
37
45
  >
38
46
  <OutlineList
47
+ inert={!isVisible}
39
48
  className={cn(
40
49
  "-top-4 max-h-[70vh] bg-background rounded-lg shadow-lg absolute overflow-auto transition-all duration-300 w-[300px] border",
41
- isHovered ? "left-[-280px] opacity-100" : "left-[300px] opacity-0",
50
+ isVisible ? "left-[-280px] opacity-100" : "left-[300px] opacity-0",
42
51
  )}
43
52
  items={items}
44
53
  activeHeaderId={activeHeaderId}
@@ -62,17 +71,20 @@ export const MiniMap: React.FC<{
62
71
  const seen = new Map<string, number>();
63
72
  return (
64
73
  <div className="flex flex-col gap-4 items-end max-h-[70vh] overflow-hidden">
65
- {items.map((item, idx) => {
74
+ {items.map((item) => {
66
75
  const identifier = "id" in item.by ? item.by.id : item.by.path;
76
+ const selector = JSON.stringify(item.by);
67
77
  // Keep track of how many times we've seen this selector
68
- const occurrences = seen.get(identifier) ?? 0;
69
- seen.set(identifier, occurrences + 1);
78
+ const occurrences = seen.get(selector) ?? 0;
79
+ seen.set(selector, occurrences + 1);
70
80
 
71
81
  return (
72
- <div
73
- key={`${identifier}-${idx}`}
82
+ <button
83
+ type="button"
84
+ aria-label={item.name}
85
+ key={`${selector}-${occurrences}`}
74
86
  className={cn(
75
- "h-[2px] bg-muted-foreground/60",
87
+ "h-[2px] bg-muted-foreground/60 focus-visible:outline focus-visible:outline-ring",
76
88
  item.level === 1 && "w-5",
77
89
  item.level === 2 && "w-4",
78
90
  item.level === 3 && "w-3",
@@ -91,24 +103,31 @@ export const MiniMap: React.FC<{
91
103
 
92
104
  export const OutlineList: React.FC<{
93
105
  className?: string;
106
+ inert?: boolean;
94
107
  items: OutlineItem[];
95
108
  activeHeaderId: string | undefined;
96
109
  activeOccurrences: number | undefined;
97
- }> = ({ items, activeHeaderId, activeOccurrences, className }) => {
110
+ }> = ({ items, activeHeaderId, activeOccurrences, className, inert }) => {
98
111
  // Map of selector to its occurrences
99
112
  const seen = new Map<string, number>();
100
113
  return (
101
- <div className={cn("flex flex-col overflow-auto py-4 pl-2", className)}>
102
- {items.map((item, idx) => {
114
+ <div
115
+ inert={inert}
116
+ className={cn("flex flex-col overflow-auto py-4 pl-2", className)}
117
+ >
118
+ {items.map((item) => {
103
119
  const identifier = "id" in item.by ? item.by.id : item.by.path;
120
+ const selector = JSON.stringify(item.by);
104
121
  // Keep track of how many times we've seen this selector
105
- const occurrences = seen.get(identifier) ?? 0;
106
- seen.set(identifier, occurrences + 1);
122
+ const occurrences = seen.get(selector) ?? 0;
123
+ seen.set(selector, occurrences + 1);
107
124
 
108
- const key = `${identifier}-${idx}`;
125
+ const key = `${selector}-${occurrences}`;
109
126
  const sharedProps = {
127
+ role: "link",
128
+ tabIndex: 0,
110
129
  className: cn(
111
- "px-2 py-1 cursor-pointer hover:bg-accent/50 hover:text-accent-foreground rounded-l",
130
+ "px-2 py-1 cursor-pointer hover:bg-accent/50 hover:text-accent-foreground rounded-l focus-visible:outline focus-visible:outline-ring",
112
131
  item.level === 1 && "font-semibold",
113
132
  item.level === 2 && "ml-3",
114
133
  item.level === 3 && "ml-6",
@@ -118,6 +137,12 @@ export const OutlineList: React.FC<{
118
137
  "text-accent-foreground",
119
138
  ),
120
139
  onClick: () => scrollToOutlineItem(item, occurrences),
140
+ onKeyDown: (e: React.KeyboardEvent) => {
141
+ if (e.target === e.currentTarget && e.key === "Enter") {
142
+ e.preventDefault();
143
+ scrollToOutlineItem(item, occurrences);
144
+ }
145
+ },
121
146
  };
122
147
 
123
148
  if (item.html) {
@@ -113,9 +113,10 @@ export function findOutlineElements(items: OutlineItem[]) {
113
113
  return items
114
114
  .map((item) => {
115
115
  const identifier = "id" in item.by ? item.by.id : item.by.path;
116
+ const selector = JSON.stringify(item.by);
116
117
  // Keep track of how many times we've seen this selector
117
- const occurrences = seen.get(identifier) ?? 0;
118
- seen.set(identifier, occurrences + 1);
118
+ const occurrences = seen.get(selector) ?? 0;
119
+ seen.set(selector, occurrences + 1);
119
120
 
120
121
  const el = findOutlineItem(item, occurrences);
121
122
  if (!el) {
@@ -31,6 +31,7 @@ import { usePackageAction } from "@/core/packages/usePackageAction";
31
31
  import { usePackageDependencies } from "@/core/packages/usePackageDependencies";
32
32
  import { useInstallPackages } from "@/core/packages/useInstallPackage";
33
33
  import { isWasm } from "@/core/wasm/utils";
34
+ import { useStateForKey } from "@/hooks/useStateForKey";
34
35
  import { ErrorBanner } from "@/plugins/impl/common/error-banner";
35
36
  import { cn } from "@/utils/cn";
36
37
  import { copyToClipboard } from "@/utils/copy";
@@ -440,15 +441,11 @@ const DependencyTree: React.FC<{
440
441
  error?: Error | null;
441
442
  sandboxBackend: "uv" | "pixi" | null;
442
443
  }> = ({ tree, error, sandboxBackend }) => {
443
- const [expandedNodes, setExpandedNodes] = React.useState<Set<string>>(
444
- new Set(),
444
+ const [expandedNodes, setExpandedNodes] = useStateForKey(
445
+ tree,
446
+ () => new Set<string>(),
445
447
  );
446
448
 
447
- // Reset tree to collapsed state when tree data changes (including refetches)
448
- React.useEffect(() => {
449
- setExpandedNodes(new Set());
450
- }, [tree]);
451
-
452
449
  if (error) {
453
450
  return <ErrorBanner error={error} />;
454
451
  }
@@ -66,6 +66,7 @@ const SecretsPanel: React.FC = () => {
66
66
  <div className="flex justify-start h-8 border-b">
67
67
  <button
68
68
  type="button"
69
+ aria-label="Add environment variable"
69
70
  className="border-r px-2 m-0 h-full hover:bg-accent hover:text-accent-foreground"
70
71
  onClick={() =>
71
72
  openModal(
@@ -22,7 +22,7 @@ import { Suspense } from "react";
22
22
  import { Spinner } from "@/components/icons/spinner";
23
23
  import { Button } from "@/components/ui/button";
24
24
  import { Tooltip } from "@/components/ui/tooltip";
25
- import { useCellActions } from "@/core/cells/cells";
25
+ import { useCellActions } from "@/core/cells/actions";
26
26
  import { useLastFocusedCellId } from "@/core/cells/focus";
27
27
  import { useRequestClient } from "@/core/network/requests";
28
28
  import { LazyAnyLanguageCodeMirror } from "@/plugins/impl/code/LazyAnyLanguageCodeMirror";
@@ -71,7 +71,11 @@ const SnippetsPanel: React.FC = () => {
71
71
  rootClassName="flex-1 border-r"
72
72
  />
73
73
  <ContributeSnippetButton>
74
- <button className="float-right px-2 m-0 h-full hover:bg-accent hover:text-accent-foreground">
74
+ <button
75
+ type="button"
76
+ aria-label="Contribute snippet"
77
+ className="float-right px-2 m-0 h-full hover:bg-accent hover:text-accent-foreground"
78
+ >
75
79
  <PlusIcon className="h-4 w-4" />
76
80
  </button>
77
81
  </ContributeSnippetButton>
@@ -121,18 +125,16 @@ const SnippetViewer: React.FC<{ snippet: Snippet; onClose: () => void }> = ({
121
125
  onClose,
122
126
  }) => {
123
127
  const { theme } = useTheme();
124
- const { createNewCell } = useCellActions();
128
+ const { createCell } = useCellActions();
125
129
  const lastFocusedCellId = useLastFocusedCellId();
126
130
 
127
131
  const handleInsertSnippet = () => {
128
132
  // Add below last focused cell in reverse order
129
133
  for (const section of snippet.sections.toReversed()) {
130
134
  if (section.code) {
131
- createNewCell({
135
+ createCell({
136
+ after: lastFocusedCellId,
132
137
  code: section.code,
133
- before: false,
134
- cellId: lastFocusedCellId ?? "__end__",
135
- // If the code already exists, skip creation
136
138
  skipIfCodeExists: true,
137
139
  });
138
140
  }
@@ -140,11 +142,7 @@ const SnippetViewer: React.FC<{ snippet: Snippet; onClose: () => void }> = ({
140
142
  };
141
143
 
142
144
  const handleInsertCode = (code: string) => {
143
- createNewCell({
144
- code,
145
- before: false,
146
- cellId: lastFocusedCellId ?? "__end__",
147
- });
145
+ createCell({ after: lastFocusedCellId, code });
148
146
  };
149
147
 
150
148
  return (
@@ -8,13 +8,13 @@ import {
8
8
  screen,
9
9
  waitFor,
10
10
  } from "@testing-library/react";
11
- import { createStore, Provider } from "jotai";
11
+ import { atom, createStore, Provider } from "jotai";
12
12
  import { userEvent } from "storybook/test";
13
13
  import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
14
14
  import { PAIR_CONFIG } from "@/__tests__/fixtures/pair-config";
15
15
  import { TooltipProvider } from "@/components/ui/tooltip";
16
16
  import { toast } from "@/components/ui/use-toast";
17
- import { type Attachment, attachmentsAtom } from "@/core/attachments/state";
17
+ import type { Attachment } from "@/core/notebook/attachments";
18
18
  import { pairConfigAtom } from "@/core/config/pair";
19
19
  import {
20
20
  type StableSessionId,
@@ -22,13 +22,19 @@ import {
22
22
  } from "@/core/kernel/session";
23
23
  import { API } from "@/core/network/api";
24
24
  import { connectionAtom } from "@/core/network/connection";
25
- import { WebSocketClosedReason, ConnectionState } from "@/core/websocket/types";
25
+ import { ConnectionState } from "@/core/session/types";
26
26
  import { Deferred } from "@/utils/Deferred";
27
27
  import { AttachmentStatus } from "../footer-items/attachment-status";
28
28
 
29
29
  vi.mock("@/components/ui/use-toast", () => ({ toast: vi.fn() }));
30
30
  vi.mock("@/core/network/api", () => ({ API: { post: vi.fn() } }));
31
31
 
32
+ const attachmentsAtom = atom<Attachment[]>([]);
33
+ vi.mock("@/core/session/active-notebook", async (importOriginal) => ({
34
+ ...(await importOriginal<typeof import("@/core/session/active-notebook")>()),
35
+ activeNotebookReplicaAtom: atom(() => ({ attachmentsAtom })),
36
+ }));
37
+
32
38
  const AGENT: Attachment = {
33
39
  id: "a1",
34
40
  kind: "agent",
@@ -342,21 +348,6 @@ describe("AttachmentStatus", () => {
342
348
  ).toBeDisabled();
343
349
  });
344
350
 
345
- it.each([
346
- { state: ConnectionState.CONNECTING },
347
- { state: ConnectionState.CLOSING },
348
- { state: ConnectionState.NOT_STARTED },
349
- {
350
- state: ConnectionState.CLOSED,
351
- code: WebSocketClosedReason.KERNEL_DISCONNECTED,
352
- reason: "offline",
353
- },
354
- ])("hides connected claims while the browser is %j", (connection) => {
355
- const { store, container } = renderStatus();
356
- act(() => store.set(connectionAtom, connection));
357
- expect(container).toBeEmptyDOMElement();
358
- });
359
-
360
351
  it("removes detached agents when the list is replaced", () => {
361
352
  const { store, container } = renderStatus([
362
353
  AGENT,
@@ -0,0 +1,92 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { act, fireEvent, render, screen } from "@testing-library/react";
4
+ import { atom, createStore, Provider } from "jotai";
5
+ import { afterEach, describe, expect, it, vi } from "vitest";
6
+ import { TooltipProvider } from "@/components/ui/tooltip";
7
+ import { createDocumentModel } from "@/core/notebook/document";
8
+ import { document } from "@/core/notebook/document/__tests__/fixtures";
9
+ import {
10
+ type FakeEdit,
11
+ fakeBackend,
12
+ } from "@/core/notebook/replica/__tests__/fixtures";
13
+ import { activeDocumentModelAtom } from "@/core/session/active-notebook";
14
+ import { DocumentSyncStatus } from "../footer-items/document-sync-status";
15
+
16
+ const lifetimes: AbortController[] = [];
17
+
18
+ afterEach(() => {
19
+ act(() => lifetimes.splice(0).forEach((controller) => controller.abort()));
20
+ });
21
+
22
+ const conflict = {
23
+ ok: false as const,
24
+ status: 409,
25
+ problem: {
26
+ type: "tag:marimo.io,2026:host/conflict",
27
+ title: "Conflict",
28
+ status: 409,
29
+ detail: "cell c1 has changed since it was read.",
30
+ },
31
+ };
32
+
33
+ function setup(edit: FakeEdit) {
34
+ const store = createStore();
35
+ const model = createDocumentModel({
36
+ store,
37
+ documentAtom: atom(document()),
38
+ backendAtom: atom(fakeBackend({ edit })),
39
+ });
40
+ const controller = new AbortController();
41
+ lifetimes.push(controller);
42
+ model.start({ signal: controller.signal });
43
+ store.set(activeDocumentModelAtom, model);
44
+ render(
45
+ <Provider store={store}>
46
+ <TooltipProvider>
47
+ <DocumentSyncStatus />
48
+ </TooltipProvider>
49
+ </Provider>,
50
+ );
51
+ return { store, model };
52
+ }
53
+
54
+ describe("DocumentSyncStatus", () => {
55
+ it("renders nothing while edits reach the server", () => {
56
+ setup(vi.fn<FakeEdit>());
57
+ expect(screen.queryByTestId("footer-document-sync")).toBeNull();
58
+ });
59
+
60
+ it("shows a rejected edit and resubmits it on click", async () => {
61
+ const accepted = document().revision;
62
+ const edit = vi
63
+ .fn<FakeEdit>()
64
+ .mockResolvedValueOnce(conflict)
65
+ .mockResolvedValueOnce({
66
+ ok: true,
67
+ status: 200,
68
+ data: {
69
+ document: { ...accepted, version: 2, etag: "document-2" },
70
+ cells: {},
71
+ },
72
+ });
73
+ const { store, model } = setup(edit);
74
+
75
+ await act(async () => {
76
+ store.set(model.editAtom, (tx) => tx.setCode("c1", "x = 2"));
77
+ await model.flush().catch(() => undefined);
78
+ });
79
+ const item = screen.getByTestId("footer-document-sync");
80
+ expect(item).toHaveAttribute(
81
+ "aria-label",
82
+ "Edits are not reaching the server",
83
+ );
84
+
85
+ await act(async () => {
86
+ fireEvent.click(item);
87
+ await Promise.resolve();
88
+ });
89
+ expect(edit).toHaveBeenCalledTimes(2);
90
+ expect(screen.queryByTestId("footer-document-sync")).toBeNull();
91
+ });
92
+ });
@@ -1,11 +1,12 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
- import { act, render, screen, within } from "@testing-library/react";
3
+ import { act, fireEvent, render, screen, within } from "@testing-library/react";
4
4
  import { createStore, Provider } from "jotai";
5
5
  import { describe, expect, it, vi } from "vitest";
6
6
  import { cellId } from "@/__tests__/branded";
7
7
  import { TooltipProvider } from "@/components/ui/tooltip";
8
8
  import { stagedAICellsAtom, visibleForTesting } from "@/core/ai/staged-cells";
9
+ import { scrollAndHighlightCell } from "../../../links/cell-link";
9
10
  import { PendingAICells } from "../pending-ai-cells";
10
11
 
11
12
  vi.mock("@/components/editor/cell/useDeleteCell", () => ({
@@ -14,8 +15,10 @@ vi.mock("@/components/editor/cell/useDeleteCell", () => ({
14
15
 
15
16
  vi.mock("@/core/cells/cells", () => ({
16
17
  getCellEditorView: vi.fn(),
18
+ }));
19
+ vi.mock("@/core/cells/actions", () => ({
17
20
  useCellActions: () => ({
18
- createNewCell: vi.fn(),
21
+ createCell: vi.fn(),
19
22
  updateCellCode: vi.fn(),
20
23
  }),
21
24
  }));
@@ -29,6 +32,62 @@ vi.mock("@/components/editor/links/cell-link", () => ({
29
32
  }));
30
33
 
31
34
  describe("PendingAICells", () => {
35
+ it("navigates between pending cells through named controls", () => {
36
+ const store = createStore();
37
+ const first = cellId("first-pending-cell");
38
+ const second = cellId("second-pending-cell");
39
+ const third = cellId("third-pending-cell");
40
+ store.set(
41
+ stagedAICellsAtom,
42
+ new Map([
43
+ [first, { type: "add_cell" }],
44
+ [second, { type: "add_cell" }],
45
+ [third, { type: "add_cell" }],
46
+ ]),
47
+ );
48
+
49
+ render(
50
+ <Provider store={store}>
51
+ <TooltipProvider>
52
+ <PendingAICells />
53
+ </TooltipProvider>
54
+ </Provider>,
55
+ );
56
+
57
+ const next = screen.getByRole("button", { name: "Next pending AI cell" });
58
+ const previous = screen.getByRole("button", {
59
+ name: "Previous pending AI cell",
60
+ });
61
+
62
+ fireEvent.click(next);
63
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
64
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
65
+
66
+ fireEvent.click(next);
67
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(second, "focus");
68
+ expect(screen.getByText("2 / 3")).toBeInTheDocument();
69
+
70
+ fireEvent.click(previous);
71
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
72
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
73
+
74
+ fireEvent.click(next);
75
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(second, "focus");
76
+ expect(screen.getByText("2 / 3")).toBeInTheDocument();
77
+
78
+ fireEvent.click(next);
79
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(third, "focus");
80
+ expect(screen.getByText("3 / 3")).toBeInTheDocument();
81
+
82
+ fireEvent.click(next);
83
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
84
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
85
+
86
+ fireEvent.click(previous);
87
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(third, "focus");
88
+ expect(screen.getByText("3 / 3")).toBeInTheDocument();
89
+ });
90
+
32
91
  it("disables bulk actions while cells are still generating", () => {
33
92
  const store = createStore();
34
93
  const generationId = Symbol("staged-cell-generation");
@@ -17,10 +17,10 @@ import {
17
17
  } from "@/components/ui/popover";
18
18
  import {
19
19
  type Attachment,
20
- agentAttachmentsAtom,
21
20
  attachmentLabel,
22
- } from "@/core/attachments/state";
23
- import { agentToast } from "@/core/attachments/toast";
21
+ useNotebookAttachments,
22
+ } from "@/core/notebook";
23
+ import { agentToast } from "@/components/editor/agent-toast";
24
24
  import { pairConfigAtom } from "@/core/config/pair";
25
25
  import { stableSessionIdAtom } from "@/core/kernel/session";
26
26
  import { API } from "@/core/network/api";
@@ -29,7 +29,7 @@ import { FooterItem } from "../footer-item";
29
29
 
30
30
  export function AttachmentStatus() {
31
31
  const hostEnabled = useAtomValue(pairConfigAtom)?.hostEnabled;
32
- const agents = useAtomValue(agentAttachmentsAtom);
32
+ const agents = useNotebookAttachments("agent");
33
33
  const [open, setOpen] = useState(false);
34
34
  const previous = useRef(new Set<string>());
35
35
  const automatic = useRef(false);
@@ -14,7 +14,7 @@ import {
14
14
  isAppConnected,
15
15
  isAppConnecting,
16
16
  isAppNotStarted,
17
- } from "@/core/websocket/connection-utils";
17
+ } from "@/core/session/connection-utils";
18
18
  import { useAsyncData } from "@/hooks/useAsyncData";
19
19
  import { useInterval } from "@/hooks/useInterval";
20
20
  import { Strings } from "@/utils/strings";
@@ -0,0 +1,54 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { atom, useAtomValue } from "jotai";
4
+ import { AlertCircleIcon } from "lucide-react";
5
+ import type React from "react";
6
+ import { activeDocumentModelAtom } from "@/core/session/active-notebook";
7
+ import { FooterItem } from "../footer-item";
8
+
9
+ /** The active notebook's document sync status, or null without a model. */
10
+ const syncStatusAtom = atom((get) => {
11
+ const model = get(activeDocumentModelAtom);
12
+ return model === null ? null : get(model.syncStatusAtom);
13
+ });
14
+
15
+ /**
16
+ * Shown when this client's edits stop reaching the server.
17
+ *
18
+ * Typing keeps working locally and Run still uses the editor's code, but
19
+ * other clients stop seeing the edits and Save fails until they are
20
+ * through. A rejected edit is retried on click.
21
+ */
22
+ export const DocumentSyncStatus: React.FC = () => {
23
+ const model = useAtomValue(activeDocumentModelAtom);
24
+ const status = useAtomValue(syncStatusAtom);
25
+ if (model === null || status === null || status.status !== "error") {
26
+ return null;
27
+ }
28
+ const retry = status.retryable
29
+ ? () => {
30
+ // The status shows the outcome, so a failed retry needs no handling.
31
+ model.retry().catch(() => undefined);
32
+ }
33
+ : undefined;
34
+ const detail = status.problem?.detail ?? status.error.message;
35
+ return (
36
+ <FooterItem
37
+ tooltip={
38
+ <div className="w-[260px] space-y-1">
39
+ <div>Edits are not reaching the server.</div>
40
+ <div className="text-muted-foreground">{detail}</div>
41
+ <div>
42
+ {retry ? "Click to retry." : "Reload the page to continue."}
43
+ </div>
44
+ </div>
45
+ }
46
+ selected={false}
47
+ onClick={retry}
48
+ aria-label="Edits are not reaching the server"
49
+ data-testid="footer-document-sync"
50
+ >
51
+ <AlertCircleIcon className="w-4 h-4 text-destructive" />
52
+ </FooterItem>
53
+ );
54
+ };
@@ -14,7 +14,7 @@ import type {
14
14
  LspRestartResponse,
15
15
  } from "@/core/network/types";
16
16
  import { isWasm } from "@/core/wasm/utils";
17
- import { isAppConnected } from "@/core/websocket/connection-utils";
17
+ import { isAppConnected } from "@/core/session/connection-utils";
18
18
  import { useAsyncData } from "@/hooks/useAsyncData";
19
19
  import { useInterval } from "@/hooks/useInterval";
20
20