@marimo-team/frontend 0.23.16-dev9 → 0.23.17-dev0

package/dist/assets/add-connection-dialog-GskQAyZw.js

@@ -0,0 +1,178 @@
+var ht=Object.defineProperty;var mt=(t,e,s)=>e in t?ht(t,e,{enumerable:!0,configurable:!0,writable:!0,value:s}):t[e]=s;var G=(t,e,s)=>mt(t,typeof e!="symbol"?e+"":e,s);import{s as ie}from"./chunk-LvLJmgfZ.js";import{u as ut}from"./useEvent-D91BmmQi.js";import{t as bt}from"./react-Bj1aDYRI.js";import{Ct as ce,E as ft,Lt as gt,Pr as yt,ur as y}from"./cells-CJLUCcQg.js";import{D as K,P as _,R as b,T as L,_ as le,i as _t,k as x}from"./zod-H_cgTO0M.js";import{t as U}from"./compiler-runtime-B3qBwwSJ.js";import{_ as xt}from"./useEventListener-C0195JsF.js";import{c as St}from"./utils-BWBBpCkT.js";import{t as X}from"./cn-DYvqRARy.js";import{t as kt}from"./jsx-runtime-Blw4afVn.js";import{a as wt,c as de,i as jt,o as pe,r as vt,s as $t,t as Ct}from"./popover-Bcz1Zikb.js";import{c as At,i as Nt,n as Rt,r as Tt,s as Lt,t as Dt}from"./select-BP75_5jJ.js";import{i as It,t as he}from"./tooltip-4bZlU7jl.js";import{t as me}from"./button-DDGiD1NW.js";import{r as W}from"./strings--BknHX9C.js";import{r as Pt,t as qt}from"./x-C-Eq0ag8.js";import{t as Et}from"./circle-plus-BluydVNx.js";import{t as Ft}from"./components-K_Mi0vdC.js";import{n as Z}from"./DeferredRequestRegistry-C8EV1PoE.js";import{t as Ot}from"./plus-AgXtozBz.js";import{t as Bt}from"./sparkles-DUCpRFlS.js";import{a as ue,c as be,i as Ut,n as fe,o as ge,p as Kt,r as ye,s as _e,u as Mt}from"./form-W6ZZMHnc.js";import{a as Wt,c as Ht,l as Gt,n as Vt,r as zt,t as Qt}from"./dialog-BCVepWLJ.js";import{n as xe}from"./ImperativeModal-fY-ZfT0b.js";import{t as V}from"./links-wcQA1Np1.js";import{t as Jt}from"./useIframeCapabilities-j6fR_dBA.js";import{r as Yt}from"./html-to-image-CN9yTa5s.js";import{o as Xt}from"./focus-B6HBbAwF.js";import{n as Se}from"./useAsyncData-Bg5iAViP.js";import{a as Zt,c as ke,i as z,o as Q,s as es,t as ts}from"./command-Dqxie4gq.js";import{c as d,o as ss,t as rs}from"./form-BYlqbYKI.js";import{t as ns}from"./request-registry-Dpk9hQrX.js";import{a as ee,c as te,d as os,f as as,i as we,l as je,n as is,o as cs,r as se,s as E,t as ve,u as $e}from"./write-secret-modal-Dtm66yVf.js";var D=ie(bt(),1),o=ie(kt(),1);function ls(t,e){if(!e)return{recommended:[],other:[...t]};let s;try{s=new RegExp(e,"i")}catch{return{recommended:[],other:[...t]}}let n=[],r=[];for(let a of t)s.test(a)?n.push(a):r.push(a);return{recommended:n,other:r}}const Ce=({value:t,onChange:e,placeholder:s,secretsOnly:n=!1,recommendedKeys:r,otherKeys:a,onCreateSecret:c,className:i,searchPlaceholder:l,formatCustomValueLabel:p=g=>`Use "${g}"`,createSecretLabel:h="Create a new secret",allowCustomValue:f,allowCreateSecret:u})=>{let[g,m]=(0,D.useState)(!1),[S,w]=(0,D.useState)(""),k=S.trim(),v=!n&&k.length>0&&k!==t&&((f==null?void 0:f(k))??!0),A=(u==null?void 0:u(k))??!0,C=t?E(t)?cs(t):t:null,N=j=>{e(te(j)),m(!1),w("")},P=j=>{e(j),m(!1),w("")},q=()=>{e(""),w(""),m(!1)};return(0,o.jsxs)("div",{className:X("flex w-full min-w-48 mb-1 items-center gap-1",i),children:[(0,o.jsxs)(Ct,{modal:!0,open:g,onOpenChange:j=>{m(j),j||w("")},children:[(0,o.jsx)(jt,{asChild:!0,children:(0,o.jsxs)("button",{type:"button",className:X("flex h-6 min-w-0 flex-1 shadow-xs-solid items-center justify-between","rounded-sm border border-input bg-background px-1.5 text-sm font-code ring-offset-background placeholder:text-muted-foreground","hover:shadow-sm-solid focus:outline-hidden focus:ring-1 focus:ring-ring focus:border-primary focus:shadow-md-solid",E(t)&&"bg-accent"),"aria-expanded":g,children:[(0,o.jsx)("span",{className:"truncate flex-1 min-w-0 text-left flex items-center gap-1.5",children:C?(0,o.jsxs)(o.Fragment,{children:[E(t)&&(0,o.jsx)(Z,{className:"h-3 w-3 shrink-0 opacity-70"}),(0,o.jsx)("span",{className:"truncate",children:C})]}):(0,o.jsx)("span",{className:"text-muted-foreground",children:s})}),(0,o.jsx)(Pt,{className:"ml-1 h-4 w-4 opacity-50 shrink-0"})]})}),(0,o.jsx)(vt,{className:"w-(--radix-popover-trigger-width) p-0",align:"start",children:(0,o.jsxs)(ts,{children:[(0,o.jsx)(Zt,{placeholder:l??(n?"Search secrets...":"Type a value or search secrets..."),rootClassName:"px-1 h-8",autoFocus:!0,value:S,onValueChange:w}),(0,o.jsxs)(es,{className:"max-h-60 overscroll-contain",children:[v&&(0,o.jsx)(z,{children:(0,o.jsx)(Q,{value:`use custom value ${k}`,onSelect:()=>P(k),children:p(k)})}),A&&(0,o.jsx)(z,{className:"mt-0",children:(0,o.jsxs)(Q,{value:`create new secret ${k}`,onSelect:()=>{let j=k||void 0;m(!1),w(""),c(j)},children:[(0,o.jsx)(Et,{className:"mr-2 h-3.5 w-3.5"}),h]})}),r.length>0&&(0,o.jsxs)(o.Fragment,{children:[(0,o.jsx)(ke,{className:"mt-0"}),(0,o.jsx)(z,{heading:"Recommended",children:r.map(j=>(0,o.jsxs)(Q,{value:j,onSelect:()=>N(j),children:[(0,o.jsx)(Z,{className:"mr-2 h-3 w-3 opacity-70"}),j]},j))})]}),a.length>0&&(0,o.jsxs)(o.Fragment,{children:[(0,o.jsx)(ke,{className:"mt-0"}),(0,o.jsx)(z,{heading:r.length>0?"Other":void 0,children:a.map(j=>(0,o.jsxs)(Q,{value:j,onSelect:()=>N(j),children:[(0,o.jsx)(Z,{className:"mr-2 h-3 w-3 opacity-70"}),j]},j))})]})]})]})})]}),t&&(0,o.jsx)("button",{type:"button","aria-label":"Clear",className:X("flex h-6 w-6 shrink-0 items-center justify-center rounded-sm","text-muted-foreground hover:bg-muted hover:text-foreground","focus:outline-hidden focus:ring-1 focus:ring-ring"),onClick:q,children:(0,o.jsx)(qt,{className:"h-3.5 w-3.5"})})]})};var ds=U(),Ae=(0,D.createContext)({providerNames:[],secretKeys:[],loading:!1,error:void 0,refreshSecrets:xt.NOOP});const Ne=()=>(0,D.use)(Ae),ps=t=>{let e=(0,ds.c)(14),{children:s}=t,n;e[0]===Symbol.for("react.memo_cache_sentinel")?(n=[],e[0]=n):n=e[0];let{data:r,isPending:a,error:c,refetch:i}=Se(gs,n);r==null||r.secretKeys,r==null||r.providerNames;let l;e[1]===(r==null?void 0:r.secretKeys)?l=e[2]:(l=(r==null?void 0:r.secretKeys)||[],e[1]=r==null?void 0:r.secretKeys,e[2]=l);let p;e[3]===(r==null?void 0:r.providerNames)?p=e[4]:(p=(r==null?void 0:r.providerNames)||[],e[3]=r==null?void 0:r.providerNames,e[4]=p);let h;e[5]!==c||e[6]!==a||e[7]!==i||e[8]!==l||e[9]!==p?(h={secretKeys:l,providerNames:p,loading:a,error:c,refreshSecrets:i},e[5]=c,e[6]=a,e[7]=i,e[8]=l,e[9]=p,e[10]=h):h=e[10];let f=h,u;return e[11]!==s||e[12]!==f?(u=(0,o.jsx)(Ae,{value:f,children:s}),e[11]=s,e[12]=f,e[13]=u):u=e[13],u},hs={isMatch:t=>{if(t instanceof le){let{optionRegex:e}=d.parse(t.description||"");return!!e}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:n,providerNames:r,refreshSecrets:a}=Ne(),{openModal:c,closeModal:i}=xe(),{label:l,description:p,placeholder:h,optionRegex:f="",inputType:u}=d.parse(t.description||""),g=u==="password",{recommended:m,other:S}=ls(n,f);return(0,o.jsx)(ue,{control:e.control,name:s,render:({field:w})=>(0,o.jsxs)(ge,{children:[(0,o.jsx)(_e,{children:l}),(0,o.jsx)(ye,{children:p}),(0,o.jsx)(fe,{children:(0,o.jsx)(Ce,{value:w.value?String(w.value):"",onChange:w.onChange,placeholder:h,secretsOnly:g,recommendedKeys:m,otherKeys:S,onCreateSecret:k=>{c((0,o.jsx)(ve,{providerNames:r,initialValue:k,onSuccess:v=>{a(),w.onChange(te(v)),i()},onClose:i}))}})}),(0,o.jsx)(be,{})]})})}},ms={isMatch:t=>{if(t instanceof le){let{special:e}=d.parse(t.description||"");return e==="secret_textarea"}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:n,providerNames:r,refreshSecrets:a}=Ne(),{openModal:c,closeModal:i}=xe(),{label:l,description:p,placeholder:h}=d.parse(t.description||"");return(0,o.jsx)(ue,{control:e.control,name:s,render:({field:f})=>{let u=f.value?String(f.value):"",g=$e(u)?as(u):u;return(0,o.jsxs)(ge,{children:[(0,o.jsxs)(_e,{className:"flex items-center gap-1",children:[l,(0,o.jsx)(he,{content:"Enter a path to a file on disk, or create a secret to paste JSON directly.",delayDuration:300,children:(0,o.jsx)("button",{type:"button",className:"inline-flex","aria-label":"Credential input help",children:(0,o.jsx)(yt,{className:"h-3.5 w-3.5"})})})]}),(0,o.jsx)(ye,{children:p}),(0,o.jsx)(fe,{children:(0,o.jsx)(Ce,{value:g,onChange:m=>{f.onChange(!m||E(m)||$e(m)?m:os(m))},placeholder:h,searchPlaceholder:"Type a file path or paste JSON",formatCustomValueLabel:m=>`Use "${m}" as a file path`,createSecretLabel:"Paste JSON credentials",allowCustomValue:m=>!we(m),allowCreateSecret:m=>!m||we(m),recommendedKeys:[],otherKeys:n,onCreateSecret:m=>{c((0,o.jsx)(ve,{providerNames:r,initialValue:m,multiline:!0,onSuccess:S=>{a(),f.onChange(te(S)),i()},onClose:i}))}})}),(0,o.jsx)(be,{})]})}})}};function us(t){return t.provider!=="env"}function bs(t){return t.name}function fs(t){return t.keys}async function gs(){let t=await ns.request({}),e=is(t.secrets).filter(us).map(bs);return{secretKeys:t.secrets.flatMap(fs).toSorted(),providerNames:e}}var H=U(),ys=[hs,ms];const re=t=>{let e=(0,H.c)(2),{children:s}=t,n;return e[0]===s?n=e[1]:(n=(0,o.jsx)("div",{className:"grid grid-cols-2 md:grid-cols-3 gap-4",children:s}),e[0]=s,e[1]=n),n},ne=t=>{let e=(0,H.c)(9),{name:s,color:n,icon:r,onSelect:a}=t,c;e[0]===n?c=e[1]:(c={backgroundColor:n},e[0]=n,e[1]=c);let i;e[2]===s?i=e[3]:(i=(0,o.jsx)("span",{className:"text-white font-medium text-lg",children:s}),e[2]=s,e[3]=i);let l;return e[4]!==r||e[5]!==a||e[6]!==c||e[7]!==i?(l=(0,o.jsxs)("button",{type:"button",className:"py-3 flex flex-col items-center justify-center gap-1 transition-all hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid",style:c,onClick:a,children:[r,i]}),e[4]=r,e[5]=a,e[6]=c,e[7]=i,e[8]=l):l=e[8],l},_s=t=>{let e=(0,H.c)(25),{onBack:s,isValid:n,libraries:r,preferredLibrary:a,onLibraryChange:c,displayNames:i,libraryLabel:l}=t,p=l===void 0?"Preferred library":l,h;e[0]===s?h=e[1]:(h=(0,o.jsx)(me,{type:"button",variant:"outline",onClick:s,children:"Back"}),e[0]=s,e[1]=h);let f=!n,u;e[2]===f?u=e[3]:(u=(0,o.jsx)(me,{type:"submit",disabled:f,children:"Add"}),e[2]=f,e[3]=u);let g;e[4]!==h||e[5]!==u?(g=(0,o.jsxs)("div",{className:"flex gap-2",children:[h,u]}),e[4]=h,e[5]=u,e[6]=g):g=e[6];let m;e[7]===Symbol.for("react.memo_cache_sentinel")?(m=(0,o.jsx)(Lt,{children:(0,o.jsx)(At,{placeholder:"Select a library"})}),e[7]=m):m=e[7];let S;e[8]===p?S=e[9]:(S=(0,o.jsxs)("div",{className:"flex flex-col gap-1 items-end",children:[m,(0,o.jsx)("span",{className:"text-xs text-muted-foreground",children:p})]}),e[8]=p,e[9]=S);let w;if(e[10]!==i||e[11]!==r){let C;e[13]===i?C=e[14]:(C=N=>(0,o.jsx)(Nt,{value:N,children:i[N]??N},N),e[13]=i,e[14]=C),w=r.map(C),e[10]=i,e[11]=r,e[12]=w}else w=e[12];let k;e[15]===w?k=e[16]:(k=(0,o.jsx)(Rt,{children:(0,o.jsx)(Tt,{children:w})}),e[15]=w,e[16]=k);let v;e[17]!==c||e[18]!==a||e[19]!==S||e[20]!==k?(v=(0,o.jsx)("div",{children:(0,o.jsxs)(Dt,{value:a,onValueChange:c,children:[S,k]})}),e[17]=c,e[18]=a,e[19]=S,e[20]=k,e[21]=v):v=e[21];let A;return e[22]!==v||e[23]!==g?(A=(0,o.jsxs)("div",{className:"flex gap-2 justify-between",children:[g,v]}),e[22]=v,e[23]=g,e[24]=A):A=e[24],A};function Re(){let t=(0,H.c)(4),{createNewCell:e}=ft(),s=Xt(),n=ut(St),r;return t[0]!==n||t[1]!==e||t[2]!==s?(r=a=>{/\bmo\./.test(a)&&Yt({autoInstantiate:n,createNewCell:e,fromCellId:s}),e({code:a,before:!1,cellId:s??"__end__",skipIfCodeExists:!0})},t[0]=n,t[1]=e,t[2]=s,t[3]=r):r=t[3],r}const Te=t=>{let e=(0,H.c)(30),{schema:s,libraries:n,preferredLibrary:r,displayNames:a,libraryLabel:c,generateCode:i,onSubmit:l,onBack:p}=t,h;e[0]===s?h=e[1]:(h=ss(s),e[0]=s,e[1]=h);let f=h,u;e[2]===s?u=e[3]:(u=Mt(s),e[2]=s,e[3]=u);let g;e[4]!==f||e[5]!==u?(g={defaultValues:f,resolver:u,reValidateMode:"onChange"},e[4]=f,e[5]=u,e[6]=g):g=e[6];let m=Kt(g),[S,w]=D.useState(r),k=Re(),v;e[7]!==i||e[8]!==k||e[9]!==l||e[10]!==S?(v=pt=>{k(i(pt,S)),l()},e[7]=i,e[8]=k,e[9]=l,e[10]=S,e[11]=v):v=e[11];let A=v,C;e[12]!==m||e[13]!==A?(C=m.handleSubmit(A),e[12]=m,e[13]=A,e[14]=C):C=e[14];let N;e[15]===Symbol.for("react.memo_cache_sentinel")?(N=(0,o.jsx)(Ut,{}),e[15]=N):N=e[15];let P;e[16]!==m||e[17]!==s?(P=(0,o.jsx)(ps,{children:(0,o.jsx)(rs,{schema:s,form:m,renderers:ys,children:N})}),e[16]=m,e[17]=s,e[18]=P):P=e[18];let q;e[19]!==a||e[20]!==m.formState.isValid||e[21]!==n||e[22]!==c||e[23]!==p||e[24]!==S?(q=(0,o.jsx)(_s,{onBack:p,isValid:m.formState.isValid,libraries:n,preferredLibrary:S,onLibraryChange:w,displayNames:a,libraryLabel:c}),e[19]=a,e[20]=m.formState.isValid,e[21]=n,e[22]=c,e[23]=p,e[24]=S,e[25]=q):q=e[25];let j;return e[26]!==C||e[27]!==P||e[28]!==q?(j=(0,o.jsxs)("form",{onSubmit:C,className:"space-y-4",children:[P,q]}),e[26]=C,e[27]=P,e[28]=q,e[29]=j):j=e[29],j};function F(){return b().optional().describe(d.of({label:"Password",inputType:"password",optionRegex:"(password|passwd|pgpassword|db.?pass(word)?)"}))}function J(t,e){let s=b();return s=e?s.nonempty():s.optional(),s=s.describe(d.of({label:t||"Token",inputType:"password",optionRegex:"(token|api.?key|access.?token|auth.?token|(^|_)pat(_|$))"})),s}function Y(){return b().optional().describe(d.of({label:"Warehouse Name",optionRegex:"(warehouse|snowflake.?warehouse)"}))}function oe(t,e){let s=b();return s=e?s.nonempty():s.optional(),s.describe(d.of({label:t||"URI",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"}))}function I(t){return b().nonempty().describe(d.of({label:t||"Host",optionRegex:"^(?!.*(kubernetes|gpg)).*(host(name)?|pghost|db.?host|database.?host|mysql.?host|postgres.?host|server.?host)"}))}function R(){return b().describe(d.of({label:"Database",optionRegex:"(database|db.?name|pgdatabase|^DB$|^DATABASE$)"}))}function xs(){return b().describe(d.of({label:"Schema",optionRegex:"(^SCHEMA$|db.?schema|postgres.?schema|pg.?schema|_SCHEMA$)"}))}function T(){return b().nonempty().describe(d.of({label:"Username",optionRegex:"(username|pguser|db.?user|^USER$|_USER$)"}))}function O(t){let e=_t().describe(d.of({label:"Port",inputType:"number",placeholder:t==null?void 0:t.toString()})).transform(Number).refine(s=>s>=0&&s<=65535,{message:"Port must be between 0 and 65535"});return t===void 0?e:e.default(t)}function Le(){return L().default(!1).describe(d.of({label:"Read Only"}))}const De=_({type:x("postgres"),host:I(),port:O(5432).optional(),database:R(),username:T(),password:F(),ssl:L().default(!1).describe(d.of({label:"Use SSL"}))}).describe(d.of({direction:"two-columns"})),Ie=_({type:x("mysql"),host:I(),port:O(3306),database:R(),username:T(),password:F(),ssl:L().default(!1).describe(d.of({label:"Use SSL"}))}).describe(d.of({direction:"two-columns"})),Pe=_({type:x("sqlite"),database:R().describe(d.of({label:"Database Path"}))}).describe(d.of({direction:"two-columns"})),qe=_({type:x("duckdb"),database:R().describe(d.of({label:"Database Path"})),read_only:Le()}).describe(d.of({direction:"two-columns"})),Ee=_({type:x("motherduck"),database:R().default("my_db").describe(d.of({label:"Database Name"})),token:J()}).describe(d.of({direction:"two-columns"})),Fe=_({type:x("snowflake"),account:b().nonempty().describe(d.of({label:"Account",optionRegex:"(snowflake.?account|^SNOWFLAKE_ACCOUNT$|account.?id)"})),warehouse:b().optional().describe(d.of({label:"Warehouse",optionRegex:"(snowflake.?warehouse|warehouse)"})),database:R(),schema:b().optional().describe(d.of({label:"Schema",optionRegex:"(snowflake.?schema|^SCHEMA$|_SCHEMA$)"})),role:b().optional().describe(d.of({label:"Role"})),authType:K("type",[_({type:x("Password"),username:T(),password:F(),enable_mfa:L().default(!1).describe(d.of({label:"Enable MFA (Duo Push)"}))}),_({type:x("SSO (Browser)"),username:T()}),_({type:x("Key Pair"),username:T(),private_key_path:b().nonempty().describe(d.of({label:"Private Key Path",placeholder:"/path/to/rsa_key.p8"})),private_key_passphrase:b().optional().describe(d.of({label:"Private Key Passphrase",inputType:"password",optionRegex:"(passphrase|private.?key.?passphrase|key.?passphrase)"}))}),_({type:x("OAuth / PAT"),token:J("Token",!0)})]).default({type:"Password",username:"username",enable_mfa:!1}).describe(d.of({special:"tabs"}))}).describe(d.of({direction:"two-columns"})),Oe=_({type:x("bigquery"),project:b().nonempty().describe(d.of({label:"Project ID",optionRegex:"(bigquery.?project|gcp.?project|google.?cloud.?project|project.?id)"})),dataset:b().nonempty().describe(d.of({label:"Dataset",optionRegex:"(bigquery.?dataset|dataset)"})),credentials_json:b().describe(d.of({label:"Credentials JSON",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),Be=_({type:x("clickhouse_connect"),host:I(),port:O(8123).optional(),username:T(),password:F(),secure:L().default(!1).describe(d.of({label:"Use HTTPs"})),proxy_path:b().optional().describe(d.of({label:"Proxy Path",placeholder:"/clickhouse"}))}).describe(d.of({direction:"two-columns"})),Ue=_({type:x("timeplus"),host:I().default("localhost"),port:O(8123).optional(),username:T().default("default"),password:F().default("")}).describe(d.of({direction:"two-columns"})),Ke=_({type:x("chdb"),database:R().describe(d.of({label:"Database Path"})),read_only:Le()}).describe(d.of({direction:"two-columns"})),Me=_({type:x("trino"),host:I(),port:O(8080),database:R(),schema:xs().optional(),username:T(),password:F(),async_support:L().default(!1).describe(d.of({label:"Async Support"}))}).describe(d.of({direction:"two-columns"})),We=_({type:x("iceberg"),name:b().describe(d.of({label:"Catalog Name"})),catalog:K("type",[_({type:x("REST"),warehouse:Y(),uri:b().optional().describe(d.of({label:"URI",placeholder:"https://",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"})),token:J()}),_({type:x("SQL"),warehouse:Y(),uri:b().optional().describe(d.of({label:"URI",placeholder:"jdbc:iceberg://host:port/database",optionRegex:"(uri|url|connection.?string|database.?url|jdbc)"}))}),_({type:x("Hive"),warehouse:Y(),uri:oe()}),_({type:x("Glue"),warehouse:Y(),uri:oe()}),_({type:x("DynamoDB"),"dynamodb.profile-name":b().optional().describe(d.of({label:"Profile Name"})),"dynamodb.region":b().optional().describe(d.of({label:"Region"})),"dynamodb.access-key-id":b().optional().describe(d.of({label:"Access Key ID",inputType:"password",optionRegex:"(access.?key.?id|aws.?access.?key)"})),"dynamodb.secret-access-key":b().optional().describe(d.of({label:"Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|aws.?secret)"})),"dynamodb.session-token":b().optional().describe(d.of({label:"Session Token",inputType:"password",optionRegex:"(session.?token|aws.?session)"}))})]).default({type:"REST",token:void 0}).describe(d.of({special:"tabs"}))}),He=_({type:x("datafusion"),sessionContext:L().optional().describe(d.of({label:"Use Session Context"}))}),Ge=_({type:x("pyspark"),host:I().optional(),port:O().optional()}),Ve=_({type:x("redshift"),host:I(),port:O(5439),connectionType:K("type",[_({type:x("IAM credentials"),region:b().describe(d.of({label:"Region"})),aws_access_key_id:b().nonempty().describe(d.of({label:"AWS Access Key ID",inputType:"password",optionRegex:"(access.?key.?id|aws.?access.?key)"})),aws_secret_access_key:b().nonempty().describe(d.of({label:"AWS Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|aws.?secret)"})),aws_session_token:b().optional().describe(d.of({label:"AWS Session Token",inputType:"password",optionRegex:"(session.?token|aws.?session)"}))}),_({type:x("DB credentials"),user:T(),password:F()})]).default({type:"IAM credentials",aws_access_key_id:"",aws_secret_access_key:"",region:""}),database:R()}).describe(d.of({direction:"two-columns"})),ze=_({type:x("databricks"),access_token:J("Access Token",!0),server_hostname:I("Server Hostname"),http_path:oe("HTTP Path",!0),catalog:b().optional().describe(d.of({label:"Catalog"})),schema:b().optional().describe(d.of({label:"Schema"}))}).describe(d.of({direction:"two-columns"})),Qe=_({type:x("supabase"),host:I(),port:O(5432).optional(),database:R(),username:T(),password:F(),disable_client_pooling:L().default(!1).describe(d.of({label:"Disable Client-Side Pooling"}))}).describe(d.of({direction:"two-columns"})),Ss=K("type",[De,Ie,Pe,qe,Ee,Fe,Oe,Be,Ue,Ke,Me,We,He,Ge,Ve,ze,Qe]),Je={sqlmodel:"SQLModel",sqlalchemy:"SQLAlchemy",duckdb:"DuckDB",clickhouse_connect:"ClickHouse Connect",chdb:"chDB",pyiceberg:"PyIceberg",ibis:"Ibis",motherduck:"MotherDuck",redshift:"Redshift",databricks:"Databricks"};var $=class{constructor(t,e,s){this.connection=t,this.orm=e,this.secrets=s}get imports(){let t=new Set(this.generateImports());switch(this.orm){case"sqlalchemy":t.add("import sqlalchemy");break;case"sqlmodel":t.add("import sqlmodel");break;case"duckdb":t.add("import duckdb");break;case"ibis":t.add("import ibis");break}return t}},ks=t=>`_${t}`,ws=class{constructor(){G(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e,s){if(t=ks(t),E(e)){let n=je(e),r=s?`os.environ.get("${n}", "${s}")`:`os.environ.get("${n}")`;return this.secrets[t]=r,t}if(s!=null){let n=`os.environ.get("${e}", "${s}")`;return this.secrets[t]=n,t}return typeof e=="number"||typeof e=="number"?`${e}`:typeof e=="boolean"?M(e):e?`"${e}"`:""}printInFString(t,e,s){if(e===void 0)return"";if(typeof e=="number")return`${e}`;if(typeof e=="boolean")return M(e);let n=this.print(t,e,s);return n.startsWith('"')&&n.endsWith('"')?n.slice(1,-1):`{${n}}`}printPassword(t,e,s,n){let r=s?this.printInFString.bind(this):this.print.bind(this);return E(t)?r(n||"password",t):r(n||"password",e,t)}getSecrets(){return this.secrets}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
+`)}},js=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'sslmode': 'require'}":"",e=this.secrets.printPassword(this.connection.password,"POSTGRES_PASSWORD",!0);return y(`
+      DATABASE_URL = f"postgresql://${this.secrets.printInFString("username",this.connection.username)}:${e}@${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}/${this.secrets.printInFString("database",this.connection.database)}"
+      engine = ${this.orm}.create_engine(DATABASE_URL${t})
+    `)}},vs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'ssl': {'ssl-mode': 'preferred'}}":"",e=this.secrets.printPassword(this.connection.password,"MYSQL_PASSWORD",!0),s=this.secrets.printInFString("database",this.connection.database);return y(`
+      DATABASE_URL = f"mysql+pymysql://${this.secrets.printInFString("username",this.connection.username)}:${e}@${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}/${s}"
+      engine = ${this.orm}.create_engine(DATABASE_URL${t})
+    `)}},$s=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.connection.database?this.secrets.printInFString("database",this.connection.database):"";return y(`
+      ${t.startsWith("{")&&t.endsWith("}")?`DATABASE_URL = f"sqlite:///${t}"`:`DATABASE_URL = "sqlite:///${t}"`}
+      engine = ${this.orm}.create_engine(DATABASE_URL)
+    `)}},Cs=class extends ${generateImports(){return["from snowflake.sqlalchemy import URL"]}generateConnectionCode(){let{authType:t,account:e,database:s,warehouse:n,schema:r,role:a}=this.connection,c={account:this.secrets.print("account",e),database:this.secrets.print("database",s),warehouse:n?this.secrets.print("warehouse",n):void 0,schema:r?this.secrets.print("schema",r):void 0,role:a?this.secrets.print("role",a):void 0};switch(t.type){case"Password":{let i=this.secrets.printPassword(t.password,"SNOWFLAKE_PASSWORD",!1),l={...c,user:this.secrets.print("user",t.username),password:i};return t.enable_mfa?y(`
+            engine = ${this.orm}.create_engine(
+              URL(
+${B(l,p=>`                ${p}`)},
+              ),
+              connect_args={"authenticator": "username_password_mfa"},
+            )
+          `):y(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${B(l,p=>`              ${p}`)},
+            )
+          )
+        `)}case"SSO (Browser)":{let i={...c,user:this.secrets.print("user",t.username),authenticator:'"externalbrowser"'};return y(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${B(i,l=>`              ${l}`)},
+            )
+          )
+        `)}case"Key Pair":{let i={...c,user:this.secrets.print("user",t.username)},l=this.secrets.print("private_key_path",t.private_key_path),p=t.private_key_passphrase?this.secrets.printPassword(t.private_key_passphrase,"SNOWFLAKE_PRIVATE_KEY_PASSPHRASE",!1):void 0,h=['              "authenticator": "SNOWFLAKE_JWT"',`              "private_key_file": ${l}`];return p&&h.push(`              "private_key_file_pwd": ${p}`),y(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${B(i,f=>`              ${f}`)},
+            ),
+            connect_args={
+${h.join(`,
+`)},
+            },
+          )
+        `)}case"OAuth / PAT":{let i=this.secrets.printPassword(t.token,"SNOWFLAKE_TOKEN",!1),l={...c,authenticator:'"oauth"',token:i};return y(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${B(l,p=>`              ${p}`)},
+            )
+          )
+        `)}default:W(t)}}},As=class extends ${get credential(){return ee(this.connection.credentials_json,t=>this.secrets.print("credentials_json",t))}generateImports(){return this.credential.kind==="path"?[]:["import json"]}generateConnectionCode(){let t=this.secrets.printInFString("project",this.connection.project),e=this.secrets.printInFString("dataset",this.connection.dataset),s=this.credential;return s.kind==="path"?y(`
+        engine = ${this.orm}.create_engine(f"bigquery://${t}/${e}", credentials_path="${se(s.path)}")
+      `):y(`
+      credentials = ${s.expr}
+      engine = ${this.orm}.create_engine(f"bigquery://${t}/${e}", credentials_info=credentials)
+    `)}},Ns=class extends ${generateImports(){return[]}generateConnectionCode(){return y(`
+      DATABASE_URL = "${this.secrets.printInFString("database",this.connection.database||":memory:")}"
+      engine = ${this.orm}.connect(DATABASE_URL, read_only=${M(this.connection.read_only)})
+    `)}},Rs=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printInFString("database",this.connection.database);return this.connection.token?y(`
+      conn = duckdb.connect("md:${t}", config={"motherduck_token": ${this.secrets.printPassword(this.connection.token,"MOTHERDUCK_TOKEN",!1)}})
+    `):y(`
+        conn = duckdb.connect("md:${t}")
+        `)}},Ts=class extends ${generateImports(){return["import clickhouse_connect"]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"CLICKHOUSE_PASSWORD",!1),e={host:this.secrets.print("host",this.connection.host),user:this.secrets.print("user",this.connection.username),secure:this.secrets.print("secure",this.connection.secure),port:this.connection.port?this.secrets.print("port",this.connection.port):void 0,password:this.connection.password?t:void 0,proxy_path:this.connection.proxy_path?this.secrets.print("proxy_path",this.connection.proxy_path):void 0};return y(`
+      engine = ${this.orm}.get_client(
+${B(e,s=>`        ${s}`)},
+      )
+    `)}},Ls=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"TIMEPLUS_PASSWORD",!0);return y(`
+      DATABASE_URL = f"timeplus://${this.secrets.printInFString("username",this.connection.username)}:${t}@${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}"
+      engine = ${this.orm}.create_engine(DATABASE_URL)
+    `)}},Ds=class extends ${generateImports(){return["import chdb"]}generateConnectionCode(){let t=this.secrets.print("database",this.connection.database)||'""';return y(`
+      engine = ${this.orm}.connect(${t}, read_only=${M(this.connection.read_only)})
+    `)}},Is=class extends ${generateImports(){return this.connection.async_support?["import aiotrino"]:["import trino.sqlalchemy"]}generateConnectionCode(){let t=this.connection.async_support?"aiotrino":"trino",e=this.connection.schema?`/${this.connection.schema}`:"",s=this.secrets.printInFString("username",this.connection.username),n=this.secrets.printInFString("host",this.connection.host),r=this.secrets.printInFString("port",this.connection.port),a=this.secrets.printInFString("database",this.connection.database),c=this.secrets.printPassword(this.connection.password,"TRINO_PASSWORD",!0);return y(`
+      engine = ${this.orm}.create_engine(f"${t}://${s}:${c}@${n}:${r}/${a}${e}")
+    `)}},Ps=class extends ${getCatalogOptions(){return Object.fromEntries(Object.entries(this.connection.catalog).filter(([t,e])=>e!=null&&e!==""&&t!=="type"))}generateImports(){if(Object.keys(this.getCatalogOptions()).length===0)return["from pyiceberg.catalog import load_catalog"];switch(this.connection.catalog.type){case"REST":return["from pyiceberg.catalog.rest import RestCatalog"];case"SQL":return["from pyiceberg.catalog.sql import SqlCatalog"];case"Hive":return["from pyiceberg.catalog.hive import HiveCatalog"];case"Glue":return["from pyiceberg.catalog.glue import GlueCatalog"];case"DynamoDB":return["from pyiceberg.catalog.dynamodb import DynamoDBCatalog"];default:W(this.connection.catalog)}}generateConnectionCode(){let t=`"${this.connection.name}"`,e=this.getCatalogOptions();if(Object.keys(e).length===0)return`catalog = load_catalog(${t})`;for(let[n,r]of Object.entries(e))E(r)?e[n]=this.secrets.print(n,r):typeof r=="string"&&(e[n]=`"${r}"`);let s=Ms(e,n=>`              ${n}`);switch(this.connection.catalog.type){case"REST":return y(`
+          catalog = RestCatalog(
+            ${t},
+            **{
+${s}
+            },
+          )
+        `);case"SQL":return y(`
+          catalog = SqlCatalog(
+            ${t},
+            **{
+${s}
+            },
+          )
+        `);case"Hive":return y(`
+          catalog = HiveCatalog(
+            ${t},
+            **{
+${s}
+            },
+          )
+        `);case"Glue":return y(`
+          catalog = GlueCatalog(
+            ${t},
+            **{
+${s}
+            },
+          )
+        `);case"DynamoDB":return y(`
+          catalog = DynamoDBCatalog(
+            ${t},
+            **{
+${s}
+            },
+          )
+        `);default:W(this.connection.catalog)}}},qs=class extends ${generateImports(){return["from datafusion import SessionContext"]}generateConnectionCode(){return this.connection.sessionContext?y(`
+        ctx = SessionContext()
+        # Sample table
+        _ = ctx.from_pydict({"a": [1, 2, 3]}, "my_table")
+
+        con = ibis.datafusion.connect(ctx)
+      `):y(`
+      con = ibis.datafusion.connect()
+    `)}},Es=class extends ${generateImports(){return["from pyspark.sql import SparkSession"]}generateConnectionCode(){return this.connection.host||this.connection.port?y(`
+        session = SparkSession.builder.remote(f"sc://${this.secrets.printInFString("host",this.connection.host)}:${this.secrets.printInFString("port",this.connection.port)}").getOrCreate()
+        con = ibis.pyspark.connect(session)
+      `):y(`
+      con = ibis.pyspark.connect()
+    `)}},Fs=class extends ${generateImports(){return["import redshift_connector"]}generateConnectionCode(){let t=this.secrets.print("host",this.connection.host),e=this.secrets.print("port",this.connection.port),s=this.secrets.print("database",this.connection.database);if(this.connection.connectionType.type==="IAM credentials"){let a=this.secrets.print("aws_access_key_id",this.connection.connectionType.aws_access_key_id),c=this.secrets.print("aws_secret_access_key",this.connection.connectionType.aws_secret_access_key),i=this.connection.connectionType.aws_session_token?this.secrets.print("aws_session_token",this.connection.connectionType.aws_session_token):void 0;return y(`
+        con = redshift_connector.connect(
+${B({iam:!0,host:t,port:e,region:`"${this.connection.connectionType.region}"`,database:s,access_key_id:a,secret_access_key:c,...i&&{session_token:i}},l=>`          ${l}`)},
+        )
+      `)}let n=this.connection.connectionType.user?this.secrets.print("user",this.connection.connectionType.user):void 0,r=this.connection.connectionType.password?this.secrets.printPassword(this.connection.connectionType.password,"REDSHIFT_PASSWORD",!1):void 0;return y(`
+      con = redshift_connector.connect(
+${B({host:t,port:e,database:s,...n&&{user:n},...r&&{password:r}},a=>`          ${a}`)},
+      )
+    `)}},Os=class extends ${generateImports(){return[]}generateConnectionCode(){let t=this.orm!=="ibis",e=this.secrets.printPassword(this.connection.access_token,"DATABRICKS_ACCESS_TOKEN",t,"access_token"),s=this.secrets.printPassword(this.connection.server_hostname,"DATABRICKS_SERVER_HOSTNAME",t,"server_hostname"),n=this.secrets.printPassword(this.connection.http_path,"DATABRICKS_HTTP_PATH",t,"http_path"),r=this.connection.catalog?this.secrets.printInFString("catalog",this.connection.catalog):void 0,a=this.connection.schema?this.secrets.printInFString("schema",this.connection.schema):void 0,c=`databricks://token:${e}@${s}?http_path=${n}`;return r&&(c+=`&catalog=${r}`),a&&(c+=`&schema=${a}`),this.orm==="ibis"?y(`
+        engine = ibis.databricks.connect(
+          server_hostname=${s},
+          http_path=${n},${r?`
+          catalog=${r},`:""}${a?`
+          schema=${a},`:""}
+          access_token=${e}
+        )
+      `):y(`
+      DATABASE_URL = f"${c}"
+      engine = ${this.orm}.create_engine(DATABASE_URL)
+      `)}},Bs=class extends ${generateImports(){return this.connection.disable_client_pooling?["from sqlalchemy.pool import NullPool"]:[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"SUPABASE_PASSWORD",!0),e=this.secrets.printInFString("username",this.connection.username),s=this.secrets.printInFString("host",this.connection.host),n=this.secrets.printInFString("port",this.connection.port),r=this.secrets.printInFString("database",this.connection.database),a=this.connection.disable_client_pooling?", poolclass=NullPool":"";return y(`
+      DATABASE_URL = f"postgresql+psycopg2://${e}:${t}@${s}:${n}/${r}?sslmode=require"
+      engine = ${this.orm}.create_engine(DATABASE_URL${a})
+    `)}},Us=class{constructor(){G(this,"secrets",new ws)}createGenerator(t,e){switch(t.type){case"postgres":return new js(t,e,this.secrets);case"mysql":return new vs(t,e,this.secrets);case"sqlite":return new $s(t,e,this.secrets);case"snowflake":return new Cs(t,e,this.secrets);case"bigquery":return new As(t,e,this.secrets);case"duckdb":return new Ns(t,e,this.secrets);case"motherduck":return new Rs(t,e,this.secrets);case"clickhouse_connect":return new Ts(t,e,this.secrets);case"timeplus":return new Ls(t,e,this.secrets);case"chdb":return new Ds(t,e,this.secrets);case"trino":return new Is(t,e,this.secrets);case"iceberg":return new Ps(t,e,this.secrets);case"datafusion":return new qs(t,e,this.secrets);case"pyspark":return new Es(t,e,this.secrets);case"redshift":return new Fs(t,e,this.secrets);case"databricks":return new Os(t,e,this.secrets);case"supabase":return new Bs(t,e,this.secrets);default:W(t)}}};function Ks(t,e){if(!(e in Je))throw Error(`Unsupported library: ${e}`);Ss.parse(t);let s=new Us,n=s.createGenerator(t,e),r=n.generateConnectionCode(),a=s.secrets,c=[...new Set([...a.imports,...n.imports])].toSorted();c.push("");let i=a.formatSecrets();return i&&c.push(i),c.push(r.trim()),c.join(`
+`)}function M(t){return t.toString().charAt(0).toUpperCase()+t.toString().slice(1)}function B(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,n])=>e(typeof n=="boolean"?`${s}=${M(n)}`:`${s}=${n}`)).join(`,
+`)}function Ms(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,n])=>{let r=`"${s}"`;return e(typeof n=="boolean"?`${r}: ${M(n)}`:`${r}: ${n}`)}).join(`,
+`)}var Ye=U(),Xe=[{name:"PostgreSQL",schema:De,color:"#336791",logo:"postgres",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"MySQL",schema:Ie,color:"#00758F",logo:"mysql",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"SQLite",schema:Pe,color:"#003B57",logo:"sqlite",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DuckDB",schema:qe,color:"#FFD700",logo:"duckdb",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"MotherDuck",schema:Ee,color:"#ff9538",logo:"motherduck",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"Snowflake",schema:Fe,color:"#29B5E8",logo:"snowflake",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse",schema:Be,color:"#2C2C1D",logo:"clickhouse",connectionLibraries:{libraries:["clickhouse_connect"],preferred:"clickhouse_connect"}},{name:"Timeplus",schema:Ue,color:"#B83280",logo:"timeplus",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"BigQuery",schema:Oe,color:"#4285F4",logo:"bigquery",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse Embedded",schema:Ke,color:"#f2b611",logo:"clickhouse",connectionLibraries:{libraries:["chdb"],preferred:"chdb"}},{name:"Trino",schema:Me,color:"#d466b6",logo:"trino",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DataFusion",schema:He,color:"#202A37",logo:"datafusion",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"PySpark",schema:Ge,color:"#1C5162",logo:"pyspark",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"Redshift",schema:Ve,color:"#522BAE",logo:"redshift",connectionLibraries:{libraries:["redshift"],preferred:"redshift"}},{name:"Databricks",schema:ze,color:"#c41e0c",logo:"databricks",connectionLibraries:{libraries:["sqlalchemy","sqlmodel","ibis"],preferred:"sqlalchemy"}},{name:"Supabase",schema:Qe,color:"#238F5F",logo:"supabase",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}}],Ze=[{name:"Iceberg",schema:We,color:"#000000",logo:"iceberg",connectionLibraries:{libraries:["pyiceberg"],preferred:"pyiceberg"}}],Ws=[...Xe,...Ze],Hs=t=>{let e=(0,Ye.c)(12),{onSelect:s}=t,n;e[0]===s?n=e[1]:(n=Xe.map(p=>{let{name:h,schema:f,color:u,logo:g}=p;return(0,o.jsx)(ne,{name:h,color:u,icon:(0,o.jsx)(ce,{name:g,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(f)},h)}),e[0]=s,e[1]=n);let r;e[2]===n?r=e[3]:(r=(0,o.jsx)(re,{children:n}),e[2]=n,e[3]=r);let a;e[4]===Symbol.for("react.memo_cache_sentinel")?(a=(0,o.jsxs)("h4",{className:"font-semibold text-muted-foreground text-lg flex items-center gap-4 my-2",children:["Data Catalogs",(0,o.jsx)("hr",{className:"flex-1"})]}),e[4]=a):a=e[4];let c;e[5]===s?c=e[6]:(c=Ze.map(p=>{let{name:h,schema:f,color:u,logo:g}=p;return(0,o.jsx)(ne,{name:h,color:u,icon:(0,o.jsx)(ce,{name:g,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(f)},h)}),e[5]=s,e[6]=c);let i;e[7]===c?i=e[8]:(i=(0,o.jsx)(re,{children:c}),e[7]=c,e[8]=i);let l;return e[9]!==r||e[10]!==i?(l=(0,o.jsxs)(o.Fragment,{children:[r,a,i]}),e[9]=r,e[10]=i,e[11]=l):l=e[11],l};const Gs=t=>{let e=(0,Ye.c)(13),{onSubmit:s,header:n}=t,[r,a]=(0,D.useState)(null);if(!r){let u;e[0]===Symbol.for("react.memo_cache_sentinel")?(u=(0,o.jsx)("div",{children:(0,o.jsx)(Hs,{onSelect:a})}),e[0]=u):u=e[0];let g;return e[1]===n?g=e[2]:(g=(0,o.jsxs)(o.Fragment,{children:[n,u]}),e[1]=n,e[2]=g),g}let c;e[3]===r?c=e[4]:(c=Ws.find(u=>u.schema===r),e[3]=r,e[4]=c);let i=c==null?void 0:c.connectionLibraries,l;e[5]===(i==null?void 0:i.libraries)?l=e[6]:(l=(i==null?void 0:i.libraries)??[],e[5]=i==null?void 0:i.libraries,e[6]=l);let p=(i==null?void 0:i.preferred)??"sqlalchemy",h;e[7]===Symbol.for("react.memo_cache_sentinel")?(h=()=>a(null),e[7]=h):h=e[7];let f;return e[8]!==s||e[9]!==r||e[10]!==l||e[11]!==p?(f=(0,o.jsx)(Te,{schema:r,libraries:l,preferredLibrary:p,displayNames:Je,libraryLabel:"Preferred connection library",generateCode:Vs,onSubmit:s,onBack:h}),e[8]=s,e[9]=r,e[10]=l,e[11]=p,e[12]=f):f=e[12],f};function Vs(t,e){return Ks(t,e)}var zs=U();async function Qs(){return(await gt.request({})).sources}function Js(){let t=(0,zs.c)(1),e;return t[0]===Symbol.for("react.memo_cache_sentinel")?(e=[],t[0]=e):e=t[0],Se(Qs,e)}var ae=U();const Ys=t=>{let e=(0,ae.c)(8),{sources:s,onAdd:n}=t;if(s.length===0)return null;let r;e[0]===Symbol.for("react.memo_cache_sentinel")?(r=(0,o.jsxs)("div",{className:"mr-1 flex items-center gap-1.5",children:[(0,o.jsx)(Bt,{className:"h-3.5 w-3.5 text-muted-foreground"}),(0,o.jsx)("h3",{id:"quick-add-data-sources-title",className:"text-sm font-medium",children:"Quick add"})]}),e[0]=r):r=e[0];let a;if(e[1]!==n||e[2]!==s){let i;e[4]===n?i=e[5]:(i=l=>(0,o.jsx)(he,{side:"bottom",content:(0,o.jsx)(Xs,{source:l}),children:(0,o.jsxs)("button",{type:"button","aria-label":`Add ${l.displayName} connection`,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",onClick:()=>n(l),children:[(0,o.jsx)(Ot,{className:"h-3 w-3"}),l.displayName]})},l.id),e[4]=n,e[5]=i),a=s.map(i),e[1]=n,e[2]=s,e[3]=a}else a=e[3];let c;return e[6]===a?c=e[7]:(c=(0,o.jsx)("section",{"aria-labelledby":"quick-add-data-sources-title",className:"rounded-md border bg-muted/30 px-3 py-2",children:(0,o.jsxs)("div",{className:"flex flex-wrap items-center gap-2",children:[r,(0,o.jsx)(It,{delayDuration:200,children:a})]})}),e[6]=a,e[7]=c),c};var Xs=t=>{let e=(0,ae.c)(17),{source:s}=t,n;e[0]===s.displayName?n=e[1]:(n=(0,o.jsx)("div",{className:"font-medium",children:s.displayName}),e[0]=s.displayName,e[1]=n);let r;e[2]===s.origins?r=e[3]:(r=s.origins.map(er).join(" and "),e[2]=s.origins,e[3]=r);let a;e[4]===r?a=e[5]:(a=(0,o.jsxs)("div",{className:"text-xs text-muted-foreground",children:["Detected from"," ",r]}),e[4]=r,e[5]=a);let c;e[6]!==n||e[7]!==a?(c=(0,o.jsxs)("div",{children:[n,a]}),e[6]=n,e[7]=a,e[8]=c):c=e[8];let i;e[9]===s.configuration?i=e[10]:(i=s.configuration.map(tr),e[9]=s.configuration,e[10]=i);let l;e[11]===i?l=e[12]:(l=(0,o.jsx)("dl",{className:"grid grid-cols-[auto_1fr] gap-x-3 gap-y-1 text-xs",children:i}),e[11]=i,e[12]=l);let p;e[13]===Symbol.for("react.memo_cache_sentinel")?(p=(0,o.jsx)("div",{className:"text-xs text-muted-foreground",children:"Click to add a configured cell."}),e[13]=p):p=e[13];let h;return e[14]!==c||e[15]!==l?(h=(0,o.jsxs)("div",{className:"min-w-64 space-y-2 py-1",children:[c,l,p]}),e[14]=c,e[15]=l,e[16]=h):h=e[16],h};const Zs=t=>{let e=(0,ae.c)(8),{onSubmit:s}=t,n=Re(),{data:r}=Js(),a;e[0]===r?a=e[1]:(a=r??[],e[0]=r,e[1]=a);let c;e[2]!==n||e[3]!==s?(c=l=>{n(l.code),s()},e[2]=n,e[3]=s,e[4]=c):c=e[4];let i;return e[5]!==a||e[6]!==c?(i=(0,o.jsx)(Ys,{sources:a,onAdd:c}),e[5]=a,e[6]=c,e[7]=i):i=e[7],i};function er(t){return t.label}function tr(t){return(0,o.jsxs)("div",{className:"contents",children:[(0,o.jsx)("dt",{className:"text-muted-foreground",children:t.field}),(0,o.jsx)("dd",{children:(0,o.jsx)("code",{children:t.value.kind==="environment-variable"?`os.environ["${t.value.name}"]`:t.value.value})})]},t.value.kind==="environment-variable"?t.value.name:`${t.field}:${t.value.value}`)}var et=()=>K("type",[_({type:x("Access keys"),access_key_id:b().optional().describe(d.of({label:"Access Key ID",description:"Leave empty to use the default credential chain",inputType:"password",optionRegex:"(access.?key.?id|object.?storage.?key|aws.?access.?key)"})),secret_access_key:b().optional().describe(d.of({label:"Secret Access Key",inputType:"password",optionRegex:"(secret.?access.?key|object.?storage.?secret|aws.?secret)"}))}),_({type:x("Container credentials"),container_credentials_full_uri:b().nonempty().describe(d.of({label:"Credentials URI",description:"Endpoint that vends credentials to the container",placeholder:"http://169.254.170.23/v1/credentials",optionRegex:"container.?credentials"})),container_authorization_token_file:b().nonempty().describe(d.of({label:"Authorization Token File",description:"File holding the token sent to the credentials URI",placeholder:"/var/run/secrets/token",optionRegex:"(container.?authorization.?token.?file|container.?auth.?token)"}))})]).default({type:"Access keys"}).describe(d.of({label:"Credentials",special:"tabs"}));const tt=_({type:x("s3"),bucket:b().nonempty().describe(d.of({label:"Bucket",placeholder:"my-bucket",optionRegex:"(bucket|s3.?bucket)"})),region:b().optional().describe(d.of({label:"Region",placeholder:"us-east-1",optionRegex:"(region|aws.?region)"})),endpoint_url:b().optional().describe(d.of({label:"Endpoint URL",description:"Ignored if the AWS_ENDPOINT_URL_S3 environment variable is set",placeholder:"https://s3.amazonaws.com",optionRegex:"(endpoint|s3.?url|s3.?endpoint)"})),allow_http:L().default(!1).describe(d.of({label:"Allow HTTP",description:"Required for plain-http (non-TLS) endpoints"})),auth:et()}).describe(d.of({direction:"two-columns"})),st=_({type:x("gcs"),bucket:b().nonempty().describe(d.of({label:"Bucket",placeholder:"my-bucket",optionRegex:"(bucket|gcs.?bucket|google.?bucket)"})),service_account_key:b().optional().describe(d.of({label:"Service Account Key (JSON)",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),rt=_({type:x("azure"),container:b().nonempty().describe(d.of({label:"Container",placeholder:"my-container"})),account_name:b().nonempty().describe(d.of({label:"Account Name",placeholder:"storageaccount",optionRegex:"(azure.?account|account.?name|storage.?account)"})),account_key:b().optional().describe(d.of({label:"Account Key",inputType:"password",optionRegex:"(azure.?key|account.?key|storage.?key)"}))}).describe(d.of({direction:"two-columns"})),nt=_({type:x("coreweave"),bucket:b().nonempty().describe(d.of({label:"Bucket",placeholder:"bucket-name"})),region:b().nonempty().describe(d.of({label:"Region",placeholder:"US-EAST-04A"})),auth:et()}).describe(d.of({direction:"two-columns"})),ot=_({type:x("gdrive"),credentials_json:b().optional().describe(d.of({label:"Service Account JSON",description:"Leave empty to use browser-based authentication",special:"secret_textarea"}))}).describe(d.of({direction:"two-columns"})),sr=K("type",[tt,st,rt,nt,ot]),rr={obstore:"obstore",fsspec:"fsspec"};var nr=class{constructor(){G(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e){if(e===void 0||e==="")return"";let s=`_${t}`;if(E(e)){let n=je(e);return this.secrets[s]=`os.environ.get("${n}")`,s}return`"${e}"`}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
+`)}};function at(t,e){if((t==null?void 0:t.type)==="Container credentials")return[`    container_credentials_full_uri=${e.print("container_credentials_full_uri",t.container_credentials_full_uri)},`,`    container_authorization_token_file=${e.print("container_authorization_token_file",t.container_authorization_token_file)},`];let s=[];return t!=null&&t.access_key_id&&s.push(`    access_key_id=${e.print("access_key_id",t.access_key_id)},`),t!=null&&t.secret_access_key&&s.push(`    secret_access_key=${e.print("secret_access_key",t.secret_access_key)},`),s}function or(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import S3Store"]),r=[];return t.region&&r.push(`    region=${e.print("region",t.region)},`),r.push(...at(t.auth,e)),t.endpoint_url&&r.push(`    endpoint_url=${e.print("endpoint_url",t.endpoint_url)},`),t.allow_http&&r.push("    allow_http=True,"),{imports:n,code:y(`
+    store = S3Store(${s},${r.length>0?`
+${r.join(`
+`)}
+`:""})
+  `)}}function ar(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import GCSStore"]);if(!t.service_account_key)return{imports:n,code:y(`
+        store = GCSStore(${s})
+      `)};let r=ee(t.service_account_key,a=>e.print("service_account_key",a));return r.kind==="path"?{imports:n,code:y(`
+      store = GCSStore(${s},
+          service_account_path="${se(r.path)}",
+      )
+    `)}:(n.add("import json"),{imports:n,code:y(`
+    _credentials = ${r.expr}
+    store = GCSStore(${s},
+        service_account_key=_credentials,
+    )
+  `)})}function ir(t,e){let s=e.print("container",t.container),n=e.print("account_name",t.account_name),r=new Set(["from obstore.store import AzureStore"]),a=[`account_name=${n},`];return t.account_key&&a.push(`account_key=${e.print("account_key",t.account_key)},`),{imports:r,code:`store = AzureStore(${s},
+${a.map(c=>`    ${c}`).join(`
+`)}
+)`}}function cr(t,e){let s=e.print("bucket",t.bucket),n=new Set(["from obstore.store import S3Store"]),r=[`    region=${e.print("region",t.region)},`];return r.push(...at(t.auth,e)),r.push(`    endpoint="https://${t.bucket}.cwobject.com",`,"    virtual_hosted_style_request=True,"),{imports:n,code:y(`
+    store = S3Store(${s},${`
+${r.join(`
+`)}
+`})
+  `)}}function lr(t,e){let{secrets:s,isEmbedded:n=!1}=e,r=new Set(["from gdrive_fsspec import GoogleDriveFileSystem"]);if(t.credentials_json){let a=ee(t.credentials_json,c=>s.print("credentials_json",c));return a.kind==="path"?{imports:r,code:y(`
+        fs = GoogleDriveFileSystem(creds="${se(a.path)}", token="service_account", use_listings_cache=False, skip_instance_cache=True)
+      `)}:(r.add("import json"),{imports:r,code:y(`
+      _creds = ${a.expr}
+      fs = GoogleDriveFileSystem(creds=_creds, token="service_account", use_listings_cache=False, skip_instance_cache=True)
+    `)})}return{imports:r,code:y(n?`
+        fs = GoogleDriveFileSystem(use_listings_cache=False, skip_instance_cache=True, auth_kwargs={"use_local_webserver": False})
+        mo.output.clear_console()
+      `:`
+        fs = GoogleDriveFileSystem(use_listings_cache=False, skip_instance_cache=True)
+      `)}}function dr(t,e){sr.parse(t);let s=new nr,n;switch(t.type){case"s3":n=or(t,s);break;case"gcs":n=ar(t,s);break;case"azure":n=ir(t,s);break;case"coreweave":n=cr(t,s);break;case"gdrive":n=lr(t,{secrets:s,isEmbedded:e.isEmbedded});break;default:W(t)}let r=[...new Set([...s.imports,...n.imports])].toSorted();r.push("");let a=s.formatSecrets();return a&&r.push(a),r.push(n.code.trim()),r.join(`
+`)}var it=U(),pr="#232F3E",ct=[{name:"Amazon S3",schema:tt,protocol:"s3",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Cloud Storage",schema:st,protocol:"gcs",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Azure Blob Storage",schema:rt,protocol:"azure",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"CoreWeave",schema:nt,protocol:"coreweave",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Drive",schema:ot,protocol:"gdrive",storageLibraries:{libraries:["fsspec"],preferred:"fsspec"}}],hr=t=>{let e=(0,it.c)(4),{onSelect:s}=t,n;e[0]===s?n=e[1]:(n=ct.map(a=>{let{name:c,schema:i,protocol:l}=a;return(0,o.jsx)(ne,{name:c,color:pr,icon:(0,o.jsx)(Ft,{protocol:l,forceDark:!0,className:"w-7.5 h-7.5"}),onSelect:()=>s(i)},c)}),e[0]=s,e[1]=n);let r;return e[2]===n?r=e[3]:(r=(0,o.jsx)(re,{children:n}),e[2]=n,e[3]=r),r};const mr=t=>{let e=(0,it.c)(16),{onSubmit:s,header:n}=t,{isEmbedded:r}=Jt(),[a,c]=(0,D.useState)(null);if(!a){let m;e[0]===Symbol.for("react.memo_cache_sentinel")?(m=(0,o.jsx)("div",{children:(0,o.jsx)(hr,{onSelect:c})}),e[0]=m):m=e[0];let S;return e[1]===n?S=e[2]:(S=(0,o.jsxs)(o.Fragment,{children:[n,m]}),e[1]=n,e[2]=S),S}let i;e[3]===a?i=e[4]:(i=ct.find(m=>m.schema===a),e[3]=a,e[4]=i);let l=i==null?void 0:i.storageLibraries,p;e[5]===(l==null?void 0:l.libraries)?p=e[6]:(p=(l==null?void 0:l.libraries)??[],e[5]=l==null?void 0:l.libraries,e[6]=p);let h=(l==null?void 0:l.preferred)??"obstore",f;e[7]===r?f=e[8]:(f=(m,S)=>dr(m,{library:S,isEmbedded:r}),e[7]=r,e[8]=f);let u;e[9]===Symbol.for("react.memo_cache_sentinel")?(u=()=>c(null),e[9]=u):u=e[9];let g;return e[10]!==s||e[11]!==a||e[12]!==p||e[13]!==h||e[14]!==f?(g=(0,o.jsx)(Te,{schema:a,libraries:p,preferredLibrary:h,displayNames:rr,libraryLabel:"Preferred storage library",generateCode:f,onSubmit:s,onBack:u}),e[10]=s,e[11]=a,e[12]=p,e[13]=h,e[14]=f,e[15]=g):g=e[15],g};var lt=U();const ur=t=>{let e=(0,lt.c)(9),{children:s,defaultTab:n}=t,r=n===void 0?"databases":n,[a,c]=(0,D.useState)(!1),i;e[0]===s?i=e[1]:(i=(0,o.jsx)(Gt,{asChild:!0,children:s}),e[0]=s,e[1]=i);let l;e[2]===Symbol.for("react.memo_cache_sentinel")?(l=()=>c(!1),e[2]=l):l=e[2];let p;e[3]===r?p=e[4]:(p=(0,o.jsx)(dt,{defaultTab:r,onClose:l}),e[3]=r,e[4]=p);let h;return e[5]!==a||e[6]!==i||e[7]!==p?(h=(0,o.jsxs)(Qt,{open:a,onOpenChange:c,children:[i,p]}),e[5]=a,e[6]=i,e[7]=p,e[8]=h):h=e[8],h},dt=t=>{let e=(0,lt.c)(23),{defaultTab:s,onClose:n}=t,[r,a]=(0,D.useState)(s===void 0?"databases":s),c;e[0]===Symbol.for("react.memo_cache_sentinel")?(c=(0,o.jsxs)($t,{className:"w-full mb-4",children:[(0,o.jsx)(de,{value:"databases",className:"flex-1",children:"Databases & Catalogs"}),(0,o.jsx)(de,{value:"storage",className:"flex-1",children:"Remote Storages"})]}),e[0]=c):c=e[0];let i=c,l;e[1]===r?l=e[2]:(l=r==="databases"?(0,o.jsxs)(o.Fragment,{children:["Don't see your database or connection method? A"," ",(0,o.jsx)(V,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"code snippet"})," ","is all you need."]}):(0,o.jsxs)(o.Fragment,{children:["Don't see your storage or connection method? A"," ",(0,o.jsx)(V,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/#creating-a-storage-connection",children:"code snippet"})," ","is all you need."]}),e[1]=r,e[2]=l);let p=l,h;e[3]===Symbol.for("react.memo_cache_sentinel")?(h=(0,o.jsx)(Ht,{children:"Add Connection"}),e[3]=h):h=e[3];let f;e[4]===Symbol.for("react.memo_cache_sentinel")?(f=(0,o.jsx)(V,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"database, data catalog"}),e[4]=f):f=e[4];let u;e[5]===Symbol.for("react.memo_cache_sentinel")?(u=(0,o.jsx)(V,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/",children:"remote storage"}),e[5]=u):u=e[5];let g;e[6]===p?g=e[7]:(g=(0,o.jsxs)(Wt,{children:[h,(0,o.jsxs)(zt,{children:["Connect to a"," ",f," ","or"," ",u," ","directly from your notebook.",(0,o.jsx)("span",{className:"block",children:p})]})]}),e[6]=p,e[7]=g);let m;e[8]===n?m=e[9]:(m=(0,o.jsx)(Zs,{onSubmit:n}),e[8]=n,e[9]=m);let S;e[10]===Symbol.for("react.memo_cache_sentinel")?(S=C=>a(C),e[10]=S):S=e[10];let w;e[11]===n?w=e[12]:(w=(0,o.jsx)(pe,{value:"databases",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,o.jsx)(Gs,{onSubmit:n,header:i})}),e[11]=n,e[12]=w);let k;e[13]===n?k=e[14]:(k=(0,o.jsx)(pe,{value:"storage",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,o.jsx)(mr,{onSubmit:n,header:i})}),e[13]=n,e[14]=k);let v;e[15]!==r||e[16]!==w||e[17]!==k?(v=(0,o.jsxs)(wt,{value:r,onValueChange:S,children:[w,k]}),e[15]=r,e[16]=w,e[17]=k,e[18]=v):v=e[18];let A;return e[19]!==v||e[20]!==g||e[21]!==m?(A=(0,o.jsxs)(Vt,{className:"max-h-[75vh] overflow-y-auto",children:[g,m,v]}),e[19]=v,e[20]=g,e[21]=m,e[22]=A):A=e[22],A};export{dt as n,ur as t};