@marimo-team/frontend 0.23.17-dev5 → 0.23.17-dev50

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 (326) hide show
  1. package/dist/assets/{CellStatus-DIW20_Z8.js → CellStatus-XlxL4ctu.js} +1 -1
  2. package/dist/assets/{ConnectedDataExplorerComponent-C4zEgCZh.js → ConnectedDataExplorerComponent-BF5CJvB2.js} +1 -1
  3. package/dist/assets/{ErrorBoundary-DzTD8LVq.js → ErrorBoundary-BfMsWVQU.js} +1 -1
  4. package/dist/assets/ImageComparisonComponent-CukPDP1p.js +1 -0
  5. package/dist/assets/{ImperativeModal-fY-ZfT0b.js → ImperativeModal-Ci0icDPt.js} +1 -1
  6. package/dist/assets/{JsonOutput-Bk2exFh0.js → JsonOutput-W3f6H6VQ.js} +13 -13
  7. package/dist/assets/{LazyAnyLanguageCodeMirror-A-Ant1Ad.js → LazyAnyLanguageCodeMirror-Dcvy_p3-.js} +2 -2
  8. package/dist/assets/{MarimoErrorOutput-L_o2Knzo.js → MarimoErrorOutput-COZo8cfk.js} +5 -5
  9. package/dist/assets/{RSPContexts-WiCgGm__.js → RSPContexts-IrDQoFki.js} +1 -1
  10. package/dist/assets/{RenderHTML-BTDLQMKg.js → RenderHTML-CQC8I348.js} +1 -1
  11. package/dist/assets/{RunButton-DiHX9YjB.js → RunButton-DRxV8u_T.js} +1 -1
  12. package/dist/assets/add-cell-with-ai-CuLILhtV.js +107 -0
  13. package/dist/assets/{add-connection-dialog-BZaJAQCM.js → add-connection-dialog-Dld_PDdZ.js} +31 -31
  14. package/dist/assets/{agent-panel-CRxiqCgP.js → agent-panel-3Ta5EPgw.js} +6 -6
  15. package/dist/assets/ai-model-dropdown-CIqhiHxf.js +5 -0
  16. package/dist/assets/{alert-dialog-Bb8Lz8ko.js → alert-dialog-DMh8_LWc.js} +1 -1
  17. package/dist/assets/{any-language-editor-BIeaxtdD.js → any-language-editor-DUQN91an.js} +1 -1
  18. package/dist/assets/{app-config-button-BS4LSo1l.js → app-config-button-EyzXs1Nt.js} +1 -1
  19. package/dist/assets/{blob-B9eg-RfN.js → blob-BzgGkt3B.js} +1 -1
  20. package/dist/assets/{button-DDGiD1NW.js → button-BB1AV4Rb.js} +1 -1
  21. package/dist/assets/{cache-panel-B2B1ePao.js → cache-panel-hpL5o4Y7.js} +1 -1
  22. package/dist/assets/{capabilities-DoEg0XwL.js → capabilities-LI6_PoMn.js} +1 -1
  23. package/dist/assets/{cell-editor-OC7h_nZ3.js → cell-editor-C5qD14SG.js} +11 -11
  24. package/dist/assets/{cell-link-D0aELsL_.js → cell-link-uPmFMRTh.js} +1 -1
  25. package/dist/assets/cells-BE1kpmPF.css +2 -0
  26. package/dist/assets/{cells-DLYs3NiJ.js → cells-DxL32Atg.js} +60 -60
  27. package/dist/assets/chat-display-CBpsXLHT.js +1 -0
  28. package/dist/assets/chat-panel-Dzq-QTrj.js +4 -0
  29. package/dist/assets/chat-ui-BcI4PAQ1.js +4 -0
  30. package/dist/assets/{column-preview-CnqeJb-x.js → column-preview-Dxxs9Brz.js} +1 -1
  31. package/dist/assets/{command-Dqxie4gq.js → command-CPFHYoYk.js} +1 -1
  32. package/dist/assets/command-palette-B1EO8oVR.js +1 -0
  33. package/dist/assets/{common-D_beLbpm.js → common-aHaDUD0-.js} +1 -1
  34. package/dist/assets/components-Bm1d4vm9.js +1 -0
  35. package/dist/assets/{components-CxEGXaTB.js → components-RV25kGIW.js} +1 -1
  36. package/dist/assets/config-BVfo0E1-.js +1 -0
  37. package/dist/assets/config-Bm5B9wGP.js +1 -0
  38. package/dist/assets/{copy-Bxb1_Gtw.js → copy-B-LLR1Yy.js} +1 -1
  39. package/dist/assets/{copy-icon-neDQUo4Q.js → copy-icon-DWG8Y-xW.js} +1 -1
  40. package/dist/assets/{createReducer-DGx5Af-C.js → createReducer-DNehfY0t.js} +1 -1
  41. package/dist/assets/{datasource-Bfg8PKXp.js → datasource-D0eBPOXe.js} +2 -2
  42. package/dist/assets/{dates-C2LWzsCZ.js → dates-BzOh4kS_.js} +1 -1
  43. package/dist/assets/defaultLocale-t77eoiwF.js +1 -0
  44. package/dist/assets/{dependency-graph-panel-CB5ie0zu.js → dependency-graph-panel-BKWUzXHX.js} +2 -2
  45. package/dist/assets/{diagram-PSM6KHXK-DIXxW5au.js → diagram-PSM6KHXK-CrTUa1Xy.js} +1 -1
  46. package/dist/assets/{dialog-BCVepWLJ.js → dialog-88bLE398.js} +1 -1
  47. package/dist/assets/{dist-CPjTeT8p.js → dist-6qzebh8f.js} +1 -1
  48. package/dist/assets/dist-BLqRx0W-.js +1 -0
  49. package/dist/assets/{dist-BP4hpPAE.js → dist-BVr4HQPR.js} +1 -1
  50. package/dist/assets/{dist-DnjsPBRJ.js → dist-BWXIz8tq.js} +1 -1
  51. package/dist/assets/{dist-Cros4IAo.js → dist-BWbS__vR.js} +1 -1
  52. package/dist/assets/dist-BZehnCk2.js +1 -0
  53. package/dist/assets/dist-BaRgF7DU.js +1 -0
  54. package/dist/assets/{dist-DYEuVkEX.js → dist-BblMEWTQ.js} +1 -1
  55. package/dist/assets/dist-BlMeFaGB.js +1 -0
  56. package/dist/assets/{dist-CrKffqgx.js → dist-BmYa2llD.js} +1 -1
  57. package/dist/assets/{dist-9o0Xjd5v.js → dist-Btp8vRFr.js} +1 -1
  58. package/dist/assets/{dist-C-3FVUQA.js → dist-BvMWjDxm.js} +1 -1
  59. package/dist/assets/{dist-DeJBvhAc.js → dist-ByfZ_2N7.js} +1 -1
  60. package/dist/assets/dist-C-g2AKOo.js +1 -0
  61. package/dist/assets/{dist-DQl27SFq.js → dist-C52BqSeO.js} +1 -1
  62. package/dist/assets/{dist-CiVShZKM.js → dist-C7rDfTPJ.js} +1 -1
  63. package/dist/assets/{dist-BKgHjzML.js → dist-C830fIIj.js} +1 -1
  64. package/dist/assets/{dist-fLBYOt9R.js → dist-CHN_sQZw.js} +1 -1
  65. package/dist/assets/dist-CLFhXj8x.js +1 -0
  66. package/dist/assets/{dist-D47O3FBX.js → dist-CLQWOmD7.js} +1 -1
  67. package/dist/assets/dist-CO4dulTL.js +1 -0
  68. package/dist/assets/dist-CPw3iq_l.js +1 -0
  69. package/dist/assets/dist-CQtN-vVI.js +1 -0
  70. package/dist/assets/dist-CZE9FE_5.js +1 -0
  71. package/dist/assets/dist-CexuM4NA.js +1 -0
  72. package/dist/assets/{dist-DJuRuTXr.js → dist-CfiLpZZj.js} +1 -1
  73. package/dist/assets/dist-CjSt6gRp.js +1 -0
  74. package/dist/assets/dist-Cr4q2U0d.js +1 -0
  75. package/dist/assets/{dist-C6_DArtj.js → dist-CxaoIFrU.js} +1 -1
  76. package/dist/assets/{dist-aRc_dRed.js → dist-D5nYeU7Q.js} +1 -1
  77. package/dist/assets/{dist-BJZAIIze.js → dist-DX04AOlt.js} +1 -1
  78. package/dist/assets/dist-DkZos4_s.js +1 -0
  79. package/dist/assets/{dist-DI1-Z60I.js → dist-DmCNk9gq.js} +1 -1
  80. package/dist/assets/{dist-kLs0VbW_.js → dist-DmjT6iaH.js} +1 -1
  81. package/dist/assets/{dist-CvWC93qL.js → dist-DrY8fZeD.js} +1 -1
  82. package/dist/assets/{dist-l4RTzHX-.js → dist-Drexw1Ir.js} +1 -1
  83. package/dist/assets/{dist-DrBqBDHF.js → dist-DtNSqHzW.js} +1 -1
  84. package/dist/assets/dist-DxQA7Dn1.js +1 -0
  85. package/dist/assets/{dist-BLN8PNJV.js → dist-Dza6FJtV.js} +1 -1
  86. package/dist/assets/{dist-8SH6SWEG.js → dist-HrXjBuXP.js} +1 -1
  87. package/dist/assets/dist-J00F7mRY.js +1 -0
  88. package/dist/assets/{dist-DcwTQG1A.js → dist-Jbd9zUbK.js} +1 -1
  89. package/dist/assets/{dist-CZvbHnyT.js → dist-bvfx5OQE.js} +1 -1
  90. package/dist/assets/{dist-12S8I4CH.js → dist-gUHullA_.js} +11 -11
  91. package/dist/assets/dist-nYKCNtEX.js +1 -0
  92. package/dist/assets/dist-oK8dffBy.js +1 -0
  93. package/dist/assets/{documentation-panel-Dii7ouzj.js → documentation-panel-DotG4o5W.js} +1 -1
  94. package/dist/assets/{download-DenADH0w.js → download-Tklnz6pn.js} +3 -3
  95. package/dist/assets/{dropdown-menu-dHmZR8bI.js → dropdown-menu-BGBlPKNz.js} +1 -1
  96. package/dist/assets/{edit-page-B0EhuiM-.js → edit-page-HdLqqfXC.js} +6 -6
  97. package/dist/assets/{error-banner-BSxEOIn2.js → error-banner-BkIr-0dH.js} +1 -1
  98. package/dist/assets/{error-entries-4XYBryls.js → error-entries-BvUR_22-.js} +3 -3
  99. package/dist/assets/{error-panel-xYZ4lX0A.js → error-panel-v0YdlA0h.js} +1 -1
  100. package/dist/assets/{errors-PgHhDa-M.js → errors-DCPV979Q.js} +1 -1
  101. package/dist/assets/{es-gYgfF8y5.js → es-Dl2xAwTv.js} +1 -1
  102. package/dist/assets/{esm-B30N1GS0.js → esm-Bmzgrl_5.js} +1 -1
  103. package/dist/assets/{esm-BDGraAqp.js → esm-Nb9m4lqg.js} +1 -1
  104. package/dist/assets/{field-BsxD35VE.js → field-BFgvUZyh.js} +1 -1
  105. package/dist/assets/file-explorer-panel-DSZY0Q7h.js +37 -0
  106. package/dist/assets/{file-icons-mgXTzM1c.js → file-icons-NkObNkr_.js} +1 -1
  107. package/dist/assets/{file-name-input-Dq7Z_QmU.js → file-name-input-BQJ9U9lU.js} +1 -1
  108. package/dist/assets/{floating-outline-ypOHt2wD.js → floating-outline-BHmregl1.js} +1 -1
  109. package/dist/assets/{focus-0L1nX7M9.js → focus-BBiJct2G.js} +1 -1
  110. package/dist/assets/{form-oz4Z280Z.js → form-BGoLLcgk.js} +1 -1
  111. package/dist/assets/{form-BsW7zr4t.js → form-BIUQB1NI.js} +1 -1
  112. package/dist/assets/{formats-DzmLpc9A.js → formats-DNYtnQk7.js} +1 -1
  113. package/dist/assets/{formatting-COz4XEaB.js → formatting-qNOrQYRN.js} +1 -1
  114. package/dist/assets/{fullscreen-CTDqNQr6.js → fullscreen-B51geQ_6.js} +1 -1
  115. package/dist/assets/{gallery-page-DbuxknAw.js → gallery-page-DBHHOraF.js} +1 -1
  116. package/dist/assets/{ganttDiagram-JELNMOA3-DaS60lKW.js → ganttDiagram-JELNMOA3-C9KBLmAA.js} +1 -1
  117. package/dist/assets/{glide-data-editor-CejKKgnN.js → glide-data-editor-CHN4-xUu.js} +1 -1
  118. package/dist/assets/{home-page-BOP1dDcM.js → home-page-DLeE6pH_.js} +2 -2
  119. package/dist/assets/{html-to-image-D_C-hjtC.js → html-to-image-ChNWEUlR.js} +1 -1
  120. package/dist/assets/index-BxVymcMs.js +38 -0
  121. package/dist/assets/index-WoaGtlKd.css +2 -0
  122. package/dist/assets/{input-JxFwu-3G.js → input-_AaMXWgD.js} +1 -1
  123. package/dist/assets/{kiosk-mode-BvKMuigM.js → kiosk-mode-RTwj6XH4.js} +1 -1
  124. package/dist/assets/{label-CqjrpwRC.js → label-DUD3SZnk.js} +1 -1
  125. package/dist/assets/{layout-yfmf2ZBx.js → layout-QkEddeXT.js} +4 -4
  126. package/dist/assets/{linear-BPv4krGt.js → linear-B1ROUUtm.js} +1 -1
  127. package/dist/assets/{links-ClNQSIal.js → links-DBc28bsl.js} +1 -1
  128. package/dist/assets/{logs-panel-OMd1Z9CL.js → logs-panel-Cdcjc_fq.js} +1 -1
  129. package/dist/assets/markdown-export-CNu5ZTPR.js +1 -0
  130. package/dist/assets/{markdown-renderer-BfLNJGNj.js → markdown-renderer-DpskAUxO.js} +1 -1
  131. package/dist/assets/{menu-items-CCi6aOPm.js → menu-items-Db-bgz92.js} +1 -1
  132. package/dist/assets/{mermaid-B8gNHyGy.js → mermaid-OU2hLna0.js} +3 -3
  133. package/dist/assets/{name-cell-input-COtjRghz.js → name-cell-input-DoxL2EXI.js} +1 -1
  134. package/dist/assets/{numbers-rmawRqQP.js → numbers-D9g2ArIv.js} +1 -1
  135. package/dist/assets/{outline-panel-CIGf4SLb.js → outline-panel-BxjI-YSj.js} +1 -1
  136. package/dist/assets/{packages-panel-DLuMuNuG.js → packages-panel-CiKYP9Ij.js} +1 -1
  137. package/dist/assets/{pair-with-agent-modal-C5OhJ8rX.js → pair-with-agent-modal-1b5Eh8Xo.js} +1 -1
  138. package/dist/assets/{panel-accordion-state-D1UP17Lk.js → panel-accordion-state-BkWYwckU.js} +1 -1
  139. package/dist/assets/{panels-B0w23XJS.js → panels-Swk0l08v.js} +2 -2
  140. package/dist/assets/{popover-Bcz1Zikb.js → popover-DHbixh3D.js} +1 -1
  141. package/dist/assets/{precisionRound-i2ImZxgc.js → precisionRound-DORAUfVY.js} +1 -1
  142. package/dist/assets/process-output-DJonUYp2.js +1 -0
  143. package/dist/assets/{quadrantDiagram-AYHSOK5B-DJuGB95-.js → quadrantDiagram-AYHSOK5B-Duf8sGaa.js} +1 -1
  144. package/dist/assets/{radio-group-BGXYy3u5.js → radio-group-ByLzIKjm.js} +1 -1
  145. package/dist/assets/react-vega-J-J2akIt.js +191 -0
  146. package/dist/assets/{readonly-code-display-DdABj8tH.js → readonly-code-display-B_ZHScA7.js} +1 -1
  147. package/dist/assets/{readonly-python-code-DUjqXwhi.js → readonly-python-code-C4GeGU7O.js} +1 -1
  148. package/dist/assets/{renderShortcut-Bl8_FVn8.js → renderShortcut-DnqsU4zj.js} +1 -1
  149. package/dist/assets/{reveal-component-BjgfnO81.js → reveal-component-De_cjq3p.js} +2 -2
  150. package/dist/assets/{run-page-DDty5GvQ.js → run-page-DmXT_-MQ.js} +1 -1
  151. package/dist/assets/{runs-s9t2f_Tg.js → runs-C6avq6Gi.js} +1 -1
  152. package/dist/assets/scratchpad-panel-BtGHIRdJ.js +1 -0
  153. package/dist/assets/{scroll-to-bottom-button-C8yfT4wq.js → scroll-to-bottom-button-CRMPUjYZ.js} +1 -1
  154. package/dist/assets/{secrets-panel-Cz17_MLf.js → secrets-panel-DmAwH7vI.js} +1 -1
  155. package/dist/assets/{select-BP75_5jJ.js → select-Clmc8Hw2.js} +1 -1
  156. package/dist/assets/{session-BmukRjSi.js → session-BtBr2zr3.js} +1 -1
  157. package/dist/assets/{session-panel-aMoQPTCB.js → session-panel-kBLydN0h.js} +1 -1
  158. package/dist/assets/{share-852X_cqy.js → share-DSOv_SvY.js} +1 -1
  159. package/dist/assets/{skeleton-CPuDx7yn.js → skeleton-CJymG2v3.js} +1 -1
  160. package/dist/assets/{snippets-panel-DBDXr-XP.js → snippets-panel-l_HnIWUj.js} +1 -1
  161. package/dist/assets/{spec-C7opZZRp.js → spec-CdDzA5ef.js} +1 -1
  162. package/dist/assets/{state-DCtLlFJq.js → state-B4lC29et.js} +1 -1
  163. package/dist/assets/{state-C4kiwedg.js → state-BkoVfBqt.js} +1 -1
  164. package/dist/assets/{state-hlEqpZfH.js → state-D9xpDCxa.js} +2 -2
  165. package/dist/assets/state-DDimGh2o.js +1 -0
  166. package/dist/assets/{strings--BknHX9C.js → strings-DMKwTex-.js} +1 -1
  167. package/dist/assets/{swiper-component-DXW8snlt.js → swiper-component-pSAP3cRf.js} +1 -1
  168. package/dist/assets/switch-8FxRs-OU.js +1 -0
  169. package/dist/assets/{terminal-CaHTVNrz.js → terminal-DPQJ6hNg.js} +1 -1
  170. package/dist/assets/{textarea-DtCcemo9.js → textarea-BzWvGIrJ.js} +1 -1
  171. package/dist/assets/{time-p47i6wvp.js → time-BO-naW1O.js} +1 -1
  172. package/dist/assets/tooltip-BMi4WFDl.js +61 -0
  173. package/dist/assets/{tooltip-4bZlU7jl.js → tooltip-BcgbFR7q.js} +1 -1
  174. package/dist/assets/{tracing-CStM7RGI.js → tracing-DCuWLKnk.js} +1 -1
  175. package/dist/assets/{tracing-panel-BTLzu8L-.js → tracing-panel-dVtThkzL.js} +2 -2
  176. package/dist/assets/{tree-actions-BwwVyp3l.js → tree-actions-o2iyMmKN.js} +1 -1
  177. package/dist/assets/{useBoolean-DIi7jQLa.js → useBoolean-sASJEpGA.js} +1 -1
  178. package/dist/assets/{useCellActionButton-uxVyv3JK.js → useCellActionButton-n2-2xb-5.js} +1 -1
  179. package/dist/assets/{useDeleteCell-Cy23OP4e.js → useDeleteCell-DH0weyin.js} +1 -1
  180. package/dist/assets/{useDependencyPanelTab-DNfjp_-X.js → useDependencyPanelTab-DQZmbkd-.js} +1 -1
  181. package/dist/assets/{useEventListener-C0195JsF.js → useEventListener-B_wSLgFR.js} +1 -1
  182. package/dist/assets/{useHotkey-DLZvrRZT.js → useHotkey-DJyYmk5P.js} +1 -1
  183. package/dist/assets/{useIframeCapabilities-j6fR_dBA.js → useIframeCapabilities-D_6fB_8O.js} +1 -1
  184. package/dist/assets/{useInstallPackage-WiYBwhUO.js → useInstallPackage-DyNrAhrx.js} +1 -1
  185. package/dist/assets/{useInterval-CnQfyMa9.js → useInterval-CjGsg4ie.js} +1 -1
  186. package/dist/assets/useNotebookActions-BUQta3nX.js +12 -0
  187. package/dist/assets/useRunCells-8xYRh7wH.js +1 -0
  188. package/dist/assets/{useSplitCell-BK5mnkkL.js → useSplitCell-BvgvIZaz.js} +1 -1
  189. package/dist/assets/{useTheme-Bxv9B31g.js → useTheme-BR31Gjbc.js} +1 -1
  190. package/dist/assets/utils-C5VWChv9.js +1 -0
  191. package/dist/assets/{vega-component-BGiCExka.js → vega-component-DG9G1qk2.js} +1 -1
  192. package/dist/assets/{vega-embed-container-DxSqUOCK.js → vega-embed-container-BJRQVehC.js} +1 -1
  193. package/dist/assets/vega-loader.browser-CAfmqKEc.js +6 -0
  194. package/dist/assets/{write-secret-modal-Dtm66yVf.js → write-secret-modal-DwCUuNRy.js} +1 -1
  195. package/dist/assets/{xychartDiagram-PRI3JC2R-B8A0hi-Q.js → xychartDiagram-PRI3JC2R-ClPpSFDI.js} +1 -1
  196. package/dist/assets/{zod-H_cgTO0M.js → zod-CIOjj3iA.js} +5 -5
  197. package/dist/index.html +87 -87
  198. package/package.json +7 -7
  199. package/src/__mocks__/requests.ts +5 -0
  200. package/src/components/ai/ai-model-dropdown.tsx +2 -2
  201. package/src/components/app-config/user-config-form.tsx +26 -0
  202. package/src/components/chat/__tests__/capabilities-popover.test.tsx +38 -0
  203. package/src/components/chat/__tests__/chat-abort.test.ts +51 -0
  204. package/src/components/chat/__tests__/use-chat-controller-id.test.tsx +39 -0
  205. package/src/components/chat/capabilities-popover.tsx +68 -0
  206. package/src/components/chat/chat-abort.ts +44 -0
  207. package/src/components/chat/chat-display.tsx +17 -0
  208. package/src/components/chat/chat-panel.tsx +107 -50
  209. package/src/components/chat/chat-utils.ts +1 -1
  210. package/src/components/chat/use-chat-controller-id.ts +24 -0
  211. package/src/components/data-table/charts/__tests__/schemas.test.ts +35 -0
  212. package/src/components/data-table/charts/schemas.ts +5 -2
  213. package/src/components/databases/icons/huggingface.svg +8 -0
  214. package/src/components/datasets/__tests__/icons.test.ts +21 -0
  215. package/src/components/datasets/icons.tsx +10 -0
  216. package/src/components/datasources/__tests__/components.test.tsx +29 -0
  217. package/src/components/datasources/components.tsx +8 -3
  218. package/src/components/editor/actions/export-dialog/__tests__/export-dialog.test.tsx +35 -0
  219. package/src/components/editor/actions/export-dialog/__tests__/state.test.ts +61 -0
  220. package/src/components/editor/actions/export-dialog/export-command.ts +4 -9
  221. package/src/components/editor/actions/export-dialog/export-dialog.tsx +27 -23
  222. package/src/components/editor/actions/export-dialog/format-notice.tsx +28 -0
  223. package/src/components/editor/actions/export-dialog/state.ts +23 -5
  224. package/src/components/editor/actions/export-dialog/use-export-dialog.ts +7 -15
  225. package/src/components/editor/actions/useConfigActions.tsx +24 -0
  226. package/src/components/editor/ai/add-cell-with-ai.tsx +2 -2
  227. package/src/components/editor/ai/ai-completion-editor.tsx +2 -2
  228. package/src/components/editor/chrome/panels/outline/__tests__/useActiveOutline.test.ts +26 -0
  229. package/src/components/editor/chrome/wrapper/useOpenAiAssistant.ts +2 -2
  230. package/src/components/editor/code/__tests__/readonly-python-code.test.tsx +114 -2
  231. package/src/components/editor/connections/storage/__tests__/__snapshots__/as-code.test.ts.snap +14 -0
  232. package/src/components/editor/connections/storage/__tests__/as-code.test.ts +20 -0
  233. package/src/components/editor/connections/storage/add-storage-form.tsx +10 -0
  234. package/src/components/editor/connections/storage/as-code.ts +19 -1
  235. package/src/components/editor/connections/storage/schemas.ts +19 -0
  236. package/src/components/storage/__tests__/storage-snippets.test.ts +88 -0
  237. package/src/components/storage/components.tsx +2 -0
  238. package/src/components/storage/storage-snippets.ts +58 -0
  239. package/src/core/__tests__/mode.test.ts +85 -0
  240. package/src/core/ai/__tests__/config.test.tsx +96 -0
  241. package/src/core/ai/config.ts +16 -22
  242. package/src/core/ai/constants.ts +7 -0
  243. package/src/core/ai/staged-cells.ts +3 -0
  244. package/src/core/ai/state.ts +10 -0
  245. package/src/core/codemirror/__tests__/setup.test.ts +1 -1
  246. package/src/core/codemirror/cm.ts +2 -2
  247. package/src/core/codemirror/code-lens/__tests__/extension.test.ts +4 -5
  248. package/src/core/codemirror/code-lens/extension.ts +3 -3
  249. package/src/core/codemirror/find-replace/search-highlight.ts +8 -3
  250. package/src/core/codemirror/language/languages/sql/__tests__/renderers.test.tsx +44 -0
  251. package/src/core/codemirror/language/languages/sql/renderers.tsx +3 -3
  252. package/src/core/codemirror/theme/__tests__/light.test.ts +59 -123
  253. package/src/core/codemirror/theme/dark.ts +6 -5
  254. package/src/core/codemirror/theme/light.ts +6 -4
  255. package/src/core/config/__tests__/config-schema.test.ts +2 -0
  256. package/src/core/config/config-schema.ts +1 -0
  257. package/src/core/config/feature-flag.tsx +0 -2
  258. package/src/core/dom/outline.ts +25 -2
  259. package/src/core/hotkeys/hotkeys.ts +2 -0
  260. package/src/core/mode.ts +11 -9
  261. package/src/core/network/__tests__/api.test.ts +132 -17
  262. package/src/core/network/__tests__/export-filename.test.ts +43 -0
  263. package/src/core/network/__tests__/requests-lazy.test.ts +1 -0
  264. package/src/core/network/api.ts +47 -12
  265. package/src/core/network/export-filename.ts +37 -0
  266. package/src/core/network/markdown-export.ts +19 -0
  267. package/src/core/network/requests-network.ts +29 -5
  268. package/src/core/network/types.ts +4 -0
  269. package/src/core/storage/types.ts +1 -0
  270. package/src/css/app/codemirror.css +11 -7
  271. package/src/css/globals.css +8 -0
  272. package/src/plugins/impl/DataEditorPlugin.tsx +2 -9
  273. package/src/plugins/impl/__tests__/DataEditorPlugin.test.ts +18 -0
  274. package/src/plugins/impl/chat/chat-ui.tsx +51 -3
  275. package/src/plugins/impl/data-frames/__tests__/schema.test.ts +30 -0
  276. package/src/plugins/impl/data-frames/schema.ts +4 -1
  277. package/src/plugins/impl/image-comparison/ImageComparisonComponent.tsx +20 -1
  278. package/src/plugins/impl/image-comparison/__tests__/ImageComparisonComponent.test.tsx +58 -0
  279. package/src/stories/theme.stories.tsx +71 -1
  280. package/dist/assets/ImageComparisonComponent-COvcNAUu.js +0 -1
  281. package/dist/assets/add-cell-with-ai-BU5EDuHt.js +0 -107
  282. package/dist/assets/ai-model-dropdown-CLCa-gsg.js +0 -5
  283. package/dist/assets/cells-BVzo2MHf.css +0 -2
  284. package/dist/assets/chat-display-7CzBSZRm.js +0 -1
  285. package/dist/assets/chat-panel-Dd5WiR1a.js +0 -4
  286. package/dist/assets/chat-ui-CYqqqoSC.js +0 -4
  287. package/dist/assets/command-palette-BBlSC8AZ.js +0 -1
  288. package/dist/assets/components-K_Mi0vdC.js +0 -1
  289. package/dist/assets/config-BJVx3eBW.js +0 -1
  290. package/dist/assets/config-DAL-jiRM.js +0 -1
  291. package/dist/assets/defaultLocale-DOVIEEPN.js +0 -1
  292. package/dist/assets/dist-B99-E4qp.js +0 -1
  293. package/dist/assets/dist-Bfm9U3vp.js +0 -1
  294. package/dist/assets/dist-BiwuSVkx.js +0 -1
  295. package/dist/assets/dist-BrZjJXDA.js +0 -1
  296. package/dist/assets/dist-CaZf6AoY.js +0 -1
  297. package/dist/assets/dist-Cgoy5550.js +0 -1
  298. package/dist/assets/dist-Csf5i4KY.js +0 -1
  299. package/dist/assets/dist-D6Iobl3P.js +0 -1
  300. package/dist/assets/dist-D6WgndTj.js +0 -1
  301. package/dist/assets/dist-DcXy-1eJ.js +0 -1
  302. package/dist/assets/dist-DsfPGbgk.js +0 -1
  303. package/dist/assets/dist-DtbtRAqz.js +0 -1
  304. package/dist/assets/dist-I3qeQWuP.js +0 -1
  305. package/dist/assets/dist-aeKlrIMv.js +0 -1
  306. package/dist/assets/dist-gDi2I7IJ.js +0 -1
  307. package/dist/assets/dist-miDa4-cz.js +0 -1
  308. package/dist/assets/dist-nDEBxpL6.js +0 -1
  309. package/dist/assets/dist-oUL4XFdB.js +0 -1
  310. package/dist/assets/file-explorer-panel-4h-BKKaM.js +0 -26
  311. package/dist/assets/house-DS0r7dNZ.js +0 -1
  312. package/dist/assets/index-BXlfeAku.js +0 -38
  313. package/dist/assets/index-DYycoCjG.css +0 -2
  314. package/dist/assets/process-output-Cm_vuKBw.js +0 -1
  315. package/dist/assets/react-vega-BLc5RhC7.js +0 -188
  316. package/dist/assets/scratchpad-panel-DJkO5OrI.js +0 -1
  317. package/dist/assets/state-Brswj8bD.js +0 -1
  318. package/dist/assets/switch-CONP-vq2.js +0 -1
  319. package/dist/assets/tooltip-B7U4xkIN.js +0 -61
  320. package/dist/assets/useNotebookActions-BHaEvPEM.js +0 -12
  321. package/dist/assets/useRunCells-kmbsc2R5.js +0 -1
  322. package/dist/assets/utils-BWBBpCkT.js +0 -1
  323. package/dist/assets/vega-loader.browser-Dun7Qe4B.js +0 -6
  324. /package/dist/assets/{bundle.esm-D0TWQ6iT.js → bundle.esm-CvMga-X6.js} +0 -0
  325. /package/dist/assets/{icons-CIPE8AIY.js → icons-NFYKylj6.js} +0 -0
  326. /package/dist/assets/{useDeepCompareMemoize-CSMoa-ga.js → useDeepCompareMemoize-BLRnan2Y.js} +0 -0
@@ -1,14 +1,14 @@
1
- var ht=Object.defineProperty;var mt=(t,e,s)=>e in t?ht(t,e,{enumerable:!0,configurable:!0,writable:!0,value:s}):t[e]=s;var G=(t,e,s)=>mt(t,typeof e!="symbol"?e+"":e,s);import{s as ie}from"./chunk-LvLJmgfZ.js";import{u as ut}from"./useEvent-D91BmmQi.js";import{t as bt}from"./react-Bj1aDYRI.js";import{Ct as ce,E as ft,Lt as gt,Pr as yt,ur as y}from"./cells-DLYs3NiJ.js";import{D as K,P as _,R as f,T as L,_ as le,i as _t,k as x}from"./zod-H_cgTO0M.js";import{t as U}from"./compiler-runtime-B3qBwwSJ.js";import{_ as xt}from"./useEventListener-C0195JsF.js";import{c as St}from"./utils-BWBBpCkT.js";import{t as V}from"./cn-DYvqRARy.js";import{t as kt}from"./jsx-runtime-Blw4afVn.js";import{a as wt,c as de,i as jt,o as pe,r as vt,s as $t,t as Ct}from"./popover-Bcz1Zikb.js";import{c as At,i as Nt,n as Rt,r as Tt,s as Lt,t as Dt}from"./select-BP75_5jJ.js";import{i as It,t as he}from"./tooltip-4bZlU7jl.js";import{t as me}from"./button-DDGiD1NW.js";import{r as W}from"./strings--BknHX9C.js";import{r as Pt,t as qt}from"./x-C-Eq0ag8.js";import{t as Et}from"./circle-plus-BluydVNx.js";import{t as Ft}from"./components-K_Mi0vdC.js";import{n as Z}from"./DeferredRequestRegistry-C8EV1PoE.js";import{t as Ot}from"./plus-AgXtozBz.js";import{t as Bt}from"./sparkles-DUCpRFlS.js";import{a as ue,c as be,i as Ut,n as fe,o as ge,p as Kt,r as ye,s as _e,u as Mt}from"./form-BsW7zr4t.js";import{a as Wt,c as Ht,l as Gt,n as Vt,r as zt,t as Qt}from"./dialog-BCVepWLJ.js";import{n as xe}from"./ImperativeModal-fY-ZfT0b.js";import{t as z}from"./links-wcQA1Np1.js";import{t as Jt}from"./useIframeCapabilities-j6fR_dBA.js";import{r as Yt}from"./html-to-image-D_C-hjtC.js";import{o as Xt}from"./focus-0L1nX7M9.js";import{n as Se}from"./useAsyncData-Bg5iAViP.js";import{a as Zt,c as ke,i as Q,o as J,s as es,t as ts}from"./command-Dqxie4gq.js";import{c as d,o as ss,t as rs}from"./form-oz4Z280Z.js";import{t as ns}from"./request-registry-Dpk9hQrX.js";import{a as ee,c as te,d as os,f as as,i as we,l as je,n as is,o as cs,r as se,s as E,t as ve,u as $e}from"./write-secret-modal-Dtm66yVf.js";var D=ie(bt(),1),o=ie(kt(),1);function ls(t,e){if(!e)return{recommended:[],other:[...t]};let s;try{s=new RegExp(e,"i")}catch{return{recommended:[],other:[...t]}}let n=[],r=[];for(let a of t)s.test(a)?n.push(a):r.push(a);return{recommended:n,other:r}}const Ce=({value:t,onChange:e,placeholder:s,secretsOnly:n=!1,recommendedKeys:r,otherKeys:a,onCreateSecret:c,className:i,searchPlaceholder:l,formatCustomValueLabel:p=g=>`Use "${g}"`,createSecretLabel:h="Create a new secret",allowCustomValue:u,allowCreateSecret:b})=>{let[g,m]=(0,D.useState)(!1),[S,w]=(0,D.useState)(""),k=S.trim(),v=!n&&k.length>0&&k!==t&&((u==null?void 0:u(k))??!0),A=(b==null?void 0:b(k))??!0,C=t?E(t)?cs(t):t:null,N=j=>{e(te(j)),m(!1),w("")},P=j=>{e(j),m(!1),w("")},q=()=>{e(""),w(""),m(!1)};return(0,o.jsxs)("div",{className:V("flex w-full min-w-48 mb-1 items-center gap-1",i),children:[(0,o.jsxs)(Ct,{modal:!0,open:g,onOpenChange:j=>{m(j),j||w("")},children:[(0,o.jsx)(jt,{asChild:!0,children:(0,o.jsxs)("button",{type:"button",className:V("flex h-6 min-w-0 flex-1 shadow-xs-solid items-center justify-between","rounded-sm border border-input bg-background px-1.5 text-sm font-code ring-offset-background placeholder:text-muted-foreground","hover:shadow-sm-solid focus:outline-hidden focus:ring-1 focus:ring-ring focus:border-primary focus:shadow-md-solid",E(t)&&"bg-accent"),"aria-expanded":g,children:[(0,o.jsx)("span",{className:"truncate flex-1 min-w-0 text-left flex items-center gap-1.5",children:C?(0,o.jsxs)(o.Fragment,{children:[E(t)&&(0,o.jsx)(Z,{className:"h-3 w-3 shrink-0 opacity-70"}),(0,o.jsx)("span",{className:"truncate",children:C})]}):(0,o.jsx)("span",{className:"text-muted-foreground",children:s})}),(0,o.jsx)(Pt,{className:"ml-1 h-4 w-4 opacity-50 shrink-0"})]})}),(0,o.jsx)(vt,{className:"w-(--radix-popover-trigger-width) p-0",align:"start",children:(0,o.jsxs)(ts,{children:[(0,o.jsx)(Zt,{placeholder:l??(n?"Search secrets...":"Type a value or search secrets..."),rootClassName:"px-1 h-8",autoFocus:!0,value:S,onValueChange:w}),(0,o.jsxs)(es,{className:"max-h-60 overscroll-contain",children:[v&&(0,o.jsx)(Q,{children:(0,o.jsx)(J,{value:`use custom value ${k}`,onSelect:()=>P(k),children:p(k)})}),A&&(0,o.jsx)(Q,{className:"mt-0",children:(0,o.jsxs)(J,{value:`create new secret ${k}`,onSelect:()=>{let j=k||void 0;m(!1),w(""),c(j)},children:[(0,o.jsx)(Et,{className:"mr-2 h-3.5 w-3.5"}),h]})}),r.length>0&&(0,o.jsxs)(o.Fragment,{children:[(0,o.jsx)(ke,{className:"mt-0"}),(0,o.jsx)(Q,{heading:"Recommended",children:r.map(j=>(0,o.jsxs)(J,{value:j,onSelect:()=>N(j),children:[(0,o.jsx)(Z,{className:"mr-2 h-3 w-3 opacity-70"}),j]},j))})]}),a.length>0&&(0,o.jsxs)(o.Fragment,{children:[(0,o.jsx)(ke,{className:"mt-0"}),(0,o.jsx)(Q,{heading:r.length>0?"Other":void 0,children:a.map(j=>(0,o.jsxs)(J,{value:j,onSelect:()=>N(j),children:[(0,o.jsx)(Z,{className:"mr-2 h-3 w-3 opacity-70"}),j]},j))})]})]})]})})]}),t&&(0,o.jsx)("button",{type:"button","aria-label":"Clear",className:V("flex h-6 w-6 shrink-0 items-center justify-center rounded-sm","text-muted-foreground hover:bg-muted hover:text-foreground","focus:outline-hidden focus:ring-1 focus:ring-ring"),onClick:q,children:(0,o.jsx)(qt,{className:"h-3.5 w-3.5"})})]})};var ds=U(),Ae=(0,D.createContext)({providerNames:[],secretKeys:[],loading:!1,error:void 0,refreshSecrets:xt.NOOP});const Ne=()=>(0,D.use)(Ae),ps=t=>{let e=(0,ds.c)(14),{children:s}=t,n;e[0]===Symbol.for("react.memo_cache_sentinel")?(n=[],e[0]=n):n=e[0];let{data:r,isPending:a,error:c,refetch:i}=Se(gs,n);r==null||r.secretKeys,r==null||r.providerNames;let l;e[1]===(r==null?void 0:r.secretKeys)?l=e[2]:(l=(r==null?void 0:r.secretKeys)||[],e[1]=r==null?void 0:r.secretKeys,e[2]=l);let p;e[3]===(r==null?void 0:r.providerNames)?p=e[4]:(p=(r==null?void 0:r.providerNames)||[],e[3]=r==null?void 0:r.providerNames,e[4]=p);let h;e[5]!==c||e[6]!==a||e[7]!==i||e[8]!==l||e[9]!==p?(h={secretKeys:l,providerNames:p,loading:a,error:c,refreshSecrets:i},e[5]=c,e[6]=a,e[7]=i,e[8]=l,e[9]=p,e[10]=h):h=e[10];let u=h,b;return e[11]!==s||e[12]!==u?(b=(0,o.jsx)(Ae,{value:u,children:s}),e[11]=s,e[12]=u,e[13]=b):b=e[13],b},hs={isMatch:t=>{if(t instanceof le){let{optionRegex:e}=d.parse(t.description||"");return!!e}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:n,providerNames:r,refreshSecrets:a}=Ne(),{openModal:c,closeModal:i}=xe(),{label:l,description:p,placeholder:h,optionRegex:u="",inputType:b}=d.parse(t.description||""),g=b==="password",{recommended:m,other:S}=ls(n,u);return(0,o.jsx)(ue,{control:e.control,name:s,render:({field:w})=>(0,o.jsxs)(ge,{children:[(0,o.jsx)(_e,{children:l}),(0,o.jsx)(ye,{children:p}),(0,o.jsx)(fe,{children:(0,o.jsx)(Ce,{value:w.value?String(w.value):"",onChange:w.onChange,placeholder:h,secretsOnly:g,recommendedKeys:m,otherKeys:S,onCreateSecret:k=>{c((0,o.jsx)(ve,{providerNames:r,initialValue:k,onSuccess:v=>{a(),w.onChange(te(v)),i()},onClose:i}))}})}),(0,o.jsx)(be,{})]})})}},ms={isMatch:t=>{if(t instanceof le){let{special:e}=d.parse(t.description||"");return e==="secret_textarea"}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:n,providerNames:r,refreshSecrets:a}=Ne(),{openModal:c,closeModal:i}=xe(),{label:l,description:p,placeholder:h}=d.parse(t.description||"");return(0,o.jsx)(ue,{control:e.control,name:s,render:({field:u})=>{let b=u.value?String(u.value):"",g=$e(b)?as(b):b;return(0,o.jsxs)(ge,{children:[(0,o.jsxs)(_e,{className:"flex items-center gap-1",children:[l,(0,o.jsx)(he,{content:"Enter a path to a file on disk, or create a secret to paste JSON directly.",delayDuration:300,children:(0,o.jsx)("button",{type:"button",className:"inline-flex","aria-label":"Credential input help",children:(0,o.jsx)(yt,{className:"h-3.5 w-3.5"})})})]}),(0,o.jsx)(ye,{children:p}),(0,o.jsx)(fe,{children:(0,o.jsx)(Ce,{value:g,onChange:m=>{u.onChange(!m||E(m)||$e(m)?m:os(m))},placeholder:h,searchPlaceholder:"Type a file path or paste JSON",formatCustomValueLabel:m=>`Use "${m}" as a file path`,createSecretLabel:"Paste JSON credentials",allowCustomValue:m=>!we(m),allowCreateSecret:m=>!m||we(m),recommendedKeys:[],otherKeys:n,onCreateSecret:m=>{c((0,o.jsx)(ve,{providerNames:r,initialValue:m,multiline:!0,onSuccess:S=>{a(),u.onChange(te(S)),i()},onClose:i}))}})}),(0,o.jsx)(be,{})]})}})}};function us(t){return t.provider!=="env"}function bs(t){return t.name}function fs(t){return t.keys}async function gs(){let t=await ns.request({}),e=is(t.secrets).filter(us).map(bs);return{secretKeys:t.secrets.flatMap(fs).toSorted(),providerNames:e}}var H=U(),ys=[hs,ms];const re=t=>{let e=(0,H.c)(2),{children:s}=t,n;return e[0]===s?n=e[1]:(n=(0,o.jsx)("div",{className:"grid grid-cols-2 md:grid-cols-3 gap-4",children:s}),e[0]=s,e[1]=n),n},ne=t=>{let e=(0,H.c)(9),{name:s,color:n,icon:r,onSelect:a}=t,c;e[0]===n?c=e[1]:(c={backgroundColor:n},e[0]=n,e[1]=c);let i;e[2]===s?i=e[3]:(i=(0,o.jsx)("span",{className:"text-white font-medium text-lg",children:s}),e[2]=s,e[3]=i);let l;return e[4]!==r||e[5]!==a||e[6]!==c||e[7]!==i?(l=(0,o.jsxs)("button",{type:"button",className:"py-3 flex flex-col items-center justify-center gap-1 transition-all hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid",style:c,onClick:a,children:[r,i]}),e[4]=r,e[5]=a,e[6]=c,e[7]=i,e[8]=l):l=e[8],l},_s=t=>{let e=(0,H.c)(25),{onBack:s,isValid:n,libraries:r,preferredLibrary:a,onLibraryChange:c,displayNames:i,libraryLabel:l}=t,p=l===void 0?"Preferred library":l,h;e[0]===s?h=e[1]:(h=(0,o.jsx)(me,{type:"button",variant:"outline",onClick:s,children:"Back"}),e[0]=s,e[1]=h);let u=!n,b;e[2]===u?b=e[3]:(b=(0,o.jsx)(me,{type:"submit",disabled:u,children:"Add"}),e[2]=u,e[3]=b);let g;e[4]!==h||e[5]!==b?(g=(0,o.jsxs)("div",{className:"flex gap-2",children:[h,b]}),e[4]=h,e[5]=b,e[6]=g):g=e[6];let m;e[7]===Symbol.for("react.memo_cache_sentinel")?(m=(0,o.jsx)(Lt,{children:(0,o.jsx)(At,{placeholder:"Select a library"})}),e[7]=m):m=e[7];let S;e[8]===p?S=e[9]:(S=(0,o.jsxs)("div",{className:"flex flex-col gap-1 items-end",children:[m,(0,o.jsx)("span",{className:"text-xs text-muted-foreground",children:p})]}),e[8]=p,e[9]=S);let w;if(e[10]!==i||e[11]!==r){let C;e[13]===i?C=e[14]:(C=N=>(0,o.jsx)(Nt,{value:N,children:i[N]??N},N),e[13]=i,e[14]=C),w=r.map(C),e[10]=i,e[11]=r,e[12]=w}else w=e[12];let k;e[15]===w?k=e[16]:(k=(0,o.jsx)(Rt,{children:(0,o.jsx)(Tt,{children:w})}),e[15]=w,e[16]=k);let v;e[17]!==c||e[18]!==a||e[19]!==S||e[20]!==k?(v=(0,o.jsx)("div",{children:(0,o.jsxs)(Dt,{value:a,onValueChange:c,children:[S,k]})}),e[17]=c,e[18]=a,e[19]=S,e[20]=k,e[21]=v):v=e[21];let A;return e[22]!==v||e[23]!==g?(A=(0,o.jsxs)("div",{className:"flex gap-2 justify-between",children:[g,v]}),e[22]=v,e[23]=g,e[24]=A):A=e[24],A};function Re(){let t=(0,H.c)(4),{createNewCell:e}=ft(),s=Xt(),n=ut(St),r;return t[0]!==n||t[1]!==e||t[2]!==s?(r=a=>{/\bmo\./.test(a)&&Yt({autoInstantiate:n,createNewCell:e,fromCellId:s}),e({code:a,before:!1,cellId:s??"__end__",skipIfCodeExists:!0})},t[0]=n,t[1]=e,t[2]=s,t[3]=r):r=t[3],r}const Te=t=>{let e=(0,H.c)(30),{schema:s,libraries:n,preferredLibrary:r,displayNames:a,libraryLabel:c,generateCode:i,onSubmit:l,onBack:p}=t,h;e[0]===s?h=e[1]:(h=ss(s),e[0]=s,e[1]=h);let u=h,b;e[2]===s?b=e[3]:(b=Mt(s),e[2]=s,e[3]=b);let g;e[4]!==u||e[5]!==b?(g={defaultValues:u,resolver:b,reValidateMode:"onChange"},e[4]=u,e[5]=b,e[6]=g):g=e[6];let m=Kt(g),[S,w]=D.useState(r),k=Re(),v;e[7]!==i||e[8]!==k||e[9]!==l||e[10]!==S?(v=pt=>{k(i(pt,S)),l()},e[7]=i,e[8]=k,e[9]=l,e[10]=S,e[11]=v):v=e[11];let A=v,C;e[12]!==m||e[13]!==A?(C=m.handleSubmit(A),e[12]=m,e[13]=A,e[14]=C):C=e[14];let N;e[15]===Symbol.for("react.memo_cache_sentinel")?(N=(0,o.jsx)(Ut,{}),e[15]=N):N=e[15];let P;e[16]!==m||e[17]!==s?(P=(0,o.jsx)(ps,{children:(0,o.jsx)(rs,{schema:s,form:m,renderers:ys,children:N})}),e[16]=m,e[17]=s,e[18]=P):P=e[18];let q;e[19]!==a||e[20]!==m.formState.isValid||e[21]!==n||e[22]!==c||e[23]!==p||e[24]!==S?(q=(0,o.jsx)(_s,{onBack:p,isValid:m.formState.isValid,libraries:n,preferredLibrary:S,onLibraryChange:w,displayNames:a,libraryLabel:c}),e[19]=a,e[20]=m.formState.isValid,e[21]=n,e[22]=c,e[23]=p,e[24]=S,e[25]=q):q=e[25];let j;return e[26]!==C||e[27]!==P||e[28]!==q?(j=(0,o.jsxs)("form",{onSubmit:C,className:"space-y-4",children:[P,q]}),e[26]=C,e[27]=P,e[28]=q,e[29]=j):j=e[29],j};function F(){return f().optional().describe(d.of({label:"Password",inputType:"password",optionRegex:"(password|passwd|pgpassword|db.?pass(word)?)"}))}function Y(t,e){let s=f();return s=e?s.nonempty():s.optional(),s=s.describe(d.of({label:t||"Token",inputType:"password",optionRegex:"(token|api.?key|access.?token|auth.?token|(^|_)pat(_|$))"})),s}function X(){return f().optional().describe(d.of({label:"Warehouse Name",optionRegex:"(warehouse|snowflake.?warehouse)"}))}function oe(t,e){let s=f();return s=e?s.nonempty():s.optional(),s.describe(d.of({label:t||"URI",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"}))}function I(t){return f().nonempty().describe(d.of({label:t||"Host",optionRegex:"^(?!.*(kubernetes|gpg)).*(host(name)?|pghost|db.?host|database.?host|mysql.?host|postgres.?host|server.?host)"}))}function R(){return f().describe(d.of({label:"Database",optionRegex:"(database|db.?name|pgdatabase|^DB$|^DATABASE$)"}))}function xs(){return f().describe(d.of({label:"Schema",optionRegex:"(^SCHEMA$|db.?schema|postgres.?schema|pg.?schema|_SCHEMA$)"}))}function T(){return f().nonempty().describe(d.of({label:"Username",optionRegex:"(username|pguser|db.?user|^USER$|_USER$)"}))}function O(t){let e=_t().describe(d.of({label:"Port",inputType:"number",placeholder:t==null?void 0:t.toString()})).transform(Number).refine(s=>s>=0&&s<=65535,{message:"Port must be between 0 and 65535"});return t===void 0?e:e.default(t)}function Le(){return L().default(!1).describe(d.of({label:"Read Only"}))}const De=_({type:x("postgres"),host:I(),port:O(5432).optional(),database:R(),username:T(),password:F(),ssl:L().default(!1).describe(d.of({label:"Use SSL"}))}).describe(d.of({direction:"two-columns"})),Ie=_({type:x("mysql"),host:I(),port:O(3306),database:R(),username:T(),password:F(),ssl:L().default(!1).describe(d.of({label:"Use SSL"}))}).describe(d.of({direction:"two-columns"})),Pe=_({type:x("sqlite"),database:R().describe(d.of({label:"Database Path"}))}).describe(d.of({direction:"two-columns"})),qe=_({type:x("duckdb"),database:R().describe(d.of({label:"Database Path"})),read_only:Le()}).describe(d.of({direction:"two-columns"})),Ee=_({type:x("motherduck"),database:R().default("my_db").describe(d.of({label:"Database Name"})),token:Y()}).describe(d.of({direction:"two-columns"})),Fe=_({type:x("snowflake"),account:f().nonempty().describe(d.of({label:"Account",optionRegex:"(snowflake.?account|^SNOWFLAKE_ACCOUNT$|account.?id)"})),warehouse:f().optional().describe(d.of({label:"Warehouse",optionRegex:"(snowflake.?warehouse|warehouse)"})),database:R(),schema:f().optional().describe(d.of({label:"Schema",optionRegex:"(snowflake.?schema|^SCHEMA$|_SCHEMA$)"})),role:f().optional().describe(d.of({label:"Role"})),authType:K("type",[_({type:x("Password"),username:T(),password:F(),enable_mfa:L().default(!1).describe(d.of({label:"Enable MFA (Duo Push)"}))}),_({type:x("SSO (Browser)"),username:T()}),_({type:x("Key Pair"),username:T(),private_key_path:f().nonempty().describe(d.of({label:"Private Key Path",placeholder:"/path/to/rsa_key.p8"})),private_key_passphrase:f().optional().describe(d.of({label:"Private Key Passphrase",inputType:"password",optionRegex:"(passphrase|private.?key.?passphrase|key.?passphrase)"}))}),_({type:x("OAuth / PAT"),token:Y("Token",!0)})]).default({type:"Password",username:"username",enable_mfa:!1}).describe(d.of({special:"tabs"}))}).describe(d.of({direction:"two-columns"})),Oe=_({type:x("bigquery"),project:f().nonempty().describe(d.of({label:"Project ID",optionRegex:"(bigquery.?project|gcp.?project|google.?cloud.?project|project.?id)"})),dataset:f().nonempty().describe(d.of({label:"Dataset",optionRegex:"(bigquery.?dataset|dataset)"})),credentials_json:f().describe(d.of({label:"Credentials JSON",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),Be=_({type:x("clickhouse_connect"),host:I(),port:O(8123).optional(),username:T(),password:F(),secure:L().default(!1).describe(d.of({label:"Use HTTPs"})),proxy_path:f().optional().describe(d.of({label:"Proxy Path",placeholder:"/clickhouse"}))}).describe(d.of({direction:"two-columns"})),Ue=_({type:x("timeplus"),host:I().default("localhost"),port:O(8123).optional(),username:T().default("default"),password:F().default("")}).describe(d.of({direction:"two-columns"})),Ke=_({type:x("chdb"),database:R().describe(d.of({label:"Database Path"})),read_only:Le()}).describe(d.of({direction:"two-columns"})),Me=_({type:x("trino"),host:I(),port:O(8080),database:R(),schema:xs().optional(),username:T(),password:F(),async_support:L().default(!1).describe(d.of({label:"Async Support"}))}).describe(d.of({direction:"two-columns"})),We=_({type:x("iceberg"),name:f().describe(d.of({label:"Catalog Name"})),catalog:K("type",[_({type:x("REST"),warehouse:X(),uri:f().optional().describe(d.of({label:"URI",placeholder:"https://",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"})),token:Y()}),_({type:x("SQL"),warehouse:X(),uri:f().optional().describe(d.of({label:"URI",placeholder:"jdbc:iceberg://host:port/database",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"}))}),_({type:x("Hive"),warehouse:X(),uri:oe()}),_({type:x("Glue"),warehouse:X(),uri:oe()}),_({type:x("DynamoDB"),"dynamodb.profile-name":f().optional().describe(d.of({label:"Profile Name"})),"dynamodb.region":f().optional().describe(d.of({label:"Region"})),"dynamodb.access-key-id":f().optional().describe(d.of({label:"Access Key ID",inputType:"password",optionRegex:"(access.?key.?id|aws.?access.?key)"})),"dynamodb.secret-access-key":f().optional().describe(d.of({label:"Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|aws.?secret)"})),"dynamodb.session-token":f().optional().describe(d.of({label:"Session Token",inputType:"password",optionRegex:"(session.?token|aws.?session)"}))})]).default({type:"REST",token:void 0}).describe(d.of({special:"tabs"}))}),He=_({type:x("datafusion"),sessionContext:L().optional().describe(d.of({label:"Use Session Context"}))}),Ge=_({type:x("pyspark"),host:I().optional(),port:O().optional()}),Ve=_({type:x("redshift"),host:I(),port:O(5439),connectionType:K("type",[_({type:x("IAM credentials"),region:f().describe(d.of({label:"Region"})),aws_access_key_id:f().nonempty().describe(d.of({label:"AWS Access Key ID",inputType:"password",optionRegex:"(access.?key.?id|aws.?access.?key)"})),aws_secret_access_key:f().nonempty().describe(d.of({label:"AWS Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|aws.?secret)"})),aws_session_token:f().optional().describe(d.of({label:"AWS Session Token",inputType:"password",optionRegex:"(session.?token|aws.?session)"}))}),_({type:x("DB credentials"),user:T(),password:F()})]).default({type:"IAM credentials",aws_access_key_id:"",aws_secret_access_key:"",region:""}),database:R()}).describe(d.of({direction:"two-columns"})),ze=_({type:x("databricks"),access_token:Y("Access Token",!0),server_hostname:I("Server Hostname"),http_path:oe("HTTP Path",!0),catalog:f().optional().describe(d.of({label:"Catalog"})),schema:f().optional().describe(d.of({label:"Schema"}))}).describe(d.of({direction:"two-columns"})),Qe=_({type:x("supabase"),host:I(),port:O(5432).optional(),database:R(),username:T(),password:F(),disable_client_pooling:L().default(!1).describe(d.of({label:"Disable Client-Side Pooling"}))}).describe(d.of({direction:"two-columns"})),Ss=K("type",[De,Ie,Pe,qe,Ee,Fe,Oe,Be,Ue,Ke,Me,We,He,Ge,Ve,ze,Qe]),Je={sqlmodel:"SQLModel",sqlalchemy:"SQLAlchemy",duckdb:"DuckDB",clickhouse_connect:"ClickHouse Connect",chdb:"chDB",pyiceberg:"PyIceberg",ibis:"Ibis",motherduck:"MotherDuck",redshift:"Redshift",databricks:"Databricks"};var $=class{constructor(t,e,s){this.connection=t,this.orm=e,this.secrets=s}get imports(){let t=new Set(this.generateImports());switch(this.orm){case"sqlalchemy":t.add("import sqlalchemy");break;case"sqlmodel":t.add("import sqlmodel");break;case"duckdb":t.add("import duckdb");break;case"ibis":t.add("import ibis");break}return t}},ks=t=>`_${t}`,ws=class{constructor(){G(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e,s){if(t=ks(t),E(e)){let n=je(e),r=s?`os.environ.get("${n}", "${s}")`:`os.environ.get("${n}")`;return this.secrets[t]=r,t}if(s!=null){let n=`os.environ.get("${e}", "${s}")`;return this.secrets[t]=n,t}return typeof e=="number"||typeof e=="number"?`${e}`:typeof e=="boolean"?M(e):e?`"${e}"`:""}printInFString(t,e,s){if(e===void 0)return"";if(typeof e=="number")return`${e}`;if(typeof e=="boolean")return M(e);let n=this.print(t,e,s);return n.startsWith('"')&&n.endsWith('"')?n.slice(1,-1):`{${n}}`}printPassword(t,e,s,n){let r=s?this.printInFString.bind(this):this.print.bind(this);return E(t)?r(n||"password",t):r(n||"password",e,t)}getSecrets(){return this.secrets}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
2
- `)}},js=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'sslmode': 'require'}":"",e=this.secrets.printPassword(this.connection.password,"POSTGRES_PASSWORD",!0);return y(`
1
+ var ut=Object.defineProperty;var mt=(t,e,s)=>e in t?ut(t,e,{enumerable:!0,configurable:!0,writable:!0,value:s}):t[e]=s;var G=(t,e,s)=>mt(t,typeof e!="symbol"?e+"":e,s);import{s as ie}from"./chunk-LvLJmgfZ.js";import{u as bt}from"./useEvent-D91BmmQi.js";import{t as gt}from"./react-Bj1aDYRI.js";import{Ct as ce,E as ft,Fr as yt,Lt as _t,ur as y}from"./cells-DxL32Atg.js";import{D as K,L as g,P as _,T as L,_ as le,i as xt,k as x}from"./zod-CIOjj3iA.js";import{t as U}from"./compiler-runtime-B3qBwwSJ.js";import{_ as St}from"./useEventListener-B_wSLgFR.js";import{c as kt}from"./utils-C5VWChv9.js";import{t as V}from"./cn-DYvqRARy.js";import{t as wt}from"./jsx-runtime-Blw4afVn.js";import{a as jt,c as de,i as vt,o as pe,r as $t,s as Ct,t as At}from"./popover-DHbixh3D.js";import{c as Nt,i as Rt,n as Tt,r as Lt,s as Dt,t as It}from"./select-Clmc8Hw2.js";import{i as Pt,t as he}from"./tooltip-BcgbFR7q.js";import{t as ue}from"./button-BB1AV4Rb.js";import{r as M}from"./strings-DMKwTex-.js";import{r as qt,t as Ft}from"./x-C-Eq0ag8.js";import{t as Et}from"./circle-plus-BluydVNx.js";import{t as Ot}from"./components-Bm1d4vm9.js";import{n as Z}from"./DeferredRequestRegistry-C8EV1PoE.js";import{t as Bt}from"./plus-AgXtozBz.js";import{t as Ut}from"./sparkles-DUCpRFlS.js";import{a as me,c as be,i as Kt,n as ge,o as fe,p as Ht,r as ye,s as _e,u as Mt}from"./form-BIUQB1NI.js";import{a as Wt,c as Gt,l as Vt,n as zt,r as Qt,t as Jt}from"./dialog-88bLE398.js";import{n as xe}from"./ImperativeModal-Ci0icDPt.js";import{t as z}from"./links-wcQA1Np1.js";import{t as Yt}from"./useIframeCapabilities-D_6fB_8O.js";import{r as Xt}from"./html-to-image-ChNWEUlR.js";import{o as Zt}from"./focus-BBiJct2G.js";import{n as Se}from"./useAsyncData-Bg5iAViP.js";import{a as es,c as ke,i as Q,o as J,s as ts,t as ss}from"./command-CPFHYoYk.js";import{c as d,o as rs,t as ns}from"./form-BGoLLcgk.js";import{t as os}from"./request-registry-Dpk9hQrX.js";import{a as ee,c as te,d as as,f as is,i as we,l as je,n as cs,o as ls,r as se,s as F,t as ve,u as $e}from"./write-secret-modal-DwCUuNRy.js";var D=ie(gt(),1),o=ie(wt(),1);function ds(t,e){if(!e)return{recommended:[],other:[...t]};let s;try{s=new RegExp(e,"i")}catch{return{recommended:[],other:[...t]}}let n=[],r=[];for(let a of t)s.test(a)?n.push(a):r.push(a);return{recommended:n,other:r}}const Ce=({value:t,onChange:e,placeholder:s,secretsOnly:n=!1,recommendedKeys:r,otherKeys:a,onCreateSecret:c,className:i,searchPlaceholder:l,formatCustomValueLabel:p=f=>`Use "${f}"`,createSecretLabel:h="Create a new secret",allowCustomValue:m,allowCreateSecret:b})=>{let[f,u]=(0,D.useState)(!1),[S,w]=(0,D.useState)(""),k=S.trim(),v=!n&&k.length>0&&k!==t&&((m==null?void 0:m(k))??!0),A=(b==null?void 0:b(k))??!0,C=t?F(t)?ls(t):t:null,N=j=>{e(te(j)),u(!1),w("")},P=j=>{e(j),u(!1),w("")},q=()=>{e(""),w(""),u(!1)};return(0,o.jsxs)("div",{className:V("flex w-full min-w-48 mb-1 items-center gap-1",i),children:[(0,o.jsxs)(At,{modal:!0,open:f,onOpenChange:j=>{u(j),j||w("")},children:[(0,o.jsx)(vt,{asChild:!0,children:(0,o.jsxs)("button",{type:"button",className:V("flex h-6 min-w-0 flex-1 shadow-xs-solid items-center justify-between","rounded-sm border border-input bg-background px-1.5 text-sm font-code ring-offset-background placeholder:text-muted-foreground","hover:shadow-sm-solid focus:outline-hidden focus:ring-1 focus:ring-ring focus:border-primary focus:shadow-md-solid",F(t)&&"bg-accent"),"aria-expanded":f,children:[(0,o.jsx)("span",{className:"truncate flex-1 min-w-0 text-left flex items-center gap-1.5",children:C?(0,o.jsxs)(o.Fragment,{children:[F(t)&&(0,o.jsx)(Z,{className:"h-3 w-3 shrink-0 opacity-70"}),(0,o.jsx)("span",{className:"truncate",children:C})]}):(0,o.jsx)("span",{className:"text-muted-foreground",children:s})}),(0,o.jsx)(qt,{className:"ml-1 h-4 w-4 opacity-50 shrink-0"})]})}),(0,o.jsx)($t,{className:"w-(--radix-popover-trigger-width) p-0",align:"start",children:(0,o.jsxs)(ss,{children:[(0,o.jsx)(es,{placeholder:l??(n?"Search secrets...":"Type a value or search secrets..."),rootClassName:"px-1 h-8",autoFocus:!0,value:S,onValueChange:w}),(0,o.jsxs)(ts,{className:"max-h-60 overscroll-contain",children:[v&&(0,o.jsx)(Q,{children:(0,o.jsx)(J,{value:`use custom value ${k}`,onSelect:()=>P(k),children:p(k)})}),A&&(0,o.jsx)(Q,{className:"mt-0",children:(0,o.jsxs)(J,{value:`create new secret ${k}`,onSelect:()=>{let j=k||void 0;u(!1),w(""),c(j)},children:[(0,o.jsx)(Et,{className:"mr-2 h-3.5 w-3.5"}),h]})}),r.length>0&&(0,o.jsxs)(o.Fragment,{children:[(0,o.jsx)(ke,{className:"mt-0"}),(0,o.jsx)(Q,{heading:"Recommended",children:r.map(j=>(0,o.jsxs)(J,{value:j,onSelect:()=>N(j),children:[(0,o.jsx)(Z,{className:"mr-2 h-3 w-3 opacity-70"}),j]},j))})]}),a.length>0&&(0,o.jsxs)(o.Fragment,{children:[(0,o.jsx)(ke,{className:"mt-0"}),(0,o.jsx)(Q,{heading:r.length>0?"Other":void 0,children:a.map(j=>(0,o.jsxs)(J,{value:j,onSelect:()=>N(j),children:[(0,o.jsx)(Z,{className:"mr-2 h-3 w-3 opacity-70"}),j]},j))})]})]})]})})]}),t&&(0,o.jsx)("button",{type:"button","aria-label":"Clear",className:V("flex h-6 w-6 shrink-0 items-center justify-center rounded-sm","text-muted-foreground hover:bg-muted hover:text-foreground","focus:outline-hidden focus:ring-1 focus:ring-ring"),onClick:q,children:(0,o.jsx)(Ft,{className:"h-3.5 w-3.5"})})]})};var ps=U(),Ae=(0,D.createContext)({providerNames:[],secretKeys:[],loading:!1,error:void 0,refreshSecrets:St.NOOP});const Ne=()=>(0,D.use)(Ae),hs=t=>{let e=(0,ps.c)(14),{children:s}=t,n;e[0]===Symbol.for("react.memo_cache_sentinel")?(n=[],e[0]=n):n=e[0];let{data:r,isPending:a,error:c,refetch:i}=Se(ys,n);r==null||r.secretKeys,r==null||r.providerNames;let l;e[1]===(r==null?void 0:r.secretKeys)?l=e[2]:(l=(r==null?void 0:r.secretKeys)||[],e[1]=r==null?void 0:r.secretKeys,e[2]=l);let p;e[3]===(r==null?void 0:r.providerNames)?p=e[4]:(p=(r==null?void 0:r.providerNames)||[],e[3]=r==null?void 0:r.providerNames,e[4]=p);let h;e[5]!==c||e[6]!==a||e[7]!==i||e[8]!==l||e[9]!==p?(h={secretKeys:l,providerNames:p,loading:a,error:c,refreshSecrets:i},e[5]=c,e[6]=a,e[7]=i,e[8]=l,e[9]=p,e[10]=h):h=e[10];let m=h,b;return e[11]!==s||e[12]!==m?(b=(0,o.jsx)(Ae,{value:m,children:s}),e[11]=s,e[12]=m,e[13]=b):b=e[13],b},us={isMatch:t=>{if(t instanceof le){let{optionRegex:e}=d.parse(t.description||"");return!!e}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:n,providerNames:r,refreshSecrets:a}=Ne(),{openModal:c,closeModal:i}=xe(),{label:l,description:p,placeholder:h,optionRegex:m="",inputType:b}=d.parse(t.description||""),f=b==="password",{recommended:u,other:S}=ds(n,m);return(0,o.jsx)(me,{control:e.control,name:s,render:({field:w})=>(0,o.jsxs)(fe,{children:[(0,o.jsx)(_e,{children:l}),(0,o.jsx)(ye,{children:p}),(0,o.jsx)(ge,{children:(0,o.jsx)(Ce,{value:w.value?String(w.value):"",onChange:w.onChange,placeholder:h,secretsOnly:f,recommendedKeys:u,otherKeys:S,onCreateSecret:k=>{c((0,o.jsx)(ve,{providerNames:r,initialValue:k,onSuccess:v=>{a(),w.onChange(te(v)),i()},onClose:i}))}})}),(0,o.jsx)(be,{})]})})}},ms={isMatch:t=>{if(t instanceof le){let{special:e}=d.parse(t.description||"");return e==="secret_textarea"}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:n,providerNames:r,refreshSecrets:a}=Ne(),{openModal:c,closeModal:i}=xe(),{label:l,description:p,placeholder:h}=d.parse(t.description||"");return(0,o.jsx)(me,{control:e.control,name:s,render:({field:m})=>{let b=m.value?String(m.value):"",f=$e(b)?is(b):b;return(0,o.jsxs)(fe,{children:[(0,o.jsxs)(_e,{className:"flex items-center gap-1",children:[l,(0,o.jsx)(he,{content:"Enter a path to a file on disk, or create a secret to paste JSON directly.",delayDuration:300,children:(0,o.jsx)("button",{type:"button",className:"inline-flex","aria-label":"Credential input help",children:(0,o.jsx)(yt,{className:"h-3.5 w-3.5"})})})]}),(0,o.jsx)(ye,{children:p}),(0,o.jsx)(ge,{children:(0,o.jsx)(Ce,{value:f,onChange:u=>{m.onChange(!u||F(u)||$e(u)?u:as(u))},placeholder:h,searchPlaceholder:"Type a file path or paste JSON",formatCustomValueLabel:u=>`Use "${u}" as a file path`,createSecretLabel:"Paste JSON credentials",allowCustomValue:u=>!we(u),allowCreateSecret:u=>!u||we(u),recommendedKeys:[],otherKeys:n,onCreateSecret:u=>{c((0,o.jsx)(ve,{providerNames:r,initialValue:u,multiline:!0,onSuccess:S=>{a(),m.onChange(te(S)),i()},onClose:i}))}})}),(0,o.jsx)(be,{})]})}})}};function bs(t){return t.provider!=="env"}function gs(t){return t.name}function fs(t){return t.keys}async function ys(){let t=await os.request({}),e=cs(t.secrets).filter(bs).map(gs);return{secretKeys:t.secrets.flatMap(fs).toSorted(),providerNames:e}}var W=U(),_s=[us,ms];const re=t=>{let e=(0,W.c)(2),{children:s}=t,n;return e[0]===s?n=e[1]:(n=(0,o.jsx)("div",{className:"grid grid-cols-2 md:grid-cols-3 gap-4",children:s}),e[0]=s,e[1]=n),n},ne=t=>{let e=(0,W.c)(9),{name:s,color:n,icon:r,onSelect:a}=t,c;e[0]===n?c=e[1]:(c={backgroundColor:n},e[0]=n,e[1]=c);let i;e[2]===s?i=e[3]:(i=(0,o.jsx)("span",{className:"text-white font-medium text-lg",children:s}),e[2]=s,e[3]=i);let l;return e[4]!==r||e[5]!==a||e[6]!==c||e[7]!==i?(l=(0,o.jsxs)("button",{type:"button",className:"py-3 flex flex-col items-center justify-center gap-1 transition-all hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid",style:c,onClick:a,children:[r,i]}),e[4]=r,e[5]=a,e[6]=c,e[7]=i,e[8]=l):l=e[8],l},xs=t=>{let e=(0,W.c)(25),{onBack:s,isValid:n,libraries:r,preferredLibrary:a,onLibraryChange:c,displayNames:i,libraryLabel:l}=t,p=l===void 0?"Preferred library":l,h;e[0]===s?h=e[1]:(h=(0,o.jsx)(ue,{type:"button",variant:"outline",onClick:s,children:"Back"}),e[0]=s,e[1]=h);let m=!n,b;e[2]===m?b=e[3]:(b=(0,o.jsx)(ue,{type:"submit",disabled:m,children:"Add"}),e[2]=m,e[3]=b);let f;e[4]!==h||e[5]!==b?(f=(0,o.jsxs)("div",{className:"flex gap-2",children:[h,b]}),e[4]=h,e[5]=b,e[6]=f):f=e[6];let u;e[7]===Symbol.for("react.memo_cache_sentinel")?(u=(0,o.jsx)(Dt,{children:(0,o.jsx)(Nt,{placeholder:"Select a library"})}),e[7]=u):u=e[7];let S;e[8]===p?S=e[9]:(S=(0,o.jsxs)("div",{className:"flex flex-col gap-1 items-end",children:[u,(0,o.jsx)("span",{className:"text-xs text-muted-foreground",children:p})]}),e[8]=p,e[9]=S);let w;if(e[10]!==i||e[11]!==r){let C;e[13]===i?C=e[14]:(C=N=>(0,o.jsx)(Rt,{value:N,children:i[N]??N},N),e[13]=i,e[14]=C),w=r.map(C),e[10]=i,e[11]=r,e[12]=w}else w=e[12];let k;e[15]===w?k=e[16]:(k=(0,o.jsx)(Tt,{children:(0,o.jsx)(Lt,{children:w})}),e[15]=w,e[16]=k);let v;e[17]!==c||e[18]!==a||e[19]!==S||e[20]!==k?(v=(0,o.jsx)("div",{children:(0,o.jsxs)(It,{value:a,onValueChange:c,children:[S,k]})}),e[17]=c,e[18]=a,e[19]=S,e[20]=k,e[21]=v):v=e[21];let A;return e[22]!==v||e[23]!==f?(A=(0,o.jsxs)("div",{className:"flex gap-2 justify-between",children:[f,v]}),e[22]=v,e[23]=f,e[24]=A):A=e[24],A};function Re(){let t=(0,W.c)(4),{createNewCell:e}=ft(),s=Zt(),n=bt(kt),r;return t[0]!==n||t[1]!==e||t[2]!==s?(r=a=>{/\bmo\./.test(a)&&Xt({autoInstantiate:n,createNewCell:e,fromCellId:s}),e({code:a,before:!1,cellId:s??"__end__",skipIfCodeExists:!0})},t[0]=n,t[1]=e,t[2]=s,t[3]=r):r=t[3],r}const Te=t=>{let e=(0,W.c)(30),{schema:s,libraries:n,preferredLibrary:r,displayNames:a,libraryLabel:c,generateCode:i,onSubmit:l,onBack:p}=t,h;e[0]===s?h=e[1]:(h=rs(s),e[0]=s,e[1]=h);let m=h,b;e[2]===s?b=e[3]:(b=Mt(s),e[2]=s,e[3]=b);let f;e[4]!==m||e[5]!==b?(f={defaultValues:m,resolver:b,reValidateMode:"onChange"},e[4]=m,e[5]=b,e[6]=f):f=e[6];let u=Ht(f),[S,w]=D.useState(r),k=Re(),v;e[7]!==i||e[8]!==k||e[9]!==l||e[10]!==S?(v=ht=>{k(i(ht,S)),l()},e[7]=i,e[8]=k,e[9]=l,e[10]=S,e[11]=v):v=e[11];let A=v,C;e[12]!==u||e[13]!==A?(C=u.handleSubmit(A),e[12]=u,e[13]=A,e[14]=C):C=e[14];let N;e[15]===Symbol.for("react.memo_cache_sentinel")?(N=(0,o.jsx)(Kt,{}),e[15]=N):N=e[15];let P;e[16]!==u||e[17]!==s?(P=(0,o.jsx)(hs,{children:(0,o.jsx)(ns,{schema:s,form:u,renderers:_s,children:N})}),e[16]=u,e[17]=s,e[18]=P):P=e[18];let q;e[19]!==a||e[20]!==u.formState.isValid||e[21]!==n||e[22]!==c||e[23]!==p||e[24]!==S?(q=(0,o.jsx)(xs,{onBack:p,isValid:u.formState.isValid,libraries:n,preferredLibrary:S,onLibraryChange:w,displayNames:a,libraryLabel:c}),e[19]=a,e[20]=u.formState.isValid,e[21]=n,e[22]=c,e[23]=p,e[24]=S,e[25]=q):q=e[25];let j;return e[26]!==C||e[27]!==P||e[28]!==q?(j=(0,o.jsxs)("form",{onSubmit:C,className:"space-y-4",children:[P,q]}),e[26]=C,e[27]=P,e[28]=q,e[29]=j):j=e[29],j};function E(){return g().optional().describe(d.of({label:"Password",inputType:"password",optionRegex:"(password|passwd|pgpassword|db.?pass(word)?)"}))}function Y(t,e){let s=g();return s=e?s.nonempty():s.optional(),s=s.describe(d.of({label:t||"Token",inputType:"password",optionRegex:"(token|api.?key|access.?token|auth.?token|(^|_)pat(_|$))"})),s}function X(){return g().optional().describe(d.of({label:"Warehouse Name",optionRegex:"(warehouse|snowflake.?warehouse)"}))}function oe(t,e){let s=g();return s=e?s.nonempty():s.optional(),s.describe(d.of({label:t||"URI",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"}))}function I(t){return g().nonempty().describe(d.of({label:t||"Host",optionRegex:"^(?!.*(kubernetes|gpg)).*(host(name)?|pghost|db.?host|database.?host|mysql.?host|postgres.?host|server.?host)"}))}function R(){return g().describe(d.of({label:"Database",optionRegex:"(database|db.?name|pgdatabase|^DB$|^DATABASE$)"}))}function Ss(){return g().describe(d.of({label:"Schema",optionRegex:"(^SCHEMA$|db.?schema|postgres.?schema|pg.?schema|_SCHEMA$)"}))}function T(){return g().nonempty().describe(d.of({label:"Username",optionRegex:"(username|pguser|db.?user|^USER$|_USER$)"}))}function O(t){let e=xt().describe(d.of({label:"Port",inputType:"number",placeholder:t==null?void 0:t.toString()})).transform(Number).refine(s=>s>=0&&s<=65535,{message:"Port must be between 0 and 65535"});return t===void 0?e:e.default(t)}function Le(){return L().default(!1).describe(d.of({label:"Read Only"}))}const De=_({type:x("postgres"),host:I(),port:O(5432).optional(),database:R(),username:T(),password:E(),ssl:L().default(!1).describe(d.of({label:"Use SSL"}))}).describe(d.of({direction:"two-columns"})),Ie=_({type:x("mysql"),host:I(),port:O(3306),database:R(),username:T(),password:E(),ssl:L().default(!1).describe(d.of({label:"Use SSL"}))}).describe(d.of({direction:"two-columns"})),Pe=_({type:x("sqlite"),database:R().describe(d.of({label:"Database Path"}))}).describe(d.of({direction:"two-columns"})),qe=_({type:x("duckdb"),database:R().describe(d.of({label:"Database Path"})),read_only:Le()}).describe(d.of({direction:"two-columns"})),Fe=_({type:x("motherduck"),database:R().default("my_db").describe(d.of({label:"Database Name"})),token:Y()}).describe(d.of({direction:"two-columns"})),Ee=_({type:x("snowflake"),account:g().nonempty().describe(d.of({label:"Account",optionRegex:"(snowflake.?account|^SNOWFLAKE_ACCOUNT$|account.?id)"})),warehouse:g().optional().describe(d.of({label:"Warehouse",optionRegex:"(snowflake.?warehouse|warehouse)"})),database:R(),schema:g().optional().describe(d.of({label:"Schema",optionRegex:"(snowflake.?schema|^SCHEMA$|_SCHEMA$)"})),role:g().optional().describe(d.of({label:"Role"})),authType:K("type",[_({type:x("Password"),username:T(),password:E(),enable_mfa:L().default(!1).describe(d.of({label:"Enable MFA (Duo Push)"}))}),_({type:x("SSO (Browser)"),username:T()}),_({type:x("Key Pair"),username:T(),private_key_path:g().nonempty().describe(d.of({label:"Private Key Path",placeholder:"/path/to/rsa_key.p8"})),private_key_passphrase:g().optional().describe(d.of({label:"Private Key Passphrase",inputType:"password",optionRegex:"(passphrase|private.?key.?passphrase|key.?passphrase)"}))}),_({type:x("OAuth / PAT"),token:Y("Token",!0)})]).default({type:"Password",username:"username",enable_mfa:!1}).describe(d.of({special:"tabs"}))}).describe(d.of({direction:"two-columns"})),Oe=_({type:x("bigquery"),project:g().nonempty().describe(d.of({label:"Project ID",optionRegex:"(bigquery.?project|gcp.?project|google.?cloud.?project|project.?id)"})),dataset:g().nonempty().describe(d.of({label:"Dataset",optionRegex:"(bigquery.?dataset|dataset)"})),credentials_json:g().describe(d.of({label:"Credentials JSON",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),Be=_({type:x("clickhouse_connect"),host:I(),port:O(8123).optional(),username:T(),password:E(),secure:L().default(!1).describe(d.of({label:"Use HTTPs"})),proxy_path:g().optional().describe(d.of({label:"Proxy Path",placeholder:"/clickhouse"}))}).describe(d.of({direction:"two-columns"})),Ue=_({type:x("timeplus"),host:I().default("localhost"),port:O(8123).optional(),username:T().default("default"),password:E().default("")}).describe(d.of({direction:"two-columns"})),Ke=_({type:x("chdb"),database:R().describe(d.of({label:"Database Path"})),read_only:Le()}).describe(d.of({direction:"two-columns"})),He=_({type:x("trino"),host:I(),port:O(8080),database:R(),schema:Ss().optional(),username:T(),password:E(),async_support:L().default(!1).describe(d.of({label:"Async Support"}))}).describe(d.of({direction:"two-columns"})),Me=_({type:x("iceberg"),name:g().describe(d.of({label:"Catalog Name"})),catalog:K("type",[_({type:x("REST"),warehouse:X(),uri:g().optional().describe(d.of({label:"URI",placeholder:"https://",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"})),token:Y()}),_({type:x("SQL"),warehouse:X(),uri:g().optional().describe(d.of({label:"URI",placeholder:"jdbc:iceberg://host:port/database",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"}))}),_({type:x("Hive"),warehouse:X(),uri:oe()}),_({type:x("Glue"),warehouse:X(),uri:oe()}),_({type:x("DynamoDB"),"dynamodb.profile-name":g().optional().describe(d.of({label:"Profile Name"})),"dynamodb.region":g().optional().describe(d.of({label:"Region"})),"dynamodb.access-key-id":g().optional().describe(d.of({label:"Access Key ID",inputType:"password",optionRegex:"(access.?key.?id|aws.?access.?key)"})),"dynamodb.secret-access-key":g().optional().describe(d.of({label:"Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|aws.?secret)"})),"dynamodb.session-token":g().optional().describe(d.of({label:"Session Token",inputType:"password",optionRegex:"(session.?token|aws.?session)"}))})]).default({type:"REST",token:void 0}).describe(d.of({special:"tabs"}))}),We=_({type:x("datafusion"),sessionContext:L().optional().describe(d.of({label:"Use Session Context"}))}),Ge=_({type:x("pyspark"),host:I().optional(),port:O().optional()}),Ve=_({type:x("redshift"),host:I(),port:O(5439),connectionType:K("type",[_({type:x("IAM credentials"),region:g().describe(d.of({label:"Region"})),aws_access_key_id:g().nonempty().describe(d.of({label:"AWS Access Key ID",inputType:"password",optionRegex:"(access.?key.?id|aws.?access.?key)"})),aws_secret_access_key:g().nonempty().describe(d.of({label:"AWS Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|aws.?secret)"})),aws_session_token:g().optional().describe(d.of({label:"AWS Session Token",inputType:"password",optionRegex:"(session.?token|aws.?session)"}))}),_({type:x("DB credentials"),user:T(),password:E()})]).default({type:"IAM credentials",aws_access_key_id:"",aws_secret_access_key:"",region:""}),database:R()}).describe(d.of({direction:"two-columns"})),ze=_({type:x("databricks"),access_token:Y("Access Token",!0),server_hostname:I("Server Hostname"),http_path:oe("HTTP Path",!0),catalog:g().optional().describe(d.of({label:"Catalog"})),schema:g().optional().describe(d.of({label:"Schema"}))}).describe(d.of({direction:"two-columns"})),Qe=_({type:x("supabase"),host:I(),port:O(5432).optional(),database:R(),username:T(),password:E(),disable_client_pooling:L().default(!1).describe(d.of({label:"Disable Client-Side Pooling"}))}).describe(d.of({direction:"two-columns"})),ks=K("type",[De,Ie,Pe,qe,Fe,Ee,Oe,Be,Ue,Ke,He,Me,We,Ge,Ve,ze,Qe]),Je={sqlmodel:"SQLModel",sqlalchemy:"SQLAlchemy",duckdb:"DuckDB",clickhouse_connect:"ClickHouse Connect",chdb:"chDB",pyiceberg:"PyIceberg",ibis:"Ibis",motherduck:"MotherDuck",redshift:"Redshift",databricks:"Databricks"};var $=class{constructor(t,e,s){this.connection=t,this.orm=e,this.secrets=s}get imports(){let t=new Set(this.generateImports());switch(this.orm){case"sqlalchemy":t.add("import sqlalchemy");break;case"sqlmodel":t.add("import sqlmodel");break;case"duckdb":t.add("import duckdb");break;case"ibis":t.add("import ibis");break}return t}},ws=t=>`_${t}`,js=class{constructor(){G(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e,s){if(t=ws(t),F(e)){let n=je(e),r=s?`os.environ.get("${n}", "${s}")`:`os.environ.get("${n}")`;return this.secrets[t]=r,t}if(s!=null){let n=`os.environ.get("${e}", "${s}")`;return this.secrets[t]=n,t}return typeof e=="number"||typeof e=="number"?`${e}`:typeof e=="boolean"?H(e):e?`"${e}"`:""}printInFString(t,e,s){if(e===void 0)return"";if(typeof e=="number")return`${e}`;if(typeof e=="boolean")return H(e);let n=this.print(t,e,s);return n.startsWith('"')&&n.endsWith('"')?n.slice(1,-1):`{${n}}`}printPassword(t,e,s,n){let r=s?this.printInFString.bind(this):this.print.bind(this);return F(t)?r(n||"password",t):r(n||"password",e,t)}getSecrets(){return this.secrets}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
2
+ `)}},vs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'sslmode': 'require'}":"",e=this.secrets.printPassword(this.connection.password,"POSTGRES_PASSWORD",!0);return y(`
3
3
  DATABASE_URL = f"postgresql://${this.secrets.printInFString("username",this.connection.username)}:${e}@${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}/${this.secrets.printInFString("database",this.connection.database)}"
4
4
  engine = ${this.orm}.create_engine(DATABASE_URL${t})
5
- `)}},vs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'ssl': {'ssl-mode': 'preferred'}}":"",e=this.secrets.printPassword(this.connection.password,"MYSQL_PASSWORD",!0),s=this.secrets.printInFString("database",this.connection.database);return y(`
5
+ `)}},$s=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'ssl': {'ssl-mode': 'preferred'}}":"",e=this.secrets.printPassword(this.connection.password,"MYSQL_PASSWORD",!0),s=this.secrets.printInFString("database",this.connection.database);return y(`
6
6
  DATABASE_URL = f"mysql+pymysql://${this.secrets.printInFString("username",this.connection.username)}:${e}@${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}/${s}"
7
7
  engine = ${this.orm}.create_engine(DATABASE_URL${t})
8
- `)}},$s=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.database?this.secrets.printInFString("database",this.connection.database):"";return y(`
8
+ `)}},Cs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.database?this.secrets.printInFString("database",this.connection.database):"";return y(`
9
9
  ${t.startsWith("{")&&t.endsWith("}")?`DATABASE_URL = f"sqlite:///${t}"`:`DATABASE_URL = "sqlite:///${t}"`}
10
10
  engine = ${this.orm}.create_engine(DATABASE_URL)
11
- `)}},Cs=class extends ${generateImports(){return["from snowflake.sqlalchemy import URL"]}generateConnectionCode(){let{authType:t,account:e,database:s,warehouse:n,schema:r,role:a}=this.connection,c={account:this.secrets.print("account",e),database:this.secrets.print("database",s),warehouse:n?this.secrets.print("warehouse",n):void 0,schema:r?this.secrets.print("schema",r):void 0,role:a?this.secrets.print("role",a):void 0};switch(t.type){case"Password":{let i=this.secrets.printPassword(t.password,"SNOWFLAKE_PASSWORD",!1),l={...c,user:this.secrets.print("user",t.username),password:i};return t.enable_mfa?y(`
11
+ `)}},As=class extends ${generateImports(){return["from snowflake.sqlalchemy import URL"]}generateConnectionCode(){let{authType:t,account:e,database:s,warehouse:n,schema:r,role:a}=this.connection,c={account:this.secrets.print("account",e),database:this.secrets.print("database",s),warehouse:n?this.secrets.print("warehouse",n):void 0,schema:r?this.secrets.print("schema",r):void 0,role:a?this.secrets.print("role",a):void 0};switch(t.type){case"Password":{let i=this.secrets.printPassword(t.password,"SNOWFLAKE_PASSWORD",!1),l={...c,user:this.secrets.print("user",t.username),password:i};return t.enable_mfa?y(`
12
12
  engine = ${this.orm}.create_engine(
13
13
  URL(
14
14
  ${B(l,p=>` ${p}`)},
@@ -30,7 +30,7 @@ ${B(i,l=>` ${l}`)},
30
30
  `)}case"Key Pair":{let i={...c,user:this.secrets.print("user",t.username)},l=this.secrets.print("private_key_path",t.private_key_path),p=t.private_key_passphrase?this.secrets.printPassword(t.private_key_passphrase,"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE",!1):void 0,h=[' "authenticator": "SNOWFLAKE_JWT"',` "private_key_file": ${l}`];return p&&h.push(` "private_key_file_pwd": ${p}`),y(`
31
31
  engine = ${this.orm}.create_engine(
32
32
  URL(
33
- ${B(i,u=>` ${u}`)},
33
+ ${B(i,m=>` ${m}`)},
34
34
  ),
35
35
  connect_args={
36
36
  ${h.join(`,
@@ -43,30 +43,30 @@ ${h.join(`,
43
43
  ${B(l,p=>` ${p}`)},
44
44
  )
45
45
  )
46
- `)}default:W(t)}}},As=class extends ${get credential(){return ee(this.connection.credentials_json,t=>this.secrets.print("credentials_json",t))}generateImports(){return this.credential.kind==="path"?[]:["import json"]}generateConnectionCode(){let t=this.secrets.printInFString("project",this.connection.project),e=this.secrets.printInFString("dataset",this.connection.dataset),s=this.credential;return s.kind==="path"?y(`
46
+ `)}default:M(t)}}},Ns=class extends ${get credential(){return ee(this.connection.credentials_json,t=>this.secrets.print("credentials_json",t))}generateImports(){return this.credential.kind==="path"?[]:["import json"]}generateConnectionCode(){let t=this.secrets.printInFString("project",this.connection.project),e=this.secrets.printInFString("dataset",this.connection.dataset),s=this.credential;return s.kind==="path"?y(`
47
47
  engine = ${this.orm}.create_engine(f"bigquery://${t}/${e}", credentials_path="${se(s.path)}")
48
48
  `):y(`
49
49
  credentials = ${s.expr}
50
50
  engine = ${this.orm}.create_engine(f"bigquery://${t}/${e}", credentials_info=credentials)
51
- `)}},Ns=class extends ${generateImports(){return[]}generateConnectionCode(){return y(`
51
+ `)}},Rs=class extends ${generateImports(){return[]}generateConnectionCode(){return y(`
52
52
  DATABASE_URL = "${this.secrets.printInFString("database",this.connection.database||":memory:")}"
53
- engine = ${this.orm}.connect(DATABASE_URL, read_only=${M(this.connection.read_only)})
54
- `)}},Rs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printInFString("database",this.connection.database);return this.connection.token?y(`
53
+ engine = ${this.orm}.connect(DATABASE_URL, read_only=${H(this.connection.read_only)})
54
+ `)}},Ts=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printInFString("database",this.connection.database);return this.connection.token?y(`
55
55
  conn = duckdb.connect("md:${t}", config={"motherduck_token": ${this.secrets.printPassword(this.connection.token,"MOTHERDUCK_TOKEN",!1)}})
56
56
  `):y(`
57
57
  conn = duckdb.connect("md:${t}")
58
- `)}},Ts=class extends ${generateImports(){return["import clickhouse_connect"]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"CLICKHOUSE_PASSWORD",!1),e={host:this.secrets.print("host",this.connection.host),user:this.secrets.print("user",this.connection.username),secure:this.secrets.print("secure",this.connection.secure),port:this.connection.port?this.secrets.print("port",this.connection.port):void 0,password:this.connection.password?t:void 0,proxy_path:this.connection.proxy_path?this.secrets.print("proxy_path",this.connection.proxy_path):void 0};return y(`
58
+ `)}},Ls=class extends ${generateImports(){return["import clickhouse_connect"]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"CLICKHOUSE_PASSWORD",!1),e={host:this.secrets.print("host",this.connection.host),user:this.secrets.print("user",this.connection.username),secure:this.secrets.print("secure",this.connection.secure),port:this.connection.port?this.secrets.print("port",this.connection.port):void 0,password:this.connection.password?t:void 0,proxy_path:this.connection.proxy_path?this.secrets.print("proxy_path",this.connection.proxy_path):void 0};return y(`
59
59
  engine = ${this.orm}.get_client(
60
60
  ${B(e,s=>` ${s}`)},
61
61
  )
62
- `)}},Ls=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"TIMEPLUS_PASSWORD",!0);return y(`
62
+ `)}},Ds=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"TIMEPLUS_PASSWORD",!0);return y(`
63
63
  DATABASE_URL = f"timeplus://${this.secrets.printInFString("username",this.connection.username)}:${t}@${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}"
64
64
  engine = ${this.orm}.create_engine(DATABASE_URL)
65
- `)}},Ds=class extends ${generateImports(){return["import chdb"]}generateConnectionCode(){let t=this.secrets.print("database",this.connection.database)||'""';return y(`
66
- engine = ${this.orm}.connect(${t}, read_only=${M(this.connection.read_only)})
67
- `)}},Is=class extends ${generateImports(){return this.connection.async_support?["import aiotrino"]:["import trino.sqlalchemy"]}generateConnectionCode(){let t=this.connection.async_support?"aiotrino":"trino",e=this.connection.schema?`/${this.connection.schema}`:"",s=this.secrets.printInFString("username",this.connection.username),n=this.secrets.printInFString("host",this.connection.host),r=this.secrets.printInFString("port",this.connection.port),a=this.secrets.printInFString("database",this.connection.database),c=this.secrets.printPassword(this.connection.password,"TRINO_PASSWORD",!0);return y(`
65
+ `)}},Is=class extends ${generateImports(){return["import chdb"]}generateConnectionCode(){let t=this.secrets.print("database",this.connection.database)||'""';return y(`
66
+ engine = ${this.orm}.connect(${t}, read_only=${H(this.connection.read_only)})
67
+ `)}},Ps=class extends ${generateImports(){return this.connection.async_support?["import aiotrino"]:["import trino.sqlalchemy"]}generateConnectionCode(){let t=this.connection.async_support?"aiotrino":"trino",e=this.connection.schema?`/${this.connection.schema}`:"",s=this.secrets.printInFString("username",this.connection.username),n=this.secrets.printInFString("host",this.connection.host),r=this.secrets.printInFString("port",this.connection.port),a=this.secrets.printInFString("database",this.connection.database),c=this.secrets.printPassword(this.connection.password,"TRINO_PASSWORD",!0);return y(`
68
68
  engine = ${this.orm}.create_engine(f"${t}://${s}:${c}@${n}:${r}/${a}${e}")
69
- `)}},Ps=class extends ${getCatalogOptions(){return Object.fromEntries(Object.entries(this.connection.catalog).filter(([t,e])=>e!=null&&e!==""&&t!=="type"))}generateImports(){if(Object.keys(this.getCatalogOptions()).length===0)return["from pyiceberg.catalog import load_catalog"];switch(this.connection.catalog.type){case"REST":return["from pyiceberg.catalog.rest import RestCatalog"];case"SQL":return["from pyiceberg.catalog.sql import SqlCatalog"];case"Hive":return["from pyiceberg.catalog.hive import HiveCatalog"];case"Glue":return["from pyiceberg.catalog.glue import GlueCatalog"];case"DynamoDB":return["from pyiceberg.catalog.dynamodb import DynamoDBCatalog"];default:W(this.connection.catalog)}}generateConnectionCode(){let t=`"${this.connection.name}"`,e=this.getCatalogOptions();if(Object.keys(e).length===0)return`catalog = load_catalog(${t})`;for(let[n,r]of Object.entries(e))E(r)?e[n]=this.secrets.print(n,r):typeof r=="string"&&(e[n]=`"${r}"`);let s=Ms(e,n=>` ${n}`);switch(this.connection.catalog.type){case"REST":return y(`
69
+ `)}},qs=class extends ${getCatalogOptions(){return Object.fromEntries(Object.entries(this.connection.catalog).filter(([t,e])=>e!=null&&e!==""&&t!=="type"))}generateImports(){if(Object.keys(this.getCatalogOptions()).length===0)return["from pyiceberg.catalog import load_catalog"];switch(this.connection.catalog.type){case"REST":return["from pyiceberg.catalog.rest import RestCatalog"];case"SQL":return["from pyiceberg.catalog.sql import SqlCatalog"];case"Hive":return["from pyiceberg.catalog.hive import HiveCatalog"];case"Glue":return["from pyiceberg.catalog.glue import GlueCatalog"];case"DynamoDB":return["from pyiceberg.catalog.dynamodb import DynamoDBCatalog"];default:M(this.connection.catalog)}}generateConnectionCode(){let t=`"${this.connection.name}"`,e=this.getCatalogOptions();if(Object.keys(e).length===0)return`catalog = load_catalog(${t})`;for(let[n,r]of Object.entries(e))F(r)?e[n]=this.secrets.print(n,r):typeof r=="string"&&(e[n]=`"${r}"`);let s=Ms(e,n=>` ${n}`);switch(this.connection.catalog.type){case"REST":return y(`
70
70
  catalog = RestCatalog(
71
71
  ${t},
72
72
  **{
@@ -101,7 +101,7 @@ ${s}
101
101
  ${s}
102
102
  },
103
103
  )
104
- `);default:W(this.connection.catalog)}}},qs=class extends ${generateImports(){return["from datafusion import SessionContext"]}generateConnectionCode(){return this.connection.sessionContext?y(`
104
+ `);default:M(this.connection.catalog)}}},Fs=class extends ${generateImports(){return["from datafusion import SessionContext"]}generateConnectionCode(){return this.connection.sessionContext?y(`
105
105
  ctx = SessionContext()
106
106
  # Sample table
107
107
  _ = ctx.from_pydict({"a": [1, 2, 3]}, "my_table")
@@ -114,7 +114,7 @@ ${s}
114
114
  con = ibis.pyspark.connect(session)
115
115
  `):y(`
116
116
  con = ibis.pyspark.connect()
117
- `)}},Fs=class extends ${generateImports(){return["import redshift_connector"]}generateConnectionCode(){let t=this.secrets.print("host",this.connection.host),e=this.secrets.print("port",this.connection.port),s=this.secrets.print("database",this.connection.database);if(this.connection.connectionType.type==="IAM credentials"){let a=this.secrets.print("aws_access_key_id",this.connection.connectionType.aws_access_key_id),c=this.secrets.print("aws_secret_access_key",this.connection.connectionType.aws_secret_access_key),i=this.connection.connectionType.aws_session_token?this.secrets.print("aws_session_token",this.connection.connectionType.aws_session_token):void 0;return y(`
117
+ `)}},Os=class extends ${generateImports(){return["import redshift_connector"]}generateConnectionCode(){let t=this.secrets.print("host",this.connection.host),e=this.secrets.print("port",this.connection.port),s=this.secrets.print("database",this.connection.database);if(this.connection.connectionType.type==="IAM credentials"){let a=this.secrets.print("aws_access_key_id",this.connection.connectionType.aws_access_key_id),c=this.secrets.print("aws_secret_access_key",this.connection.connectionType.aws_secret_access_key),i=this.connection.connectionType.aws_session_token?this.secrets.print("aws_session_token",this.connection.connectionType.aws_session_token):void 0;return y(`
118
118
  con = redshift_connector.connect(
119
119
  ${B({iam:!0,host:t,port:e,region:`"${this.connection.connectionType.region}"`,database:s,access_key_id:a,secret_access_key:c,...i&&{session_token:i}},l=>` ${l}`)},
120
120
  )
@@ -122,7 +122,7 @@ ${B({iam:!0,host:t,port:e,region:`"${this.connection.connectionType.region}"`,da
122
122
  con = redshift_connector.connect(
123
123
  ${B({host:t,port:e,database:s,...n&&{user:n},...r&&{password:r}},a=>` ${a}`)},
124
124
  )
125
- `)}},Os=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.orm!=="ibis",e=this.secrets.printPassword(this.connection.access_token,"DATABRICKS_ACCESS_TOKEN",t,"access_token"),s=this.secrets.printPassword(this.connection.server_hostname,"DATABRICKS_SERVER_HOSTNAME",t,"server_hostname"),n=this.secrets.printPassword(this.connection.http_path,"DATABRICKS_HTTP_PATH",t,"http_path"),r=this.connection.catalog?this.secrets.printInFString("catalog",this.connection.catalog):void 0,a=this.connection.schema?this.secrets.printInFString("schema",this.connection.schema):void 0,c=`databricks://token:${e}@${s}?http_path=${n}`;return r&&(c+=`&catalog=${r}`),a&&(c+=`&schema=${a}`),this.orm==="ibis"?y(`
125
+ `)}},Bs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.orm!=="ibis",e=this.secrets.printPassword(this.connection.access_token,"DATABRICKS_ACCESS_TOKEN",t,"access_token"),s=this.secrets.printPassword(this.connection.server_hostname,"DATABRICKS_SERVER_HOSTNAME",t,"server_hostname"),n=this.secrets.printPassword(this.connection.http_path,"DATABRICKS_HTTP_PATH",t,"http_path"),r=this.connection.catalog?this.secrets.printInFString("catalog",this.connection.catalog):void 0,a=this.connection.schema?this.secrets.printInFString("schema",this.connection.schema):void 0,c=`databricks://token:${e}@${s}?http_path=${n}`;return r&&(c+=`&catalog=${r}`),a&&(c+=`&schema=${a}`),this.orm==="ibis"?y(`
126
126
  engine = ibis.databricks.connect(
127
127
  server_hostname=${s},
128
128
  http_path=${n},${r?`
@@ -133,19 +133,19 @@ ${B({host:t,port:e,database:s,...n&&{user:n},...r&&{password:r}},a=>` $
133
133
  `):y(`
134
134
  DATABASE_URL = f"${c}"
135
135
  engine = ${this.orm}.create_engine(DATABASE_URL)
136
- `)}},Bs=class extends ${generateImports(){return this.connection.disable_client_pooling?["from sqlalchemy.pool import NullPool"]:[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"SUPABASE_PASSWORD",!0),e=this.secrets.printInFString("username",this.connection.username),s=this.secrets.printInFString("host",this.connection.host),n=this.secrets.printInFString("port",this.connection.port),r=this.secrets.printInFString("database",this.connection.database),a=this.connection.disable_client_pooling?", poolclass=NullPool":"";return y(`
136
+ `)}},Us=class extends ${generateImports(){return this.connection.disable_client_pooling?["from sqlalchemy.pool import NullPool"]:[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"SUPABASE_PASSWORD",!0),e=this.secrets.printInFString("username",this.connection.username),s=this.secrets.printInFString("host",this.connection.host),n=this.secrets.printInFString("port",this.connection.port),r=this.secrets.printInFString("database",this.connection.database),a=this.connection.disable_client_pooling?", poolclass=NullPool":"";return y(`
137
137
  DATABASE_URL = f"postgresql+psycopg2://${e}:${t}@${s}:${n}/${r}?sslmode=require"
138
138
  engine = ${this.orm}.create_engine(DATABASE_URL${a})
139
- `)}},Us=class{constructor(){G(this,"secrets",new ws)}createGenerator(t,e){switch(t.type){case"postgres":return new js(t,e,this.secrets);case"mysql":return new vs(t,e,this.secrets);case"sqlite":return new $s(t,e,this.secrets);case"snowflake":return new Cs(t,e,this.secrets);case"bigquery":return new As(t,e,this.secrets);case"duckdb":return new Ns(t,e,this.secrets);case"motherduck":return new Rs(t,e,this.secrets);case"clickhouse_connect":return new Ts(t,e,this.secrets);case"timeplus":return new Ls(t,e,this.secrets);case"chdb":return new Ds(t,e,this.secrets);case"trino":return new Is(t,e,this.secrets);case"iceberg":return new Ps(t,e,this.secrets);case"datafusion":return new qs(t,e,this.secrets);case"pyspark":return new Es(t,e,this.secrets);case"redshift":return new Fs(t,e,this.secrets);case"databricks":return new Os(t,e,this.secrets);case"supabase":return new Bs(t,e,this.secrets);default:W(t)}}};function Ks(t,e){if(!(e in Je))throw Error(`Unsupported library: ${e}`);Ss.parse(t);let s=new Us,n=s.createGenerator(t,e),r=n.generateConnectionCode(),a=s.secrets,c=[...new Set([...a.imports,...n.imports])].toSorted();c.push("");let i=a.formatSecrets();return i&&c.push(i),c.push(r.trim()),c.join(`
140
- `)}function M(t){return t.toString().charAt(0).toUpperCase()+t.toString().slice(1)}function B(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,n])=>e(typeof n=="boolean"?`${s}=${M(n)}`:`${s}=${n}`)).join(`,
141
- `)}function Ms(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,n])=>{let r=`"${s}"`;return e(typeof n=="boolean"?`${r}: ${M(n)}`:`${r}: ${n}`)}).join(`,
142
- `)}var Ye=U(),Xe=[{name:"PostgreSQL",schema:De,color:"#336791",logo:"postgres",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"MySQL",schema:Ie,color:"#00758F",logo:"mysql",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"SQLite",schema:Pe,color:"#003B57",logo:"sqlite",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DuckDB",schema:qe,color:"#FFD700",logo:"duckdb",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"MotherDuck",schema:Ee,color:"#ff9538",logo:"motherduck",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"Snowflake",schema:Fe,color:"#29B5E8",logo:"snowflake",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse",schema:Be,color:"#2C2C1D",logo:"clickhouse",connectionLibraries:{libraries:["clickhouse_connect"],preferred:"clickhouse_connect"}},{name:"Timeplus",schema:Ue,color:"#B83280",logo:"timeplus",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"BigQuery",schema:Oe,color:"#4285F4",logo:"bigquery",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse Embedded",schema:Ke,color:"#f2b611",logo:"clickhouse",connectionLibraries:{libraries:["chdb"],preferred:"chdb"}},{name:"Trino",schema:Me,color:"#d466b6",logo:"trino",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DataFusion",schema:He,color:"#202A37",logo:"datafusion",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"PySpark",schema:Ge,color:"#1C5162",logo:"pyspark",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"Redshift",schema:Ve,color:"#522BAE",logo:"redshift",connectionLibraries:{libraries:["redshift"],preferred:"redshift"}},{name:"Databricks",schema:ze,color:"#c41e0c",logo:"databricks",connectionLibraries:{libraries:["sqlalchemy","sqlmodel","ibis"],preferred:"sqlalchemy"}},{name:"Supabase",schema:Qe,color:"#238F5F",logo:"supabase",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}}],Ze=[{name:"Iceberg",schema:We,color:"#000000",logo:"iceberg",connectionLibraries:{libraries:["pyiceberg"],preferred:"pyiceberg"}}],Ws=[...Xe,...Ze],Hs=t=>{let e=(0,Ye.c)(12),{onSelect:s}=t,n;e[0]===s?n=e[1]:(n=Xe.map(p=>{let{name:h,schema:u,color:b,logo:g}=p;return(0,o.jsx)(ne,{name:h,color:b,icon:(0,o.jsx)(ce,{name:g,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(u)},h)}),e[0]=s,e[1]=n);let r;e[2]===n?r=e[3]:(r=(0,o.jsx)(re,{children:n}),e[2]=n,e[3]=r);let a;e[4]===Symbol.for("react.memo_cache_sentinel")?(a=(0,o.jsxs)("h4",{className:"font-semibold text-muted-foreground text-lg flex items-center gap-4 my-2",children:["Data Catalogs",(0,o.jsx)("hr",{className:"flex-1"})]}),e[4]=a):a=e[4];let c;e[5]===s?c=e[6]:(c=Ze.map(p=>{let{name:h,schema:u,color:b,logo:g}=p;return(0,o.jsx)(ne,{name:h,color:b,icon:(0,o.jsx)(ce,{name:g,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(u)},h)}),e[5]=s,e[6]=c);let i;e[7]===c?i=e[8]:(i=(0,o.jsx)(re,{children:c}),e[7]=c,e[8]=i);let l;return e[9]!==r||e[10]!==i?(l=(0,o.jsxs)(o.Fragment,{children:[r,a,i]}),e[9]=r,e[10]=i,e[11]=l):l=e[11],l};const Gs=t=>{let e=(0,Ye.c)(13),{onSubmit:s,header:n}=t,[r,a]=(0,D.useState)(null);if(!r){let b;e[0]===Symbol.for("react.memo_cache_sentinel")?(b=(0,o.jsx)("div",{children:(0,o.jsx)(Hs,{onSelect:a})}),e[0]=b):b=e[0];let g;return e[1]===n?g=e[2]:(g=(0,o.jsxs)(o.Fragment,{children:[n,b]}),e[1]=n,e[2]=g),g}let c;e[3]===r?c=e[4]:(c=Ws.find(b=>b.schema===r),e[3]=r,e[4]=c);let i=c==null?void 0:c.connectionLibraries,l;e[5]===(i==null?void 0:i.libraries)?l=e[6]:(l=(i==null?void 0:i.libraries)??[],e[5]=i==null?void 0:i.libraries,e[6]=l);let p=(i==null?void 0:i.preferred)??"sqlalchemy",h;e[7]===Symbol.for("react.memo_cache_sentinel")?(h=()=>a(null),e[7]=h):h=e[7];let u;return e[8]!==s||e[9]!==r||e[10]!==l||e[11]!==p?(u=(0,o.jsx)(Te,{schema:r,libraries:l,preferredLibrary:p,displayNames:Je,libraryLabel:"Preferred connection library",generateCode:Vs,onSubmit:s,onBack:h}),e[8]=s,e[9]=r,e[10]=l,e[11]=p,e[12]=u):u=e[12],u};function Vs(t,e){return Ks(t,e)}var zs=U();async function Qs(){return(await gt.request({})).sources}function Js(){let t=(0,zs.c)(1),e;return t[0]===Symbol.for("react.memo_cache_sentinel")?(e=[],t[0]=e):e=t[0],Se(Qs,e)}var ae=U();const Ys=t=>{let e=(0,ae.c)(13),{className:s,sources:n,onAdd:r}=t;if(n.length===0)return null;let a;e[0]===s?a=e[1]:(a=V("rounded-full bg-[linear-gradient(135deg,var(--blue-2),var(--purple-3))] px-3 py-2",s),e[0]=s,e[1]=a);let c;e[2]===Symbol.for("react.memo_cache_sentinel")?(c=(0,o.jsxs)("div",{className:"mr-1 flex items-center gap-1.5",children:[(0,o.jsx)(Bt,{className:"h-3.5 w-3.5 text-(--blue-9)"}),(0,o.jsx)("h3",{id:"quick-add-data-sources-title",className:"text-sm",children:"Quick add"})]}),e[2]=c):c=e[2];let i;if(e[3]!==r||e[4]!==n){let h;e[6]===r?h=e[7]:(h=u=>(0,o.jsx)(he,{side:"bottom",content:(0,o.jsx)(Xs,{source:u}),children:(0,o.jsxs)("button",{type:"button","aria-label":`Add ${u.displayName} connection`,className:"inline-flex items-center gap-1 rounded-full border border-border/60 bg-background px-2.5 py-1 text-xs font-medium text-foreground/90 transition-colors hover:border-(--blue-7) focus:outline-hidden focus:ring-2 focus:ring-ring focus:ring-offset-2",onClick:()=>r(u),children:[(0,o.jsx)(Ot,{className:"h-3 w-3 text-muted-foreground"}),u.displayName]})},u.id),e[6]=r,e[7]=h),i=n.map(h),e[3]=r,e[4]=n,e[5]=i}else i=e[5];let l;e[8]===i?l=e[9]:(l=(0,o.jsxs)("div",{className:"flex flex-wrap items-center gap-2",children:[c,(0,o.jsx)(It,{delayDuration:200,children:i})]}),e[8]=i,e[9]=l);let p;return e[10]!==a||e[11]!==l?(p=(0,o.jsx)("section",{"aria-labelledby":"quick-add-data-sources-title",className:a,children:l}),e[10]=a,e[11]=l,e[12]=p):p=e[12],p};var Xs=t=>{let e=(0,ae.c)(17),{source:s}=t,n;e[0]===s.displayName?n=e[1]:(n=(0,o.jsx)("div",{className:"font-medium",children:s.displayName}),e[0]=s.displayName,e[1]=n);let r;e[2]===s.origins?r=e[3]:(r=s.origins.map(er).join(" and "),e[2]=s.origins,e[3]=r);let a;e[4]===r?a=e[5]:(a=(0,o.jsxs)("div",{className:"text-xs text-muted-foreground",children:["Detected from"," ",r]}),e[4]=r,e[5]=a);let c;e[6]!==n||e[7]!==a?(c=(0,o.jsxs)("div",{children:[n,a]}),e[6]=n,e[7]=a,e[8]=c):c=e[8];let i;e[9]===s.configuration?i=e[10]:(i=s.configuration.map(tr),e[9]=s.configuration,e[10]=i);let l;e[11]===i?l=e[12]:(l=(0,o.jsx)("dl",{className:"grid grid-cols-[auto_1fr] gap-x-3 gap-y-1 text-xs",children:i}),e[11]=i,e[12]=l);let p;e[13]===Symbol.for("react.memo_cache_sentinel")?(p=(0,o.jsx)("div",{className:"text-xs text-muted-foreground",children:"Click to add a configured cell."}),e[13]=p):p=e[13];let h;return e[14]!==c||e[15]!==l?(h=(0,o.jsxs)("div",{className:"min-w-64 space-y-2 py-1",children:[c,l,p]}),e[14]=c,e[15]=l,e[16]=h):h=e[16],h};const Zs=t=>{let e=(0,ae.c)(9),{onSubmit:s,className:n}=t,r=Re(),{data:a}=Js(),c;e[0]===a?c=e[1]:(c=a??[],e[0]=a,e[1]=c);let i;e[2]!==r||e[3]!==s?(i=p=>{r(p.code),s()},e[2]=r,e[3]=s,e[4]=i):i=e[4];let l;return e[5]!==n||e[6]!==c||e[7]!==i?(l=(0,o.jsx)(Ys,{className:n,sources:c,onAdd:i}),e[5]=n,e[6]=c,e[7]=i,e[8]=l):l=e[8],l};function er(t){return t.label}function tr(t){return(0,o.jsxs)("div",{className:"contents",children:[(0,o.jsx)("dt",{className:"text-muted-foreground",children:t.field}),(0,o.jsx)("dd",{children:(0,o.jsx)("code",{children:t.value.kind==="environment-variable"?`os.environ["${t.value.name}"]`:t.value.value})})]},t.value.kind==="environment-variable"?t.value.name:`${t.field}:${t.value.value}`)}var et=()=>K("type",[_({type:x("Access keys"),access_key_id:f().optional().describe(d.of({label:"Access Key ID",description:"Leave empty to use the default credential chain",inputType:"password",optionRegex:"(access.?key.?id|object.?storage.?key|aws.?access.?key)"})),secret_access_key:f().optional().describe(d.of({label:"Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|object.?storage.?secret|aws.?secret)"}))}),_({type:x("Container credentials"),container_credentials_full_uri:f().nonempty().describe(d.of({label:"Credentials URI",description:"Endpoint that vends credentials to the container",placeholder:"http://169.254.170.23/v1/credentials",optionRegex:"container.?credentials"})),container_authorization_token_file:f().nonempty().describe(d.of({label:"Authorization Token File",description:"File holding the token sent to the credentials URI",placeholder:"/var/run/secrets/token",optionRegex:"(container.?authorization.?token.?file|container.?auth.?token)"}))})]).default({type:"Access keys"}).describe(d.of({label:"Credentials",special:"tabs"}));const tt=_({type:x("s3"),bucket:f().nonempty().describe(d.of({label:"Bucket",placeholder:"my-bucket",optionRegex:"(bucket|s3.?bucket)"})),region:f().optional().describe(d.of({label:"Region",placeholder:"us-east-1",optionRegex:"(region|aws.?region)"})),endpoint_url:f().optional().describe(d.of({label:"Endpoint URL",description:"Ignored if the AWS_ENDPOINT_URL_S3 environment variable is set",placeholder:"https://s3.amazonaws.com",optionRegex:"(endpoint|s3.?url|s3.?endpoint)"})),allow_http:L().default(!1).describe(d.of({label:"Allow HTTP",description:"Required for plain-http (non-TLS) endpoints"})),auth:et()}).describe(d.of({direction:"two-columns"})),st=_({type:x("gcs"),bucket:f().nonempty().describe(d.of({label:"Bucket",placeholder:"my-bucket",optionRegex:"(bucket|gcs.?bucket|google.?bucket)"})),service_account_key:f().optional().describe(d.of({label:"Service Account Key (JSON)",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),rt=_({type:x("azure"),container:f().nonempty().describe(d.of({label:"Container",placeholder:"my-container"})),account_name:f().nonempty().describe(d.of({label:"Account Name",placeholder:"storageaccount",optionRegex:"(azure.?account|account.?name|storage.?account)"})),account_key:f().optional().describe(d.of({label:"Account Key",inputType:"password",optionRegex:"(azure.?key|account.?key|storage.?key)"}))}).describe(d.of({direction:"two-columns"})),nt=_({type:x("coreweave"),bucket:f().nonempty().describe(d.of({label:"Bucket",placeholder:"bucket-name"})),region:f().nonempty().describe(d.of({label:"Region",placeholder:"US-EAST-04A"})),auth:et()}).describe(d.of({direction:"two-columns"})),ot=_({type:x("gdrive"),credentials_json:f().optional().describe(d.of({label:"Service Account JSON",description:"Leave empty to use browser-based authentication",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),sr=K("type",[tt,st,rt,nt,ot]),rr={obstore:"obstore",fsspec:"fsspec"};var nr=class{constructor(){G(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e){if(e===void 0||e==="")return"";let s=`_${t}`;if(E(e)){let n=je(e);return this.secrets[s]=`os.environ.get("${n}")`,s}return`"${e}"`}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
143
- `)}};function at(t,e){if((t==null?void 0:t.type)==="Container credentials")return[` container_credentials_full_uri=${e.print("container_credentials_full_uri",t.container_credentials_full_uri)},`,` container_authorization_token_file=${e.print("container_authorization_token_file",t.container_authorization_token_file)},`];let s=[];return t!=null&&t.access_key_id&&s.push(` access_key_id=${e.print("access_key_id",t.access_key_id)},`),t!=null&&t.secret_access_key&&s.push(` secret_access_key=${e.print("secret_access_key",t.secret_access_key)},`),s}function or(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import S3Store"]),r=[];return t.region&&r.push(` region=${e.print("region",t.region)},`),r.push(...at(t.auth,e)),t.endpoint_url&&r.push(` endpoint_url=${e.print("endpoint_url",t.endpoint_url)},`),t.allow_http&&r.push(" allow_http=True,"),{imports:n,code:y(`
139
+ `)}},Ks=class{constructor(){G(this,"secrets",new js)}createGenerator(t,e){switch(t.type){case"postgres":return new vs(t,e,this.secrets);case"mysql":return new $s(t,e,this.secrets);case"sqlite":return new Cs(t,e,this.secrets);case"snowflake":return new As(t,e,this.secrets);case"bigquery":return new Ns(t,e,this.secrets);case"duckdb":return new Rs(t,e,this.secrets);case"motherduck":return new Ts(t,e,this.secrets);case"clickhouse_connect":return new Ls(t,e,this.secrets);case"timeplus":return new Ds(t,e,this.secrets);case"chdb":return new Is(t,e,this.secrets);case"trino":return new Ps(t,e,this.secrets);case"iceberg":return new qs(t,e,this.secrets);case"datafusion":return new Fs(t,e,this.secrets);case"pyspark":return new Es(t,e,this.secrets);case"redshift":return new Os(t,e,this.secrets);case"databricks":return new Bs(t,e,this.secrets);case"supabase":return new Us(t,e,this.secrets);default:M(t)}}};function Hs(t,e){if(!(e in Je))throw Error(`Unsupported library: ${e}`);ks.parse(t);let s=new Ks,n=s.createGenerator(t,e),r=n.generateConnectionCode(),a=s.secrets,c=[...new Set([...a.imports,...n.imports])].toSorted();c.push("");let i=a.formatSecrets();return i&&c.push(i),c.push(r.trim()),c.join(`
140
+ `)}function H(t){return t.toString().charAt(0).toUpperCase()+t.toString().slice(1)}function B(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,n])=>e(typeof n=="boolean"?`${s}=${H(n)}`:`${s}=${n}`)).join(`,
141
+ `)}function Ms(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,n])=>{let r=`"${s}"`;return e(typeof n=="boolean"?`${r}: ${H(n)}`:`${r}: ${n}`)}).join(`,
142
+ `)}var Ye=U(),Xe=[{name:"PostgreSQL",schema:De,color:"#336791",logo:"postgres",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"MySQL",schema:Ie,color:"#00758F",logo:"mysql",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"SQLite",schema:Pe,color:"#003B57",logo:"sqlite",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DuckDB",schema:qe,color:"#FFD700",logo:"duckdb",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"MotherDuck",schema:Fe,color:"#ff9538",logo:"motherduck",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"Snowflake",schema:Ee,color:"#29B5E8",logo:"snowflake",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse",schema:Be,color:"#2C2C1D",logo:"clickhouse",connectionLibraries:{libraries:["clickhouse_connect"],preferred:"clickhouse_connect"}},{name:"Timeplus",schema:Ue,color:"#B83280",logo:"timeplus",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"BigQuery",schema:Oe,color:"#4285F4",logo:"bigquery",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse Embedded",schema:Ke,color:"#f2b611",logo:"clickhouse",connectionLibraries:{libraries:["chdb"],preferred:"chdb"}},{name:"Trino",schema:He,color:"#d466b6",logo:"trino",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DataFusion",schema:We,color:"#202A37",logo:"datafusion",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"PySpark",schema:Ge,color:"#1C5162",logo:"pyspark",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"Redshift",schema:Ve,color:"#522BAE",logo:"redshift",connectionLibraries:{libraries:["redshift"],preferred:"redshift"}},{name:"Databricks",schema:ze,color:"#c41e0c",logo:"databricks",connectionLibraries:{libraries:["sqlalchemy","sqlmodel","ibis"],preferred:"sqlalchemy"}},{name:"Supabase",schema:Qe,color:"#238F5F",logo:"supabase",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}}],Ze=[{name:"Iceberg",schema:Me,color:"#000000",logo:"iceberg",connectionLibraries:{libraries:["pyiceberg"],preferred:"pyiceberg"}}],Ws=[...Xe,...Ze],Gs=t=>{let e=(0,Ye.c)(12),{onSelect:s}=t,n;e[0]===s?n=e[1]:(n=Xe.map(p=>{let{name:h,schema:m,color:b,logo:f}=p;return(0,o.jsx)(ne,{name:h,color:b,icon:(0,o.jsx)(ce,{name:f,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(m)},h)}),e[0]=s,e[1]=n);let r;e[2]===n?r=e[3]:(r=(0,o.jsx)(re,{children:n}),e[2]=n,e[3]=r);let a;e[4]===Symbol.for("react.memo_cache_sentinel")?(a=(0,o.jsxs)("h4",{className:"font-semibold text-muted-foreground text-lg flex items-center gap-4 my-2",children:["Data Catalogs",(0,o.jsx)("hr",{className:"flex-1"})]}),e[4]=a):a=e[4];let c;e[5]===s?c=e[6]:(c=Ze.map(p=>{let{name:h,schema:m,color:b,logo:f}=p;return(0,o.jsx)(ne,{name:h,color:b,icon:(0,o.jsx)(ce,{name:f,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(m)},h)}),e[5]=s,e[6]=c);let i;e[7]===c?i=e[8]:(i=(0,o.jsx)(re,{children:c}),e[7]=c,e[8]=i);let l;return e[9]!==r||e[10]!==i?(l=(0,o.jsxs)(o.Fragment,{children:[r,a,i]}),e[9]=r,e[10]=i,e[11]=l):l=e[11],l};const Vs=t=>{let e=(0,Ye.c)(13),{onSubmit:s,header:n}=t,[r,a]=(0,D.useState)(null);if(!r){let b;e[0]===Symbol.for("react.memo_cache_sentinel")?(b=(0,o.jsx)("div",{children:(0,o.jsx)(Gs,{onSelect:a})}),e[0]=b):b=e[0];let f;return e[1]===n?f=e[2]:(f=(0,o.jsxs)(o.Fragment,{children:[n,b]}),e[1]=n,e[2]=f),f}let c;e[3]===r?c=e[4]:(c=Ws.find(b=>b.schema===r),e[3]=r,e[4]=c);let i=c==null?void 0:c.connectionLibraries,l;e[5]===(i==null?void 0:i.libraries)?l=e[6]:(l=(i==null?void 0:i.libraries)??[],e[5]=i==null?void 0:i.libraries,e[6]=l);let p=(i==null?void 0:i.preferred)??"sqlalchemy",h;e[7]===Symbol.for("react.memo_cache_sentinel")?(h=()=>a(null),e[7]=h):h=e[7];let m;return e[8]!==s||e[9]!==r||e[10]!==l||e[11]!==p?(m=(0,o.jsx)(Te,{schema:r,libraries:l,preferredLibrary:p,displayNames:Je,libraryLabel:"Preferred connection library",generateCode:zs,onSubmit:s,onBack:h}),e[8]=s,e[9]=r,e[10]=l,e[11]=p,e[12]=m):m=e[12],m};function zs(t,e){return Hs(t,e)}var Qs=U();async function Js(){return(await _t.request({})).sources}function Ys(){let t=(0,Qs.c)(1),e;return t[0]===Symbol.for("react.memo_cache_sentinel")?(e=[],t[0]=e):e=t[0],Se(Js,e)}var ae=U();const Xs=t=>{let e=(0,ae.c)(13),{className:s,sources:n,onAdd:r}=t;if(n.length===0)return null;let a;e[0]===s?a=e[1]:(a=V("rounded-full bg-[linear-gradient(135deg,var(--blue-2),var(--purple-3))] px-3 py-2",s),e[0]=s,e[1]=a);let c;e[2]===Symbol.for("react.memo_cache_sentinel")?(c=(0,o.jsxs)("div",{className:"mr-1 flex items-center gap-1.5",children:[(0,o.jsx)(Ut,{className:"h-3.5 w-3.5 text-(--blue-9)"}),(0,o.jsx)("h3",{id:"quick-add-data-sources-title",className:"text-sm",children:"Quick add"})]}),e[2]=c):c=e[2];let i;if(e[3]!==r||e[4]!==n){let h;e[6]===r?h=e[7]:(h=m=>(0,o.jsx)(he,{side:"bottom",content:(0,o.jsx)(Zs,{source:m}),children:(0,o.jsxs)("button",{type:"button","aria-label":`Add ${m.displayName} connection`,className:"inline-flex items-center gap-1 rounded-full border border-border/60 bg-background px-2.5 py-1 text-xs font-medium text-foreground/90 transition-colors hover:border-(--blue-7) focus:outline-hidden focus:ring-2 focus:ring-ring focus:ring-offset-2",onClick:()=>r(m),children:[(0,o.jsx)(Bt,{className:"h-3 w-3 text-muted-foreground"}),m.displayName]})},m.id),e[6]=r,e[7]=h),i=n.map(h),e[3]=r,e[4]=n,e[5]=i}else i=e[5];let l;e[8]===i?l=e[9]:(l=(0,o.jsxs)("div",{className:"flex flex-wrap items-center gap-2",children:[c,(0,o.jsx)(Pt,{delayDuration:200,children:i})]}),e[8]=i,e[9]=l);let p;return e[10]!==a||e[11]!==l?(p=(0,o.jsx)("section",{"aria-labelledby":"quick-add-data-sources-title",className:a,children:l}),e[10]=a,e[11]=l,e[12]=p):p=e[12],p};var Zs=t=>{let e=(0,ae.c)(17),{source:s}=t,n;e[0]===s.displayName?n=e[1]:(n=(0,o.jsx)("div",{className:"font-medium",children:s.displayName}),e[0]=s.displayName,e[1]=n);let r;e[2]===s.origins?r=e[3]:(r=s.origins.map(tr).join(" and "),e[2]=s.origins,e[3]=r);let a;e[4]===r?a=e[5]:(a=(0,o.jsxs)("div",{className:"text-xs text-muted-foreground",children:["Detected from"," ",r]}),e[4]=r,e[5]=a);let c;e[6]!==n||e[7]!==a?(c=(0,o.jsxs)("div",{children:[n,a]}),e[6]=n,e[7]=a,e[8]=c):c=e[8];let i;e[9]===s.configuration?i=e[10]:(i=s.configuration.map(sr),e[9]=s.configuration,e[10]=i);let l;e[11]===i?l=e[12]:(l=(0,o.jsx)("dl",{className:"grid grid-cols-[auto_1fr] gap-x-3 gap-y-1 text-xs",children:i}),e[11]=i,e[12]=l);let p;e[13]===Symbol.for("react.memo_cache_sentinel")?(p=(0,o.jsx)("div",{className:"text-xs text-muted-foreground",children:"Click to add a configured cell."}),e[13]=p):p=e[13];let h;return e[14]!==c||e[15]!==l?(h=(0,o.jsxs)("div",{className:"min-w-64 space-y-2 py-1",children:[c,l,p]}),e[14]=c,e[15]=l,e[16]=h):h=e[16],h};const er=t=>{let e=(0,ae.c)(9),{onSubmit:s,className:n}=t,r=Re(),{data:a}=Ys(),c;e[0]===a?c=e[1]:(c=a??[],e[0]=a,e[1]=c);let i;e[2]!==r||e[3]!==s?(i=p=>{r(p.code),s()},e[2]=r,e[3]=s,e[4]=i):i=e[4];let l;return e[5]!==n||e[6]!==c||e[7]!==i?(l=(0,o.jsx)(Xs,{className:n,sources:c,onAdd:i}),e[5]=n,e[6]=c,e[7]=i,e[8]=l):l=e[8],l};function tr(t){return t.label}function sr(t){return(0,o.jsxs)("div",{className:"contents",children:[(0,o.jsx)("dt",{className:"text-muted-foreground",children:t.field}),(0,o.jsx)("dd",{children:(0,o.jsx)("code",{children:t.value.kind==="environment-variable"?`os.environ["${t.value.name}"]`:t.value.value})})]},t.value.kind==="environment-variable"?t.value.name:`${t.field}:${t.value.value}`)}var et=()=>K("type",[_({type:x("Access keys"),access_key_id:g().optional().describe(d.of({label:"Access Key ID",description:"Leave empty to use the default credential chain",inputType:"password",optionRegex:"(access.?key.?id|object.?storage.?key|aws.?access.?key)"})),secret_access_key:g().optional().describe(d.of({label:"Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|object.?storage.?secret|aws.?secret)"}))}),_({type:x("Container credentials"),container_credentials_full_uri:g().nonempty().describe(d.of({label:"Credentials URI",description:"Endpoint that vends credentials to the container",placeholder:"http://169.254.170.23/v1/credentials",optionRegex:"container.?credentials"})),container_authorization_token_file:g().nonempty().describe(d.of({label:"Authorization Token File",description:"File holding the token sent to the credentials URI",placeholder:"/var/run/secrets/token",optionRegex:"(container.?authorization.?token.?file|container.?auth.?token)"}))})]).default({type:"Access keys"}).describe(d.of({label:"Credentials",special:"tabs"}));const tt=_({type:x("s3"),bucket:g().nonempty().describe(d.of({label:"Bucket",placeholder:"my-bucket",optionRegex:"(bucket|s3.?bucket)"})),region:g().optional().describe(d.of({label:"Region",placeholder:"us-east-1",optionRegex:"(region|aws.?region)"})),endpoint_url:g().optional().describe(d.of({label:"Endpoint URL",description:"Ignored if the AWS_ENDPOINT_URL_S3 environment variable is set",placeholder:"https://s3.amazonaws.com",optionRegex:"(endpoint|s3.?url|s3.?endpoint)"})),allow_http:L().default(!1).describe(d.of({label:"Allow HTTP",description:"Required for plain-http (non-TLS) endpoints"})),auth:et()}).describe(d.of({direction:"two-columns"})),st=_({type:x("gcs"),bucket:g().nonempty().describe(d.of({label:"Bucket",placeholder:"my-bucket",optionRegex:"(bucket|gcs.?bucket|google.?bucket)"})),service_account_key:g().optional().describe(d.of({label:"Service Account Key (JSON)",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),rt=_({type:x("azure"),container:g().nonempty().describe(d.of({label:"Container",placeholder:"my-container"})),account_name:g().nonempty().describe(d.of({label:"Account Name",placeholder:"storageaccount",optionRegex:"(azure.?account|account.?name|storage.?account)"})),account_key:g().optional().describe(d.of({label:"Account Key",inputType:"password",optionRegex:"(azure.?key|account.?key|storage.?key)"}))}).describe(d.of({direction:"two-columns"})),nt=_({type:x("coreweave"),bucket:g().nonempty().describe(d.of({label:"Bucket",placeholder:"bucket-name"})),region:g().nonempty().describe(d.of({label:"Region",placeholder:"US-EAST-04A"})),auth:et()}).describe(d.of({direction:"two-columns"})),ot=_({type:x("gdrive"),credentials_json:g().optional().describe(d.of({label:"Service Account JSON",description:"Leave empty to use browser-based authentication",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),at=_({type:x("huggingface"),token:g().optional().describe(d.of({label:"Access Token",description:"Leave empty to use the HF_TOKEN environment variable or cached login",inputType:"password",optionRegex:"(hf.?token|hugging.?face.?token|hub.?token)"}))}).describe(d.of({direction:"two-columns"})),rr=K("type",[tt,st,rt,nt,ot,at]),nr={obstore:"obstore",fsspec:"fsspec",huggingface_hub:"huggingface_hub"};var or=class{constructor(){G(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e){if(e===void 0||e==="")return"";let s=`_${t}`;if(F(e)){let n=je(e);return this.secrets[s]=`os.environ.get("${n}")`,s}return`"${e}"`}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
143
+ `)}};function it(t,e){if((t==null?void 0:t.type)==="Container credentials")return[` container_credentials_full_uri=${e.print("container_credentials_full_uri",t.container_credentials_full_uri)},`,` container_authorization_token_file=${e.print("container_authorization_token_file",t.container_authorization_token_file)},`];let s=[];return t!=null&&t.access_key_id&&s.push(` access_key_id=${e.print("access_key_id",t.access_key_id)},`),t!=null&&t.secret_access_key&&s.push(` secret_access_key=${e.print("secret_access_key",t.secret_access_key)},`),s}function ar(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import S3Store"]),r=[];return t.region&&r.push(` region=${e.print("region",t.region)},`),r.push(...it(t.auth,e)),t.endpoint_url&&r.push(` endpoint_url=${e.print("endpoint_url",t.endpoint_url)},`),t.allow_http&&r.push(" allow_http=True,"),{imports:n,code:y(`
144
144
  store = S3Store(${s},${r.length>0?`
145
145
  ${r.join(`
146
146
  `)}
147
147
  `:""})
148
- `)}}function ar(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import GCSStore"]);if(!t.service_account_key)return{imports:n,code:y(`
148
+ `)}}function ir(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import GCSStore"]);if(!t.service_account_key)return{imports:n,code:y(`
149
149
  store = GCSStore(${s})
150
150
  `)};let r=ee(t.service_account_key,a=>e.print("service_account_key",a));return r.kind==="path"?{imports:n,code:y(`
151
151
  store = GCSStore(${s},
@@ -156,15 +156,15 @@ ${r.join(`
156
156
  store = GCSStore(${s},
157
157
  service_account_key=_credentials,
158
158
  )
159
- `)})}function ir(t,e){let s=e.print("container",t.container),n=e.print("account_name",t.account_name),r=new Set(["from obstore.store import AzureStore"]),a=[`account_name=${n},`];return t.account_key&&a.push(`account_key=${e.print("account_key",t.account_key)},`),{imports:r,code:`store = AzureStore(${s},
159
+ `)})}function cr(t,e){let s=e.print("container",t.container),n=e.print("account_name",t.account_name),r=new Set(["from obstore.store import AzureStore"]),a=[`account_name=${n},`];return t.account_key&&a.push(`account_key=${e.print("account_key",t.account_key)},`),{imports:r,code:`store = AzureStore(${s},
160
160
  ${a.map(c=>` ${c}`).join(`
161
161
  `)}
162
- )`}}function cr(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import S3Store"]),r=[` region=${e.print("region",t.region)},`];return r.push(...at(t.auth,e)),r.push(` endpoint="https://${t.bucket}.cwobject.com",`," virtual_hosted_style_request=True,"),{imports:n,code:y(`
162
+ )`}}function lr(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import S3Store"]),r=[` region=${e.print("region",t.region)},`];return r.push(...it(t.auth,e)),r.push(` endpoint="https://${t.bucket}.cwobject.com",`," virtual_hosted_style_request=True,"),{imports:n,code:y(`
163
163
  store = S3Store(${s},${`
164
164
  ${r.join(`
165
165
  `)}
166
166
  `})
167
- `)}}function lr(t,e){let{secrets:s,isEmbedded:n=!1}=e,r=new Set(["from gdrive_fsspec import GoogleDriveFileSystem"]);if(t.credentials_json){let a=ee(t.credentials_json,c=>s.print("credentials_json",c));return a.kind==="path"?{imports:r,code:y(`
167
+ `)}}function dr(t,e){let{secrets:s,isEmbedded:n=!1}=e,r=new Set(["from gdrive_fsspec import GoogleDriveFileSystem"]);if(t.credentials_json){let a=ee(t.credentials_json,c=>s.print("credentials_json",c));return a.kind==="path"?{imports:r,code:y(`
168
168
  fs = GoogleDriveFileSystem(creds="${se(a.path)}", token="service_account", use_listings_cache=False, skip_instance_cache=True)
169
169
  `)}:(r.add("import json"),{imports:r,code:y(`
170
170
  _creds = ${a.expr}
@@ -174,5 +174,5 @@ ${r.join(`
174
174
  mo.output.clear_console()
175
175
  `:`
176
176
  fs = GoogleDriveFileSystem(use_listings_cache=False, skip_instance_cache=True)
177
- `)}}function dr(t,e){sr.parse(t);let s=new nr,n;switch(t.type){case"s3":n=or(t,s);break;case"gcs":n=ar(t,s);break;case"azure":n=ir(t,s);break;case"coreweave":n=cr(t,s);break;case"gdrive":n=lr(t,{secrets:s,isEmbedded:e.isEmbedded});break;default:W(t)}let r=[...new Set([...s.imports,...n.imports])].toSorted();r.push("");let a=s.formatSecrets();return a&&r.push(a),r.push(n.code.trim()),r.join(`
178
- `)}var it=U(),pr="#232F3E",ct=[{name:"Amazon S3",schema:tt,protocol:"s3",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Cloud Storage",schema:st,protocol:"gcs",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Azure Blob Storage",schema:rt,protocol:"azure",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"CoreWeave",schema:nt,protocol:"coreweave",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Drive",schema:ot,protocol:"gdrive",storageLibraries:{libraries:["fsspec"],preferred:"fsspec"}}],hr=t=>{let e=(0,it.c)(4),{onSelect:s}=t,n;e[0]===s?n=e[1]:(n=ct.map(a=>{let{name:c,schema:i,protocol:l}=a;return(0,o.jsx)(ne,{name:c,color:pr,icon:(0,o.jsx)(Ft,{protocol:l,forceDark:!0,className:"w-7.5 h-7.5"}),onSelect:()=>s(i)},c)}),e[0]=s,e[1]=n);let r;return e[2]===n?r=e[3]:(r=(0,o.jsx)(re,{children:n}),e[2]=n,e[3]=r),r};const mr=t=>{let e=(0,it.c)(16),{onSubmit:s,header:n}=t,{isEmbedded:r}=Jt(),[a,c]=(0,D.useState)(null);if(!a){let m;e[0]===Symbol.for("react.memo_cache_sentinel")?(m=(0,o.jsx)("div",{children:(0,o.jsx)(hr,{onSelect:c})}),e[0]=m):m=e[0];let S;return e[1]===n?S=e[2]:(S=(0,o.jsxs)(o.Fragment,{children:[n,m]}),e[1]=n,e[2]=S),S}let i;e[3]===a?i=e[4]:(i=ct.find(m=>m.schema===a),e[3]=a,e[4]=i);let l=i==null?void 0:i.storageLibraries,p;e[5]===(l==null?void 0:l.libraries)?p=e[6]:(p=(l==null?void 0:l.libraries)??[],e[5]=l==null?void 0:l.libraries,e[6]=p);let h=(l==null?void 0:l.preferred)??"obstore",u;e[7]===r?u=e[8]:(u=(m,S)=>dr(m,{library:S,isEmbedded:r}),e[7]=r,e[8]=u);let b;e[9]===Symbol.for("react.memo_cache_sentinel")?(b=()=>c(null),e[9]=b):b=e[9];let g;return e[10]!==s||e[11]!==a||e[12]!==p||e[13]!==h||e[14]!==u?(g=(0,o.jsx)(Te,{schema:a,libraries:p,preferredLibrary:h,displayNames:rr,libraryLabel:"Preferred storage library",generateCode:u,onSubmit:s,onBack:b}),e[10]=s,e[11]=a,e[12]=p,e[13]=h,e[14]=u,e[15]=g):g=e[15],g};var lt=U();const ur=t=>{let e=(0,lt.c)(9),{children:s,defaultTab:n}=t,r=n===void 0?"databases":n,[a,c]=(0,D.useState)(!1),i;e[0]===s?i=e[1]:(i=(0,o.jsx)(Gt,{asChild:!0,children:s}),e[0]=s,e[1]=i);let l;e[2]===Symbol.for("react.memo_cache_sentinel")?(l=()=>c(!1),e[2]=l):l=e[2];let p;e[3]===r?p=e[4]:(p=(0,o.jsx)(dt,{defaultTab:r,onClose:l}),e[3]=r,e[4]=p);let h;return e[5]!==a||e[6]!==i||e[7]!==p?(h=(0,o.jsxs)(Qt,{open:a,onOpenChange:c,children:[i,p]}),e[5]=a,e[6]=i,e[7]=p,e[8]=h):h=e[8],h},dt=t=>{let e=(0,lt.c)(23),{defaultTab:s,onClose:n}=t,[r,a]=(0,D.useState)(s===void 0?"databases":s),c;e[0]===Symbol.for("react.memo_cache_sentinel")?(c=(0,o.jsxs)($t,{className:"w-full mb-4",children:[(0,o.jsx)(de,{value:"databases",className:"flex-1",children:"Databases & Catalogs"}),(0,o.jsx)(de,{value:"storage",className:"flex-1",children:"Remote Storages"})]}),e[0]=c):c=e[0];let i=c,l;e[1]===r?l=e[2]:(l=r==="databases"?(0,o.jsxs)(o.Fragment,{children:["Don't see your database or connection method? A"," ",(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"code snippet"})," ","is all you need."]}):(0,o.jsxs)(o.Fragment,{children:["Don't see your storage or connection method? A"," ",(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/#creating-a-storage-connection",children:"code snippet"})," ","is all you need."]}),e[1]=r,e[2]=l);let p=l,h;e[3]===Symbol.for("react.memo_cache_sentinel")?(h=(0,o.jsx)(Ht,{children:"Add Connection"}),e[3]=h):h=e[3];let u;e[4]===Symbol.for("react.memo_cache_sentinel")?(u=(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"database, data catalog"}),e[4]=u):u=e[4];let b;e[5]===Symbol.for("react.memo_cache_sentinel")?(b=(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/",children:"remote storage"}),e[5]=b):b=e[5];let g;e[6]===p?g=e[7]:(g=(0,o.jsxs)(Wt,{children:[h,(0,o.jsxs)(zt,{children:["Connect to a"," ",u," ","or"," ",b," ","directly from your notebook.",(0,o.jsx)("span",{className:"block",children:p})]})]}),e[6]=p,e[7]=g);let m;e[8]===n?m=e[9]:(m=(0,o.jsx)(Zs,{onSubmit:n,className:"-mt-2"}),e[8]=n,e[9]=m);let S;e[10]===Symbol.for("react.memo_cache_sentinel")?(S=C=>a(C),e[10]=S):S=e[10];let w;e[11]===n?w=e[12]:(w=(0,o.jsx)(pe,{value:"databases",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,o.jsx)(Gs,{onSubmit:n,header:i})}),e[11]=n,e[12]=w);let k;e[13]===n?k=e[14]:(k=(0,o.jsx)(pe,{value:"storage",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,o.jsx)(mr,{onSubmit:n,header:i})}),e[13]=n,e[14]=k);let v;e[15]!==r||e[16]!==w||e[17]!==k?(v=(0,o.jsxs)(wt,{value:r,onValueChange:S,children:[w,k]}),e[15]=r,e[16]=w,e[17]=k,e[18]=v):v=e[18];let A;return e[19]!==v||e[20]!==g||e[21]!==m?(A=(0,o.jsxs)(Vt,{className:"max-h-[75vh] overflow-y-auto",children:[g,m,v]}),e[19]=v,e[20]=g,e[21]=m,e[22]=A):A=e[22],A};export{dt as n,ur as t};
177
+ `)}}function pr(t,e){let s=new Set(["from huggingface_hub import HfApi"]);return t.token?{imports:s,code:`hf = HfApi(token=${e.print("token",t.token)})`}:{imports:s,code:"hf = HfApi()"}}function hr(t,e){rr.parse(t);let s=new or,n;switch(t.type){case"s3":n=ar(t,s);break;case"gcs":n=ir(t,s);break;case"azure":n=cr(t,s);break;case"coreweave":n=lr(t,s);break;case"gdrive":n=dr(t,{secrets:s,isEmbedded:e.isEmbedded});break;case"huggingface":n=pr(t,s);break;default:M(t)}let r=[...new Set([...s.imports,...n.imports])].toSorted();r.push("");let a=s.formatSecrets();return a&&r.push(a),r.push(n.code.trim()),r.join(`
178
+ `)}var ct=U(),ur="#232F3E",lt=[{name:"Amazon S3",schema:tt,protocol:"s3",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Cloud Storage",schema:st,protocol:"gcs",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Azure Blob Storage",schema:rt,protocol:"azure",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"CoreWeave",schema:nt,protocol:"coreweave",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Drive",schema:ot,protocol:"gdrive",storageLibraries:{libraries:["fsspec"],preferred:"fsspec"}},{name:"Hugging Face Hub",schema:at,protocol:"hf",storageLibraries:{libraries:["huggingface_hub"],preferred:"huggingface_hub"}}],mr=t=>{let e=(0,ct.c)(4),{onSelect:s}=t,n;e[0]===s?n=e[1]:(n=lt.map(a=>{let{name:c,schema:i,protocol:l}=a;return(0,o.jsx)(ne,{name:c,color:ur,icon:(0,o.jsx)(Ot,{protocol:l,forceDark:!0,className:"w-7.5 h-7.5"}),onSelect:()=>s(i)},c)}),e[0]=s,e[1]=n);let r;return e[2]===n?r=e[3]:(r=(0,o.jsx)(re,{children:n}),e[2]=n,e[3]=r),r};const br=t=>{let e=(0,ct.c)(16),{onSubmit:s,header:n}=t,{isEmbedded:r}=Yt(),[a,c]=(0,D.useState)(null);if(!a){let u;e[0]===Symbol.for("react.memo_cache_sentinel")?(u=(0,o.jsx)("div",{children:(0,o.jsx)(mr,{onSelect:c})}),e[0]=u):u=e[0];let S;return e[1]===n?S=e[2]:(S=(0,o.jsxs)(o.Fragment,{children:[n,u]}),e[1]=n,e[2]=S),S}let i;e[3]===a?i=e[4]:(i=lt.find(u=>u.schema===a),e[3]=a,e[4]=i);let l=i==null?void 0:i.storageLibraries,p;e[5]===(l==null?void 0:l.libraries)?p=e[6]:(p=(l==null?void 0:l.libraries)??[],e[5]=l==null?void 0:l.libraries,e[6]=p);let h=(l==null?void 0:l.preferred)??"obstore",m;e[7]===r?m=e[8]:(m=(u,S)=>hr(u,{library:S,isEmbedded:r}),e[7]=r,e[8]=m);let b;e[9]===Symbol.for("react.memo_cache_sentinel")?(b=()=>c(null),e[9]=b):b=e[9];let f;return e[10]!==s||e[11]!==a||e[12]!==p||e[13]!==h||e[14]!==m?(f=(0,o.jsx)(Te,{schema:a,libraries:p,preferredLibrary:h,displayNames:nr,libraryLabel:"Preferred storage library",generateCode:m,onSubmit:s,onBack:b}),e[10]=s,e[11]=a,e[12]=p,e[13]=h,e[14]=m,e[15]=f):f=e[15],f};var dt=U();const gr=t=>{let e=(0,dt.c)(9),{children:s,defaultTab:n}=t,r=n===void 0?"databases":n,[a,c]=(0,D.useState)(!1),i;e[0]===s?i=e[1]:(i=(0,o.jsx)(Vt,{asChild:!0,children:s}),e[0]=s,e[1]=i);let l;e[2]===Symbol.for("react.memo_cache_sentinel")?(l=()=>c(!1),e[2]=l):l=e[2];let p;e[3]===r?p=e[4]:(p=(0,o.jsx)(pt,{defaultTab:r,onClose:l}),e[3]=r,e[4]=p);let h;return e[5]!==a||e[6]!==i||e[7]!==p?(h=(0,o.jsxs)(Jt,{open:a,onOpenChange:c,children:[i,p]}),e[5]=a,e[6]=i,e[7]=p,e[8]=h):h=e[8],h},pt=t=>{let e=(0,dt.c)(23),{defaultTab:s,onClose:n}=t,[r,a]=(0,D.useState)(s===void 0?"databases":s),c;e[0]===Symbol.for("react.memo_cache_sentinel")?(c=(0,o.jsxs)(Ct,{className:"w-full mb-4",children:[(0,o.jsx)(de,{value:"databases",className:"flex-1",children:"Databases & Catalogs"}),(0,o.jsx)(de,{value:"storage",className:"flex-1",children:"Remote Storages"})]}),e[0]=c):c=e[0];let i=c,l;e[1]===r?l=e[2]:(l=r==="databases"?(0,o.jsxs)(o.Fragment,{children:["Don't see your database or connection method? A"," ",(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"code snippet"})," ","is all you need."]}):(0,o.jsxs)(o.Fragment,{children:["Don't see your storage or connection method? A"," ",(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/#creating-a-storage-connection",children:"code snippet"})," ","is all you need."]}),e[1]=r,e[2]=l);let p=l,h;e[3]===Symbol.for("react.memo_cache_sentinel")?(h=(0,o.jsx)(Gt,{children:"Add Connection"}),e[3]=h):h=e[3];let m;e[4]===Symbol.for("react.memo_cache_sentinel")?(m=(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"database, data catalog"}),e[4]=m):m=e[4];let b;e[5]===Symbol.for("react.memo_cache_sentinel")?(b=(0,o.jsx)(z,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/",children:"remote storage"}),e[5]=b):b=e[5];let f;e[6]===p?f=e[7]:(f=(0,o.jsxs)(Wt,{children:[h,(0,o.jsxs)(Qt,{children:["Connect to a"," ",m," ","or"," ",b," ","directly from your notebook.",(0,o.jsx)("span",{className:"block",children:p})]})]}),e[6]=p,e[7]=f);let u;e[8]===n?u=e[9]:(u=(0,o.jsx)(er,{onSubmit:n,className:"-mt-2"}),e[8]=n,e[9]=u);let S;e[10]===Symbol.for("react.memo_cache_sentinel")?(S=C=>a(C),e[10]=S):S=e[10];let w;e[11]===n?w=e[12]:(w=(0,o.jsx)(pe,{value:"databases",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,o.jsx)(Vs,{onSubmit:n,header:i})}),e[11]=n,e[12]=w);let k;e[13]===n?k=e[14]:(k=(0,o.jsx)(pe,{value:"storage",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,o.jsx)(br,{onSubmit:n,header:i})}),e[13]=n,e[14]=k);let v;e[15]!==r||e[16]!==w||e[17]!==k?(v=(0,o.jsxs)(jt,{value:r,onValueChange:S,children:[w,k]}),e[15]=r,e[16]=w,e[17]=k,e[18]=v):v=e[18];let A;return e[19]!==v||e[20]!==f||e[21]!==u?(A=(0,o.jsxs)(zt,{className:"max-h-[75vh] overflow-y-auto",children:[f,u,v]}),e[19]=v,e[20]=f,e[21]=u,e[22]=A):A=e[22],A};export{pt as n,gr as t};