@marimo-team/islands 0.25.3-dev24 → 0.25.3-dev245

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 (757) hide show
  1. package/dist/{ConnectedDataExplorerComponent-DfG3PRaJ.js → ConnectedDataExplorerComponent-BBihxHrr.js} +120 -116
  2. package/dist/{ErrorBoundary-LHjDFqXG.js → ErrorBoundary-DiPu-nOq.js} +16 -16
  3. package/dist/_baseSet-BfI5FjBM.js +328 -0
  4. package/dist/{abnfDiagram-VCTEODGH-B3w-XWX8.js → abnfDiagram-VCTEODGH-DBcJVxYP.js} +2 -2
  5. package/dist/any-language-editor-jGaR0EAo.js +75 -0
  6. package/dist/{architectureDiagram-5GKGNRK7-C18U9A6k.js → architectureDiagram-5GKGNRK7-BcHZXF4S.js} +1 -1
  7. package/dist/{arrays-CN1djC4k.js → arrays-DoQ-AILx.js} +15 -27
  8. package/dist/assets/__vite-browser-external-BKBpimif.js +1 -0
  9. package/dist/assets/worker-CgoMtD-_.js +113 -0
  10. package/dist/{blockDiagram-I7D4REHJ-Cg25UrxT.js → blockDiagram-I7D4REHJ-Dkhe6_PP.js} +1 -1
  11. package/dist/{button-CvjkwdSk.js → button-B6RjDKJx.js} +227 -217
  12. package/dist/{capabilities-CQ5IjhzL.js → capabilities-CkVzWJEX.js} +2 -2
  13. package/dist/{chat-ui-w8SVynkn.js → chat-ui-DgaVxFrc.js} +1309 -1307
  14. package/dist/{check-BMKBBrdc.js → check-Z_vtI3OL.js} +1 -1
  15. package/dist/{chunk-IMKFNOWR-CrS5bzqW.js → chunk-IMKFNOWR-CTZlwBAA.js} +1 -1
  16. package/dist/{chunk-SHT3W25Y-StpvPvUG.js → chunk-SHT3W25Y-53g_RZHm.js} +1 -1
  17. package/dist/{chunk-SVP7TREG-CBXjbYZC.js → chunk-SVP7TREG-P6abNGV_.js} +1 -1
  18. package/dist/{chunk-TICWLB2K-Dp6bIb-q.js → chunk-TICWLB2K-BTJJqkyF.js} +1 -1
  19. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  20. package/dist/{classDiagram-ZZMXUADV-BFYuxV5g.js → classDiagram-ZZMXUADV-BUGqOrTm.js} +2 -2
  21. package/dist/{classDiagram-v2-VYDZK3BY-D_tjXVMl.js → classDiagram-v2-VYDZK3BY-CXpQvO-s.js} +2 -2
  22. package/dist/code-visibility-B955Xe77.js +134 -0
  23. package/dist/{common-1pRFQo1N.js → common-CivG9Zso.js} +9980 -9749
  24. package/dist/config-DbjzjwxX.js +636 -0
  25. package/dist/{copy-icon-BgUO8jsi.js → copy-icon-CG57cxwb.js} +3 -3
  26. package/dist/createReducer-DePYJ0F3.js +63 -0
  27. package/dist/{cynefinDiagram-5FMLGOSQ-DSgYrHB2.js → cynefinDiagram-5FMLGOSQ-DOcnl8wW.js} +1 -1
  28. package/dist/{dagre-GXQ25YYZ-GYlruCBf.js → dagre-GXQ25YYZ-CuHR9QfG.js} +3 -3
  29. package/dist/{dagre-BGSfXNEb.js → dagre-mqiMAsFY.js} +531 -680
  30. package/dist/{data-grid-overlay-editor-UR0x91Ij.js → data-grid-overlay-editor-BUnD6AGE.js} +5 -5
  31. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  32. package/dist/{debounce-C3jyATpc.js → debounce-DrhhziJv.js} +2 -2
  33. package/dist/{diagram-S7CK7UJ4-DRu5LX5_.js → diagram-S7CK7UJ4-BPiG4kso.js} +1 -1
  34. package/dist/{diagram-UQ7AKVKN-DD7uiejP.js → diagram-UQ7AKVKN-_piBOvOC.js} +1 -1
  35. package/dist/{diagram-VX7I27RA-ByH9KAJJ.js → diagram-VX7I27RA-DCzVeJ8M.js} +1 -1
  36. package/dist/{diagram-Z3DM3KII-BsQHPRke.js → diagram-Z3DM3KII-D64CF1jQ.js} +1 -1
  37. package/dist/{dialog-Cc479ZEH.js → dialog-DsZ49mmU.js} +26 -26
  38. package/dist/{dist-Bvs24R_E.js → dist-11T5wq3_.js} +31 -30
  39. package/dist/{dist-F8Ep36rk.js → dist-17ugwGZc.js} +59 -58
  40. package/dist/{dist-BMQf5zIS.js → dist-7cRjf--y.js} +123 -122
  41. package/dist/{dist-diIx5qJW.js → dist-B8pkfkzc.js} +31 -30
  42. package/dist/{dist-CN_BeG2I.js → dist-B9UoM_wq.js} +58 -57
  43. package/dist/{dist-CLdpGQVr.js → dist-BGuMTs1i.js} +118 -117
  44. package/dist/{dist-BK4Oj99o.js → dist-BHW11fni.js} +18 -17
  45. package/dist/dist-BLY4m1Wi.js +850 -0
  46. package/dist/{dist-PXYI_yuP.js → dist-BNlcVoxs.js} +24 -23
  47. package/dist/{dist-D6d86eNJ.js → dist-BQYe5Bju.js} +47 -46
  48. package/dist/{dist-CkTTIdBe.js → dist-BX1mn8na.js} +337 -337
  49. package/dist/{dist-DiDibOYr.js → dist-Bh0PTKUW.js} +151 -150
  50. package/dist/{dist-CSJ0MDRw.js → dist-CKaXK92p.js} +56 -55
  51. package/dist/{dist-C8fAKSH1.js → dist-CLhwfhpJ.js} +53 -52
  52. package/dist/{dist-DoE4XJBC.js → dist-CYeqK4Xw.js} +9801 -10649
  53. package/dist/{dist-BLfKmR-l.js → dist-Cxlp3WH6.js} +16 -15
  54. package/dist/{dist-D-wwqlzJ.js → dist-DSJPGj42.js} +158 -157
  55. package/dist/{dist-BorMD-uX.js → dist-DU9ICsWZ.js} +75 -74
  56. package/dist/{dist-DBtSl3sf.js → dist-DWv0ALBi.js} +47 -46
  57. package/dist/{dist-OR1OnN-y.js → dist-DiBpWEsB.js} +31 -30
  58. package/dist/{dist-CrSP_H1M.js → dist-DiLFTlnl.js} +242 -242
  59. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  60. package/dist/{dist-CSbh1PNP.js → dist-cVDV5IuI.js} +23 -22
  61. package/dist/{dist-BuaRHxOc.js → dist-fRfaOoq8.js} +44 -43
  62. package/dist/{dist-Cg7gzw-a.js → dist-gTTI-9Px.js} +78 -77
  63. package/dist/dropdown-menu-BRiE7Vnp.js +6083 -0
  64. package/dist/{ebnfDiagram-PWID7BFC-CQRK5_eT.js → ebnfDiagram-PWID7BFC-BapQPML-.js} +2 -2
  65. package/dist/{erDiagram-RLTQ6QDP-BKtWrDVh.js → erDiagram-RLTQ6QDP-CNQvC2_Q.js} +1 -1
  66. package/dist/{error-banner-CFPHFw4h.js → error-banner-IsonnvA0.js} +84 -83
  67. package/dist/error-entries-CRPfxMqm.js +2505 -0
  68. package/dist/{esm-D2gTsXEV.js → esm-BgKjMvWw.js} +530 -529
  69. package/dist/{esm-AMFssoNp.js → esm-CHjVSexr.js} +328 -329
  70. package/dist/feedback-button-WpSRKq8m.js +374 -0
  71. package/dist/{flowDiagram-HODETNUW-1IN-ih2E.js → flowDiagram-HODETNUW-BBt6S6oe.js} +2 -2
  72. package/dist/{formats-X4mMQWzX.js → formats-BK4VefeF.js} +6 -6
  73. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  74. package/dist/{glide-data-editor-DPjObn47.js → glide-data-editor-CORsDYy3.js} +417 -415
  75. package/dist/{glide-utils-DM9zBO4W.js → glide-utils-CQddGpa-.js} +16 -16
  76. package/dist/{globals-DOHFyDcV.js → globals-B3f1L5ta.js} +3 -3
  77. package/dist/{graphlib-BmY_CCV9.js → graphlib-nlwT3Wvy.js} +177 -220
  78. package/dist/hasIn-C20JsQ2m.js +28 -0
  79. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  80. package/dist/{error-entries-Damuuclv.js → html-to-image-Gx4q3ZoD.js} +26911 -26769
  81. package/dist/{infoDiagram-27XIBGKW-CkB-zr4p.js → infoDiagram-27XIBGKW-Hd8Ifz82.js} +1 -1
  82. package/dist/isArrayLikeObject-B-HXCsex.js +626 -0
  83. package/dist/isEmpty-CdX7MTE1.js +14 -0
  84. package/dist/{ishikawaDiagram-5VMMS53U-wlyLjBmM.js → ishikawaDiagram-5VMMS53U-CP7hnHWx.js} +1 -1
  85. package/dist/{kanban-definition-UXKFOSKX-C2usY0i-.js → kanban-definition-UXKFOSKX-CX91sa9b.js} +1 -1
  86. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  87. package/dist/{loader-BQOfjTIJ.js → loader-BisjOLiW.js} +12 -12
  88. package/dist/main.js +10234 -8104
  89. package/dist/{mermaid-BA-Z6wGL.js → mermaid-Djb3Ni9t.js} +54 -54
  90. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  91. package/dist/{mindmap-definition-YA3MSWOX-nKOWS8RR.js → mindmap-definition-YA3MSWOX-DgCNo07O.js} +1 -1
  92. package/dist/{now-jM8-M7IR.js → now-BKZCLE5P.js} +12 -13
  93. package/dist/{number-overlay-editor-B3cSnrr6.js → number-overlay-editor-B5LIxGos.js} +3 -3
  94. package/dist/{dist-CN8zI-XC.js → objectWithoutPropertiesLoose-ai5NUqGa.js} +262 -240
  95. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  96. package/dist/{once-BtE-wKRL.js → once-NnpX_HnX.js} +1 -1
  97. package/dist/{pegDiagram-XKGWAZYB-Dac9EIKD.js → pegDiagram-XKGWAZYB-SGJhnIKo.js} +2 -2
  98. package/dist/pick-tCszMUbm.js +35 -0
  99. package/dist/{pieDiagram-E7YTZNPT-B7S-DsW_.js → pieDiagram-E7YTZNPT-1stXIWCQ.js} +1 -1
  100. package/dist/precisionRound-DhAzeRcm.js +49 -0
  101. package/dist/{process-output-CTPTtxXB.js → process-output-D2ISIyIs.js} +4 -4
  102. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  103. package/dist/{railroadDiagram-O6MQD6OU-Cfa5Q-hr.js → railroadDiagram-O6MQD6OU-DY1mojzB.js} +2 -2
  104. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  105. package/dist/{requirementDiagram-BXWQKSXE-d8zXCJIk.js → requirementDiagram-BXWQKSXE-gHzhUp0D.js} +1 -1
  106. package/dist/{reveal-component-CFFzSsPX.js → reveal-component-DvM7Lxss.js} +430 -407
  107. package/dist/schemas-CdscWNy_.js +4747 -0
  108. package/dist/select-BSxlITBo.js +1200 -0
  109. package/dist/{sequenceDiagram-WJ2MYXX4-b6pei3lJ.js → sequenceDiagram-WJ2MYXX4-rjPtD9fB.js} +1 -1
  110. package/dist/{spec-BJzgmvVa.js → spec-Dq8RNplp.js} +41 -11
  111. package/dist/{stateDiagram-D77RDMKH-Be-xSpaZ.js → stateDiagram-D77RDMKH-DMNxjUxV.js} +4 -4
  112. package/dist/{stateDiagram-v2-MP3YSRHH-5Y-FxesJ.js → stateDiagram-v2-MP3YSRHH-CumAaqGo.js} +2 -2
  113. package/dist/static-state-BldGd99T.js +72 -0
  114. package/dist/strings-BGW-fXVJ.js +507 -0
  115. package/dist/style.css +1 -1
  116. package/dist/{swimlanes-42K2YHIH-nRRl4NU0.js → swimlanes-42K2YHIH-BU14_Tej.js} +1 -1
  117. package/dist/{swimlanesDiagram-VR7AAH4N-BS79510V.js → swimlanesDiagram-VR7AAH4N-CieAnPqJ.js} +2 -2
  118. package/dist/{swiper-component-DKntyn8p.js → swiper-component-Dir1o8ml.js} +2 -2
  119. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  120. package/dist/{timeline-definition-24CTP7MA-Cfzf8WPf.js → timeline-definition-24CTP7MA-CrVj4E1K.js} +1 -1
  121. package/dist/toast-BWIPi3aF.js +491 -0
  122. package/dist/{tooltip-CtBZ5q8p.js → tooltip-C3UnvsRe.js} +329 -305
  123. package/dist/use-restore-focus-BWO0z1lo.js +1672 -0
  124. package/dist/{useAsyncData-BD534XmW.js → useAsyncData-DsZMvNXn.js} +15 -14
  125. package/dist/{useDateFormatter-DkajGwwd.js → useDateFormatter-ChB4eyim.js} +2 -2
  126. package/dist/{useDeepCompareMemoize-C72_RNsu.js → useDeepCompareMemoize-KMBBux8C.js} +1 -1
  127. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-cMNQ80CK.js} +1 -1
  128. package/dist/useLifecycle-BquH9ET-.js +112 -0
  129. package/dist/useTheme-DgOLVISn.js +337 -0
  130. package/dist/utils-DXkj82sK.js +813 -0
  131. package/dist/{vega-component-DfcYgj-Y.js → vega-component-Bwg_Kuh-.js} +171 -170
  132. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  133. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  134. package/dist/{vennDiagram-4TSXK5OY-CeY6Z1xZ.js → vennDiagram-4TSXK5OY-ZLxQ8yhm.js} +1 -1
  135. package/dist/{wardleyDiagram-VM6X3IG4-Cxs9NtVi.js → wardleyDiagram-VM6X3IG4-D88pcLY1.js} +1 -1
  136. package/dist/{xychartDiagram-S5SC5T6Z-DNWbvPK7.js → xychartDiagram-S5SC5T6Z-_13d4g8g.js} +2 -2
  137. package/package.json +16 -20
  138. package/src/__mocks__/document.tsx +63 -0
  139. package/src/__mocks__/notebook.ts +3 -4
  140. package/src/__mocks__/partial.ts +26 -0
  141. package/src/__mocks__/requests.ts +1 -4
  142. package/src/__tests__/main.test.tsx +24 -1
  143. package/src/__tests__/module-entry.test.ts +44 -0
  144. package/src/__tests__/mount.test.ts +1 -21
  145. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  146. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  147. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  148. package/src/components/app-config/ai-config.tsx +155 -149
  149. package/src/components/app-config/app-config-button.tsx +6 -5
  150. package/src/components/app-config/app-config-form.tsx +5 -9
  151. package/src/components/app-config/common.tsx +7 -9
  152. package/src/components/app-config/data-form.tsx +112 -113
  153. package/src/components/app-config/mcp-config.tsx +30 -53
  154. package/src/components/app-config/user-config-form.css +91 -0
  155. package/src/components/app-config/user-config-form.tsx +266 -349
  156. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  157. package/src/components/chat/__tests__/notebook-access-selector.test.tsx +64 -0
  158. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  159. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  160. package/src/components/chat/acp/blocks.tsx +3 -2
  161. package/src/components/chat/acp/session-tabs.tsx +8 -3
  162. package/src/components/chat/capabilities-popover.tsx +1 -5
  163. package/src/components/chat/chat-components.tsx +2 -0
  164. package/src/components/chat/chat-panel.tsx +23 -104
  165. package/src/components/chat/chat-utils.ts +4 -8
  166. package/src/components/chat/notebook-access-selector.stories.tsx +22 -0
  167. package/src/components/chat/notebook-access-selector.tsx +100 -0
  168. package/src/components/data-table/TableBottomBar.tsx +4 -1
  169. package/src/components/data-table/TableTopBar.tsx +2 -0
  170. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  171. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  172. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  173. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  174. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  175. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  176. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  177. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  178. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  179. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  180. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  181. package/src/components/data-table/cell-selection/feature.ts +2 -2
  182. package/src/components/data-table/cell-selection/types.ts +2 -2
  183. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  184. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  185. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  186. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  187. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  188. package/src/components/data-table/charts/charts.tsx +40 -30
  189. package/src/components/data-table/charts/storage.ts +16 -5
  190. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  191. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  192. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  193. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  194. package/src/components/data-table/columns.tsx +45 -46
  195. package/src/components/data-table/context-menu.tsx +36 -20
  196. package/src/components/data-table/data-table.tsx +17 -1
  197. package/src/components/data-table/date-popover.tsx +6 -4
  198. package/src/components/data-table/export-actions.tsx +76 -48
  199. package/src/components/data-table/pagination.tsx +44 -33
  200. package/src/components/data-table/range-focus/atoms.ts +4 -6
  201. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  202. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  203. package/src/components/data-table/renderers.tsx +16 -7
  204. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  205. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  206. package/src/components/data-table/search-highlight.tsx +79 -0
  207. package/src/components/data-table/types.ts +1 -0
  208. package/src/components/data-table/url-detector.tsx +49 -9
  209. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  210. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  211. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  212. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  213. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  214. package/src/components/datasources/column-preview.tsx +17 -12
  215. package/src/components/datasources/datasources.tsx +56 -43
  216. package/src/components/datasources/utils.ts +32 -23
  217. package/src/components/debugger/debugger-code.tsx +5 -0
  218. package/src/components/dependency-graph/custom-node.tsx +1 -1
  219. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  220. package/src/components/dependency-graph/elements.ts +24 -21
  221. package/src/components/dependency-graph/minimap-content.tsx +13 -19
  222. package/src/components/dependency-graph/panels.tsx +4 -2
  223. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  224. package/src/components/editor/SortableCell.tsx +1 -3
  225. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  226. package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
  227. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -14
  228. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  229. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -2
  230. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  231. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  232. package/src/components/editor/actions/__tests__/pair-with-agent-commands.test.ts +3 -3
  233. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +40 -10
  234. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +6 -10
  235. package/src/components/editor/actions/__tests__/useCellActionButton.test.tsx +113 -0
  236. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  237. package/src/components/editor/actions/pair-with-agent-commands.ts +10 -6
  238. package/src/components/editor/actions/pair-with-agent-modal.tsx +24 -21
  239. package/src/components/editor/actions/pair-with-agent.tsx +15 -10
  240. package/src/components/editor/actions/useCellActionButton.tsx +36 -31
  241. package/src/components/editor/actions/useConfigActions.tsx +3 -4
  242. package/src/components/editor/actions/useNotebookActions.tsx +25 -29
  243. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  244. package/src/components/editor/actions/useSetCodeVisibility.ts +11 -12
  245. package/src/components/editor/actions/useSetOutputsExpanded.ts +14 -19
  246. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  247. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  248. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  249. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  250. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  251. package/src/components/editor/alerts/stdin-blocking-alert.tsx +1 -1
  252. package/src/components/editor/app-container.tsx +1 -1
  253. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  254. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  255. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  256. package/src/components/editor/cell/RunButton.tsx +2 -2
  257. package/src/components/editor/cell/StagedAICell.tsx +10 -2
  258. package/src/components/editor/cell/StopButton.tsx +2 -2
  259. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +4 -2
  260. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  261. package/src/components/editor/cell/cell-actions.tsx +12 -1
  262. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  263. package/src/components/editor/cell/code/cell-editor.tsx +46 -165
  264. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  265. package/src/components/editor/cell/collapse.tsx +11 -4
  266. package/src/components/editor/cell/useAddCell.ts +4 -8
  267. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  268. package/src/components/editor/cell/useRunCells.ts +2 -5
  269. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  270. package/src/components/editor/cell/useSplitCell.tsx +10 -6
  271. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  272. package/src/components/editor/chrome/components/feedback-button.tsx +2 -8
  273. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  274. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  275. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  276. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  277. package/src/components/editor/chrome/panels/dependency-graph-panel.tsx +1 -1
  278. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  279. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  280. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  281. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  282. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  283. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  284. package/src/components/editor/chrome/panels/session-panel.tsx +1 -1
  285. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  286. package/src/components/editor/chrome/panels/variable-panel.tsx +1 -5
  287. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +6 -9
  288. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  289. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +62 -3
  290. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
  291. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  292. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  293. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  294. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  295. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  296. package/src/components/editor/chrome/wrapper/footer.tsx +2 -2
  297. package/src/components/editor/chrome/wrapper/minimap-state.ts +1 -1
  298. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  299. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  300. package/src/components/editor/columns/__tests__/cell-column.test.tsx +1 -1
  301. package/src/components/editor/columns/cell-column.tsx +1 -1
  302. package/src/components/editor/columns/sortable-column.tsx +2 -2
  303. package/src/components/editor/connections/components.tsx +5 -10
  304. package/src/components/editor/controls/Controls.tsx +4 -4
  305. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  306. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  307. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  308. package/src/components/editor/controls/command-palette.tsx +22 -10
  309. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  310. package/src/components/editor/documentation.css +34 -11
  311. package/src/components/editor/dynamic-favicon.tsx +21 -40
  312. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +25 -20
  313. package/src/components/editor/errors/auto-fix.tsx +4 -8
  314. package/src/components/editor/errors/send-error-report-button.tsx +3 -6
  315. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  316. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  317. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  318. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  319. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  320. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  321. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  322. package/src/components/editor/header/__tests__/filename-form.test.tsx +1 -4
  323. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  324. package/src/components/editor/header/app-header.tsx +1 -1
  325. package/src/components/editor/header/filename-form.tsx +1 -1
  326. package/src/components/editor/header/status.tsx +40 -20
  327. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  328. package/src/components/editor/links/cell-link-list.tsx +1 -1
  329. package/src/components/editor/links/cell-link.tsx +17 -7
  330. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  331. package/src/components/editor/navigation/__tests__/navigation.test.ts +108 -85
  332. package/src/components/editor/navigation/__tests__/selection.test.ts +43 -21
  333. package/src/components/editor/navigation/clipboard.ts +17 -14
  334. package/src/components/editor/navigation/focus-utils.ts +23 -4
  335. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +17 -33
  336. package/src/components/editor/navigation/navigation.ts +71 -70
  337. package/src/components/editor/navigation/selection.ts +30 -29
  338. package/src/components/editor/notebook-banner.tsx +1 -0
  339. package/src/components/editor/notebook-cell.tsx +29 -32
  340. package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
  341. package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
  342. package/src/components/editor/output/Outputs.css +8 -0
  343. package/src/components/editor/output/VideoOutput.tsx +6 -0
  344. package/src/components/editor/output/__tests__/VideoOutput.test.tsx +131 -0
  345. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  346. package/src/components/editor/output/ansi-reduce.ts +33 -20
  347. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  348. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  349. package/src/components/editor/package-alert.tsx +14 -7
  350. package/src/components/editor/renderers/cell-array.tsx +22 -33
  351. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  352. package/src/components/editor/renderers/slides-layout/__tests__/plugin.test.ts +0 -1
  353. package/src/components/editor/renderers/slides-layout/slides-layout.tsx +1 -1
  354. package/src/components/editor/renderers/vertical-layout/__tests__/useFocusFirstEditor.test.ts +4 -4
  355. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  356. package/src/components/editor/renderers/vertical-layout/useFocusFirstEditor.ts +1 -1
  357. package/src/components/editor/renderers/vertical-layout/vertical-layout.tsx +1 -1
  358. package/src/components/editor/stdin-blocking-alert.tsx +1 -1
  359. package/src/components/home/components.tsx +17 -31
  360. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  361. package/src/components/icons/large-spinner.tsx +3 -14
  362. package/src/components/icons/spinner.tsx +1 -1
  363. package/src/components/markdown/markdown-renderer.tsx +4 -8
  364. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  365. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  366. package/src/components/pages/edit-page.tsx +19 -17
  367. package/src/components/pages/home-page.tsx +164 -91
  368. package/src/components/scratchpad/scratchpad.test.tsx +89 -0
  369. package/src/components/scratchpad/scratchpad.tsx +43 -32
  370. package/src/components/scratchpad/state.ts +7 -0
  371. package/src/components/slides/__tests__/minimap-actions.test.tsx +22 -29
  372. package/src/components/slides/__tests__/minimap.test.ts +38 -19
  373. package/src/components/slides/__tests__/slide-scroll-hint.test.tsx +37 -0
  374. package/src/components/slides/minimap.tsx +17 -12
  375. package/src/components/slides/reveal-component.tsx +58 -26
  376. package/src/components/slides/reveal-slides.css +5 -4
  377. package/src/components/slides/slide-cell-view.tsx +4 -5
  378. package/src/components/slides/slide-form.tsx +1 -0
  379. package/src/components/slides/slide-scroll-hint.tsx +3 -1
  380. package/src/components/sort/SortableCellsProvider.tsx +8 -25
  381. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  382. package/src/components/tracing/tracing.tsx +46 -42
  383. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  384. package/src/components/ui/__tests__/fullscreen.test.tsx +44 -2
  385. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  386. package/src/components/ui/alert-dialog.tsx +1 -1
  387. package/src/components/ui/draggable-popover.tsx +24 -2
  388. package/src/components/ui/fullscreen.tsx +25 -1
  389. package/src/components/ui/range-slider.tsx +22 -9
  390. package/src/core/MarimoApp.tsx +3 -1
  391. package/src/core/ai/__tests__/staged-cells.test.ts +32 -33
  392. package/src/core/ai/context/providers/__tests__/cell-output.test.ts +8 -12
  393. package/src/core/ai/context/providers/cell-output.ts +2 -2
  394. package/src/core/ai/context/providers/datasource.ts +6 -2
  395. package/src/core/ai/staged-cells.ts +18 -19
  396. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +40 -88
  397. package/src/core/ai/tools/__tests__/registry.test.ts +17 -0
  398. package/src/core/ai/tools/base.ts +18 -15
  399. package/src/core/ai/tools/edit-notebook-tool.ts +86 -80
  400. package/src/core/ai/tools/registry.ts +13 -10
  401. package/src/core/ai/tools/sample-tool.ts +6 -7
  402. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -37
  403. package/src/core/cells/__tests__/actions.test.ts +624 -35
  404. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  405. package/src/core/cells/__tests__/arrange.test.ts +432 -0
  406. package/src/core/cells/__tests__/cell.test.ts +39 -0
  407. package/src/core/cells/__tests__/cells.test.ts +251 -2540
  408. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  409. package/src/core/cells/__tests__/document.test.ts +164 -0
  410. package/src/core/cells/__tests__/editor.test.ts +57 -0
  411. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  412. package/src/core/cells/__tests__/harness.ts +106 -0
  413. package/src/core/cells/__tests__/layout.test.ts +177 -0
  414. package/src/core/cells/__tests__/notebook-outline.test.ts +122 -0
  415. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  416. package/src/core/cells/__tests__/session.test.ts +445 -211
  417. package/src/core/cells/__tests__/utils.test.ts +9 -108
  418. package/src/core/cells/actions.ts +675 -6
  419. package/src/core/cells/add-missing-import.ts +47 -43
  420. package/src/core/cells/arrange.ts +529 -0
  421. package/src/core/cells/cell.ts +14 -3
  422. package/src/core/cells/cells.ts +126 -1363
  423. package/src/core/cells/columns.ts +38 -0
  424. package/src/core/cells/document.ts +178 -0
  425. package/src/core/cells/edit.ts +30 -0
  426. package/src/core/cells/editor.ts +82 -0
  427. package/src/core/cells/layout.ts +221 -0
  428. package/src/core/cells/outputs.ts +12 -28
  429. package/src/core/cells/pending-delete-service.ts +3 -7
  430. package/src/core/cells/session.ts +76 -53
  431. package/src/core/cells/types.ts +22 -6
  432. package/src/core/cells/utils.ts +10 -8
  433. package/src/core/codemirror/__tests__/format.test.ts +0 -34
  434. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  435. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  436. package/src/core/codemirror/ai/resources.ts +1 -1
  437. package/src/core/codemirror/cells/__tests__/extensions.test.ts +2 -6
  438. package/src/core/codemirror/cells/extensions.ts +30 -28
  439. package/src/core/codemirror/cells/state.ts +1 -6
  440. package/src/core/codemirror/copilot/__tests__/getCodes.test.ts +6 -7
  441. package/src/core/codemirror/copilot/getCodes.ts +2 -2
  442. package/src/core/codemirror/format.ts +6 -23
  443. package/src/core/codemirror/keymaps/vim.ts +4 -8
  444. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  445. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  446. package/src/core/codemirror/language/cell-code.ts +114 -0
  447. package/src/core/codemirror/language/extension.ts +52 -35
  448. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  449. package/src/core/codemirror/language/languages/python.ts +16 -10
  450. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  451. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  452. package/src/core/codemirror/language/types.ts +7 -7
  453. package/src/core/codemirror/language/utils.ts +9 -39
  454. package/src/core/codemirror/lsp/__tests__/notebook-lsp.test.ts +1 -2
  455. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  456. package/src/core/codemirror/lsp/utils.ts +1 -1
  457. package/src/core/codemirror/misc/__tests__/paste.test.ts +3 -1
  458. package/src/core/codemirror/misc/paste.ts +1 -1
  459. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  460. package/src/core/codemirror/readonly/extension.ts +2 -2
  461. package/src/core/codemirror/utils.ts +1 -0
  462. package/src/core/config/__tests__/app-config.test.ts +64 -0
  463. package/src/core/config/__tests__/config-schema.test.ts +56 -6
  464. package/src/core/config/config-schema.ts +17 -4
  465. package/src/core/config/config.ts +65 -20
  466. package/src/core/config/feature-flag.tsx +0 -2
  467. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  468. package/src/core/datasets/data-source-discovery.ts +4 -6
  469. package/src/core/documentation/doc-lookup.ts +1 -1
  470. package/src/core/edit-app.tsx +21 -28
  471. package/src/core/errors/error-entries.ts +4 -1
  472. package/src/core/export/hooks.ts +1 -1
  473. package/src/core/export/layout.ts +2 -2
  474. package/src/core/hotkeys/actions.ts +2 -4
  475. package/src/core/islands/__tests__/bootstrap.test.ts +69 -17
  476. package/src/core/islands/__tests__/bridge.test.ts +85 -27
  477. package/src/core/islands/bootstrap.ts +46 -53
  478. package/src/core/islands/bridge.ts +84 -25
  479. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  480. package/src/core/islands/components/web-components.tsx +1 -1
  481. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  482. package/src/core/islands/worker/controller.ts +4 -10
  483. package/src/core/islands/worker/worker.tsx +33 -33
  484. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  485. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  486. package/src/core/kernel/handlers.ts +19 -72
  487. package/src/core/meta/__tests__/code-visibility.test.tsx +78 -70
  488. package/src/core/meta/code-visibility.ts +18 -9
  489. package/src/core/network/__tests__/connection.test.ts +1 -1
  490. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  491. package/src/core/network/connection-notice.ts +1 -1
  492. package/src/core/network/connection.ts +1 -1
  493. package/src/core/network/requests-lazy.ts +2 -5
  494. package/src/core/network/requests-network.ts +5 -31
  495. package/src/core/network/requests-static.ts +1 -4
  496. package/src/core/network/requests-toasting.tsx +1 -4
  497. package/src/core/network/types.ts +6 -14
  498. package/src/core/network/useConnectionNotice.ts +8 -2
  499. package/src/core/notebook/__tests__/operations.test.ts +51 -0
  500. package/src/core/notebook/attachments.ts +38 -0
  501. package/src/core/notebook/document/__tests__/codemirror.test.ts +758 -0
  502. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  503. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  504. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  505. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  506. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  507. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  508. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  509. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  510. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  511. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  512. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  513. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  514. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  515. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  516. package/src/core/notebook/document/__tests__/user.ts +61 -0
  517. package/src/core/notebook/document/codemirror.ts +216 -0
  518. package/src/core/notebook/document/draft.ts +502 -0
  519. package/src/core/notebook/document/hooks.tsx +118 -0
  520. package/src/core/notebook/document/index.ts +24 -0
  521. package/src/core/notebook/document/model.ts +102 -0
  522. package/src/core/notebook/document/select.ts +41 -0
  523. package/src/core/notebook/document/state.ts +378 -0
  524. package/src/core/notebook/document/sync.ts +369 -0
  525. package/src/core/notebook/index.ts +19 -0
  526. package/src/core/notebook/operations.ts +30 -0
  527. package/src/core/notebook/replica/AGENTS.md +11 -0
  528. package/src/core/{host → notebook/replica}/__tests__/attachments.test.ts +9 -10
  529. package/src/core/{host → notebook/replica}/__tests__/document.test.ts +18 -16
  530. package/src/core/notebook/replica/__tests__/executions.test.ts +201 -0
  531. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  532. package/src/core/notebook/replica/__tests__/replica.test.ts +559 -0
  533. package/src/core/notebook/replica/__tests__/results.test.ts +193 -0
  534. package/src/core/notebook/replica/__tests__/runtime.test.ts +78 -0
  535. package/src/core/notebook/replica/backend.ts +19 -0
  536. package/src/core/notebook/replica/connection.ts +42 -0
  537. package/src/core/notebook/replica/index.ts +20 -0
  538. package/src/core/notebook/replica/replica.ts +336 -0
  539. package/src/core/{host → notebook/replica}/resources/attachments.ts +6 -3
  540. package/src/core/{host → notebook/replica}/resources/document.ts +3 -3
  541. package/src/core/notebook/replica/resources/executions.ts +157 -0
  542. package/src/core/notebook/replica/resources/results.ts +48 -0
  543. package/src/core/notebook/replica/resources/runtime.ts +14 -0
  544. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  545. package/src/core/packages/useSandboxController.ts +1 -1
  546. package/src/core/run-app.tsx +48 -51
  547. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  548. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  549. package/src/core/runtime/__tests__/runtime.test.ts +32 -74
  550. package/src/core/runtime/adapter.ts +1 -1
  551. package/src/core/runtime/config.ts +2 -2
  552. package/src/core/runtime/runtime.ts +33 -46
  553. package/src/core/runtime/types.ts +0 -2
  554. package/src/core/saving/__tests__/document.test.tsx +123 -0
  555. package/src/core/saving/__tests__/state.test.ts +111 -0
  556. package/src/core/saving/__tests__/useAutoSave.test.tsx +67 -0
  557. package/src/core/saving/document.ts +33 -0
  558. package/src/core/saving/filename.ts +1 -1
  559. package/src/core/saving/save-component.tsx +39 -60
  560. package/src/core/saving/state.ts +38 -39
  561. package/src/core/saving/useAutoSave.ts +17 -53
  562. package/src/core/session/__tests__/backend.test.ts +236 -0
  563. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +184 -0
  564. package/src/core/{websocket → session}/__tests__/useNotebookSession.hook.test.tsx +213 -262
  565. package/src/core/session/active-notebook.ts +28 -0
  566. package/src/core/session/backend.ts +123 -0
  567. package/src/core/session/embedded.ts +7 -0
  568. package/src/core/{websocket → session}/types.ts +2 -2
  569. package/src/core/session/useNotebookReplica.ts +74 -0
  570. package/src/core/{websocket → session}/useNotebookSession.tsx +77 -64
  571. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  572. package/src/core/state/jotai.ts +1 -1
  573. package/src/core/static/__tests__/files.test.ts +38 -10
  574. package/src/core/static/files.ts +2 -0
  575. package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
  576. package/src/core/transports/__tests__/server.test.ts +408 -0
  577. package/src/core/transports/__tests__/static.test.ts +306 -0
  578. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  579. package/src/core/{websocket → transports}/connection-state.ts +9 -16
  580. package/src/core/transports/server.ts +152 -0
  581. package/src/core/transports/static.ts +166 -0
  582. package/src/core/transports/wasm.ts +97 -0
  583. package/src/core/vscode/__tests__/vscode-bindings.test.ts +125 -0
  584. package/src/core/vscode/vscode-bindings.ts +14 -2
  585. package/src/core/wasm/__tests__/bridge.test.ts +105 -65
  586. package/src/core/wasm/__tests__/worker.test.ts +187 -0
  587. package/src/core/wasm/bridge.ts +69 -72
  588. package/src/core/wasm/rpc.ts +15 -3
  589. package/src/core/wasm/store.ts +3 -3
  590. package/src/core/wasm/worker/bootstrap.ts +55 -51
  591. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  592. package/src/core/wasm/worker/save-worker.ts +9 -10
  593. package/src/core/wasm/worker/types.ts +29 -18
  594. package/src/core/wasm/worker/worker.ts +89 -45
  595. package/src/css/app/codemirror.css +1 -1
  596. package/src/css/common.css +1 -1
  597. package/src/css/globals.css +66 -22
  598. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  599. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  600. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  601. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  602. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  603. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  604. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  605. package/src/hooks/useAsyncData.ts +3 -3
  606. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  607. package/src/hooks/useDebounce.ts +6 -10
  608. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  609. package/src/hooks/useDelayElapsed.ts +14 -10
  610. package/src/hooks/useElapsedTime.ts +3 -3
  611. package/src/hooks/useEventListener.ts +1 -2
  612. package/src/hooks/useInputHistory.ts +4 -7
  613. package/src/hooks/useInternalStateWithSync.ts +6 -6
  614. package/src/hooks/useResizeObserver.ts +4 -7
  615. package/src/hooks/useScript.ts +67 -70
  616. package/src/hooks/useStateForKey.ts +93 -0
  617. package/src/mount.tsx +20 -15
  618. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  619. package/src/plugins/core/registerReactComponent.tsx +1 -1
  620. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  621. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  622. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  623. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  624. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  625. package/src/plugins/impl/FormPlugin.tsx +1 -1
  626. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  627. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  628. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  629. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
  630. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  631. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  632. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  633. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  634. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  635. package/src/plugins/impl/anywidget/model.ts +3 -1
  636. package/src/plugins/impl/anywidget/registry.ts +18 -14
  637. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  638. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  639. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  640. package/src/plugins/impl/chat/chat-ui.tsx +14 -11
  641. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  642. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  643. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  644. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  645. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  646. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  647. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  648. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  649. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  650. package/src/plugins/impl/data-frames/panel.tsx +44 -16
  651. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  652. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  653. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  654. package/src/plugins/impl/panel/utils.ts +17 -0
  655. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  656. package/src/plugins/impl/plotly/selection.ts +25 -19
  657. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  658. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  659. package/src/plugins/layout/DownloadPlugin.tsx +7 -11
  660. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  661. package/src/plugins/layout/__test__/DownloadPlugin.test.tsx +205 -0
  662. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  663. package/src/stories/button.stories.tsx +1 -0
  664. package/src/stories/cell.stories.tsx +2 -5
  665. package/src/stories/data-table.stories.tsx +31 -0
  666. package/src/stories/editor.stories.tsx +24 -17
  667. package/src/stories/form-wrapper.stories.tsx +2 -2
  668. package/src/stories/layout/vertical/one-column.stories.tsx +28 -23
  669. package/src/stories/runtime-story.tsx +23 -17
  670. package/src/stories/select.stories.tsx +2 -2
  671. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  672. package/src/utils/__tests__/arrays.test.ts +35 -0
  673. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  674. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
  675. package/src/utils/__tests__/download.test.tsx +183 -23
  676. package/src/utils/__tests__/numbers.test.ts +105 -1
  677. package/src/utils/__tests__/objects.test.ts +21 -0
  678. package/src/utils/arrays.ts +11 -2
  679. package/src/utils/blob.ts +1 -1
  680. package/src/utils/createReducer.ts +53 -26
  681. package/src/utils/download.ts +67 -6
  682. package/src/utils/maps.ts +0 -3
  683. package/src/utils/numbers.ts +79 -18
  684. package/src/utils/objects.ts +4 -9
  685. package/src/utils/tracer.ts +9 -5
  686. package/src/utils/zod-utils.ts +1 -3
  687. package/dist/_baseMerge-BiGRwuqv.js +0 -165
  688. package/dist/any-language-editor-D4q-IFcM.js +0 -76
  689. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  690. package/dist/assets/worker-HPY9fbhz.js +0 -114
  691. package/dist/code-visibility-HTfE7PV1.js +0 -138
  692. package/dist/dropdown-menu-DHP9P4VE.js +0 -4678
  693. package/dist/feedback-button-BL7Ed1J3.js +0 -379
  694. package/dist/hasIn-o2Us0uVx.js +0 -132
  695. package/dist/html-to-image-BdA59IUD.js +0 -3251
  696. package/dist/isArrayLikeObject-BZl-15ir.js +0 -450
  697. package/dist/isEmpty-BSZ1LaUo.js +0 -35
  698. package/dist/isSymbol-kQcfYbNX.js +0 -8
  699. package/dist/label-DoysmLMW.js +0 -281
  700. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  701. package/dist/objectWithoutPropertiesLoose-BRDcTsfB.js +0 -23
  702. package/dist/pick-CJeoxYOS.js +0 -51
  703. package/dist/precisionRound-DMIVQrt2.js +0 -97
  704. package/dist/select-DemoKT4G.js +0 -1344
  705. package/dist/toDate-B5mIgxwZ.js +0 -1185
  706. package/dist/toast-PkdZraM8.js +0 -1830
  707. package/dist/use-restore-focus-CKjfFjLM.js +0 -6369
  708. package/dist/useLifecycle-BGWzTwHK.js +0 -169
  709. package/dist/useTheme-BhQvec43.js +0 -904
  710. package/src/components/editor/cell/code/__tests__/rtc-editor-config.test.ts +0 -50
  711. package/src/components/editor/chrome/wrapper/footer-items/rtc-status.tsx +0 -65
  712. package/src/components/editor/navigation/__tests__/state.test.ts +0 -81
  713. package/src/components/editor/navigation/state.ts +0 -65
  714. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +0 -656
  715. package/src/core/ai/tools/run-cells-tool.ts +0 -303
  716. package/src/core/attachments/state.ts +0 -18
  717. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  718. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  719. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  720. package/src/core/cells/document-changes.ts +0 -944
  721. package/src/core/codemirror/rtc/extension.ts +0 -453
  722. package/src/core/codemirror/rtc/loro/__tests__/sync.test.ts +0 -52
  723. package/src/core/codemirror/rtc/loro/awareness.ts +0 -531
  724. package/src/core/codemirror/rtc/loro/colors.ts +0 -38
  725. package/src/core/codemirror/rtc/loro/sync.ts +0 -168
  726. package/src/core/host/AGENTS.md +0 -11
  727. package/src/core/host/__tests__/events.test.ts +0 -46
  728. package/src/core/host/__tests__/fixtures.ts +0 -39
  729. package/src/core/host/__tests__/notebook-client.test.ts +0 -433
  730. package/src/core/host/__tests__/protocol.test.ts +0 -214
  731. package/src/core/host/__tests__/stream.test.ts +0 -209
  732. package/src/core/host/connection.ts +0 -81
  733. package/src/core/host/events.ts +0 -61
  734. package/src/core/host/notebook-client.ts +0 -290
  735. package/src/core/host/protocol.ts +0 -1943
  736. package/src/core/host/resources/resource.ts +0 -47
  737. package/src/core/host/stream.ts +0 -211
  738. package/src/core/rtc/state.ts +0 -34
  739. package/src/core/state/__tests__/jotai.test.ts +0 -25
  740. package/src/core/wasm/worker/message-buffer.ts +0 -33
  741. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  742. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -170
  743. package/src/core/websocket/state.ts +0 -7
  744. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  745. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  746. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  747. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  748. package/src/core/websocket/transports/basic.ts +0 -102
  749. package/src/core/websocket/transports/create.ts +0 -31
  750. package/src/core/websocket/transports/sse.ts +0 -308
  751. package/src/core/websocket/transports/transport.ts +0 -50
  752. package/src/core/websocket/transports/ws.ts +0 -98
  753. package/src/core/websocket/useNotebookClient.ts +0 -88
  754. package/src/utils/__tests__/id-tree.test.ts +0 -2145
  755. package/src/utils/id-tree.tsx +0 -1134
  756. /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
  757. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -2,17 +2,29 @@
2
2
 
3
3
  import { act, fireEvent, render, screen } from "@testing-library/react";
4
4
  import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
5
+ import { createStore, Provider } from "jotai";
5
6
  import { SetupMocks } from "@/__mocks__/common";
7
+ import { mockDocumentModel } from "@/__mocks__/document";
6
8
  import { cellId } from "@/__tests__/branded";
7
9
  import { TooltipProvider } from "@/components/ui/tooltip";
8
10
  import type { WithResponse } from "@/core/cells/types";
11
+ import { DocumentProvider } from "@/core/notebook/document";
9
12
  import type { OutputMessage } from "@/core/kernel/messages";
10
13
  import { CONSOLE_CLEAR_DEBOUNCE_MS, ConsoleOutput } from "../ConsoleOutput";
11
14
 
12
15
  SetupMocks.resizeObserver();
13
16
 
17
+ const store = createStore();
18
+ const { model } = mockDocumentModel(store, null);
19
+
14
20
  const renderWithProvider = (ui: React.ReactElement) => {
15
- return render(<TooltipProvider>{ui}</TooltipProvider>);
21
+ return render(
22
+ <Provider store={store}>
23
+ <DocumentProvider model={model}>
24
+ <TooltipProvider>{ui}</TooltipProvider>
25
+ </DocumentProvider>
26
+ </Provider>,
27
+ );
16
28
  };
17
29
 
18
30
  describe("ConsoleOutput integration", () => {
@@ -104,13 +116,17 @@ describe("ConsoleOutput pdb history", () => {
104
116
  ];
105
117
 
106
118
  rerender(
107
- <TooltipProvider>
108
- <ConsoleOutput
109
- {...defaultProps}
110
- consoleOutputs={outputs2}
111
- onSubmitDebugger={onSubmitDebugger}
112
- />
113
- </TooltipProvider>,
119
+ <Provider store={store}>
120
+ <DocumentProvider model={model}>
121
+ <TooltipProvider>
122
+ <ConsoleOutput
123
+ {...defaultProps}
124
+ consoleOutputs={outputs2}
125
+ onSubmitDebugger={onSubmitDebugger}
126
+ />
127
+ </TooltipProvider>
128
+ </DocumentProvider>
129
+ </Provider>,
114
130
  );
115
131
 
116
132
  // New StdInput mounted — press ArrowUp to recall previous command
@@ -138,13 +154,17 @@ describe("ConsoleOutput pdb history", () => {
138
154
  fireEvent.keyDown(nameInput, { key: "Enter" });
139
155
 
140
156
  rerender(
141
- <TooltipProvider>
142
- <ConsoleOutput
143
- {...defaultProps}
144
- consoleOutputs={[stdinPrompt("Name: ", "marimo"), passwordPrompt]}
145
- onSubmitDebugger={onSubmitDebugger}
146
- />
147
- </TooltipProvider>,
157
+ <Provider store={store}>
158
+ <DocumentProvider model={model}>
159
+ <TooltipProvider>
160
+ <ConsoleOutput
161
+ {...defaultProps}
162
+ consoleOutputs={[stdinPrompt("Name: ", "marimo"), passwordPrompt]}
163
+ onSubmitDebugger={onSubmitDebugger}
164
+ />
165
+ </TooltipProvider>
166
+ </DocumentProvider>
167
+ </Provider>,
148
168
  );
149
169
  const input = screen.getByTestId("console-input");
150
170
  expect(input).toHaveAttribute("type", "password");
@@ -161,16 +181,20 @@ describe("ConsoleOutput pdb history", () => {
161
181
  expect(input).toHaveValue("");
162
182
 
163
183
  rerender(
164
- <TooltipProvider>
165
- <ConsoleOutput
166
- {...defaultProps}
167
- consoleOutputs={[
168
- stdinPrompt("Name: ", "marimo"),
169
- { ...passwordPrompt, response: "" },
170
- stdinPrompt("Name: "),
171
- ]}
172
- />
173
- </TooltipProvider>,
184
+ <Provider store={store}>
185
+ <DocumentProvider model={model}>
186
+ <TooltipProvider>
187
+ <ConsoleOutput
188
+ {...defaultProps}
189
+ consoleOutputs={[
190
+ stdinPrompt("Name: ", "marimo"),
191
+ { ...passwordPrompt, response: "" },
192
+ stdinPrompt("Name: "),
193
+ ]}
194
+ />
195
+ </TooltipProvider>
196
+ </DocumentProvider>
197
+ </Provider>,
174
198
  );
175
199
  const nextInput = screen.getByTestId("console-input");
176
200
  fireEvent.keyDown(nextInput, { key: "ArrowUp" });
@@ -239,13 +263,17 @@ describe("ConsoleOutput pdb history", () => {
239
263
  stdinPrompt("(Pdb) "),
240
264
  ];
241
265
  rerender(
242
- <TooltipProvider>
243
- <ConsoleOutput
244
- {...defaultProps}
245
- consoleOutputs={outputs2}
246
- onSubmitDebugger={onSubmitDebugger}
247
- />
248
- </TooltipProvider>,
266
+ <Provider store={store}>
267
+ <DocumentProvider model={model}>
268
+ <TooltipProvider>
269
+ <ConsoleOutput
270
+ {...defaultProps}
271
+ consoleOutputs={outputs2}
272
+ onSubmitDebugger={onSubmitDebugger}
273
+ />
274
+ </TooltipProvider>
275
+ </DocumentProvider>
276
+ </Provider>,
249
277
  );
250
278
 
251
279
  // Submit an empty value; this should NOT enter the history stack.
@@ -258,13 +286,17 @@ describe("ConsoleOutput pdb history", () => {
258
286
  stdinPrompt("(Pdb) "),
259
287
  ];
260
288
  rerender(
261
- <TooltipProvider>
262
- <ConsoleOutput
263
- {...defaultProps}
264
- consoleOutputs={outputs3}
265
- onSubmitDebugger={onSubmitDebugger}
266
- />
267
- </TooltipProvider>,
289
+ <Provider store={store}>
290
+ <DocumentProvider model={model}>
291
+ <TooltipProvider>
292
+ <ConsoleOutput
293
+ {...defaultProps}
294
+ consoleOutputs={outputs3}
295
+ onSubmitDebugger={onSubmitDebugger}
296
+ />
297
+ </TooltipProvider>
298
+ </DocumentProvider>
299
+ </Provider>,
268
300
  );
269
301
 
270
302
  // ArrowUp should jump back to "step", skipping the empty submission.
@@ -299,13 +331,17 @@ describe("ConsoleOutput pdb history", () => {
299
331
  ];
300
332
 
301
333
  rerender(
302
- <TooltipProvider>
303
- <ConsoleOutput
304
- {...defaultProps}
305
- consoleOutputs={outputs2}
306
- onSubmitDebugger={onSubmitDebugger}
307
- />
308
- </TooltipProvider>,
334
+ <Provider store={store}>
335
+ <DocumentProvider model={model}>
336
+ <TooltipProvider>
337
+ <ConsoleOutput
338
+ {...defaultProps}
339
+ consoleOutputs={outputs2}
340
+ onSubmitDebugger={onSubmitDebugger}
341
+ />
342
+ </TooltipProvider>
343
+ </DocumentProvider>
344
+ </Provider>,
309
345
  );
310
346
 
311
347
  // Submit "print(x)"
@@ -321,13 +357,17 @@ describe("ConsoleOutput pdb history", () => {
321
357
  ];
322
358
 
323
359
  rerender(
324
- <TooltipProvider>
325
- <ConsoleOutput
326
- {...defaultProps}
327
- consoleOutputs={outputs3}
328
- onSubmitDebugger={onSubmitDebugger}
329
- />
330
- </TooltipProvider>,
360
+ <Provider store={store}>
361
+ <DocumentProvider model={model}>
362
+ <TooltipProvider>
363
+ <ConsoleOutput
364
+ {...defaultProps}
365
+ consoleOutputs={outputs3}
366
+ onSubmitDebugger={onSubmitDebugger}
367
+ />
368
+ </TooltipProvider>
369
+ </DocumentProvider>
370
+ </Provider>,
331
371
  );
332
372
 
333
373
  // ArrowUp should show most recent command first
@@ -370,13 +410,17 @@ describe("ConsoleOutput pdb history", () => {
370
410
  // Same outputs, but the cell isn't interrupted -- this is a real
371
411
  // bare-Enter submission, so we should render the (empty) placeholder.
372
412
  rerender(
373
- <TooltipProvider>
374
- <ConsoleOutput
375
- {...defaultProps}
376
- consoleOutputs={interruptedOutputs}
377
- interrupted={false}
378
- />
379
- </TooltipProvider>,
413
+ <Provider store={store}>
414
+ <DocumentProvider model={model}>
415
+ <TooltipProvider>
416
+ <ConsoleOutput
417
+ {...defaultProps}
418
+ consoleOutputs={interruptedOutputs}
419
+ interrupted={false}
420
+ />
421
+ </TooltipProvider>
422
+ </DocumentProvider>
423
+ </Provider>,
380
424
  );
381
425
 
382
426
  expect(screen.getByLabelText("stdin response")).toBeInTheDocument();
@@ -426,9 +470,13 @@ describe("ConsoleOutput debounced clearing", () => {
426
470
 
427
471
  // Clear outputs (simulates cell re-run)
428
472
  rerender(
429
- <TooltipProvider>
430
- <ConsoleOutput {...defaultProps} consoleOutputs={[]} />
431
- </TooltipProvider>,
473
+ <Provider store={store}>
474
+ <DocumentProvider model={model}>
475
+ <TooltipProvider>
476
+ <ConsoleOutput {...defaultProps} consoleOutputs={[]} />
477
+ </TooltipProvider>
478
+ </DocumentProvider>
479
+ </Provider>,
432
480
  );
433
481
 
434
482
  // Old output should still be visible during debounce period
@@ -437,9 +485,13 @@ describe("ConsoleOutput debounced clearing", () => {
437
485
  // New outputs arrive before debounce fires
438
486
  const outputs2 = [createOutput("new output")];
439
487
  rerender(
440
- <TooltipProvider>
441
- <ConsoleOutput {...defaultProps} consoleOutputs={outputs2} />
442
- </TooltipProvider>,
488
+ <Provider store={store}>
489
+ <DocumentProvider model={model}>
490
+ <TooltipProvider>
491
+ <ConsoleOutput {...defaultProps} consoleOutputs={outputs2} />
492
+ </TooltipProvider>
493
+ </DocumentProvider>
494
+ </Provider>,
443
495
  );
444
496
 
445
497
  // New output should be shown immediately
@@ -458,9 +510,13 @@ describe("ConsoleOutput debounced clearing", () => {
458
510
 
459
511
  // Clear outputs
460
512
  rerender(
461
- <TooltipProvider>
462
- <ConsoleOutput {...defaultProps} consoleOutputs={[]} />
463
- </TooltipProvider>,
513
+ <Provider store={store}>
514
+ <DocumentProvider model={model}>
515
+ <TooltipProvider>
516
+ <ConsoleOutput {...defaultProps} consoleOutputs={[]} />
517
+ </TooltipProvider>
518
+ </DocumentProvider>
519
+ </Provider>,
464
520
  );
465
521
 
466
522
  // Still visible during debounce
@@ -55,6 +55,7 @@ import {
55
55
  DropdownMenu,
56
56
  DropdownMenuCheckboxItem,
57
57
  DropdownMenuContent,
58
+ DropdownMenuItem,
58
59
  DropdownMenuTrigger,
59
60
  } from "../ui/dropdown-menu";
60
61
  import { ExternalLink } from "../ui/links";
@@ -158,6 +159,7 @@ export const PackageAlert: React.FC = () => {
158
159
  <SourceBadge source={packageAlert.source} />
159
160
  </span>
160
161
  <Button
162
+ aria-label="Dismiss missing packages alert"
161
163
  variant="text"
162
164
  data-testid="remove-banner-button"
163
165
  size="icon"
@@ -171,12 +173,12 @@ export const PackageAlert: React.FC = () => {
171
173
  <p>The following packages were not found:</p>
172
174
  <table className="list-disc ml-2 mt-1">
173
175
  <tbody>
174
- {packageAlert.packages.map((pkg, index) => {
176
+ {packageAlert.packages.map((pkg) => {
175
177
  const parsed = parsePackageSpecifier(pkg);
176
178
  const currentExtras = selectedExtras[pkg] || parsed.extras;
177
179
 
178
180
  return (
179
- <tr className="font-mono text-sm" key={index}>
181
+ <tr className="font-mono text-sm" key={pkg}>
180
182
  <td className="pr-2 py-1 align-middle">
181
183
  <BoxIcon size="1rem" />
182
184
  </td>
@@ -270,6 +272,7 @@ export const PackageAlert: React.FC = () => {
270
272
  <SourceBadge source={packageAlert.source} />
271
273
  </span>
272
274
  <Button
275
+ aria-label={`Dismiss ${title}`}
273
276
  variant="text"
274
277
  data-testid="remove-banner-button"
275
278
  size="icon"
@@ -294,7 +297,7 @@ export const PackageAlert: React.FC = () => {
294
297
  </>
295
298
  )}
296
299
  <ul className="list-disc ml-2 mt-1">
297
- {Object.entries(packageAlert.packages).map(([pkg, st], index) => (
300
+ {Object.entries(packageAlert.packages).map(([pkg, st]) => (
298
301
  <li
299
302
  className={cn(
300
303
  "flex items-center gap-1 font-mono text-sm",
@@ -305,7 +308,7 @@ export const PackageAlert: React.FC = () => {
305
308
  status === "failed" &&
306
309
  "text-muted-foreground",
307
310
  )}
308
- key={index}
311
+ key={pkg}
309
312
  >
310
313
  <ProgressIcon status={st} />
311
314
  {pkg}
@@ -590,14 +593,18 @@ const ExtrasSelector: React.FC<ExtrasSelectorProps> = ({
590
593
  <div className="p-2 bg-popover border-b border-border">
591
594
  <div className="flex flex-wrap gap-1 p-1 min-h-[24px]">
592
595
  {selectedExtras.map((extra) => (
593
- <span
596
+ <DropdownMenuItem
594
597
  key={extra}
598
+ aria-label={`Remove ${extra}`}
595
599
  className="inline-flex items-center gap-1 px-1 py-0.5 text-sm font-mono border border-muted-foreground/30 hover:border-muted-foreground/60 rounded-sm cursor-pointer group transition-colors"
596
- onClick={() => handleExtraToggle(extra, false)}
600
+ onSelect={(e) => {
601
+ e.preventDefault();
602
+ handleExtraToggle(extra, false);
603
+ }}
597
604
  >
598
605
  {extra}
599
606
  <XIcon className="w-3 h-3 opacity-60 group-hover:opacity-100" />
600
- </span>
607
+ </DropdownMenuItem>
601
608
  ))}
602
609
  </div>
603
610
  </div>
@@ -29,14 +29,10 @@ import { canInteractWithAppAtom } from "@/core/network/connection";
29
29
  import { useBoolean } from "@/hooks/useBoolean";
30
30
  import { cn } from "@/utils/cn";
31
31
  import { Functions } from "@/utils/functions";
32
- import type { CellColumnId } from "@/utils/id-tree";
32
+ import { type CellColumnId, findColumn, useLayout } from "@/core/cells/layout";
33
33
  import { invariant } from "@/utils/invariant";
34
- import {
35
- columnIdsAtom,
36
- useCellActions,
37
- useCellIds,
38
- useScrollKey,
39
- } from "../../../core/cells/cells";
34
+ import { useCellIds, useScrollKey } from "../../../core/cells/cells";
35
+ import { useCellActions } from "../../../core/cells/actions";
40
36
  import { formatAll } from "../../../core/codemirror/format";
41
37
  import type { AppConfig, UserConfig } from "../../../core/config/config-schema";
42
38
  import type { AppMode } from "../../../core/mode";
@@ -60,7 +56,7 @@ interface CellArrayProps {
60
56
  }
61
57
 
62
58
  export const CellArray: React.FC<CellArrayProps> = (props) => {
63
- const columnIds = useAtomValue(columnIdsAtom);
59
+ const columnIds = useLayout().columns.map((column) => column.id);
64
60
 
65
61
  // Setup context for sorting
66
62
  return (
@@ -115,7 +111,7 @@ const CellArrayInternal: React.FC<CellArrayProps> = ({
115
111
 
116
112
  const cellIds = useCellIds();
117
113
  const scrollKey = useScrollKey();
118
- const columnIds = cellIds.getColumnIds();
114
+ const columnIds = useLayout().columns.map((column) => column.id);
119
115
 
120
116
  // Scroll to a cell targeted by a previous action
121
117
  const scrollToTarget = actions.scrollToTarget;
@@ -139,7 +135,7 @@ const CellArrayInternal: React.FC<CellArrayProps> = ({
139
135
  <StdinBlockingAlert />
140
136
  <NotebookBanner width={appConfig.width} />
141
137
  {/* Only show if not cells, otherwise running a single cell will start the connection */}
142
- {cellIds.idLength === 0 && <NotStartedConnectionAlert />}
138
+ {cellIds.length === 0 && <NotStartedConnectionAlert />}
143
139
  <div
144
140
  className={cn(
145
141
  appConfig.width === "columns" &&
@@ -188,11 +184,11 @@ const CellColumn: React.FC<{
188
184
  hideControls,
189
185
  }) => {
190
186
  const cellIds = useCellIds();
191
- const column = cellIds.get(columnId);
187
+ const column = findColumn(useLayout(), columnId);
192
188
  invariant(column, `Expected column for: ${columnId}`);
193
189
 
194
- const hasOnlyOneCell = cellIds.hasOnlyOneId();
195
- const hasSetupCell = cellIds.inOrderIds.includes(SETUP_CELL_ID);
190
+ const hasOnlyOneCell = cellIds.length === 1;
191
+ const hasSetupCell = cellIds.includes(SETUP_CELL_ID);
196
192
  const isPresenting = mode === "present";
197
193
 
198
194
  return (
@@ -218,7 +214,7 @@ const CellColumn: React.FC<{
218
214
  >
219
215
  <SortableContext
220
216
  id={`column-${index + 1}`}
221
- items={column.topLevelIds}
217
+ items={[...column.visible]}
222
218
  strategy={verticalListSortingStrategy}
223
219
  >
224
220
  {/* Render the setup cell first, always */}
@@ -237,7 +233,7 @@ const CellColumn: React.FC<{
237
233
  />
238
234
  )}
239
235
 
240
- {column.topLevelIds.map((cellId) => {
236
+ {column.visible.map((cellId) => {
241
237
  // Skip the setup cell later
242
238
  if (cellId === SETUP_CELL_ID) {
243
239
  return null;
@@ -252,8 +248,8 @@ const CellColumn: React.FC<{
252
248
  canDelete={!hasOnlyOneCell}
253
249
  mode={mode}
254
250
  userConfig={userConfig}
255
- isCollapsed={column.isCollapsed(cellId)}
256
- collapseCount={column.getCount(cellId)}
251
+ isCollapsed={column.folds.has(cellId)}
252
+ collapseCount={column.folds.get(cellId)?.length ?? 0}
257
253
  canMoveX={appConfig.width === "columns"}
258
254
  />
259
255
  );
@@ -267,7 +263,7 @@ const AddCellButtons: React.FC<{
267
263
  columnId: CellColumnId;
268
264
  className?: string;
269
265
  }> = ({ columnId, className }) => {
270
- const { createNewCell } = useCellActions();
266
+ const { createCell } = useCellActions();
271
267
  const [isAiButtonOpen, isAiButtonOpenActions] = useBoolean(false);
272
268
  const aiEnabled = useAtomValue(aiEnabledAtom);
273
269
  const aiFeaturesEnabled = useAtomValue(aiFeaturesEnabledAtom);
@@ -291,12 +287,7 @@ const AddCellButtons: React.FC<{
291
287
  variant="text"
292
288
  size="sm"
293
289
  disabled={!canInteractWithApp}
294
- onClick={() =>
295
- createNewCell({
296
- cellId: { type: "__end__", columnId },
297
- before: false,
298
- })
299
- }
290
+ onClick={() => createCell({ endOf: columnId })}
300
291
  >
301
292
  <SquareCodeIcon className="mr-2 size-4 shrink-0" />
302
293
  Python
@@ -307,13 +298,12 @@ const AddCellButtons: React.FC<{
307
298
  size="sm"
308
299
  disabled={!canInteractWithApp}
309
300
  onClick={() => {
310
- maybeAddMarimoImport({ autoInstantiate: true, createNewCell });
301
+ maybeAddMarimoImport({ autoInstantiate: true, createCell });
311
302
 
312
- createNewCell({
313
- cellId: { type: "__end__", columnId },
314
- before: false,
303
+ createCell({
304
+ endOf: columnId,
315
305
  code: LanguageAdapters.markdown.defaultCode,
316
- hideCode: MARKDOWN_INITIAL_HIDE_CODE,
306
+ config: { hide_code: MARKDOWN_INITIAL_HIDE_CODE },
317
307
  });
318
308
  }}
319
309
  >
@@ -326,11 +316,10 @@ const AddCellButtons: React.FC<{
326
316
  size="sm"
327
317
  disabled={!canInteractWithApp}
328
318
  onClick={() => {
329
- maybeAddMarimoImport({ autoInstantiate: true, createNewCell });
319
+ maybeAddMarimoImport({ autoInstantiate: true, createCell });
330
320
 
331
- createNewCell({
332
- cellId: { type: "__end__", columnId },
333
- before: false,
321
+ createCell({
322
+ endOf: columnId,
334
323
  code: LanguageAdapters.sql.defaultCode,
335
324
  });
336
325
  }}
@@ -190,12 +190,18 @@ export const GridLayoutRenderer: React.FC<Props> = ({
190
190
  cells: [...cellLayouts, dropped],
191
191
  });
192
192
  }}
193
- onDragStart={(_layout, _oldItem, _newItem, _placeholder, event) => {
194
- dragProps.onDragStart(event);
195
- }}
196
- onDrag={(_layout, _oldItem, _newItem, _placeholder, event) => {
197
- dragProps.onDragMove(event);
198
- }}
193
+ onDragStart={
194
+ // oxlint-disable-next-line marimo/prefer-object-params -- React Grid Layout supplies the event as the fifth argument.
195
+ (_layout, _oldItem, _newItem, _placeholder, event) => {
196
+ dragProps.onDragStart(event);
197
+ }
198
+ }
199
+ onDrag={
200
+ // oxlint-disable-next-line marimo/prefer-object-params -- React Grid Layout supplies the event as the fifth argument.
201
+ (_layout, _oldItem, _newItem, _placeholder, event) => {
202
+ dragProps.onDragMove(event);
203
+ }
204
+ }
199
205
  onDragStop={() => {
200
206
  dragProps.onDragStop();
201
207
  }}
@@ -17,7 +17,6 @@ function makeCell(id: string, code = "print('hi')"): CellData {
17
17
  lastCodeRun: null,
18
18
  lastExecutionTime: null,
19
19
  config: { hide_code: false, disabled: false, column: null },
20
- serializedEditorState: null,
21
20
  };
22
21
  }
23
22
 
@@ -1,7 +1,7 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
  import React, { useMemo, useState } from "react";
3
3
  import { useAtomValue } from "jotai";
4
- import { numColumnsAtom } from "@/core/cells/cells";
4
+ import { numColumnsAtom } from "@/core/cells/layout";
5
5
  import type { CellId } from "@/core/cells/ids";
6
6
  import { kioskModeAtom } from "@/core/mode";
7
7
  import type { ICellRendererProps } from "../types";
@@ -25,7 +25,7 @@ describe("useFocusFirstEditor", () => {
25
25
  const getNotebookSpy = vi
26
26
  .spyOn(cellsModule, "getNotebook")
27
27
  .mockReturnValue({
28
- cellIds: { iterateTopLevelIds: ["cell-1", "cell-2"] },
28
+ cellIds: ["cell-1", "cell-2"],
29
29
  cellData: {
30
30
  "cell-1": { config: { hide_code: false } },
31
31
  "cell-2": { config: { hide_code: false } },
@@ -52,7 +52,7 @@ describe("useFocusFirstEditor", () => {
52
52
  const getNotebookSpy = vi
53
53
  .spyOn(cellsModule, "getNotebook")
54
54
  .mockReturnValue({
55
- cellIds: { iterateTopLevelIds: ["cell-1", "cell-2"] },
55
+ cellIds: ["cell-1", "cell-2"],
56
56
  cellData: {
57
57
  "cell-1": { config: { hide_code: true } }, // hidden
58
58
  "cell-2": { config: { hide_code: false } }, // visible
@@ -94,7 +94,7 @@ describe("useFocusFirstEditor", () => {
94
94
  // Mock getNotebook
95
95
  const mockEditor = { focus: vi.fn() };
96
96
  vi.spyOn(cellsModule, "getNotebook").mockReturnValue({
97
- cellIds: { iterateTopLevelIds: [] },
97
+ cellIds: [],
98
98
  cellData: {},
99
99
  cellHandles: {
100
100
  "cell-123": { current: { editorView: mockEditor } },
@@ -129,7 +129,7 @@ describe("useFocusFirstEditor", () => {
129
129
 
130
130
  const mockEditor = { focus: vi.fn() };
131
131
  vi.spyOn(cellsModule, "getNotebook").mockReturnValue({
132
- cellIds: { iterateTopLevelIds: ["cell-1"] },
132
+ cellIds: ["cell-1"],
133
133
  cellData: {
134
134
  "cell-1": { config: { hide_code: false } },
135
135
  },
@@ -12,7 +12,13 @@ interface SidebarToggleProps {
12
12
  export const SidebarToggle = ({ isOpen, toggle }: SidebarToggleProps) => {
13
13
  return (
14
14
  <div className="invisible lg:visible absolute top-[12px] right-[16px] z-20">
15
- <Button onClick={toggle} className="w-10 h-8" variant="ghost" size="icon">
15
+ <Button
16
+ aria-label={isOpen ? "Collapse sidebar" : "Expand sidebar"}
17
+ onClick={toggle}
18
+ className="w-10 h-8"
19
+ variant="ghost"
20
+ size="icon"
21
+ >
16
22
  <ChevronLeft
17
23
  className={cn(
18
24
  "h-5 w-5 transition-transform ease-in-out duration-700",
@@ -52,7 +52,7 @@ function focusFirstEditor() {
52
52
  const { cellIds, cellData, cellHandles } = getNotebook();
53
53
 
54
54
  // Focus on the first cell if it's been mounted and is not hidden
55
- for (const cellId of cellIds.iterateTopLevelIds) {
55
+ for (const cellId of cellIds) {
56
56
  const handle = cellHandles[cellId];
57
57
  const hidden = cellData[cellId]?.config.hide_code;
58
58
  if (!hidden && handle?.current?.editorView) {
@@ -81,7 +81,7 @@ const VerticalLayoutRenderer: React.FC<VerticalLayoutProps> = ({
81
81
  getInitialShowCode({ showCodeInRunModePreference, kioskMode }),
82
82
  );
83
83
 
84
- const canShowCode = useNotebookCodeAvailable(cells);
84
+ const canShowCode = useNotebookCodeAvailable();
85
85
 
86
86
  const handleToggleShowCode = () => {
87
87
  setShowCode((currentShowCode) => {
@@ -13,7 +13,7 @@ const hasBlockingStdinAtom = atom((get) => {
13
13
  const notebook = get(notebookAtom);
14
14
 
15
15
  // Check each cell in order
16
- for (const cellId of notebook.cellIds.inOrderIds) {
16
+ for (const cellId of notebook.cellIds) {
17
17
  // Check if the cell is idle and has unresolved stdin
18
18
  const runtime = notebook.cellRuntime[cellId];
19
19
  if (runtime.status === "idle") {