@marimo-team/frontend 0.23.16-dev51 → 0.23.16-dev53

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 (101) hide show
  1. package/dist/assets/{CellStatus-DciB5rbr.js → CellStatus-BfPw99P2.js} +1 -1
  2. package/dist/assets/{JsonOutput-BM3vve9F.js → JsonOutput-E3o1Uk0e.js} +12 -12
  3. package/dist/assets/{MarimoErrorOutput-Bh0y53Du.js → MarimoErrorOutput-D9Dypu7Q.js} +4 -4
  4. package/dist/assets/{RenderHTML-B9jx7Dsm.js → RenderHTML-DRJxD5FQ.js} +1 -1
  5. package/dist/assets/{RunButton-DMtHbFuA.js → RunButton-y7Y9rjmF.js} +1 -1
  6. package/dist/assets/{add-cell-with-ai-BcRYriMM.js → add-cell-with-ai-ByfOWv0Z.js} +12 -12
  7. package/dist/assets/add-connection-dialog-DVSFAYXK.js +178 -0
  8. package/dist/assets/{agent-panel-B31W_VwU.js → agent-panel-B2OKzocQ.js} +4 -4
  9. package/dist/assets/{ai-model-dropdown-B3D05oQh.js → ai-model-dropdown-Dr2QbDlZ.js} +3 -3
  10. package/dist/assets/{app-config-button-D_mjovwU.js → app-config-button-DScmpqcV.js} +1 -1
  11. package/dist/assets/{blob-BJREJiQp.js → blob-BXcsY70H.js} +1 -1
  12. package/dist/assets/{cell-editor-BOhZHoLm.js → cell-editor-BuGvBHQI.js} +11 -11
  13. package/dist/assets/cell-link-DGv7Q4Mc.js +1 -0
  14. package/dist/assets/{cells-Jr-5Pc9f.js → cells-CMc2NsXm.js} +61 -61
  15. package/dist/assets/{chat-display-Djy1Jt6b.js → chat-display-CeQftKrl.js} +1 -1
  16. package/dist/assets/{chat-panel-B2nYSE_l.js → chat-panel-BP5hfvS7.js} +4 -4
  17. package/dist/assets/{chat-ui-DlG0En7w.js → chat-ui-DoA0lexh.js} +4 -4
  18. package/dist/assets/{column-preview-BUHlI5va.js → column-preview-BeR1SI6Z.js} +1 -1
  19. package/dist/assets/{command-palette-rEJmXCZM.js → command-palette-DZSYeVCO.js} +1 -1
  20. package/dist/assets/{common-CZc3fQI4.js → common-NWgkp5L1.js} +1 -1
  21. package/dist/assets/{components-GAYCj6Mx.js → components-Mbf0MEIC.js} +1 -1
  22. package/dist/assets/{datasource-AWkkAeiS.js → datasource-Wt8yUynT.js} +2 -2
  23. package/dist/assets/{dependency-graph-panel-wcqTPq6b.js → dependency-graph-panel-B7TXWKYs.js} +4 -4
  24. package/dist/assets/{documentation-panel-Eu8MIkoU.js → documentation-panel-D9MZUKaD.js} +1 -1
  25. package/dist/assets/{download-CQ7iZFJL.js → download-U_wgQ8mC.js} +2 -2
  26. package/dist/assets/{edit-page-Cw7nN6HC.js → edit-page-BBc9BGYT.js} +6 -6
  27. package/dist/assets/error-entries-Bs30Z6uG.js +8 -0
  28. package/dist/assets/{error-panel-BZzr0PA1.js → error-panel-KY679tDx.js} +1 -1
  29. package/dist/assets/{file-explorer-panel-C1W7g_Wp.js → file-explorer-panel-9y0hzNAa.js} +4 -4
  30. package/dist/assets/{file-icons-BAqZR08W.js → file-icons-D5l3kSYq.js} +1 -1
  31. package/dist/assets/{file-name-input-0MiZVw5r.js → file-name-input-D_xhjKFp.js} +1 -1
  32. package/dist/assets/{floating-outline-C6g75zrP.js → floating-outline-BdgJT2B9.js} +1 -1
  33. package/dist/assets/{focus-C0HXXg3V.js → focus-D64hL1tZ.js} +1 -1
  34. package/dist/assets/{form-DT44Rmhp.js → form-CoqwGbkO.js} +1 -1
  35. package/dist/assets/{form-C051zAev.js → form-b4jQBV3T.js} +1 -1
  36. package/dist/assets/{home-page-CJz9IHZ2.js → home-page-tVWY9YdF.js} +2 -2
  37. package/dist/assets/{html-to-image-CfjhKLA4.js → html-to-image-Dm7huiQj.js} +2 -2
  38. package/dist/assets/index-8Nlyq3mV.css +2 -0
  39. package/dist/assets/{index-CrEgp1mR.js → index-Dz8dLbnI.js} +6 -6
  40. package/dist/assets/{layout-D4bpsgVD.js → layout-DKUs0ozP.js} +5 -5
  41. package/dist/assets/{logs-panel-CsaY3O1m.js → logs-panel-BzS7VSh0.js} +1 -1
  42. package/dist/assets/{markdown-renderer-BgqeO4JR.js → markdown-renderer-uWTrQId1.js} +2 -2
  43. package/dist/assets/{name-cell-input-DcU8Rfrz.js → name-cell-input-BgxcWTfV.js} +1 -1
  44. package/dist/assets/{outline-panel-D-CHPZQM.js → outline-panel-BT3zzwfa.js} +1 -1
  45. package/dist/assets/{packages-panel-dq7QVZl1.js → packages-panel-BKfeGuve.js} +1 -1
  46. package/dist/assets/panels-4jSJxEYp.js +4 -0
  47. package/dist/assets/{process-output-NbwnUu05.js → process-output-DE8TTXPC.js} +1 -1
  48. package/dist/assets/{radio-group-rhQpGwU-.js → radio-group-Da80KtoT.js} +1 -1
  49. package/dist/assets/readonly-code-display-B8zhuSov.js +1 -0
  50. package/dist/assets/{readonly-python-code-Q_1Y3VyZ.js → readonly-python-code-Bmo6cMSq.js} +1 -1
  51. package/dist/assets/{reveal-component-CedpDwM8.js → reveal-component-Cv7WZ4r6.js} +2 -2
  52. package/dist/assets/{run-page-CGoux5cj.js → run-page-CA21zsME.js} +1 -1
  53. package/dist/assets/scratchpad-panel-Br7RFJkK.js +1 -0
  54. package/dist/assets/{session-panel-OIot3eyB.js → session-panel-BiUht4zM.js} +1 -1
  55. package/dist/assets/{skeleton-BRvjFlRu.js → skeleton-CY-B1asm.js} +1 -1
  56. package/dist/assets/{snippets-panel-YFyH2hUM.js → snippets-panel-CjiTlCpD.js} +1 -1
  57. package/dist/assets/{state-bHyw5u3R.js → state-BJ7IdtXW.js} +1 -1
  58. package/dist/assets/state-TgAqts79.js +3 -0
  59. package/dist/assets/{tracing-CDMM1BB2.js → tracing-DmQ7KQ3A.js} +1 -1
  60. package/dist/assets/{tracing-panel-BH5Zbvl9.js → tracing-panel-C1TOV9kj.js} +2 -2
  61. package/dist/assets/{tree-actions-DDveUIHK.js → tree-actions-enwITSy7.js} +1 -1
  62. package/dist/assets/useCellActionButton-CXvU9GmI.js +1 -0
  63. package/dist/assets/{useDeleteCell-BIopCngo.js → useDeleteCell-Bp3S6pL8.js} +1 -1
  64. package/dist/assets/{useDependencyPanelTab-snM337f9.js → useDependencyPanelTab-CewYv8cP.js} +1 -1
  65. package/dist/assets/{useNotebookActions-CsZfWFKy.js → useNotebookActions-l9NAqPKm.js} +4 -4
  66. package/dist/assets/useRunCells-B-KIwrVZ.js +1 -0
  67. package/dist/assets/{useSplitCell-BTmKY1CU.js → useSplitCell-B0jvSFsl.js} +1 -1
  68. package/dist/index.html +24 -24
  69. package/package.json +1 -1
  70. package/src/__mocks__/requests.ts +1 -0
  71. package/src/components/editor/connections/__tests__/quick-add-data-sources.test.tsx +113 -0
  72. package/src/components/editor/connections/add-connection-dialog.tsx +2 -0
  73. package/src/components/editor/connections/quick-add-data-sources.tsx +106 -0
  74. package/src/core/codemirror/language/__tests__/sql.test.ts +191 -7
  75. package/src/core/codemirror/language/languages/sql/sql.ts +41 -13
  76. package/src/core/datasets/data-source-discovery.ts +5 -0
  77. package/src/core/datasets/request-registry.ts +11 -0
  78. package/src/core/islands/bootstrap.ts +1 -0
  79. package/src/core/islands/bridge.ts +1 -0
  80. package/src/core/kernel/messages.ts +2 -0
  81. package/src/core/network/__tests__/requests-lazy.test.ts +15 -0
  82. package/src/core/network/__tests__/requests-network.test.ts +14 -0
  83. package/src/core/network/requests-lazy.ts +1 -0
  84. package/src/core/network/requests-network.ts +8 -0
  85. package/src/core/network/requests-static.ts +1 -0
  86. package/src/core/network/requests-toasting.tsx +1 -0
  87. package/src/core/network/types.ts +2 -0
  88. package/src/core/wasm/bridge.ts +10 -0
  89. package/src/core/websocket/useMarimoKernelConnection.tsx +4 -0
  90. package/src/hooks/__tests__/useDataSourceDiscovery.test.ts +80 -0
  91. package/src/hooks/useDataSourceDiscovery.ts +18 -0
  92. package/dist/assets/add-connection-dialog-syjRSNPT.js +0 -178
  93. package/dist/assets/cell-link-DqNEv3f9.js +0 -1
  94. package/dist/assets/error-entries-Cr8uVukN.js +0 -8
  95. package/dist/assets/index-D6Dp_PFj.css +0 -2
  96. package/dist/assets/panels-DWoniKhE.js +0 -4
  97. package/dist/assets/readonly-code-display-CVY2f7EZ.js +0 -1
  98. package/dist/assets/scratchpad-panel-BLuF25yR.js +0 -1
  99. package/dist/assets/state-DB_fxbhn.js +0 -3
  100. package/dist/assets/useCellActionButton-D5ix_g2a.js +0 -1
  101. package/dist/assets/useRunCells-CF_V8X2K.js +0 -1
package/dist/index.html CHANGED
@@ -66,7 +66,7 @@
66
66
  <marimo-server-token data-token="{{ server_token }}" hidden></marimo-server-token>
67
67
  <!-- /TODO -->
68
68
  <title>{{ title }}</title>
69
- <script type="module" crossorigin src="./assets/index-CrEgp1mR.js"></script>
69
+ <script type="module" crossorigin src="./assets/index-Dz8dLbnI.js"></script>
70
70
  <link rel="modulepreload" crossorigin href="./assets/preload-helper-TXkS1QI3.js">
71
71
  <link rel="modulepreload" crossorigin href="./assets/objects-BlG0ZrO0.js">
72
72
  <link rel="modulepreload" crossorigin href="./assets/chunk-LvLJmgfZ.js">
@@ -134,7 +134,7 @@
134
134
  <link rel="modulepreload" crossorigin href="./assets/debounce-OiXznToA.js">
135
135
  <link rel="modulepreload" crossorigin href="./assets/database-zap-CKO8g4BU.js">
136
136
  <link rel="modulepreload" crossorigin href="./assets/main-BNL5rxQw.js">
137
- <link rel="modulepreload" crossorigin href="./assets/cells-Jr-5Pc9f.js">
137
+ <link rel="modulepreload" crossorigin href="./assets/cells-CMc2NsXm.js">
138
138
  <link rel="modulepreload" crossorigin href="./assets/kbd-C5HnBH5U.js">
139
139
  <link rel="modulepreload" crossorigin href="./assets/useInstallPackage-WiYBwhUO.js">
140
140
  <link rel="modulepreload" crossorigin href="./assets/alert-dialog-Bb8Lz8ko.js">
@@ -147,40 +147,40 @@
147
147
  <link rel="modulepreload" crossorigin href="./assets/usePress-5hBv4sjD.js">
148
148
  <link rel="modulepreload" crossorigin href="./assets/input-JxFwu-3G.js">
149
149
  <link rel="modulepreload" crossorigin href="./assets/ImperativeModal-fY-ZfT0b.js">
150
- <link rel="modulepreload" crossorigin href="./assets/cell-link-DqNEv3f9.js">
150
+ <link rel="modulepreload" crossorigin href="./assets/cell-link-DGv7Q4Mc.js">
151
151
  <link rel="modulepreload" crossorigin href="./assets/config-BJVx3eBW.js">
152
152
  <link rel="modulepreload" crossorigin href="./assets/alert-jlwhGxSb.js">
153
153
  <link rel="modulepreload" crossorigin href="./assets/chevron-right-BS_NcSch.js">
154
154
  <link rel="modulepreload" crossorigin href="./assets/dropdown-menu-dHmZR8bI.js">
155
155
  <link rel="modulepreload" crossorigin href="./assets/links-wcQA1Np1.js">
156
- <link rel="modulepreload" crossorigin href="./assets/useRunCells-CF_V8X2K.js">
156
+ <link rel="modulepreload" crossorigin href="./assets/useRunCells-B-KIwrVZ.js">
157
157
  <link rel="modulepreload" crossorigin href="./assets/copy-Bxb1_Gtw.js">
158
158
  <link rel="modulepreload" crossorigin href="./assets/copy-3iHFdt22.js">
159
159
  <link rel="modulepreload" crossorigin href="./assets/copy-icon-neDQUo4Q.js">
160
- <link rel="modulepreload" crossorigin href="./assets/RenderHTML-B9jx7Dsm.js">
161
- <link rel="modulepreload" crossorigin href="./assets/datasource-AWkkAeiS.js">
162
- <link rel="modulepreload" crossorigin href="./assets/state-bHyw5u3R.js">
160
+ <link rel="modulepreload" crossorigin href="./assets/RenderHTML-DRJxD5FQ.js">
161
+ <link rel="modulepreload" crossorigin href="./assets/datasource-Wt8yUynT.js">
162
+ <link rel="modulepreload" crossorigin href="./assets/state-BJ7IdtXW.js">
163
163
  <link rel="modulepreload" crossorigin href="./assets/hat-glasses-BttHlyvs.js">
164
164
  <link rel="modulepreload" crossorigin href="./assets/package-Bxz_8e44.js">
165
165
  <link rel="modulepreload" crossorigin href="./assets/sparkles-DUCpRFlS.js">
166
- <link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-Bh0y53Du.js">
166
+ <link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-D9Dypu7Q.js">
167
167
  <link rel="modulepreload" crossorigin href="./assets/ErrorBoundary-DzTD8LVq.js">
168
168
  <link rel="modulepreload" crossorigin href="./assets/eye-off-BOZA7gbn.js">
169
169
  <link rel="modulepreload" crossorigin href="./assets/refresh-cw-CYq4mK5a.js">
170
- <link rel="modulepreload" crossorigin href="./assets/tree-actions-DDveUIHK.js">
170
+ <link rel="modulepreload" crossorigin href="./assets/tree-actions-enwITSy7.js">
171
171
  <link rel="modulepreload" crossorigin href="./assets/spinner-D-V4psAa.js">
172
- <link rel="modulepreload" crossorigin href="./assets/html-to-image-CfjhKLA4.js">
173
- <link rel="modulepreload" crossorigin href="./assets/focus-C0HXXg3V.js">
172
+ <link rel="modulepreload" crossorigin href="./assets/html-to-image-Dm7huiQj.js">
173
+ <link rel="modulepreload" crossorigin href="./assets/focus-D64hL1tZ.js">
174
174
  <link rel="modulepreload" crossorigin href="./assets/useAsyncData-Bg5iAViP.js">
175
175
  <link rel="modulepreload" crossorigin href="./assets/LazyAnyLanguageCodeMirror-DPOEues4.js">
176
176
  <link rel="modulepreload" crossorigin href="./assets/micromark-factory-space-Dx_STMJD.js">
177
177
  <link rel="modulepreload" crossorigin href="./assets/chunk-5FQGJX7Z-BEbM01OT.js">
178
- <link rel="modulepreload" crossorigin href="./assets/markdown-renderer-BgqeO4JR.js">
178
+ <link rel="modulepreload" crossorigin href="./assets/markdown-renderer-uWTrQId1.js">
179
179
  <link rel="modulepreload" crossorigin href="./assets/command-Dqxie4gq.js">
180
180
  <link rel="modulepreload" crossorigin href="./assets/field-BsxD35VE.js">
181
181
  <link rel="modulepreload" crossorigin href="./assets/popover-Bcz1Zikb.js">
182
182
  <link rel="modulepreload" crossorigin href="./assets/errors-PgHhDa-M.js">
183
- <link rel="modulepreload" crossorigin href="./assets/download-CQ7iZFJL.js">
183
+ <link rel="modulepreload" crossorigin href="./assets/download-U_wgQ8mC.js">
184
184
  <link rel="modulepreload" crossorigin href="./assets/table-CH2plyy9.js">
185
185
  <link rel="modulepreload" crossorigin href="./assets/useIframeCapabilities-j6fR_dBA.js">
186
186
  <link rel="modulepreload" crossorigin href="./assets/error-banner-BSxEOIn2.js">
@@ -201,34 +201,34 @@
201
201
  <link rel="modulepreload" crossorigin href="./assets/plus-AgXtozBz.js">
202
202
  <link rel="modulepreload" crossorigin href="./assets/trash-2-72W0TZ3L.js">
203
203
  <link rel="modulepreload" crossorigin href="./assets/react-resizable-panels.browser.esm-Z-MD3n3J.js">
204
- <link rel="modulepreload" crossorigin href="./assets/JsonOutput-BM3vve9F.js">
204
+ <link rel="modulepreload" crossorigin href="./assets/JsonOutput-E3o1Uk0e.js">
205
205
  <link rel="modulepreload" crossorigin href="./assets/chart-no-axes-column-iDRxid4h.js">
206
206
  <link rel="modulepreload" crossorigin href="./assets/square-function-DNNRTPQJ.js">
207
207
  <link rel="modulepreload" crossorigin href="./assets/spec-C7opZZRp.js">
208
- <link rel="modulepreload" crossorigin href="./assets/column-preview-BUHlI5va.js">
208
+ <link rel="modulepreload" crossorigin href="./assets/column-preview-BeR1SI6Z.js">
209
209
  <link rel="modulepreload" crossorigin href="./assets/icons-CIPE8AIY.js">
210
- <link rel="modulepreload" crossorigin href="./assets/radio-group-rhQpGwU-.js">
211
- <link rel="modulepreload" crossorigin href="./assets/floating-outline-C6g75zrP.js">
210
+ <link rel="modulepreload" crossorigin href="./assets/radio-group-Da80KtoT.js">
211
+ <link rel="modulepreload" crossorigin href="./assets/floating-outline-BdgJT2B9.js">
212
212
  <link rel="modulepreload" crossorigin href="./assets/switch-OVajJBPR.js">
213
213
  <link rel="modulepreload" crossorigin href="./assets/objectWithoutPropertiesLoose-CboCOq4o.js">
214
214
  <link rel="modulepreload" crossorigin href="./assets/esm-B30N1GS0.js">
215
- <link rel="modulepreload" crossorigin href="./assets/readonly-python-code-Q_1Y3VyZ.js">
215
+ <link rel="modulepreload" crossorigin href="./assets/readonly-python-code-Bmo6cMSq.js">
216
216
  <link rel="modulepreload" crossorigin href="./assets/file-headphone-DAu6VO6j.js">
217
217
  <link rel="modulepreload" crossorigin href="./assets/file-CJH9JNGH.js">
218
218
  <link rel="modulepreload" crossorigin href="./assets/image-5XI2VU-q.js">
219
- <link rel="modulepreload" crossorigin href="./assets/file-icons-BAqZR08W.js">
219
+ <link rel="modulepreload" crossorigin href="./assets/file-icons-D5l3kSYq.js">
220
220
  <link rel="modulepreload" crossorigin href="./assets/events-90X9nql5.js">
221
221
  <link rel="modulepreload" crossorigin href="./assets/share-852X_cqy.js">
222
- <link rel="modulepreload" crossorigin href="./assets/blob-BJREJiQp.js">
222
+ <link rel="modulepreload" crossorigin href="./assets/blob-BXcsY70H.js">
223
223
  <link rel="modulepreload" crossorigin href="./assets/memoize-Tp7rARFe.js">
224
224
  <link rel="modulepreload" crossorigin href="./assets/get-C-qh_et5.js">
225
225
  <link rel="modulepreload" crossorigin href="./assets/_baseSet-CsW0tABf.js">
226
- <link rel="modulepreload" crossorigin href="./assets/state-DB_fxbhn.js">
226
+ <link rel="modulepreload" crossorigin href="./assets/state-TgAqts79.js">
227
227
  <link rel="modulepreload" crossorigin href="./assets/label-CqjrpwRC.js">
228
- <link rel="modulepreload" crossorigin href="./assets/form-C051zAev.js">
228
+ <link rel="modulepreload" crossorigin href="./assets/form-b4jQBV3T.js">
229
229
  <link rel="modulepreload" crossorigin href="./assets/textarea-DZB2bzVL.js">
230
230
  <link rel="modulepreload" crossorigin href="./assets/refresh-ccw-sLaKP2vj.js">
231
- <link rel="modulepreload" crossorigin href="./assets/form-DT44Rmhp.js">
231
+ <link rel="modulepreload" crossorigin href="./assets/form-CoqwGbkO.js">
232
232
  <link rel="modulepreload" crossorigin href="./assets/renderShortcut-Bl8_FVn8.js">
233
233
  <link rel="modulepreload" crossorigin href="./assets/useBoolean-CFj86iac.js">
234
234
  <link rel="modulepreload" crossorigin href="./assets/useDeepCompareMemoize-CSMoa-ga.js">
@@ -250,7 +250,7 @@
250
250
  <link rel="stylesheet" crossorigin href="./assets/cells-BVzo2MHf.css">
251
251
  <link rel="stylesheet" crossorigin href="./assets/markdown-renderer-DdDKmWlR.css">
252
252
  <link rel="stylesheet" crossorigin href="./assets/JsonOutput-BQtz7-np.css">
253
- <link rel="stylesheet" crossorigin href="./assets/index-D6Dp_PFj.css">
253
+ <link rel="stylesheet" crossorigin href="./assets/index-8Nlyq3mV.css">
254
254
  </head>
255
255
  <body>
256
256
  <div id="root"></div>
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@marimo-team/frontend",
3
- "version": "0.23.16-dev51",
3
+ "version": "0.23.16-dev53",
4
4
  "main": "dist/main.js",
5
5
  "types": "dist/index.d.ts",
6
6
  "type": "module",
@@ -41,6 +41,7 @@ export const MockRequestClient = {
41
41
  previewSQLTableList: vi.fn().mockResolvedValue({ tables: [] }),
42
42
  previewSQLSchemaList: vi.fn().mockResolvedValue({ schemas: [] }),
43
43
  previewDataSourceConnection: vi.fn().mockResolvedValue({}),
44
+ discoverDataSources: vi.fn().mockResolvedValue({}),
44
45
  validateSQL: vi.fn().mockResolvedValue({}),
45
46
  openFile: vi.fn().mockResolvedValue({}),
46
47
  getUsageStats: vi.fn().mockResolvedValue({}),
@@ -0,0 +1,113 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { fireEvent, render, screen, waitFor } from "@testing-library/react";
4
+ import { describe, expect, it, vi } from "vitest";
5
+ import type { DetectedDataSource } from "@/core/datasets/data-source-discovery";
6
+ import { QuickAddDataSources } from "../quick-add-data-sources";
7
+
8
+ const sources: DetectedDataSource[] = [
9
+ {
10
+ id: "postgres-libpq-environment",
11
+ integration: "postgres",
12
+ category: "database",
13
+ displayName: "PostgreSQL",
14
+ confidence: "high",
15
+ origins: [{ type: "environment", label: "Kernel environment" }],
16
+ configuration: [
17
+ {
18
+ field: "Host",
19
+ value: { kind: "environment-variable", name: "PGHOST" },
20
+ },
21
+ {
22
+ field: "Username",
23
+ value: { kind: "environment-variable", name: "PGUSER" },
24
+ },
25
+ {
26
+ field: "Database",
27
+ value: { kind: "environment-variable", name: "PGDATABASE" },
28
+ },
29
+ ],
30
+ code: "engine = create_engine()",
31
+ },
32
+ {
33
+ id: "pyiceberg-prod",
34
+ integration: "pyiceberg",
35
+ category: "catalog",
36
+ displayName: "PyIceberg (prod)",
37
+ confidence: "high",
38
+ origins: [
39
+ {
40
+ type: "configuration",
41
+ label: "Resolved PyIceberg configuration",
42
+ },
43
+ ],
44
+ configuration: [
45
+ {
46
+ field: "Catalog",
47
+ value: { kind: "safe-literal", value: "prod" },
48
+ },
49
+ {
50
+ field: "Type",
51
+ value: { kind: "safe-literal", value: "REST" },
52
+ },
53
+ ],
54
+ code: 'catalog = load_catalog("prod")',
55
+ },
56
+ ];
57
+
58
+ describe("QuickAddDataSources", () => {
59
+ it("does not render an empty section", () => {
60
+ const { container } = render(
61
+ <QuickAddDataSources sources={[]} onAdd={vi.fn()} />,
62
+ );
63
+
64
+ expect(container).toBeEmptyDOMElement();
65
+ });
66
+
67
+ it("renders detected sources as clickable tags", () => {
68
+ const onAdd = vi.fn();
69
+
70
+ render(<QuickAddDataSources sources={sources} onAdd={onAdd} />);
71
+ fireEvent.click(
72
+ screen.getByRole("button", {
73
+ name: "Add PostgreSQL connection",
74
+ }),
75
+ );
76
+
77
+ expect(screen.getByText("Quick add")).toBeInTheDocument();
78
+ expect(onAdd).toHaveBeenCalledWith(sources[0]);
79
+ });
80
+
81
+ it("shows environment references on hover", async () => {
82
+ render(<QuickAddDataSources sources={sources} onAdd={vi.fn()} />);
83
+ const tag = screen.getByRole("button", {
84
+ name: "Add PostgreSQL connection",
85
+ });
86
+
87
+ fireEvent.pointerMove(tag);
88
+ fireEvent.mouseOver(tag);
89
+
90
+ await waitFor(() => {
91
+ expect(screen.getAllByText('os.environ["PGHOST"]')).not.toHaveLength(0);
92
+ });
93
+ expect(screen.getAllByText('os.environ["PGUSER"]')).not.toHaveLength(0);
94
+ expect(screen.getAllByText('os.environ["PGDATABASE"]')).not.toHaveLength(0);
95
+ });
96
+
97
+ it("shows safe configuration metadata on hover", async () => {
98
+ render(<QuickAddDataSources sources={sources} onAdd={vi.fn()} />);
99
+ const tag = screen.getByRole("button", {
100
+ name: "Add PyIceberg (prod) connection",
101
+ });
102
+
103
+ fireEvent.pointerMove(tag);
104
+ fireEvent.mouseOver(tag);
105
+
106
+ await waitFor(() => {
107
+ expect(
108
+ screen.getAllByText("Detected from Resolved PyIceberg configuration"),
109
+ ).not.toHaveLength(0);
110
+ });
111
+ expect(screen.getAllByText("REST")).not.toHaveLength(0);
112
+ });
113
+ });
@@ -12,6 +12,7 @@ import {
12
12
  import { ExternalLink } from "@/components/ui/links";
13
13
  import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
14
14
  import { AddDatabaseForm } from "./database/add-database-form";
15
+ import { AutoDiscoveredDataSources } from "./quick-add-data-sources";
15
16
  import { AddStorageForm } from "./storage/add-storage-form";
16
17
 
17
18
  type ConnectionTab = "databases" | "storage";
@@ -86,6 +87,7 @@ export const AddConnectionDialogContent: React.FC<{
86
87
  <span className="block">{codeSnippetHint}</span>
87
88
  </DialogDescription>
88
89
  </DialogHeader>
90
+ <AutoDiscoveredDataSources onSubmit={onClose} />
89
91
  <Tabs
90
92
  value={activeTab}
91
93
  onValueChange={(v) => setActiveTab(v as ConnectionTab)}
@@ -0,0 +1,106 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { PlusIcon, SparklesIcon } from "lucide-react";
4
+ import { Tooltip, TooltipProvider } from "@/components/ui/tooltip";
5
+ import type { DetectedDataSource } from "@/core/datasets/data-source-discovery";
6
+ import { useDataSourceDiscovery } from "@/hooks/useDataSourceDiscovery";
7
+ import { useInsertCode } from "./components";
8
+
9
+ export const QuickAddDataSources: React.FC<{
10
+ sources: DetectedDataSource[];
11
+ onAdd: (source: DetectedDataSource) => void;
12
+ }> = ({ sources, onAdd }) => {
13
+ if (sources.length === 0) {
14
+ return null;
15
+ }
16
+
17
+ return (
18
+ <section
19
+ aria-labelledby="quick-add-data-sources-title"
20
+ className="rounded-md border bg-muted/30 px-3 py-2"
21
+ >
22
+ <div className="flex flex-wrap items-center gap-2">
23
+ <div className="mr-1 flex items-center gap-1.5">
24
+ <SparklesIcon className="h-3.5 w-3.5 text-muted-foreground" />
25
+ <h3 id="quick-add-data-sources-title" className="text-sm font-medium">
26
+ Quick add
27
+ </h3>
28
+ </div>
29
+ <TooltipProvider delayDuration={200}>
30
+ {sources.map((source) => (
31
+ <Tooltip
32
+ key={source.id}
33
+ side="bottom"
34
+ content={<DetectedDataSourceDetails source={source} />}
35
+ >
36
+ <button
37
+ type="button"
38
+ aria-label={`Add ${source.displayName} connection`}
39
+ className="inline-flex items-center gap-1 rounded-full border border-(--blue-8) bg-(--blue-2) px-2.5 py-1 text-xs font-semibold text-(--blue-11) transition-colors hover:bg-(--blue-3) focus:outline-hidden focus:ring-2 focus:ring-ring focus:ring-offset-2"
40
+ onClick={() => onAdd(source)}
41
+ >
42
+ <PlusIcon className="h-3 w-3" />
43
+ {source.displayName}
44
+ </button>
45
+ </Tooltip>
46
+ ))}
47
+ </TooltipProvider>
48
+ </div>
49
+ </section>
50
+ );
51
+ };
52
+
53
+ const DetectedDataSourceDetails: React.FC<{
54
+ source: DetectedDataSource;
55
+ }> = ({ source }) => (
56
+ <div className="min-w-64 space-y-2 py-1">
57
+ <div>
58
+ <div className="font-medium">{source.displayName}</div>
59
+ <div className="text-xs text-muted-foreground">
60
+ Detected from{" "}
61
+ {source.origins.map((origin) => origin.label).join(" and ")}
62
+ </div>
63
+ </div>
64
+ <dl className="grid grid-cols-[auto_1fr] gap-x-3 gap-y-1 text-xs">
65
+ {source.configuration.map((item) => (
66
+ <div
67
+ className="contents"
68
+ key={
69
+ item.value.kind === "environment-variable"
70
+ ? item.value.name
71
+ : `${item.field}:${item.value.value}`
72
+ }
73
+ >
74
+ <dt className="text-muted-foreground">{item.field}</dt>
75
+ <dd>
76
+ <code>
77
+ {item.value.kind === "environment-variable"
78
+ ? `os.environ["${item.value.name}"]`
79
+ : item.value.value}
80
+ </code>
81
+ </dd>
82
+ </div>
83
+ ))}
84
+ </dl>
85
+ <div className="text-xs text-muted-foreground">
86
+ Click to add a configured cell.
87
+ </div>
88
+ </div>
89
+ );
90
+
91
+ export const AutoDiscoveredDataSources: React.FC<{
92
+ onSubmit: () => void;
93
+ }> = ({ onSubmit }) => {
94
+ const insertCode = useInsertCode();
95
+ const { data } = useDataSourceDiscovery();
96
+
97
+ return (
98
+ <QuickAddDataSources
99
+ sources={data ?? []}
100
+ onAdd={(source) => {
101
+ insertCode(source.code);
102
+ onSubmit();
103
+ }}
104
+ />
105
+ );
106
+ };
@@ -27,6 +27,7 @@ import type { HotkeyProvider } from "@/core/hotkeys/hotkeys";
27
27
  import { store } from "@/core/state/jotai";
28
28
  import type { PlaceholderType } from "../../config/types";
29
29
  import { TestSQLCompletionStore } from "../languages/sql/completion-store";
30
+ import * as sqlMode from "../languages/sql/sql-mode";
30
31
  import {
31
32
  exportedForTesting,
32
33
  SQLLanguageAdapter,
@@ -971,13 +972,63 @@ describe("SQL analysis features", () => {
971
972
  });
972
973
 
973
974
  describe("CustomSqlParser", () => {
975
+ beforeEach(() => {
976
+ store.set(dataSourceConnectionsAtom, {
977
+ connectionsMap: new Map(),
978
+ latestEngineSelected: DUCKDB_ENGINE,
979
+ });
980
+ });
981
+
974
982
  afterEach(() => {
975
983
  vi.useRealTimers();
976
984
  vi.restoreAllMocks();
985
+ store.set(dataSourceConnectionsAtom, {
986
+ connectionsMap: new Map(),
987
+ latestEngineSelected: DUCKDB_ENGINE,
988
+ });
977
989
  });
978
990
 
991
+ const createNamedConnectionState = ({
992
+ doc,
993
+ engine,
994
+ dialect,
995
+ }: {
996
+ doc: string;
997
+ engine: ConnectionName;
998
+ dialect: string;
999
+ }) => {
1000
+ store.set(dataSourceConnectionsAtom, {
1001
+ connectionsMap: new Map([
1002
+ [
1003
+ engine,
1004
+ {
1005
+ name: engine,
1006
+ dialect,
1007
+ display_name: engine,
1008
+ source: dialect,
1009
+ databases: [],
1010
+ },
1011
+ ],
1012
+ ]),
1013
+ latestEngineSelected: engine,
1014
+ });
1015
+ return EditorState.create({
1016
+ doc,
1017
+ extensions: [
1018
+ languageMetadataField.init(() => ({
1019
+ dataframeName: "_df",
1020
+ quotePrefix: "f",
1021
+ commentLines: [],
1022
+ showOutput: true,
1023
+ engine,
1024
+ })),
1025
+ ],
1026
+ });
1027
+ };
1028
+
979
1029
  it("uses backend DuckDB validation", async () => {
980
1030
  vi.useFakeTimers();
1031
+ vi.spyOn(sqlMode, "getSQLMode").mockReturnValue("validate");
981
1032
  const error = {
982
1033
  message: "Backend syntax error",
983
1034
  line: 1,
@@ -1009,13 +1060,146 @@ describe("CustomSqlParser", () => {
1009
1060
  await vi.runAllTimersAsync();
1010
1061
 
1011
1062
  await expect(result).resolves.toEqual([error]);
1012
- expect(request).toHaveBeenCalledWith(
1013
- expect.objectContaining({
1014
- engine: DUCKDB_ENGINE,
1015
- onlyParse: true,
1016
- query: "SELECT",
1017
- }),
1018
- );
1063
+ // No connection is registered for the internal engine, so the dialect
1064
+ // falls back to DEFAULT_PARSER_DIALECT ("DuckDB").
1065
+ expect(request).toHaveBeenCalledWith({
1066
+ dialect: "DuckDB",
1067
+ engine: DUCKDB_ENGINE,
1068
+ onlyParse: false,
1069
+ query: "SELECT",
1070
+ });
1071
+ });
1072
+
1073
+ it("resolves superseded validation requests instead of leaving them pending", async () => {
1074
+ vi.useFakeTimers();
1075
+ vi.spyOn(sqlMode, "getSQLMode").mockReturnValue("validate");
1076
+ const error = {
1077
+ message: "Backend syntax error",
1078
+ line: 1,
1079
+ column: 1,
1080
+ severity: "error" as const,
1081
+ };
1082
+ vi.spyOn(ValidateSQL, "request").mockResolvedValue({
1083
+ error: null,
1084
+ parse_result: { success: false, errors: [error] },
1085
+ request_id: "request-id",
1086
+ validate_result: null,
1087
+ });
1088
+ const state = EditorState.create({
1089
+ doc: "SELECT",
1090
+ extensions: [
1091
+ languageMetadataField.init(() => ({
1092
+ dataframeName: "_df",
1093
+ quotePrefix: "f",
1094
+ commentLines: [],
1095
+ showOutput: true,
1096
+ engine: DUCKDB_ENGINE,
1097
+ })),
1098
+ ],
1099
+ });
1100
+ const parser = new exportedForTesting.CustomSqlParser();
1101
+ parser.setFocusState(true);
1102
+
1103
+ // Simulate a rapid edit: a second validation call arrives before the
1104
+ // first call's internal debounce timer has fired. The first call should
1105
+ // resolve (with no errors) rather than hang forever.
1106
+ const first = parser.validateSql("SELECT 1", { state });
1107
+ await vi.advanceTimersByTimeAsync(100);
1108
+ const second = parser.validateSql("SELECT 2", { state });
1109
+ await vi.runAllTimersAsync();
1110
+
1111
+ await expect(first).resolves.toEqual([]);
1112
+ await expect(second).resolves.toEqual([error]);
1113
+ });
1114
+
1115
+ it("uses backend validation for a named DuckDB connection", async () => {
1116
+ vi.useFakeTimers();
1117
+ vi.spyOn(sqlMode, "getSQLMode").mockReturnValue("validate");
1118
+ const query =
1119
+ 'INSERT OR IGNORE INTO "rebalance-dates" (Timestamp) VALUES (CURRENT_TIMESTAMP);';
1120
+ const error = {
1121
+ message: "Named DuckDB backend result",
1122
+ line: 1,
1123
+ column: 1,
1124
+ severity: "error" as const,
1125
+ };
1126
+ const request = vi.spyOn(ValidateSQL, "request").mockResolvedValue({
1127
+ error: null,
1128
+ parse_result: { success: false, errors: [error] },
1129
+ request_id: "request-id",
1130
+ validate_result: null,
1131
+ });
1132
+ const state = createNamedConnectionState({
1133
+ doc: query,
1134
+ engine: "con" as ConnectionName,
1135
+ dialect: "duckdb",
1136
+ });
1137
+ const parser = new exportedForTesting.CustomSqlParser();
1138
+ parser.setFocusState(true);
1139
+
1140
+ const result = parser.validateSql(query, { state });
1141
+ await vi.runAllTimersAsync();
1142
+
1143
+ await expect(result).resolves.toEqual([error]);
1144
+ expect(request).toHaveBeenCalledWith({
1145
+ dialect: "DuckDB",
1146
+ engine: "con",
1147
+ onlyParse: true,
1148
+ query,
1149
+ });
1150
+ });
1151
+
1152
+ it("skips client-side parsing for named DuckDB SQL", async () => {
1153
+ const query = "CHECKPOINT;";
1154
+ const state = createNamedConnectionState({
1155
+ doc: query,
1156
+ engine: "con" as ConnectionName,
1157
+ dialect: "duckdb",
1158
+ });
1159
+ const parser = new exportedForTesting.CustomSqlParser();
1160
+
1161
+ await expect(parser.parse(query, { state })).resolves.toEqual({
1162
+ success: true,
1163
+ errors: [],
1164
+ });
1165
+ });
1166
+
1167
+ it("preserves client-side parsing for non-DuckDB connections", async () => {
1168
+ const query = "SELECT TOP 1 * FROM users";
1169
+ const state = createNamedConnectionState({
1170
+ doc: query,
1171
+ engine: "postgres_con" as ConnectionName,
1172
+ dialect: "postgres",
1173
+ });
1174
+ const parser = new exportedForTesting.CustomSqlParser();
1175
+
1176
+ await expect(parser.parse(query, { state })).resolves.toMatchObject({
1177
+ success: false,
1178
+ });
1179
+ });
1180
+
1181
+ it("preserves client-side parsing for an unregistered connection", async () => {
1182
+ // No entry is registered in the data source connections store for this
1183
+ // engine, so its dialect cannot be resolved (unlike a known non-DuckDB
1184
+ // dialect, which resolves to a non-null, non-DuckDB value).
1185
+ const query = "SELECT TOP 1 * FROM users";
1186
+ const state = EditorState.create({
1187
+ doc: query,
1188
+ extensions: [
1189
+ languageMetadataField.init(() => ({
1190
+ dataframeName: "_df",
1191
+ quotePrefix: "f",
1192
+ commentLines: [],
1193
+ showOutput: true,
1194
+ engine: "unregistered_con" as ConnectionName,
1195
+ })),
1196
+ ],
1197
+ });
1198
+ const parser = new exportedForTesting.CustomSqlParser();
1199
+
1200
+ await expect(parser.parse(query, { state })).resolves.toMatchObject({
1201
+ success: false,
1202
+ });
1019
1203
  });
1020
1204
  });
1021
1205