@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.
- package/dist/assets/{CellStatus-DciB5rbr.js → CellStatus-BfPw99P2.js} +1 -1
- package/dist/assets/{JsonOutput-BM3vve9F.js → JsonOutput-E3o1Uk0e.js} +12 -12
- package/dist/assets/{MarimoErrorOutput-Bh0y53Du.js → MarimoErrorOutput-D9Dypu7Q.js} +4 -4
- package/dist/assets/{RenderHTML-B9jx7Dsm.js → RenderHTML-DRJxD5FQ.js} +1 -1
- package/dist/assets/{RunButton-DMtHbFuA.js → RunButton-y7Y9rjmF.js} +1 -1
- package/dist/assets/{add-cell-with-ai-BcRYriMM.js → add-cell-with-ai-ByfOWv0Z.js} +12 -12
- package/dist/assets/add-connection-dialog-DVSFAYXK.js +178 -0
- package/dist/assets/{agent-panel-B31W_VwU.js → agent-panel-B2OKzocQ.js} +4 -4
- package/dist/assets/{ai-model-dropdown-B3D05oQh.js → ai-model-dropdown-Dr2QbDlZ.js} +3 -3
- package/dist/assets/{app-config-button-D_mjovwU.js → app-config-button-DScmpqcV.js} +1 -1
- package/dist/assets/{blob-BJREJiQp.js → blob-BXcsY70H.js} +1 -1
- package/dist/assets/{cell-editor-BOhZHoLm.js → cell-editor-BuGvBHQI.js} +11 -11
- package/dist/assets/cell-link-DGv7Q4Mc.js +1 -0
- package/dist/assets/{cells-Jr-5Pc9f.js → cells-CMc2NsXm.js} +61 -61
- package/dist/assets/{chat-display-Djy1Jt6b.js → chat-display-CeQftKrl.js} +1 -1
- package/dist/assets/{chat-panel-B2nYSE_l.js → chat-panel-BP5hfvS7.js} +4 -4
- package/dist/assets/{chat-ui-DlG0En7w.js → chat-ui-DoA0lexh.js} +4 -4
- package/dist/assets/{column-preview-BUHlI5va.js → column-preview-BeR1SI6Z.js} +1 -1
- package/dist/assets/{command-palette-rEJmXCZM.js → command-palette-DZSYeVCO.js} +1 -1
- package/dist/assets/{common-CZc3fQI4.js → common-NWgkp5L1.js} +1 -1
- package/dist/assets/{components-GAYCj6Mx.js → components-Mbf0MEIC.js} +1 -1
- package/dist/assets/{datasource-AWkkAeiS.js → datasource-Wt8yUynT.js} +2 -2
- package/dist/assets/{dependency-graph-panel-wcqTPq6b.js → dependency-graph-panel-B7TXWKYs.js} +4 -4
- package/dist/assets/{documentation-panel-Eu8MIkoU.js → documentation-panel-D9MZUKaD.js} +1 -1
- package/dist/assets/{download-CQ7iZFJL.js → download-U_wgQ8mC.js} +2 -2
- package/dist/assets/{edit-page-Cw7nN6HC.js → edit-page-BBc9BGYT.js} +6 -6
- package/dist/assets/error-entries-Bs30Z6uG.js +8 -0
- package/dist/assets/{error-panel-BZzr0PA1.js → error-panel-KY679tDx.js} +1 -1
- package/dist/assets/{file-explorer-panel-C1W7g_Wp.js → file-explorer-panel-9y0hzNAa.js} +4 -4
- package/dist/assets/{file-icons-BAqZR08W.js → file-icons-D5l3kSYq.js} +1 -1
- package/dist/assets/{file-name-input-0MiZVw5r.js → file-name-input-D_xhjKFp.js} +1 -1
- package/dist/assets/{floating-outline-C6g75zrP.js → floating-outline-BdgJT2B9.js} +1 -1
- package/dist/assets/{focus-C0HXXg3V.js → focus-D64hL1tZ.js} +1 -1
- package/dist/assets/{form-DT44Rmhp.js → form-CoqwGbkO.js} +1 -1
- package/dist/assets/{form-C051zAev.js → form-b4jQBV3T.js} +1 -1
- package/dist/assets/{home-page-CJz9IHZ2.js → home-page-tVWY9YdF.js} +2 -2
- package/dist/assets/{html-to-image-CfjhKLA4.js → html-to-image-Dm7huiQj.js} +2 -2
- package/dist/assets/index-8Nlyq3mV.css +2 -0
- package/dist/assets/{index-CrEgp1mR.js → index-Dz8dLbnI.js} +6 -6
- package/dist/assets/{layout-D4bpsgVD.js → layout-DKUs0ozP.js} +5 -5
- package/dist/assets/{logs-panel-CsaY3O1m.js → logs-panel-BzS7VSh0.js} +1 -1
- package/dist/assets/{markdown-renderer-BgqeO4JR.js → markdown-renderer-uWTrQId1.js} +2 -2
- package/dist/assets/{name-cell-input-DcU8Rfrz.js → name-cell-input-BgxcWTfV.js} +1 -1
- package/dist/assets/{outline-panel-D-CHPZQM.js → outline-panel-BT3zzwfa.js} +1 -1
- package/dist/assets/{packages-panel-dq7QVZl1.js → packages-panel-BKfeGuve.js} +1 -1
- package/dist/assets/panels-4jSJxEYp.js +4 -0
- package/dist/assets/{process-output-NbwnUu05.js → process-output-DE8TTXPC.js} +1 -1
- package/dist/assets/{radio-group-rhQpGwU-.js → radio-group-Da80KtoT.js} +1 -1
- package/dist/assets/readonly-code-display-B8zhuSov.js +1 -0
- package/dist/assets/{readonly-python-code-Q_1Y3VyZ.js → readonly-python-code-Bmo6cMSq.js} +1 -1
- package/dist/assets/{reveal-component-CedpDwM8.js → reveal-component-Cv7WZ4r6.js} +2 -2
- package/dist/assets/{run-page-CGoux5cj.js → run-page-CA21zsME.js} +1 -1
- package/dist/assets/scratchpad-panel-Br7RFJkK.js +1 -0
- package/dist/assets/{session-panel-OIot3eyB.js → session-panel-BiUht4zM.js} +1 -1
- package/dist/assets/{skeleton-BRvjFlRu.js → skeleton-CY-B1asm.js} +1 -1
- package/dist/assets/{snippets-panel-YFyH2hUM.js → snippets-panel-CjiTlCpD.js} +1 -1
- package/dist/assets/{state-bHyw5u3R.js → state-BJ7IdtXW.js} +1 -1
- package/dist/assets/state-TgAqts79.js +3 -0
- package/dist/assets/{tracing-CDMM1BB2.js → tracing-DmQ7KQ3A.js} +1 -1
- package/dist/assets/{tracing-panel-BH5Zbvl9.js → tracing-panel-C1TOV9kj.js} +2 -2
- package/dist/assets/{tree-actions-DDveUIHK.js → tree-actions-enwITSy7.js} +1 -1
- package/dist/assets/useCellActionButton-CXvU9GmI.js +1 -0
- package/dist/assets/{useDeleteCell-BIopCngo.js → useDeleteCell-Bp3S6pL8.js} +1 -1
- package/dist/assets/{useDependencyPanelTab-snM337f9.js → useDependencyPanelTab-CewYv8cP.js} +1 -1
- package/dist/assets/{useNotebookActions-CsZfWFKy.js → useNotebookActions-l9NAqPKm.js} +4 -4
- package/dist/assets/useRunCells-B-KIwrVZ.js +1 -0
- package/dist/assets/{useSplitCell-BTmKY1CU.js → useSplitCell-B0jvSFsl.js} +1 -1
- package/dist/index.html +24 -24
- package/package.json +1 -1
- package/src/__mocks__/requests.ts +1 -0
- package/src/components/editor/connections/__tests__/quick-add-data-sources.test.tsx +113 -0
- package/src/components/editor/connections/add-connection-dialog.tsx +2 -0
- package/src/components/editor/connections/quick-add-data-sources.tsx +106 -0
- package/src/core/codemirror/language/__tests__/sql.test.ts +191 -7
- package/src/core/codemirror/language/languages/sql/sql.ts +41 -13
- package/src/core/datasets/data-source-discovery.ts +5 -0
- package/src/core/datasets/request-registry.ts +11 -0
- package/src/core/islands/bootstrap.ts +1 -0
- package/src/core/islands/bridge.ts +1 -0
- package/src/core/kernel/messages.ts +2 -0
- package/src/core/network/__tests__/requests-lazy.test.ts +15 -0
- package/src/core/network/__tests__/requests-network.test.ts +14 -0
- package/src/core/network/requests-lazy.ts +1 -0
- package/src/core/network/requests-network.ts +8 -0
- package/src/core/network/requests-static.ts +1 -0
- package/src/core/network/requests-toasting.tsx +1 -0
- package/src/core/network/types.ts +2 -0
- package/src/core/wasm/bridge.ts +10 -0
- package/src/core/websocket/useMarimoKernelConnection.tsx +4 -0
- package/src/hooks/__tests__/useDataSourceDiscovery.test.ts +80 -0
- package/src/hooks/useDataSourceDiscovery.ts +18 -0
- package/dist/assets/add-connection-dialog-syjRSNPT.js +0 -178
- package/dist/assets/cell-link-DqNEv3f9.js +0 -1
- package/dist/assets/error-entries-Cr8uVukN.js +0 -8
- package/dist/assets/index-D6Dp_PFj.css +0 -2
- package/dist/assets/panels-DWoniKhE.js +0 -4
- package/dist/assets/readonly-code-display-CVY2f7EZ.js +0 -1
- package/dist/assets/scratchpad-panel-BLuF25yR.js +0 -1
- package/dist/assets/state-DB_fxbhn.js +0 -3
- package/dist/assets/useCellActionButton-D5ix_g2a.js +0 -1
- 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-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
161
|
-
<link rel="modulepreload" crossorigin href="./assets/datasource-
|
|
162
|
-
<link rel="modulepreload" crossorigin href="./assets/state-
|
|
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-
|
|
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-
|
|
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-
|
|
173
|
-
<link rel="modulepreload" crossorigin href="./assets/focus-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
211
|
-
<link rel="modulepreload" crossorigin href="./assets/floating-outline-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
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
|
@@ -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
|
-
|
|
1013
|
-
|
|
1014
|
-
|
|
1015
|
-
|
|
1016
|
-
|
|
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
|
|