@marimo-team/frontend 0.24.3-dev3 → 0.24.3-dev4
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-LoTwk7fR.js → CellStatus-0QNZS4By.js} +1 -1
- package/dist/assets/{JsonOutput-CzIGPEZQ.js → JsonOutput-D2ZWLZs4.js} +7 -7
- package/dist/assets/{MarimoErrorOutput-DqpoYOvw.js → MarimoErrorOutput-BlmIJ9WY.js} +2 -2
- package/dist/assets/{RenderHTML-DwhMAd9l.js → RenderHTML-DnPB4qpJ.js} +1 -1
- package/dist/assets/{RunButton-BlpbweyH.js → RunButton-C7TxE07w.js} +1 -1
- package/dist/assets/{add-cell-with-ai-R-3DS79K.js → add-cell-with-ai-OQ4sSXg4.js} +12 -12
- package/dist/assets/{add-connection-dialog-BGfpw5lC.js → add-connection-dialog-MTOC2ycO.js} +73 -73
- package/dist/assets/{agent-panel-Cgt_K0tD.js → agent-panel-BdEDIHdc.js} +4 -4
- package/dist/assets/ai-model-dropdown-CmJEEhSC.js +3 -0
- package/dist/assets/app-config-button-W8Pytgb5.js +1 -0
- package/dist/assets/{blob-B2d_dHwI.js → blob-BrVH7IJp.js} +1 -1
- package/dist/assets/{cell-editor-Dw09J9yD.js → cell-editor-C7IiingQ.js} +8 -8
- package/dist/assets/{cell-link-6burT2XF.js → cell-link-CbHXmrt8.js} +1 -1
- package/dist/assets/{cells-Cp-G34Rz.js → cells-BBhW3FXj.js} +1 -1
- package/dist/assets/{chat-display-Bu405iyc.js → chat-display-CIpYPLlF.js} +1 -1
- package/dist/assets/{chat-panel-BlUwNY5j.js → chat-panel-CK3X_KPy.js} +2 -2
- package/dist/assets/{chat-ui-DGiVDzNh.js → chat-ui-BJwtx1SR.js} +4 -4
- package/dist/assets/{column-preview-riSXbYe_.js → column-preview-Coa5PyHk.js} +1 -1
- package/dist/assets/{command-palette-CCVzJcyf.js → command-palette-_xpwMniA.js} +1 -1
- package/dist/assets/{common-CHR7w7aQ.js → common-KqtqJsml.js} +1 -1
- package/dist/assets/{components-C7mWrmXr.js → components-Cpv2z-yV.js} +1 -1
- package/dist/assets/{datasource-DrODSIs0.js → datasource-B3qF2oaW.js} +2 -2
- package/dist/assets/{dependency-graph-panel-AY5BEVH2.js → dependency-graph-panel-Av1CMZ61.js} +3 -3
- package/dist/assets/{documentation-panel-CaJF7e-j.js → documentation-panel-BEz3Hjfu.js} +1 -1
- package/dist/assets/{download-DMxOzI7b.js → download-BgO9EN-W.js} +1 -1
- package/dist/assets/edit-page-DslcxB67.js +9 -0
- package/dist/assets/error-entries-BFzSRy80.js +5 -0
- package/dist/assets/{error-panel-Bhlvt6_H.js → error-panel-BtLTJDUq.js} +1 -1
- package/dist/assets/{file-explorer-panel-s0ohSG_L.js → file-explorer-panel-DEyChgqS.js} +1 -1
- package/dist/assets/{file-icons-DdGYuADJ.js → file-icons-BhotQqrU.js} +1 -1
- package/dist/assets/{file-name-input-K4MKESI_.js → file-name-input-CH-hNP6F.js} +1 -1
- package/dist/assets/{floating-outline-DaUHnl_e.js → floating-outline-Cmk3XDJi.js} +1 -1
- package/dist/assets/{home-page-BLd6rnbk.js → home-page-CUBuu5Go.js} +1 -1
- package/dist/assets/index-C5h1aWhw.js +29 -0
- package/dist/assets/{layout-DWnrxIo2.js → layout-CsLuqwD3.js} +5 -5
- package/dist/assets/{logs-panel-BGS2v3ba.js → logs-panel-BV4VZ_uH.js} +1 -1
- package/dist/assets/{markdown-renderer-BZKlbm2Q.js → markdown-renderer-751VNlrQ.js} +1 -1
- package/dist/assets/{name-cell-input-BMCdK-el.js → name-cell-input-DexJ7ZIH.js} +1 -1
- package/dist/assets/{outline-panel-Dr5u73V6.js → outline-panel-QU3nj1VW.js} +1 -1
- package/dist/assets/packages-panel-D7oDuI6v.js +1 -0
- package/dist/assets/panel-accordion-state-Dl0VcB79.js +1 -0
- package/dist/assets/panels-nM4QKfEI.js +4 -0
- package/dist/assets/process-output-CPUOlFhp.js +1 -0
- package/dist/assets/readonly-code-display-DP7DxRA1.js +1 -0
- package/dist/assets/{readonly-python-code-DpIyjHqY.js → readonly-python-code-CAYyabaV.js} +1 -1
- package/dist/assets/{reveal-component-nZIAapoD.js → reveal-component-D9DygPds.js} +2 -2
- package/dist/assets/rotate-cw-CESctpLh.js +1 -0
- package/dist/assets/{run-page-9pl8qhQR.js → run-page-CzO-uMUn.js} +1 -1
- package/dist/assets/{scratchpad-panel-BleGt2KM.js → scratchpad-panel-7YBL6E95.js} +1 -1
- package/dist/assets/session-panel-hnFaUP0Y.js +1 -0
- package/dist/assets/share-BZW3tzEX.js +1 -0
- package/dist/assets/{skeleton-BKeh8hf3.js → skeleton-CM7HxSva.js} +1 -1
- package/dist/assets/{snippets-panel-D5yy7kj0.js → snippets-panel-AdjuR23O.js} +1 -1
- package/dist/assets/state-DOrY1Jyf.js +1 -0
- package/dist/assets/state-DQHafeTV.js +3 -0
- package/dist/assets/{tracing-DNk3gmWv.js → tracing-CUlvXr_H.js} +1 -1
- package/dist/assets/{tracing-panel-Yqazr6w_.js → tracing-panel-CjyayE-l.js} +2 -2
- package/dist/assets/{tree-actions-CG5g7aUQ.js → tree-actions-mfuAcK1x.js} +1 -1
- package/dist/assets/{useAddCell-DdB_uBif.js → useAddCell-BR8kYnb6.js} +1 -1
- package/dist/assets/{useCellActionButton-B31neI3m.js → useCellActionButton-BhREXvvn.js} +1 -1
- package/dist/assets/{useDeleteCell-Bz49_cFj.js → useDeleteCell-DwaN6KAw.js} +1 -1
- package/dist/assets/useDependencyPanelTab-I-7B78cW.js +1 -0
- package/dist/assets/useInstallPackage-IUt497GC.js +1 -0
- package/dist/assets/useNotebookActions-D5_YLR42.js +10 -0
- package/dist/assets/useRunCells-BMQuitYY.js +1 -0
- package/dist/assets/{useSplitCell-Dy2_DmJR.js → useSplitCell-B7AvHoya.js} +1 -1
- package/dist/index.html +22 -22
- package/package.json +1 -1
- package/src/__mocks__/requests.ts +7 -2
- package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +143 -0
- package/src/components/editor/chrome/panels/packages-panel.tsx +145 -51
- package/src/components/editor/package-alert.tsx +25 -1
- package/src/core/packages/__tests__/useInstallPackage.test.tsx +47 -1
- package/src/core/packages/toast-components.tsx +26 -0
- package/src/core/packages/useInstallPackage.ts +7 -2
- package/src/core/wasm/bridge.ts +1 -0
- package/dist/assets/ai-model-dropdown-DCmiuk7x.js +0 -3
- package/dist/assets/app-config-button-DUJC6b6x.js +0 -1
- package/dist/assets/edit-page-xlMlcLnI.js +0 -9
- package/dist/assets/error-entries-Cv2m1yhc.js +0 -5
- package/dist/assets/index-4n22QjCD.js +0 -29
- package/dist/assets/packages-panel-BdpRg9tX.js +0 -1
- package/dist/assets/panel-accordion-state-DjUTN7i5.js +0 -1
- package/dist/assets/panels-CWn0Yh1D.js +0 -4
- package/dist/assets/process-output-Ck7-P_xp.js +0 -1
- package/dist/assets/readonly-code-display--S2daDa6.js +0 -1
- package/dist/assets/session-panel-DRGPlL5J.js +0 -1
- package/dist/assets/share-CPtBiNgr.js +0 -1
- package/dist/assets/state-B6AklCzD.js +0 -1
- package/dist/assets/state-Di9TbuV9.js +0 -3
- package/dist/assets/useDependencyPanelTab-C_ps0r--.js +0 -1
- package/dist/assets/useInstallPackage-DsD08SLK.js +0 -1
- package/dist/assets/useNotebookActions-CQtbk9CB.js +0 -10
- package/dist/assets/useRunCells-CX7oW5zy.js +0 -1
|
@@ -1,166 +1,166 @@
|
|
|
1
|
-
import{a as e}from"./rolldown-runtime-B0Z9INg1.js";import{t}from"./react-CBDbhulC.js";import{
|
|
2
|
-
`)}},Ee=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.connection.ssl?`, connect_args={'sslmode': 'require'}`:``,t=this.secrets.printPassword(this.connection.password,`POSTGRES_PASSWORD`,!0),
|
|
3
|
-
DATABASE_URL = f"postgresql://${
|
|
1
|
+
import{a as e}from"./rolldown-runtime-B0Z9INg1.js";import{t}from"./react-CBDbhulC.js";import{Ci as n,Et as r,It as i,Jo as a,Lt as o,Rn as s,Rt as c,Ui as l,jn as u,zt as d}from"./cells-BBhW3FXj.js";import{Cn as f,In as p,kt as m,ln as h,zt as g}from"./schemas-aTUejmFi.js";import{t as _}from"./compiler-runtime-CYIjoINj.js";import{t as v}from"./cn-ldLVeaqm.js";import{t as y}from"./jsx-runtime-B74pBk57.js";import{i as ee,n as te,r as ne,t as re}from"./tabs-Cy1qXLzH.js";import{i as b,t as ie}from"./tooltip-BUtPzKB4.js";import{r as x}from"./strings-CReV1AYG.js";import{t as ae}from"./components-DUYCXngU.js";import{E as oe,M as se,O as ce,T as le,a as S,j as ue,l as de,p as C,r as w,s as T,w as fe}from"./write-secret-modal-D6zoZgLY.js";import{t as pe}from"./plus-C4J9gIE1.js";import{t as me}from"./useIframeCapabilities-C7Di8f2y.js";var E=e(t(),1),D=_();function O(){return p().optional().describe(u.of({label:`Password`,inputType:`password`,optionRegex:`(password|passwd|pgpassword|db.?pass(word)?)`}))}function k(e,t){let n=p();return n=t?n.nonempty():n.optional(),n=n.describe(u.of({label:e||`Token`,inputType:`password`,optionRegex:`(token|api.?key|access.?token|auth.?token|(^|_)pat(_|$))`})),n}function A(){return p().optional().describe(u.of({label:`Warehouse Name`,optionRegex:`(warehouse|snowflake.?warehouse)`}))}function j(e,t){let n=p();return n=t?n.nonempty():n.optional(),n.describe(u.of({label:e||`URI`,optionRegex:`(uri|url|connection.?string|database.?url|jdbc)`}))}function M(e){return p().nonempty().describe(u.of({label:e||`Host`,optionRegex:`^(?!.*(kubernetes|gpg)).*(host(name)?|pghost|db.?host|database.?host|mysql.?host|postgres.?host|server.?host)`}))}function N(){return p().describe(u.of({label:`Database`,optionRegex:`(database|db.?name|pgdatabase|^DB$|^DATABASE$)`}))}function he(){return p().describe(u.of({label:`Schema`,optionRegex:`(^SCHEMA$|db.?schema|postgres.?schema|pg.?schema|_SCHEMA$)`}))}function P(){return p().nonempty().describe(u.of({label:`Username`,optionRegex:`(username|pguser|db.?user|^USER$|_USER$)`}))}function F(e){let t=a().describe(u.of({label:`Port`,inputType:`number`,placeholder:e?.toString()})).transform(Number).refine(e=>e>=0&&e<=65535,{message:`Port must be between 0 and 65535`});return e===void 0?t:t.default(e)}function ge(){return m().default(!1).describe(u.of({label:`Read Only`}))}var _e=f({type:h(`postgres`),host:M(),port:F(5432).optional(),database:N(),username:P(),password:O(),ssl:m().default(!1).describe(u.of({label:`Use SSL`}))}).describe(u.of({direction:`two-columns`})),ve=f({type:h(`mysql`),host:M(),port:F(3306),database:N(),username:P(),password:O(),ssl:m().default(!1).describe(u.of({label:`Use SSL`}))}).describe(u.of({direction:`two-columns`})),ye=f({type:h(`sqlite`),database:N().describe(u.of({label:`Database Path`}))}).describe(u.of({direction:`two-columns`})),be=f({type:h(`duckdb`),database:N().describe(u.of({label:`Database Path`})),read_only:ge()}).describe(u.of({direction:`two-columns`})),I=f({type:h(`motherduck`),database:N().default(`my_db`).describe(u.of({label:`Database Name`})),token:k()}).describe(u.of({direction:`two-columns`})),L=f({type:h(`snowflake`),account:p().nonempty().describe(u.of({label:`Account`,optionRegex:`(snowflake.?account|^SNOWFLAKE_ACCOUNT$|account.?id)`})),warehouse:p().optional().describe(u.of({label:`Warehouse`,optionRegex:`(snowflake.?warehouse|warehouse)`})),database:N(),schema:p().optional().describe(u.of({label:`Schema`,optionRegex:`(snowflake.?schema|^SCHEMA$|_SCHEMA$)`})),role:p().optional().describe(u.of({label:`Role`})),authType:g(`type`,[f({type:h(`Password`),username:P(),password:O(),enable_mfa:m().default(!1).describe(u.of({label:`Enable MFA (Duo Push)`}))}),f({type:h(`SSO (Browser)`),username:P()}),f({type:h(`Key Pair`),username:P(),private_key_path:p().nonempty().describe(u.of({label:`Private Key Path`,placeholder:`/path/to/rsa_key.p8`})),private_key_passphrase:p().optional().describe(u.of({label:`Private Key Passphrase`,inputType:`password`,optionRegex:`(passphrase|private.?key.?passphrase|key.?passphrase)`}))}),f({type:h(`OAuth / PAT`),token:k(`Token`,!0)})]).default({type:`Password`,username:`username`,enable_mfa:!1}).describe(u.of({special:`tabs`}))}).describe(u.of({direction:`two-columns`})),R=f({type:h(`bigquery`),project:p().nonempty().describe(u.of({label:`Project ID`,optionRegex:`(bigquery.?project|gcp.?project|google.?cloud.?project|project.?id)`})),dataset:p().nonempty().describe(u.of({label:`Dataset`,optionRegex:`(bigquery.?dataset|dataset)`})),credentials_json:p().describe(u.of({label:`Credentials JSON`,special:`secret_textarea`}))}).describe(u.of({direction:`two-columns`})),z=f({type:h(`clickhouse_connect`),host:M(),port:F(8123).optional(),username:P(),password:O(),secure:m().default(!1).describe(u.of({label:`Use HTTPs`})),proxy_path:p().optional().describe(u.of({label:`Proxy Path`,placeholder:`/clickhouse`}))}).describe(u.of({direction:`two-columns`})),B=f({type:h(`timeplus`),host:M().default(`localhost`),port:F(8123).optional(),username:P().default(`default`),password:O().default(``)}).describe(u.of({direction:`two-columns`})),V=f({type:h(`chdb`),database:N().describe(u.of({label:`Database Path`})),read_only:ge()}).describe(u.of({direction:`two-columns`})),H=f({type:h(`trino`),host:M(),port:F(8080),database:N(),schema:he().optional(),username:P(),password:O(),async_support:m().default(!1).describe(u.of({label:`Async Support`}))}).describe(u.of({direction:`two-columns`})),U=f({type:h(`iceberg`),name:p().describe(u.of({label:`Catalog Name`})),catalog:g(`type`,[f({type:h(`REST`),warehouse:A(),uri:p().optional().describe(u.of({label:`URI`,placeholder:`https://`,optionRegex:`(uri|url|connection.?string|database.?url|jdbc)`})),token:k()}),f({type:h(`SQL`),warehouse:A(),uri:p().optional().describe(u.of({label:`URI`,placeholder:`jdbc:iceberg://host:port/database`,optionRegex:`(uri|url|connection.?string|database.?url|jdbc)`}))}),f({type:h(`Hive`),warehouse:A(),uri:j()}),f({type:h(`Glue`),warehouse:A(),uri:j()}),f({type:h(`DynamoDB`),"dynamodb.profile-name":p().optional().describe(u.of({label:`Profile Name`})),"dynamodb.region":p().optional().describe(u.of({label:`Region`})),"dynamodb.access-key-id":p().optional().describe(u.of({label:`Access Key ID`,inputType:`password`,optionRegex:`(access.?key.?id|aws.?access.?key)`})),"dynamodb.secret-access-key":p().optional().describe(u.of({label:`Secret Access Key`,inputType:`password`,optionRegex:`(secret.?access.?key|aws.?secret)`})),"dynamodb.session-token":p().optional().describe(u.of({label:`Session Token`,inputType:`password`,optionRegex:`(session.?token|aws.?session)`}))})]).default({type:`REST`,token:void 0}).describe(u.of({special:`tabs`}))}),W=f({type:h(`datafusion`),sessionContext:m().optional().describe(u.of({label:`Use Session Context`}))}),G=f({type:h(`pyspark`),host:M().optional(),port:F().optional()}),K=f({type:h(`redshift`),host:M(),port:F(5439),connectionType:g(`type`,[f({type:h(`IAM credentials`),region:p().describe(u.of({label:`Region`})),aws_access_key_id:p().nonempty().describe(u.of({label:`AWS Access Key ID`,inputType:`password`,optionRegex:`(access.?key.?id|aws.?access.?key)`})),aws_secret_access_key:p().nonempty().describe(u.of({label:`AWS Secret Access Key`,inputType:`password`,optionRegex:`(secret.?access.?key|aws.?secret)`})),aws_session_token:p().optional().describe(u.of({label:`AWS Session Token`,inputType:`password`,optionRegex:`(session.?token|aws.?session)`}))}),f({type:h(`DB credentials`),user:P(),password:O()})]).default({type:`IAM credentials`,aws_access_key_id:``,aws_secret_access_key:``,region:``}),database:N()}).describe(u.of({direction:`two-columns`})),q=f({type:h(`databricks`),access_token:k(`Access Token`,!0),server_hostname:M(`Server Hostname`),http_path:j(`HTTP Path`,!0),catalog:p().optional().describe(u.of({label:`Catalog`})),schema:p().optional().describe(u.of({label:`Schema`}))}).describe(u.of({direction:`two-columns`})),xe=f({type:h(`supabase`),host:M(),port:F(5432).optional(),database:N(),username:P(),password:O(),disable_client_pooling:m().default(!1).describe(u.of({label:`Disable Client-Side Pooling`}))}).describe(u.of({direction:`two-columns`})),Se=g(`type`,[_e,ve,ye,be,I,L,R,z,B,V,H,U,W,G,K,q,xe]),Ce={sqlmodel:`SQLModel`,sqlalchemy:`SQLAlchemy`,duckdb:`DuckDB`,clickhouse_connect:`ClickHouse Connect`,chdb:`chDB`,pyiceberg:`PyIceberg`,ibis:`Ibis`,motherduck:`MotherDuck`,redshift:`Redshift`,databricks:`Databricks`},J=class{connection;orm;secrets;constructor(e,t,n){this.connection=e,this.orm=t,this.secrets=n}get imports(){let e=new Set(this.generateImports());switch(this.orm){case`sqlalchemy`:e.add(`import sqlalchemy`);break;case`sqlmodel`:e.add(`import sqlmodel`);break;case`duckdb`:e.add(`import duckdb`);break;case`ibis`:e.add(`import ibis`)}return e}},we=e=>`_${e}`,Te=class{secrets={};get imports(){return Object.keys(this.secrets).length===0?new Set:new Set([`import os`])}print(e,t,n){if(e=we(e),T(t)){let r=de(t),i=n?`os.environ.get("${r}", "${n}")`:`os.environ.get("${r}")`;return this.secrets[e]=i,e}if(n!=null){let r=`os.environ.get("${t}", "${n}")`;return this.secrets[e]=r,e}return typeof t==`number`||typeof t==`number`?`${t}`:typeof t==`boolean`?Y(t):t?`"${t}"`:``}printInFString(e,t,n){if(t===void 0)return``;if(typeof t==`number`)return`${t}`;if(typeof t==`boolean`)return Y(t);let r=this.print(e,t,n);return r.startsWith(`"`)&&r.endsWith(`"`)?r.slice(1,-1):`{${r}}`}printPassword(e,t,n,r){let i=n?this.printInFString.bind(this):this.print.bind(this);return T(e)?i(r||`password`,e):i(r||`password`,t,e)}getSecrets(){return this.secrets}formatSecrets(){return Object.keys(this.secrets).length===0?``:Object.entries(this.secrets).map(([e,t])=>`${e} = ${t}`).join(`
|
|
2
|
+
`)}},Ee=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.connection.ssl?`, connect_args={'sslmode': 'require'}`:``,t=this.secrets.printPassword(this.connection.password,`POSTGRES_PASSWORD`,!0),r=this.secrets.printInFString(`username`,this.connection.username),i=this.secrets.printInFString(`host`,this.connection.host),a=this.secrets.printInFString(`port`,this.connection.port),o=this.secrets.printInFString(`database`,this.connection.database);return n(`
|
|
3
|
+
DATABASE_URL = f"postgresql://${r}:${t}@${i}:${a}/${o}"
|
|
4
4
|
engine = ${this.orm}.create_engine(DATABASE_URL${e})
|
|
5
|
-
`)}},De=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.connection.ssl?`, connect_args={'ssl': {'ssl-mode': 'preferred'}}`:``,t=this.secrets.printPassword(this.connection.password,`MYSQL_PASSWORD`,!0),
|
|
6
|
-
DATABASE_URL = f"mysql+pymysql://${
|
|
5
|
+
`)}},De=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.connection.ssl?`, connect_args={'ssl': {'ssl-mode': 'preferred'}}`:``,t=this.secrets.printPassword(this.connection.password,`MYSQL_PASSWORD`,!0),r=this.secrets.printInFString(`database`,this.connection.database),i=this.secrets.printInFString(`username`,this.connection.username),a=this.secrets.printInFString(`host`,this.connection.host),o=this.secrets.printInFString(`port`,this.connection.port);return n(`
|
|
6
|
+
DATABASE_URL = f"mysql+pymysql://${i}:${t}@${a}:${o}/${r}"
|
|
7
7
|
engine = ${this.orm}.create_engine(DATABASE_URL${e})
|
|
8
|
-
`)}},Oe=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.connection.database?this.secrets.printInFString(`database`,this.connection.database):``,t=e.startsWith(`{`)&&e.endsWith(`}`)?`DATABASE_URL = f"sqlite:///${e}"`:`DATABASE_URL = "sqlite:///${e}"`;return
|
|
8
|
+
`)}},Oe=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.connection.database?this.secrets.printInFString(`database`,this.connection.database):``,t=e.startsWith(`{`)&&e.endsWith(`}`)?`DATABASE_URL = f"sqlite:///${e}"`:`DATABASE_URL = "sqlite:///${e}"`;return n(`
|
|
9
9
|
${t}
|
|
10
10
|
engine = ${this.orm}.create_engine(DATABASE_URL)
|
|
11
|
-
`)}},ke=class extends J{generateImports(){return[`from snowflake.sqlalchemy import URL`]}generateConnectionCode(){let{authType:e,account:t,database:
|
|
11
|
+
`)}},ke=class extends J{generateImports(){return[`from snowflake.sqlalchemy import URL`]}generateConnectionCode(){let{authType:e,account:t,database:r,warehouse:i,schema:a,role:o}=this.connection,s={account:this.secrets.print(`account`,t),database:this.secrets.print(`database`,r),warehouse:i?this.secrets.print(`warehouse`,i):void 0,schema:a?this.secrets.print(`schema`,a):void 0,role:o?this.secrets.print(`role`,o):void 0};switch(e.type){case`Password`:{let t=this.secrets.printPassword(e.password,`SNOWFLAKE_PASSWORD`,!1),r={...s,user:this.secrets.print(`user`,e.username),password:t};return e.enable_mfa?n(`
|
|
12
12
|
engine = ${this.orm}.create_engine(
|
|
13
13
|
URL(
|
|
14
|
-
${X(
|
|
14
|
+
${X(r,e=>` ${e}`)},
|
|
15
15
|
),
|
|
16
16
|
connect_args={"authenticator": "username_password_mfa"},
|
|
17
17
|
)
|
|
18
|
-
`):
|
|
18
|
+
`):n(`
|
|
19
19
|
engine = ${this.orm}.create_engine(
|
|
20
20
|
URL(
|
|
21
|
-
${X(
|
|
21
|
+
${X(r,e=>` ${e}`)},
|
|
22
22
|
)
|
|
23
23
|
)
|
|
24
|
-
`)}case`SSO (Browser)`:{let t={...
|
|
24
|
+
`)}case`SSO (Browser)`:{let t={...s,user:this.secrets.print(`user`,e.username),authenticator:`"externalbrowser"`};return n(`
|
|
25
25
|
engine = ${this.orm}.create_engine(
|
|
26
26
|
URL(
|
|
27
27
|
${X(t,e=>` ${e}`)},
|
|
28
28
|
)
|
|
29
29
|
)
|
|
30
|
-
`)}case`Key Pair`:{let t={...
|
|
30
|
+
`)}case`Key Pair`:{let t={...s,user:this.secrets.print(`user`,e.username)},r=this.secrets.print(`private_key_path`,e.private_key_path),i=e.private_key_passphrase?this.secrets.printPassword(e.private_key_passphrase,`SNOWFLAKE_PRIVATE_KEY_PASSPHRASE`,!1):void 0,a=[` "authenticator": "SNOWFLAKE_JWT"`,` "private_key_file": ${r}`];return i&&a.push(` "private_key_file_pwd": ${i}`),n(`
|
|
31
31
|
engine = ${this.orm}.create_engine(
|
|
32
32
|
URL(
|
|
33
33
|
${X(t,e=>` ${e}`)},
|
|
34
34
|
),
|
|
35
35
|
connect_args={
|
|
36
|
-
${
|
|
36
|
+
${a.join(`,
|
|
37
37
|
`)},
|
|
38
38
|
},
|
|
39
39
|
)
|
|
40
|
-
`)}case`OAuth / PAT`:{let t=this.secrets.printPassword(e.token,`SNOWFLAKE_TOKEN`,!1),
|
|
40
|
+
`)}case`OAuth / PAT`:{let t=this.secrets.printPassword(e.token,`SNOWFLAKE_TOKEN`,!1),r={...s,authenticator:`"oauth"`,token:t};return n(`
|
|
41
41
|
engine = ${this.orm}.create_engine(
|
|
42
42
|
URL(
|
|
43
|
-
${X(
|
|
43
|
+
${X(r,e=>` ${e}`)},
|
|
44
44
|
)
|
|
45
45
|
)
|
|
46
|
-
`)}default:x(e)}}},Ae=class extends J{get credential(){return S(this.connection.credentials_json,e=>this.secrets.print(`credentials_json`,e))}generateImports(){return this.credential.kind===`path`?[]:[`import json`]}generateConnectionCode(){let e=this.secrets.printInFString(`project`,this.connection.project),t=this.secrets.printInFString(`dataset`,this.connection.dataset),
|
|
47
|
-
engine = ${this.orm}.create_engine(f"bigquery://${e}/${t}", credentials_path="${w(
|
|
48
|
-
`):
|
|
49
|
-
credentials = ${
|
|
46
|
+
`)}default:x(e)}}},Ae=class extends J{get credential(){return S(this.connection.credentials_json,e=>this.secrets.print(`credentials_json`,e))}generateImports(){return this.credential.kind===`path`?[]:[`import json`]}generateConnectionCode(){let e=this.secrets.printInFString(`project`,this.connection.project),t=this.secrets.printInFString(`dataset`,this.connection.dataset),r=this.credential;return r.kind===`path`?n(`
|
|
47
|
+
engine = ${this.orm}.create_engine(f"bigquery://${e}/${t}", credentials_path="${w(r.path)}")
|
|
48
|
+
`):n(`
|
|
49
|
+
credentials = ${r.expr}
|
|
50
50
|
engine = ${this.orm}.create_engine(f"bigquery://${e}/${t}", credentials_info=credentials)
|
|
51
|
-
`)}},je=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.secrets.printInFString(`database`,this.connection.database||`:memory:`);return
|
|
51
|
+
`)}},je=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.secrets.printInFString(`database`,this.connection.database||`:memory:`);return n(`
|
|
52
52
|
DATABASE_URL = "${e}"
|
|
53
53
|
engine = ${this.orm}.connect(DATABASE_URL, read_only=${Y(this.connection.read_only)})
|
|
54
|
-
`)}},Me=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.secrets.printInFString(`database`,this.connection.database);if(!this.connection.token)return
|
|
54
|
+
`)}},Me=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.secrets.printInFString(`database`,this.connection.database);if(!this.connection.token)return n(`
|
|
55
55
|
conn = duckdb.connect("md:${e}")
|
|
56
|
-
`);let t=this.secrets.printPassword(this.connection.token,`MOTHERDUCK_TOKEN`,!1);return
|
|
56
|
+
`);let t=this.secrets.printPassword(this.connection.token,`MOTHERDUCK_TOKEN`,!1);return n(`
|
|
57
57
|
conn = duckdb.connect("md:${e}", config={"motherduck_token": ${t}})
|
|
58
|
-
`)}},Ne=class extends J{generateImports(){return[`import clickhouse_connect`]}generateConnectionCode(){let e=this.secrets.printPassword(this.connection.password,`CLICKHOUSE_PASSWORD`,!1),t={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?e:void 0,proxy_path:this.connection.proxy_path?this.secrets.print(`proxy_path`,this.connection.proxy_path):void 0};return
|
|
58
|
+
`)}},Ne=class extends J{generateImports(){return[`import clickhouse_connect`]}generateConnectionCode(){let e=this.secrets.printPassword(this.connection.password,`CLICKHOUSE_PASSWORD`,!1),t={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?e:void 0,proxy_path:this.connection.proxy_path?this.secrets.print(`proxy_path`,this.connection.proxy_path):void 0};return n(`
|
|
59
59
|
engine = ${this.orm}.get_client(
|
|
60
60
|
${X(t,e=>` ${e}`)},
|
|
61
61
|
)
|
|
62
|
-
`)}},Pe=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.secrets.printPassword(this.connection.password,`TIMEPLUS_PASSWORD`,!0),t=this.secrets.printInFString(`username`,this.connection.username),
|
|
63
|
-
DATABASE_URL = f"timeplus://${t}:${e}@${
|
|
62
|
+
`)}},Pe=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.secrets.printPassword(this.connection.password,`TIMEPLUS_PASSWORD`,!0),t=this.secrets.printInFString(`username`,this.connection.username),r=this.secrets.printInFString(`host`,this.connection.host),i=this.secrets.printInFString(`port`,this.connection.port);return n(`
|
|
63
|
+
DATABASE_URL = f"timeplus://${t}:${e}@${r}:${i}"
|
|
64
64
|
engine = ${this.orm}.create_engine(DATABASE_URL)
|
|
65
|
-
`)}},Fe=class extends J{generateImports(){return[`import chdb`]}generateConnectionCode(){let e=this.secrets.print(`database`,this.connection.database)||`""`;return
|
|
65
|
+
`)}},Fe=class extends J{generateImports(){return[`import chdb`]}generateConnectionCode(){let e=this.secrets.print(`database`,this.connection.database)||`""`;return n(`
|
|
66
66
|
engine = ${this.orm}.connect(${e}, read_only=${Y(this.connection.read_only)})
|
|
67
|
-
`)}},Ie=class extends J{generateImports(){return this.connection.async_support?[`import aiotrino`]:[`import trino.sqlalchemy`]}generateConnectionCode(){let e=this.connection.async_support?`aiotrino`:`trino`,t=this.connection.schema?`/${this.connection.schema}`:``,
|
|
68
|
-
engine = ${this.orm}.create_engine(f"${e}://${
|
|
69
|
-
`)}},Le=class extends J{getCatalogOptions(){return Object.fromEntries(Object.entries(this.connection.catalog).filter(([e,t])=>t!=null&&t!==``&&e!==`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:x(this.connection.catalog)}}generateConnectionCode(){let e=`"${this.connection.name}"`,t=this.getCatalogOptions();if(Object.keys(t).length===0)return`catalog = load_catalog(${e})`;for(let[e,n]of Object.entries(t))T(n)?t[e]=this.secrets.print(e,n):typeof n==`string`&&(t[e]=`"${n}"`);let
|
|
67
|
+
`)}},Ie=class extends J{generateImports(){return this.connection.async_support?[`import aiotrino`]:[`import trino.sqlalchemy`]}generateConnectionCode(){let e=this.connection.async_support?`aiotrino`:`trino`,t=this.connection.schema?`/${this.connection.schema}`:``,r=this.secrets.printInFString(`username`,this.connection.username),i=this.secrets.printInFString(`host`,this.connection.host),a=this.secrets.printInFString(`port`,this.connection.port),o=this.secrets.printInFString(`database`,this.connection.database),s=this.secrets.printPassword(this.connection.password,`TRINO_PASSWORD`,!0);return n(`
|
|
68
|
+
engine = ${this.orm}.create_engine(f"${e}://${r}:${s}@${i}:${a}/${o}${t}")
|
|
69
|
+
`)}},Le=class extends J{getCatalogOptions(){return Object.fromEntries(Object.entries(this.connection.catalog).filter(([e,t])=>t!=null&&t!==``&&e!==`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:x(this.connection.catalog)}}generateConnectionCode(){let e=`"${this.connection.name}"`,t=this.getCatalogOptions();if(Object.keys(t).length===0)return`catalog = load_catalog(${e})`;for(let[e,n]of Object.entries(t))T(n)?t[e]=this.secrets.print(e,n):typeof n==`string`&&(t[e]=`"${n}"`);let r=Ge(t,e=>` ${e}`);switch(this.connection.catalog.type){case`REST`:return n(`
|
|
70
70
|
catalog = RestCatalog(
|
|
71
71
|
${e},
|
|
72
|
-
**{\n${
|
|
72
|
+
**{\n${r}
|
|
73
73
|
},
|
|
74
74
|
)
|
|
75
|
-
`);case`SQL`:return
|
|
75
|
+
`);case`SQL`:return n(`
|
|
76
76
|
catalog = SqlCatalog(
|
|
77
77
|
${e},
|
|
78
|
-
**{\n${
|
|
78
|
+
**{\n${r}
|
|
79
79
|
},
|
|
80
80
|
)
|
|
81
|
-
`);case`Hive`:return
|
|
81
|
+
`);case`Hive`:return n(`
|
|
82
82
|
catalog = HiveCatalog(
|
|
83
83
|
${e},
|
|
84
|
-
**{\n${
|
|
84
|
+
**{\n${r}
|
|
85
85
|
},
|
|
86
86
|
)
|
|
87
|
-
`);case`Glue`:return
|
|
87
|
+
`);case`Glue`:return n(`
|
|
88
88
|
catalog = GlueCatalog(
|
|
89
89
|
${e},
|
|
90
|
-
**{\n${
|
|
90
|
+
**{\n${r}
|
|
91
91
|
},
|
|
92
92
|
)
|
|
93
|
-
`);case`DynamoDB`:return
|
|
93
|
+
`);case`DynamoDB`:return n(`
|
|
94
94
|
catalog = DynamoDBCatalog(
|
|
95
95
|
${e},
|
|
96
|
-
**{\n${
|
|
96
|
+
**{\n${r}
|
|
97
97
|
},
|
|
98
98
|
)
|
|
99
|
-
`);default:x(this.connection.catalog)}}},Re=class extends J{generateImports(){return[`from datafusion import SessionContext`]}generateConnectionCode(){return this.connection.sessionContext?
|
|
99
|
+
`);default:x(this.connection.catalog)}}},Re=class extends J{generateImports(){return[`from datafusion import SessionContext`]}generateConnectionCode(){return this.connection.sessionContext?n(`
|
|
100
100
|
ctx = SessionContext()
|
|
101
101
|
# Sample table
|
|
102
102
|
_ = ctx.from_pydict({"a": [1, 2, 3]}, "my_table")
|
|
103
103
|
|
|
104
104
|
con = ibis.datafusion.connect(ctx)
|
|
105
|
-
`):
|
|
105
|
+
`):n(`
|
|
106
106
|
con = ibis.datafusion.connect()
|
|
107
|
-
`)}},ze=class extends J{generateImports(){return[`from pyspark.sql import SparkSession`]}generateConnectionCode(){if(this.connection.host||this.connection.port){let e=this.secrets.printInFString(`host`,this.connection.host),t=this.secrets.printInFString(`port`,this.connection.port);return
|
|
107
|
+
`)}},ze=class extends J{generateImports(){return[`from pyspark.sql import SparkSession`]}generateConnectionCode(){if(this.connection.host||this.connection.port){let e=this.secrets.printInFString(`host`,this.connection.host),t=this.secrets.printInFString(`port`,this.connection.port);return n(`
|
|
108
108
|
session = SparkSession.builder.remote(f"sc://${e}:${t}").getOrCreate()
|
|
109
109
|
con = ibis.pyspark.connect(session)
|
|
110
|
-
`)}return
|
|
110
|
+
`)}return n(`
|
|
111
111
|
con = ibis.pyspark.connect()
|
|
112
|
-
`)}},Be=class extends J{generateImports(){return[`import redshift_connector`]}generateConnectionCode(){let e=this.secrets.print(`host`,this.connection.host),t=this.secrets.print(`port`,this.connection.port),
|
|
112
|
+
`)}},Be=class extends J{generateImports(){return[`import redshift_connector`]}generateConnectionCode(){let e=this.secrets.print(`host`,this.connection.host),t=this.secrets.print(`port`,this.connection.port),r=this.secrets.print(`database`,this.connection.database);if(this.connection.connectionType.type===`IAM credentials`){let i=this.secrets.print(`aws_access_key_id`,this.connection.connectionType.aws_access_key_id),a=this.secrets.print(`aws_secret_access_key`,this.connection.connectionType.aws_secret_access_key),o=this.connection.connectionType.aws_session_token?this.secrets.print(`aws_session_token`,this.connection.connectionType.aws_session_token):void 0,s={iam:!0,host:e,port:t,region:`"${this.connection.connectionType.region}"`,database:r,access_key_id:i,secret_access_key:a,...o&&{session_token:o}};return n(`
|
|
113
113
|
con = redshift_connector.connect(
|
|
114
|
-
${X(
|
|
114
|
+
${X(s,e=>` ${e}`)},
|
|
115
115
|
)
|
|
116
|
-
`)}let
|
|
116
|
+
`)}let i=this.connection.connectionType.user?this.secrets.print(`user`,this.connection.connectionType.user):void 0,a=this.connection.connectionType.password?this.secrets.printPassword(this.connection.connectionType.password,`REDSHIFT_PASSWORD`,!1):void 0,o={host:e,port:t,database:r,...i&&{user:i},...a&&{password:a}};return n(`
|
|
117
117
|
con = redshift_connector.connect(
|
|
118
|
-
${X(
|
|
118
|
+
${X(o,e=>` ${e}`)},
|
|
119
119
|
)
|
|
120
|
-
`)}},Ve=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.orm!==`ibis`,t=this.secrets.printPassword(this.connection.access_token,`DATABRICKS_ACCESS_TOKEN`,e,`access_token`),
|
|
120
|
+
`)}},Ve=class extends J{generateImports(){return[]}generateConnectionCode(){let e=this.orm!==`ibis`,t=this.secrets.printPassword(this.connection.access_token,`DATABRICKS_ACCESS_TOKEN`,e,`access_token`),r=this.secrets.printPassword(this.connection.server_hostname,`DATABRICKS_SERVER_HOSTNAME`,e,`server_hostname`),i=this.secrets.printPassword(this.connection.http_path,`DATABRICKS_HTTP_PATH`,e,`http_path`),a=this.connection.catalog?this.secrets.printInFString(`catalog`,this.connection.catalog):void 0,o=this.connection.schema?this.secrets.printInFString(`schema`,this.connection.schema):void 0,s=`databricks://token:${t}@${r}?http_path=${i}`;if(a&&(s+=`&catalog=${a}`),o&&(s+=`&schema=${o}`),this.orm===`ibis`){let e=a?`\n catalog=${a},`:``,s=o?`\n schema=${o},`:``;return n(`
|
|
121
121
|
engine = ibis.databricks.connect(
|
|
122
|
-
server_hostname=${
|
|
123
|
-
http_path=${
|
|
122
|
+
server_hostname=${r},
|
|
123
|
+
http_path=${i},${e}${s}
|
|
124
124
|
access_token=${t}
|
|
125
125
|
)
|
|
126
|
-
`)}return
|
|
127
|
-
DATABASE_URL = f"${
|
|
126
|
+
`)}return n(`
|
|
127
|
+
DATABASE_URL = f"${s}"
|
|
128
128
|
engine = ${this.orm}.create_engine(DATABASE_URL)
|
|
129
|
-
`)}},He=class extends J{generateImports(){return this.connection.disable_client_pooling?[`from sqlalchemy.pool import NullPool`]:[]}generateConnectionCode(){let e=this.secrets.printPassword(this.connection.password,`SUPABASE_PASSWORD`,!0),t=this.secrets.printInFString(`username`,this.connection.username),
|
|
130
|
-
DATABASE_URL = f"postgresql+psycopg2://${t}:${e}@${
|
|
131
|
-
engine = ${this.orm}.create_engine(DATABASE_URL${
|
|
129
|
+
`)}},He=class extends J{generateImports(){return this.connection.disable_client_pooling?[`from sqlalchemy.pool import NullPool`]:[]}generateConnectionCode(){let e=this.secrets.printPassword(this.connection.password,`SUPABASE_PASSWORD`,!0),t=this.secrets.printInFString(`username`,this.connection.username),r=this.secrets.printInFString(`host`,this.connection.host),i=this.secrets.printInFString(`port`,this.connection.port),a=this.secrets.printInFString(`database`,this.connection.database),o=this.connection.disable_client_pooling?`, poolclass=NullPool`:``;return n(`
|
|
130
|
+
DATABASE_URL = f"postgresql+psycopg2://${t}:${e}@${r}:${i}/${a}?sslmode=require"
|
|
131
|
+
engine = ${this.orm}.create_engine(DATABASE_URL${o})
|
|
132
132
|
`)}},Ue=class{secrets=new Te;createGenerator(e,t){switch(e.type){case`postgres`:return new Ee(e,t,this.secrets);case`mysql`:return new De(e,t,this.secrets);case`sqlite`:return new Oe(e,t,this.secrets);case`snowflake`:return new ke(e,t,this.secrets);case`bigquery`:return new Ae(e,t,this.secrets);case`duckdb`:return new je(e,t,this.secrets);case`motherduck`:return new Me(e,t,this.secrets);case`clickhouse_connect`:return new Ne(e,t,this.secrets);case`timeplus`:return new Pe(e,t,this.secrets);case`chdb`:return new Fe(e,t,this.secrets);case`trino`:return new Ie(e,t,this.secrets);case`iceberg`:return new Le(e,t,this.secrets);case`datafusion`:return new Re(e,t,this.secrets);case`pyspark`:return new ze(e,t,this.secrets);case`redshift`:return new Be(e,t,this.secrets);case`databricks`:return new Ve(e,t,this.secrets);case`supabase`:return new He(e,t,this.secrets);default:x(e)}}};function We(e,t){if(!(t in Ce))throw Error(`Unsupported library: ${t}`);Se.parse(e);let n=new Ue,r=n.createGenerator(e,t),i=r.generateConnectionCode(),a=n.secrets,o=[...new Set([...a.imports,...r.imports])].toSorted();o.push(``);let s=a.formatSecrets();return s&&o.push(s),o.push(i.trim()),o.join(`
|
|
133
133
|
`)}function Y(e){return e.toString().charAt(0).toUpperCase()+e.toString().slice(1)}function X(e,t){return Object.entries(e).filter(([,e])=>e!=null&&e!==``).map(([e,n])=>t(typeof n==`boolean`?`${e}=${Y(n)}`:`${e}=${n}`)).join(`,
|
|
134
134
|
`)}function Ge(e,t){return Object.entries(e).filter(([,e])=>e!=null&&e!==``).map(([e,n])=>{let r=`"${e}"`;return t(typeof n==`boolean`?`${r}: ${Y(n)}`:`${r}: ${n}`)}).join(`,
|
|
135
|
-
`)}var Z=y(),Ke=[{name:`PostgreSQL`,schema:_e,color:`#336791`,logo:`postgres`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`MySQL`,schema:ve,color:`#00758F`,logo:`mysql`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`SQLite`,schema:ye,color:`#003B57`,logo:`sqlite`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`DuckDB`,schema:be,color:`#FFD700`,logo:`duckdb`,connectionLibraries:{libraries:[`duckdb`],preferred:`duckdb`}},{name:`MotherDuck`,schema:I,color:`#ff9538`,logo:`motherduck`,connectionLibraries:{libraries:[`duckdb`],preferred:`duckdb`}},{name:`Snowflake`,schema:L,color:`#29B5E8`,logo:`snowflake`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`ClickHouse`,schema:z,color:`#2C2C1D`,logo:`clickhouse`,connectionLibraries:{libraries:[`clickhouse_connect`],preferred:`clickhouse_connect`}},{name:`Timeplus`,schema:B,color:`#B83280`,logo:`timeplus`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`BigQuery`,schema:R,color:`#4285F4`,logo:`bigquery`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`ClickHouse Embedded`,schema:V,color:`#f2b611`,logo:`clickhouse`,connectionLibraries:{libraries:[`chdb`],preferred:`chdb`}},{name:`Trino`,schema:H,color:`#d466b6`,logo:`trino`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`DataFusion`,schema:W,color:`#202A37`,logo:`datafusion`,connectionLibraries:{libraries:[`ibis`],preferred:`ibis`}},{name:`PySpark`,schema:G,color:`#1C5162`,logo:`pyspark`,connectionLibraries:{libraries:[`ibis`],preferred:`ibis`}},{name:`Redshift`,schema:K,color:`#522BAE`,logo:`redshift`,connectionLibraries:{libraries:[`redshift`],preferred:`redshift`}},{name:`Databricks`,schema:q,color:`#c41e0c`,logo:`databricks`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`,`ibis`],preferred:`sqlalchemy`}},{name:`Supabase`,schema:xe,color:`#238F5F`,logo:`supabase`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}}],qe=[{name:`Iceberg`,schema:U,color:`#000000`,logo:`iceberg`,connectionLibraries:{libraries:[`pyiceberg`],preferred:`pyiceberg`}}],Je=[...Ke,...qe],Ye=[...new Set(Je.flatMap(({name:e,logo:t})=>[e,t]))],Xe=e=>{let t=(0,D.c)(12),{onSelect:n}=e,r;t[0]===n?r=t[1]:(r=Ke.map(e=>{let{name:t,schema:r,color:i,logo:s}=e;return(0,Z.jsx)(a,{name:t,color:i,icon:(0,Z.jsx)(o,{name:s,className:`w-8 h-8 text-white brightness-0 invert dark:invert`}),onSelect:()=>n(r)},t)}),t[0]=n,t[1]=r);let i;t[2]===r?i=t[3]:(i=(0,Z.jsx)(s,{children:r}),t[2]=r,t[3]=i);let c;t[4]===Symbol.for(`react.memo_cache_sentinel`)?(c=(0,Z.jsxs)(`h4`,{className:`font-semibold text-muted-foreground text-lg flex items-center gap-4 my-2`,children:[`Data Catalogs`,(0,Z.jsx)(`hr`,{className:`flex-1`})]}),t[4]=c):c=t[4];let l;t[5]===n?l=t[6]:(l=qe.map(e=>{let{name:t,schema:r,color:i,logo:s}=e;return(0,Z.jsx)(a,{name:t,color:i,icon:(0,Z.jsx)(o,{name:s,className:`w-8 h-8 text-white brightness-0 invert dark:invert`}),onSelect:()=>n(r)},t)}),t[5]=n,t[6]=l);let u;t[7]===l?u=t[8]:(u=(0,Z.jsx)(s,{children:l}),t[7]=l,t[8]=u);let d;return t[9]!==i||t[10]!==u?(d=(0,Z.jsxs)(Z.Fragment,{children:[i,c,u]}),t[9]=i,t[10]=u,t[11]=d):d=t[11],d},Ze=e=>{let t=(0,D.c)(13),{onSubmit:n,header:i}=e,[a,o]=(0,E.useState)(null);if(!a){let e;t[0]===Symbol.for(`react.memo_cache_sentinel`)?(e=(0,Z.jsx)(`div`,{children:(0,Z.jsx)(Xe,{onSelect:o})}),t[0]=e):e=t[0];let n;return t[1]===i?n=t[2]:(n=(0,Z.jsxs)(Z.Fragment,{children:[i,e]}),t[1]=i,t[2]=n),n}let s;t[3]===a?s=t[4]:(s=Je.find(e=>e.schema===a),t[3]=a,t[4]=s);let c=s?.connectionLibraries,l;t[5]===c?.libraries?l=t[6]:(l=c?.libraries??[],t[5]=c?.libraries,t[6]=l);let u=c?.preferred??`sqlalchemy`,d;t[7]===Symbol.for(`react.memo_cache_sentinel`)?(d=()=>o(null),t[7]=d):d=t[7];let f;return t[8]!==n||t[9]!==a||t[10]!==l||t[11]!==u?(f=(0,Z.jsx)(r,{schema:a,libraries:l,preferredLibrary:u,displayNames:Ce,libraryLabel:`Preferred connection library`,generateCode:Qe,onSubmit:n,onBack:d}),t[8]=n,t[9]=a,t[10]=l,t[11]=u,t[12]=f):f=t[12],f};function Qe(e,t){return We(e,t)}var $e=e=>{let t=(0,D.c)(19),{className:n,sources:r,onAdd:a}=e,o=(0,E.useId)();if(r.length===0)return null;let s;t[0]===n?s=t[1]:(s=v(`rounded-full bg-[linear-gradient(135deg,var(--blue-2),var(--purple-3))] px-3 py-2 dark:bg-[linear-gradient(135deg,var(--blue-2),var(--purple-2))]`,n),t[0]=n,t[1]=s);let c;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(c=(0,Z.jsx)(i,{className:`h-3.5 w-3.5 text-(--blue-9) dark:text-(--blue-10)`}),t[2]=c):c=t[2];let l;t[3]===o?l=t[4]:(l=(0,Z.jsxs)(`div`,{className:`mr-1 flex items-center gap-1.5`,children:[c,(0,Z.jsx)(`h3`,{id:o,className:`text-sm`,children:`Quick add`})]}),t[3]=o,t[4]=l);let u;if(t[5]!==a||t[6]!==r){let e;t[8]===a?e=t[9]:(e=e=>(0,Z.jsx)(ie,{side:`bottom`,content:(0,Z.jsx)(et,{source:e}),children:(0,Z.jsxs)(`button`,{type:`button`,"aria-label":`Add ${e.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 dark:border-border dark:bg-background/60 dark:hover:bg-accent/80`,onClick:()=>a(e),children:[(0,Z.jsx)(pe,{className:`h-3 w-3 text-muted-foreground`}),e.displayName]})},e.id),t[8]=a,t[9]=e),u=r.map(e),t[5]=a,t[6]=r,t[7]=u}else u=t[7];let d;t[10]===u?d=t[11]:(d=(0,Z.jsx)(b,{delayDuration:200,children:u}),t[10]=u,t[11]=d);let f;t[12]!==l||t[13]!==d?(f=(0,Z.jsxs)(`div`,{className:`flex flex-wrap items-center gap-2`,children:[l,d]}),t[12]=l,t[13]=d,t[14]=f):f=t[14];let p;return t[15]!==s||t[16]!==f||t[17]!==o?(p=(0,Z.jsx)(`section`,{"aria-labelledby":o,className:s,children:f}),t[15]=s,t[16]=f,t[17]=o,t[18]=p):p=t[18],p},et=e=>{let t=(0,D.c)(17),{source:n}=e,r;t[0]===n.displayName?r=t[1]:(r=(0,Z.jsx)(`div`,{className:`font-medium`,children:n.displayName}),t[0]=n.displayName,t[1]=r);let i;t[2]===n.origins?i=t[3]:(i=n.origins.map(nt).join(` and `),t[2]=n.origins,t[3]=i);let a;t[4]===i?a=t[5]:(a=(0,Z.jsxs)(`div`,{className:`text-xs text-muted-foreground`,children:[`Detected from`,` `,i]}),t[4]=i,t[5]=a);let o;t[6]!==r||t[7]!==a?(o=(0,Z.jsxs)(`div`,{children:[r,a]}),t[6]=r,t[7]=a,t[8]=o):o=t[8];let s;t[9]===n.configuration?s=t[10]:(s=n.configuration.map(rt),t[9]=n.configuration,t[10]=s);let c;t[11]===s?c=t[12]:(c=(0,Z.jsx)(`dl`,{className:`grid grid-cols-[auto_1fr] gap-x-3 gap-y-1 text-xs`,children:s}),t[11]=s,t[12]=c);let l;t[13]===Symbol.for(`react.memo_cache_sentinel`)?(l=(0,Z.jsx)(`div`,{className:`text-xs text-muted-foreground`,children:`Click to add a configured cell.`}),t[13]=l):l=t[13];let u;return t[14]!==o||t[15]!==c?(u=(0,Z.jsxs)(`div`,{className:`min-w-64 space-y-2 py-1`,children:[o,c,l]}),t[14]=o,t[15]=c,t[16]=u):u=t[16],u},tt=e=>{let t=(0,D.c)(4),{onSubmit:n,className:r,group:i}=e,a=c(n),o=u(i),s;return t[0]!==a||t[1]!==r||t[2]!==o?(s=(0,Z.jsx)($e,{className:r,sources:o,onAdd:a}),t[0]=a,t[1]=r,t[2]=o,t[3]=s):s=t[3],s};function nt(e){return e.label}function rt(e){return(0,Z.jsxs)(`div`,{className:`contents`,children:[(0,Z.jsx)(`dt`,{className:`text-muted-foreground`,children:e.field}),(0,Z.jsx)(`dd`,{children:(0,Z.jsx)(`code`,{children:e.value.kind===`environment-variable`?`os.environ["${e.value.name}"]`:e.value.value})})]},e.value.kind===`environment-variable`?e.value.name:`${e.field}:${e.value.value}`)}var it=()=>g(`type`,[f({type:h(`Access keys`),access_key_id:p().optional().describe(n.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:p().optional().describe(n.of({label:`Secret Access Key`,inputType:`password`,optionRegex:`(secret.?access.?key|object.?storage.?secret|aws.?secret)`}))}),f({type:h(`Container credentials`),container_credentials_full_uri:p().nonempty().describe(n.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:p().nonempty().describe(n.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(n.of({label:`Credentials`,special:`tabs`})),at=f({type:h(`s3`),bucket:p().nonempty().describe(n.of({label:`Bucket`,placeholder:`my-bucket`,optionRegex:`(bucket|s3.?bucket)`})),region:p().optional().describe(n.of({label:`Region`,placeholder:`us-east-1`,optionRegex:`(region|aws.?region)`})),endpoint_url:p().optional().describe(n.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:m().default(!1).describe(n.of({label:`Allow HTTP`,description:`Required for plain-http (non-TLS) endpoints`})),auth:it()}).describe(n.of({direction:`two-columns`})),ot=f({type:h(`gcs`),bucket:p().nonempty().describe(n.of({label:`Bucket`,placeholder:`my-bucket`,optionRegex:`(bucket|gcs.?bucket|google.?bucket)`})),service_account_key:p().optional().describe(n.of({label:`Service Account Key (JSON)`,special:`secret_textarea`}))}).describe(n.of({direction:`two-columns`})),Q=f({type:h(`azure`),container:p().nonempty().describe(n.of({label:`Container`,placeholder:`my-container`})),account_name:p().nonempty().describe(n.of({label:`Account Name`,placeholder:`storageaccount`,optionRegex:`(azure.?account|account.?name|storage.?account)`})),account_key:p().optional().describe(n.of({label:`Account Key`,inputType:`password`,optionRegex:`(azure.?key|account.?key|storage.?key)`}))}).describe(n.of({direction:`two-columns`})),st=f({type:h(`coreweave`),bucket:p().nonempty().describe(n.of({label:`Bucket`,placeholder:`bucket-name`})),region:p().nonempty().describe(n.of({label:`Region`,placeholder:`US-EAST-04A`})),auth:it()}).describe(n.of({direction:`two-columns`})),ct=f({type:h(`gdrive`),credentials_json:p().optional().describe(n.of({label:`Service Account JSON`,description:`Leave empty to use browser-based authentication`,special:`secret_textarea`}))}).describe(n.of({direction:`two-columns`})),lt=f({type:h(`huggingface`),token:p().optional().describe(n.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(n.of({direction:`two-columns`}));function ut(e){return e!=null&&e!==``}var dt=f({type:h(`github`),org:p().nonempty().describe(n.of({label:`Organization or user`})),repo:p().nonempty().describe(n.of({label:`Repository`})),sha:p().optional().describe(n.of({label:`Branch, tag, or commit`,placeholder:`main`,description:`Leave empty to use the default branch`})),username:p().optional().describe(n.of({label:`Username`,description:`Set this together with access token`,optionRegex:`(github.?username|github.?user)`})),token:p().optional().describe(n.of({label:`Access Token`,description:`Set this together with username. Required for private repos and to avoid GitHub's rate limits.`,inputType:`password`,optionRegex:`(github.?token|gh.?token|github.?pat)`}))}).refine(e=>ut(e.username)===ut(e.token),{message:`Username and access token are required together`,path:[`token`]}).describe(n.of({direction:`two-columns`})),ft=g(`type`,[at,ot,Q,st,ct,lt,dt]),pt={obstore:`obstore`,fsspec:`fsspec`,huggingface_hub:`huggingface_hub`},mt=class{secrets={};get imports(){return Object.keys(this.secrets).length===0?new Set:new Set([`import os`])}print(e,t){if(t===void 0||t===``)return``;let n=`_${e}`;if(T(t)){let e=de(t);return this.secrets[n]=`os.environ.get("${e}")`,n}return`"${w(t)}"`}formatSecrets(){return Object.keys(this.secrets).length===0?``:Object.entries(this.secrets).map(([e,t])=>`${e} = ${t}`).join(`
|
|
136
|
-
`)}};function ht(e,t){if(e?.type===`Container credentials`)return[` container_credentials_full_uri=${t.print(`container_credentials_full_uri`,e.container_credentials_full_uri)},`,` container_authorization_token_file=${t.print(`container_authorization_token_file`,e.container_authorization_token_file)},`];let n=[];return e?.access_key_id&&n.push(` access_key_id=${t.print(`access_key_id`,e.access_key_id)},`),e?.secret_access_key&&n.push(` secret_access_key=${t.print(`secret_access_key`,e.secret_access_key)},`),n}function gt(e,t){let
|
|
137
|
-
`)}\n`:``;return{imports:
|
|
138
|
-
store = S3Store(${
|
|
139
|
-
`)}}function _t(e,t){let
|
|
140
|
-
store = GCSStore(${
|
|
141
|
-
`)};let
|
|
142
|
-
store = GCSStore(${
|
|
143
|
-
service_account_path="${w(
|
|
135
|
+
`)}var Z=y(),Ke=[{name:`PostgreSQL`,schema:_e,color:`#336791`,logo:`postgres`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`MySQL`,schema:ve,color:`#00758F`,logo:`mysql`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`SQLite`,schema:ye,color:`#003B57`,logo:`sqlite`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`DuckDB`,schema:be,color:`#FFD700`,logo:`duckdb`,connectionLibraries:{libraries:[`duckdb`],preferred:`duckdb`}},{name:`MotherDuck`,schema:I,color:`#ff9538`,logo:`motherduck`,connectionLibraries:{libraries:[`duckdb`],preferred:`duckdb`}},{name:`Snowflake`,schema:L,color:`#29B5E8`,logo:`snowflake`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`ClickHouse`,schema:z,color:`#2C2C1D`,logo:`clickhouse`,connectionLibraries:{libraries:[`clickhouse_connect`],preferred:`clickhouse_connect`}},{name:`Timeplus`,schema:B,color:`#B83280`,logo:`timeplus`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`BigQuery`,schema:R,color:`#4285F4`,logo:`bigquery`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`ClickHouse Embedded`,schema:V,color:`#f2b611`,logo:`clickhouse`,connectionLibraries:{libraries:[`chdb`],preferred:`chdb`}},{name:`Trino`,schema:H,color:`#d466b6`,logo:`trino`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}},{name:`DataFusion`,schema:W,color:`#202A37`,logo:`datafusion`,connectionLibraries:{libraries:[`ibis`],preferred:`ibis`}},{name:`PySpark`,schema:G,color:`#1C5162`,logo:`pyspark`,connectionLibraries:{libraries:[`ibis`],preferred:`ibis`}},{name:`Redshift`,schema:K,color:`#522BAE`,logo:`redshift`,connectionLibraries:{libraries:[`redshift`],preferred:`redshift`}},{name:`Databricks`,schema:q,color:`#c41e0c`,logo:`databricks`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`,`ibis`],preferred:`sqlalchemy`}},{name:`Supabase`,schema:xe,color:`#238F5F`,logo:`supabase`,connectionLibraries:{libraries:[`sqlalchemy`,`sqlmodel`],preferred:`sqlalchemy`}}],qe=[{name:`Iceberg`,schema:U,color:`#000000`,logo:`iceberg`,connectionLibraries:{libraries:[`pyiceberg`],preferred:`pyiceberg`}}],Je=[...Ke,...qe],Ye=[...new Set(Je.flatMap(({name:e,logo:t})=>[e,t]))],Xe=e=>{let t=(0,D.c)(12),{onSelect:n}=e,r;t[0]===n?r=t[1]:(r=Ke.map(e=>{let{name:t,schema:r,color:i,logo:a}=e;return(0,Z.jsx)(o,{name:t,color:i,icon:(0,Z.jsx)(s,{name:a,className:`w-8 h-8 text-white brightness-0 invert dark:invert`}),onSelect:()=>n(r)},t)}),t[0]=n,t[1]=r);let i;t[2]===r?i=t[3]:(i=(0,Z.jsx)(c,{children:r}),t[2]=r,t[3]=i);let a;t[4]===Symbol.for(`react.memo_cache_sentinel`)?(a=(0,Z.jsxs)(`h4`,{className:`font-semibold text-muted-foreground text-lg flex items-center gap-4 my-2`,children:[`Data Catalogs`,(0,Z.jsx)(`hr`,{className:`flex-1`})]}),t[4]=a):a=t[4];let l;t[5]===n?l=t[6]:(l=qe.map(e=>{let{name:t,schema:r,color:i,logo:a}=e;return(0,Z.jsx)(o,{name:t,color:i,icon:(0,Z.jsx)(s,{name:a,className:`w-8 h-8 text-white brightness-0 invert dark:invert`}),onSelect:()=>n(r)},t)}),t[5]=n,t[6]=l);let u;t[7]===l?u=t[8]:(u=(0,Z.jsx)(c,{children:l}),t[7]=l,t[8]=u);let d;return t[9]!==i||t[10]!==u?(d=(0,Z.jsxs)(Z.Fragment,{children:[i,a,u]}),t[9]=i,t[10]=u,t[11]=d):d=t[11],d},Ze=e=>{let t=(0,D.c)(13),{onSubmit:n,header:r}=e,[a,o]=(0,E.useState)(null);if(!a){let e;t[0]===Symbol.for(`react.memo_cache_sentinel`)?(e=(0,Z.jsx)(`div`,{children:(0,Z.jsx)(Xe,{onSelect:o})}),t[0]=e):e=t[0];let n;return t[1]===r?n=t[2]:(n=(0,Z.jsxs)(Z.Fragment,{children:[r,e]}),t[1]=r,t[2]=n),n}let s;t[3]===a?s=t[4]:(s=Je.find(e=>e.schema===a),t[3]=a,t[4]=s);let c=s?.connectionLibraries,l;t[5]===c?.libraries?l=t[6]:(l=c?.libraries??[],t[5]=c?.libraries,t[6]=l);let u=c?.preferred??`sqlalchemy`,d;t[7]===Symbol.for(`react.memo_cache_sentinel`)?(d=()=>o(null),t[7]=d):d=t[7];let f;return t[8]!==n||t[9]!==a||t[10]!==l||t[11]!==u?(f=(0,Z.jsx)(i,{schema:a,libraries:l,preferredLibrary:u,displayNames:Ce,libraryLabel:`Preferred connection library`,generateCode:Qe,onSubmit:n,onBack:d}),t[8]=n,t[9]=a,t[10]=l,t[11]=u,t[12]=f):f=t[12],f};function Qe(e,t){return We(e,t)}var $e=e=>{let t=(0,D.c)(19),{className:n,sources:r,onAdd:i}=e,a=(0,E.useId)();if(r.length===0)return null;let o;t[0]===n?o=t[1]:(o=v(`rounded-full bg-[linear-gradient(135deg,var(--blue-2),var(--purple-3))] px-3 py-2 dark:bg-[linear-gradient(135deg,var(--blue-2),var(--purple-2))]`,n),t[0]=n,t[1]=o);let s;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(s=(0,Z.jsx)(l,{className:`h-3.5 w-3.5 text-(--blue-9) dark:text-(--blue-10)`}),t[2]=s):s=t[2];let c;t[3]===a?c=t[4]:(c=(0,Z.jsxs)(`div`,{className:`mr-1 flex items-center gap-1.5`,children:[s,(0,Z.jsx)(`h3`,{id:a,className:`text-sm`,children:`Quick add`})]}),t[3]=a,t[4]=c);let u;if(t[5]!==i||t[6]!==r){let e;t[8]===i?e=t[9]:(e=e=>(0,Z.jsx)(ie,{side:`bottom`,content:(0,Z.jsx)(et,{source:e}),children:(0,Z.jsxs)(`button`,{type:`button`,"aria-label":`Add ${e.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 dark:border-border dark:bg-background/60 dark:hover:bg-accent/80`,onClick:()=>i(e),children:[(0,Z.jsx)(pe,{className:`h-3 w-3 text-muted-foreground`}),e.displayName]})},e.id),t[8]=i,t[9]=e),u=r.map(e),t[5]=i,t[6]=r,t[7]=u}else u=t[7];let d;t[10]===u?d=t[11]:(d=(0,Z.jsx)(b,{delayDuration:200,children:u}),t[10]=u,t[11]=d);let f;t[12]!==c||t[13]!==d?(f=(0,Z.jsxs)(`div`,{className:`flex flex-wrap items-center gap-2`,children:[c,d]}),t[12]=c,t[13]=d,t[14]=f):f=t[14];let p;return t[15]!==o||t[16]!==f||t[17]!==a?(p=(0,Z.jsx)(`section`,{"aria-labelledby":a,className:o,children:f}),t[15]=o,t[16]=f,t[17]=a,t[18]=p):p=t[18],p},et=e=>{let t=(0,D.c)(17),{source:n}=e,r;t[0]===n.displayName?r=t[1]:(r=(0,Z.jsx)(`div`,{className:`font-medium`,children:n.displayName}),t[0]=n.displayName,t[1]=r);let i;t[2]===n.origins?i=t[3]:(i=n.origins.map(nt).join(` and `),t[2]=n.origins,t[3]=i);let a;t[4]===i?a=t[5]:(a=(0,Z.jsxs)(`div`,{className:`text-xs text-muted-foreground`,children:[`Detected from`,` `,i]}),t[4]=i,t[5]=a);let o;t[6]!==r||t[7]!==a?(o=(0,Z.jsxs)(`div`,{children:[r,a]}),t[6]=r,t[7]=a,t[8]=o):o=t[8];let s;t[9]===n.configuration?s=t[10]:(s=n.configuration.map(rt),t[9]=n.configuration,t[10]=s);let c;t[11]===s?c=t[12]:(c=(0,Z.jsx)(`dl`,{className:`grid grid-cols-[auto_1fr] gap-x-3 gap-y-1 text-xs`,children:s}),t[11]=s,t[12]=c);let l;t[13]===Symbol.for(`react.memo_cache_sentinel`)?(l=(0,Z.jsx)(`div`,{className:`text-xs text-muted-foreground`,children:`Click to add a configured cell.`}),t[13]=l):l=t[13];let u;return t[14]!==o||t[15]!==c?(u=(0,Z.jsxs)(`div`,{className:`min-w-64 space-y-2 py-1`,children:[o,c,l]}),t[14]=o,t[15]=c,t[16]=u):u=t[16],u},tt=e=>{let t=(0,D.c)(4),{onSubmit:n,className:i,group:a}=e,o=d(n),s=r(a),c;return t[0]!==o||t[1]!==i||t[2]!==s?(c=(0,Z.jsx)($e,{className:i,sources:s,onAdd:o}),t[0]=o,t[1]=i,t[2]=s,t[3]=c):c=t[3],c};function nt(e){return e.label}function rt(e){return(0,Z.jsxs)(`div`,{className:`contents`,children:[(0,Z.jsx)(`dt`,{className:`text-muted-foreground`,children:e.field}),(0,Z.jsx)(`dd`,{children:(0,Z.jsx)(`code`,{children:e.value.kind===`environment-variable`?`os.environ["${e.value.name}"]`:e.value.value})})]},e.value.kind===`environment-variable`?e.value.name:`${e.field}:${e.value.value}`)}var it=()=>g(`type`,[f({type:h(`Access keys`),access_key_id:p().optional().describe(u.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:p().optional().describe(u.of({label:`Secret Access Key`,inputType:`password`,optionRegex:`(secret.?access.?key|object.?storage.?secret|aws.?secret)`}))}),f({type:h(`Container credentials`),container_credentials_full_uri:p().nonempty().describe(u.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:p().nonempty().describe(u.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(u.of({label:`Credentials`,special:`tabs`})),at=f({type:h(`s3`),bucket:p().nonempty().describe(u.of({label:`Bucket`,placeholder:`my-bucket`,optionRegex:`(bucket|s3.?bucket)`})),region:p().optional().describe(u.of({label:`Region`,placeholder:`us-east-1`,optionRegex:`(region|aws.?region)`})),endpoint_url:p().optional().describe(u.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:m().default(!1).describe(u.of({label:`Allow HTTP`,description:`Required for plain-http (non-TLS) endpoints`})),auth:it()}).describe(u.of({direction:`two-columns`})),ot=f({type:h(`gcs`),bucket:p().nonempty().describe(u.of({label:`Bucket`,placeholder:`my-bucket`,optionRegex:`(bucket|gcs.?bucket|google.?bucket)`})),service_account_key:p().optional().describe(u.of({label:`Service Account Key (JSON)`,special:`secret_textarea`}))}).describe(u.of({direction:`two-columns`})),Q=f({type:h(`azure`),container:p().nonempty().describe(u.of({label:`Container`,placeholder:`my-container`})),account_name:p().nonempty().describe(u.of({label:`Account Name`,placeholder:`storageaccount`,optionRegex:`(azure.?account|account.?name|storage.?account)`})),account_key:p().optional().describe(u.of({label:`Account Key`,inputType:`password`,optionRegex:`(azure.?key|account.?key|storage.?key)`}))}).describe(u.of({direction:`two-columns`})),st=f({type:h(`coreweave`),bucket:p().nonempty().describe(u.of({label:`Bucket`,placeholder:`bucket-name`})),region:p().nonempty().describe(u.of({label:`Region`,placeholder:`US-EAST-04A`})),auth:it()}).describe(u.of({direction:`two-columns`})),ct=f({type:h(`gdrive`),credentials_json:p().optional().describe(u.of({label:`Service Account JSON`,description:`Leave empty to use browser-based authentication`,special:`secret_textarea`}))}).describe(u.of({direction:`two-columns`})),lt=f({type:h(`huggingface`),token:p().optional().describe(u.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(u.of({direction:`two-columns`}));function ut(e){return e!=null&&e!==``}var dt=f({type:h(`github`),org:p().nonempty().describe(u.of({label:`Organization or user`})),repo:p().nonempty().describe(u.of({label:`Repository`})),sha:p().optional().describe(u.of({label:`Branch, tag, or commit`,placeholder:`main`,description:`Leave empty to use the default branch`})),username:p().optional().describe(u.of({label:`Username`,description:`Set this together with access token`,optionRegex:`(github.?username|github.?user)`})),token:p().optional().describe(u.of({label:`Access Token`,description:`Set this together with username. Required for private repos and to avoid GitHub's rate limits.`,inputType:`password`,optionRegex:`(github.?token|gh.?token|github.?pat)`}))}).refine(e=>ut(e.username)===ut(e.token),{message:`Username and access token are required together`,path:[`token`]}).describe(u.of({direction:`two-columns`})),ft=g(`type`,[at,ot,Q,st,ct,lt,dt]),pt={obstore:`obstore`,fsspec:`fsspec`,huggingface_hub:`huggingface_hub`},mt=class{secrets={};get imports(){return Object.keys(this.secrets).length===0?new Set:new Set([`import os`])}print(e,t){if(t===void 0||t===``)return``;let n=`_${e}`;if(T(t)){let e=de(t);return this.secrets[n]=`os.environ.get("${e}")`,n}return`"${w(t)}"`}formatSecrets(){return Object.keys(this.secrets).length===0?``:Object.entries(this.secrets).map(([e,t])=>`${e} = ${t}`).join(`
|
|
136
|
+
`)}};function ht(e,t){if(e?.type===`Container credentials`)return[` container_credentials_full_uri=${t.print(`container_credentials_full_uri`,e.container_credentials_full_uri)},`,` container_authorization_token_file=${t.print(`container_authorization_token_file`,e.container_authorization_token_file)},`];let n=[];return e?.access_key_id&&n.push(` access_key_id=${t.print(`access_key_id`,e.access_key_id)},`),e?.secret_access_key&&n.push(` secret_access_key=${t.print(`secret_access_key`,e.secret_access_key)},`),n}function gt(e,t){let r=t.print(`bucket`,e.bucket),i=new Set([`from obstore.store import S3Store`]),a=[];e.region&&a.push(` region=${t.print(`region`,e.region)},`),a.push(...ht(e.auth,t)),e.endpoint_url&&a.push(` endpoint_url=${t.print(`endpoint_url`,e.endpoint_url)},`),e.allow_http&&a.push(` allow_http=True,`);let o=a.length>0?`\n${a.join(`
|
|
137
|
+
`)}\n`:``;return{imports:i,code:n(`
|
|
138
|
+
store = S3Store(${r},${o})
|
|
139
|
+
`)}}function _t(e,t){let r=t.print(`bucket`,e.bucket),i=new Set([`from obstore.store import GCSStore`]);if(!e.service_account_key)return{imports:i,code:n(`
|
|
140
|
+
store = GCSStore(${r})
|
|
141
|
+
`)};let a=S(e.service_account_key,e=>t.print(`service_account_key`,e));return a.kind===`path`?{imports:i,code:n(`
|
|
142
|
+
store = GCSStore(${r},
|
|
143
|
+
service_account_path="${w(a.path)}",
|
|
144
144
|
)
|
|
145
|
-
`)}:(
|
|
146
|
-
_credentials = ${
|
|
147
|
-
store = GCSStore(${
|
|
145
|
+
`)}:(i.add(`import json`),{imports:i,code:n(`
|
|
146
|
+
_credentials = ${a.expr}
|
|
147
|
+
store = GCSStore(${r},
|
|
148
148
|
service_account_key=_credentials,
|
|
149
149
|
)
|
|
150
150
|
`)})}function vt(e,t){let n=t.print(`container`,e.container),r=t.print(`account_name`,e.account_name),i=new Set([`from obstore.store import AzureStore`]),a=[`account_name=${r},`];return e.account_key&&a.push(`account_key=${t.print(`account_key`,e.account_key)},`),{imports:i,code:`store = AzureStore(${n},\n${a.map(e=>` ${e}`).join(`
|
|
151
|
-
`)}\n)`}}function yt(e,t){let
|
|
152
|
-
`)}\n`;return{imports:
|
|
153
|
-
store = S3Store(${
|
|
154
|
-
`)}}function bt(e,t){let{secrets:
|
|
151
|
+
`)}\n)`}}function yt(e,t){let r=t.print(`bucket`,e.bucket),i=new Set([`from obstore.store import S3Store`]),a=[` region=${t.print(`region`,e.region)},`];a.push(...ht(e.auth,t)),a.push(` endpoint="https://${e.bucket}.cwobject.com",`,` virtual_hosted_style_request=True,`);let o=`\n${a.join(`
|
|
152
|
+
`)}\n`;return{imports:i,code:n(`
|
|
153
|
+
store = S3Store(${r},${o})
|
|
154
|
+
`)}}function bt(e,t){let{secrets:r,isEmbedded:i=!1}=t,a=new Set([`from gdrive_fsspec import GoogleDriveFileSystem`]);if(e.credentials_json){let t=S(e.credentials_json,e=>r.print(`credentials_json`,e));return t.kind===`path`?{imports:a,code:n(`
|
|
155
155
|
fs = GoogleDriveFileSystem(creds="${w(t.path)}", token="service_account", use_listings_cache=False, skip_instance_cache=True)
|
|
156
|
-
`)}:(
|
|
156
|
+
`)}:(a.add(`import json`),{imports:a,code:n(`
|
|
157
157
|
_creds = ${t.expr}
|
|
158
158
|
fs = GoogleDriveFileSystem(creds=_creds, token="service_account", use_listings_cache=False, skip_instance_cache=True)
|
|
159
|
-
`)})}return{imports:
|
|
159
|
+
`)})}return{imports:a,code:n(i?`
|
|
160
160
|
fs = GoogleDriveFileSystem(use_listings_cache=False, skip_instance_cache=True, auth_kwargs={"use_local_webserver": False})
|
|
161
161
|
mo.output.clear_console()
|
|
162
162
|
`:`
|
|
163
163
|
fs = GoogleDriveFileSystem(use_listings_cache=False, skip_instance_cache=True)
|
|
164
164
|
`)}}function xt(e,t){let n=new Set([`from huggingface_hub import HfApi`]);return e.token?{imports:n,code:`hf = HfApi(token=${t.print(`token`,e.token)})`}:{imports:n,code:`hf = HfApi()`}}function St(e,t){let n=new Set([`from fsspec.implementations.github import GithubFileSystem`]),r=[`org=${t.print(`org`,e.org)},`,`repo=${t.print(`repo`,e.repo)},`];return e.sha&&r.push(`sha=${t.print(`sha`,e.sha)},`),e.username&&e.token&&r.push(`username=${t.print(`username`,e.username)},`,`token=${t.print(`token`,e.token)},`),{imports:n,code:`fs = GithubFileSystem(\n${r.map(e=>` ${e}`).join(`
|
|
165
165
|
`)}\n)`}}function Ct(e,t){ft.parse(e);let n=new mt,r;switch(e.type){case`s3`:r=gt(e,n);break;case`gcs`:r=_t(e,n);break;case`azure`:r=vt(e,n);break;case`coreweave`:r=yt(e,n);break;case`gdrive`:r=bt(e,{secrets:n,isEmbedded:t.isEmbedded});break;case`huggingface`:r=xt(e,n);break;case`github`:r=St(e,n);break;default:x(e)}let i=[...new Set([...n.imports,...r.imports])].toSorted();i.push(``);let a=n.formatSecrets();return a&&i.push(a),i.push(r.code.trim()),i.join(`
|
|
166
|
-
`)}var wt=`#232F3E`,$=[{name:`Amazon S3`,schema:at,protocol:`s3`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`Google Cloud Storage`,schema:ot,protocol:`gcs`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`Azure Blob Storage`,schema:Q,protocol:`azure`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`CoreWeave`,schema:st,protocol:`coreweave`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`Google Drive`,schema:ct,protocol:`gdrive`,storageLibraries:{libraries:[`fsspec`],preferred:`fsspec`}},{name:`Hugging Face Hub`,schema:lt,protocol:`hf`,storageLibraries:{libraries:[`huggingface_hub`],preferred:`huggingface_hub`}},{name:`GitHub`,schema:dt,protocol:`github`,storageLibraries:{libraries:[`fsspec`],preferred:`fsspec`}}],Tt=[...new Set($.flatMap(({name:e,protocol:t})=>[e,t]))],Et=e=>{let t=(0,D.c)(4),{onSelect:n}=e,r;t[0]===n?r=t[1]:(r=$.map(e=>{let{name:t,schema:r,protocol:i}=e;return(0,Z.jsx)(
|
|
166
|
+
`)}var wt=`#232F3E`,$=[{name:`Amazon S3`,schema:at,protocol:`s3`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`Google Cloud Storage`,schema:ot,protocol:`gcs`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`Azure Blob Storage`,schema:Q,protocol:`azure`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`CoreWeave`,schema:st,protocol:`coreweave`,storageLibraries:{libraries:[`obstore`],preferred:`obstore`}},{name:`Google Drive`,schema:ct,protocol:`gdrive`,storageLibraries:{libraries:[`fsspec`],preferred:`fsspec`}},{name:`Hugging Face Hub`,schema:lt,protocol:`hf`,storageLibraries:{libraries:[`huggingface_hub`],preferred:`huggingface_hub`}},{name:`GitHub`,schema:dt,protocol:`github`,storageLibraries:{libraries:[`fsspec`],preferred:`fsspec`}}],Tt=[...new Set($.flatMap(({name:e,protocol:t})=>[e,t]))],Et=e=>{let t=(0,D.c)(4),{onSelect:n}=e,r;t[0]===n?r=t[1]:(r=$.map(e=>{let{name:t,schema:r,protocol:i}=e;return(0,Z.jsx)(o,{name:t,color:wt,icon:(0,Z.jsx)(ae,{protocol:i,forceDark:!0,className:`w-7.5 h-7.5 text-white`}),onSelect:()=>n(r)},t)}),t[0]=n,t[1]=r);let i;return t[2]===r?i=t[3]:(i=(0,Z.jsx)(c,{children:r}),t[2]=r,t[3]=i),i},Dt=e=>{let t=(0,D.c)(16),{onSubmit:n,header:r}=e,{isEmbedded:a}=me(),[o,s]=(0,E.useState)(null);if(!o){let e;t[0]===Symbol.for(`react.memo_cache_sentinel`)?(e=(0,Z.jsx)(`div`,{children:(0,Z.jsx)(Et,{onSelect:s})}),t[0]=e):e=t[0];let n;return t[1]===r?n=t[2]:(n=(0,Z.jsxs)(Z.Fragment,{children:[r,e]}),t[1]=r,t[2]=n),n}let c;t[3]===o?c=t[4]:(c=$.find(e=>e.schema===o),t[3]=o,t[4]=c);let l=c?.storageLibraries,u;t[5]===l?.libraries?u=t[6]:(u=l?.libraries??[],t[5]=l?.libraries,t[6]=u);let d=l?.preferred??`obstore`,f;t[7]===a?f=t[8]:(f=(e,t)=>Ct(e,{library:t,isEmbedded:a}),t[7]=a,t[8]=f);let p;t[9]===Symbol.for(`react.memo_cache_sentinel`)?(p=()=>s(null),t[9]=p):p=t[9];let m;return t[10]!==n||t[11]!==o||t[12]!==u||t[13]!==d||t[14]!==f?(m=(0,Z.jsx)(i,{schema:o,libraries:u,preferredLibrary:d,displayNames:pt,libraryLabel:`Preferred storage library`,generateCode:f,onSubmit:n,onBack:p}),t[10]=n,t[11]=o,t[12]=u,t[13]=d,t[14]=f,t[15]=m):m=t[15],m},Ot=e=>{let t=(0,D.c)(9),{children:n,defaultTab:r}=e,i=r===void 0?`databases`:r,[a,o]=(0,E.useState)(!1),s;t[0]===n?s=t[1]:(s=(0,Z.jsx)(se,{asChild:!0,children:n}),t[0]=n,t[1]=s);let c;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(c=()=>o(!1),t[2]=c):c=t[2];let l;t[3]===i?l=t[4]:(l=(0,Z.jsx)(kt,{defaultTab:i,onClose:c}),t[3]=i,t[4]=l);let u;return t[5]!==a||t[6]!==s||t[7]!==l?(u=(0,Z.jsxs)(fe,{open:a,onOpenChange:o,children:[s,l]}),t[5]=a,t[6]=s,t[7]=l,t[8]=u):u=t[8],u},kt=e=>{let t=(0,D.c)(24),{defaultTab:n,onClose:r}=e,[i,a]=(0,E.useState)(n===void 0?`databases`:n),o;t[0]===Symbol.for(`react.memo_cache_sentinel`)?(o=(0,Z.jsxs)(ne,{className:`w-full mb-4`,children:[(0,Z.jsx)(ee,{value:`databases`,className:`flex-1`,children:`Databases & Catalogs`}),(0,Z.jsx)(ee,{value:`storage`,className:`flex-1`,children:`Remote Storages`})]}),t[0]=o):o=t[0];let s=o,c;t[1]===i?c=t[2]:(c=i===`databases`?(0,Z.jsxs)(Z.Fragment,{children:[`Don't see your database or connection method? A`,` `,(0,Z.jsx)(C,{href:`https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database`,children:`code snippet`}),` `,`is all you need.`]}):(0,Z.jsxs)(Z.Fragment,{children:[`Don't see your storage or connection method? A`,` `,(0,Z.jsx)(C,{href:`https://docs.marimo.io/guides/working_with_data/remote_storage/#creating-a-storage-connection`,children:`code snippet`}),` `,`is all you need.`]}),t[1]=i,t[2]=c);let l=c,u;t[3]===Symbol.for(`react.memo_cache_sentinel`)?(u=(0,Z.jsx)(ue,{children:`Add Connection`}),t[3]=u):u=t[3];let d;t[4]===Symbol.for(`react.memo_cache_sentinel`)?(d=(0,Z.jsx)(C,{href:`https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database`,children:`database, data catalog`}),t[4]=d):d=t[4];let f;t[5]===Symbol.for(`react.memo_cache_sentinel`)?(f=(0,Z.jsx)(C,{href:`https://docs.marimo.io/guides/working_with_data/remote_storage/`,children:`remote storage`}),t[5]=f):f=t[5];let p;t[6]===l?p=t[7]:(p=(0,Z.jsxs)(ce,{children:[u,(0,Z.jsxs)(oe,{children:[`Connect to a`,` `,d,` `,`or`,` `,f,` `,`directly from your notebook.`,(0,Z.jsx)(`span`,{className:`block`,children:l})]})]}),t[6]=l,t[7]=p);let m=i===`storage`?`storage`:`database`,h;t[8]!==r||t[9]!==m?(h=(0,Z.jsx)(tt,{onSubmit:r,className:`-mt-2`,group:m}),t[8]=r,t[9]=m,t[10]=h):h=t[10];let g;t[11]===Symbol.for(`react.memo_cache_sentinel`)?(g=e=>a(e),t[11]=g):g=t[11];let _;t[12]===r?_=t[13]:(_=(0,Z.jsx)(te,{value:`databases`,className:`mt-0 focus-visible:ring-0 focus-visible:ring-offset-0`,children:(0,Z.jsx)(Ze,{onSubmit:r,header:s})}),t[12]=r,t[13]=_);let v;t[14]===r?v=t[15]:(v=(0,Z.jsx)(te,{value:`storage`,className:`mt-0 focus-visible:ring-0 focus-visible:ring-offset-0`,children:(0,Z.jsx)(Dt,{onSubmit:r,header:s})}),t[14]=r,t[15]=v);let y;t[16]!==i||t[17]!==_||t[18]!==v?(y=(0,Z.jsxs)(re,{value:i,onValueChange:g,children:[_,v]}),t[16]=i,t[17]=_,t[18]=v,t[19]=y):y=t[19];let b;return t[20]!==y||t[21]!==p||t[22]!==h?(b=(0,Z.jsxs)(le,{className:`max-h-[75vh] overflow-y-auto`,children:[p,h,y]}),t[20]=y,t[21]=p,t[22]=h,t[23]=b):b=t[23],b};export{Ye as i,kt as n,Tt as r,Ot as t};
|