@marimo-team/islands 0.25.3-dev12 → 0.25.3-dev122

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 (391) hide show
  1. package/dist/{ConnectedDataExplorerComponent-CNMliVzw.js → ConnectedDataExplorerComponent-CFzqIsUg.js} +14 -14
  2. package/dist/{ErrorBoundary-CUd8VnYI.js → ErrorBoundary-Mby9UXAS.js} +8 -8
  3. package/dist/{abnfDiagram-VCTEODGH-D-Ez91gV.js → abnfDiagram-VCTEODGH-Di400kYu.js} +2 -2
  4. package/dist/any-language-editor-k2B8YI3Y.js +75 -0
  5. package/dist/{architectureDiagram-5GKGNRK7-CTpcNtO8.js → architectureDiagram-5GKGNRK7-BdTJcsNJ.js} +1 -1
  6. package/dist/{arrays-CN1djC4k.js → arrays-R6JsImiT.js} +8 -0
  7. package/dist/assets/__vite-browser-external-SaDYcg7x.js +1 -0
  8. package/dist/assets/{worker-HPY9fbhz.js → worker-CfME3uyi.js} +3 -2
  9. package/dist/{button-CvjkwdSk.js → button-BxB3RDjN.js} +1 -1
  10. package/dist/{capabilities-CQ5IjhzL.js → capabilities-BgR0osfN.js} +2 -2
  11. package/dist/{chat-ui-DaHynk8g.js → chat-ui-DA8zEZ1I.js} +1141 -1141
  12. package/dist/{check-EDwSwP89.js → check-P35dc2dH.js} +1 -1
  13. package/dist/{chunk-IMKFNOWR-DGxi-zeo.js → chunk-IMKFNOWR-mtD7aJBw.js} +1 -1
  14. package/dist/{chunk-SHT3W25Y-CV9J3rAB.js → chunk-SHT3W25Y-DybB1ShH.js} +1 -1
  15. package/dist/{chunk-SVP7TREG-BL-9Fqgg.js → chunk-SVP7TREG-DNhhdCex.js} +1 -1
  16. package/dist/{chunk-TICWLB2K-Cu_XMjDM.js → chunk-TICWLB2K-NhAQhlDf.js} +1 -1
  17. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  18. package/dist/{classDiagram-ZZMXUADV-BXXV13Hq.js → classDiagram-ZZMXUADV-GTL0l5xS.js} +2 -2
  19. package/dist/{classDiagram-v2-VYDZK3BY-B7FXJ1Cp.js → classDiagram-v2-VYDZK3BY-BWrM2ruI.js} +2 -2
  20. package/dist/{code-visibility-P0Ornave.js → code-visibility-3U6ZTDXf.js} +6 -6
  21. package/dist/{common-BtO98PuE.js → common-CKeY7M4s.js} +5559 -5453
  22. package/dist/{copy-icon-Ft7t9bqz.js → copy-icon-CPoxUG7G.js} +3 -3
  23. package/dist/{cynefinDiagram-5FMLGOSQ-CH96rwba.js → cynefinDiagram-5FMLGOSQ-CX22wa7c.js} +1 -1
  24. package/dist/{dagre-GXQ25YYZ-BD3PmpFY.js → dagre-GXQ25YYZ-Df6Pp-cl.js} +1 -1
  25. package/dist/{data-grid-overlay-editor-B801aMeG.js → data-grid-overlay-editor-YfAOLpti.js} +5 -5
  26. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  27. package/dist/{diagram-S7CK7UJ4-BsSJJldB.js → diagram-S7CK7UJ4-CctU9kVx.js} +1 -1
  28. package/dist/{diagram-UQ7AKVKN-muQtNZc4.js → diagram-UQ7AKVKN-FVHjCF9j.js} +1 -1
  29. package/dist/{diagram-VX7I27RA-CdhuZAqR.js → diagram-VX7I27RA-BSDbwFFv.js} +1 -1
  30. package/dist/{diagram-Z3DM3KII-BD8mXjOj.js → diagram-Z3DM3KII-BX8kW7Nc.js} +1 -1
  31. package/dist/{dialog-CEBFZNyc.js → dialog-C1PcpCqX.js} +3 -3
  32. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  33. package/dist/{dropdown-menu-CP_qCYHC.js → dropdown-menu-B_7xD2X4.js} +177 -169
  34. package/dist/{ebnfDiagram-PWID7BFC-BIN1sweE.js → ebnfDiagram-PWID7BFC-DtapdLet.js} +2 -2
  35. package/dist/{erDiagram-RLTQ6QDP-Ctp5cfoN.js → erDiagram-RLTQ6QDP-BAru9jHt.js} +1 -1
  36. package/dist/{error-banner-D-WCIiMt.js → error-banner-Sl2FlSJ4.js} +2 -2
  37. package/dist/{error-entries-Bv6bZPF3.js → error-entries-g9iQiVIT.js} +844 -811
  38. package/dist/{feedback-button-Bpu-arCP.js → feedback-button-Y-QYprZR.js} +29 -29
  39. package/dist/{flowDiagram-HODETNUW-3EfLrlFb.js → flowDiagram-HODETNUW-CjaeFeea.js} +2 -2
  40. package/dist/{formats-27CW_UP4.js → formats-V8ifA0Bk.js} +3 -3
  41. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  42. package/dist/{glide-data-editor-XZB3u4tH.js → glide-data-editor--N1kxdkm.js} +370 -368
  43. package/dist/{glide-utils-DJyqh0EM.js → glide-utils-Bt8_pBa2.js} +16 -16
  44. package/dist/{globals-BIQc0DK0.js → globals-CNvCn9Ut.js} +3 -3
  45. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  46. package/dist/{html-to-image-Bf8QMGi4.js → html-to-image-Bg5ZoSO-.js} +29 -29
  47. package/dist/{infoDiagram-27XIBGKW-DipQoHxl.js → infoDiagram-27XIBGKW-B1hmUggC.js} +1 -1
  48. package/dist/{ishikawaDiagram-5VMMS53U-DgFMjSGz.js → ishikawaDiagram-5VMMS53U-DUeT3t-d.js} +1 -1
  49. package/dist/{kanban-definition-UXKFOSKX-w1ACVjgl.js → kanban-definition-UXKFOSKX-DEUnNNgc.js} +1 -1
  50. package/dist/{label-BDKF3Afc.js → label-BRnUC0NX.js} +2 -2
  51. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  52. package/dist/{loader-BQOfjTIJ.js → loader-Bm8cy3e8.js} +12 -12
  53. package/dist/main.js +3272 -3030
  54. package/dist/{mermaid-CHo49Yv-.js → mermaid-B0fXpsNP.js} +38 -38
  55. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  56. package/dist/{mindmap-definition-YA3MSWOX-CrT0KSfX.js → mindmap-definition-YA3MSWOX-CPexlhDQ.js} +1 -1
  57. package/dist/{number-overlay-editor-CEO_7tD7.js → number-overlay-editor-COjoFXgy.js} +3 -3
  58. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  59. package/dist/{once-BtE-wKRL.js → once-eG8Vmvjd.js} +1 -1
  60. package/dist/{pegDiagram-XKGWAZYB-BqvywNJA.js → pegDiagram-XKGWAZYB-CMpp1OOQ.js} +2 -2
  61. package/dist/{pieDiagram-E7YTZNPT-BfxJ_XCc.js → pieDiagram-E7YTZNPT-DlMWg5t1.js} +1 -1
  62. package/dist/precisionRound-DhAzeRcm.js +49 -0
  63. package/dist/{process-output-BZg56CCb.js → process-output-BtVPCGeI.js} +4 -4
  64. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  65. package/dist/{railroadDiagram-O6MQD6OU-8vFLVTPu.js → railroadDiagram-O6MQD6OU-DBJfseeO.js} +2 -2
  66. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  67. package/dist/{requirementDiagram-BXWQKSXE-a_MxGvxe.js → requirementDiagram-BXWQKSXE-BC7V639r.js} +1 -1
  68. package/dist/{reveal-component-LPCUNcgp.js → reveal-component-BKaSM0Qr.js} +23 -23
  69. package/dist/{select-BButFi-d.js → select-B0LRDH0g.js} +4 -4
  70. package/dist/{sequenceDiagram-WJ2MYXX4-CXFMnxGh.js → sequenceDiagram-WJ2MYXX4-CgesU6F-.js} +1 -1
  71. package/dist/{spec-BwarBpUJ.js → spec-DTiOPKyk.js} +41 -11
  72. package/dist/{stateDiagram-D77RDMKH-DQeA_CwI.js → stateDiagram-D77RDMKH-lH_BUhs3.js} +2 -2
  73. package/dist/{stateDiagram-v2-MP3YSRHH-BxxQucm1.js → stateDiagram-v2-MP3YSRHH-8LL6FrBe.js} +2 -2
  74. package/dist/style.css +1 -1
  75. package/dist/{swimlanes-42K2YHIH-Do6L-cYg.js → swimlanes-42K2YHIH-Cuf9B9kE.js} +1 -1
  76. package/dist/{swimlanesDiagram-VR7AAH4N-DUzPvSVU.js → swimlanesDiagram-VR7AAH4N-BoMMtOPh.js} +2 -2
  77. package/dist/{swiper-component-DKntyn8p.js → swiper-component-DfhRvAFi.js} +2 -2
  78. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  79. package/dist/{timeline-definition-24CTP7MA-D3c46WM5.js → timeline-definition-24CTP7MA-yzzw4TF4.js} +1 -1
  80. package/dist/{toDate-UtiwmnH2.js → toDate-DKH0_kMo.js} +15 -18
  81. package/dist/{toast-B9UMEi8u.js → toast-CFWpihJU.js} +4 -4
  82. package/dist/{tooltip-BO1REbqs.js → tooltip-CKO1pHJo.js} +2 -2
  83. package/dist/{use-restore-focus-BEM7sFWl.js → use-restore-focus-DBCITLeF.js} +1 -1
  84. package/dist/{useAsyncData-Deh5kszT.js → useAsyncData-D0mLJstx.js} +15 -14
  85. package/dist/{useDateFormatter-D8XruNGG.js → useDateFormatter-c_yJtkAe.js} +2 -2
  86. package/dist/{useDeepCompareMemoize-DXwzRINI.js → useDeepCompareMemoize-BpPWi3G3.js} +1 -1
  87. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-BkdeVdXC.js} +1 -1
  88. package/dist/{useLifecycle-j7D9uD1o.js → useLifecycle-Dum8WSXq.js} +3 -3
  89. package/dist/{useTheme-CRnjkku0.js → useTheme-CfeaVw5D.js} +6 -6
  90. package/dist/{vega-component-BeF9dfyO.js → vega-component-DgytCKrL.js} +147 -147
  91. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  92. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  93. package/dist/{vennDiagram-4TSXK5OY-BSZD8JcZ.js → vennDiagram-4TSXK5OY-eJOrTqqC.js} +1 -1
  94. package/dist/{wardleyDiagram-VM6X3IG4-48cUWkfA.js → wardleyDiagram-VM6X3IG4-TILPvee6.js} +1 -1
  95. package/dist/{xychartDiagram-S5SC5T6Z-CswYgYeu.js → xychartDiagram-S5SC5T6Z-DUPvRwwJ.js} +2 -2
  96. package/package.json +12 -12
  97. package/src/__tests__/host-imports.test.ts +1 -0
  98. package/src/__tests__/mount.test.ts +0 -20
  99. package/src/components/app-config/ai-config.tsx +7 -2
  100. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  101. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  102. package/src/components/chat/acp/blocks.tsx +2 -2
  103. package/src/components/chat/chat-components.tsx +2 -0
  104. package/src/components/chat/chat-panel.tsx +9 -8
  105. package/src/components/chat/chat-utils.ts +4 -8
  106. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  107. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  108. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  109. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  110. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  111. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  112. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  113. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  114. package/src/components/data-table/cell-selection/feature.ts +2 -2
  115. package/src/components/data-table/cell-selection/types.ts +2 -2
  116. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  117. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  118. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  119. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  120. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  121. package/src/components/data-table/charts/charts.tsx +40 -30
  122. package/src/components/data-table/charts/storage.ts +16 -5
  123. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  124. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  125. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  126. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  127. package/src/components/data-table/columns.tsx +45 -46
  128. package/src/components/data-table/context-menu.tsx +36 -20
  129. package/src/components/data-table/data-table.tsx +7 -1
  130. package/src/components/data-table/date-popover.tsx +6 -4
  131. package/src/components/data-table/export-actions.tsx +76 -48
  132. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +2 -2
  133. package/src/components/data-table/search-highlight.tsx +79 -0
  134. package/src/components/data-table/types.ts +1 -0
  135. package/src/components/data-table/url-detector.tsx +49 -9
  136. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  137. package/src/components/datasources/__tests__/column-preview.test.tsx +63 -2
  138. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  139. package/src/components/datasources/column-preview.tsx +8 -2
  140. package/src/components/datasources/datasources.tsx +47 -35
  141. package/src/components/debugger/debugger-code.tsx +5 -0
  142. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  143. package/src/components/dependency-graph/elements.ts +24 -21
  144. package/src/components/dependency-graph/panels.tsx +2 -1
  145. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  146. package/src/components/editor/SortableCell.tsx +1 -3
  147. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  148. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +24 -0
  149. package/src/components/editor/actions/pair-with-agent-modal.tsx +18 -15
  150. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  151. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  152. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  153. package/src/components/editor/cell/CreateCellButton.tsx +8 -10
  154. package/src/components/editor/cell/DeleteButton.tsx +1 -0
  155. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  156. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  157. package/src/components/editor/cell/code/cell-editor.tsx +1 -1
  158. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  159. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  160. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +30 -1
  161. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  162. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  163. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +12 -18
  164. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +58 -1
  165. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  166. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  167. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  168. package/src/components/editor/connections/components.tsx +1 -1
  169. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  170. package/src/components/editor/controls/command-palette.tsx +22 -10
  171. package/src/components/editor/dynamic-favicon.tsx +21 -40
  172. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +12 -23
  173. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  174. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  175. package/src/components/editor/file-tree/file-explorer.tsx +10 -5
  176. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  177. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  178. package/src/components/editor/notebook-banner.tsx +1 -0
  179. package/src/components/editor/output/MarimoTracebackOutput.tsx +2 -2
  180. package/src/components/editor/output/ansi-reduce.ts +33 -20
  181. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  182. package/src/components/editor/package-alert.tsx +6 -4
  183. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  184. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  185. package/src/components/home/components.tsx +17 -31
  186. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  187. package/src/components/icons/large-spinner.tsx +3 -14
  188. package/src/components/icons/spinner.tsx +1 -1
  189. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  190. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  191. package/src/components/pages/home-page.tsx +164 -91
  192. package/src/components/scratchpad/scratchpad.tsx +2 -2
  193. package/src/components/slides/reveal-component.tsx +4 -2
  194. package/src/components/slides/slide-cell-view.tsx +1 -1
  195. package/src/components/slides/slide-form.tsx +1 -0
  196. package/src/components/tracing/tracing.tsx +2 -2
  197. package/src/components/ui/alert-dialog.tsx +1 -1
  198. package/src/core/ai/context/providers/datasource.ts +6 -2
  199. package/src/core/ai/tools/base.ts +16 -7
  200. package/src/core/ai/tools/edit-notebook-tool.ts +47 -45
  201. package/src/core/ai/tools/registry.ts +13 -8
  202. package/src/core/ai/tools/sample-tool.ts +6 -7
  203. package/src/core/attachments/state.ts +5 -8
  204. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -0
  205. package/src/core/cells/__tests__/cell.test.ts +39 -0
  206. package/src/core/cells/__tests__/cells.test.ts +3 -1
  207. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  208. package/src/core/cells/__tests__/notebook-outline.test.ts +113 -0
  209. package/src/core/cells/__tests__/session.test.ts +117 -36
  210. package/src/core/cells/cell.ts +14 -3
  211. package/src/core/cells/cells.ts +3 -1
  212. package/src/core/cells/pending-delete-service.ts +2 -6
  213. package/src/core/cells/session.ts +1 -1
  214. package/src/core/cells/types.ts +10 -1
  215. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  216. package/src/core/codemirror/cells/__tests__/extensions.test.ts +15 -0
  217. package/src/core/codemirror/cells/extensions.ts +7 -3
  218. package/src/core/codemirror/keymaps/vim.ts +2 -6
  219. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  220. package/src/core/codemirror/language/extension.ts +8 -8
  221. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  222. package/src/core/codemirror/language/languages/python.ts +16 -10
  223. package/src/core/codemirror/language/languages/sql/sql.ts +9 -7
  224. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  225. package/src/core/codemirror/language/types.ts +7 -7
  226. package/src/core/codemirror/rtc/extension.ts +24 -12
  227. package/src/core/codemirror/rtc/loro/awareness.ts +34 -19
  228. package/src/core/codemirror/rtc/loro/sync.ts +1 -0
  229. package/src/core/codemirror/utils.ts +1 -0
  230. package/src/core/datasets/__tests__/data-source-discovery.test.ts +4 -4
  231. package/src/core/datasets/data-source-discovery.ts +4 -6
  232. package/src/core/document/__tests__/codemirror.test.ts +483 -0
  233. package/src/core/document/__tests__/draft.test.ts +322 -0
  234. package/src/core/document/__tests__/fixtures.ts +92 -0
  235. package/src/core/document/__tests__/hooks.test.tsx +386 -0
  236. package/src/core/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  237. package/src/core/document/__tests__/model.test.ts +158 -0
  238. package/src/core/document/__tests__/protocol-fixtures.ts +38 -0
  239. package/src/core/document/__tests__/recovery-scenarios.test.ts +230 -0
  240. package/src/core/document/__tests__/scenario-fixture.ts +42 -0
  241. package/src/core/document/__tests__/scenarios.test.ts +205 -0
  242. package/src/core/document/__tests__/state.test.ts +438 -0
  243. package/src/core/document/__tests__/submission.test.ts +203 -0
  244. package/src/core/document/__tests__/test-host.test.ts +92 -0
  245. package/src/core/document/__tests__/test-host.ts +247 -0
  246. package/src/core/document/__tests__/test-notebook.ts +139 -0
  247. package/src/core/document/__tests__/user.ts +61 -0
  248. package/src/core/document/codemirror.ts +311 -0
  249. package/src/core/document/draft.ts +472 -0
  250. package/src/core/document/hooks.tsx +118 -0
  251. package/src/core/document/index.ts +22 -0
  252. package/src/core/document/model.ts +52 -0
  253. package/src/core/document/select.ts +41 -0
  254. package/src/core/document/state.ts +374 -0
  255. package/src/core/document/sync.ts +363 -0
  256. package/src/core/host/AGENTS.md +12 -2
  257. package/src/core/host/__tests__/attachments.test.ts +56 -0
  258. package/src/core/host/__tests__/document.test.ts +126 -0
  259. package/src/core/host/__tests__/fixtures.ts +41 -0
  260. package/src/core/host/__tests__/notebook-client.test.ts +511 -132
  261. package/src/core/host/__tests__/static.test.ts +167 -0
  262. package/src/core/host/__tests__/wasm.test.ts +152 -0
  263. package/src/core/host/backend.ts +10 -0
  264. package/src/core/host/connection.ts +1 -40
  265. package/src/core/host/notebook-client.ts +204 -133
  266. package/src/core/host/resources/attachments.ts +22 -0
  267. package/src/core/host/resources/document.ts +101 -0
  268. package/src/core/host/static.ts +70 -0
  269. package/src/core/host/wasm.ts +92 -0
  270. package/src/core/hotkeys/actions.ts +2 -4
  271. package/src/core/islands/__tests__/bootstrap.test.ts +0 -26
  272. package/src/core/islands/bootstrap.ts +0 -1
  273. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  274. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  275. package/src/core/network/connection.ts +4 -2
  276. package/src/core/network/useConnectionNotice.ts +8 -2
  277. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  278. package/src/core/runtime/__tests__/runtime.test.ts +15 -58
  279. package/src/core/runtime/runtime.ts +25 -42
  280. package/src/core/runtime/types.ts +0 -2
  281. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  282. package/src/core/state/jotai.ts +1 -1
  283. package/src/core/static/__tests__/files.test.ts +38 -10
  284. package/src/core/static/files.ts +2 -0
  285. package/src/core/wasm/bridge.ts +38 -10
  286. package/src/core/wasm/store.ts +3 -3
  287. package/src/core/wasm/worker/bootstrap.ts +2 -1
  288. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  289. package/src/core/wasm/worker/types.ts +11 -0
  290. package/src/core/wasm/worker/worker.ts +48 -0
  291. package/src/core/websocket/__tests__/connection-state.test.ts +0 -12
  292. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +153 -70
  293. package/src/core/websocket/__tests__/useNotebookSession.hook.test.tsx +151 -428
  294. package/src/core/websocket/connection-state.ts +0 -10
  295. package/src/core/websocket/state.ts +7 -0
  296. package/src/core/websocket/transports/__tests__/notebook.test.ts +502 -0
  297. package/src/core/websocket/transports/notebook.ts +213 -0
  298. package/src/core/websocket/useNotebookClient.ts +38 -29
  299. package/src/core/websocket/useNotebookSession.tsx +1 -14
  300. package/src/css/globals.css +5 -0
  301. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  302. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  303. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  304. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  305. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  306. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  307. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  308. package/src/hooks/useAsyncData.ts +3 -3
  309. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  310. package/src/hooks/useDebounce.ts +6 -10
  311. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  312. package/src/hooks/useDelayElapsed.ts +14 -10
  313. package/src/hooks/useElapsedTime.ts +3 -3
  314. package/src/hooks/useEventListener.ts +1 -2
  315. package/src/hooks/useInputHistory.ts +4 -7
  316. package/src/hooks/useInternalStateWithSync.ts +6 -6
  317. package/src/hooks/useResizeObserver.ts +4 -7
  318. package/src/hooks/useScript.ts +67 -70
  319. package/src/hooks/useStateForKey.ts +93 -0
  320. package/src/mount.tsx +0 -4
  321. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  322. package/src/plugins/core/registerReactComponent.tsx +1 -1
  323. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  324. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  325. package/src/plugins/impl/DataTablePlugin.tsx +29 -33
  326. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  327. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  328. package/src/plugins/impl/FormPlugin.tsx +1 -1
  329. package/src/plugins/impl/RangeSliderPlugin.tsx +0 -1
  330. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  331. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  332. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +117 -1
  333. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  334. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  335. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +2 -0
  336. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  337. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  338. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  339. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  340. package/src/plugins/impl/chat/chat-ui.tsx +12 -10
  341. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  342. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  343. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  344. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  345. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  346. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  347. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  348. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  349. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  350. package/src/plugins/impl/panel/utils.ts +17 -0
  351. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  352. package/src/plugins/impl/plotly/selection.ts +25 -19
  353. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  354. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  355. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  356. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  357. package/src/stories/button.stories.tsx +1 -0
  358. package/src/stories/editor.stories.tsx +24 -17
  359. package/src/stories/form-wrapper.stories.tsx +2 -2
  360. package/src/stories/select.stories.tsx +2 -2
  361. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  362. package/src/utils/__tests__/arrays.test.ts +35 -0
  363. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  364. package/src/utils/__tests__/numbers.test.ts +48 -1
  365. package/src/utils/__tests__/objects.test.ts +21 -0
  366. package/src/utils/arrays.ts +10 -0
  367. package/src/utils/blob.ts +1 -1
  368. package/src/utils/numbers.ts +28 -17
  369. package/src/utils/objects.ts +3 -7
  370. package/src/utils/once.ts +19 -19
  371. package/src/utils/tracer.ts +9 -5
  372. package/src/utils/zod-utils.ts +1 -3
  373. package/dist/any-language-editor-DZ_YTVus.js +0 -76
  374. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  375. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  376. package/dist/precisionRound-DMIVQrt2.js +0 -97
  377. package/src/core/host/__tests__/events.test.ts +0 -46
  378. package/src/core/host/__tests__/protocol.test.ts +0 -214
  379. package/src/core/host/events.ts +0 -60
  380. package/src/core/host/protocol.ts +0 -1943
  381. package/src/core/state/__tests__/jotai.test.ts +0 -25
  382. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  383. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  384. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  385. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  386. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  387. package/src/core/websocket/transports/basic.ts +0 -102
  388. package/src/core/websocket/transports/create.ts +0 -31
  389. package/src/core/websocket/transports/sse.ts +0 -308
  390. package/src/core/websocket/transports/transport.ts +0 -50
  391. package/src/core/websocket/transports/ws.ts +0 -98
@@ -8,27 +8,31 @@ import {
8
8
  screen,
9
9
  waitFor,
10
10
  } from "@testing-library/react";
11
- import { createStore, Provider } from "jotai";
11
+ import { atom, createStore, Provider } from "jotai";
12
12
  import { userEvent } from "storybook/test";
13
13
  import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
14
14
  import { PAIR_CONFIG } from "@/__tests__/fixtures/pair-config";
15
15
  import { TooltipProvider } from "@/components/ui/tooltip";
16
16
  import { toast } from "@/components/ui/use-toast";
17
- import { type Attachment, attachmentsAtom } from "@/core/attachments/state";
17
+ import type { Attachment } from "@/core/attachments/state";
18
18
  import { pairConfigAtom } from "@/core/config/pair";
19
+ import { createNotebookClient } from "@/core/host/notebook-client";
19
20
  import {
20
21
  type StableSessionId,
21
22
  stableSessionIdAtom,
22
23
  } from "@/core/kernel/session";
23
24
  import { API } from "@/core/network/api";
24
25
  import { connectionAtom } from "@/core/network/connection";
25
- import { WebSocketClosedReason, ConnectionState } from "@/core/websocket/types";
26
+ import { notebookClientAtom } from "@/core/websocket/state";
27
+ import { ConnectionState } from "@/core/websocket/types";
26
28
  import { Deferred } from "@/utils/Deferred";
27
29
  import { AttachmentStatus } from "../footer-items/attachment-status";
28
30
 
29
31
  vi.mock("@/components/ui/use-toast", () => ({ toast: vi.fn() }));
30
32
  vi.mock("@/core/network/api", () => ({ API: { post: vi.fn() } }));
31
33
 
34
+ const attachmentsAtom = atom<Attachment[]>([]);
35
+
32
36
  const AGENT: Attachment = {
33
37
  id: "a1",
34
38
  kind: "agent",
@@ -41,6 +45,11 @@ function renderStatus(attachments: Attachment[] = [AGENT], preview = true) {
41
45
  const store = createStore();
42
46
  store.set(connectionAtom, { state: ConnectionState.OPEN });
43
47
  store.set(pairConfigAtom, { ...PAIR_CONFIG, hostEnabled: preview });
48
+ store.set(notebookClientAtom, {
49
+ ...createNotebookClient(store).client,
50
+ connectionAtom: atom({ state: ConnectionState.OPEN }),
51
+ attachmentsAtom,
52
+ });
44
53
  store.set(attachmentsAtom, attachments);
45
54
  store.set(stableSessionIdAtom, "stable/test" as StableSessionId);
46
55
  return {
@@ -342,21 +351,6 @@ describe("AttachmentStatus", () => {
342
351
  ).toBeDisabled();
343
352
  });
344
353
 
345
- it.each([
346
- { state: ConnectionState.CONNECTING },
347
- { state: ConnectionState.CLOSING },
348
- { state: ConnectionState.NOT_STARTED },
349
- {
350
- state: ConnectionState.CLOSED,
351
- code: WebSocketClosedReason.KERNEL_DISCONNECTED,
352
- reason: "offline",
353
- },
354
- ])("hides connected claims while the browser is %j", (connection) => {
355
- const { store, container } = renderStatus();
356
- act(() => store.set(connectionAtom, connection));
357
- expect(container).toBeEmptyDOMElement();
358
- });
359
-
360
354
  it("removes detached agents when the list is replaced", () => {
361
355
  const { store, container } = renderStatus([
362
356
  AGENT,
@@ -1,11 +1,12 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
- import { act, render, screen, within } from "@testing-library/react";
3
+ import { act, fireEvent, render, screen, within } from "@testing-library/react";
4
4
  import { createStore, Provider } from "jotai";
5
5
  import { describe, expect, it, vi } from "vitest";
6
6
  import { cellId } from "@/__tests__/branded";
7
7
  import { TooltipProvider } from "@/components/ui/tooltip";
8
8
  import { stagedAICellsAtom, visibleForTesting } from "@/core/ai/staged-cells";
9
+ import { scrollAndHighlightCell } from "../../../links/cell-link";
9
10
  import { PendingAICells } from "../pending-ai-cells";
10
11
 
11
12
  vi.mock("@/components/editor/cell/useDeleteCell", () => ({
@@ -29,6 +30,62 @@ vi.mock("@/components/editor/links/cell-link", () => ({
29
30
  }));
30
31
 
31
32
  describe("PendingAICells", () => {
33
+ it("navigates between pending cells through named controls", () => {
34
+ const store = createStore();
35
+ const first = cellId("first-pending-cell");
36
+ const second = cellId("second-pending-cell");
37
+ const third = cellId("third-pending-cell");
38
+ store.set(
39
+ stagedAICellsAtom,
40
+ new Map([
41
+ [first, { type: "add_cell" }],
42
+ [second, { type: "add_cell" }],
43
+ [third, { type: "add_cell" }],
44
+ ]),
45
+ );
46
+
47
+ render(
48
+ <Provider store={store}>
49
+ <TooltipProvider>
50
+ <PendingAICells />
51
+ </TooltipProvider>
52
+ </Provider>,
53
+ );
54
+
55
+ const next = screen.getByRole("button", { name: "Next pending AI cell" });
56
+ const previous = screen.getByRole("button", {
57
+ name: "Previous pending AI cell",
58
+ });
59
+
60
+ fireEvent.click(next);
61
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
62
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
63
+
64
+ fireEvent.click(next);
65
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(second, "focus");
66
+ expect(screen.getByText("2 / 3")).toBeInTheDocument();
67
+
68
+ fireEvent.click(previous);
69
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
70
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
71
+
72
+ fireEvent.click(next);
73
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(second, "focus");
74
+ expect(screen.getByText("2 / 3")).toBeInTheDocument();
75
+
76
+ fireEvent.click(next);
77
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(third, "focus");
78
+ expect(screen.getByText("3 / 3")).toBeInTheDocument();
79
+
80
+ fireEvent.click(next);
81
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
82
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
83
+
84
+ fireEvent.click(previous);
85
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(third, "focus");
86
+ expect(screen.getByText("3 / 3")).toBeInTheDocument();
87
+ });
88
+
32
89
  it("disables bulk actions while cells are still generating", () => {
33
90
  const store = createStore();
34
91
  const generationId = Symbol("staged-cell-generation");
@@ -26,13 +26,7 @@ import { isWasm } from "@/core/wasm/utils";
26
26
  import { cn } from "@/utils/cn";
27
27
  import { FooterItem } from "../footer-item";
28
28
 
29
- interface RuntimeSettingsProps {
30
- className?: string;
31
- }
32
-
33
- export const RuntimeSettings: React.FC<RuntimeSettingsProps> = ({
34
- className,
35
- }) => {
29
+ export const RuntimeSettings: React.FC = () => {
36
30
  const { saveUserConfig } = useRequestClient();
37
31
  const [config, setUserConfig] = useResolvedMarimoConfig();
38
32
 
@@ -79,7 +79,12 @@ export const PendingAICells: React.FC = () => {
79
79
  <SparklesIcon className="h-4 w-4 text-primary" />
80
80
 
81
81
  <div className="flex items-center">
82
- <Button variant="ghost" size="icon" onClick={() => clickNext("up")}>
82
+ <Button
83
+ aria-label="Previous pending AI cell"
84
+ variant="ghost"
85
+ size="icon"
86
+ onClick={() => clickNext("down")}
87
+ >
83
88
  <ChevronUp className="h-3.5 w-3.5" />
84
89
  </Button>
85
90
  <span className="text-xs font-mono min-w-14 text-center">
@@ -87,7 +92,12 @@ export const PendingAICells: React.FC = () => {
87
92
  ? `${listStagedCells.length} pending`
88
93
  : `${currentIndex + 1} / ${listStagedCells.length}`}
89
94
  </span>
90
- <Button variant="ghost" size="icon" onClick={() => clickNext("down")}>
95
+ <Button
96
+ aria-label="Next pending AI cell"
97
+ variant="ghost"
98
+ size="icon"
99
+ onClick={() => clickNext("up")}
100
+ >
91
101
  <ChevronDown className="h-3.5 w-3.5" />
92
102
  </Button>
93
103
  </div>
@@ -15,7 +15,7 @@ let storedValue: string | null = null;
15
15
  */
16
16
  export function createStorage(location: "left" | "bottom"): PanelGroupStorage {
17
17
  return {
18
- getItem(name) {
18
+ getItem(_name) {
19
19
  if (!storedValue) {
20
20
  return storedValue ?? null;
21
21
  }
@@ -48,7 +48,7 @@ export const SelectorButton: React.FC<{
48
48
  }> = ({ name, color, icon, onSelect }) => (
49
49
  <button
50
50
  type="button"
51
- className="py-3 flex flex-col items-center justify-center gap-1 transition-all hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid"
51
+ className="py-3 flex flex-col items-center justify-center gap-1 transition-[scale,filter,box-shadow] hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid"
52
52
  style={{ backgroundColor: color }}
53
53
  onClick={onSelect}
54
54
  >
@@ -0,0 +1,53 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { fireEvent, render, screen, waitFor } from "@testing-library/react";
4
+ import { createStore, Provider } from "jotai";
5
+ import { expect, it, vi } from "vitest";
6
+ import { useNotebookActions } from "../../actions/useNotebookActions";
7
+ import CommandPalette from "../command-palette";
8
+
9
+ vi.mock("../../actions/useNotebookActions", () => ({
10
+ useNotebookActions: vi.fn(),
11
+ }));
12
+ vi.mock("../../actions/useCellActionButton", () => ({
13
+ useCellActionButtons: () => [],
14
+ }));
15
+ vi.mock("../../actions/useConfigActions", () => ({
16
+ useConfigActions: () => [],
17
+ }));
18
+
19
+ it("builds current actions only while open and keeps the opening shortcut active", async () => {
20
+ HTMLElement.prototype.scrollIntoView = vi.fn();
21
+ const firstAction = vi.fn();
22
+ const nextAction = vi.fn();
23
+ vi.mocked(useNotebookActions).mockReturnValue([
24
+ { label: "First action", handle: firstAction },
25
+ ]);
26
+ render(
27
+ <Provider store={createStore()}>
28
+ <CommandPalette />
29
+ </Provider>,
30
+ );
31
+ expect(useNotebookActions).not.toHaveBeenCalled();
32
+
33
+ fireEvent.keyDown(document, { key: "k", ctrlKey: true });
34
+ await screen.findByPlaceholderText("Type to search...");
35
+ fireEvent.click(screen.getByText("First action"));
36
+ await waitFor(() => expect(firstAction).toHaveBeenCalledOnce());
37
+ await waitFor(() =>
38
+ expect(screen.queryByPlaceholderText("Type to search...")).toBeNull(),
39
+ );
40
+
41
+ vi.mocked(useNotebookActions)
42
+ .mockClear()
43
+ .mockReturnValue([{ label: "Updated action", handle: nextAction }]);
44
+ fireEvent.keyDown(document, { key: "x" });
45
+ expect(useNotebookActions).not.toHaveBeenCalled();
46
+
47
+ fireEvent.keyDown(document, { key: "k", ctrlKey: true });
48
+ await screen.findByPlaceholderText("Type to search...");
49
+ expect(screen.queryByText("First action")).toBeNull();
50
+ fireEvent.click(screen.getByText("Updated action"));
51
+ await waitFor(() => expect(nextAction).toHaveBeenCalledOnce());
52
+ expect(firstAction).toHaveBeenCalledOnce();
53
+ });
@@ -1,6 +1,6 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
- import { useAtom, useAtomValue } from "jotai";
3
+ import { useAtom, useAtomValue, useSetAtom } from "jotai";
4
4
  import {
5
5
  CommandDialog,
6
6
  CommandEmpty,
@@ -32,6 +32,25 @@ import { commandPaletteAtom } from "./state";
32
32
 
33
33
  const CommandPalette = () => {
34
34
  const [open, setOpen] = useAtom(commandPaletteAtom);
35
+ const hotkeys = useAtomValue(hotkeysAtom);
36
+
37
+ useEventListener(document, "keydown", (e) => {
38
+ if (parseShortcut(hotkeys.getHotkey("global.commandPalette").key)(e)) {
39
+ e.preventDefault();
40
+ setOpen((open) => !open);
41
+ }
42
+ });
43
+
44
+ return (
45
+ <CommandDialog open={open} onOpenChange={setOpen}>
46
+ <CommandPaletteContent />
47
+ </CommandDialog>
48
+ );
49
+ };
50
+
51
+ // The dialog mounts these subscriptions only while its content is present.
52
+ const CommandPaletteContent = () => {
53
+ const setOpen = useSetAtom(commandPaletteAtom);
35
54
  const registeredActions = useRegisteredActions();
36
55
  const lastFocusedCell = useAtomValue(lastFocusedCellAtom);
37
56
  const hotkeys = useAtomValue(hotkeysAtom);
@@ -55,13 +74,6 @@ const CommandPalette = () => {
55
74
  const { recentCommands, addRecentCommand } = useRecentCommands();
56
75
  const recentCommandsSet = new Set(recentCommands);
57
76
 
58
- useEventListener(document, "keydown", (e) => {
59
- if (parseShortcut(hotkeys.getHotkey("global.commandPalette").key)(e)) {
60
- e.preventDefault();
61
- setOpen((open) => !open);
62
- }
63
- });
64
-
65
77
  const renderShortcutCommandItem = (
66
78
  shortcut: HotkeyAction,
67
79
  props: {
@@ -142,7 +154,7 @@ const CommandPalette = () => {
142
154
  };
143
155
 
144
156
  return (
145
- <CommandDialog open={open} onOpenChange={setOpen}>
157
+ <>
146
158
  <CommandInput placeholder="Type to search..." />
147
159
  <CommandList>
148
160
  <CommandEmpty>No results found.</CommandEmpty>
@@ -211,7 +223,7 @@ const CommandPalette = () => {
211
223
  })}
212
224
  </CommandGroup>
213
225
  </CommandList>
214
- </CommandDialog>
226
+ </>
215
227
  );
216
228
  };
217
229
 
@@ -1,8 +1,9 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
3
  import { usePrevious } from "@dnd-kit/utilities";
4
+ import { useAtomValue } from "jotai";
4
5
  import { useEffect } from "react";
5
- import { useCellErrors } from "@/core/cells/cells";
6
+ import { cellErrorCount } from "@/core/cells/cells";
6
7
  import { useEventListener } from "@/hooks/useEventListener";
7
8
  import successFaviconUrl from "../../assets/circle-check.ico";
8
9
  import runningFaviconUrl from "../../assets/circle-play.ico";
@@ -15,13 +16,6 @@ const FAVICON_PATHS = {
15
16
  error: errorFaviconUrl,
16
17
  } as const;
17
18
 
18
- // Cache favicon object URLs lazily
19
- type FaviconKey = keyof typeof FAVICON_PATHS;
20
-
21
- async function getFaviconUrl(key: FaviconKey): Promise<string> {
22
- return FAVICON_PATHS[key];
23
- }
24
-
25
19
  function getOrCreateFavicon(): HTMLLinkElement {
26
20
  const existingFavicon =
27
21
  document.querySelector<HTMLLinkElement>("link[rel~='icon']");
@@ -44,16 +38,16 @@ function maybeSendNotification(numErrors: number) {
44
38
  return;
45
39
  }
46
40
 
47
- const sendNotification = async () => {
41
+ const sendNotification = () => {
48
42
  if (numErrors === 0) {
49
43
  new Notification("Execution completed", {
50
44
  body: "Your notebook run completed successfully.",
51
- icon: await getFaviconUrl("success"),
45
+ icon: FAVICON_PATHS.success,
52
46
  });
53
47
  } else {
54
48
  new Notification("Execution failed", {
55
49
  body: `Your notebook run encountered ${numErrors} error(s).`,
56
- icon: await getFaviconUrl("error"),
50
+ icon: FAVICON_PATHS.error,
57
51
  });
58
52
  }
59
53
  };
@@ -75,7 +69,7 @@ function maybeSendNotification(numErrors: number) {
75
69
 
76
70
  export const DynamicFavicon = (props: Props) => {
77
71
  const { isRunning } = props;
78
- const errors = useCellErrors();
72
+ const numErrors = useAtomValue(cellErrorCount);
79
73
 
80
74
  useEffect(() => {
81
75
  const favicon = getOrCreateFavicon();
@@ -86,46 +80,33 @@ export const DynamicFavicon = (props: Props) => {
86
80
  return;
87
81
  }
88
82
 
89
- const updateFavicon = async () => {
90
- let key: FaviconKey;
91
- // When notebook is running, display running favicon
92
- if (isRunning) {
93
- key = "running";
94
- } else {
95
- // When run is complete, display success or error favicon
96
- key = errors.length === 0 ? "success" : "error";
97
- }
98
- favicon.href = await getFaviconUrl(key);
83
+ const key = isRunning ? "running" : numErrors === 0 ? "success" : "error";
84
+ favicon.href = FAVICON_PATHS[key];
99
85
 
100
- // If notebook is in focus, reset favicon after 3 seconds
101
- // If not in focus, the focus event listener handles it
102
- if (!document.hasFocus()) {
103
- return;
104
- }
105
-
106
- const timeoutId = setTimeout(async () => {
107
- favicon.href = await getFaviconUrl("idle");
108
- }, 3000);
109
-
110
- return () => clearTimeout(timeoutId);
111
- };
86
+ if (!document.hasFocus()) {
87
+ // The focus listener resets the favicon when the user returns.
88
+ return;
89
+ }
112
90
 
113
- updateFavicon();
114
- }, [isRunning, errors]);
91
+ const timeoutId = setTimeout(() => {
92
+ favicon.href = FAVICON_PATHS.idle;
93
+ }, 3000);
94
+ return () => clearTimeout(timeoutId);
95
+ }, [isRunning, numErrors]);
115
96
 
116
97
  // Send user notification when run has completed
117
98
  const prevRunning = usePrevious(isRunning) ?? isRunning;
118
99
  useEffect(() => {
119
100
  if (prevRunning && !isRunning) {
120
- maybeSendNotification(errors.length);
101
+ maybeSendNotification(numErrors);
121
102
  }
122
- }, [errors, prevRunning, isRunning]);
103
+ }, [numErrors, prevRunning, isRunning]);
123
104
 
124
105
  // When notebook comes back in focus, reset favicon
125
- useEventListener(window, "focus", async (_) => {
106
+ useEventListener(window, "focus", () => {
126
107
  if (!isRunning) {
127
108
  const favicon = getOrCreateFavicon();
128
- favicon.href = await getFaviconUrl("idle");
109
+ favicon.href = FAVICON_PATHS.idle;
129
110
  }
130
111
  });
131
112
 
@@ -8,7 +8,7 @@ import {
8
8
  screen,
9
9
  waitFor,
10
10
  } from "@testing-library/react";
11
- import { createStore, Provider, useAtomValue } from "jotai";
11
+ import { atom, createStore, Provider, useAtomValue } from "jotai";
12
12
  import type React from "react";
13
13
  import { beforeEach, describe, expect, it, vi } from "vitest";
14
14
  import { MockNotebook } from "@/__mocks__/notebook";
@@ -16,7 +16,7 @@ import { cellId } from "@/__tests__/branded";
16
16
  import { PAIR_CONFIG } from "@/__tests__/fixtures/pair-config";
17
17
  import { TooltipProvider } from "@/components/ui/tooltip";
18
18
  import { toast } from "@/components/ui/use-toast";
19
- import { type Attachment, attachmentsAtom } from "@/core/attachments/state";
19
+ import type { Attachment } from "@/core/attachments/state";
20
20
  import { exportedForTesting, notebookAtom } from "@/core/cells/cells";
21
21
  import {
22
22
  type CellOutputMessage,
@@ -25,6 +25,7 @@ import {
25
25
  } from "@/core/cells/error-report";
26
26
  import { pairConfigAtom } from "@/core/config/pair";
27
27
  import { notebookStateFromSession } from "@/core/cells/session";
28
+ import { createNotebookClient } from "@/core/host/notebook-client";
28
29
  import type { MarimoError, OutputMessage } from "@/core/kernel/messages";
29
30
  import {
30
31
  type StableSessionId,
@@ -32,7 +33,8 @@ import {
32
33
  } from "@/core/kernel/session";
33
34
  import { API } from "@/core/network/api";
34
35
  import { connectionAtom } from "@/core/network/connection";
35
- import { WebSocketClosedReason, ConnectionState } from "@/core/websocket/types";
36
+ import { notebookClientAtom } from "@/core/websocket/state";
37
+ import { ConnectionState } from "@/core/websocket/types";
36
38
  import { Deferred } from "@/utils/Deferred";
37
39
  import { HTTPError } from "@/utils/errors";
38
40
  import { MarimoErrorOutput } from "../../output/MarimoErrorOutput";
@@ -48,6 +50,8 @@ vi.mock("@/components/editor/chrome/wrapper/useOpenAiAssistant", () => ({
48
50
 
49
51
  const CELL_ID = cellId("test-cell");
50
52
  const STABLE_ID = "stable/test" as StableSessionId;
53
+ const attachmentsAtom = atom<Attachment[]>([]);
54
+
51
55
  const AGENT: Attachment = {
52
56
  id: "a1",
53
57
  kind: "agent",
@@ -93,6 +97,11 @@ function renderButton({
93
97
  const store = createStore();
94
98
  store.set(connectionAtom, { state: ConnectionState.OPEN });
95
99
  store.set(pairConfigAtom, { ...PAIR_CONFIG, hostEnabled: preview });
100
+ store.set(notebookClientAtom, {
101
+ ...createNotebookClient(store).client,
102
+ connectionAtom: atom({ state: ConnectionState.OPEN }),
103
+ attachmentsAtom,
104
+ });
96
105
  store.set(attachmentsAtom, attachments);
97
106
  store.set(stableSessionIdAtom, stableId);
98
107
  const wrapper = ({ children }: { children: React.ReactNode }) => (
@@ -177,26 +186,6 @@ describe("SendErrorReportButton", () => {
177
186
  expect(screen.getByRole("button")).toBeDisabled();
178
187
  });
179
188
 
180
- it.each([
181
- { state: ConnectionState.CONNECTING },
182
- { state: ConnectionState.CLOSING },
183
- { state: ConnectionState.NOT_STARTED },
184
- {
185
- state: ConnectionState.CLOSED,
186
- code: WebSocketClosedReason.KERNEL_DISCONNECTED,
187
- reason: "offline",
188
- },
189
- ])(
190
- "does not send from a stale snapshot while the browser is %j",
191
- (connection) => {
192
- const { store } = renderButton();
193
- act(() => store.set(connectionAtom, connection));
194
- expect(screen.getByRole("button")).toBeDisabled();
195
- send();
196
- expect(API.post).not.toHaveBeenCalled();
197
- },
198
- );
199
-
200
189
  it.each([
201
190
  [new HTTPError(409, "Conflict"), "No agent attached"],
202
191
  [new HTTPError(500, "Server error"), "Could not send error"],
@@ -137,5 +137,11 @@ describe("FileExplorer upload destination", () => {
137
137
  }),
138
138
  ).toBeVisible();
139
139
  expect(screen.queryByText(".hidden")).not.toBeInTheDocument();
140
+
141
+ fireEvent.click(screen.getByTestId("file-explorer-hidden-files-button"));
142
+ expect(await screen.findByText(".hidden")).toBeVisible();
143
+ expect(
144
+ screen.getByRole("button", { name: "Upload files to workspace" }),
145
+ ).toBeVisible();
140
146
  });
141
147
  });
@@ -0,0 +1,91 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { act, renderHook, waitFor } from "@testing-library/react";
4
+ import { beforeEach, describe, expect, it, vi } from "vitest";
5
+ import type { EditRequests } from "@/core/network/types";
6
+ import { Deferred } from "@/utils/Deferred";
7
+ import { RequestingTree } from "../requesting-tree";
8
+ import { useFileSearch } from "../use-file-search";
9
+
10
+ const sendSearchFiles = vi.hoisted(() =>
11
+ vi.fn<EditRequests["sendSearchFiles"]>(),
12
+ );
13
+ vi.mock("@/core/network/requests", () => ({
14
+ useRequestClient: () => ({ sendSearchFiles }),
15
+ }));
16
+
17
+ describe("useFileSearch", () => {
18
+ let tree: RequestingTree;
19
+
20
+ beforeEach(async () => {
21
+ sendSearchFiles.mockReset();
22
+ tree = new RequestingTree({
23
+ getRoots: vi.fn().mockResolvedValue({
24
+ roots: [{ path: "/root", name: "workspace", isPrimary: true }],
25
+ }),
26
+ listFiles: vi.fn().mockResolvedValue({ files: [], root: "/root" }),
27
+ createFileOrFolder: vi.fn(),
28
+ deleteFileOrFolder: vi.fn(),
29
+ copyFileOrFolder: vi.fn(),
30
+ renameFileOrFolder: vi.fn(),
31
+ });
32
+ await tree.initialize(() => undefined);
33
+ });
34
+
35
+ it("derives loading for a refetch and a changed hidden-file filter", async () => {
36
+ sendSearchFiles.mockResolvedValue({
37
+ files: [],
38
+ query: "file",
39
+ totalFound: 0,
40
+ });
41
+ const { result, rerender } = renderHook(
42
+ ({ showHiddenFiles }) =>
43
+ useFileSearch({ query: "file", tree, showHiddenFiles }),
44
+ { initialProps: { showHiddenFiles: false } },
45
+ );
46
+ await waitFor(() => expect(result.current.state.status).toBe("success"));
47
+
48
+ const pending = new Deferred<
49
+ Awaited<ReturnType<EditRequests["sendSearchFiles"]>>
50
+ >();
51
+ sendSearchFiles.mockReturnValue(pending.promise);
52
+ act(() => result.current.refetch());
53
+ expect(result.current.state.status).toBe("loading");
54
+ await act(async () =>
55
+ pending.resolve({ files: [], query: "file", totalFound: 0 }),
56
+ );
57
+ expect(result.current.state.status).toBe("success");
58
+
59
+ sendSearchFiles.mockReturnValue(new Promise(() => undefined));
60
+ rerender({ showHiddenFiles: true });
61
+ expect(result.current.state.status).toBe("loading");
62
+ await waitFor(() =>
63
+ expect(sendSearchFiles).toHaveBeenLastCalledWith(
64
+ expect.objectContaining({ includeHidden: true }),
65
+ ),
66
+ );
67
+ });
68
+
69
+ it("keeps superseded scans serial and ignores their results", async () => {
70
+ const first = new Deferred<
71
+ Awaited<ReturnType<EditRequests["sendSearchFiles"]>>
72
+ >();
73
+ sendSearchFiles
74
+ .mockReturnValueOnce(first.promise)
75
+ .mockResolvedValue({ files: [], query: "file", totalFound: 0 });
76
+ const { result } = renderHook(() =>
77
+ useFileSearch({ query: "file", tree, showHiddenFiles: false }),
78
+ );
79
+ await waitFor(() => expect(sendSearchFiles).toHaveBeenCalledOnce());
80
+
81
+ act(() => result.current.refetch());
82
+ expect(sendSearchFiles).toHaveBeenCalledOnce();
83
+ expect(result.current.state.status).toBe("loading");
84
+ await act(async () =>
85
+ first.resolve({ files: [], query: "file", totalFound: 0 }),
86
+ );
87
+
88
+ expect(sendSearchFiles).toHaveBeenCalledTimes(2);
89
+ expect(result.current.state.status).toBe("success");
90
+ });
91
+ });
@@ -201,11 +201,16 @@ export const FileExplorer: React.FC<{
201
201
  () => filterHiddenTree(data, showHiddenFiles),
202
202
  [data, showHiddenFiles],
203
203
  );
204
- React.useEffect(() => {
205
- if (selectedFolder && !treeContainsId(visibleData, selectedFolder.id)) {
206
- setSelectedFolder(null);
207
- }
208
- }, [selectedFolder, visibleData]);
204
+ const selectedFolderId = selectedFolder?.id;
205
+ const selectedFolderVisible = React.useMemo(
206
+ () =>
207
+ selectedFolderId === undefined ||
208
+ treeContainsId(visibleData, selectedFolderId),
209
+ [visibleData, selectedFolderId],
210
+ );
211
+ if (!selectedFolderVisible) {
212
+ setSelectedFolder(null);
213
+ }
209
214
  const contextValue = React.useMemo<FileExplorerContextValue>(
210
215
  () => ({
211
216
  tree,
@@ -9,7 +9,7 @@ import {
9
9
  SaveIcon,
10
10
  } from "lucide-react";
11
11
  import type React from "react";
12
- import { useEffect, useRef, useState } from "react";
12
+ import { useEffect, useLayoutEffect, useRef, useState } from "react";
13
13
  import { renderShortcut } from "@/components/shortcuts/renderShortcut";
14
14
  import { Alert, AlertDescription } from "@/components/ui/alert";
15
15
  import { Button } from "@/components/ui/button";
@@ -82,7 +82,9 @@ export const FileViewer: React.FC<Props> = ({ file, onOpenNotebook }) => {
82
82
  // carry no draft to preserve.
83
83
  // We use a ref for internalValue so we don't call this effect on each keystroke
84
84
  const internalValueRef = useRef<string>(internalValue);
85
- internalValueRef.current = internalValue;
85
+ useLayoutEffect(() => {
86
+ internalValueRef.current = internalValue;
87
+ }, [internalValue]);
86
88
  useEffect(() => {
87
89
  return () => {
88
90
  if (