@marimo-team/islands 0.25.3-dev26 → 0.25.3-dev261

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 (796) hide show
  1. package/dist/{ConnectedDataExplorerComponent-DfG3PRaJ.js → ConnectedDataExplorerComponent-DjlwdMQT.js} +120 -116
  2. package/dist/{ErrorBoundary-MDk-fuIJ.js → ErrorBoundary-m3vF1P4J.js} +16 -16
  3. package/dist/_baseSet-BfI5FjBM.js +328 -0
  4. package/dist/{abnfDiagram-VCTEODGH-B3w-XWX8.js → abnfDiagram-VCTEODGH-DzrZZXRD.js} +2 -2
  5. package/dist/any-language-editor-Be7dA2VK.js +75 -0
  6. package/dist/{architectureDiagram-5GKGNRK7-C18U9A6k.js → architectureDiagram-5GKGNRK7-iF_snyFL.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-0-RqY1DW.js} +229 -218
  12. package/dist/{capabilities-CQ5IjhzL.js → capabilities-B6yy_CBN.js} +2 -2
  13. package/dist/{chat-ui-w8SVynkn.js → chat-ui-Bl_I-evE.js} +822 -755
  14. package/dist/{check-BMKBBrdc.js → check-CSvxZk5I.js} +1 -1
  15. package/dist/{chunk-IMKFNOWR-CrS5bzqW.js → chunk-IMKFNOWR-B7xytD4B.js} +1 -1
  16. package/dist/{chunk-SHT3W25Y-StpvPvUG.js → chunk-SHT3W25Y-I8ZFP5L1.js} +1 -1
  17. package/dist/{chunk-SVP7TREG-CBXjbYZC.js → chunk-SVP7TREG-Di-uAPdT.js} +1 -1
  18. package/dist/{chunk-TICWLB2K-Dp6bIb-q.js → chunk-TICWLB2K-BFpUy0ud.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-Cqm4JGpx.js} +2 -2
  21. package/dist/{classDiagram-v2-VYDZK3BY-D_tjXVMl.js → classDiagram-v2-VYDZK3BY-DFHRJpN3.js} +2 -2
  22. package/dist/code-visibility-TqQbHmZ5.js +134 -0
  23. package/dist/{common-Bo1ahZJz.js → common-BY_oazL9.js} +9850 -9619
  24. package/dist/config-mwEjAmvP.js +636 -0
  25. package/dist/{copy-icon-BgUO8jsi.js → copy-icon-DD8bdw4L.js} +3 -3
  26. package/dist/createReducer-C_syF8Td.js +67 -0
  27. package/dist/{cynefinDiagram-5FMLGOSQ-DSgYrHB2.js → cynefinDiagram-5FMLGOSQ-cuGOeTBC.js} +1 -1
  28. package/dist/{dagre-GXQ25YYZ-GYlruCBf.js → dagre-GXQ25YYZ-DJ76aC4k.js} +3 -3
  29. package/dist/{dagre-BGSfXNEb.js → dagre-mqiMAsFY.js} +531 -680
  30. package/dist/{data-grid-overlay-editor-BcbqpDrL.js → data-grid-overlay-editor-CLQPFjyJ.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-DLLc0sw7.js} +1 -1
  34. package/dist/{diagram-UQ7AKVKN-DD7uiejP.js → diagram-UQ7AKVKN-Tz_d6qmk.js} +1 -1
  35. package/dist/{diagram-VX7I27RA-ByH9KAJJ.js → diagram-VX7I27RA-D7SzPrDN.js} +1 -1
  36. package/dist/{diagram-Z3DM3KII-BsQHPRke.js → diagram-Z3DM3KII-qr37XV58.js} +1 -1
  37. package/dist/{dialog-Cc479ZEH.js → dialog-BHYPX87J.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-Blu7V6Mk.js +6083 -0
  64. package/dist/{ebnfDiagram-PWID7BFC-CQRK5_eT.js → ebnfDiagram-PWID7BFC-BKOQb4J0.js} +2 -2
  65. package/dist/{erDiagram-RLTQ6QDP-BKtWrDVh.js → erDiagram-RLTQ6QDP-CM15Vq68.js} +1 -1
  66. package/dist/{error-banner-CFPHFw4h.js → error-banner-DbRFOd2O.js} +84 -83
  67. package/dist/error-entries-DFhMo601.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-DYk1f0In.js +374 -0
  71. package/dist/{flowDiagram-HODETNUW-1IN-ih2E.js → flowDiagram-HODETNUW-Dt3N5yFT.js} +2 -2
  72. package/dist/{formats-X4mMQWzX.js → formats-gGospKWJ.js} +6 -6
  73. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  74. package/dist/{glide-data-editor-h-P3W8Rc.js → glide-data-editor-D30zzaiO.js} +417 -415
  75. package/dist/{glide-utils-DM9zBO4W.js → glide-utils-qbkRZpEy.js} +16 -16
  76. package/dist/{globals-4XAS0Mmd.js → globals-BA0s7TxM.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-BPs9Vo-H.js} +26928 -26781
  81. package/dist/{infoDiagram-27XIBGKW-CkB-zr4p.js → infoDiagram-27XIBGKW-C7axJMCl.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-CdC9Eur7.js} +1 -1
  85. package/dist/{kanban-definition-UXKFOSKX-C2usY0i-.js → kanban-definition-UXKFOSKX-BHnMIgD7.js} +1 -1
  86. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  87. package/dist/{loader-BQOfjTIJ.js → loader-BVEtQ4F3.js} +12 -12
  88. package/dist/main.js +10241 -8111
  89. package/dist/{mermaid-BA-Z6wGL.js → mermaid-DFYN_PXd.js} +54 -54
  90. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  91. package/dist/{mindmap-definition-YA3MSWOX-nKOWS8RR.js → mindmap-definition-YA3MSWOX-BagHoCGU.js} +1 -1
  92. package/dist/{now-jM8-M7IR.js → now-BKZCLE5P.js} +12 -13
  93. package/dist/{number-overlay-editor-DVBm-B8V.js → number-overlay-editor-B1nffNQ2.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-8IIZczip.js} +2 -2
  98. package/dist/pick-tCszMUbm.js +35 -0
  99. package/dist/{pieDiagram-E7YTZNPT-B7S-DsW_.js → pieDiagram-E7YTZNPT-CE6dKBIr.js} +1 -1
  100. package/dist/precisionRound-DhAzeRcm.js +49 -0
  101. package/dist/{process-output-CTPTtxXB.js → process-output-c6xaHokq.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-CkexPq3b.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-B0-OkuAe.js} +1 -1
  106. package/dist/{reveal-component-DtPzCmeW.js → reveal-component-fg3-aYjk.js} +430 -407
  107. package/dist/schemas-CdscWNy_.js +4747 -0
  108. package/dist/select-CcidPqlM.js +1200 -0
  109. package/dist/{sequenceDiagram-WJ2MYXX4-b6pei3lJ.js → sequenceDiagram-WJ2MYXX4-Dxysg8DX.js} +1 -1
  110. package/dist/{spec-BJzgmvVa.js → spec-ZwfZcgsm.js} +41 -11
  111. package/dist/{stateDiagram-D77RDMKH-Be-xSpaZ.js → stateDiagram-D77RDMKH-DTEN3-OJ.js} +4 -4
  112. package/dist/{stateDiagram-v2-MP3YSRHH-5Y-FxesJ.js → stateDiagram-v2-MP3YSRHH-vktk3Xc5.js} +2 -2
  113. package/dist/static-state-BldGd99T.js +72 -0
  114. package/dist/strings-BYJNyKmg.js +507 -0
  115. package/dist/style.css +1 -1
  116. package/dist/{swimlanes-42K2YHIH-nRRl4NU0.js → swimlanes-42K2YHIH-Cjveh9OM.js} +1 -1
  117. package/dist/{swimlanesDiagram-VR7AAH4N-BS79510V.js → swimlanesDiagram-VR7AAH4N-Bdr7Ev_2.js} +2 -2
  118. package/dist/{swiper-component-DKntyn8p.js → swiper-component-Dvj-goaD.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-KIuDcefn.js} +1 -1
  121. package/dist/toast-CkB5Jq9K.js +491 -0
  122. package/dist/{tooltip-CtBZ5q8p.js → tooltip-Bzl4k00l.js} +329 -305
  123. package/dist/use-restore-focus-CmflVLWJ.js +1672 -0
  124. package/dist/{useAsyncData-BD534XmW.js → useAsyncData-DsZMvNXn.js} +15 -14
  125. package/dist/{useDateFormatter-DkajGwwd.js → useDateFormatter-CYo5fm8s.js} +2 -2
  126. package/dist/{useDeepCompareMemoize-C72_RNsu.js → useDeepCompareMemoize-KMBBux8C.js} +1 -1
  127. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-BIolGsF1.js} +1 -1
  128. package/dist/useLifecycle-GNgvNwI-.js +112 -0
  129. package/dist/useTheme-BZ2GGLgm.js +339 -0
  130. package/dist/utils-DXkj82sK.js +813 -0
  131. package/dist/{vega-component-DfcYgj-Y.js → vega-component-DOtem7Hf.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-DrIOUHfb.js} +1 -1
  135. package/dist/{wardleyDiagram-VM6X3IG4-Cxs9NtVi.js → wardleyDiagram-VM6X3IG4-CJNjf4OH.js} +1 -1
  136. package/dist/{xychartDiagram-S5SC5T6Z-DNWbvPK7.js → xychartDiagram-S5SC5T6Z-DEshFho-.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__/fixtures/pair-config.ts +1 -1
  143. package/src/__tests__/main.test.tsx +24 -1
  144. package/src/__tests__/module-entry.test.ts +44 -0
  145. package/src/__tests__/mount.test.ts +1 -21
  146. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  147. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  148. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  149. package/src/components/app-config/__tests__/state.test.tsx +50 -0
  150. package/src/components/app-config/ai-config.tsx +156 -150
  151. package/src/components/app-config/app-config-button.tsx +6 -5
  152. package/src/components/app-config/app-config-form.tsx +5 -9
  153. package/src/components/app-config/common.tsx +7 -9
  154. package/src/components/app-config/data-form.tsx +112 -113
  155. package/src/components/app-config/mcp-config.tsx +30 -53
  156. package/src/components/app-config/user-config-form.css +91 -0
  157. package/src/components/app-config/user-config-form.tsx +266 -349
  158. package/src/components/chat/__tests__/capabilities-popover.test.tsx +0 -1
  159. package/src/components/chat/__tests__/chat-display.test.tsx +125 -0
  160. package/src/components/chat/__tests__/chat-markdown.test.tsx +43 -0
  161. package/src/components/chat/__tests__/chat-utils.test.ts +51 -1
  162. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  163. package/src/components/chat/__tests__/notebook-access-selector.test.tsx +64 -0
  164. package/src/components/chat/acp/__tests__/agent-panel-initialization.test.tsx +1 -1
  165. package/src/components/chat/acp/__tests__/agent-panel.test.tsx +22 -17
  166. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  167. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  168. package/src/components/chat/acp/agent-panel.tsx +4 -3
  169. package/src/components/chat/acp/blocks.tsx +13 -12
  170. package/src/components/chat/acp/common.tsx +4 -5
  171. package/src/components/chat/acp/model-selector.tsx +1 -1
  172. package/src/components/chat/acp/session-tabs.tsx +8 -3
  173. package/src/components/chat/capabilities-popover.tsx +2 -11
  174. package/src/components/chat/chat-components.tsx +9 -4
  175. package/src/components/chat/chat-display.tsx +51 -6
  176. package/src/components/chat/chat-history-popover.tsx +6 -1
  177. package/src/components/chat/chat-markdown.tsx +15 -0
  178. package/src/components/chat/chat-panel.tsx +119 -214
  179. package/src/components/chat/chat-typography.css +108 -0
  180. package/src/components/chat/chat-utils.ts +12 -11
  181. package/src/components/chat/notebook-access-selector.stories.tsx +22 -0
  182. package/src/components/chat/notebook-access-selector.tsx +100 -0
  183. package/src/components/chat/reasoning-accordion.tsx +3 -3
  184. package/src/components/chat/tool-call/tool-activity-group.tsx +49 -0
  185. package/src/components/chat/tool-call/tool-history-row.tsx +3 -30
  186. package/src/components/chat/tool-call/tool-result.tsx +1 -1
  187. package/src/components/chat/use-chat-generation.ts +19 -0
  188. package/src/components/data-table/TableBottomBar.tsx +4 -1
  189. package/src/components/data-table/TableTopBar.tsx +2 -0
  190. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  191. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  192. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  193. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  194. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  195. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  196. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  197. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  198. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  199. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  200. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  201. package/src/components/data-table/cell-selection/feature.ts +2 -2
  202. package/src/components/data-table/cell-selection/types.ts +2 -2
  203. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  204. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  205. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  206. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  207. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  208. package/src/components/data-table/charts/charts.tsx +40 -30
  209. package/src/components/data-table/charts/storage.ts +16 -5
  210. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  211. package/src/components/data-table/column-formatting/types.ts +1 -2
  212. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  213. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  214. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  215. package/src/components/data-table/columns.tsx +45 -46
  216. package/src/components/data-table/context-menu.tsx +36 -20
  217. package/src/components/data-table/data-table.tsx +17 -1
  218. package/src/components/data-table/date-popover.tsx +6 -4
  219. package/src/components/data-table/export-actions.tsx +76 -48
  220. package/src/components/data-table/pagination.tsx +44 -33
  221. package/src/components/data-table/range-focus/atoms.ts +4 -6
  222. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  223. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  224. package/src/components/data-table/renderers.tsx +16 -7
  225. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  226. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  227. package/src/components/data-table/search-highlight.tsx +79 -0
  228. package/src/components/data-table/types.ts +1 -0
  229. package/src/components/data-table/url-detector.tsx +49 -9
  230. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  231. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  232. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  233. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  234. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  235. package/src/components/datasources/column-preview.tsx +17 -12
  236. package/src/components/datasources/datasources.tsx +56 -43
  237. package/src/components/datasources/utils.ts +32 -23
  238. package/src/components/debugger/debugger-code.tsx +5 -0
  239. package/src/components/dependency-graph/custom-node.tsx +1 -1
  240. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  241. package/src/components/dependency-graph/elements.ts +24 -21
  242. package/src/components/dependency-graph/minimap-content.tsx +13 -19
  243. package/src/components/dependency-graph/panels.tsx +4 -2
  244. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  245. package/src/components/editor/SortableCell.tsx +1 -3
  246. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  247. package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
  248. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -14
  249. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  250. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -2
  251. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  252. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  253. package/src/components/editor/actions/__tests__/pair-with-agent-commands.test.ts +21 -40
  254. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +41 -14
  255. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +6 -24
  256. package/src/components/editor/actions/__tests__/useCellActionButton.test.tsx +113 -0
  257. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  258. package/src/components/editor/actions/pair-with-agent-commands.ts +10 -7
  259. package/src/components/editor/actions/pair-with-agent-modal.tsx +29 -30
  260. package/src/components/editor/actions/pair-with-agent.tsx +16 -39
  261. package/src/components/editor/actions/useCellActionButton.tsx +37 -32
  262. package/src/components/editor/actions/useConfigActions.tsx +3 -4
  263. package/src/components/editor/actions/useNotebookActions.tsx +25 -29
  264. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  265. package/src/components/editor/actions/useSetCodeVisibility.ts +11 -12
  266. package/src/components/editor/actions/useSetOutputsExpanded.ts +14 -19
  267. package/src/components/editor/ai/ai-completion-editor.tsx +2 -1
  268. package/src/components/editor/ai/prompt-input.tsx +189 -0
  269. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  270. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  271. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  272. package/src/components/editor/alerts/stdin-blocking-alert.tsx +1 -1
  273. package/src/components/editor/app-container.tsx +1 -1
  274. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  275. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  276. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  277. package/src/components/editor/cell/RunButton.tsx +2 -2
  278. package/src/components/editor/cell/StagedAICell.tsx +13 -5
  279. package/src/components/editor/cell/StopButton.tsx +2 -2
  280. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  281. package/src/components/editor/cell/cell-actions.tsx +12 -1
  282. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  283. package/src/components/editor/cell/code/cell-editor.tsx +59 -165
  284. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  285. package/src/components/editor/cell/collapse.tsx +11 -4
  286. package/src/components/editor/cell/useAddCell.ts +4 -8
  287. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  288. package/src/components/editor/cell/useRunCells.ts +2 -5
  289. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  290. package/src/components/editor/cell/useSplitCell.tsx +10 -6
  291. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  292. package/src/components/editor/chrome/components/feedback-button.tsx +2 -8
  293. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  294. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  295. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  296. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  297. package/src/components/editor/chrome/panels/dependency-graph-panel.tsx +1 -1
  298. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  299. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  300. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  301. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  302. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  303. package/src/components/editor/chrome/panels/panel-header.tsx +13 -0
  304. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  305. package/src/components/editor/chrome/panels/session-panel.tsx +1 -1
  306. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  307. package/src/components/editor/chrome/panels/variable-panel.tsx +1 -5
  308. package/src/components/editor/chrome/types.ts +1 -1
  309. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +7 -9
  310. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  311. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +102 -35
  312. package/src/components/editor/chrome/wrapper/__tests__/useAskAiAboutCell.test.tsx +58 -0
  313. package/src/components/editor/chrome/wrapper/app-chrome.tsx +89 -72
  314. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
  315. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  316. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  317. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  318. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  319. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  320. package/src/components/editor/chrome/wrapper/footer.tsx +2 -2
  321. package/src/components/editor/chrome/wrapper/minimap-state.ts +1 -1
  322. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +17 -10
  323. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  324. package/src/components/editor/chrome/wrapper/useAskAiAboutCell.ts +36 -0
  325. package/src/components/editor/columns/__tests__/cell-column.test.tsx +1 -1
  326. package/src/components/editor/columns/cell-column.tsx +1 -1
  327. package/src/components/editor/columns/sortable-column.tsx +2 -2
  328. package/src/components/editor/connections/components.tsx +5 -10
  329. package/src/components/editor/controls/Controls.tsx +4 -4
  330. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  331. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  332. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  333. package/src/components/editor/controls/command-palette.tsx +22 -10
  334. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  335. package/src/components/editor/documentation.css +34 -11
  336. package/src/components/editor/dynamic-favicon.tsx +21 -40
  337. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +25 -20
  338. package/src/components/editor/errors/auto-fix.tsx +4 -8
  339. package/src/components/editor/errors/send-error-report-button.tsx +3 -6
  340. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  341. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  342. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  343. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  344. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  345. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  346. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  347. package/src/components/editor/header/__tests__/filename-form.test.tsx +1 -4
  348. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  349. package/src/components/editor/header/app-header.tsx +1 -1
  350. package/src/components/editor/header/filename-form.tsx +1 -1
  351. package/src/components/editor/header/status.tsx +40 -20
  352. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  353. package/src/components/editor/links/cell-link-list.tsx +1 -1
  354. package/src/components/editor/links/cell-link.tsx +17 -7
  355. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  356. package/src/components/editor/navigation/__tests__/navigation.test.ts +108 -85
  357. package/src/components/editor/navigation/__tests__/selection.test.ts +43 -21
  358. package/src/components/editor/navigation/clipboard.ts +17 -14
  359. package/src/components/editor/navigation/focus-utils.ts +23 -4
  360. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +17 -33
  361. package/src/components/editor/navigation/navigation.ts +71 -70
  362. package/src/components/editor/navigation/selection.ts +30 -29
  363. package/src/components/editor/notebook-banner.tsx +1 -0
  364. package/src/components/editor/notebook-cell.tsx +29 -32
  365. package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
  366. package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
  367. package/src/components/editor/output/Outputs.css +8 -0
  368. package/src/components/editor/output/VideoOutput.tsx +6 -0
  369. package/src/components/editor/output/__tests__/VideoOutput.test.tsx +131 -0
  370. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  371. package/src/components/editor/output/ansi-reduce.ts +33 -20
  372. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  373. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  374. package/src/components/editor/package-alert.tsx +14 -7
  375. package/src/components/editor/renderers/__tests__/add-cell-buttons.test.tsx +18 -0
  376. package/src/components/editor/renderers/cell-array.tsx +32 -100
  377. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  378. package/src/components/editor/renderers/slides-layout/__tests__/plugin.test.ts +0 -1
  379. package/src/components/editor/renderers/slides-layout/slides-layout.tsx +1 -1
  380. package/src/components/editor/renderers/vertical-layout/__tests__/useFocusFirstEditor.test.ts +4 -4
  381. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  382. package/src/components/editor/renderers/vertical-layout/useFocusFirstEditor.ts +1 -1
  383. package/src/components/editor/renderers/vertical-layout/vertical-layout.tsx +1 -1
  384. package/src/components/editor/stdin-blocking-alert.tsx +1 -1
  385. package/src/components/home/components.tsx +17 -31
  386. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  387. package/src/components/icons/large-spinner.tsx +3 -14
  388. package/src/components/icons/spinner.tsx +1 -1
  389. package/src/components/markdown/markdown-renderer.tsx +39 -28
  390. package/src/components/mcp/mcp-status-indicator.tsx +18 -14
  391. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  392. package/src/components/pages/edit-page.tsx +19 -17
  393. package/src/components/pages/home-page.tsx +164 -91
  394. package/src/components/scratchpad/scratchpad.test.tsx +89 -0
  395. package/src/components/scratchpad/scratchpad.tsx +43 -32
  396. package/src/components/scratchpad/state.ts +7 -0
  397. package/src/components/slides/__tests__/minimap-actions.test.tsx +22 -29
  398. package/src/components/slides/__tests__/minimap.test.ts +38 -19
  399. package/src/components/slides/__tests__/slide-scroll-hint.test.tsx +37 -0
  400. package/src/components/slides/minimap.tsx +17 -12
  401. package/src/components/slides/reveal-component.tsx +58 -26
  402. package/src/components/slides/reveal-slides.css +5 -4
  403. package/src/components/slides/slide-cell-view.tsx +4 -5
  404. package/src/components/slides/slide-form.tsx +1 -0
  405. package/src/components/slides/slide-scroll-hint.tsx +3 -1
  406. package/src/components/sort/SortableCellsProvider.tsx +8 -25
  407. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  408. package/src/components/tracing/tracing.tsx +46 -42
  409. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  410. package/src/components/ui/__tests__/fullscreen.test.tsx +44 -2
  411. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  412. package/src/components/ui/alert-dialog.tsx +1 -1
  413. package/src/components/ui/draggable-popover.tsx +24 -2
  414. package/src/components/ui/fullscreen.tsx +25 -1
  415. package/src/components/ui/range-slider.tsx +22 -9
  416. package/src/core/MarimoApp.tsx +3 -1
  417. package/src/core/ai/__tests__/staged-cells.test.ts +15 -482
  418. package/src/core/ai/context/context.ts +2 -0
  419. package/src/core/ai/context/providers/__tests__/cell-output.test.ts +8 -12
  420. package/src/core/ai/context/providers/__tests__/cell.test.ts +49 -0
  421. package/src/core/ai/context/providers/cell-output.ts +2 -2
  422. package/src/core/ai/context/providers/cell.ts +54 -0
  423. package/src/core/ai/context/providers/datasource.ts +6 -2
  424. package/src/core/ai/staged-cells.ts +17 -294
  425. package/src/core/ai/state.ts +3 -0
  426. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +40 -88
  427. package/src/core/ai/tools/__tests__/registry.test.ts +17 -0
  428. package/src/core/ai/tools/base.ts +18 -15
  429. package/src/core/ai/tools/edit-notebook-tool.ts +86 -80
  430. package/src/core/ai/tools/registry.ts +13 -10
  431. package/src/core/ai/tools/sample-tool.ts +6 -7
  432. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -37
  433. package/src/core/cells/__tests__/actions.test.ts +624 -35
  434. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  435. package/src/core/cells/__tests__/arrange.test.ts +432 -0
  436. package/src/core/cells/__tests__/cell.test.ts +39 -0
  437. package/src/core/cells/__tests__/cells.test.ts +251 -2540
  438. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  439. package/src/core/cells/__tests__/document.test.ts +164 -0
  440. package/src/core/cells/__tests__/editor.test.ts +57 -0
  441. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  442. package/src/core/cells/__tests__/harness.ts +106 -0
  443. package/src/core/cells/__tests__/layout.test.ts +177 -0
  444. package/src/core/cells/__tests__/notebook-outline.test.ts +122 -0
  445. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  446. package/src/core/cells/__tests__/session.test.ts +445 -211
  447. package/src/core/cells/__tests__/utils.test.ts +9 -108
  448. package/src/core/cells/actions.ts +675 -6
  449. package/src/core/cells/add-missing-import.ts +47 -43
  450. package/src/core/cells/arrange.ts +529 -0
  451. package/src/core/cells/cell.ts +14 -3
  452. package/src/core/cells/cells.ts +126 -1363
  453. package/src/core/cells/columns.ts +38 -0
  454. package/src/core/cells/document.ts +178 -0
  455. package/src/core/cells/edit.ts +30 -0
  456. package/src/core/cells/editor.ts +82 -0
  457. package/src/core/cells/layout.ts +221 -0
  458. package/src/core/cells/outputs.ts +12 -28
  459. package/src/core/cells/pending-delete-service.ts +3 -7
  460. package/src/core/cells/session.ts +76 -53
  461. package/src/core/cells/types.ts +22 -6
  462. package/src/core/cells/utils.ts +10 -8
  463. package/src/core/codemirror/__tests__/format.test.ts +0 -34
  464. package/src/core/codemirror/__tests__/setup.test.ts +61 -29
  465. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  466. package/src/core/codemirror/ai/resources.ts +1 -1
  467. package/src/core/codemirror/cells/__tests__/extensions.test.ts +2 -6
  468. package/src/core/codemirror/cells/extensions.ts +30 -28
  469. package/src/core/codemirror/cells/state.ts +2 -6
  470. package/src/core/codemirror/cm.ts +5 -1
  471. package/src/core/codemirror/config/types.ts +1 -1
  472. package/src/core/codemirror/copilot/__tests__/getCodes.test.ts +6 -7
  473. package/src/core/codemirror/copilot/getCodes.ts +2 -2
  474. package/src/core/codemirror/format.ts +6 -23
  475. package/src/core/codemirror/keymaps/vim.ts +4 -8
  476. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  477. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  478. package/src/core/codemirror/language/cell-code.ts +114 -0
  479. package/src/core/codemirror/language/extension.ts +52 -35
  480. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  481. package/src/core/codemirror/language/languages/python.ts +19 -13
  482. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  483. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  484. package/src/core/codemirror/language/types.ts +7 -7
  485. package/src/core/codemirror/language/utils.ts +9 -39
  486. package/src/core/codemirror/lsp/__tests__/notebook-lsp.test.ts +1 -2
  487. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  488. package/src/core/codemirror/lsp/utils.ts +1 -1
  489. package/src/core/codemirror/misc/__tests__/paste.test.ts +3 -1
  490. package/src/core/codemirror/misc/paste.ts +1 -1
  491. package/src/core/codemirror/placeholder/__tests__/extensions.test.ts +16 -0
  492. package/src/core/codemirror/placeholder/extensions.ts +57 -19
  493. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  494. package/src/core/codemirror/readonly/extension.ts +2 -2
  495. package/src/core/codemirror/utils.ts +1 -0
  496. package/src/core/config/__tests__/app-config.test.ts +64 -0
  497. package/src/core/config/__tests__/config-schema.test.ts +56 -6
  498. package/src/core/config/config-schema.ts +17 -4
  499. package/src/core/config/config.ts +65 -20
  500. package/src/core/config/feature-flag.tsx +0 -2
  501. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  502. package/src/core/datasets/data-source-discovery.ts +4 -6
  503. package/src/core/documentation/doc-lookup.ts +1 -1
  504. package/src/core/edit-app.tsx +21 -28
  505. package/src/core/errors/error-entries.ts +4 -1
  506. package/src/core/export/hooks.ts +1 -1
  507. package/src/core/export/layout.ts +2 -2
  508. package/src/core/hotkeys/actions.ts +2 -4
  509. package/src/core/hotkeys/hotkeys.ts +2 -1
  510. package/src/core/islands/__tests__/bootstrap.test.ts +69 -17
  511. package/src/core/islands/__tests__/bridge.test.ts +85 -27
  512. package/src/core/islands/bootstrap.ts +46 -53
  513. package/src/core/islands/bridge.ts +84 -25
  514. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  515. package/src/core/islands/components/web-components.tsx +1 -1
  516. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  517. package/src/core/islands/worker/controller.ts +4 -10
  518. package/src/core/islands/worker/worker.tsx +33 -33
  519. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  520. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  521. package/src/core/kernel/handlers.ts +19 -72
  522. package/src/core/meta/__tests__/code-visibility.test.tsx +78 -70
  523. package/src/core/meta/code-visibility.ts +18 -9
  524. package/src/core/network/__tests__/connection.test.ts +1 -1
  525. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  526. package/src/core/network/connection-notice.ts +1 -1
  527. package/src/core/network/connection.ts +1 -1
  528. package/src/core/network/requests-lazy.ts +2 -5
  529. package/src/core/network/requests-network.ts +5 -31
  530. package/src/core/network/requests-static.ts +1 -4
  531. package/src/core/network/requests-toasting.tsx +1 -4
  532. package/src/core/network/types.ts +6 -14
  533. package/src/core/network/useConnectionNotice.ts +8 -2
  534. package/src/core/notebook/__tests__/operations.test.ts +51 -0
  535. package/src/core/notebook/attachments.ts +38 -0
  536. package/src/core/notebook/document/__tests__/codemirror.test.ts +759 -0
  537. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  538. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  539. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  540. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  541. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  542. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  543. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  544. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  545. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  546. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  547. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  548. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  549. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  550. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  551. package/src/core/notebook/document/__tests__/user.ts +61 -0
  552. package/src/core/notebook/document/codemirror.ts +216 -0
  553. package/src/core/notebook/document/draft.ts +502 -0
  554. package/src/core/notebook/document/hooks.tsx +118 -0
  555. package/src/core/notebook/document/index.ts +24 -0
  556. package/src/core/notebook/document/model.ts +102 -0
  557. package/src/core/notebook/document/select.ts +41 -0
  558. package/src/core/notebook/document/state.ts +378 -0
  559. package/src/core/notebook/document/sync.ts +369 -0
  560. package/src/core/notebook/index.ts +19 -0
  561. package/src/core/notebook/operations.ts +30 -0
  562. package/src/core/notebook/replica/AGENTS.md +11 -0
  563. package/src/core/{host → notebook/replica}/__tests__/attachments.test.ts +9 -10
  564. package/src/core/{host → notebook/replica}/__tests__/document.test.ts +18 -16
  565. package/src/core/notebook/replica/__tests__/executions.test.ts +200 -0
  566. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  567. package/src/core/notebook/replica/__tests__/replica.test.ts +559 -0
  568. package/src/core/notebook/replica/__tests__/results.test.ts +193 -0
  569. package/src/core/notebook/replica/__tests__/runtime.test.ts +78 -0
  570. package/src/core/notebook/replica/backend.ts +19 -0
  571. package/src/core/notebook/replica/connection.ts +42 -0
  572. package/src/core/notebook/replica/index.ts +20 -0
  573. package/src/core/notebook/replica/replica.ts +336 -0
  574. package/src/core/{host → notebook/replica}/resources/attachments.ts +6 -3
  575. package/src/core/{host → notebook/replica}/resources/document.ts +3 -3
  576. package/src/core/notebook/replica/resources/executions.ts +157 -0
  577. package/src/core/notebook/replica/resources/results.ts +48 -0
  578. package/src/core/notebook/replica/resources/runtime.ts +14 -0
  579. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  580. package/src/core/packages/useSandboxController.ts +1 -1
  581. package/src/core/run-app.tsx +48 -51
  582. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  583. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  584. package/src/core/runtime/__tests__/runtime.test.ts +32 -74
  585. package/src/core/runtime/adapter.ts +1 -1
  586. package/src/core/runtime/config.ts +2 -2
  587. package/src/core/runtime/runtime.ts +33 -46
  588. package/src/core/runtime/types.ts +0 -2
  589. package/src/core/saving/__tests__/document.test.tsx +123 -0
  590. package/src/core/saving/__tests__/state.test.ts +111 -0
  591. package/src/core/saving/__tests__/useAutoSave.test.tsx +67 -0
  592. package/src/core/saving/document.ts +33 -0
  593. package/src/core/saving/filename.ts +1 -1
  594. package/src/core/saving/save-component.tsx +39 -60
  595. package/src/core/saving/state.ts +38 -39
  596. package/src/core/saving/useAutoSave.ts +17 -53
  597. package/src/core/session/__tests__/backend.test.ts +236 -0
  598. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +184 -0
  599. package/src/core/{websocket → session}/__tests__/useNotebookSession.hook.test.tsx +213 -262
  600. package/src/core/session/active-notebook.ts +28 -0
  601. package/src/core/session/backend.ts +123 -0
  602. package/src/core/session/embedded.ts +7 -0
  603. package/src/core/{websocket → session}/types.ts +2 -2
  604. package/src/core/session/useNotebookReplica.ts +74 -0
  605. package/src/core/{websocket → session}/useNotebookSession.tsx +77 -64
  606. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  607. package/src/core/state/jotai.ts +1 -1
  608. package/src/core/static/__tests__/files.test.ts +38 -10
  609. package/src/core/static/files.ts +2 -0
  610. package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
  611. package/src/core/transports/__tests__/server.test.ts +408 -0
  612. package/src/core/transports/__tests__/static.test.ts +306 -0
  613. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  614. package/src/core/{websocket → transports}/connection-state.ts +9 -16
  615. package/src/core/transports/server.ts +152 -0
  616. package/src/core/transports/static.ts +166 -0
  617. package/src/core/transports/wasm.ts +97 -0
  618. package/src/core/vscode/__tests__/vscode-bindings.test.ts +125 -0
  619. package/src/core/vscode/vscode-bindings.ts +14 -2
  620. package/src/core/wasm/__tests__/bridge.test.ts +210 -64
  621. package/src/core/wasm/__tests__/worker.test.ts +188 -0
  622. package/src/core/wasm/bridge.ts +119 -72
  623. package/src/core/wasm/rpc.ts +15 -3
  624. package/src/core/wasm/store.ts +3 -3
  625. package/src/core/wasm/worker/bootstrap.ts +55 -51
  626. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  627. package/src/core/wasm/worker/save-worker.ts +9 -10
  628. package/src/core/wasm/worker/types.ts +30 -18
  629. package/src/core/wasm/worker/worker.ts +118 -72
  630. package/src/css/app/codemirror.css +1 -1
  631. package/src/css/common.css +1 -1
  632. package/src/css/globals.css +66 -22
  633. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  634. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  635. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  636. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  637. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  638. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  639. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  640. package/src/hooks/useAsyncData.ts +3 -3
  641. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  642. package/src/hooks/useDebounce.ts +6 -10
  643. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  644. package/src/hooks/useDelayElapsed.ts +14 -10
  645. package/src/hooks/useElapsedTime.ts +3 -3
  646. package/src/hooks/useEventListener.ts +1 -2
  647. package/src/hooks/useInputHistory.ts +4 -7
  648. package/src/hooks/useInternalStateWithSync.ts +6 -6
  649. package/src/hooks/useResizeObserver.ts +4 -7
  650. package/src/hooks/useScript.ts +67 -70
  651. package/src/hooks/useStateForKey.ts +93 -0
  652. package/src/mount.tsx +20 -15
  653. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  654. package/src/plugins/core/registerReactComponent.tsx +1 -1
  655. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  656. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  657. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  658. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  659. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  660. package/src/plugins/impl/FormPlugin.tsx +1 -1
  661. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  662. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  663. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  664. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
  665. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  666. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  667. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  668. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  669. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  670. package/src/plugins/impl/anywidget/model.ts +3 -1
  671. package/src/plugins/impl/anywidget/registry.ts +18 -14
  672. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  673. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  674. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  675. package/src/plugins/impl/chat/chat-ui.tsx +15 -12
  676. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  677. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  678. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  679. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  680. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  681. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  682. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  683. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  684. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  685. package/src/plugins/impl/data-frames/panel.tsx +44 -16
  686. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  687. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  688. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  689. package/src/plugins/impl/panel/utils.ts +17 -0
  690. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  691. package/src/plugins/impl/plotly/selection.ts +25 -19
  692. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  693. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  694. package/src/plugins/layout/DownloadPlugin.tsx +7 -11
  695. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  696. package/src/plugins/layout/__test__/DownloadPlugin.test.tsx +205 -0
  697. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  698. package/src/stories/button.stories.tsx +1 -0
  699. package/src/stories/cell.stories.tsx +2 -5
  700. package/src/stories/data-table.stories.tsx +31 -0
  701. package/src/stories/editor.stories.tsx +24 -17
  702. package/src/stories/form-wrapper.stories.tsx +2 -2
  703. package/src/stories/layout/vertical/one-column.stories.tsx +28 -23
  704. package/src/stories/runtime-story.tsx +23 -17
  705. package/src/stories/select.stories.tsx +2 -2
  706. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  707. package/src/utils/__tests__/arrays.test.ts +35 -0
  708. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  709. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +127 -1
  710. package/src/utils/__tests__/download.test.tsx +183 -23
  711. package/src/utils/__tests__/numbers.test.ts +105 -1
  712. package/src/utils/__tests__/objects.test.ts +21 -0
  713. package/src/utils/arrays.ts +11 -2
  714. package/src/utils/blob.ts +1 -1
  715. package/src/utils/createReducer.ts +115 -79
  716. package/src/utils/download.ts +67 -6
  717. package/src/utils/maps.ts +0 -3
  718. package/src/utils/numbers.ts +79 -18
  719. package/src/utils/objects.ts +4 -9
  720. package/src/utils/tracer.ts +9 -5
  721. package/src/utils/zod-utils.ts +1 -3
  722. package/dist/_baseMerge-BiGRwuqv.js +0 -165
  723. package/dist/any-language-editor-D4q-IFcM.js +0 -76
  724. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  725. package/dist/assets/worker-HPY9fbhz.js +0 -114
  726. package/dist/code-visibility-XgDUZz2f.js +0 -138
  727. package/dist/dropdown-menu-DHP9P4VE.js +0 -4678
  728. package/dist/feedback-button-CoWyGXbB.js +0 -379
  729. package/dist/hasIn-o2Us0uVx.js +0 -132
  730. package/dist/html-to-image-BdA59IUD.js +0 -3251
  731. package/dist/isArrayLikeObject-BZl-15ir.js +0 -450
  732. package/dist/isEmpty-BSZ1LaUo.js +0 -35
  733. package/dist/isSymbol-kQcfYbNX.js +0 -8
  734. package/dist/label-DoysmLMW.js +0 -281
  735. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  736. package/dist/objectWithoutPropertiesLoose-BRDcTsfB.js +0 -23
  737. package/dist/pick-CJeoxYOS.js +0 -51
  738. package/dist/precisionRound-DMIVQrt2.js +0 -97
  739. package/dist/select-DemoKT4G.js +0 -1344
  740. package/dist/toDate-B5mIgxwZ.js +0 -1185
  741. package/dist/toast-PkdZraM8.js +0 -1830
  742. package/dist/use-restore-focus-CKjfFjLM.js +0 -6369
  743. package/dist/useLifecycle-BGWzTwHK.js +0 -169
  744. package/dist/useTheme-BhQvec43.js +0 -904
  745. package/src/components/editor/ai/__tests__/staged-cell-submission.test.ts +0 -106
  746. package/src/components/editor/ai/add-cell-with-ai.tsx +0 -506
  747. package/src/components/editor/ai/staged-cell-submission.ts +0 -46
  748. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +0 -64
  749. package/src/components/editor/cell/code/__tests__/rtc-editor-config.test.ts +0 -50
  750. package/src/components/editor/chrome/wrapper/footer-items/rtc-status.tsx +0 -65
  751. package/src/components/editor/navigation/__tests__/state.test.ts +0 -81
  752. package/src/components/editor/navigation/state.ts +0 -65
  753. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +0 -656
  754. package/src/core/ai/tools/run-cells-tool.ts +0 -303
  755. package/src/core/attachments/state.ts +0 -18
  756. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  757. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  758. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  759. package/src/core/cells/document-changes.ts +0 -944
  760. package/src/core/codemirror/rtc/extension.ts +0 -453
  761. package/src/core/codemirror/rtc/loro/__tests__/sync.test.ts +0 -52
  762. package/src/core/codemirror/rtc/loro/awareness.ts +0 -531
  763. package/src/core/codemirror/rtc/loro/colors.ts +0 -38
  764. package/src/core/codemirror/rtc/loro/sync.ts +0 -168
  765. package/src/core/host/AGENTS.md +0 -11
  766. package/src/core/host/__tests__/events.test.ts +0 -46
  767. package/src/core/host/__tests__/fixtures.ts +0 -39
  768. package/src/core/host/__tests__/notebook-client.test.ts +0 -433
  769. package/src/core/host/__tests__/protocol.test.ts +0 -214
  770. package/src/core/host/__tests__/stream.test.ts +0 -209
  771. package/src/core/host/connection.ts +0 -81
  772. package/src/core/host/events.ts +0 -61
  773. package/src/core/host/notebook-client.ts +0 -290
  774. package/src/core/host/protocol.ts +0 -1943
  775. package/src/core/host/resources/resource.ts +0 -47
  776. package/src/core/host/stream.ts +0 -211
  777. package/src/core/rtc/state.ts +0 -34
  778. package/src/core/state/__tests__/jotai.test.ts +0 -25
  779. package/src/core/wasm/worker/message-buffer.ts +0 -33
  780. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  781. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -170
  782. package/src/core/websocket/state.ts +0 -7
  783. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  784. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  785. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  786. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  787. package/src/core/websocket/transports/basic.ts +0 -102
  788. package/src/core/websocket/transports/create.ts +0 -31
  789. package/src/core/websocket/transports/sse.ts +0 -308
  790. package/src/core/websocket/transports/transport.ts +0 -50
  791. package/src/core/websocket/transports/ws.ts +0 -98
  792. package/src/core/websocket/useNotebookClient.ts +0 -88
  793. package/src/utils/__tests__/id-tree.test.ts +0 -2145
  794. package/src/utils/id-tree.tsx +0 -1134
  795. /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
  796. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -0,0 +1,759 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { history, undo } from "@codemirror/commands";
4
+ import { MySQL, PostgreSQL } from "@codemirror/lang-sql";
5
+ import { language as activeLanguage } from "@codemirror/language";
6
+ import { EditorSelection, EditorState } from "@codemirror/state";
7
+ import { EditorView } from "@codemirror/view";
8
+ import type { CellId } from "@marimo-team/client/protocol";
9
+ import { atom, createStore } from "jotai/vanilla";
10
+ import { afterEach, describe, expect, it, vi } from "vitest";
11
+ import { cellId as brandedCellId } from "@/__tests__/branded";
12
+ import { createNotebookActions } from "@/core/cells/cells";
13
+ import { OverridingHotkeyProvider } from "@/core/hotkeys/hotkeys";
14
+ import { cellActionsState } from "../../../codemirror/cells/state";
15
+ import { cellConfigExtension } from "../../../codemirror/config/extension";
16
+ import {
17
+ adaptiveLanguageConfiguration,
18
+ languageAdapterState,
19
+ switchLanguage,
20
+ } from "../../../codemirror/language/extension";
21
+ import {
22
+ languageMetadataField,
23
+ setLanguageMetadata,
24
+ } from "../../../codemirror/language/metadata";
25
+ import {
26
+ editorShowsCellCode,
27
+ getCellCode,
28
+ } from "../../../codemirror/language/cell-code";
29
+ import { SCHEMA_CACHE } from "../../../codemirror/language/languages/sql/completion-store";
30
+ import type { Document } from "../../replica/resources/document";
31
+ import { cellCodeBinding } from "../codemirror";
32
+ import { createDocumentModel } from "../model";
33
+ import { fakeBackend } from "../../replica/__tests__/fixtures";
34
+ import { cell, document, withCells } from "./fixtures";
35
+
36
+ const views: EditorView[] = [];
37
+ const lifetimes: AbortController[] = [];
38
+
39
+ afterEach(() => {
40
+ for (const view of views.splice(0)) {
41
+ view.destroy();
42
+ }
43
+ lifetimes.splice(0).forEach((controller) => controller.abort());
44
+ });
45
+
46
+ const completionConfig = {
47
+ activate_on_typing: true,
48
+ signature_hint_on_typing: false,
49
+ copilot: false,
50
+ codeium_api_key: null,
51
+ } as const;
52
+
53
+ /** The language configuration a marimo cell editor installs. */
54
+ function languages(id: CellId) {
55
+ const options = {
56
+ cellId: brandedCellId(id),
57
+ completionConfig,
58
+ hotkeys: new OverridingHotkeyProvider({}),
59
+ placeholderType: "marimo-import",
60
+ lspConfig: {},
61
+ } as const;
62
+ return [
63
+ cellActionsState.of({
64
+ ...createNotebookActions(vi.fn()),
65
+ toggleHideCode: () => false,
66
+ askAiAboutCell: () => undefined,
67
+ aiCellCompletion: () => false,
68
+ createManyBelow: vi.fn(),
69
+ addOrAppendSetupCell: vi.fn(),
70
+ setCellCode: vi.fn(),
71
+ createCell: vi.fn(),
72
+ moveToNextCell: vi.fn(),
73
+ addSetupCell: vi.fn(),
74
+ deleteCells: vi.fn(),
75
+ undoDeleteCell: vi.fn(),
76
+ focusCell: vi.fn(),
77
+ focusTopCell: vi.fn(),
78
+ focusBottomCell: vi.fn(),
79
+ collapseCell: vi.fn(),
80
+ expandCell: vi.fn(),
81
+ collapseAllCells: vi.fn(),
82
+ expandAllCells: vi.fn(),
83
+ showCellIfHidden: vi.fn(),
84
+ moveCell: vi.fn(),
85
+ moveCellToIndex: vi.fn(),
86
+ moveCellsRelativeTo: vi.fn(),
87
+ dropCellOverCell: vi.fn(),
88
+ dropCellOverColumn: vi.fn(),
89
+ sendToTop: vi.fn(),
90
+ sendToBottom: vi.fn(),
91
+ addColumn: vi.fn(),
92
+ addColumnBreakpoint: vi.fn(),
93
+ deleteColumn: vi.fn(),
94
+ mergeAllColumns: vi.fn(),
95
+ compactColumns: vi.fn(),
96
+ moveColumn: vi.fn(),
97
+ splitCell: vi.fn(),
98
+ undoSplitCell: vi.fn(),
99
+ onRun: vi.fn(),
100
+ deleteCell: vi.fn(),
101
+ afterToggleMarkdown: vi.fn(),
102
+ afterToggleSQL: vi.fn(),
103
+ saveNotebook: vi.fn(),
104
+ }),
105
+ adaptiveLanguageConfiguration(options),
106
+ cellConfigExtension({ ...options, diagnosticsConfig: {} }),
107
+ ];
108
+ }
109
+
110
+ function setup({
111
+ initial = document(),
112
+ cellId = "c1",
113
+ doc = "",
114
+ adapters = false,
115
+ }: {
116
+ initial?: Document | null;
117
+ cellId?: CellId;
118
+ doc?: string;
119
+ adapters?: boolean;
120
+ } = {}) {
121
+ const store = createStore();
122
+ const source = atom(initial);
123
+ const model = createDocumentModel({
124
+ store,
125
+ documentAtom: source,
126
+ backendAtom: atom(fakeBackend({ edit: () => new Promise(() => {}) })),
127
+ });
128
+ const controller = new AbortController();
129
+ lifetimes.push(controller);
130
+ model.start({ signal: controller.signal });
131
+ const editor = (text = doc) => {
132
+ const view = new EditorView({
133
+ state: EditorState.create({
134
+ doc: text,
135
+ extensions: [
136
+ history(),
137
+ adapters ? languages(cellId) : [],
138
+ cellCodeBinding({ store, model, cellId }),
139
+ ],
140
+ }),
141
+ });
142
+ views.push(view);
143
+ return view;
144
+ };
145
+ const view = editor();
146
+ return {
147
+ store,
148
+ source,
149
+ model,
150
+ view,
151
+ editor,
152
+ code: () => store.get(model.documentAtom)?.cells.get(cellId)?.code,
153
+ status: () => store.get(model.syncStatusAtom),
154
+ type: (insert: string) =>
155
+ view.dispatch({
156
+ changes: { from: view.state.doc.length, insert },
157
+ userEvent: "input.type",
158
+ }),
159
+ };
160
+ }
161
+
162
+ // Inbound changes apply in a microtask.
163
+ const settle = () => new Promise<void>((resolve) => setTimeout(resolve, 0));
164
+
165
+ describe("cellCodeBinding", () => {
166
+ it("seeds the editor from the cell outside undo history", async () => {
167
+ const { view, status } = setup();
168
+ await settle();
169
+ expect(view.state.doc.toString()).toBe("x = 1");
170
+ expect(undo(view)).toBe(false);
171
+ expect(status()).toEqual({ status: "idle" });
172
+ });
173
+
174
+ it("writes local changes to the draft before the update returns", async () => {
175
+ const { code, status, type } = setup();
176
+ await settle();
177
+ type(" + 1");
178
+ expect(code()).toBe("x = 1 + 1");
179
+ expect(status()).toEqual({ status: "pending" });
180
+ });
181
+
182
+ it("shows a change another subscriber makes while a push notifies", async () => {
183
+ const { store, model, view, type } = setup();
184
+ await settle();
185
+ // Another subscriber normalizes code synchronously, inside the push.
186
+ const unsubscribe = store.sub(model.documentAtom, () => {
187
+ const code = store.get(model.documentAtom)?.cells.get("c1")?.code;
188
+ if (code?.includes(" + ")) {
189
+ const normalized = code.replaceAll(" + ", "+");
190
+ store.set(model.editAtom, (tx) => tx.setCode("c1", normalized));
191
+ }
192
+ });
193
+ type(" + 1");
194
+ await settle();
195
+ expect(view.state.doc.toString()).toBe("x = 1+1");
196
+ unsubscribe();
197
+ });
198
+
199
+ it("keeps the selection between separate host edits", async () => {
200
+ const { store, source, view } = setup({
201
+ initial: withCells(document(), { c1: cell("c1", "a = 1\nb = 2\nc = 3") }),
202
+ });
203
+ await settle();
204
+ view.dispatch({ selection: EditorSelection.cursor(8) });
205
+
206
+ store.set(
207
+ source,
208
+ withCells(document(), { c1: cell("c1", "A = 1\nb = 2\nC = 3") }),
209
+ );
210
+ await settle();
211
+
212
+ expect(view.state.doc.toString()).toBe("A = 1\nb = 2\nC = 3");
213
+ expect(view.state.selection.main.head).toBe(8);
214
+ });
215
+
216
+ it("waits for composition to end before applying host changes", async () => {
217
+ const { store, source, view } = setup();
218
+ await settle();
219
+ let composing = true;
220
+ Object.defineProperty(view, "composing", { get: () => composing });
221
+
222
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 2") }));
223
+ await settle();
224
+ expect(view.state.doc.toString()).toBe("x = 1");
225
+
226
+ // A composition can end without an update, so the binding watches for it.
227
+ composing = false;
228
+ view.contentDOM.dispatchEvent(new Event("compositionend"));
229
+ await settle();
230
+ expect(view.state.doc.toString()).toBe("x = 2");
231
+ });
232
+
233
+ it("applies a waiting host change on the first update after composition", async () => {
234
+ const { store, source, view } = setup();
235
+ await settle();
236
+ let composing = true;
237
+ Object.defineProperty(view, "composing", { get: () => composing });
238
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 2") }));
239
+ await settle();
240
+
241
+ composing = false;
242
+ view.dispatch({ selection: EditorSelection.cursor(0) });
243
+ await settle();
244
+ expect(view.state.doc.toString()).toBe("x = 2");
245
+ });
246
+
247
+ it("keeps two editors on one cell in step", async () => {
248
+ const { editor, type } = setup();
249
+ const other = editor();
250
+ await settle();
251
+
252
+ type("!");
253
+ await settle();
254
+
255
+ expect(other.state.doc.toString()).toBe("x = 1!");
256
+ });
257
+
258
+ it("shows host code that returns to what this editor wrote", async () => {
259
+ const { store, source, view, type } = setup();
260
+ await settle();
261
+ type("!");
262
+ // The host confirms the edit, which releases the draft.
263
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 1!") }));
264
+ await settle();
265
+
266
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 2") }));
267
+ await settle();
268
+ expect(view.state.doc.toString()).toBe("x = 2");
269
+
270
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 1!") }));
271
+ await settle();
272
+ expect(view.state.doc.toString()).toBe("x = 1!");
273
+ });
274
+
275
+ it("follows a cell once its code is available", async () => {
276
+ const absent = setup({ cellId: "c3" });
277
+ const withheld = setup({
278
+ initial: withCells(document(), { c1: cell("c1", "", { code: null }) }),
279
+ });
280
+ await settle();
281
+
282
+ absent.store.set(
283
+ absent.source,
284
+ withCells(document(), { c3: cell("c3", "new") }),
285
+ );
286
+ withheld.store.set(
287
+ withheld.source,
288
+ withCells(document(), { c1: cell("c1", "revealed") }),
289
+ );
290
+ await settle();
291
+
292
+ expect(absent.view.state.doc.toString()).toBe("new");
293
+ expect(withheld.view.state.doc.toString()).toBe("revealed");
294
+ });
295
+
296
+ it("shows the host's code after typing while the document was unavailable", async () => {
297
+ const { store, source, view, type, code } = setup();
298
+ await settle();
299
+
300
+ // All in one task, so no pull runs while the document is away.
301
+ store.set(source, null);
302
+ type(" + 1");
303
+ store.set(source, document());
304
+ await settle();
305
+
306
+ expect(view.state.doc.toString()).toBe(code());
307
+ });
308
+
309
+ it("leaves the editor alone for an absent cell or withheld source", async () => {
310
+ const absent = setup({ cellId: "c9", doc: "local" });
311
+ const withheld = setup({
312
+ initial: withCells(document(), { c1: cell("c1", "", { code: null }) }),
313
+ doc: "local",
314
+ });
315
+ await settle();
316
+
317
+ for (const { view, type, status } of [absent, withheld]) {
318
+ type("!");
319
+ expect(view.state.doc.toString()).toBe("local!");
320
+ expect(status()).toEqual({ status: "idle" });
321
+ }
322
+ });
323
+
324
+ it("stops following the cell when the view is destroyed", async () => {
325
+ const { store, source, view } = setup();
326
+ await settle();
327
+
328
+ // The change arrives before the view goes, and would apply just after.
329
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 2") }));
330
+ view.destroy();
331
+ await settle();
332
+
333
+ expect(view.state.doc.toString()).toBe("x = 1");
334
+ });
335
+
336
+ describe("with language adapters", () => {
337
+ const language = (view: EditorView) =>
338
+ view.state.field(languageAdapterState).type;
339
+ /** What the editor's adapter reads back from cell code. */
340
+ const readBack = (view: EditorView, code: string | null | undefined) => {
341
+ const [text, , metadata] = view.state
342
+ .field(languageAdapterState)
343
+ .transformIn(code ?? "");
344
+ return { text, metadata };
345
+ };
346
+
347
+ it("edits markdown as markdown and writes Python with its metadata", async () => {
348
+ const { view, code, type } = setup({
349
+ adapters: true,
350
+ initial: withCells(document(), {
351
+ c1: cell("c1", 'mo.md(f"""hi {x}""")'),
352
+ }),
353
+ });
354
+ await settle();
355
+ expect(language(view)).toBe("markdown");
356
+ expect(view.state.doc.toString()).toBe("hi {x}");
357
+
358
+ type("!");
359
+ expect(readBack(view, code())).toMatchObject({
360
+ text: "hi {x}!",
361
+ metadata: { quotePrefix: "f" },
362
+ });
363
+ });
364
+
365
+ it("keeps Python mode for host code without rewriting it", async () => {
366
+ const { store, source, view, code, status } = setup({ adapters: true });
367
+ await settle();
368
+ expect(language(view)).toBe("python");
369
+
370
+ store.set(
371
+ source,
372
+ withCells(document(), { c1: cell("c1", 'mo.md("hi")') }),
373
+ );
374
+ await settle();
375
+
376
+ expect(language(view)).toBe("python");
377
+ expect(view.state.doc.toString()).toBe('mo.md("hi")');
378
+ expect(code()).toBe('mo.md("hi")');
379
+ expect(status()).toEqual({ status: "idle" });
380
+ });
381
+
382
+ it("keeps Python mode while typing mo.md() and receiving its echo", async () => {
383
+ const { store, source, view, type } = setup({
384
+ adapters: true,
385
+ initial: withCells(document(), { c1: cell("c1", "") }),
386
+ });
387
+ await settle();
388
+ type("mo.md");
389
+ await settle();
390
+ type("()");
391
+ await settle();
392
+ store.set(source, withCells(document(), { c1: cell("c1", "mo.md()") }));
393
+ await settle();
394
+
395
+ expect(language(view)).toBe("python");
396
+ expect(view.state.doc.toString()).toBe("mo.md()");
397
+ });
398
+
399
+ it("falls back to Python when host code no longer fits Markdown", async () => {
400
+ const { store, source, view, code, status } = setup({
401
+ adapters: true,
402
+ initial: withCells(document(), { c1: cell("c1", 'mo.md("hi")') }),
403
+ });
404
+ await settle();
405
+ expect(language(view)).toBe("markdown");
406
+
407
+ store.set(source, withCells(document(), { c1: cell("c1", "x = 1") }));
408
+ await settle();
409
+
410
+ expect(language(view)).toBe("python");
411
+ expect(view.state.doc.toString()).toBe("x = 1");
412
+ expect(code()).toBe("x = 1");
413
+ expect(status()).toEqual({ status: "idle" });
414
+ });
415
+
416
+ it("preserves Python when typing before the initial pull", async () => {
417
+ const { view, type } = setup({
418
+ adapters: true,
419
+ initial: withCells(document(), { c1: cell("c1", "") }),
420
+ });
421
+ type("mo.md()");
422
+ await settle();
423
+
424
+ expect(language(view)).toBe("python");
425
+ expect(view.state.doc.toString()).toBe("mo.md()");
426
+ });
427
+
428
+ it("writes changes to language metadata alone", async () => {
429
+ const { view, code } = setup({
430
+ adapters: true,
431
+ initial: withCells(document(), {
432
+ c1: cell("c1", 'mo.md("""\n# Title\n""")'),
433
+ }),
434
+ });
435
+ await settle();
436
+
437
+ view.dispatch({
438
+ effects: setLanguageMetadata.of({
439
+ ...view.state.field(languageMetadataField),
440
+ quotePrefix: "r",
441
+ }),
442
+ });
443
+ expect(readBack(view, code())).toMatchObject({
444
+ text: "# Title",
445
+ metadata: { quotePrefix: "r" },
446
+ });
447
+ });
448
+
449
+ it("applies a host change to language metadata alone", async () => {
450
+ const { store, source, view } = setup({
451
+ adapters: true,
452
+ initial: withCells(document(), {
453
+ c1: cell("c1", 'mo.md(r"""hi""")'),
454
+ }),
455
+ });
456
+ await settle();
457
+
458
+ store.set(
459
+ source,
460
+ withCells(document(), { c1: cell("c1", 'mo.md(f"""hi""")') }),
461
+ );
462
+ await settle();
463
+
464
+ expect(view.state.doc.toString()).toBe("hi");
465
+ expect(view.state.field(languageMetadataField)).toMatchObject({
466
+ quotePrefix: "f",
467
+ });
468
+ });
469
+
470
+ it("does not rewrite code when the editor switches language as it opens", async () => {
471
+ const original = 'mo.md("hello")';
472
+ const { view, code, status } = setup({
473
+ adapters: true,
474
+ doc: original,
475
+ initial: withCells(document(), { c1: cell("c1", original) }),
476
+ });
477
+ // The cell editor switches to the language its code implies on mount.
478
+ switchLanguage(view, { language: "markdown" });
479
+ await settle();
480
+
481
+ expect(code()).toBe(original);
482
+ expect(status()).toEqual({ status: "idle" });
483
+ });
484
+
485
+ it("keeps typed text that the adapter would normalize", async () => {
486
+ const { view, type } = setup({
487
+ adapters: true,
488
+ initial: withCells(document(), {
489
+ c1: cell("c1", 'mo.md("""\nhi\n""")'),
490
+ }),
491
+ });
492
+ await settle();
493
+
494
+ type("\n");
495
+ await settle();
496
+ type(" ");
497
+ await settle();
498
+
499
+ expect(view.state.doc.toString()).toBe("hi\n ");
500
+ });
501
+
502
+ it("keeps the language a person chose when the cell changes otherwise", async () => {
503
+ const { store, model, view } = setup({
504
+ adapters: true,
505
+ initial: withCells(document(), {
506
+ c1: cell("c1", 'mo.md("""\nhi\n""")'),
507
+ }),
508
+ });
509
+ await settle();
510
+ switchLanguage(view, { language: "python" });
511
+ await settle();
512
+
513
+ store.set(model.editAtom, (tx) => tx.setName("c1", "renamed"));
514
+ await settle();
515
+
516
+ expect(language(view)).toBe("python");
517
+ });
518
+
519
+ it("edits SQL as SQL and keeps its output name", async () => {
520
+ const { view, code, type } = setup({
521
+ adapters: true,
522
+ initial: withCells(document(), {
523
+ c1: cell("c1", 'df = mo.sql(f"""SELECT 1""")'),
524
+ }),
525
+ });
526
+ await settle();
527
+ expect(language(view)).toBe("sql");
528
+ expect(view.state.doc.toString()).toBe("SELECT 1");
529
+
530
+ type(" LIMIT 1");
531
+ expect(readBack(view, code())).toMatchObject({
532
+ text: "SELECT 1 LIMIT 1",
533
+ metadata: { dataframeName: "df" },
534
+ });
535
+ });
536
+
537
+ it("sets the SQL dialect from the host's engine", async () => {
538
+ const { store, source, view } = setup({
539
+ adapters: true,
540
+ initial: withCells(document(), {
541
+ c1: cell("c1", 'df = mo.sql("SELECT 1")'),
542
+ }),
543
+ });
544
+ await Promise.resolve();
545
+ const getDialect = SCHEMA_CACHE.getDialect.bind(SCHEMA_CACHE);
546
+ const dialects = new Map([
547
+ ["warehouse", PostgreSQL],
548
+ ["lake", MySQL],
549
+ ]);
550
+ const dialect = vi
551
+ .spyOn(SCHEMA_CACHE, "getDialect")
552
+ .mockImplementation(
553
+ (engine) => dialects.get(engine) ?? getDialect(engine),
554
+ );
555
+ try {
556
+ store.set(
557
+ source,
558
+ withCells(document(), {
559
+ c1: cell("c1", '_df = mo.sql(f"""SELECT 1""", engine=warehouse)'),
560
+ }),
561
+ );
562
+ await Promise.resolve();
563
+ expect(view.state.facet(activeLanguage)).toBe(PostgreSQL.language);
564
+
565
+ store.set(
566
+ source,
567
+ withCells(document(), {
568
+ c1: cell("c1", '_df = mo.sql(f"""SELECT 1""", engine=lake)'),
569
+ }),
570
+ );
571
+ await Promise.resolve();
572
+ expect(view.state.facet(activeLanguage)).toBe(MySQL.language);
573
+ } finally {
574
+ dialect.mockRestore();
575
+ }
576
+ });
577
+ });
578
+ });
579
+
580
+ describe("cellCodeBinding before the cell has source", () => {
581
+ it("still infers the language when the source first arrives", async () => {
582
+ const { store, view, source, type } = setup({
583
+ initial: null,
584
+ adapters: true,
585
+ });
586
+ type("x");
587
+ await settle();
588
+ store.set(source, withCells(document(), { c1: cell("c1", 'mo.md("hi")') }));
589
+ await settle();
590
+ expect(view.state.field(languageAdapterState).type).toBe("markdown");
591
+ });
592
+ });
593
+
594
+ describe("cellCodeBinding under random interleavings", () => {
595
+ // A small deterministic generator, so a failure names the seed to replay.
596
+ function generator(seed: number) {
597
+ let state = seed >>> 0;
598
+ return () => {
599
+ state = (Math.imul(state, 1664525) + 1013904223) >>> 0;
600
+ return state / 2 ** 32;
601
+ };
602
+ }
603
+
604
+ const TOKENS = ["x", " ", "\n", "= 1", "mo.md(", ")", '"""', "#", "y + 2"];
605
+ const REMOTE = [
606
+ "x = 2",
607
+ 'mo.md(r"""hi""")',
608
+ "y = x + 1\nz = y",
609
+ "",
610
+ "# other",
611
+ ];
612
+
613
+ async function stress(seed: number, adapters: boolean) {
614
+ const random = generator(seed);
615
+ const pick = <T>(items: readonly T[]) =>
616
+ items[Math.floor(random() * items.length)];
617
+ const { store, source, model, view, editor, code } = setup({
618
+ adapters,
619
+ initial: withCells(document(), { c1: cell("c1", "x = 1") }),
620
+ });
621
+ const other = editor();
622
+ await settle();
623
+ let composing = false;
624
+ Object.defineProperty(view, "composing", { get: () => composing });
625
+ let version = 1;
626
+ let hostCode = "x = 1";
627
+ const host = (next: string | null) => {
628
+ version += 1;
629
+ store.set(
630
+ source,
631
+ withCells(document(), {
632
+ c1:
633
+ next === null
634
+ ? null
635
+ : cell("c1", next, { version, etag: `c1-${version}` }),
636
+ }),
637
+ );
638
+ };
639
+ const agrees = (state = view.state) => {
640
+ const current = code();
641
+ return typeof current !== "string" || editorShowsCellCode(state, current);
642
+ };
643
+ // Every write to the model is one local or host step. A binding that
644
+ // echoed its own pulls back as pushes would write many more.
645
+ let writes = 0;
646
+ let steps = 0;
647
+ // Checks inside the loop collect here, so one assertion reports them all.
648
+ const failures: string[] = [];
649
+ const check = (label: string, ok: boolean) => {
650
+ if (!ok) {
651
+ failures.push(label);
652
+ }
653
+ };
654
+ const unsubscribe = store.sub(model.documentAtom, () => {
655
+ writes += 1;
656
+ });
657
+
658
+ for (let step = 0; step < 200; step++) {
659
+ const roll = random();
660
+ steps += 1;
661
+ if (roll < 0.35) {
662
+ const length = view.state.doc.length;
663
+ const from = Math.floor(random() * (length + 1));
664
+ view.dispatch({
665
+ changes: { from, insert: pick(TOKENS) },
666
+ userEvent: "input.type",
667
+ });
668
+ // A local change reaches the model before the update returns.
669
+ check(`step ${step} after typing`, agrees());
670
+ } else if (roll < 0.45) {
671
+ const length = view.state.doc.length;
672
+ if (length > 0) {
673
+ const from = Math.floor(random() * length);
674
+ const to = Math.min(length, from + 1 + Math.floor(random() * 5));
675
+ view.dispatch({
676
+ changes: { from, to },
677
+ userEvent: "delete.backward",
678
+ });
679
+ check(`step ${step} after deleting`, agrees());
680
+ }
681
+ } else if (roll < 0.6) {
682
+ // The host confirms what this client wrote.
683
+ const current = code();
684
+ if (typeof current === "string") {
685
+ hostCode = current;
686
+ host(hostCode);
687
+ }
688
+ } else if (roll < 0.75) {
689
+ // Another client changes the cell.
690
+ hostCode = pick(REMOTE);
691
+ host(hostCode);
692
+ } else if (roll < 0.8) {
693
+ // The cell leaves the document and returns.
694
+ host(null);
695
+ await settle();
696
+ host(hostCode);
697
+ } else if (roll < 0.88) {
698
+ composing = !composing;
699
+ if (!composing) {
700
+ view.contentDOM.dispatchEvent(new Event("compositionend"));
701
+ }
702
+ } else if (roll < 0.94) {
703
+ other.dispatch({
704
+ changes: { from: other.state.doc.length, insert: pick(TOKENS) },
705
+ userEvent: "input.type",
706
+ });
707
+ } else if (roll < 0.97) {
708
+ // The host withholds the cell's source for a while.
709
+ version += 1;
710
+ store.set(
711
+ source,
712
+ withCells(document(), {
713
+ c1: cell("c1", "", { code: null, version, etag: `c1-${version}` }),
714
+ }),
715
+ );
716
+ } else {
717
+ // The document goes away and comes back.
718
+ store.set(source, null);
719
+ await settle();
720
+ host(hostCode);
721
+ }
722
+ if (random() < 0.5) {
723
+ await settle();
724
+ check(`step ${step} after settling`, composing || agrees());
725
+ check(`step ${step} other editor`, agrees(other.state));
726
+ }
727
+ }
728
+ unsubscribe();
729
+ check(`${writes} writes for ${steps} steps`, writes <= 2 * steps);
730
+
731
+ composing = false;
732
+ view.contentDOM.dispatchEvent(new Event("compositionend"));
733
+ host(hostCode);
734
+ await settle();
735
+ await settle();
736
+ const current = code();
737
+ check("cell present at the end", typeof current === "string");
738
+ check("editor agrees at the end", agrees());
739
+ check("other editor agrees at the end", agrees(other.state));
740
+ // Without adapters the agreement is plain text equality.
741
+ check(
742
+ "text equal at the end",
743
+ adapters || getCellCode(view.state) === current,
744
+ );
745
+ return failures;
746
+ }
747
+
748
+ const seeds = Array.from({ length: 16 }, (_, i) => i + 1);
749
+ for (const seed of seeds) {
750
+ it(`keeps editor and model in agreement, seed ${seed}`, async () => {
751
+ expect(await stress(seed, false)).toEqual([]);
752
+ });
753
+ }
754
+ for (const seed of seeds) {
755
+ it(`keeps editor and model in agreement with language adapters, seed ${seed}`, async () => {
756
+ expect(await stress(seed, true)).toEqual([]);
757
+ });
758
+ }
759
+ });