@marimo-team/frontend 0.20.5-dev62 → 0.20.5-dev64

package/src/components/editor/connections/database/as-code.ts

@@ -293,34 +293,124 @@ class SnowflakeGenerator extends CodeGenerator<"snowflake"> {
   }
 
   generateConnectionCode(): string {
-    const password = this.secrets.printPassword(
-      this.connection.password,
-      "SNOWFLAKE_PASSWORD",
-      false,
-    );
-    const params = {
-      account: this.secrets.print("account", this.connection.account),
-      user: this.secrets.print("user", this.connection.username),
-      database: this.secrets.print("database", this.connection.database),
-      warehouse: this.connection.warehouse
-        ? this.secrets.print("warehouse", this.connection.warehouse)
-        : undefined,
-      schema: this.connection.schema
-        ? this.secrets.print("schema", this.connection.schema)
-        : undefined,
-      role: this.connection.role
-        ? this.secrets.print("role", this.connection.role)
+    const { authType, account, database, warehouse, schema, role } =
+      this.connection;
+    const baseParams: Record<string, string | undefined> = {
+      account: this.secrets.print("account", account),
+      database: this.secrets.print("database", database),
+      warehouse: warehouse
+        ? this.secrets.print("warehouse", warehouse)
         : undefined,
-      password: password,
+      schema: schema ? this.secrets.print("schema", schema) : undefined,
+      role: role ? this.secrets.print("role", role) : undefined,
     };
 
-    return dedent(`
-      engine = ${this.orm}.create_engine(
-        URL(
-${formatUrlParams(params, (inner) => `          ${inner}`)},
-        )
-      )
-    `);
+    switch (authType.type) {
+      case "Password": {
+        const password = this.secrets.printPassword(
+          authType.password,
+          "SNOWFLAKE_PASSWORD",
+          false,
+        );
+        const params = {
+          ...baseParams,
+          user: this.secrets.print("user", authType.username),
+          password,
+        };
+        if (authType.enable_mfa) {
+          return dedent(`
+            engine = ${this.orm}.create_engine(
+              URL(
+${formatUrlParams(params, (inner) => `                ${inner}`)},
+              ),
+              connect_args={"authenticator": "username_password_mfa"},
+            )
+          `);
+        }
+        return dedent(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${formatUrlParams(params, (inner) => `              ${inner}`)},
+            )
+          )
+        `);
+      }
+
+      case "SSO (Browser)": {
+        const params = {
+          ...baseParams,
+          user: this.secrets.print("user", authType.username),
+          authenticator: '"externalbrowser"',
+        };
+        return dedent(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${formatUrlParams(params, (inner) => `              ${inner}`)},
+            )
+          )
+        `);
+      }
+
+      case "Key Pair": {
+        const params = {
+          ...baseParams,
+          user: this.secrets.print("user", authType.username),
+        };
+        const privateKeyPath = this.secrets.print(
+          "private_key_path",
+          authType.private_key_path,
+        );
+        const passphrase = authType.private_key_passphrase
+          ? this.secrets.printPassword(
+              authType.private_key_passphrase,
+              "SNOWFLAKE_PRIVATE_KEY_PASSPHRASE",
+              false,
+            )
+          : undefined;
+        const connectArgLines = [
+          `              "authenticator": "SNOWFLAKE_JWT"`,
+          `              "private_key_file": ${privateKeyPath}`,
+        ];
+        if (passphrase) {
+          connectArgLines.push(
+            `              "private_key_file_pwd": ${passphrase}`,
+          );
+        }
+        return dedent(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${formatUrlParams(params, (inner) => `              ${inner}`)},
+            ),
+            connect_args={
+${connectArgLines.join(",\n")},
+            },
+          )
+        `);
+      }
+
+      case "OAuth / PAT": {
+        const token = this.secrets.printPassword(
+          authType.token,
+          "SNOWFLAKE_TOKEN",
+          false,
+        );
+        const params = {
+          ...baseParams,
+          authenticator: '"oauth"',
+          token,
+        };
+        return dedent(`
+          engine = ${this.orm}.create_engine(
+            URL(
+${formatUrlParams(params, (inner) => `              ${inner}`)},
+            )
+          )
+        `);
+      }
+
+      default:
+        assertNever(authType);
+    }
   }
 }
 

package/src/components/editor/connections/database/schemas.ts

@@ -218,12 +218,59 @@ export const SnowflakeConnectionSchema = z
           optionRegex: ".*snowflake.*",
         }),
       ),
-    username: usernameField(),
-    password: passwordField(),
     role: z
       .string()
       .optional()
       .describe(FieldOptions.of({ label: "Role" })),
+    authType: z
+      .discriminatedUnion("type", [
+        z.object({
+          type: z.literal("Password"),
+          username: usernameField(),
+          password: passwordField(),
+          enable_mfa: z
+            .boolean()
+            .default(false)
+            .describe(FieldOptions.of({ label: "Enable MFA (Duo Push)" })),
+        }),
+        z.object({
+          type: z.literal("SSO (Browser)"),
+          username: usernameField(),
+        }),
+        z.object({
+          type: z.literal("Key Pair"),
+          username: usernameField(),
+          private_key_path: z
+            .string()
+            .nonempty()
+            .describe(
+              FieldOptions.of({
+                label: "Private Key Path",
+                placeholder: "/path/to/rsa_key.p8",
+              }),
+            ),
+          private_key_passphrase: z
+            .string()
+            .optional()
+            .describe(
+              FieldOptions.of({
+                label: "Private Key Passphrase",
+                inputType: "password",
+                optionRegex: ".*passphrase.*",
+              }),
+            ),
+        }),
+        z.object({
+          type: z.literal("OAuth / PAT"),
+          token: tokenField("Token", true),
+        }),
+      ])
+      .default({
+        type: "Password",
+        username: "username",
+        enable_mfa: false,
+      })
+      .describe(FieldOptions.of({ special: "tabs" })),
   })
   .describe(FieldOptions.of({ direction: "two-columns" }));
 

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

@@ -1,138 +0,0 @@
-var Ge=Object.defineProperty;var Ve=(t,e,s)=>e in t?Ge(t,e,{enumerable:!0,configurable:!0,writable:!0,value:s}):t[e]=s;var K=(t,e,s)=>Ve(t,typeof e!="symbol"?e+"":e,s);import{s as oe}from"./chunk-LvLJmgfZ.js";import{t as Ze}from"./react-Bj1aDYRI.js";import{Rn as g,T as Qe,Tt as ne}from"./cells-5wegzPD4.js";import{D as H,P as y,R as u,T as q,_ as ce,i as Je,k as _,m as Ye}from"./zod-H_cgTO0M.js";import{t as P}from"./compiler-runtime-B3qBwwSJ.js";import{t as z}from"./assertNever-CTRGz-ul.js";import{t as Xe}from"./_createAggregator-zT4tQxzT.js";import{h as et}from"./useEventListener-BGihH8bs.js";import{t as tt}from"./jsx-runtime-icT_Ltz2.js";import{t as Z}from"./button-BZAv3veu.js";import{t as W}from"./cn-BoaCR37F.js";import{t as st}from"./createLucideIcon-Cm-Z7dYr.js";import{t as rt}from"./circle-plus-CGG-gArM.js";import{a as Q,c as ie,o as ot,p as nt,r as ct,t as it}from"./dropdown-menu-CJ-Wwa2I.js";import{t as at}from"./database-zap-DTWCDKdn.js";import{t as lt}from"./github-BVtI-3F1.js";import{m as dt,o as pt,t as ht,u as i}from"./form-p9cBnWwo.js";import{n as mt}from"./DeferredRequestRegistry-O6RDJKs0.js";import{o as ut,r as bt}from"./input-C5eatTLl.js";import{c as ft,d as gt,f as yt,h as _t,l as wt,n as St,p as xt,s as kt,t as vt,u as $t,v as jt}from"./azure-a3T26UGw.js";import{r as Ct}from"./useTheme-Bp1J8nG-.js";import{c as At,i as Rt,n as Tt,r as Lt,s as Dt,t as It}from"./select-Ch6m1k2G.js";import{a as qt,c as Nt,l as Ft,n as Pt,r as Bt,t as Et}from"./dialog-DCGn-_Ip.js";import{c as ae,l as Mt,s as Ot,u as le}from"./popover-JNsSzROl.js";import{n as Ut}from"./ImperativeModal-b5nJtLod.js";import{t as de}from"./links-CoGM-Inf.js";import{n as Kt}from"./useAsyncData-a8f2Q_nk.js";import{o as Ht}from"./focus-BjrLbkmu.js";import{t as zt}from"./request-registry-C4HbFS_V.js";import{n as Wt,t as Gt}from"./write-secret-modal-s2JWi0j9.js";var Vt=Xe(function(t,e,s){t[s?0:1].push(e)},function(){return[[],[]]}),J=st("hard-drive",[["line",{x1:"22",x2:"2",y1:"12",y2:"12",key:"1y58io"}],["path",{d:"M5.45 5.11 2 12v6a2 2 0 0 0 2 2h16a2 2 0 0 0 2-2v-6l-3.45-6.89A2 2 0 0 0 16.76 4H7.24a2 2 0 0 0-1.79 1.11z",key:"oot6mr"}],["line",{x1:"6",x2:"6.01",y1:"16",y2:"16",key:"sgf278"}],["line",{x1:"10",x2:"10.01",y1:"16",y2:"16",key:"1l4acy"}]]),Y="env:";function Ir(t){return t}function M(t){return typeof t=="string"?t.startsWith(Y):!1}function X(t){return`${Y}${t}`}function pe(t){return t.replace(Y,"")}var Zt=P(),B=oe(Ze(),1),n=oe(tt(),1),he=(0,B.createContext)({providerNames:[],secretKeys:[],loading:!1,error:void 0,refreshSecrets:et.NOOP});const Qt=()=>(0,B.use)(he),Jt=t=>{let e=(0,Zt.c)(14),{children:s}=t,o;e[0]===Symbol.for("react.memo_cache_sentinel")?(o=[],e[0]=o):o=e[0];let{data:r,isPending:l,error:a,refetch:c}=Kt(ss,o),d;e[1]===(r==null?void 0:r.secretKeys)?d=e[2]:(d=(r==null?void 0:r.secretKeys)||[],e[1]=r==null?void 0:r.secretKeys,e[2]=d);let m;e[3]===(r==null?void 0:r.providerNames)?m=e[4]:(m=(r==null?void 0:r.providerNames)||[],e[3]=r==null?void 0:r.providerNames,e[4]=m);let p;e[5]!==a||e[6]!==l||e[7]!==c||e[8]!==d||e[9]!==m?(p={secretKeys:d,providerNames:m,loading:l,error:a,refreshSecrets:c},e[5]=a,e[6]=l,e[7]=c,e[8]=d,e[9]=m,e[10]=p):p=e[10];let f;return e[11]!==s||e[12]!==p?(f=(0,n.jsx)(he,{value:p,children:s}),e[11]=s,e[12]=p,e[13]=f):f=e[13],f},Yt={isMatch:t=>{if(t instanceof ce||t instanceof Ye){let{optionRegex:e}=i.parse(t.description||"");return!!e}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:o,providerNames:r,refreshSecrets:l}=Qt(),{openModal:a,closeModal:c}=Ut(),{label:d,description:m,optionRegex:p=""}=i.parse(t.description||""),[f,h]=Vt(o,b=>new RegExp(p,"i").test(b));return(0,n.jsx)($t,{control:e.control,name:s,render:({field:b})=>(0,n.jsxs)(gt,{children:[(0,n.jsx)(yt,{children:d}),(0,n.jsx)(ft,{children:m}),(0,n.jsx)(kt,{children:(0,n.jsxs)("div",{className:"flex gap-2",children:[t instanceof ce?(0,n.jsx)(bt,{...b,value:b.value,onChange:b.onChange,className:W("flex-1")}):(0,n.jsx)(ut,{...b,value:b.value,onChange:b.onChange,className:"flex-1"}),(0,n.jsxs)(it,{children:[(0,n.jsx)(nt,{asChild:!0,children:(0,n.jsx)(Z,{variant:"outline",size:"icon",className:W(M(b.value)&&"bg-accent"),children:(0,n.jsx)(mt,{className:"h-3 w-3"})})}),(0,n.jsxs)(ct,{align:"end",className:"max-h-60 overflow-y-auto",children:[(0,n.jsxs)(Q,{onSelect:()=>{a((0,n.jsx)(Gt,{providerNames:r,onSuccess:w=>{l(),b.onChange(X(w)),c()},onClose:c}))},children:[(0,n.jsx)(rt,{className:"mr-2 h-3.5 w-3.5"}),"Create a new secret"]}),f.length>0&&(0,n.jsxs)(n.Fragment,{children:[(0,n.jsx)(ie,{}),(0,n.jsx)(ot,{children:"Recommended"})]}),f.map(w=>(0,n.jsx)(Q,{onSelect:()=>b.onChange(X(w)),children:w},w)),h.length>0&&(0,n.jsx)(ie,{}),h.map(w=>(0,n.jsx)(Q,{onSelect:()=>b.onChange(X(w)),children:w},w))]})]})]})}),(0,n.jsx)(xt,{})]})})}};function Xt(t){return t.provider!=="env"}function es(t){return t.name}function ts(t){return t.keys}async function ss(){let t=await zt.request({}),e=Wt(t.secrets).filter(Xt).map(es);return{secretKeys:t.secrets.flatMap(ts).sort(),providerNames:e}}var O=P(),rs=[Yt];const ee=t=>{let e=(0,O.c)(2),{children:s}=t,o;return e[0]===s?o=e[1]:(o=(0,n.jsx)("div",{className:"grid grid-cols-2 md:grid-cols-3 gap-4",children:s}),e[0]=s,e[1]=o),o},te=t=>{let e=(0,O.c)(9),{name:s,color:o,icon:r,onSelect:l}=t,a;e[0]===o?a=e[1]:(a={backgroundColor:o},e[0]=o,e[1]=a);let c;e[2]===s?c=e[3]:(c=(0,n.jsx)("span",{className:"text-white font-medium text-lg",children:s}),e[2]=s,e[3]=c);let d;return e[4]!==r||e[5]!==l||e[6]!==a||e[7]!==c?(d=(0,n.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:a,onClick:l,children:[r,c]}),e[4]=r,e[5]=l,e[6]=a,e[7]=c,e[8]=d):d=e[8],d},os=t=>{let e=(0,O.c)(25),{onBack:s,isValid:o,libraries:r,preferredLibrary:l,onLibraryChange:a,displayNames:c,libraryLabel:d}=t,m=d===void 0?"Preferred library":d,p;e[0]===s?p=e[1]:(p=(0,n.jsx)(Z,{type:"button",variant:"outline",onClick:s,children:"Back"}),e[0]=s,e[1]=p);let f=!o,h;e[2]===f?h=e[3]:(h=(0,n.jsx)(Z,{type:"submit",disabled:f,children:"Add"}),e[2]=f,e[3]=h);let b;e[4]!==p||e[5]!==h?(b=(0,n.jsxs)("div",{className:"flex gap-2",children:[p,h]}),e[4]=p,e[5]=h,e[6]=b):b=e[6];let w;e[7]===Symbol.for("react.memo_cache_sentinel")?(w=(0,n.jsx)(Dt,{children:(0,n.jsx)(At,{placeholder:"Select a library"})}),e[7]=w):w=e[7];let x;e[8]===m?x=e[9]:(x=(0,n.jsxs)("div",{className:"flex flex-col gap-1 items-end",children:[w,(0,n.jsx)("span",{className:"text-xs text-muted-foreground",children:m})]}),e[8]=m,e[9]=x);let R;if(e[10]!==c||e[11]!==r){let v;e[13]===c?v=e[14]:(v=C=>(0,n.jsx)(Rt,{value:C,children:c[C]??C},C),e[13]=c,e[14]=v),R=r.map(v),e[10]=c,e[11]=r,e[12]=R}else R=e[12];let $;e[15]===R?$=e[16]:($=(0,n.jsx)(Tt,{children:(0,n.jsx)(Lt,{children:R})}),e[15]=R,e[16]=$);let k;e[17]!==a||e[18]!==l||e[19]!==x||e[20]!==$?(k=(0,n.jsx)("div",{children:(0,n.jsxs)(It,{value:l,onValueChange:a,children:[x,$]})}),e[17]=a,e[18]=l,e[19]=x,e[20]=$,e[21]=k):k=e[21];let T;return e[22]!==k||e[23]!==b?(T=(0,n.jsxs)("div",{className:"flex gap-2 justify-between",children:[b,k]}),e[22]=k,e[23]=b,e[24]=T):T=e[24],T};function ns(){let t=(0,O.c)(3),{createNewCell:e}=Qe(),s=Ht(),o;return t[0]!==e||t[1]!==s?(o=r=>{e({code:r,before:!1,cellId:s??"__end__",skipIfCodeExists:!0})},t[0]=e,t[1]=s,t[2]=o):o=t[2],o}const me=t=>{let e=(0,O.c)(30),{schema:s,libraries:o,preferredLibrary:r,displayNames:l,libraryLabel:a,generateCode:c,onSubmit:d,onBack:m}=t,p;e[0]===s?p=e[1]:(p=pt(s),e[0]=s,e[1]=p);let f=p,h;e[2]===s?h=e[3]:(h=_t(s),e[2]=s,e[3]=h);let b;e[4]!==f||e[5]!==h?(b={defaultValues:f,resolver:h,reValidateMode:"onChange"},e[4]=f,e[5]=h,e[6]=b):b=e[6];let w=jt(b),[x,R]=B.useState(r),$=ns(),k;e[7]!==c||e[8]!==$||e[9]!==d||e[10]!==x?(k=We=>{$(c(We,x)),d()},e[7]=c,e[8]=$,e[9]=d,e[10]=x,e[11]=k):k=e[11];let T=k,v;e[12]!==w||e[13]!==T?(v=w.handleSubmit(T),e[12]=w,e[13]=T,e[14]=v):v=e[14];let C;e[15]===Symbol.for("react.memo_cache_sentinel")?(C=(0,n.jsx)(wt,{}),e[15]=C):C=e[15];let N;e[16]!==w||e[17]!==s?(N=(0,n.jsx)(Jt,{children:(0,n.jsx)(ht,{schema:s,form:w,renderers:rs,children:C})}),e[16]=w,e[17]=s,e[18]=N):N=e[18];let F;e[19]!==l||e[20]!==w.formState.isValid||e[21]!==o||e[22]!==a||e[23]!==m||e[24]!==x?(F=(0,n.jsx)(os,{onBack:m,isValid:w.formState.isValid,libraries:o,preferredLibrary:x,onLibraryChange:R,displayNames:l,libraryLabel:a}),e[19]=l,e[20]=w.formState.isValid,e[21]=o,e[22]=a,e[23]=m,e[24]=x,e[25]=F):F=e[25];let U;return e[26]!==v||e[27]!==N||e[28]!==F?(U=(0,n.jsxs)("form",{onSubmit:v,className:"space-y-4",children:[N,F]}),e[26]=v,e[27]=N,e[28]=F,e[29]=U):U=e[29],U};function L(){return u().optional().describe(i.of({label:"Password",inputType:"password",placeholder:"password",optionRegex:".*password.*"}))}function se(t,e){let s=u();return s=e?s.nonempty():s.optional(),s=s.describe(i.of({label:t||"Token",inputType:"password",placeholder:"token",optionRegex:".*token.*"})),s}function G(){return u().optional().describe(i.of({label:"Warehouse Name",placeholder:"warehouse",optionRegex:".*warehouse.*"}))}function re(t,e){let s=u();return s=e?s.nonempty():s.optional(),s.describe(i.of({label:t||"URI",optionRegex:".*uri.*"}))}function A(t){return u().nonempty().describe(i.of({label:t||"Host",placeholder:"localhost",optionRegex:".*host.*"}))}function j(){return u().describe(i.of({label:"Database",placeholder:"db name",optionRegex:".*database.*"}))}function cs(){return u().describe(i.of({label:"Schema",placeholder:"schema name",optionRegex:".*schema.*"}))}function D(){return u().nonempty().describe(i.of({label:"Username",placeholder:"username",optionRegex:".*username.*"}))}function I(t){let e=Je().describe(i.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 ue(){return q().default(!1).describe(i.of({label:"Read Only"}))}const be=y({type:_("postgres"),host:A(),port:I(5432).optional(),database:j(),username:D(),password:L(),ssl:q().default(!1).describe(i.of({label:"Use SSL"}))}).describe(i.of({direction:"two-columns"})),fe=y({type:_("mysql"),host:A(),port:I(3306),database:j(),username:D(),password:L(),ssl:q().default(!1).describe(i.of({label:"Use SSL"}))}).describe(i.of({direction:"two-columns"})),ge=y({type:_("sqlite"),database:j().describe(i.of({label:"Database Path"}))}).describe(i.of({direction:"two-columns"})),ye=y({type:_("duckdb"),database:j().describe(i.of({label:"Database Path"})),read_only:ue()}).describe(i.of({direction:"two-columns"})),_e=y({type:_("motherduck"),database:j().default("my_db").describe(i.of({label:"Database Name"})),token:se()}).describe(i.of({direction:"two-columns"})),we=y({type:_("snowflake"),account:u().nonempty().describe(i.of({label:"Account",optionRegex:".*snowflake.*"})),warehouse:u().optional().describe(i.of({label:"Warehouse",optionRegex:".*snowflake.*"})),database:j(),schema:u().optional().describe(i.of({label:"Schema",optionRegex:".*snowflake.*"})),username:D(),password:L(),role:u().optional().describe(i.of({label:"Role"}))}).describe(i.of({direction:"two-columns"})),Se=y({type:_("bigquery"),project:u().nonempty().describe(i.of({label:"Project ID",optionRegex:".*bigquery.*"})),dataset:u().nonempty().describe(i.of({label:"Dataset",optionRegex:".*bigquery.*"})),credentials_json:u().describe(i.of({label:"Credentials JSON",inputType:"textarea"}))}).describe(i.of({direction:"two-columns"})),xe=y({type:_("clickhouse_connect"),host:A(),port:I(8123).optional(),username:D(),password:L(),secure:q().default(!1).describe(i.of({label:"Use HTTPs"})),proxy_path:u().optional().describe(i.of({label:"Proxy Path",placeholder:"/clickhouse"}))}).describe(i.of({direction:"two-columns"})),ke=y({type:_("timeplus"),host:A().default("localhost"),port:I(8123).optional(),username:D().default("default"),password:L().default("")}).describe(i.of({direction:"two-columns"})),ve=y({type:_("chdb"),database:j().describe(i.of({label:"Database Path"})),read_only:ue()}).describe(i.of({direction:"two-columns"})),$e=y({type:_("trino"),host:A(),port:I(8080),database:j(),schema:cs().optional(),username:D(),password:L(),async_support:q().default(!1).describe(i.of({label:"Async Support"}))}).describe(i.of({direction:"two-columns"})),je=y({type:_("iceberg"),name:u().describe(i.of({label:"Catalog Name"})),catalog:H("type",[y({type:_("REST"),warehouse:G(),uri:u().optional().describe(i.of({label:"URI",placeholder:"https://",optionRegex:".*uri.*"})),token:se()}),y({type:_("SQL"),warehouse:G(),uri:u().optional().describe(i.of({label:"URI",placeholder:"jdbc:iceberg://host:port/database",optionRegex:".*uri.*"}))}),y({type:_("Hive"),warehouse:G(),uri:re()}),y({type:_("Glue"),warehouse:G(),uri:re()}),y({type:_("DynamoDB"),"dynamodb.profile-name":u().optional().describe(i.of({label:"Profile Name"})),"dynamodb.region":u().optional().describe(i.of({label:"Region"})),"dynamodb.access-key-id":u().optional().describe(i.of({label:"Access Key ID"})),"dynamodb.secret-access-key":u().optional().describe(i.of({label:"Secret Access Key",inputType:"password"})),"dynamodb.session-token":u().optional().describe(i.of({label:"Session Token",inputType:"password"}))})]).default({type:"REST",token:void 0}).describe(i.of({special:"tabs"}))}),Ce=y({type:_("datafusion"),sessionContext:q().optional().describe(i.of({label:"Use Session Context"}))}),Ae=y({type:_("pyspark"),host:A().optional(),port:I().optional()}),Re=y({type:_("redshift"),host:A(),port:I(5439),connectionType:H("type",[y({type:_("IAM credentials"),region:u().describe(i.of({label:"Region"})),aws_access_key_id:u().nonempty().describe(i.of({label:"AWS Access Key ID",inputType:"password",optionRegex:".*aws_access_key_id.*"})),aws_secret_access_key:u().nonempty().describe(i.of({label:"AWS Secret Access Key",inputType:"password",optionRegex:".*aws_secret_access_key.*"})),aws_session_token:u().optional().describe(i.of({label:"AWS Session Token",inputType:"password",optionRegex:".*aws_session_token.*"}))}),y({type:_("DB credentials"),user:D(),password:L()})]).default({type:"IAM credentials",aws_access_key_id:"",aws_secret_access_key:"",region:""}),database:j()}).describe(i.of({direction:"two-columns"})),Te=y({type:_("databricks"),access_token:se("Access Token",!0),server_hostname:A("Server Hostname"),http_path:re("HTTP Path",!0),catalog:u().optional().describe(i.of({label:"Catalog"})),schema:u().optional().describe(i.of({label:"Schema"}))}).describe(i.of({direction:"two-columns"})),Le=y({type:_("supabase"),host:A(),port:I(5432).optional(),database:j(),username:D(),password:L(),disable_client_pooling:q().default(!1).describe(i.of({label:"Disable Client-Side Pooling"}))}).describe(i.of({direction:"two-columns"})),is=H("type",[be,fe,ge,ye,_e,we,Se,xe,ke,ve,$e,je,Ce,Ae,Re,Te,Le]),De={sqlmodel:"SQLModel",sqlalchemy:"SQLAlchemy",duckdb:"DuckDB",clickhouse_connect:"ClickHouse Connect",chdb:"chDB",pyiceberg:"PyIceberg",ibis:"Ibis",motherduck:"MotherDuck",redshift:"Redshift",databricks:"Databricks"};var S=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}},as=t=>`_${t}`,ls=class{constructor(){K(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e,s){if(t=as(t),M(e)){let o=pe(e),r=s?`os.environ.get("${o}", "${s}")`:`os.environ.get("${o}")`;return this.secrets[t]=r,t}if(s!=null){let o=`os.environ.get("${e}", "${s}")`;return this.secrets[t]=o,t}return typeof e=="number"||typeof e=="number"?`${e}`:typeof e=="boolean"?E(e):e?`"${e}"`:""}printInFString(t,e,s){if(e===void 0)return"";if(typeof e=="number")return`${e}`;if(typeof e=="boolean")return E(e);let o=this.print(t,e,s);return o.startsWith('"')&&o.endsWith('"')?o.slice(1,-1):`{${o}}`}printPassword(t,e,s,o){let r=s?this.printInFString.bind(this):this.print.bind(this);return M(t)?r(o||"password",t):r(o||"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(`
-`)}},ds=class extends S{generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'sslmode': 'require'}":"",e=this.secrets.printPassword(this.connection.password,"POSTGRES_PASSWORD",!0);return g(`
-      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})
-    `)}},ps=class extends S{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 g(`
-      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})
-    `)}},hs=class extends S{generateImports(){return[]}generateConnectionCode(){let t=this.connection.database?this.secrets.printInFString("database",this.connection.database):"";return g(`
-      ${t.startsWith("{")&&t.endsWith("}")?`DATABASE_URL = f"sqlite:///${t}"`:`DATABASE_URL = "sqlite:///${t}"`}
-      engine = ${this.orm}.create_engine(DATABASE_URL)
-    `)}},ms=class extends S{generateImports(){return["from snowflake.sqlalchemy import URL"]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"SNOWFLAKE_PASSWORD",!1),e={account:this.secrets.print("account",this.connection.account),user:this.secrets.print("user",this.connection.username),database:this.secrets.print("database",this.connection.database),warehouse:this.connection.warehouse?this.secrets.print("warehouse",this.connection.warehouse):void 0,schema:this.connection.schema?this.secrets.print("schema",this.connection.schema):void 0,role:this.connection.role?this.secrets.print("role",this.connection.role):void 0,password:t};return g(`
-      engine = ${this.orm}.create_engine(
-        URL(
-${V(e,s=>`          ${s}`)},
-        )
-      )
-    `)}},us=class extends S{generateImports(){return["import json"]}generateConnectionCode(){let t=this.secrets.printInFString("project",this.connection.project),e=this.secrets.printInFString("dataset",this.connection.dataset);return g(`
-      credentials = json.loads("""${this.connection.credentials_json}""")
-      engine = ${this.orm}.create_engine(f"bigquery://${t}/${e}", credentials_info=credentials)
-    `)}},bs=class extends S{generateImports(){return[]}generateConnectionCode(){return g(`
-      DATABASE_URL = "${this.secrets.printInFString("database",this.connection.database||":memory:")}"
-      engine = ${this.orm}.connect(DATABASE_URL, read_only=${E(this.connection.read_only)})
-    `)}},fs=class extends S{generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printInFString("database",this.connection.database);return this.connection.token?g(`
-      conn = duckdb.connect("md:${t}", config={"motherduck_token": ${this.secrets.printPassword(this.connection.token,"MOTHERDUCK_TOKEN",!1)}})
-    `):g(`
-        conn = duckdb.connect("md:${t}")
-        `)}},gs=class extends S{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 g(`
-      engine = ${this.orm}.get_client(
-${V(e,s=>`        ${s}`)},
-      )
-    `)}},ys=class extends S{generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"TIMEPLUS_PASSWORD",!0);return g(`
-      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)
-    `)}},_s=class extends S{generateImports(){return["import chdb"]}generateConnectionCode(){let t=this.secrets.print("database",this.connection.database)||'""';return g(`
-      engine = ${this.orm}.connect(${t}, read_only=${E(this.connection.read_only)})
-    `)}},ws=class extends S{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),o=this.secrets.printInFString("host",this.connection.host),r=this.secrets.printInFString("port",this.connection.port),l=this.secrets.printInFString("database",this.connection.database),a=this.secrets.printPassword(this.connection.password,"TRINO_PASSWORD",!0);return g(`
-      engine = ${this.orm}.create_engine(f"${t}://${s}:${a}@${o}:${r}/${l}${e}")
-    `)}},Ss=class extends S{generateImports(){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:z(this.connection.catalog)}}generateConnectionCode(){let t={...this.connection.catalog};t=Object.fromEntries(Object.entries(t).filter(([o,r])=>r!=null&&r!==""&&o!=="type"));for(let[o,r]of Object.entries(t))M(r)?t[o]=this.secrets.print(o,r):typeof r=="string"&&(t[o]=`"${r}"`);let e=Rs(t,o=>`              ${o}`),s=`"${this.connection.name}"`;switch(this.connection.catalog.type){case"REST":return g(`
-          catalog = RestCatalog(
-            ${s},
-            **{
-${e}
-            },
-          )
-        `);case"SQL":return g(`
-          catalog = SqlCatalog(
-            ${s},
-            **{
-${e}
-            },
-          )
-        `);case"Hive":return g(`
-          catalog = HiveCatalog(
-            ${s},
-            **{
-${e}
-            },
-          )
-        `);case"Glue":return g(`
-          catalog = GlueCatalog(
-            ${s},
-            **{
-${e}
-            },
-          )
-        `);case"DynamoDB":return g(`
-          catalog = DynamoDBCatalog(
-            ${s},
-            **{
-${e}
-            },
-          )
-        `);default:z(this.connection.catalog)}}},xs=class extends S{generateImports(){return["from datafusion import SessionContext"]}generateConnectionCode(){return this.connection.sessionContext?g(`
-        ctx = SessionContext()
-        # Sample table
-        _ = ctx.from_pydict({"a": [1, 2, 3]}, "my_table")
-
-        con = ibis.datafusion.connect(ctx)
-      `):g(`
-      con = ibis.datafusion.connect()
-    `)}},ks=class extends S{generateImports(){return["from pyspark.sql import SparkSession"]}generateConnectionCode(){return this.connection.host||this.connection.port?g(`
-        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)
-      `):g(`
-      con = ibis.pyspark.connect()
-    `)}},vs=class extends S{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 l=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),c=this.connection.connectionType.aws_session_token?this.secrets.print("aws_session_token",this.connection.connectionType.aws_session_token):void 0;return g(`
-        con = redshift_connector.connect(
-${V({iam:!0,host:t,port:e,region:`"${this.connection.connectionType.region}"`,database:s,access_key_id:l,secret_access_key:a,...c&&{session_token:c}},d=>`          ${d}`)},
-        )
-      `)}let o=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 g(`
-      con = redshift_connector.connect(
-${V({host:t,port:e,database:s,...o&&{user:o},...r&&{password:r}},l=>`          ${l}`)},
-      )
-    `)}},$s=class extends S{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"),o=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,l=this.connection.schema?this.secrets.printInFString("schema",this.connection.schema):void 0,a=`databricks://token:${e}@${s}?http_path=${o}`;return r&&(a+=`&catalog=${r}`),l&&(a+=`&schema=${l}`),this.orm==="ibis"?g(`
-        engine = ibis.databricks.connect(
-          server_hostname=${s},
-          http_path=${o},${r?`
-          catalog=${r},`:""}${l?`
-          schema=${l},`:""}
-          access_token=${e}
-        )
-      `):g(`
-      DATABASE_URL = f"${a}"
-      engine = ${this.orm}.create_engine(DATABASE_URL)
-      `)}},js=class extends S{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),o=this.secrets.printInFString("port",this.connection.port),r=this.secrets.printInFString("database",this.connection.database),l=this.connection.disable_client_pooling?", poolclass=NullPool":"";return g(`
-      DATABASE_URL = f"postgresql+psycopg2://${e}:${t}@${s}:${o}/${r}?sslmode=require"
-      engine = ${this.orm}.create_engine(DATABASE_URL${l})
-    `)}},Cs=class{constructor(){K(this,"secrets",new ls)}createGenerator(t,e){switch(t.type){case"postgres":return new ds(t,e,this.secrets);case"mysql":return new ps(t,e,this.secrets);case"sqlite":return new hs(t,e,this.secrets);case"snowflake":return new ms(t,e,this.secrets);case"bigquery":return new us(t,e,this.secrets);case"duckdb":return new bs(t,e,this.secrets);case"motherduck":return new fs(t,e,this.secrets);case"clickhouse_connect":return new gs(t,e,this.secrets);case"timeplus":return new ys(t,e,this.secrets);case"chdb":return new _s(t,e,this.secrets);case"trino":return new ws(t,e,this.secrets);case"iceberg":return new Ss(t,e,this.secrets);case"datafusion":return new xs(t,e,this.secrets);case"pyspark":return new ks(t,e,this.secrets);case"redshift":return new vs(t,e,this.secrets);case"databricks":return new $s(t,e,this.secrets);case"supabase":return new js(t,e,this.secrets);default:z(t)}}};function As(t,e){if(!(e in De))throw Error(`Unsupported library: ${e}`);is.parse(t);let s=new Cs,o=s.createGenerator(t,e),r=o.generateConnectionCode(),l=s.secrets,a=[...new Set([...l.imports,...o.imports])].sort();a.push("");let c=l.formatSecrets();return c&&a.push(c),a.push(r.trim()),a.join(`
-`)}function E(t){return t.toString().charAt(0).toUpperCase()+t.toString().slice(1)}function V(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,o])=>e(typeof o=="boolean"?`${s}=${E(o)}`:`${s}=${o}`)).join(`,
-`)}function Rs(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,o])=>{let r=`"${s}"`;return e(typeof o=="boolean"?`${r}: ${E(o)}`:`${r}: ${o}`)}).join(`,
-`)}var Ie=P(),qe=[{name:"PostgreSQL",schema:be,color:"#336791",logo:"postgres",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"MySQL",schema:fe,color:"#00758F",logo:"mysql",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"SQLite",schema:ge,color:"#003B57",logo:"sqlite",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DuckDB",schema:ye,color:"#FFD700",logo:"duckdb",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"MotherDuck",schema:_e,color:"#ff9538",logo:"motherduck",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"Snowflake",schema:we,color:"#29B5E8",logo:"snowflake",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse",schema:xe,color:"#2C2C1D",logo:"clickhouse",connectionLibraries:{libraries:["clickhouse_connect"],preferred:"clickhouse_connect"}},{name:"Timeplus",schema:ke,color:"#B83280",logo:"timeplus",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"BigQuery",schema:Se,color:"#4285F4",logo:"bigquery",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse Embedded",schema:ve,color:"#f2b611",logo:"clickhouse",connectionLibraries:{libraries:["chdb"],preferred:"chdb"}},{name:"Trino",schema:$e,color:"#d466b6",logo:"trino",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DataFusion",schema:Ce,color:"#202A37",logo:"datafusion",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"PySpark",schema:Ae,color:"#1C5162",logo:"pyspark",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"Redshift",schema:Re,color:"#522BAE",logo:"redshift",connectionLibraries:{libraries:["redshift"],preferred:"redshift"}},{name:"Databricks",schema:Te,color:"#c41e0c",logo:"databricks",connectionLibraries:{libraries:["sqlalchemy","sqlmodel","ibis"],preferred:"sqlalchemy"}},{name:"Supabase",schema:Le,color:"#238F5F",logo:"supabase",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}}],Ne=[{name:"Iceberg",schema:je,color:"#000000",logo:"iceberg",connectionLibraries:{libraries:["pyiceberg"],preferred:"pyiceberg"}}],Ts=[...qe,...Ne],Ls=t=>{let e=(0,Ie.c)(12),{onSelect:s}=t,o;e[0]===s?o=e[1]:(o=qe.map(m=>{let{name:p,schema:f,color:h,logo:b}=m;return(0,n.jsx)(te,{name:p,color:h,icon:(0,n.jsx)(ne,{name:b,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(f)},p)}),e[0]=s,e[1]=o);let r;e[2]===o?r=e[3]:(r=(0,n.jsx)(ee,{children:o}),e[2]=o,e[3]=r);let l;e[4]===Symbol.for("react.memo_cache_sentinel")?(l=(0,n.jsxs)("h4",{className:"font-semibold text-muted-foreground text-lg flex items-center gap-4 my-2",children:["Data Catalogs",(0,n.jsx)("hr",{className:"flex-1"})]}),e[4]=l):l=e[4];let a;e[5]===s?a=e[6]:(a=Ne.map(m=>{let{name:p,schema:f,color:h,logo:b}=m;return(0,n.jsx)(te,{name:p,color:h,icon:(0,n.jsx)(ne,{name:b,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(f)},p)}),e[5]=s,e[6]=a);let c;e[7]===a?c=e[8]:(c=(0,n.jsx)(ee,{children:a}),e[7]=a,e[8]=c);let d;return e[9]!==r||e[10]!==c?(d=(0,n.jsxs)(n.Fragment,{children:[r,l,c]}),e[9]=r,e[10]=c,e[11]=d):d=e[11],d};const Ds=t=>{let e=(0,Ie.c)(13),{onSubmit:s,header:o}=t,[r,l]=(0,B.useState)(null);if(!r){let h;e[0]===Symbol.for("react.memo_cache_sentinel")?(h=(0,n.jsx)("div",{children:(0,n.jsx)(Ls,{onSelect:l})}),e[0]=h):h=e[0];let b;return e[1]===o?b=e[2]:(b=(0,n.jsxs)(n.Fragment,{children:[o,h]}),e[1]=o,e[2]=b),b}let a;e[3]===r?a=e[4]:(a=Ts.find(h=>h.schema===r),e[3]=r,e[4]=a);let c=a==null?void 0:a.connectionLibraries,d;e[5]===(c==null?void 0:c.libraries)?d=e[6]:(d=(c==null?void 0:c.libraries)??[],e[5]=c==null?void 0:c.libraries,e[6]=d);let m=(c==null?void 0:c.preferred)??"sqlalchemy",p;e[7]===Symbol.for("react.memo_cache_sentinel")?(p=()=>l(null),e[7]=p):p=e[7];let f;return e[8]!==s||e[9]!==r||e[10]!==d||e[11]!==m?(f=(0,n.jsx)(me,{schema:r,libraries:d,preferredLibrary:m,displayNames:De,libraryLabel:"Preferred connection library",generateCode:Is,onSubmit:s,onBack:p}),e[8]=s,e[9]=r,e[10]=d,e[11]=m,e[12]=f):f=e[12],f};function Is(t,e){return As(t,e)}var qs="data:image/svg+xml,%3csvg%20xmlns='http://www.w3.org/2000/svg'%20xml:space='preserve'%20id='Layer_1'%20x='0'%20y='0'%20version='1.1'%20viewBox='0.02%20102.6%20511.9%20306.4'%3e%3cstyle%3e.st1{fill-rule:evenodd;clip-rule:evenodd;fill:%23f90}%3c/style%3e%3cpath%20d='M144.3%20214.1c0%206.3.7%2011.4%201.9%2015.2%201.4%203.7%203.1%207.8%205.4%2012.3.9%201.4%201.2%202.7%201.2%203.9%200%201.7-1%203.4-3.2%205.1l-10.7%207.2c-1.5%201-3.1%201.5-4.4%201.5-1.7%200-3.4-.9-5.1-2.4-2.4-2.6-4.4-5.3-6.1-8-1.7-2.9-3.4-6.1-5.3-10-13.3%2015.7-30%2023.5-50.1%2023.5-14.3%200-25.7-4.1-34.1-12.3-8.3-8.2-12.6-19.1-12.6-32.7%200-14.5%205.1-26.2%2015.5-35.1S60.8%20169%2078.4%20169c5.8%200%2011.7.5%2018.1%201.4s12.8%202.2%2019.6%203.7v-12.4c0-12.9-2.7-22-8-27.2-5.4-5.3-14.6-7.8-27.8-7.8-6%200-12.1.7-18.4%202.2s-12.4%203.4-18.4%205.8c-2.7%201.2-4.8%201.9-6%202.2s-2%20.5-2.7.5c-2.4%200-3.6-1.7-3.6-5.3v-8.3c0-2.7.3-4.8%201.2-6s2.4-2.4%204.8-3.6c6-3.1%2013.1-5.6%2021.5-7.7%208.3-2.2%2017.2-3.2%2026.6-3.2%2020.3%200%2035.1%204.6%2044.6%2013.8%209.4%209.2%2014.1%2023.2%2014.1%2041.9v55.2h.3zM75.2%20240c5.6%200%2011.4-1%2017.5-3.1%206.1-2%2011.6-5.8%2016.2-10.9%202.7-3.2%204.8-6.8%205.8-10.9s1.7-9%201.7-14.8v-7.2c-4.9-1.2-10.2-2.2-15.7-2.9-5.4-.7-10.7-1-16-1-11.4%200-19.8%202.2-25.4%206.8S51%20207.1%2051%20215.6c0%208%202%2014%206.3%2018.1%204.1%204.2%2010%206.3%2017.9%206.3m136.7%2018.4c-3.1%200-5.1-.5-6.5-1.7-1.4-1-2.6-3.4-3.6-6.6l-40-131.6c-1-3.4-1.5-5.6-1.5-6.8%200-2.7%201.4-4.3%204.1-4.3h16.7c3.2%200%205.4.5%206.6%201.7%201.4%201%202.4%203.4%203.4%206.6l28.6%20112.7%2026.6-112.7c.9-3.4%201.9-5.6%203.2-6.6%201.4-1%203.7-1.7%206.8-1.7H270c3.2%200%205.4.5%206.8%201.7%201.4%201%202.6%203.4%203.2%206.6l26.9%20114.1%2029.5-114.1c1-3.4%202.2-5.6%203.4-6.6%201.4-1%203.6-1.7%206.6-1.7h15.8c2.7%200%204.3%201.4%204.3%204.3%200%20.9-.2%201.7-.3%202.7-.2%201-.5%202.4-1.2%204.3l-41%20131.6q-1.5%205.1-3.6%206.6c-1.4%201-3.6%201.7-6.5%201.7h-14.6c-3.2%200-5.4-.5-6.8-1.7s-2.6-3.4-3.2-6.8l-26.4-109.8L236.7%20250c-.9%203.4-1.9%205.6-3.2%206.8-1.4%201.2-3.7%201.7-6.8%201.7h-14.8zm218.8%204.6c-8.9%200-17.7-1-26.2-3.1-8.5-2-15.2-4.3-19.6-6.8-2.7-1.5-4.6-3.2-5.3-4.8s-1-3.2-1-4.8v-8.7c0-3.6%201.4-5.3%203.9-5.3%201%200%202%20.2%203.1.5%201%20.3%202.6%201%204.3%201.7%205.8%202.6%2012.1%204.6%2018.7%206%206.8%201.4%2013.5%202%2020.3%202%2010.7%200%2019.1-1.9%2024.9-5.6s8.9-9.2%208.9-16.2c0-4.8-1.5-8.7-4.6-11.9s-8.9-6.1-17.2-8.9l-24.7-7.7c-12.4-3.9-21.6-9.7-27.2-17.4-5.6-7.5-8.5-15.8-8.5-24.7%200-7.2%201.5-13.5%204.6-18.9s7.2-10.2%2012.3-14c5.1-3.9%2010.9-6.8%2017.7-8.9%206.8-2%2014-2.9%2021.5-2.9%203.7%200%207.7.2%2011.4.7%203.9.5%207.5%201.2%2011.1%201.9%203.4.9%206.6%201.7%209.7%202.7s5.4%202%207.2%203.1c2.4%201.4%204.1%202.7%205.1%204.3%201%201.4%201.5%203.2%201.5%205.6v8c0%203.6-1.4%205.4-3.9%205.4-1.4%200-3.6-.7-6.5-2q-14.55-6.6-32.7-6.6c-9.7%200-17.4%201.5-22.6%204.8s-8%208.2-8%2015.2c0%204.8%201.7%208.9%205.1%2012.1s9.7%206.5%2018.7%209.4l24.2%207.7c12.3%203.9%2021.1%209.4%2026.4%2016.3s7.8%2015%207.8%2023.8c0%207.3-1.5%2014-4.4%2019.8-3.1%205.8-7.2%2010.9-12.4%2015-5.3%204.3-11.6%207.3-18.9%209.5-8%202.5-16%203.7-24.7%203.7'%20style='fill:%23fff'/%3e%3cpath%20fill='%23fff'%20d='M462.9%20345.7c-56%2041.4-137.4%2063.3-207.4%2063.3-98.1%200-186.5-36.3-253.2-96.6-5.3-4.8-.5-11.2%205.8-7.5%2072.2%2041.9%20161.3%2067.3%20253.4%2067.3%2062.2%200%20130.4-12.9%20193.3-39.5%209.3-4.2%2017.3%206.2%208.1%2013'%20class='st1'/%3e%3cpath%20fill='%23fff'%20d='M486.2%20319.2c-7.2-9.2-47.3-4.4-65.6-2.2-5.4.7-6.3-4.1-1.4-7.7%2032-22.5%2084.6-16%2090.8-8.5%206.1%207.7-1.7%2060.3-31.7%2085.5-4.6%203.9-9%201.9-7-3.2%206.9-16.9%2022.1-54.9%2014.9-63.9'%20class='st1'/%3e%3c/svg%3e",Ns="data:image/svg+xml,%3csvg%20xmlns='http://www.w3.org/2000/svg'%20viewBox='0%200%20128%20128'%3e%3cpath%20fill='%23FFF'%20d='m115.679%2069.288-15.591-8.94-2.689-1.163-63.781.436v32.381h82.061z'/%3e%3cpath%20fill='%23F38020'%20d='M87.295%2089.022c.763-2.617.472-5.015-.8-6.796-1.163-1.635-3.125-2.58-5.488-2.689l-44.737-.581c-.291%200-.545-.145-.691-.363s-.182-.509-.109-.8c.145-.436.581-.763%201.054-.8l45.137-.581c5.342-.254%2011.157-4.579%2013.192-9.885l2.58-6.723c.109-.291.145-.581.073-.872-2.906-13.158-14.644-22.97-28.672-22.97-12.938%200-23.913%208.359-27.838%2019.952a13.35%2013.35%200%200%200-9.267-2.58c-6.215.618-11.193%205.597-11.811%2011.811-.145%201.599-.036%203.162.327%204.615C10.104%2070.051%202%2078.337%202%2088.549c0%20.909.073%201.817.182%202.726a.895.895%200%200%200%20.872.763h82.57c.472%200%20.909-.327%201.054-.8l.617-2.216z'/%3e%3cpath%20fill='%23FAAE40'%20d='M101.542%2060.275c-.4%200-.836%200-1.236.036-.291%200-.545.218-.654.509l-1.744%206.069c-.763%202.617-.472%205.015.8%206.796%201.163%201.635%203.125%202.58%205.488%202.689l9.522.581c.291%200%20.545.145.691.363.145.218.182.545.109.8-.145.436-.581.763-1.054.8l-9.924.582c-5.379.254-11.157%204.579-13.192%209.885l-.727%201.853c-.145.363.109.727.509.727h34.089c.4%200%20.763-.254.872-.654.581-2.108.909-4.325.909-6.614%200-13.447-10.975-24.422-24.458-24.422'/%3e%3c/svg%3e",Fs="data:image/svg+xml,%3csvg%20xmlns:xlink='http://www.w3.org/1999/xlink'%20xmlns='http://www.w3.org/2000/svg'%20id='Layer_1'%20data-name='Layer%201'%20viewBox='0%200%20224.09%20161.1'%20width='224.09'%20height='161.1'%20%3e%3cdefs%3e%3cstyle%3e%20.cls-1,%20.cls-2%20{%20fill:%20%23000;%20stroke-width:%200px;%20}%20.cls-2%20{%20fill-rule:%20evenodd;%20}%20%3c/style%3e%3c/defs%3e%3cpath%20class='cls-2'%20d='M92.1,112.3h-27.07c-1.65,0-3.3-.99-4.29-2.64l-12.55-22.47c-.66-1.65-.66-3.3,0-4.62l12.55-22.47c.99-1.65,2.64-2.64,4.29-2.64h15.52c1.65,0,3.3-.99,4.29-2.64l11.23-20.16c.99-1.65-.33-3.63-1.98-3.63h-41.6c-3.63,0-6.6,1.98-8.58,4.96l-24.76,44.28c-1.65,2.97-1.65,6.61,0,9.58l24.76,43.94c1.65,2.97,4.95,4.96,8.58,4.96h53.15c1.98,0,2.97-1.98,1.98-3.64l-11.55-20.49c-.33-1.65-1.98-2.31-3.96-2.31Z'/%3e%3cpath%20class='cls-1'%20d='M165.4,50.51l-10.57-17.84c-.99-1.65-3.3-1.65-4.29,0l-44.57,79.3c-.66,1.65-.66,3.3,0,4.63l11.55,20.82c.99,1.65,3.3,1.65,4.29,0l43.58-77.32c1.65-2.97,1.65-6.61,0-9.58Z'/%3e%3cpath%20class='cls-2'%20d='M111.58,86.86l13.21-23.46c.66-1.32.66-3.3,0-4.96l-15.19-26.1c-.99-1.65-3.3-1.65-4.29,0l-12.88,22.8c-.66,1.32-.66,3.3,0,4.63l14.86,26.76c.99,1.98,3.63,1.98,4.29.33Z'/%3e%3cpath%20class='cls-1'%20d='M204.35,50.51l-10.57-17.84c-.99-1.65-3.3-1.65-4.29,0l-44.57,79.3c-.66,1.65-.66,3.3,0,4.63l11.89,20.82c.99,1.65,3.3,1.65,4.29,0l43.58-77.32c1.65-2.97,1.65-6.61-.33-9.58Z'/%3e%3c/svg%3e",Ps="data:image/svg+xml,%3csvg%20xmlns:xlink='http://www.w3.org/1999/xlink'%20xmlns='http://www.w3.org/2000/svg'%20id='Layer_1'%20data-name='Layer%201'%20viewBox='0%200%20224.09%20161.1'%20width='224.09'%20height='161.1'%3e%3cdefs%3e%3cstyle%3e%20.cls-1,%20.cls-2%20{%20fill:%20%23fff;%20stroke-width:%200px;%20}%20.cls-2%20{%20fill-rule:%20evenodd;%20}%20%3c/style%3e%3c/defs%3e%3cpath%20class='cls-2'%20d='M92.1,112.3h-27.07c-1.65,0-3.3-.99-4.29-2.64l-12.55-22.47c-.66-1.65-.66-3.3,0-4.62l12.55-22.47c.99-1.65,2.64-2.64,4.29-2.64h15.52c1.65,0,3.3-.99,4.29-2.64l11.23-20.16c.99-1.65-.33-3.63-1.98-3.63h-41.6c-3.63,0-6.6,1.98-8.58,4.96l-24.76,44.28c-1.65,2.97-1.65,6.61,0,9.58l24.76,43.94c1.65,2.97,4.95,4.96,8.58,4.96h53.15c1.98,0,2.97-1.98,1.98-3.64l-11.55-20.49c-.33-1.65-1.98-2.31-3.96-2.31Z'%3e%3c/path%3e%3cpath%20class='cls-1'%20d='M165.4,50.51l-10.57-17.84c-.99-1.65-3.3-1.65-4.29,0l-44.57,79.3c-.66,1.65-.66,3.3,0,4.63l11.55,20.82c.99,1.65,3.3,1.65,4.29,0l43.58-77.32c1.65-2.97,1.65-6.61,0-9.58Z'%3e%3c/path%3e%3cpath%20class='cls-2'%20d='M111.58,86.86l13.21-23.46c.66-1.32.66-3.3,0-4.96l-15.19-26.1c-.99-1.65-3.3-1.65-4.29,0l-12.88,22.8c-.66,1.32-.66,3.3,0,4.63l14.86,26.76c.99,1.98,3.63,1.98,4.29.33Z'%3e%3c/path%3e%3cpath%20class='cls-1'%20d='M204.35,50.51l-10.57-17.84c-.99-1.65-3.3-1.65-4.29,0l-44.57,79.3c-.66,1.65-.66,3.3,0,4.63l11.89,20.82c.99,1.65,3.3,1.65,4.29,0l43.58-77.32c1.65-2.97,1.65-6.61-.33-9.58Z'%3e%3c/path%3e%3c/svg%3e",Bs="data:image/svg+xml,%3c?xml%20version='1.0'%20encoding='UTF-8'?%3e%3csvg%20id='standard_product_icon'%20xmlns='http://www.w3.org/2000/svg'%20version='1.1'%20viewBox='0%200%20512%20512'%3e%3c!--%20Generator:%20Adobe%20Illustrator%2029.1.0,%20SVG%20Export%20Plug-In%20.%20SVG%20Version:%202.1.0%20Build%20142)%20--%3e%3cdefs%3e%3cstyle%3e%20.st0%20{%20fill:%20none;%20}%20.st1%20{%20fill:%20%234285f4;%20}%20.st2%20{%20fill:%20%2334a853;%20}%20.st3%20{%20fill:%20%23fbbc04;%20}%20.st4%20{%20fill:%20%23ea4335;%20}%20%3c/style%3e%3c/defs%3e%3cg%20id='bounding_box'%3e%3crect%20class='st0'%20width='512'%20height='512'/%3e%3c/g%3e%3cg%20id='art'%3e%3cpath%20class='st2'%20d='M442,277.9H70c-8.8,0-16,7.2-16,16v148.1c0,8.8,7.2,16,16,16h107.2c1.5.5,3.1.7,4.8.7s3.3-.3,4.8-.7h255.2c8.8,0,16-7.2,16-16v-148.1c0-8.8-7.2-16-16-16ZM86,309.9h80v116.1h-80v-116.1ZM426,425.9h-228v-116.1h228v116.1Z'/%3e%3cpath%20class='st3'%20d='M442,54H70c-8.8,0-16,7.2-16,16v148.8c0,8.8,7.2,16,16,16h372c8.8,0,16-7.2,16-16V70c0-8.8-7.2-16-16-16ZM86,86h80v116.8h-80v-116.8ZM426,202.8h-228v-116.8h228v116.8Z'/%3e%3cpath%20class='st4'%20d='M442,234.8h-16V86H54v-16c0-8.8,7.2-16,16-16h372c8.8,0,16,7.2,16,16v148.8c0,8.8-7.2,16-16,16Z'/%3e%3cpath%20class='st1'%20d='M442,457.9h-16v-148.1H54v-16c0-8.8,7.2-16,16-16h372c8.8,0,16,7.2,16,16v148.1c0,8.8-7.2,16-16,16Z'/%3e%3ccircle%20class='st4'%20cx='349'%20cy='144.4'%20r='37'/%3e%3ccircle%20class='st1'%20cx='349'%20cy='367.9'%20r='37'/%3e%3c/g%3e%3c/svg%3e",Es="data:image/svg+xml,%3csvg%20viewBox='0%200%2087.3%2078'%20xmlns='http://www.w3.org/2000/svg'%3e%3cpath%20d='m6.6%2066.85%203.85%206.65c.8%201.4%201.95%202.5%203.3%203.3l13.75-23.8h-27.5c0%201.55.4%203.1%201.2%204.5z'%20fill='%230066da'/%3e%3cpath%20d='m43.65%2025-13.75-23.8c-1.35.8-2.5%201.9-3.3%203.3l-25.4%2044a9.06%209.06%200%200%200%20-1.2%204.5h27.5z'%20fill='%2300ac47'/%3e%3cpath%20d='m73.55%2076.8c1.35-.8%202.5-1.9%203.3-3.3l1.6-2.75%207.65-13.25c.8-1.4%201.2-2.95%201.2-4.5h-27.502l5.852%2011.5z'%20fill='%23ea4335'/%3e%3cpath%20d='m43.65%2025%2013.75-23.8c-1.35-.8-2.9-1.2-4.5-1.2h-18.5c-1.6%200-3.15.45-4.5%201.2z'%20fill='%2300832d'/%3e%3cpath%20d='m59.8%2053h-32.3l-13.75%2023.8c1.35.8%202.9%201.2%204.5%201.2h50.8c1.6%200%203.15-.45%204.5-1.2z'%20fill='%232684fc'/%3e%3cpath%20d='m73.4%2026.5-12.7-22c-.8-1.4-1.95-2.5-3.3-3.3l-13.75%2023.8%2016.15%2028h27.45c0-1.55-.4-3.1-1.2-4.5z'%20fill='%23ffba00'/%3e%3c/svg%3e",Ms=P(),Os={s3:{src:St,dark:qs},coreweave:{src:Fs,dark:Ps},cloudflare:{src:Ns},azure:{src:vt},gcs:{src:Bs},http:dt,file:J,"in-memory":at,gdrive:{src:Es},github:lt};const Fe=t=>{let e=(0,Ms.c)(11),{protocol:s,forceDark:o,className:r}=t,{theme:l}=Ct(),a=Os[s.toLowerCase()]??J;if("src"in a){let p=a.src;a.dark&&(o||l==="dark")&&(p=a.dark);let f=p,h;e[0]===r?h=e[1]:(h=W("h-3.5 w-3.5",r),e[0]=r,e[1]=h);let b;return e[2]!==s||e[3]!==p||e[4]!==h?(b=(0,n.jsx)("img",{src:f,alt:s,className:h}),e[2]=s,e[3]=p,e[4]=h,e[5]=b):b=e[5],b}let c=a,d;e[6]===r?d=e[7]:(d=W("h-3.5 w-3.5",r),e[6]=r,e[7]=d);let m;return e[8]!==c||e[9]!==d?(m=(0,n.jsx)(c,{className:d}),e[8]=c,e[9]=d,e[10]=m):m=e[10],m},Pe=y({type:_("s3"),bucket:u().nonempty().describe(i.of({label:"Bucket",placeholder:"my-bucket"})),region:u().optional().describe(i.of({label:"Region",placeholder:"us-east-1"})),access_key_id:u().optional().describe(i.of({label:"Access Key ID",inputType:"password",optionRegex:".*access_key.*"})),secret_access_key:u().optional().describe(i.of({label:"Secret Access Key",inputType:"password",optionRegex:".*secret.*access.*"})),endpoint_url:u().optional().describe(i.of({label:"Endpoint URL",placeholder:"https://s3.amazonaws.com"}))}).describe(i.of({direction:"two-columns"})),Be=y({type:_("gcs"),bucket:u().nonempty().describe(i.of({label:"Bucket",placeholder:"my-bucket"})),service_account_key:u().optional().describe(i.of({label:"Service Account Key (JSON)",inputType:"textarea"}))}).describe(i.of({direction:"two-columns"})),Ee=y({type:_("azure"),container:u().nonempty().describe(i.of({label:"Container",placeholder:"my-container"})),account_name:u().nonempty().describe(i.of({label:"Account Name",placeholder:"storageaccount",optionRegex:".*account.*"})),account_key:u().optional().describe(i.of({label:"Account Key",inputType:"password",optionRegex:".*azure.*key.*"}))}).describe(i.of({direction:"two-columns"})),Me=y({type:_("coreweave"),bucket:u().nonempty().describe(i.of({label:"Bucket",placeholder:"bucket-name"})),region:u().nonempty().describe(i.of({label:"Region",placeholder:"US-EAST-04A"})),access_key_id:u().optional().describe(i.of({label:"Access Key ID",inputType:"password",optionRegex:".*object_storage_key.*"})),secret_access_key:u().optional().describe(i.of({label:"Secret Access Key",inputType:"password",optionRegex:".*object_storage_secret.*"}))}).describe(i.of({direction:"two-columns"})),Oe=y({type:_("gdrive"),credentials_json:u().optional().describe(i.of({label:"Service Account JSON",description:"Leave empty to use browser-based authentication",inputType:"textarea"}))}).describe(i.of({direction:"two-columns"})),Us=H("type",[Pe,Be,Ee,Me,Oe]),Ks={obstore:"Obstore",fsspec:"fsspec"};var Hs=class{constructor(){K(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(M(e)){let o=pe(e);return this.secrets[s]=`os.environ.get("${o}")`,s}return`"${e}"`}formatSecrets(){return Object.keys(this.secrets).length===0?"":Object.entries(this.secrets).map(([t,e])=>`${t} = ${e}`).join(`
-`)}};function zs(t,e){let s=e.print("bucket",t.bucket),o=new Set(["from obstore.store import S3Store"]),r=[];return t.region&&r.push(`    region=${e.print("region",t.region)},`),t.access_key_id&&r.push(`    access_key_id=${e.print("access_key_id",t.access_key_id)},`),t.secret_access_key&&r.push(`    secret_access_key=${e.print("secret_access_key",t.secret_access_key)},`),t.endpoint_url&&r.push(`    endpoint_url=${e.print("endpoint_url",t.endpoint_url)},`),{imports:o,code:g(`
-    store = S3Store(${s},${r.length>0?`
-${r.join(`
-`)}
-`:""})
-  `)}}function Ws(t,e){let s=e.print("bucket",t.bucket),o=new Set(["from obstore.store import GCSStore"]),r;return t.service_account_key?(o.add("import json"),r=g(`
-      _credentials = json.loads("""${t.service_account_key}""")
-      store = GCSStore(${s},
-          service_account_key=_credentials,
-      )
-    `)):r=g(`
-      store = GCSStore(${s})
-    `),{imports:o,code:r}}function Gs(t,e){let s=e.print("container",t.container),o=e.print("account_name",t.account_name),r=new Set(["from obstore.store import AzureStore"]),l=[`account_name=${o},`];return t.account_key&&l.push(`account_key=${e.print("account_key",t.account_key)},`),{imports:r,code:`store = AzureStore(${s},
-${l.map(a=>`    ${a}`).join(`
-`)}
-)`}}function Vs(t,e){let s=e.print("bucket",t.bucket),o=new Set(["from obstore.store import S3Store"]),r=[`    region=${e.print("region",t.region)},`];return t.access_key_id&&r.push(`    access_key_id=${e.print("access_key_id",t.access_key_id)},`),t.secret_access_key&&r.push(`    secret_access_key=${e.print("secret_access_key",t.secret_access_key)},`),r.push(`    endpoint="https://${t.bucket}.cwobject.com",`,"    virtual_hosted_style_request=True,"),{imports:o,code:g(`
-    store = S3Store(${s},${`
-${r.join(`
-`)}
-`})
-  `)}}function Zs(t,e){var o;let s=new Set(["from gdrive_fsspec import GoogleDriveFileSystem"]);if(t.credentials_json){s.add("import json");let r=e.print("credentials_json",t.credentials_json);return{imports:s,code:g(`
-      _creds = json.loads("""${(o=t.credentials_json)!=null&&o.startsWith("ENV:")?`{${r}}`:t.credentials_json}""")
-      fs = GoogleDriveFileSystem(creds=_creds, token="service_account")
-    `)}}return{imports:s,code:g(`
-    fs = GoogleDriveFileSystem(token="browser")
-  `)}}function Qs(t,e){Us.parse(t);let s=new Hs,o;switch(t.type){case"s3":o=zs(t,s);break;case"gcs":o=Ws(t,s);break;case"azure":o=Gs(t,s);break;case"coreweave":o=Vs(t,s);break;case"gdrive":o=Zs(t,s);break;default:z(t)}let r=[...new Set([...s.imports,...o.imports])].sort();r.push("");let l=s.formatSecrets();return l&&r.push(l),r.push(o.code.trim()),r.join(`
-`)}var Ue=P(),Js="#232F3E",Ke=[{name:"Amazon S3",schema:Pe,protocol:"s3",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Cloud Storage",schema:Be,protocol:"gcs",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Azure Blob Storage",schema:Ee,protocol:"azure",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"CoreWeave",schema:Me,protocol:"coreweave",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Drive",schema:Oe,protocol:"gdrive",storageLibraries:{libraries:["fsspec"],preferred:"fsspec"}}],Ys=t=>{let e=(0,Ue.c)(4),{onSelect:s}=t,o;e[0]===s?o=e[1]:(o=Ke.map(l=>{let{name:a,schema:c,protocol:d}=l;return(0,n.jsx)(te,{name:a,color:Js,icon:(0,n.jsx)(Fe,{protocol:d,forceDark:!0,className:"w-7.5 h-7.5"}),onSelect:()=>s(c)},a)}),e[0]=s,e[1]=o);let r;return e[2]===o?r=e[3]:(r=(0,n.jsx)(ee,{children:o}),e[2]=o,e[3]=r),r};const Xs=t=>{let e=(0,Ue.c)(13),{onSubmit:s,header:o}=t,[r,l]=(0,B.useState)(null);if(!r){let h;e[0]===Symbol.for("react.memo_cache_sentinel")?(h=(0,n.jsx)("div",{children:(0,n.jsx)(Ys,{onSelect:l})}),e[0]=h):h=e[0];let b;return e[1]===o?b=e[2]:(b=(0,n.jsxs)(n.Fragment,{children:[o,h]}),e[1]=o,e[2]=b),b}let a;e[3]===r?a=e[4]:(a=Ke.find(h=>h.schema===r),e[3]=r,e[4]=a);let c=a==null?void 0:a.storageLibraries,d;e[5]===(c==null?void 0:c.libraries)?d=e[6]:(d=(c==null?void 0:c.libraries)??[],e[5]=c==null?void 0:c.libraries,e[6]=d);let m=(c==null?void 0:c.preferred)??"obstore",p;e[7]===Symbol.for("react.memo_cache_sentinel")?(p=()=>l(null),e[7]=p):p=e[7];let f;return e[8]!==s||e[9]!==r||e[10]!==d||e[11]!==m?(f=(0,n.jsx)(me,{schema:r,libraries:d,preferredLibrary:m,displayNames:Ks,libraryLabel:"Preferred storage library",generateCode:er,onSubmit:s,onBack:p}),e[8]=s,e[9]=r,e[10]=d,e[11]=m,e[12]=f):f=e[12],f};function er(t,e){return Qs(t,e)}var He=P();const tr=t=>{let e=(0,He.c)(9),{children:s,defaultTab:o}=t,r=o===void 0?"databases":o,[l,a]=(0,B.useState)(!1),c;e[0]===s?c=e[1]:(c=(0,n.jsx)(Ft,{asChild:!0,children:s}),e[0]=s,e[1]=c);let d;e[2]===Symbol.for("react.memo_cache_sentinel")?(d=()=>a(!1),e[2]=d):d=e[2];let m;e[3]===r?m=e[4]:(m=(0,n.jsx)(ze,{defaultTab:r,onClose:d}),e[3]=r,e[4]=m);let p;return e[5]!==l||e[6]!==c||e[7]!==m?(p=(0,n.jsxs)(Et,{open:l,onOpenChange:a,children:[c,m]}),e[5]=l,e[6]=c,e[7]=m,e[8]=p):p=e[8],p},ze=t=>{let e=(0,He.c)(12),{defaultTab:s,onClose:o}=t,r=s===void 0?"databases":s,l;e[0]===Symbol.for("react.memo_cache_sentinel")?(l=(0,n.jsxs)(Mt,{className:"w-full mb-4",children:[(0,n.jsx)(le,{value:"databases",className:"flex-1",children:"Databases & Catalogs"}),(0,n.jsx)(le,{value:"storage",className:"flex-1",children:"Remote Storages"})]}),e[0]=l):l=e[0];let a=l,c;e[1]===Symbol.for("react.memo_cache_sentinel")?(c=(0,n.jsx)(Nt,{children:"Add Connection"}),e[1]=c):c=e[1];let d;e[2]===Symbol.for("react.memo_cache_sentinel")?(d=(0,n.jsx)(de,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"database, data catalog"}),e[2]=d):d=e[2];let m;e[3]===Symbol.for("react.memo_cache_sentinel")?(m=(0,n.jsxs)(qt,{children:[c,(0,n.jsxs)(Bt,{children:["Connect to a"," ",d," ","or"," ",(0,n.jsx)(de,{href:"https://docs.marimo.io/guides/working_with_data/remote_storage/",children:"remote storage"})," ","to work with data directly from your notebook."]})]}),e[3]=m):m=e[3];let p;e[4]===o?p=e[5]:(p=(0,n.jsx)(ae,{value:"databases",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,n.jsx)(Ds,{onSubmit:o,header:a})}),e[4]=o,e[5]=p);let f;e[6]===o?f=e[7]:(f=(0,n.jsx)(ae,{value:"storage",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,n.jsx)(Xs,{onSubmit:o,header:a})}),e[6]=o,e[7]=f);let h;return e[8]!==r||e[9]!==p||e[10]!==f?(h=(0,n.jsxs)(Pt,{className:"max-h-[75vh] overflow-y-auto",children:[m,(0,n.jsxs)(Ot,{defaultValue:r,children:[p,f]})]}),e[8]=r,e[9]=p,e[10]=f,e[11]=h):h=e[11],h};export{J as i,ze as n,Fe as r,tr as t};