@marimo-team/frontend 0.25.3-dev5 → 0.25.3-dev55

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 (343) hide show
  1. package/dist/assets/CellStatus-cD3S0wEA.js +1 -0
  2. package/dist/assets/{ConnectedDataExplorerComponent-t8fH_TNc.js → ConnectedDataExplorerComponent-D9UK8o9A.js} +1 -1
  3. package/dist/assets/{ErrorBoundary-B4n9EnHB.js → ErrorBoundary-DJhsFCsX.js} +2 -2
  4. package/dist/assets/JsonOutput-Pm6OCP3J.js +11 -0
  5. package/dist/assets/{LazyAnyLanguageCodeMirror-CKcCbG2G.js → LazyAnyLanguageCodeMirror-BPYE76jR.js} +2 -2
  6. package/dist/assets/MarimoErrorOutput-DlEjLCZ-.js +10 -0
  7. package/dist/assets/{RenderHTML-9bMgbYmo.js → RenderHTML-CfQ6i3QV.js} +1 -1
  8. package/dist/assets/RunButton-DvuvkgrG.js +1 -0
  9. package/dist/assets/{add-cell-with-ai-C_SruRWS.js → add-cell-with-ai-BhmuohAY.js} +7 -7
  10. package/dist/assets/{add-connection-dialog-CoE_PhuD.js → add-connection-dialog-D7EYSraq.js} +2 -2
  11. package/dist/assets/{agent-panel-0b3k4QzJ.js → agent-panel-GFhlWVub.js} +3 -3
  12. package/dist/assets/ai-model-dropdown-YT1HiXiK.js +13 -0
  13. package/dist/assets/{alert-dialog-BaPGzeOK.js → alert-dialog-C-sWmKva.js} +1 -1
  14. package/dist/assets/any-language-editor-DUMe1UCk.js +1 -0
  15. package/dist/assets/{app-config-button-iXdCWvfC.js → app-config-button-C_whRVO8.js} +1 -1
  16. package/dist/assets/arrays-0R2reF2y.js +1 -0
  17. package/dist/assets/{blob-iOBwGaP9.js → blob-e6VlOmQP.js} +1 -1
  18. package/dist/assets/{button-COI-NGrx.js → button-D8gVKwe-.js} +1 -1
  19. package/dist/assets/{cache-panel-DTY1dqvi.js → cache-panel-Byolv2ht.js} +1 -1
  20. package/dist/assets/{capabilities-BJvf-MD4.js → capabilities-Pq1YNPlK.js} +1 -1
  21. package/dist/assets/{cell-editor-CR-tZNnc.js → cell-editor-3cVIcyHK.js} +6 -6
  22. package/dist/assets/cell-link-DFqfW-Ym.js +1 -0
  23. package/dist/assets/{cells-rA_jsZCj.js → cells-T9_Y4Jlv.js} +56 -56
  24. package/dist/assets/{chat-display-luF7psRA.js → chat-display-BtaARI-N.js} +1 -1
  25. package/dist/assets/chat-panel-eT2lfQrB.js +4 -0
  26. package/dist/assets/chat-ui-nNFa2LdU.js +4 -0
  27. package/dist/assets/{circle-DXf-VWZG.js → circle-B6D_6D8w.js} +1 -1
  28. package/dist/assets/code-visibility-CV3S4xuf.js +1 -0
  29. package/dist/assets/coerce-DWjFeXOY.js +1 -0
  30. package/dist/assets/{column-preview-uosAAL4X.js → column-preview-C6uXRJry.js} +1 -1
  31. package/dist/assets/command-palette-OZPTmtR-.js +1 -0
  32. package/dist/assets/{common-lisZ0l6a.js → common-DeeLgjWQ.js} +1 -1
  33. package/dist/assets/{components-CP_UoFtT.js → components-DrU8dOzy.js} +1 -1
  34. package/dist/assets/{components-S7ZwOrvO.js → components-DyAF3_Nf.js} +1 -1
  35. package/dist/assets/{components-CXfO-F47.js → components-lDm0qi2t.js} +1 -1
  36. package/dist/assets/config-CjQ4WJnJ.js +1 -0
  37. package/dist/assets/config-DGX7JYor.js +1 -0
  38. package/dist/assets/config-DzdxZb17.js +1 -0
  39. package/dist/assets/{copy-D8O_CYCG.js → copy-KlhazlhB.js} +1 -1
  40. package/dist/assets/{copy-icon-BIiRXh_3.js → copy-icon-8-a5vovc.js} +1 -1
  41. package/dist/assets/{createReducer-DgRODTW6.js → createReducer-HKXUbqyz.js} +1 -1
  42. package/dist/assets/{datasource-V0-52XOh.js → datasource-gqFgG9Qd.js} +1 -1
  43. package/dist/assets/{dates-CfmdTL6B.js → dates-CQMwFko2.js} +1 -1
  44. package/dist/assets/{dependency-graph-panel-B0uP-_id.js → dependency-graph-panel-BFFxLgR6.js} +1 -1
  45. package/dist/assets/{dist-iaThVDkA.js → dist-jlhMTKo_.js} +1 -1
  46. package/dist/assets/{documentation-panel-DVl_ebWy.js → documentation-panel-BBpGbT-f.js} +1 -1
  47. package/dist/assets/{download-Bx55LMki.js → download-C8Gw-D0Y.js} +1 -1
  48. package/dist/assets/{dropdown-menu-C1rnl-0o.js → dropdown-menu-DS6umAnH.js} +1 -1
  49. package/dist/assets/{edit-page-CGyjT-CQ.js → edit-page-B0lN_Rib.js} +6 -6
  50. package/dist/assets/{error-banner-Cd69vvyX.js → error-banner-B5kzVf_0.js} +1 -1
  51. package/dist/assets/{error-entries-DOWpdGvI.js → error-entries-uyqOG5tg.js} +1 -1
  52. package/dist/assets/{error-panel-C-uxLOHF.js → error-panel-Cli5zPWi.js} +1 -1
  53. package/dist/assets/{errors-DCrsa1nH.js → errors-CNsgAw6m.js} +1 -1
  54. package/dist/assets/{es-Cqst59WX.js → es-N9SHpmG7.js} +1 -1
  55. package/dist/assets/{feedback-button-B99ThNJW.js → feedback-button-CvrwBGr7.js} +2 -2
  56. package/dist/assets/feedback-button-DknbRCN_.js +1 -0
  57. package/dist/assets/file-explorer-panel-BSU5ultY.js +1 -0
  58. package/dist/assets/{file-name-input-ChnBsDE_.js → file-name-input-CDrRlIiX.js} +1 -1
  59. package/dist/assets/{floating-outline-MYqQW4Vp.js → floating-outline-GQD9km0W.js} +1 -1
  60. package/dist/assets/{formats-yZ06h1Pl.js → formats-zylfwejq.js} +1 -1
  61. package/dist/assets/{formatting-BqrP-n49.js → formatting-uom1es4b.js} +1 -1
  62. package/dist/assets/{fullscreen-B9tHcfpV.js → fullscreen-BWTs8pYZ.js} +1 -1
  63. package/dist/assets/{gallery-page-DSL0J4md.js → gallery-page-CxmV1JLF.js} +1 -1
  64. package/dist/assets/{ganttDiagram-EL5Y4UJY-2n7Ig5Jm.js → ganttDiagram-EL5Y4UJY-CYqkCfFG.js} +1 -1
  65. package/dist/assets/{glide-data-editor-B1MWlKeM.js → glide-data-editor-B1kpoetH.js} +2 -2
  66. package/dist/assets/{glide-utils-CIrAxdjv.js → glide-utils-HFqhvWHI.js} +2 -2
  67. package/dist/assets/{home-page-B1M3X9Yf.js → home-page-KnJyha6O.js} +3 -3
  68. package/dist/assets/{hooks-Bj0C0zrZ.js → hooks-w6zKs3CJ.js} +1 -1
  69. package/dist/assets/{html-to-image-DK9Zl1lU.js → html-to-image-D_0rd3_B.js} +1 -1
  70. package/dist/assets/index-Bu8_lPMx.js +30 -0
  71. package/dist/assets/index-evFWHKmM.css +2 -0
  72. package/dist/assets/{input-BoBxR50a.js → input-BZdRe7pb.js} +1 -1
  73. package/dist/assets/{kiosk-mode-BZECfg9k.js → kiosk-mode-BGmMSc7A.js} +1 -1
  74. package/dist/assets/layout-sYdHI-lW.js +9 -0
  75. package/dist/assets/linear-Bma8J3iM.js +1 -0
  76. package/dist/assets/links-CXcpNIM0.js +1 -0
  77. package/dist/assets/{logs-panel-y0XnfulJ.js → logs-panel-CdUc5vtA.js} +1 -1
  78. package/dist/assets/{markdown-renderer-Q5AhbRU3.js → markdown-renderer-pKc_bFac.js} +1 -1
  79. package/dist/assets/{menu-items-BrwnTgyK.js → menu-items-BgUyNUpu.js} +1 -1
  80. package/dist/assets/{mermaid-a3xYymTy.js → mermaid-B6mqMm5_.js} +2 -2
  81. package/dist/assets/{name-cell-input-BNOnF6-0.js → name-cell-input-csH3Cy54.js} +1 -1
  82. package/dist/assets/numbers-CsZVw3Dc.js +1 -0
  83. package/dist/assets/{objects-snlMn4yk.js → objects-rLPlQ3X_.js} +1 -1
  84. package/dist/assets/{once-Haz_uxEL.js → once-6KNTRW0u.js} +1 -1
  85. package/dist/assets/{outline-panel-Be7stWwv.js → outline-panel-DhsHYTuo.js} +1 -1
  86. package/dist/assets/packages-panel-lkNX6Oov.js +1 -0
  87. package/dist/assets/{panel-accordion-state-GeMEdCPw.js → panel-accordion-state-Bco6nsUI.js} +1 -1
  88. package/dist/assets/panels-B2UqDbHy.js +107 -0
  89. package/dist/assets/precisionRound-BxlUs8G4.js +1 -0
  90. package/dist/assets/process-output-B2-bT4cq.js +1 -0
  91. package/dist/assets/{quadrantDiagram-AXDQQJYC-yBhCDI7E.js → quadrantDiagram-AXDQQJYC-D9KH5Zge.js} +1 -1
  92. package/dist/assets/react-vega-BTNeboR2.js +189 -0
  93. package/dist/assets/{readonly-code-display-YULOXf4h.js → readonly-code-display-Dsp6vWPh.js} +1 -1
  94. package/dist/assets/{readonly-python-code-CadgXgRu.js → readonly-python-code-B81AjzXM.js} +1 -1
  95. package/dist/assets/{renderShortcut-BkVJQ_Zh.js → renderShortcut-H60uiHBp.js} +1 -1
  96. package/dist/assets/{reveal-component-B9FLjTg_.js → reveal-component-CrQRQhW4.js} +37 -37
  97. package/dist/assets/run-page-BGzpa3D9.js +1 -0
  98. package/dist/assets/{runs-BLp5jezv.js → runs-6lJxKV1w.js} +1 -1
  99. package/dist/assets/{runtime-details-state-DuChzsA6.js → runtime-details-state-CznIe40L.js} +1 -1
  100. package/dist/assets/{schemas-Dv42wLFq.js → schemas-ChRqMOTL.js} +1 -1
  101. package/dist/assets/{scratchpad-panel-DtB0A8xj.js → scratchpad-panel-ClLxdqdK.js} +1 -1
  102. package/dist/assets/{scroll-to-bottom-button-1FyowpWV.js → scroll-to-bottom-button-CIkON2WJ.js} +1 -1
  103. package/dist/assets/{secrets-panel-Bctbtkrp.js → secrets-panel-Dh3v7XPE.js} +1 -1
  104. package/dist/assets/{select-DTBn5eTb.js → select-3CVGlYrC.js} +1 -1
  105. package/dist/assets/{session-DqoWmb7X.js → session-B8Y-GJnS.js} +1 -1
  106. package/dist/assets/{session-panel-B6tA28BD.js → session-panel-C7-wmZhc.js} +1 -1
  107. package/dist/assets/{snippets-panel-CQ1dn3lA.js → snippets-panel-BARpOPyS.js} +1 -1
  108. package/dist/assets/{spec-BfSBcO5B.js → spec-uCpVvaAR.js} +1 -1
  109. package/dist/assets/{state-BUsmcyaT.js → state-BRYi026f.js} +3 -3
  110. package/dist/assets/{state-uwRvJLb8.js → state-BY4_P6ko.js} +1 -1
  111. package/dist/assets/{state-Bfh_2SLl.js → state-Bye6rCo4.js} +1 -1
  112. package/dist/assets/{state-BAJYRhmO.js → state-C0irhlOZ.js} +1 -1
  113. package/dist/assets/{state-DxQnC_1I.js → state-Ck7FMD5A.js} +1 -1
  114. package/dist/assets/{state-C3oDoyFe.js → state-DaWgt-rn.js} +1 -1
  115. package/dist/assets/{strings-B2vv-ENv.js → strings-C25fhMR_.js} +1 -1
  116. package/dist/assets/{swiper-component-a07domrZ.js → swiper-component-CRGRy5_O.js} +1 -1
  117. package/dist/assets/{switch-CTaiAjrz.js → switch-IoWE8PbW.js} +1 -1
  118. package/dist/assets/{tabs-BuQlZFbt.js → tabs-CY8E4r9U.js} +1 -1
  119. package/dist/assets/{terminal-Ck_vK-34.js → terminal-Bhcn_y1N.js} +1 -1
  120. package/dist/assets/{time-MgToa8VM.js → time-DswSXBDk.js} +1 -1
  121. package/dist/assets/{toast-BGzLGnvw.js → toast-Gzxi-le_.js} +1 -1
  122. package/dist/assets/tooltip-Brya4Ghn.js +61 -0
  123. package/dist/assets/{tooltip-C9TuY18o.js → tooltip-CkGm8X7m.js} +1 -1
  124. package/dist/assets/tracing-BW2ZVnXs.js +1 -0
  125. package/dist/assets/{tracing-panel-dgnr2fIN.js → tracing-panel-UKdh_YXC.js} +2 -2
  126. package/dist/assets/{tree-actions-DAd80MbP.js → tree-actions-Cb-nUK1F.js} +1 -1
  127. package/dist/assets/{use-restore-focus-BxEGwIbk.js → use-restore-focus-DdS7Tl0F.js} +1 -1
  128. package/dist/assets/{useAddCell-DGstJK0S.js → useAddCell-Bielffe0.js} +1 -1
  129. package/dist/assets/useAsyncData-DhhdOGGd.js +1 -0
  130. package/dist/assets/useBoolean-CMgj9hnq.js +1 -0
  131. package/dist/assets/{useCellActionButton-DY9RlJTb.js → useCellActionButton-Cu34Zs6x.js} +1 -1
  132. package/dist/assets/useDebounce-k9QTpQ9-.js +1 -0
  133. package/dist/assets/useDelayElapsed-BlsDT4TV.js +1 -0
  134. package/dist/assets/{useDeleteCell-DFBGlSQk.js → useDeleteCell-BV4VemCv.js} +1 -1
  135. package/dist/assets/{useDependencyPanelTab-DydxsE5d.js → useDependencyPanelTab-BEGqRiWi.js} +1 -1
  136. package/dist/assets/{useEventListener-D9cftcHj.js → useEventListener-CjDeLqFC.js} +1 -1
  137. package/dist/assets/{useIframeCapabilities-CwTkczVN.js → useIframeCapabilities-Czr9e4uG.js} +1 -1
  138. package/dist/assets/{useInstallPackage-iX1eufdl.js → useInstallPackage-C7neMPwh.js} +1 -1
  139. package/dist/assets/{useInterval--x3s1VpI.js → useInterval-DJdQmXnq.js} +1 -1
  140. package/dist/assets/{useNotebookActions-CupIPSlg.js → useNotebookActions-OLol05t9.js} +1 -1
  141. package/dist/assets/{useRunCells-BFjFX3hz.js → useRunCells-BFeZMi7f.js} +1 -1
  142. package/dist/assets/{useSplitCell-Tqv5pPtL.js → useSplitCell-DllkNFke.js} +1 -1
  143. package/dist/assets/useStateForKey-CGMJQT8x.js +1 -0
  144. package/dist/assets/{useTheme-COhSHkCj.js → useTheme-DLtPc-39.js} +1 -1
  145. package/dist/assets/{utils-CaQYvazl.js → utils-DfElfBpz.js} +1 -1
  146. package/dist/assets/vega-component-D7JFthDD.js +1 -0
  147. package/dist/assets/{vega-embed-container-DJPNkHPb.js → vega-embed-container-DtArSJz1.js} +1 -1
  148. package/dist/assets/vega-loader.browser-Bu25TKax.js +6 -0
  149. package/dist/assets/{write-secret-modal-DzKAInL2.js → write-secret-modal-7DzehGyr.js} +1 -1
  150. package/dist/assets/{xychartDiagram-S5SC5T6Z-KbGzG6-u.js → xychartDiagram-S5SC5T6Z-CeRnBS5e.js} +1 -1
  151. package/dist/index.html +76 -74
  152. package/package.json +4 -4
  153. package/src/__tests__/fixtures/{pair-preview.ts → pair-config.ts} +3 -2
  154. package/src/__tests__/host-imports.test.ts +1 -0
  155. package/src/__tests__/mount.test.ts +8 -8
  156. package/src/components/app-config/ai-config.tsx +6 -2
  157. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  158. package/src/components/chat/chat-panel.tsx +3 -7
  159. package/src/components/chat/chat-utils.ts +4 -8
  160. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  161. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  162. package/src/components/data-table/__tests__/export-actions.test.tsx +434 -6
  163. package/src/components/data-table/charts/charts.tsx +14 -19
  164. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  165. package/src/components/data-table/columns.tsx +2 -1
  166. package/src/components/data-table/context-menu.tsx +36 -20
  167. package/src/components/data-table/date-popover.tsx +6 -4
  168. package/src/components/data-table/export-actions.tsx +267 -163
  169. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +2 -2
  170. package/src/components/data-table/schemas.ts +14 -36
  171. package/src/components/editor/Output.tsx +5 -1
  172. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  173. package/src/components/editor/actions/__tests__/pair-with-agent-commands.test.ts +80 -252
  174. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +210 -34
  175. package/src/components/editor/actions/pair-with-agent-commands.ts +42 -51
  176. package/src/components/editor/actions/pair-with-agent-modal.tsx +100 -37
  177. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  178. package/src/components/editor/cell/CreateCellButton.tsx +8 -10
  179. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  180. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  181. package/src/components/editor/cell/code/cell-editor.tsx +1 -1
  182. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  183. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  184. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +30 -1
  185. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  186. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +277 -78
  187. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +196 -41
  188. package/src/components/editor/connections/components.tsx +1 -1
  189. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  190. package/src/components/editor/controls/command-palette.tsx +22 -10
  191. package/src/components/editor/dynamic-favicon.tsx +21 -40
  192. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +139 -36
  193. package/src/components/editor/errors/send-error-report-button.tsx +10 -7
  194. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  195. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  196. package/src/components/editor/file-tree/file-explorer.tsx +10 -5
  197. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  198. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  199. package/src/components/editor/output/MarimoTracebackOutput.tsx +28 -22
  200. package/src/components/editor/output/console/ConsoleOutput.tsx +8 -18
  201. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  202. package/src/components/icons/large-spinner.tsx +3 -14
  203. package/src/components/pages/home-page.tsx +4 -4
  204. package/src/components/slides/slide-cell-view.tsx +1 -1
  205. package/src/components/tracing/tracing.tsx +2 -2
  206. package/src/core/attachments/__tests__/toast.test.tsx +58 -0
  207. package/src/core/attachments/state.ts +5 -8
  208. package/src/core/attachments/toast.ts +15 -0
  209. package/src/core/cells/__tests__/error-report.test.ts +53 -3
  210. package/src/core/cells/__tests__/notebook-outline.test.ts +113 -0
  211. package/src/core/cells/__tests__/session.test.ts +18 -3
  212. package/src/core/cells/cells.ts +2 -1
  213. package/src/core/cells/error-report.ts +14 -6
  214. package/src/core/cells/pending-delete-service.ts +2 -6
  215. package/src/core/cells/session.ts +19 -1
  216. package/src/core/config/pair.ts +5 -3
  217. package/src/core/datasets/__tests__/data-source-discovery.test.ts +4 -4
  218. package/src/core/datasets/data-source-discovery.ts +4 -6
  219. package/src/core/edit-app.tsx +2 -2
  220. package/src/core/host/AGENTS.md +11 -2
  221. package/src/core/host/__tests__/attachments.test.ts +56 -0
  222. package/src/core/host/__tests__/document.test.ts +126 -0
  223. package/src/core/host/__tests__/events.test.ts +49 -0
  224. package/src/core/host/__tests__/fixtures.ts +41 -0
  225. package/src/core/host/__tests__/notebook-client.test.ts +496 -0
  226. package/src/core/host/connection.ts +81 -0
  227. package/src/core/host/events.ts +68 -0
  228. package/src/core/host/notebook-client.ts +316 -0
  229. package/src/core/host/resources/attachments.ts +19 -0
  230. package/src/core/host/resources/document.ts +101 -0
  231. package/src/core/hotkeys/actions.ts +2 -4
  232. package/src/core/islands/__tests__/bootstrap.test.ts +0 -26
  233. package/src/core/islands/bootstrap.ts +0 -1
  234. package/src/core/network/connection.ts +4 -2
  235. package/src/core/network/useConnectionNotice.ts +8 -2
  236. package/src/core/run-app.tsx +2 -2
  237. package/src/core/runtime/__tests__/runtime.test.ts +5 -12
  238. package/src/core/runtime/runtime.ts +5 -5
  239. package/src/core/static/files.ts +2 -0
  240. package/src/core/websocket/__tests__/{useMarimoKernelConnection.test.ts → connection-state.test.ts} +15 -17
  241. package/src/core/websocket/__tests__/{useWebSocket.test.ts → createConnectionTransport.test.ts} +1 -1
  242. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +149 -0
  243. package/src/core/websocket/__tests__/{useMarimoKernelConnection.hook.test.tsx → useNotebookSession.hook.test.tsx} +168 -253
  244. package/src/core/websocket/connection-state.ts +70 -0
  245. package/src/core/websocket/state.ts +7 -0
  246. package/src/core/websocket/transports/__tests__/sse.test.ts +18 -4
  247. package/src/core/websocket/transports/create.ts +31 -0
  248. package/src/core/websocket/transports/sse.ts +14 -2
  249. package/src/core/websocket/transports/transport.ts +12 -23
  250. package/src/core/websocket/types.ts +9 -39
  251. package/src/core/websocket/useNotebookClient.ts +78 -0
  252. package/src/core/websocket/{useMarimoKernelConnection.tsx → useNotebookSession.tsx} +53 -233
  253. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  254. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  255. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  256. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  257. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  258. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  259. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  260. package/src/hooks/useAsyncData.ts +3 -3
  261. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  262. package/src/hooks/useDebounce.ts +5 -8
  263. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  264. package/src/hooks/useDelayElapsed.ts +14 -10
  265. package/src/hooks/useElapsedTime.ts +3 -3
  266. package/src/hooks/useInputHistory.ts +4 -7
  267. package/src/hooks/useInternalStateWithSync.ts +6 -6
  268. package/src/hooks/useResizeObserver.ts +4 -7
  269. package/src/hooks/useScript.ts +67 -70
  270. package/src/hooks/useStateForKey.ts +93 -0
  271. package/src/mount.tsx +3 -3
  272. package/src/plugins/core/registerReactComponent.tsx +1 -1
  273. package/src/plugins/impl/DataTablePlugin.tsx +14 -15
  274. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  275. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  276. package/src/plugins/impl/FormPlugin.tsx +1 -1
  277. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  278. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +201 -1
  279. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  280. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  281. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +2 -0
  282. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  283. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  284. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  285. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  286. package/src/plugins/impl/chat/chat-ui.tsx +10 -9
  287. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  288. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  289. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  290. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  291. package/src/plugins/impl/panel/PanelPlugin.tsx +15 -16
  292. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  293. package/src/plugins/impl/panel/utils.ts +17 -0
  294. package/src/plugins/impl/vega/make-selectable.ts +3 -6
  295. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  296. package/src/stories/editor.stories.tsx +24 -17
  297. package/src/utils/__tests__/arrays.test.ts +35 -0
  298. package/src/utils/__tests__/numbers.test.ts +48 -1
  299. package/src/utils/__tests__/objects.test.ts +21 -0
  300. package/src/utils/arrays.ts +10 -0
  301. package/src/utils/numbers.ts +28 -17
  302. package/src/utils/objects.ts +3 -7
  303. package/src/utils/once.ts +19 -19
  304. package/dist/assets/CellStatus-nZpuDkm7.js +0 -1
  305. package/dist/assets/JsonOutput-d-k8v_17.js +0 -11
  306. package/dist/assets/MarimoErrorOutput-CLYUxiKW.js +0 -10
  307. package/dist/assets/RunButton-CFQUNN5c.js +0 -1
  308. package/dist/assets/ai-model-dropdown-C6GHHEdR.js +0 -5
  309. package/dist/assets/any-language-editor-BLCAD3yx.js +0 -1
  310. package/dist/assets/arrays-BqUx5USx.js +0 -1
  311. package/dist/assets/cell-link-1CynTVGQ.js +0 -1
  312. package/dist/assets/chat-panel-DDDTnLvB.js +0 -4
  313. package/dist/assets/chat-ui-nV_m4Btr.js +0 -4
  314. package/dist/assets/code-visibility-B-_GvwR5.js +0 -1
  315. package/dist/assets/coerce-CpE6kUdX.js +0 -1
  316. package/dist/assets/command-palette-BZ13RKDJ.js +0 -1
  317. package/dist/assets/config-DX8Y2WbJ.js +0 -1
  318. package/dist/assets/config-FzDtjF1-.js +0 -1
  319. package/dist/assets/config-VPSq73n8.js +0 -1
  320. package/dist/assets/feedback-button-DlqZkJwz.js +0 -1
  321. package/dist/assets/file-explorer-panel-D3CphidS.js +0 -1
  322. package/dist/assets/index-CXIE-QEG.css +0 -2
  323. package/dist/assets/index-DnIgeCyQ.js +0 -29
  324. package/dist/assets/layout-DBuMeWh0.js +0 -9
  325. package/dist/assets/linear-0eij_aAG.js +0 -1
  326. package/dist/assets/links-DkkL9DVa.js +0 -1
  327. package/dist/assets/numbers-Mk3-CcoU.js +0 -1
  328. package/dist/assets/packages-panel-9Qs1npCZ.js +0 -1
  329. package/dist/assets/panels-B9x935By.js +0 -4
  330. package/dist/assets/precisionRound-Bq_nmf5u.js +0 -1
  331. package/dist/assets/process-output-DNFKMODr.js +0 -1
  332. package/dist/assets/react-vega-C3rYGtP6.js +0 -189
  333. package/dist/assets/run-page-CjdGaic3.js +0 -1
  334. package/dist/assets/tooltip-mT3CGest.js +0 -61
  335. package/dist/assets/tracing-BARlJ_Dk.js +0 -1
  336. package/dist/assets/useAsyncData-C7aJJz0Y.js +0 -1
  337. package/dist/assets/useBoolean-BDD6_xQQ.js +0 -1
  338. package/dist/assets/useDebounce-B_ccBZTp.js +0 -1
  339. package/dist/assets/vega-component-CnadTgoE.js +0 -1
  340. package/dist/assets/vega-loader.browser-CLRkNgm7.js +0 -6
  341. package/src/core/host/__tests__/protocol.test.ts +0 -214
  342. package/src/core/host/protocol.ts +0 -1943
  343. package/src/core/websocket/useWebSocket.tsx +0 -88
@@ -0,0 +1,148 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+ import { act, render, renderHook, waitFor } from "@testing-library/react";
3
+ import { Activity } from "react";
4
+ import { describe, expect, it, vi } from "vitest";
5
+ import { Deferred } from "@/utils/Deferred";
6
+ import { useAsyncData } from "../useAsyncData";
7
+
8
+ function AsyncDataStatus({ fetch }: { fetch: () => Promise<string> }) {
9
+ const result = useAsyncData(fetch, []);
10
+ return (
11
+ <output
12
+ data-status={result.status}
13
+ data-fetching={result.isFetching}
14
+ data-error={result.error?.message ?? ""}
15
+ >
16
+ {result.data}
17
+ </output>
18
+ );
19
+ }
20
+
21
+ describe("useAsyncData", () => {
22
+ it.each(["success", "error"] as const)(
23
+ "restarts fetching when Activity reopens after %s",
24
+ async (initialStatus) => {
25
+ const initial = new Deferred<string>();
26
+ const restarted = new Deferred<string>();
27
+ const fetch = vi
28
+ .fn(() => restarted.promise)
29
+ .mockImplementationOnce(() => initial.promise);
30
+ const view = render(
31
+ <Activity mode="visible">
32
+ <AsyncDataStatus fetch={fetch} />
33
+ </Activity>,
34
+ );
35
+ await act(async () => {
36
+ if (initialStatus === "success") {
37
+ initial.resolve("first");
38
+ } else {
39
+ initial.reject(new Error("offline"));
40
+ }
41
+ });
42
+ expect(view.getByRole("status").dataset.status).toBe(initialStatus);
43
+
44
+ view.rerender(
45
+ <Activity mode="hidden">
46
+ <AsyncDataStatus fetch={fetch} />
47
+ </Activity>,
48
+ );
49
+ view.rerender(
50
+ <Activity mode="visible">
51
+ <AsyncDataStatus fetch={fetch} />
52
+ </Activity>,
53
+ );
54
+
55
+ expect(fetch).toHaveBeenCalledTimes(2);
56
+ expect(view.getByRole("status").dataset).toMatchObject({
57
+ status: initialStatus === "success" ? "loading" : "pending",
58
+ error: "",
59
+ fetching: "true",
60
+ });
61
+ expect(view.getByRole("status").textContent).toBe(
62
+ initialStatus === "success" ? "first" : "",
63
+ );
64
+ await act(async () => restarted.resolve("latest"));
65
+ expect(view.getByRole("status").dataset).toMatchObject({
66
+ status: "success",
67
+ fetching: "false",
68
+ });
69
+ expect(view.getByRole("status").textContent).toBe("latest");
70
+ },
71
+ );
72
+
73
+ it("ignores a request canceled when Activity hides", async () => {
74
+ const canceled = new Deferred<string>();
75
+ const restarted = new Deferred<string>();
76
+ const fetch = vi
77
+ .fn(() => restarted.promise)
78
+ .mockImplementationOnce(() => canceled.promise);
79
+ const view = render(
80
+ <Activity mode="visible">
81
+ <AsyncDataStatus fetch={fetch} />
82
+ </Activity>,
83
+ );
84
+ view.rerender(
85
+ <Activity mode="hidden">
86
+ <AsyncDataStatus fetch={fetch} />
87
+ </Activity>,
88
+ );
89
+ view.rerender(
90
+ <Activity mode="visible">
91
+ <AsyncDataStatus fetch={fetch} />
92
+ </Activity>,
93
+ );
94
+ expect(fetch).toHaveBeenCalledTimes(2);
95
+ await act(async () => canceled.resolve("obsolete"));
96
+ expect(view.getByRole("status").dataset.status).toBe("pending");
97
+ expect(view.getByRole("status").textContent).toBe("");
98
+ await act(async () => restarted.resolve("latest"));
99
+ expect(view.getByRole("status").dataset.status).toBe("success");
100
+ expect(view.getByRole("status").textContent).toBe("latest");
101
+ });
102
+
103
+ it("preserves data while refetching and ignores obsolete requests", async () => {
104
+ const requests = [
105
+ new Deferred<string>(),
106
+ new Deferred<string>(),
107
+ new Deferred<string>(),
108
+ ];
109
+ const fetch = vi
110
+ .fn(() => requests[0].promise)
111
+ .mockImplementationOnce(() => requests[0].promise)
112
+ .mockImplementationOnce(() => requests[1].promise)
113
+ .mockImplementationOnce(() => requests[2].promise);
114
+ const { result, rerender } = renderHook(
115
+ ({ key }) => useAsyncData(fetch, [key]),
116
+ { initialProps: { key: 0 } },
117
+ );
118
+ expect(result.current.status).toBe("pending");
119
+ await act(async () => requests[0].resolve("first"));
120
+ expect(result.current.data).toBe("first");
121
+ rerender({ key: 1 });
122
+ expect(result.current).toMatchObject({ status: "loading", data: "first" });
123
+ rerender({ key: 2 });
124
+ await act(async () => requests[1].resolve("obsolete"));
125
+ expect(result.current).toMatchObject({ status: "loading", data: "first" });
126
+ await act(async () => requests[2].resolve("latest"));
127
+ expect(result.current).toMatchObject({ status: "success", data: "latest" });
128
+ });
129
+
130
+ it("retries after errors and allows manual updates during a request", async () => {
131
+ const request = new Deferred<string>();
132
+ const fetch = vi
133
+ .fn<() => Promise<string>>()
134
+ .mockRejectedValueOnce(new Error("offline"))
135
+ .mockImplementation(() => request.promise);
136
+ const { result } = renderHook(() => useAsyncData(fetch, []));
137
+ await waitFor(() => expect(result.current.status).toBe("error"));
138
+ act(() => result.current.refetch());
139
+ expect(result.current.status).toBe("pending");
140
+ act(() => result.current.setData("manual"));
141
+ expect(result.current).toMatchObject({ status: "success", data: "manual" });
142
+ await act(async () => request.resolve("response"));
143
+ expect(result.current).toMatchObject({
144
+ status: "success",
145
+ data: "response",
146
+ });
147
+ });
148
+ });
@@ -121,6 +121,57 @@ describe("useDebounceControlledState", () => {
121
121
 
122
122
  expect(onChange).not.toHaveBeenCalled();
123
123
  });
124
+
125
+ it("replaces an uncommitted draft when the external value changes", () => {
126
+ const onChange = vi.fn();
127
+ const { result, rerender } = renderHook(
128
+ ({ value }) =>
129
+ useDebounceControlledState({
130
+ initialValue: value,
131
+ onChange,
132
+ delay: 1000,
133
+ }),
134
+ { initialProps: { value: "initial" } },
135
+ );
136
+
137
+ act(() => result.current.onChange("draft"));
138
+ rerender({ value: "external" });
139
+ expect(result.current.value).toBe("external");
140
+ act(() => vi.runAllTimers());
141
+ expect(result.current.debouncedValue).toBe("external");
142
+ expect(onChange).not.toHaveBeenCalled();
143
+ });
144
+
145
+ it("handles NaN external values without repeatedly resetting the draft", () => {
146
+ const onChange = vi.fn();
147
+ const { result, rerender } = renderHook(
148
+ ({ value }) =>
149
+ useDebounceControlledState({ initialValue: value, onChange }),
150
+ { initialProps: { value: Number.NaN } },
151
+ );
152
+
153
+ expect(result.current.value).toBeNaN();
154
+ act(() => result.current.onChange(1));
155
+ rerender({ value: Number.NaN });
156
+ expect(result.current.value).toBe(1);
157
+ rerender({ value: 2 });
158
+ expect(result.current.value).toBe(2);
159
+ rerender({ value: Number.NaN });
160
+ expect(result.current.value).toBeNaN();
161
+ });
162
+
163
+ it("resets the draft when the external value changes from zero to negative zero", () => {
164
+ const onChange = vi.fn();
165
+ const { result, rerender } = renderHook(
166
+ ({ value }) =>
167
+ useDebounceControlledState({ initialValue: value, onChange }),
168
+ { initialProps: { value: 0 } },
169
+ );
170
+
171
+ act(() => result.current.onChange(1));
172
+ rerender({ value: -0 });
173
+ expect(result.current.value).toBe(-0);
174
+ });
124
175
  });
125
176
 
126
177
  describe("useDebouncedCallback", () => {
@@ -0,0 +1,39 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+ import { renderHook } from "@testing-library/react";
3
+ import { describe, expect, it } from "vitest";
4
+ import { useDeepCompareMemoize } from "../useDeepCompareMemoize";
5
+
6
+ describe("useDeepCompareMemoize", () => {
7
+ it("keeps equal values stable and updates changed values immediately", () => {
8
+ const initial = { nested: [1, 2] };
9
+ const { result, rerender } = renderHook(
10
+ ({ value }) => useDeepCompareMemoize(value),
11
+ {
12
+ initialProps: { value: initial },
13
+ },
14
+ );
15
+ rerender({ value: { nested: [1, 2] } });
16
+ expect(result.current).toBe(initial);
17
+ const changed = { nested: [1, 3] };
18
+ rerender({ value: changed });
19
+ expect(result.current).toBe(changed);
20
+ rerender({ value: { nested: [1, 3] } });
21
+ expect(result.current).toBe(changed);
22
+ });
23
+
24
+ it("handles inline callbacks without causing additional renders", () => {
25
+ let renders = 0;
26
+ const { result, rerender } = renderHook(
27
+ ({ value }) => {
28
+ renders++;
29
+ return useDeepCompareMemoize({ callback: () => value });
30
+ },
31
+ { initialProps: { value: "initial" } },
32
+ );
33
+
34
+ const initialRenders = renders;
35
+ rerender({ value: "changed" });
36
+ expect(result.current.callback()).toBe("changed");
37
+ expect(renders).toBe(initialRenders + 1);
38
+ });
39
+ });
@@ -52,4 +52,32 @@ describe("useDelayElapsed", () => {
52
52
  // The pending timeout must be cleared, not left dangling.
53
53
  expect(vi.getTimerCount()).toBe(0);
54
54
  });
55
+
56
+ it("handles a NaN delay without resetting on every render", () => {
57
+ const { result, rerender } = renderHook(() => useDelayElapsed(Number.NaN));
58
+ expect(result.current).toBe(false);
59
+
60
+ act(() => vi.runAllTimers());
61
+ expect(result.current).toBe(true);
62
+ rerender();
63
+ expect(result.current).toBe(true);
64
+ expect(vi.getTimerCount()).toBe(0);
65
+ });
66
+
67
+ it("restarts a completed delay after the gate is temporarily disabled", () => {
68
+ const { result, rerender } = renderHook(
69
+ ({ delay }) => useDelayElapsed(delay),
70
+ { initialProps: { delay: 1000 } },
71
+ );
72
+ act(() => vi.advanceTimersByTime(1000));
73
+ expect(result.current).toBe(true);
74
+ rerender({ delay: 0 });
75
+ expect(result.current).toBe(true);
76
+ rerender({ delay: 1000 });
77
+ expect(result.current).toBe(false);
78
+ act(() => vi.advanceTimersByTime(999));
79
+ expect(result.current).toBe(false);
80
+ act(() => vi.advanceTimersByTime(1));
81
+ expect(result.current).toBe(true);
82
+ });
55
83
  });
@@ -32,6 +32,26 @@ describe("useInternalStateWithSync", () => {
32
32
  expect(result.current[0]).toBe("new value");
33
33
  });
34
34
 
35
+ it("preserves function values without calling them and resets when they change", () => {
36
+ const initial = vi.fn(() => "initial");
37
+ const draft = vi.fn(() => "draft");
38
+ const changed = vi.fn(() => "changed");
39
+ const { result, rerender } = renderHook(
40
+ ({ value }) => useInternalStateWithSync(value),
41
+ { initialProps: { value: initial } },
42
+ );
43
+
44
+ expect(result.current[0]).toBe(initial);
45
+ act(() => result.current[1](() => draft));
46
+ rerender({ value: initial });
47
+ expect(result.current[0]).toBe(draft);
48
+ rerender({ value: changed });
49
+ expect(result.current[0]).toBe(changed);
50
+ expect(initial).not.toHaveBeenCalled();
51
+ expect(draft).not.toHaveBeenCalled();
52
+ expect(changed).not.toHaveBeenCalled();
53
+ });
54
+
35
55
  it("should not update internal state if initial value remains the same", () => {
36
56
  const { result, rerender } = renderHook(
37
57
  ({ value }) => useInternalStateWithSync(value),
@@ -0,0 +1,77 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+ import { act, renderHook } from "@testing-library/react";
3
+ import { afterEach, describe, expect, it } from "vitest";
4
+ import { useScript } from "../useScript";
5
+
6
+ afterEach(() => {
7
+ document.querySelectorAll("script").forEach((script) => script.remove());
8
+ });
9
+
10
+ describe("useScript", () => {
11
+ it("observes loading and completion of a new script", async () => {
12
+ const { result } = renderHook(() => useScript("/first.js"));
13
+ expect(result.current).toBe("loading");
14
+ await act(async () => {
15
+ document.querySelector("script")?.dispatchEvent(new Event("load"));
16
+ });
17
+ expect(result.current).toBe("ready");
18
+ });
19
+
20
+ it("uses the new source's status when the source changes", async () => {
21
+ const { result, rerender } = renderHook(({ src }) => useScript(src), {
22
+ initialProps: { src: "/first.js" },
23
+ });
24
+ await act(async () => {
25
+ document.querySelector("script")?.dispatchEvent(new Event("load"));
26
+ });
27
+ expect(result.current).toBe("ready");
28
+ rerender({ src: "/second.js" });
29
+ expect(result.current).toBe("loading");
30
+ await act(async () => {
31
+ document
32
+ .querySelector('script[src="/second.js"]')
33
+ ?.dispatchEvent(new Event("error"));
34
+ });
35
+ expect(result.current).toBe("error");
36
+ });
37
+
38
+ it("shares an existing script and only removes scripts it owns", async () => {
39
+ const owner = renderHook(() =>
40
+ useScript("/first.js", { removeOnUnmount: true }),
41
+ );
42
+ const subscriber = renderHook(() =>
43
+ useScript("/first.js", { removeOnUnmount: true }),
44
+ );
45
+ expect(document.querySelectorAll("script")).toHaveLength(1);
46
+ await act(async () => {
47
+ document.querySelector("script")?.dispatchEvent(new Event("load"));
48
+ });
49
+ expect(subscriber.result.current).toBe("ready");
50
+ subscriber.unmount();
51
+ expect(document.querySelectorAll("script")).toHaveLength(1);
52
+ owner.unmount();
53
+ expect(document.querySelectorAll("script")).toHaveLength(0);
54
+ });
55
+
56
+ it("settles for remaining subscribers when the first unmounts mid-load", async () => {
57
+ const first = renderHook(() => useScript("/shared.js"));
58
+ const second = renderHook(() => useScript("/shared.js"));
59
+ first.unmount();
60
+ expect(second.result.current).toBe("loading");
61
+ await act(async () => {
62
+ document.querySelector("script")?.dispatchEvent(new Event("load"));
63
+ });
64
+ expect(second.result.current).toBe("ready");
65
+ // A later subscriber sees the settled status
66
+ expect(renderHook(() => useScript("/shared.js")).result.current).toBe(
67
+ "ready",
68
+ );
69
+ });
70
+
71
+ it("handles sources that are not valid inside a CSS selector", () => {
72
+ const src = '/odd"name].js';
73
+ const { result } = renderHook(() => useScript(src));
74
+ expect(result.current).toBe("loading");
75
+ expect(document.querySelector("script")?.getAttribute("src")).toBe(src);
76
+ });
77
+ });
@@ -0,0 +1,257 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+ import {
3
+ act,
4
+ fireEvent,
5
+ render,
6
+ renderHook,
7
+ screen,
8
+ } from "@testing-library/react";
9
+ import {
10
+ Profiler,
11
+ startTransition,
12
+ StrictMode,
13
+ Suspense,
14
+ useLayoutEffect,
15
+ useState,
16
+ } from "react";
17
+ import { describe, expect, it, vi } from "vitest";
18
+ import { Deferred } from "@/utils/Deferred";
19
+ import { useStateForKey } from "../useStateForKey";
20
+
21
+ describe("useStateForKey", () => {
22
+ it("preserves edits for the same key and creates fresh state on key changes", () => {
23
+ const initialize = vi.fn(() => new Set<string>());
24
+ const { result, rerender } = renderHook(
25
+ ({ key }) => useStateForKey(key, initialize),
26
+ { initialProps: { key: "first" } },
27
+ );
28
+ const initial = result.current[0];
29
+ act(() => result.current[1](new Set(["expanded"])));
30
+ rerender({ key: "first" });
31
+ expect(result.current[0]).toEqual(new Set(["expanded"]));
32
+ expect(initialize).toHaveBeenCalledOnce();
33
+
34
+ rerender({ key: "second" });
35
+ expect(result.current[0]).toEqual(new Set());
36
+ expect(result.current[0]).not.toBe(initial);
37
+ expect(initialize).toHaveBeenCalledTimes(2);
38
+
39
+ rerender({ key: "first" });
40
+ expect(result.current[0]).toEqual(new Set());
41
+ expect(initialize).toHaveBeenCalledTimes(3);
42
+ });
43
+
44
+ it("preserves queued functional updates and ignores setters from another key", () => {
45
+ const { result, rerender } = renderHook(
46
+ ({ key }) => useStateForKey(key, () => 0),
47
+ { initialProps: { key: "first" } },
48
+ );
49
+ const oldSetter = result.current[1];
50
+ act(() => {
51
+ result.current[1]((value) => value + 1);
52
+ result.current[1]((value) => value + 1);
53
+ });
54
+ expect(result.current[0]).toBe(2);
55
+
56
+ rerender({ key: "second" });
57
+ const staleUpdate = vi.fn(() => 99);
58
+ act(() => oldSetter(staleUpdate));
59
+ act(() => oldSetter(99));
60
+ expect(result.current[0]).toBe(0);
61
+ expect(staleUpdate).not.toHaveBeenCalled();
62
+ });
63
+
64
+ it("uses Object.is for NaN and signed zero keys", () => {
65
+ const { result, rerender } = renderHook(
66
+ ({ key }) => useStateForKey(key, () => "initial"),
67
+ { initialProps: { key: Number.NaN } },
68
+ );
69
+ act(() => result.current[1]("draft"));
70
+ rerender({ key: Number.NaN });
71
+ expect(result.current[0]).toBe("draft");
72
+
73
+ rerender({ key: 0 });
74
+ expect(result.current[0]).toBe("initial");
75
+ act(() => result.current[1]("draft"));
76
+ rerender({ key: -0 });
77
+ expect(result.current[0]).toBe("initial");
78
+ });
79
+
80
+ it("stores function values without invoking them", () => {
81
+ const initial = vi.fn();
82
+ const changed = vi.fn();
83
+ const { result, rerender } = renderHook(
84
+ ({ key, value }) => useStateForKey(key, () => value),
85
+ { initialProps: { key: "first", value: initial } },
86
+ );
87
+ expect(result.current[0]).toBe(initial);
88
+ act(() => result.current[1](() => changed));
89
+ expect(result.current[0]).toBe(changed);
90
+ rerender({ key: "second", value: initial });
91
+ expect(result.current[0]).toBe(initial);
92
+ expect(initial).not.toHaveBeenCalled();
93
+ expect(changed).not.toHaveBeenCalled();
94
+ });
95
+
96
+ it("uses a changed initializer only after the key changes", () => {
97
+ const first = vi.fn(() => "first default");
98
+ const second = vi.fn(() => "second default");
99
+ const { result, rerender } = renderHook(
100
+ ({ key, initialize }) => useStateForKey(key, initialize),
101
+ { initialProps: { key: "A", initialize: first } },
102
+ );
103
+ act(() => result.current[1]("draft"));
104
+ rerender({ key: "A", initialize: second });
105
+ expect(result.current[0]).toBe("draft");
106
+ expect(second).not.toHaveBeenCalled();
107
+ rerender({ key: "B", initialize: second });
108
+ expect(result.current[0]).toBe("second default");
109
+ expect(second).toHaveBeenCalledOnce();
110
+ });
111
+
112
+ it("compares object keys by reference rather than their contents", () => {
113
+ const key = { id: "A" };
114
+ const { result, rerender } = renderHook(
115
+ ({ key }) => useStateForKey(key, () => 0),
116
+ { initialProps: { key } },
117
+ );
118
+ act(() => result.current[1](1));
119
+ rerender({ key });
120
+ expect(result.current[0]).toBe(1);
121
+ rerender({ key: { id: "A" } });
122
+ expect(result.current[0]).toBe(0);
123
+ });
124
+
125
+ it("keeps setters stable for the same key and accepts old setters when that key returns", () => {
126
+ const { result, rerender } = renderHook(
127
+ ({ key }) => useStateForKey(key, () => 0),
128
+ { initialProps: { key: "A" } },
129
+ );
130
+ const setA = result.current[1];
131
+ act(() => setA(1));
132
+ rerender({ key: "A" });
133
+ expect(result.current[1]).toBe(setA);
134
+ rerender({ key: "B" });
135
+ expect(result.current[1]).not.toBe(setA);
136
+ act(() => setA(2));
137
+ expect(result.current[0]).toBe(0);
138
+ rerender({ key: "A" });
139
+ act(() => setA(3));
140
+ expect(result.current[0]).toBe(3);
141
+ });
142
+
143
+ it("retries the owner without rendering or committing stale state to children", () => {
144
+ const onOwnerRender = vi.fn();
145
+ const onChildRender = vi.fn();
146
+ const onCommit = vi.fn();
147
+ const onLayout = vi.fn();
148
+ function Child({ owner, value }: { owner: string; value: string }) {
149
+ onChildRender(owner, value);
150
+ useLayoutEffect(() => onLayout(owner, value), [owner, value]);
151
+ return <span>{value}</span>;
152
+ }
153
+ function Owner({ owner }: { owner: string }) {
154
+ onOwnerRender();
155
+ const [value, setValue] = useStateForKey(owner, () => `${owner} default`);
156
+ return (
157
+ <>
158
+ <button onClick={() => setValue("draft")}>Edit</button>
159
+ <Child owner={owner} value={value} />
160
+ </>
161
+ );
162
+ }
163
+ const content = (owner: string) => (
164
+ <Profiler id="owner" onRender={onCommit}>
165
+ <Owner owner={owner} />
166
+ </Profiler>
167
+ );
168
+ const { rerender } = render(content("A"));
169
+ const button = screen.getByRole("button", { name: "Edit" });
170
+ fireEvent.click(button);
171
+ expect(screen.getByText("draft")).toBeInTheDocument();
172
+ onOwnerRender.mockClear();
173
+ onChildRender.mockClear();
174
+ onCommit.mockClear();
175
+ onLayout.mockClear();
176
+
177
+ rerender(content("B"));
178
+ expect(onOwnerRender.mock.calls.length).toBeGreaterThan(1);
179
+ expect(onChildRender.mock.calls).toEqual([["B", "B default"]]);
180
+ expect(onLayout.mock.calls).toEqual([["B", "B default"]]);
181
+ expect(onCommit).toHaveBeenCalledOnce();
182
+ expect(screen.getByRole("button", { name: "Edit" })).toBe(button);
183
+ });
184
+
185
+ it("keeps state correct when Strict Mode repeats initialization and updates", () => {
186
+ const initialize = vi.fn(() => new Set<string>());
187
+ const { result, rerender } = renderHook(
188
+ ({ key }) => useStateForKey(key, initialize),
189
+ { initialProps: { key: "A" }, wrapper: StrictMode },
190
+ );
191
+ expect(initialize.mock.calls.length).toBeGreaterThan(1);
192
+ act(() => result.current[1]((previous) => new Set([...previous, "draft"])));
193
+ expect(result.current[0]).toEqual(new Set(["draft"]));
194
+ rerender({ key: "B" });
195
+ expect(result.current[0]).toEqual(new Set());
196
+ });
197
+
198
+ it.each(["completes", "is canceled"])(
199
+ "preserves visible edits when a suspended transition %s",
200
+ async (outcome) => {
201
+ const pending = new Deferred<void>();
202
+ let blocked = true;
203
+ const suspended = vi.fn();
204
+ function Pending({ owner }: { owner: string }) {
205
+ if (owner === "B" && blocked) {
206
+ suspended();
207
+ throw pending.promise;
208
+ }
209
+ return null;
210
+ }
211
+ function Owner({ owner }: { owner: string }) {
212
+ const [value, setValue] = useStateForKey(owner, () => 0);
213
+ return (
214
+ <>
215
+ <button onClick={() => setValue((previous) => previous + 1)}>
216
+ {owner}:{value}
217
+ </button>
218
+ <Pending owner={owner} />
219
+ </>
220
+ );
221
+ }
222
+ function Harness() {
223
+ const [owner, setOwner] = useState("A");
224
+ return (
225
+ <>
226
+ <button onClick={() => startTransition(() => setOwner("B"))}>
227
+ Switch owner
228
+ </button>
229
+ <button onClick={() => setOwner("A")}>Keep owner</button>
230
+ <Suspense fallback="Loading">
231
+ <Owner owner={owner} />
232
+ </Suspense>
233
+ </>
234
+ );
235
+ }
236
+ render(<Harness />);
237
+ fireEvent.click(screen.getByRole("button", { name: "A:0" }));
238
+ fireEvent.click(screen.getByRole("button", { name: "Switch owner" }));
239
+ expect(suspended).toHaveBeenCalled();
240
+ fireEvent.click(screen.getByRole("button", { name: "A:1" }));
241
+ expect(screen.getByRole("button", { name: "A:2" })).toBeInTheDocument();
242
+ if (outcome === "is canceled") {
243
+ fireEvent.click(screen.getByRole("button", { name: "Keep owner" }));
244
+ }
245
+ await act(async () => {
246
+ blocked = false;
247
+ pending.resolve();
248
+ await pending.promise;
249
+ });
250
+ expect(
251
+ screen.getByRole("button", {
252
+ name: outcome === "completes" ? "B:0" : "A:2",
253
+ }),
254
+ ).toBeInTheDocument();
255
+ },
256
+ );
257
+ });
@@ -322,7 +322,7 @@ export function useAsyncData<T>(
322
322
  const [nonce, setNonce] = useState(0);
323
323
  const [result, setResult] = useState<
324
324
  PendingResult<T> | LoadingResult<T> | ErrorResult<T> | SuccessResult<T>
325
- >(Result.pending());
325
+ >(() => Result.pending());
326
326
 
327
327
  const asProps =
328
328
  typeof loaderOrProps === "function"
@@ -339,12 +339,12 @@ export function useAsyncData<T>(
339
339
  keepPrevious = true;
340
340
  },
341
341
  };
342
+ // Activity can restart this effect without changing dependencies, so
343
+ // loading state must follow the request lifecycle.
342
344
  setResult((prevResult) => {
343
- // If we have previous data, show reloading state
344
345
  if (prevResult.status === "success" || prevResult.status === "loading") {
345
346
  return Result.loading(prevResult.data);
346
347
  }
347
- // Otherwise, show initial loading state
348
348
  return Result.pending();
349
349
  });
350
350
  fetchStable(context)
@@ -42,11 +42,13 @@ export function useDetectedDataSources(
42
42
  dialects: [...connections.values()].map((connection) =>
43
43
  connection.dialect.toLowerCase(),
44
44
  ),
45
- storageProtocols: storageNamespaces.map((namespace) =>
46
- namespace.protocol.toLowerCase(),
45
+ storageProtocols: new Set(
46
+ storageNamespaces.map((namespace) => namespace.protocol.toLowerCase()),
47
47
  ),
48
- storageBackendTypes: storageNamespaces.map((namespace) =>
49
- namespace.backendType.toLowerCase(),
48
+ storageBackendTypes: new Set(
49
+ storageNamespaces.map((namespace) =>
50
+ namespace.backendType.toLowerCase(),
51
+ ),
50
52
  ),
51
53
  };
52
54