@marimo-team/islands 0.25.3-dev19 → 0.25.3-dev192
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.
- package/dist/{ConnectedDataExplorerComponent-CNMliVzw.js → ConnectedDataExplorerComponent-BBbOElk3.js} +21 -19
- package/dist/{ErrorBoundary-CBHB755J.js → ErrorBoundary-Dw6uRKVk.js} +19 -19
- package/dist/{abnfDiagram-VCTEODGH-D-Ez91gV.js → abnfDiagram-VCTEODGH-DhMT6Uhd.js} +2 -2
- package/dist/any-language-editor-BVFHDDQ3.js +75 -0
- package/dist/{architectureDiagram-5GKGNRK7-CTpcNtO8.js → architectureDiagram-5GKGNRK7-BVbEiJLq.js} +1 -1
- package/dist/{arrays-CN1djC4k.js → arrays-R6JsImiT.js} +8 -0
- package/dist/assets/__vite-browser-external-C8gQho07.js +1 -0
- package/dist/assets/worker-C6e9ERyd.js +113 -0
- package/dist/{button-CvjkwdSk.js → button-BQRdqQi1.js} +199 -192
- package/dist/{capabilities-CQ5IjhzL.js → capabilities-CI4xUPsp.js} +2 -2
- package/dist/{chat-ui-DaHynk8g.js → chat-ui-BguxTQEL.js} +541 -540
- package/dist/{check-EDwSwP89.js → check-C9Mp9eCK.js} +1 -1
- package/dist/{chunk-IMKFNOWR-DGxi-zeo.js → chunk-IMKFNOWR-By5hSbJJ.js} +1 -1
- package/dist/{chunk-SHT3W25Y-CV9J3rAB.js → chunk-SHT3W25Y-DFrEYG54.js} +1 -1
- package/dist/{chunk-SVP7TREG-BL-9Fqgg.js → chunk-SVP7TREG-DFXGc9ew.js} +1 -1
- package/dist/{chunk-TICWLB2K-Cu_XMjDM.js → chunk-TICWLB2K-DZdNIpoZ.js} +1 -1
- package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
- package/dist/{classDiagram-ZZMXUADV-BXXV13Hq.js → classDiagram-ZZMXUADV-Cgj_eEQA.js} +2 -2
- package/dist/{classDiagram-v2-VYDZK3BY-B7FXJ1Cp.js → classDiagram-v2-VYDZK3BY-DLAUrSqj.js} +2 -2
- package/dist/{code-visibility-CEXXuuOE.js → code-visibility-v_7ijLtx.js} +25 -25
- package/dist/{common-OLE92qsR.js → common-BUwWHpnK.js} +6994 -6819
- package/dist/{copy-icon-Ft7t9bqz.js → copy-icon-De584pZ4.js} +3 -3
- package/dist/{cynefinDiagram-5FMLGOSQ-CH96rwba.js → cynefinDiagram-5FMLGOSQ-Cikf9AKa.js} +1 -1
- package/dist/{dagre-GXQ25YYZ-BD3PmpFY.js → dagre-GXQ25YYZ-C5KEi5_l.js} +1 -1
- package/dist/{data-grid-overlay-editor-CpHX98zN.js → data-grid-overlay-editor-4IZymnXZ.js} +5 -5
- package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
- package/dist/{diagram-S7CK7UJ4-BsSJJldB.js → diagram-S7CK7UJ4-BaoBTQjR.js} +1 -1
- package/dist/{diagram-UQ7AKVKN-muQtNZc4.js → diagram-UQ7AKVKN-DDufBsUA.js} +1 -1
- package/dist/{diagram-VX7I27RA-CdhuZAqR.js → diagram-VX7I27RA-CY6RALJ8.js} +1 -1
- package/dist/{diagram-Z3DM3KII-BD8mXjOj.js → diagram-Z3DM3KII-Bc-LHsD6.js} +1 -1
- package/dist/{dialog-CEBFZNyc.js → dialog-DVRfQuhD.js} +25 -25
- package/dist/{dist-F8Ep36rk.js → dist-B1M0bGO3.js} +1 -1
- package/dist/{dist-D6d86eNJ.js → dist-BA13RJj5.js} +1 -1
- package/dist/{dist-Bvs24R_E.js → dist-BJtmSgcE.js} +48 -48
- package/dist/{dist-CN_BeG2I.js → dist-Ba2VIn1K.js} +53 -53
- package/dist/{dist-PXYI_yuP.js → dist-Bj4bmwUm.js} +12 -12
- package/dist/{dist-DoE4XJBC.js → dist-BmS-ZGjl.js} +8626 -8626
- package/dist/{dist-BorMD-uX.js → dist-Bzogh7CW.js} +87 -87
- package/dist/{dist-CN8zI-XC.js → dist-CThPVKaP.js} +1385 -1382
- package/dist/{dist-DiDibOYr.js → dist-CTi5FTiV.js} +87 -87
- package/dist/{dist-diIx5qJW.js → dist-Ck_C32ye.js} +2 -2
- package/dist/{dist-BLfKmR-l.js → dist-D1fnFp1J.js} +1 -1
- package/dist/{dist-OR1OnN-y.js → dist-D3x9__Fu.js} +46 -46
- package/dist/{dist-BMQf5zIS.js → dist-D7mrh5sO.js} +36 -36
- package/dist/{dist-CSJ0MDRw.js → dist-D9Q3xjiP.js} +1 -1
- package/dist/{dist-BuaRHxOc.js → dist-DP60Ubh8.js} +2 -2
- package/dist/{dist-CSbh1PNP.js → dist-DTv9KkpW.js} +12 -12
- package/dist/{dist-D-wwqlzJ.js → dist-DVVwOfrD.js} +109 -109
- package/dist/{dist-C8fAKSH1.js → dist-DYXM73-k.js} +36 -36
- package/dist/{dist-DBtSl3sf.js → dist-DjeybPOb.js} +1 -1
- package/dist/{dist-CrSP_H1M.js → dist-DtGn9N4h.js} +197 -197
- package/dist/{dist-CLdpGQVr.js → dist-KN80qWCP.js} +2 -2
- package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
- package/dist/{dist-Cg7gzw-a.js → dist-O0qFpEJh.js} +56 -56
- package/dist/{dist-CkTTIdBe.js → dist-OUle2RT7.js} +207 -207
- package/dist/{dist-BK4Oj99o.js → dist-T5rh6JvL.js} +1 -1
- package/dist/{dropdown-menu-CP_qCYHC.js → dropdown-menu-drDjxJyV.js} +1481 -1433
- package/dist/{ebnfDiagram-PWID7BFC-BIN1sweE.js → ebnfDiagram-PWID7BFC-Cz0J-tcC.js} +2 -2
- package/dist/{erDiagram-RLTQ6QDP-Ctp5cfoN.js → erDiagram-RLTQ6QDP-DTXTDgFu.js} +1 -1
- package/dist/{error-banner-D-WCIiMt.js → error-banner-DxJ3s8YO.js} +86 -86
- package/dist/{error-entries-Bv6bZPF3.js → error-entries-CZ1NQ6Kv.js} +4771 -4702
- package/dist/{esm-D2gTsXEV.js → esm-Bbw_Njjx.js} +575 -575
- package/dist/{esm-AMFssoNp.js → esm-mDQNEFe_.js} +354 -354
- package/dist/{feedback-button-_f7fx7zj.js → feedback-button-MpeoIn1D.js} +37 -37
- package/dist/{flowDiagram-HODETNUW-3EfLrlFb.js → flowDiagram-HODETNUW-RbwWf2zc.js} +2 -2
- package/dist/{formats-27CW_UP4.js → formats-DbXnijtH.js} +3 -3
- package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
- package/dist/{glide-data-editor-Cmu4vWKD.js → glide-data-editor-DU8_EDG5.js} +370 -368
- package/dist/{glide-utils-DJyqh0EM.js → glide-utils-D98laidb.js} +16 -16
- package/dist/{globals-1Hfhz6Dx.js → globals-BeBDOk22.js} +3 -3
- package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
- package/dist/{html-to-image-Bf8QMGi4.js → html-to-image-D_4_hQ8U.js} +668 -622
- package/dist/{infoDiagram-27XIBGKW-DipQoHxl.js → infoDiagram-27XIBGKW-DlmzZjmI.js} +1 -1
- package/dist/{ishikawaDiagram-5VMMS53U-DgFMjSGz.js → ishikawaDiagram-5VMMS53U-EN3TYP_q.js} +1 -1
- package/dist/{kanban-definition-UXKFOSKX-w1ACVjgl.js → kanban-definition-UXKFOSKX-CK_-Ru7Q.js} +1 -1
- package/dist/{label-BDKF3Afc.js → label-Qvw1E97W.js} +3 -3
- package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
- package/dist/{loader-BQOfjTIJ.js → loader-DOVB0Syl.js} +12 -12
- package/dist/main.js +9822 -7824
- package/dist/{mermaid-CHo49Yv-.js → mermaid-DU9ZRmkG.js} +38 -38
- package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
- package/dist/{mindmap-definition-YA3MSWOX-CrT0KSfX.js → mindmap-definition-YA3MSWOX-C-hqgUvF.js} +1 -1
- package/dist/{number-overlay-editor-BQN3-Nye.js → number-overlay-editor-BoIkyh6H.js} +3 -3
- package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
- package/dist/{once-BtE-wKRL.js → once-eG8Vmvjd.js} +1 -1
- package/dist/{pegDiagram-XKGWAZYB-BqvywNJA.js → pegDiagram-XKGWAZYB-BiMejYPB.js} +2 -2
- package/dist/{pieDiagram-E7YTZNPT-BfxJ_XCc.js → pieDiagram-E7YTZNPT-B1zCZ1eq.js} +1 -1
- package/dist/precisionRound-DhAzeRcm.js +49 -0
- package/dist/{process-output-BZg56CCb.js → process-output-DxSCHBWp.js} +4 -4
- package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
- package/dist/{railroadDiagram-O6MQD6OU-8vFLVTPu.js → railroadDiagram-O6MQD6OU-BLcfaASl.js} +2 -2
- package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
- package/dist/{requirementDiagram-BXWQKSXE-a_MxGvxe.js → requirementDiagram-BXWQKSXE-DtBOf52-.js} +1 -1
- package/dist/{reveal-component-nqUW7zgn.js → reveal-component-5M3a2rpj.js} +734 -711
- package/dist/{select-BButFi-d.js → select-2KOiReNY.js} +536 -457
- package/dist/{sequenceDiagram-WJ2MYXX4-CXFMnxGh.js → sequenceDiagram-WJ2MYXX4-DIwF19_n.js} +1 -1
- package/dist/{spec-BwarBpUJ.js → spec-DZDnnlgT.js} +41 -11
- package/dist/{stateDiagram-D77RDMKH-DQeA_CwI.js → stateDiagram-D77RDMKH-sWk1AdYL.js} +2 -2
- package/dist/{stateDiagram-v2-MP3YSRHH-BxxQucm1.js → stateDiagram-v2-MP3YSRHH-CiNo-OI7.js} +2 -2
- package/dist/style.css +1 -1
- package/dist/{swimlanes-42K2YHIH-Do6L-cYg.js → swimlanes-42K2YHIH-Bj3hiOY5.js} +1 -1
- package/dist/{swimlanesDiagram-VR7AAH4N-DUzPvSVU.js → swimlanesDiagram-VR7AAH4N-BzEo2r_N.js} +2 -2
- package/dist/{swiper-component-DKntyn8p.js → swiper-component-BkCm-kLg.js} +2 -2
- package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
- package/dist/{timeline-definition-24CTP7MA-D3c46WM5.js → timeline-definition-24CTP7MA-CrPQk42H.js} +1 -1
- package/dist/{toDate-UtiwmnH2.js → toDate-Cu2pIZcC.js} +37 -36
- package/dist/{toast-B9UMEi8u.js → toast-B5t9o6s_.js} +219 -201
- package/dist/{tooltip-BO1REbqs.js → tooltip-s4zZxwKe.js} +325 -301
- package/dist/{use-restore-focus-BEM7sFWl.js → use-restore-focus-BgDTd2tc.js} +1482 -1433
- package/dist/{useAsyncData-Deh5kszT.js → useAsyncData-DEhLdRIu.js} +15 -14
- package/dist/{useDateFormatter-D8XruNGG.js → useDateFormatter-D3Tkfzej.js} +2 -2
- package/dist/{useDeepCompareMemoize-DXwzRINI.js → useDeepCompareMemoize-RjZGVhBV.js} +1 -1
- package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-CSdWn6NP.js} +1 -1
- package/dist/{useLifecycle-j7D9uD1o.js → useLifecycle-DtcB2s1z.js} +28 -25
- package/dist/{useTheme-CRnjkku0.js → useTheme-CoEtr4mK.js} +301 -297
- package/dist/{vega-component-BeF9dfyO.js → vega-component-BhjTrP9-.js} +148 -148
- package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
- package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
- package/dist/{vennDiagram-4TSXK5OY-BSZD8JcZ.js → vennDiagram-4TSXK5OY-BaYMoIpj.js} +1 -1
- package/dist/{wardleyDiagram-VM6X3IG4-48cUWkfA.js → wardleyDiagram-VM6X3IG4-DjHeuj8p.js} +1 -1
- package/dist/{xychartDiagram-S5SC5T6Z-CswYgYeu.js → xychartDiagram-S5SC5T6Z-Bc_hcuXS.js} +2 -2
- package/package.json +16 -16
- package/src/__mocks__/document.tsx +54 -0
- package/src/__mocks__/notebook.ts +2 -1
- package/src/__mocks__/requests.ts +0 -3
- package/src/__tests__/module-entry.test.ts +44 -0
- package/src/__tests__/mount.test.ts +1 -21
- package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
- package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
- package/src/components/ai/ai-model-dropdown.tsx +11 -6
- package/src/components/app-config/ai-config.tsx +7 -2
- package/src/components/app-config/user-config-form.tsx +35 -0
- package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
- package/src/components/chat/__tests__/notebook-access-selector.test.tsx +64 -0
- package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
- package/src/components/chat/acp/__tests__/state.test.ts +2 -2
- package/src/components/chat/acp/blocks.tsx +3 -2
- package/src/components/chat/acp/session-tabs.tsx +8 -3
- package/src/components/chat/chat-components.tsx +2 -0
- package/src/components/chat/chat-panel.tsx +20 -101
- package/src/components/chat/chat-utils.ts +4 -8
- package/src/components/chat/notebook-access-selector.stories.tsx +22 -0
- package/src/components/chat/notebook-access-selector.tsx +100 -0
- package/src/components/data-table/TableBottomBar.tsx +4 -1
- package/src/components/data-table/TableTopBar.tsx +2 -0
- package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
- package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
- package/src/components/data-table/__tests__/columns.test.tsx +105 -0
- package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
- package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
- package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
- package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
- package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
- package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
- package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
- package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
- package/src/components/data-table/cell-selection/feature.ts +2 -2
- package/src/components/data-table/cell-selection/types.ts +2 -2
- package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
- package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
- package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
- package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
- package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
- package/src/components/data-table/charts/charts.tsx +40 -30
- package/src/components/data-table/charts/storage.ts +16 -5
- package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
- package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
- package/src/components/data-table/column-summary/column-summary.tsx +2 -4
- package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
- package/src/components/data-table/columns.tsx +45 -46
- package/src/components/data-table/context-menu.tsx +36 -20
- package/src/components/data-table/data-table.tsx +17 -1
- package/src/components/data-table/date-popover.tsx +6 -4
- package/src/components/data-table/export-actions.tsx +76 -48
- package/src/components/data-table/pagination.tsx +44 -33
- package/src/components/data-table/range-focus/atoms.ts +4 -6
- package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
- package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
- package/src/components/data-table/renderers.tsx +16 -7
- package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
- package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
- package/src/components/data-table/search-highlight.tsx +79 -0
- package/src/components/data-table/types.ts +1 -0
- package/src/components/data-table/url-detector.tsx +49 -9
- package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
- package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
- package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
- package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
- package/src/components/datasources/__tests__/utils.test.ts +57 -6
- package/src/components/datasources/column-preview.tsx +17 -12
- package/src/components/datasources/datasources.tsx +55 -42
- package/src/components/datasources/utils.ts +32 -23
- package/src/components/debugger/debugger-code.tsx +5 -0
- package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
- package/src/components/dependency-graph/elements.ts +24 -21
- package/src/components/dependency-graph/minimap-content.tsx +2 -6
- package/src/components/dependency-graph/panels.tsx +4 -2
- package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
- package/src/components/editor/SortableCell.tsx +1 -3
- package/src/components/editor/__tests__/Output.test.tsx +50 -32
- package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
- package/src/components/editor/__tests__/data-attributes.test.tsx +28 -13
- package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
- package/src/components/editor/__tests__/output-persistence.test.tsx +6 -1
- package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
- package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
- package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +34 -8
- package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +2 -6
- package/src/components/editor/actions/name-cell-input.tsx +4 -3
- package/src/components/editor/actions/pair-with-agent-commands.ts +2 -2
- package/src/components/editor/actions/pair-with-agent-modal.tsx +19 -16
- package/src/components/editor/actions/useCellActionButton.tsx +20 -26
- package/src/components/editor/actions/useNotebookActions.tsx +14 -19
- package/src/components/editor/actions/useRestartKernel.tsx +1 -1
- package/src/components/editor/actions/useSetCodeVisibility.ts +10 -11
- package/src/components/editor/actions/useSetOutputsExpanded.ts +13 -18
- package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
- package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
- package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
- package/src/components/editor/alerts/connection-notice.tsx +1 -1
- package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
- package/src/components/editor/app-container.tsx +1 -1
- package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
- package/src/components/editor/cell/CreateCellButton.tsx +15 -21
- package/src/components/editor/cell/DeleteButton.tsx +3 -2
- package/src/components/editor/cell/RunButton.tsx +2 -2
- package/src/components/editor/cell/StagedAICell.tsx +10 -2
- package/src/components/editor/cell/StopButton.tsx +2 -2
- package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +3 -1
- package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
- package/src/components/editor/cell/cell-actions.tsx +12 -1
- package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
- package/src/components/editor/cell/code/cell-editor.tsx +22 -36
- package/src/components/editor/cell/code/language-toggle.tsx +8 -7
- package/src/components/editor/cell/collapse.tsx +5 -3
- package/src/components/editor/cell/useAddCell.ts +4 -8
- package/src/components/editor/cell/useDeleteCell.tsx +10 -36
- package/src/components/editor/cell/useRunCells.ts +2 -5
- package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
- package/src/components/editor/cell/useSplitCell.tsx +2 -1
- package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
- package/src/components/editor/chrome/components/feedback-button.tsx +1 -1
- package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
- package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
- package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
- package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
- package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
- package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
- package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
- package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
- package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
- package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
- package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
- package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +9 -18
- package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
- package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +61 -2
- package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
- package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
- package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
- package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
- package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
- package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
- package/src/components/editor/chrome/wrapper/footer.tsx +2 -0
- package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
- package/src/components/editor/chrome/wrapper/storage.ts +1 -1
- package/src/components/editor/columns/sortable-column.tsx +1 -1
- package/src/components/editor/connections/components.tsx +5 -10
- package/src/components/editor/controls/Controls.tsx +3 -3
- package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
- package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
- package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
- package/src/components/editor/controls/command-palette.tsx +22 -10
- package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
- package/src/components/editor/documentation.css +34 -11
- package/src/components/editor/dynamic-favicon.tsx +21 -40
- package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +9 -23
- package/src/components/editor/errors/auto-fix.tsx +4 -8
- package/src/components/editor/errors/send-error-report-button.tsx +3 -6
- package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
- package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
- package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
- package/src/components/editor/file-tree/file-explorer.tsx +48 -15
- package/src/components/editor/file-tree/file-name-input.tsx +1 -0
- package/src/components/editor/file-tree/file-viewer.tsx +4 -2
- package/src/components/editor/file-tree/use-file-search.ts +35 -21
- package/src/components/editor/header/__tests__/status.test.tsx +5 -3
- package/src/components/editor/header/app-header.tsx +1 -1
- package/src/components/editor/header/status.tsx +40 -20
- package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
- package/src/components/editor/links/cell-link.tsx +16 -6
- package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
- package/src/components/editor/navigation/__tests__/navigation.test.ts +73 -64
- package/src/components/editor/navigation/clipboard.ts +12 -11
- package/src/components/editor/navigation/focus-utils.ts +4 -4
- package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +11 -20
- package/src/components/editor/navigation/navigation.ts +20 -28
- package/src/components/editor/notebook-banner.tsx +1 -0
- package/src/components/editor/notebook-cell.tsx +19 -13
- package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
- package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
- package/src/components/editor/output/Outputs.css +8 -0
- package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
- package/src/components/editor/output/ansi-reduce.ts +33 -20
- package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
- package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
- package/src/components/editor/package-alert.tsx +14 -7
- package/src/components/editor/renderers/cell-array.tsx +10 -17
- package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
- package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
- package/src/components/home/components.tsx +17 -31
- package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
- package/src/components/icons/large-spinner.tsx +3 -14
- package/src/components/icons/spinner.tsx +1 -1
- package/src/components/markdown/markdown-renderer.tsx +4 -8
- package/src/components/mcp/mcp-status-indicator.tsx +1 -1
- package/src/components/pages/__tests__/home-page.test.tsx +116 -0
- package/src/components/pages/edit-page.tsx +32 -17
- package/src/components/pages/home-page.tsx +164 -91
- package/src/components/scratchpad/scratchpad.test.tsx +97 -0
- package/src/components/scratchpad/scratchpad.tsx +20 -10
- package/src/components/slides/__tests__/minimap-actions.test.tsx +9 -25
- package/src/components/slides/__tests__/slide-scroll-hint.test.tsx +37 -0
- package/src/components/slides/minimap.tsx +7 -3
- package/src/components/slides/reveal-component.tsx +57 -25
- package/src/components/slides/reveal-slides.css +5 -4
- package/src/components/slides/slide-cell-view.tsx +4 -4
- package/src/components/slides/slide-form.tsx +1 -0
- package/src/components/slides/slide-scroll-hint.tsx +3 -1
- package/src/components/sort/SortableCellsProvider.tsx +2 -1
- package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
- package/src/components/tracing/tracing.tsx +45 -41
- package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
- package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
- package/src/components/ui/alert-dialog.tsx +1 -1
- package/src/components/ui/draggable-popover.tsx +24 -2
- package/src/components/ui/range-slider.tsx +22 -9
- package/src/core/ai/__tests__/staged-cells.test.ts +26 -25
- package/src/core/ai/context/providers/datasource.ts +6 -2
- package/src/core/ai/staged-cells.ts +15 -18
- package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +6 -22
- package/src/core/ai/tools/__tests__/registry.test.ts +17 -0
- package/src/core/ai/tools/base.ts +18 -15
- package/src/core/ai/tools/edit-notebook-tool.ts +75 -71
- package/src/core/ai/tools/registry.ts +13 -10
- package/src/core/ai/tools/sample-tool.ts +6 -7
- package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -0
- package/src/core/cells/__tests__/actions.test.ts +474 -34
- package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
- package/src/core/cells/__tests__/cell.test.ts +39 -0
- package/src/core/cells/__tests__/cells.test.ts +260 -1165
- package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
- package/src/core/cells/__tests__/document.test.ts +414 -0
- package/src/core/cells/__tests__/flush-document.test.ts +40 -0
- package/src/core/cells/__tests__/notebook-outline.test.ts +114 -0
- package/src/core/cells/__tests__/outputs.test.tsx +71 -53
- package/src/core/cells/__tests__/session.test.ts +117 -36
- package/src/core/cells/__tests__/with-document-save.test.ts +67 -0
- package/src/core/cells/actions.ts +391 -3
- package/src/core/cells/add-missing-import.ts +46 -42
- package/src/core/cells/cell.ts +14 -3
- package/src/core/cells/cells.ts +99 -599
- package/src/core/cells/columns.ts +38 -0
- package/src/core/cells/document.ts +392 -0
- package/src/core/cells/outputs.ts +12 -28
- package/src/core/cells/pending-delete-service.ts +2 -6
- package/src/core/cells/session.ts +1 -1
- package/src/core/cells/types.ts +22 -1
- package/src/core/codemirror/__tests__/setup.test.ts +29 -29
- package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
- package/src/core/codemirror/ai/resources.ts +1 -1
- package/src/core/codemirror/cells/__tests__/extensions.test.ts +42 -1
- package/src/core/codemirror/cells/extensions.ts +27 -9
- package/src/core/codemirror/cells/state.ts +1 -1
- package/src/core/codemirror/format.ts +6 -5
- package/src/core/codemirror/keymaps/vim.ts +4 -8
- package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
- package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
- package/src/core/codemirror/language/cell-code.ts +114 -0
- package/src/core/codemirror/language/extension.ts +52 -35
- package/src/core/codemirror/language/languages/markdown.ts +8 -6
- package/src/core/codemirror/language/languages/python.ts +16 -10
- package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
- package/src/core/codemirror/language/panel/panel.tsx +1 -0
- package/src/core/codemirror/language/types.ts +7 -7
- package/src/core/codemirror/language/utils.ts +9 -39
- package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
- package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
- package/src/core/codemirror/readonly/extension.ts +2 -2
- package/src/core/codemirror/rtc/extension.ts +24 -13
- package/src/core/codemirror/rtc/loro/awareness.ts +34 -19
- package/src/core/codemirror/rtc/loro/sync.ts +1 -0
- package/src/core/codemirror/utils.ts +1 -0
- package/src/core/config/__tests__/config-schema.test.ts +12 -5
- package/src/core/config/config-schema.ts +7 -1
- package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
- package/src/core/datasets/data-source-discovery.ts +4 -6
- package/src/core/edit-app.tsx +19 -23
- package/src/core/errors/error-entries.ts +3 -0
- package/src/core/export/hooks.ts +1 -1
- package/src/core/export/layout.ts +2 -2
- package/src/core/hotkeys/actions.ts +2 -4
- package/src/core/islands/__tests__/bootstrap.test.ts +73 -43
- package/src/core/islands/__tests__/bridge.test.ts +72 -25
- package/src/core/islands/bootstrap.ts +46 -53
- package/src/core/islands/bridge.ts +74 -23
- package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
- package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
- package/src/core/islands/worker/controller.ts +4 -10
- package/src/core/islands/worker/worker.tsx +33 -33
- package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
- package/src/core/kernel/__tests__/handlers.test.ts +7 -153
- package/src/core/kernel/handlers.ts +18 -59
- package/src/core/network/__tests__/connection.test.ts +1 -1
- package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
- package/src/core/network/connection-notice.ts +1 -1
- package/src/core/network/connection.ts +1 -1
- package/src/core/network/requests-lazy.ts +1 -4
- package/src/core/network/requests-network.ts +0 -25
- package/src/core/network/requests-static.ts +0 -3
- package/src/core/network/requests-toasting.tsx +0 -3
- package/src/core/network/types.ts +0 -12
- package/src/core/network/useConnectionNotice.ts +8 -2
- package/src/core/notebook/attachments.ts +38 -0
- package/src/core/notebook/document/__tests__/codemirror.test.ts +552 -0
- package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
- package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
- package/src/core/notebook/document/__tests__/hold.test.ts +88 -0
- package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
- package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
- package/src/core/notebook/document/__tests__/model.test.ts +247 -0
- package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
- package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
- package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
- package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
- package/src/core/notebook/document/__tests__/state.test.ts +459 -0
- package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
- package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
- package/src/core/notebook/document/__tests__/test-host.ts +247 -0
- package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
- package/src/core/notebook/document/__tests__/user.ts +61 -0
- package/src/core/notebook/document/codemirror.ts +216 -0
- package/src/core/notebook/document/draft.ts +502 -0
- package/src/core/notebook/document/hooks.tsx +118 -0
- package/src/core/notebook/document/index.ts +24 -0
- package/src/core/notebook/document/model.ts +105 -0
- package/src/core/notebook/document/select.ts +41 -0
- package/src/core/notebook/document/state.ts +378 -0
- package/src/core/notebook/document/sync.ts +418 -0
- package/src/core/notebook/index.ts +18 -0
- package/src/core/notebook/replica/AGENTS.md +11 -0
- package/src/core/notebook/replica/__tests__/attachments.test.ts +56 -0
- package/src/core/notebook/replica/__tests__/document.test.ts +126 -0
- package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
- package/src/core/notebook/replica/__tests__/replica.test.ts +550 -0
- package/src/core/notebook/replica/backend.ts +19 -0
- package/src/core/notebook/replica/connection.ts +42 -0
- package/src/core/notebook/replica/index.ts +13 -0
- package/src/core/notebook/replica/replica.ts +286 -0
- package/src/core/notebook/replica/resources/attachments.ts +22 -0
- package/src/core/notebook/replica/resources/document.ts +101 -0
- package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
- package/src/core/packages/useSandboxController.ts +1 -1
- package/src/core/run-app.tsx +48 -51
- package/src/core/runtime/__tests__/adapter.test.ts +1 -1
- package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
- package/src/core/runtime/__tests__/runtime.test.ts +32 -58
- package/src/core/runtime/adapter.ts +1 -1
- package/src/core/runtime/config.ts +2 -2
- package/src/core/runtime/runtime.ts +38 -42
- package/src/core/runtime/types.ts +0 -2
- package/src/core/saving/filename.ts +1 -1
- package/src/core/saving/save-component.tsx +10 -4
- package/src/core/saving/useAutoSave.ts +1 -1
- package/src/core/session/__tests__/backend.test.ts +235 -0
- package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +183 -0
- package/src/core/session/__tests__/useNotebookSession.hook.test.tsx +663 -0
- package/src/core/session/active-notebook.ts +11 -0
- package/src/core/session/backend.ts +125 -0
- package/src/core/{websocket → session}/types.ts +2 -2
- package/src/core/session/useNotebookReplica.ts +72 -0
- package/src/core/{websocket → session}/useNotebookSession.tsx +74 -68
- package/src/core/state/__tests__/jotai.test.tsx +108 -0
- package/src/core/state/jotai.ts +1 -1
- package/src/core/static/__tests__/files.test.ts +38 -10
- package/src/core/static/files.ts +2 -0
- package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
- package/src/core/transports/__tests__/server.test.ts +408 -0
- package/src/core/transports/__tests__/static.test.ts +175 -0
- package/src/core/transports/__tests__/wasm.test.ts +166 -0
- package/src/core/{websocket → transports}/connection-state.ts +9 -16
- package/src/core/transports/server.ts +152 -0
- package/src/core/transports/static.ts +84 -0
- package/src/core/transports/wasm.ts +97 -0
- package/src/core/wasm/__tests__/bridge.test.ts +213 -50
- package/src/core/wasm/__tests__/worker.test.ts +187 -0
- package/src/core/wasm/bridge.ts +75 -69
- package/src/core/wasm/rpc.ts +15 -3
- package/src/core/wasm/store.ts +3 -3
- package/src/core/wasm/worker/bootstrap.ts +55 -51
- package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
- package/src/core/wasm/worker/save-worker.ts +9 -10
- package/src/core/wasm/worker/types.ts +29 -18
- package/src/core/wasm/worker/worker.ts +88 -44
- package/src/css/globals.css +66 -22
- package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
- package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
- package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
- package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
- package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
- package/src/hooks/__tests__/useScript.test.tsx +77 -0
- package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
- package/src/hooks/useAsyncData.ts +3 -3
- package/src/hooks/useDataSourceDiscovery.ts +6 -4
- package/src/hooks/useDebounce.ts +6 -10
- package/src/hooks/useDeepCompareMemoize.ts +1 -1
- package/src/hooks/useDelayElapsed.ts +14 -10
- package/src/hooks/useElapsedTime.ts +3 -3
- package/src/hooks/useEventListener.ts +1 -2
- package/src/hooks/useInputHistory.ts +4 -7
- package/src/hooks/useInternalStateWithSync.ts +6 -6
- package/src/hooks/useResizeObserver.ts +4 -7
- package/src/hooks/useScript.ts +67 -70
- package/src/hooks/useStateForKey.ts +93 -0
- package/src/mount.tsx +1 -5
- package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
- package/src/plugins/core/registerReactComponent.tsx +1 -1
- package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
- package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
- package/src/plugins/impl/DataTablePlugin.tsx +69 -37
- package/src/plugins/impl/DropdownPlugin.tsx +2 -6
- package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
- package/src/plugins/impl/FormPlugin.tsx +1 -1
- package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
- package/src/plugins/impl/RefreshPlugin.tsx +1 -0
- package/src/plugins/impl/SliderPlugin.tsx +6 -8
- package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
- package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
- package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
- package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
- package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
- package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
- package/src/plugins/impl/anywidget/model.ts +3 -1
- package/src/plugins/impl/anywidget/registry.ts +18 -14
- package/src/plugins/impl/anywidget/runtime.ts +21 -9
- package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
- package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
- package/src/plugins/impl/chat/chat-ui.tsx +14 -11
- package/src/plugins/impl/code/any-language-editor.tsx +2 -2
- package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
- package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
- package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
- package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
- package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
- package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
- package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
- package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
- package/src/plugins/impl/data-frames/panel.tsx +43 -16
- package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
- package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
- package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
- package/src/plugins/impl/panel/utils.ts +17 -0
- package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
- package/src/plugins/impl/plotly/selection.ts +25 -19
- package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
- package/src/plugins/impl/vega/make-selectable.ts +4 -7
- package/src/plugins/layout/RoutesPlugin.tsx +1 -1
- package/src/stories/__tests__/editor-story.test.tsx +33 -0
- package/src/stories/button.stories.tsx +1 -0
- package/src/stories/cell.stories.tsx +1 -1
- package/src/stories/data-table.stories.tsx +31 -0
- package/src/stories/editor.stories.tsx +24 -17
- package/src/stories/form-wrapper.stories.tsx +2 -2
- package/src/stories/layout/vertical/one-column.stories.tsx +1 -1
- package/src/stories/runtime-story.tsx +23 -16
- package/src/stories/select.stories.tsx +2 -2
- package/src/stories/switchable-multi-select.stories.tsx +1 -1
- package/src/utils/__tests__/arrays.test.ts +35 -0
- package/src/utils/__tests__/batch-requests.test.ts +3 -3
- package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
- package/src/utils/__tests__/numbers.test.ts +105 -1
- package/src/utils/__tests__/objects.test.ts +21 -0
- package/src/utils/arrays.ts +11 -2
- package/src/utils/blob.ts +1 -1
- package/src/utils/createReducer.ts +53 -26
- package/src/utils/maps.ts +0 -3
- package/src/utils/numbers.ts +79 -18
- package/src/utils/objects.ts +4 -9
- package/src/utils/once.ts +19 -19
- package/src/utils/tracer.ts +9 -5
- package/src/utils/zod-utils.ts +1 -3
- package/dist/any-language-editor-DZ_YTVus.js +0 -76
- package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
- package/dist/assets/worker-HPY9fbhz.js +0 -114
- package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
- package/dist/precisionRound-DMIVQrt2.js +0 -97
- package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +0 -656
- package/src/core/ai/tools/run-cells-tool.ts +0 -303
- package/src/core/attachments/state.ts +0 -21
- package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
- package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
- package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
- package/src/core/cells/document-changes.ts +0 -944
- package/src/core/host/AGENTS.md +0 -6
- package/src/core/host/__tests__/events.test.ts +0 -46
- package/src/core/host/__tests__/notebook-client.test.ts +0 -191
- package/src/core/host/__tests__/protocol.test.ts +0 -214
- package/src/core/host/connection.ts +0 -81
- package/src/core/host/events.ts +0 -60
- package/src/core/host/notebook-client.ts +0 -226
- package/src/core/host/protocol.ts +0 -1943
- package/src/core/state/__tests__/jotai.test.ts +0 -25
- package/src/core/wasm/worker/message-buffer.ts +0 -33
- package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
- package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -140
- package/src/core/websocket/__tests__/useNotebookSession.hook.test.tsx +0 -883
- package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
- package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
- package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
- package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
- package/src/core/websocket/transports/basic.ts +0 -102
- package/src/core/websocket/transports/create.ts +0 -31
- package/src/core/websocket/transports/sse.ts +0 -308
- package/src/core/websocket/transports/transport.ts +0 -50
- package/src/core/websocket/transports/ws.ts +0 -98
- package/src/core/websocket/useNotebookClient.ts +0 -80
- /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
- /package/src/core/{websocket → session}/connection-utils.ts +0 -0
package/src/core/wasm/bridge.ts
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
/* Copyright 2026 Marimo. All rights reserved. */
|
|
2
2
|
/* oxlint-disable typescript/no-explicit-any */
|
|
3
3
|
|
|
4
|
+
import { createRPCRequestHandler } from "rpc-anywhere";
|
|
4
5
|
import { toast } from "@/components/ui/use-toast";
|
|
5
6
|
import { userConfigAtom } from "@/core/config/config";
|
|
6
7
|
import { serializeBlob } from "@/utils/blob";
|
|
@@ -8,9 +9,12 @@ import { Deferred } from "@/utils/Deferred";
|
|
|
8
9
|
import { throwNotImplemented } from "@/utils/functions";
|
|
9
10
|
import { Logger } from "@/utils/Logger";
|
|
10
11
|
import { reloadSafe } from "@/utils/reload-safe";
|
|
12
|
+
import { once } from "@/utils/once";
|
|
11
13
|
import { generateUUID } from "@/utils/uuid";
|
|
12
14
|
import { createModuleWorker } from "@/utils/worker";
|
|
13
15
|
import { notebookIsRunningAtom } from "../cells/cells";
|
|
16
|
+
import { flushDocument } from "../cells/document";
|
|
17
|
+
import { createWasmNotebookBackend } from "../transports/wasm";
|
|
14
18
|
import type { ClientCommand } from "../kernel/messages";
|
|
15
19
|
import { getInitialAppMode } from "../mode";
|
|
16
20
|
import { API } from "../network/api";
|
|
@@ -38,10 +42,9 @@ import type {
|
|
|
38
42
|
} from "../network/types";
|
|
39
43
|
import { filenameAtom } from "../saving/file-state";
|
|
40
44
|
import { store } from "../state/jotai";
|
|
41
|
-
import {
|
|
42
|
-
import type { IConnectionTransport } from "../websocket/transports/transport";
|
|
45
|
+
import { activeNotebookReplicaAtom } from "../session/active-notebook";
|
|
43
46
|
import { PyodideRouter } from "./router";
|
|
44
|
-
import { getWorkerRPC } from "./rpc";
|
|
47
|
+
import { getWorkerRPC, type PersistNotebookRequest } from "./rpc";
|
|
45
48
|
import { getWasmRuntimeConfig } from "./runtime-config";
|
|
46
49
|
import { createShareableLink } from "./share";
|
|
47
50
|
import { wasmInitStateAtom } from "./state";
|
|
@@ -68,11 +71,8 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
68
71
|
|
|
69
72
|
private rpc!: ReturnType<typeof getWorkerRPC<WorkerSchema>>;
|
|
70
73
|
private saveRpc: SaveWorker | undefined;
|
|
71
|
-
private
|
|
74
|
+
private notebookConsumers = new Map<string, (event: unknown) => void>();
|
|
72
75
|
private interruptBuffer?: Uint8Array;
|
|
73
|
-
private messageConsumer:
|
|
74
|
-
| ((message: MessageEvent<string>) => void)
|
|
75
|
-
| undefined;
|
|
76
76
|
|
|
77
77
|
public initialized = new Deferred<void>();
|
|
78
78
|
|
|
@@ -82,7 +82,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
82
82
|
return {
|
|
83
83
|
readFile: throwNotImplemented,
|
|
84
84
|
readNotebook: async () => (await notebookFileStore.readFile()) ?? "",
|
|
85
|
-
|
|
85
|
+
writeNotebook: throwNotImplemented,
|
|
86
86
|
};
|
|
87
87
|
}
|
|
88
88
|
|
|
@@ -97,7 +97,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
97
97
|
},
|
|
98
98
|
);
|
|
99
99
|
|
|
100
|
-
const rpc = getWorkerRPC<SaveWorkerSchema>(saveWorker);
|
|
100
|
+
const rpc = getWorkerRPC<SaveWorkerSchema>(saveWorker, undefined, Infinity);
|
|
101
101
|
rpc.send.bootstrap(runtimeConfig);
|
|
102
102
|
return rpc.proxy.request;
|
|
103
103
|
}
|
|
@@ -116,9 +116,28 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
116
116
|
});
|
|
117
117
|
|
|
118
118
|
// Create the RPC
|
|
119
|
-
this.rpc = getWorkerRPC<WorkerSchema>(
|
|
119
|
+
this.rpc = getWorkerRPC<WorkerSchema>(
|
|
120
|
+
worker,
|
|
121
|
+
createRPCRequestHandler({
|
|
122
|
+
persistNotebook: async (snapshot: PersistNotebookRequest) => {
|
|
123
|
+
if (!this.saveRpc) {
|
|
124
|
+
throw new Error("Notebook storage is not initialized");
|
|
125
|
+
}
|
|
126
|
+
await this.saveRpc.writeNotebook(snapshot);
|
|
127
|
+
notebookFileStore.saveFile(snapshot.contents);
|
|
128
|
+
fallbackFileStore.saveFile(snapshot.contents);
|
|
129
|
+
return null;
|
|
130
|
+
},
|
|
131
|
+
}),
|
|
132
|
+
// Synchronous Python execution can block worker requests indefinitely.
|
|
133
|
+
Infinity,
|
|
134
|
+
);
|
|
120
135
|
this.rpc.send.bootstrap(runtimeConfig);
|
|
121
136
|
|
|
137
|
+
this.rpc.addMessageListener("notebookEvent", ({ id, event }) => {
|
|
138
|
+
this.notebookConsumers.get(id)?.(JSON.parse(event));
|
|
139
|
+
});
|
|
140
|
+
|
|
122
141
|
// Listeners
|
|
123
142
|
this.rpc.addMessageListener("ready", () => {
|
|
124
143
|
this.startSession();
|
|
@@ -154,9 +173,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
154
173
|
store.set(wasmInitStateAtom, { kind: "error", message: error });
|
|
155
174
|
this.initialized.reject(new Error(error));
|
|
156
175
|
});
|
|
157
|
-
this.rpc.addMessageListener("kernelMessage", ({ message }) => {
|
|
158
|
-
this.messageConsumer?.(new MessageEvent("message", { data: message }));
|
|
159
|
-
});
|
|
160
176
|
}
|
|
161
177
|
|
|
162
178
|
private async startSession() {
|
|
@@ -208,11 +224,31 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
208
224
|
}
|
|
209
225
|
}
|
|
210
226
|
|
|
211
|
-
public
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
227
|
+
public async createNotebookBackend() {
|
|
228
|
+
await this.initialized.promise;
|
|
229
|
+
return createWasmNotebookBackend({
|
|
230
|
+
subscribe: async (cursor, emit, signal) => {
|
|
231
|
+
signal.throwIfAborted();
|
|
232
|
+
const id = generateUUID();
|
|
233
|
+
this.notebookConsumers.set(id, emit);
|
|
234
|
+
const close = once(() => {
|
|
235
|
+
signal.removeEventListener("abort", close);
|
|
236
|
+
this.notebookConsumers.delete(id);
|
|
237
|
+
void this.rpc.proxy.request
|
|
238
|
+
.notebookUnsubscribe(id)
|
|
239
|
+
.catch(Logger.error);
|
|
240
|
+
});
|
|
241
|
+
signal.addEventListener("abort", close, { once: true });
|
|
242
|
+
try {
|
|
243
|
+
await this.rpc.proxy.request.notebookSubscribe({ id, cursor });
|
|
244
|
+
return close;
|
|
245
|
+
} catch (error) {
|
|
246
|
+
close();
|
|
247
|
+
throw error;
|
|
248
|
+
}
|
|
249
|
+
},
|
|
250
|
+
call: (call) => this.rpc.proxy.request.notebookCall(call),
|
|
251
|
+
});
|
|
216
252
|
}
|
|
217
253
|
|
|
218
254
|
public sendRename: EditRequests["sendRename"] = async ({ filename }) => {
|
|
@@ -230,30 +266,23 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
230
266
|
return null;
|
|
231
267
|
};
|
|
232
268
|
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
269
|
+
private async saveHostNotebook() {
|
|
270
|
+
const replica = store.get(activeNotebookReplicaAtom);
|
|
271
|
+
const notebook = replica && store.get(replica.notebookAtom);
|
|
272
|
+
const backend = replica && store.get(replica.backendAtom);
|
|
273
|
+
if (!notebook || !backend) {
|
|
274
|
+
throw new Error("The notebook is not connected");
|
|
237
275
|
}
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
this.pendingSessionSave = this.pendingSessionSave
|
|
242
|
-
.catch(() => undefined)
|
|
243
|
-
.then(async () => {
|
|
244
|
-
await durableSave;
|
|
245
|
-
await this.rpc.proxy.request.saveNotebook(request);
|
|
246
|
-
});
|
|
247
|
-
void this.pendingSessionSave.catch((error) => {
|
|
248
|
-
Logger.error(error);
|
|
249
|
-
});
|
|
250
|
-
|
|
251
|
-
await durableSave;
|
|
252
|
-
const code = await this.readCode();
|
|
253
|
-
if (code.contents) {
|
|
254
|
-
notebookFileStore.saveFile(code.contents);
|
|
255
|
-
fallbackFileStore.saveFile(code.contents);
|
|
276
|
+
const result = await backend.notebook(notebook.id).save();
|
|
277
|
+
if (!result.ok) {
|
|
278
|
+
throw new Error(result.problem.detail ?? result.problem.title);
|
|
256
279
|
}
|
|
280
|
+
}
|
|
281
|
+
|
|
282
|
+
public sendSave: EditRequests["sendSave"] = async ({ layout }) => {
|
|
283
|
+
await flushDocument();
|
|
284
|
+
await this.rpc.proxy.request.setNotebookLayout(layout ?? null);
|
|
285
|
+
await this.saveHostNotebook();
|
|
257
286
|
return null;
|
|
258
287
|
};
|
|
259
288
|
|
|
@@ -316,14 +345,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
316
345
|
return response as FormatResponse;
|
|
317
346
|
};
|
|
318
347
|
|
|
319
|
-
public sendDeleteCell: EditRequests["sendDeleteCell"] = async (request) => {
|
|
320
|
-
await this.putControlRequest({
|
|
321
|
-
type: "delete-cell",
|
|
322
|
-
...request,
|
|
323
|
-
});
|
|
324
|
-
return null;
|
|
325
|
-
};
|
|
326
|
-
|
|
327
348
|
public sendInstallMissingPackages: EditRequests["sendInstallMissingPackages"] =
|
|
328
349
|
async (request) => {
|
|
329
350
|
this.putControlRequest({
|
|
@@ -370,14 +391,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
370
391
|
functionName: "save_app_config",
|
|
371
392
|
payload: request,
|
|
372
393
|
});
|
|
373
|
-
|
|
374
|
-
};
|
|
375
|
-
|
|
376
|
-
public saveCellConfig: EditRequests["saveCellConfig"] = async (request) => {
|
|
377
|
-
await this.putControlRequest({
|
|
378
|
-
type: "update-cell-config",
|
|
379
|
-
...request,
|
|
380
|
-
});
|
|
394
|
+
await this.saveHostNotebook();
|
|
381
395
|
return null;
|
|
382
396
|
};
|
|
383
397
|
|
|
@@ -409,7 +423,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
409
423
|
return response as Snippets;
|
|
410
424
|
};
|
|
411
425
|
|
|
412
|
-
public openFile: EditRequests["openFile"] = async (
|
|
426
|
+
public openFile: EditRequests["openFile"] = async () => {
|
|
413
427
|
const url = createShareableLink({
|
|
414
428
|
code: null,
|
|
415
429
|
baseUrl: window.location.origin,
|
|
@@ -453,7 +467,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
453
467
|
return null;
|
|
454
468
|
};
|
|
455
469
|
|
|
456
|
-
public sendInstantiate: RunRequests["sendInstantiate"] = async (
|
|
470
|
+
public sendInstantiate: RunRequests["sendInstantiate"] = async () => {
|
|
457
471
|
return null;
|
|
458
472
|
};
|
|
459
473
|
|
|
@@ -535,7 +549,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
535
549
|
public exportAsHTML: EditRequests["exportAsHTML"] = async (
|
|
536
550
|
request: ExportAsHTMLRequest,
|
|
537
551
|
) => {
|
|
538
|
-
await
|
|
552
|
+
await flushDocument();
|
|
539
553
|
const response = await this.rpc.proxy.request.bridge({
|
|
540
554
|
functionName: "export_html",
|
|
541
555
|
payload: withDevAssetUrl(request),
|
|
@@ -546,7 +560,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
546
560
|
public exportAsMarkdown: EditRequests["exportAsMarkdown"] = async (
|
|
547
561
|
request: ExportAsMarkdownRequest,
|
|
548
562
|
) => {
|
|
549
|
-
await
|
|
563
|
+
await flushDocument();
|
|
550
564
|
const response = await this.rpc.proxy.request.bridge({
|
|
551
565
|
functionName: "export_markdown",
|
|
552
566
|
payload: request,
|
|
@@ -557,7 +571,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
557
571
|
public exportAsScript: EditRequests["exportAsScript"] = async (
|
|
558
572
|
request: ExportAsScriptRequest,
|
|
559
573
|
) => {
|
|
560
|
-
await
|
|
574
|
+
await flushDocument();
|
|
561
575
|
const response = await this.rpc.proxy.request.bridge({
|
|
562
576
|
functionName: "export_script",
|
|
563
577
|
payload: request,
|
|
@@ -628,8 +642,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
628
642
|
return null;
|
|
629
643
|
};
|
|
630
644
|
|
|
631
|
-
public sendDocumentTransaction = () => Promise.resolve(null);
|
|
632
|
-
|
|
633
645
|
public addPackage: EditRequests["addPackage"] = async (request) => {
|
|
634
646
|
return this.rpc.proxy.request.addPackage(request);
|
|
635
647
|
};
|
|
@@ -719,12 +731,6 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
719
731
|
}
|
|
720
732
|
}
|
|
721
733
|
|
|
722
|
-
export function createPyodideConnection(): IConnectionTransport {
|
|
723
|
-
return BasicTransport.withProducerCallback((callback) => {
|
|
724
|
-
PyodideBridge.INSTANCE.attachMessageConsumer(callback);
|
|
725
|
-
});
|
|
726
|
-
}
|
|
727
|
-
|
|
728
734
|
// Compose the worker name. The optional "::controller" suffix tells
|
|
729
735
|
// getController.ts that the host page provides a custom /wasm/controller.js
|
|
730
736
|
// and that the dynamic import should be attempted. Hosts opt in by setting
|
package/src/core/wasm/rpc.ts
CHANGED
|
@@ -9,19 +9,31 @@ import { Logger } from "@/utils/Logger";
|
|
|
9
9
|
import type { WasmRuntimeConfig } from "./runtime-config";
|
|
10
10
|
import { TRANSPORT_ID } from "./worker/constants";
|
|
11
11
|
|
|
12
|
+
export interface PersistNotebookRequest {
|
|
13
|
+
filename: string;
|
|
14
|
+
contents: string;
|
|
15
|
+
}
|
|
16
|
+
|
|
12
17
|
export type ParentSchema = RPCSchema<{
|
|
18
|
+
requests: {
|
|
19
|
+
persistNotebook: { params: PersistNotebookRequest; response: null };
|
|
20
|
+
};
|
|
13
21
|
messages: {
|
|
14
|
-
consumerReady: {};
|
|
15
22
|
bootstrap: WasmRuntimeConfig;
|
|
16
23
|
};
|
|
17
24
|
}>;
|
|
18
25
|
|
|
19
|
-
export function getWorkerRPC<WorkerSchema extends RPCSchema>(
|
|
26
|
+
export function getWorkerRPC<WorkerSchema extends RPCSchema>(
|
|
27
|
+
worker: Worker,
|
|
28
|
+
requestHandler?: RPCOptions<ParentSchema, WorkerSchema>["requestHandler"],
|
|
29
|
+
maxRequestTime = 20_000,
|
|
30
|
+
) {
|
|
20
31
|
return createRPC<ParentSchema, WorkerSchema>({
|
|
21
32
|
transport: createWorkerTransport(worker, {
|
|
22
33
|
transportId: TRANSPORT_ID,
|
|
23
34
|
}),
|
|
24
|
-
|
|
35
|
+
requestHandler,
|
|
36
|
+
maxRequestTime,
|
|
25
37
|
_debugHooks: {
|
|
26
38
|
onSend: (message) => {
|
|
27
39
|
Logger.debug("[rpc] Parent -> Worker", message);
|
package/src/core/wasm/store.ts
CHANGED
|
@@ -41,7 +41,7 @@ const urlFileStore: FileStore = {
|
|
|
41
41
|
};
|
|
42
42
|
|
|
43
43
|
export const domElementFileStore: FileStore = {
|
|
44
|
-
saveFile(
|
|
44
|
+
saveFile(_contents: string) {
|
|
45
45
|
// Do nothing
|
|
46
46
|
},
|
|
47
47
|
readFile() {
|
|
@@ -54,7 +54,7 @@ export const domElementFileStore: FileStore = {
|
|
|
54
54
|
};
|
|
55
55
|
|
|
56
56
|
const remoteDefaultFileStore: FileStore = {
|
|
57
|
-
saveFile(
|
|
57
|
+
saveFile(_contents: string) {
|
|
58
58
|
// Do nothing
|
|
59
59
|
},
|
|
60
60
|
readFile() {
|
|
@@ -79,7 +79,7 @@ export const mountConfigFileStore: FileStore = {
|
|
|
79
79
|
};
|
|
80
80
|
|
|
81
81
|
const emptyFileStore: FileStore = {
|
|
82
|
-
saveFile(
|
|
82
|
+
saveFile(_contents: string) {
|
|
83
83
|
// Do nothing
|
|
84
84
|
},
|
|
85
85
|
readFile() {
|
|
@@ -1,15 +1,13 @@
|
|
|
1
1
|
/* Copyright 2026 Marimo. All rights reserved. */
|
|
2
2
|
import { loadPyodide, type PyodideInterface } from "pyodide";
|
|
3
3
|
import type { PyCallable, PyProxy } from "pyodide/ffi";
|
|
4
|
-
import type { UserConfig } from "@/core/config/config-schema";
|
|
5
|
-
import type { NotificationPayload } from "@/core/kernel/messages";
|
|
6
|
-
import type { JsonString } from "@/utils/json/base64";
|
|
7
4
|
import { invariant } from "../../../utils/invariant";
|
|
5
|
+
import { once } from "../../../utils/once";
|
|
8
6
|
import { Logger } from "../../../utils/Logger";
|
|
9
7
|
import { WasmFileSystem } from "./fs";
|
|
10
8
|
import { getMarimoWheel } from "./getMarimoWheel";
|
|
11
9
|
import { t } from "./tracer";
|
|
12
|
-
import type { SerializedBridge, WasmController } from "./types";
|
|
10
|
+
import type { SerializedBridge, WasmController, WasmSession } from "./types";
|
|
13
11
|
import { shouldLoadDuckDBPackages } from "../utils";
|
|
14
12
|
|
|
15
13
|
const MAKE_SNAPSHOT = false;
|
|
@@ -45,7 +43,7 @@ export class DefaultWasmController implements WasmController {
|
|
|
45
43
|
protected pyodide: PyodideInterface | null = null;
|
|
46
44
|
private packageLoadQueue = Promise.resolve();
|
|
47
45
|
private sessionGeneration = 0;
|
|
48
|
-
private
|
|
46
|
+
private sessions = new Set<WasmSession>();
|
|
49
47
|
|
|
50
48
|
public get requirePyodide() {
|
|
51
49
|
invariant(this.pyodide, "Pyodide not loaded");
|
|
@@ -85,7 +83,7 @@ export class DefaultWasmController implements WasmController {
|
|
|
85
83
|
packages: [
|
|
86
84
|
"micropip",
|
|
87
85
|
"msgspec",
|
|
88
|
-
getMarimoWheel(
|
|
86
|
+
getMarimoWheel(),
|
|
89
87
|
"Markdown",
|
|
90
88
|
"pymdown-extensions",
|
|
91
89
|
"narwhals",
|
|
@@ -138,28 +136,15 @@ export class DefaultWasmController implements WasmController {
|
|
|
138
136
|
});
|
|
139
137
|
}
|
|
140
138
|
|
|
141
|
-
public async
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
filename: string | null;
|
|
145
|
-
onMessage: (message: JsonString<NotificationPayload>) => void;
|
|
146
|
-
userConfig: UserConfig;
|
|
147
|
-
}): Promise<SerializedBridge> {
|
|
139
|
+
public async prepareSession(
|
|
140
|
+
opts: Parameters<WasmController["prepareSession"]>[0],
|
|
141
|
+
): Promise<WasmSession> {
|
|
148
142
|
const sessionGeneration = this.sessionGeneration;
|
|
149
|
-
const { code, filename,
|
|
150
|
-
// We pass down a messenger object to the code
|
|
151
|
-
// This is used to have synchronous communication between the JS and Python code
|
|
152
|
-
// Previously, we used a queue, but this would not properly flush the queue
|
|
153
|
-
// during processing of a cell's code.
|
|
154
|
-
//
|
|
155
|
-
// This adds a messenger object to the global scope (import js; js.messenger.callback)
|
|
156
|
-
self.messenger = {
|
|
157
|
-
callback: onMessage,
|
|
158
|
-
};
|
|
143
|
+
const { code, filename, queryParameters, userConfig } = opts;
|
|
159
144
|
self.query_params = queryParameters;
|
|
160
145
|
self.user_config = userConfig;
|
|
161
146
|
|
|
162
|
-
const span = t.startSpan("
|
|
147
|
+
const span = t.startSpan("prepareSession.runPython");
|
|
163
148
|
const nbFilename = filename || WasmFileSystem.NOTEBOOK_FILENAME;
|
|
164
149
|
const sessionResources = this.requirePyodide.runPython(
|
|
165
150
|
`
|
|
@@ -169,14 +154,12 @@ export class DefaultWasmController implements WasmController {
|
|
|
169
154
|
import js
|
|
170
155
|
from marimo._pyodide.bootstrap import create_session, instantiate
|
|
171
156
|
|
|
172
|
-
assert js.messenger, "messenger is not defined"
|
|
173
157
|
assert js.query_params, "query_params is not defined"
|
|
174
158
|
|
|
175
159
|
def create_session_resources():
|
|
176
160
|
session, bridge = create_session(
|
|
177
161
|
filename="${nbFilename}",
|
|
178
162
|
query_params=js.query_params.to_py(),
|
|
179
|
-
message_callback=js.messenger.callback,
|
|
180
163
|
user_config=js.user_config.to_py(),
|
|
181
164
|
)
|
|
182
165
|
session_task = None
|
|
@@ -198,6 +181,7 @@ export class DefaultWasmController implements WasmController {
|
|
|
198
181
|
if task is not None:
|
|
199
182
|
await asyncio.gather(task, return_exceptions=True)
|
|
200
183
|
if bridge is not None:
|
|
184
|
+
bridge.close()
|
|
201
185
|
bridge.session = None
|
|
202
186
|
task = None
|
|
203
187
|
kernel_task = None
|
|
@@ -235,45 +219,65 @@ export class DefaultWasmController implements WasmController {
|
|
|
235
219
|
} finally {
|
|
236
220
|
packagesProxy.destroy();
|
|
237
221
|
}
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
//
|
|
244
|
-
// a sense of responsiveness.
|
|
222
|
+
let stopped = false;
|
|
223
|
+
let stopping: Promise<void> | undefined;
|
|
224
|
+
const releaseInit = once(() => initSession.destroy());
|
|
225
|
+
const isStopped = () =>
|
|
226
|
+
stopped || sessionGeneration !== this.sessionGeneration;
|
|
227
|
+
// Load in parallel with subscription setup; only start() executes cells.
|
|
245
228
|
const dependenciesReady = this.packageLoadQueue.then(() => {
|
|
246
|
-
if (
|
|
229
|
+
if (isStopped()) {
|
|
247
230
|
return;
|
|
248
231
|
}
|
|
249
232
|
return this.loadNotebookDeps(code, foundPackages);
|
|
250
233
|
});
|
|
251
234
|
this.packageLoadQueue = dependenciesReady.catch(() => undefined);
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
235
|
+
const session: WasmSession = {
|
|
236
|
+
bridge: bridgeProxy as unknown as SerializedBridge,
|
|
237
|
+
start: once(async () => {
|
|
238
|
+
try {
|
|
239
|
+
if (isStopped()) {
|
|
240
|
+
return;
|
|
241
|
+
}
|
|
242
|
+
await dependenciesReady;
|
|
243
|
+
if (!isStopped()) {
|
|
244
|
+
initSession(userConfig.runtime.auto_instantiate);
|
|
245
|
+
}
|
|
246
|
+
} catch (error) {
|
|
247
|
+
if (!isStopped()) {
|
|
248
|
+
throw error;
|
|
249
|
+
}
|
|
250
|
+
} finally {
|
|
251
|
+
releaseInit();
|
|
256
252
|
}
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
253
|
+
}),
|
|
254
|
+
stop: () => {
|
|
255
|
+
stopped = true;
|
|
256
|
+
releaseInit();
|
|
257
|
+
stopping ??= (async () => {
|
|
258
|
+
await stopSession();
|
|
259
|
+
stopSession.destroy();
|
|
260
|
+
bridgeProxy.destroy();
|
|
261
|
+
this.sessions.delete(session);
|
|
262
|
+
})().catch((error: unknown) => {
|
|
263
|
+
stopping = undefined;
|
|
264
|
+
throw error;
|
|
265
|
+
});
|
|
266
|
+
return stopping;
|
|
267
|
+
},
|
|
268
|
+
};
|
|
269
|
+
this.sessions.add(session);
|
|
270
|
+
return session;
|
|
265
271
|
}
|
|
266
272
|
|
|
267
273
|
public async stopSession(): Promise<void> {
|
|
268
274
|
this.sessionGeneration += 1;
|
|
269
|
-
const
|
|
275
|
+
const sessions = [...this.sessions];
|
|
270
276
|
let hasFailure = false;
|
|
271
277
|
let firstFailure: unknown;
|
|
272
|
-
for (const
|
|
278
|
+
for (const session of sessions) {
|
|
273
279
|
try {
|
|
274
|
-
await stop();
|
|
275
|
-
this.activeSessionStops.delete(stop);
|
|
276
|
-
stop.destroy();
|
|
280
|
+
await session.stop();
|
|
277
281
|
} catch (error) {
|
|
278
282
|
if (!hasFailure) {
|
|
279
283
|
hasFailure = true;
|
|
@@ -7,12 +7,11 @@ import {
|
|
|
7
7
|
createWorkerParentTransport,
|
|
8
8
|
type RPCSchema,
|
|
9
9
|
} from "rpc-anywhere";
|
|
10
|
-
import type { SaveNotebookRequest } from "@/core/network/types";
|
|
11
10
|
import { decodeUtf8 } from "@/utils/strings";
|
|
12
11
|
import { Deferred } from "../../../utils/Deferred";
|
|
13
12
|
import { prettyError } from "../../../utils/errors";
|
|
14
13
|
import { Logger } from "../../../utils/Logger";
|
|
15
|
-
import type { ParentSchema } from "../rpc";
|
|
14
|
+
import type { ParentSchema, PersistNotebookRequest } from "../rpc";
|
|
16
15
|
import type { WasmRuntimeConfig } from "../runtime-config";
|
|
17
16
|
import { TRANSPORT_ID } from "./constants";
|
|
18
17
|
import { WasmFileSystem } from "./fs";
|
|
@@ -70,14 +69,14 @@ const requestHandler = createRPCRequestHandler({
|
|
|
70
69
|
await pyodideReadyPromise; // Make sure loading is done
|
|
71
70
|
return WasmFileSystem.readNotebook(self.pyodide);
|
|
72
71
|
},
|
|
73
|
-
|
|
74
|
-
await pyodideReadyPromise;
|
|
75
|
-
const
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
72
|
+
writeNotebook: async ({ filename, contents }: PersistNotebookRequest) => {
|
|
73
|
+
await pyodideReadyPromise;
|
|
74
|
+
const directory = filename.substring(0, filename.lastIndexOf("/"));
|
|
75
|
+
if (directory) {
|
|
76
|
+
self.pyodide.FS.mkdirTree(directory);
|
|
77
|
+
}
|
|
78
|
+
self.pyodide.FS.writeFile(filename, contents);
|
|
79
|
+
self.pyodide.FS.writeFile(WasmFileSystem.NOTEBOOK_FILENAME, contents);
|
|
81
80
|
await WasmFileSystem.persistFilesToRemote(self.pyodide);
|
|
82
81
|
},
|
|
83
82
|
});
|
|
@@ -2,11 +2,7 @@
|
|
|
2
2
|
import type { PyodideInterface } from "pyodide";
|
|
3
3
|
import type { WasmRuntimeConfig } from "../runtime-config";
|
|
4
4
|
import type { UserConfig } from "@/core/config/config-schema";
|
|
5
|
-
import type {
|
|
6
|
-
ClientCommand,
|
|
7
|
-
NotificationPayload,
|
|
8
|
-
} from "@/core/kernel/messages";
|
|
9
|
-
import type { JsonString } from "@/utils/json/base64";
|
|
5
|
+
import type { ClientCommand } from "@/core/kernel/messages";
|
|
10
6
|
import type {
|
|
11
7
|
CodeCompletionRequest,
|
|
12
8
|
CopyNotebookRequest,
|
|
@@ -34,7 +30,6 @@ import type {
|
|
|
34
30
|
FormatCellsRequest,
|
|
35
31
|
FormatResponse,
|
|
36
32
|
SaveAppConfigurationRequest,
|
|
37
|
-
SaveNotebookRequest,
|
|
38
33
|
SaveUserConfigurationRequest,
|
|
39
34
|
Snippets,
|
|
40
35
|
} from "../../network/types";
|
|
@@ -58,19 +53,21 @@ export interface WasmController {
|
|
|
58
53
|
code: string;
|
|
59
54
|
filename: string | null;
|
|
60
55
|
}): Promise<{ code: string; filename: string }>;
|
|
61
|
-
/**
|
|
62
|
-
|
|
63
|
-
* @param opts.code - The code to start with
|
|
64
|
-
* @param opts.fallbackCode - The code to fallback to
|
|
65
|
-
* @param opts.filename - The filename to start with
|
|
66
|
-
*/
|
|
67
|
-
startSession(opts: {
|
|
56
|
+
/** Prepare a session and load its dependencies without executing cells. */
|
|
57
|
+
prepareSession(opts: {
|
|
68
58
|
queryParameters: Record<string, string | string[]>;
|
|
69
59
|
code: string;
|
|
70
60
|
filename: string | null;
|
|
71
61
|
userConfig: UserConfig;
|
|
72
|
-
|
|
73
|
-
|
|
62
|
+
}): Promise<WasmSession>;
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
export interface WasmSession {
|
|
66
|
+
readonly bridge: SerializedBridge;
|
|
67
|
+
/** Start once dependencies are ready. Repeated calls share the same start. */
|
|
68
|
+
start(): Promise<void>;
|
|
69
|
+
/** Prevent a pending start and release the session and its bridge. */
|
|
70
|
+
stop(): Promise<void>;
|
|
74
71
|
}
|
|
75
72
|
|
|
76
73
|
export interface RawBridge {
|
|
@@ -81,7 +78,6 @@ export interface RawBridge {
|
|
|
81
78
|
read_snippets(): Promise<Snippets>;
|
|
82
79
|
get_environment_info(): Promise<EnvironmentInfo>;
|
|
83
80
|
format(request: FormatCellsRequest): Promise<FormatResponse>;
|
|
84
|
-
save(request: SaveNotebookRequest): Promise<string>;
|
|
85
81
|
copy(request: CopyNotebookRequest): Promise<string>;
|
|
86
82
|
save_app_config(request: SaveAppConfigurationRequest): Promise<string>;
|
|
87
83
|
save_user_config(request: SaveUserConfigurationRequest): Promise<null>;
|
|
@@ -114,11 +110,26 @@ export type BridgePayload<T extends keyof RawBridge> = T extends keyof RawBridge
|
|
|
114
110
|
: undefined;
|
|
115
111
|
|
|
116
112
|
export type SerializedBridge = {
|
|
117
|
-
[P in
|
|
113
|
+
[P in keyof RawBridge]: RawBridge[P] extends (
|
|
118
114
|
payload: string,
|
|
119
115
|
) => Promise<unknown>
|
|
120
116
|
? (payload: string) => Promise<string>
|
|
121
117
|
: RawBridge[P];
|
|
122
118
|
} & {
|
|
123
|
-
|
|
119
|
+
notebook: {
|
|
120
|
+
readonly id: string;
|
|
121
|
+
subscribe(
|
|
122
|
+
id: string,
|
|
123
|
+
cursor: string | null,
|
|
124
|
+
emit: (event: string) => void,
|
|
125
|
+
): Promise<void>;
|
|
126
|
+
unsubscribe(id: string): void;
|
|
127
|
+
call(call: string): Promise<string>;
|
|
128
|
+
transaction(request: string): Promise<void>;
|
|
129
|
+
set_layout(layout: string): Promise<void>;
|
|
130
|
+
set_persistence(
|
|
131
|
+
persist: (contents: string, filename: string) => Promise<void>,
|
|
132
|
+
): void;
|
|
133
|
+
destroy(): void;
|
|
134
|
+
};
|
|
124
135
|
};
|