@marimo-team/frontend 0.17.8 → 0.18.0

package/dist/assets/add-database-form-CDlwxUUe.js

@@ -0,0 +1,111 @@
+var Ue=Object.defineProperty;var Ke=(t,e,s)=>e in t?Ue(t,e,{enumerable:!0,configurable:!0,writable:!0,value:s}):t[e]=s;var oe=(t,e,s)=>Ke(t,typeof e!="symbol"?e+"":e,s);import{s as G}from"./chunk-DZLz74EQ.js";import{t as We}from"./react-cj45iF1i.js";import{$n as h,ct as He,w as Me}from"./cells-D2eha_-B.js";import{A as p,E as R,F as m,O as re,_ as ue,i as Qe,m as Ge,z as d}from"./zod-U2NFpcFA.js";import{t as be}from"./compiler-runtime-bbSa6V1W.js";import{t as Ve}from"./_createAggregator-CP8XSJVt.js";import{d as ze}from"./hotkeys-jCLisiwR.js";import{t as Je}from"./jsx-runtime-DmpivWpf.js";import{t as fe}from"./cn-BaXqCYqP.js";import{t as ie}from"./button-DCB7VoVL.js";import{t as Ye}from"./key-CaC9xuXQ.js";import{r as ce}from"./useLifecycle-DiLWmtpP.js";import{c as Xe,i as Ze,n as et,r as tt,s as st,t as nt}from"./select-BkT3QHmw.js";import{t as ot}from"./circle-plus-gt8rkr_9.js";import{a as ae,c as ge,o as rt,p as it,r as ct,t as at}from"./dropdown-menu-CN5aVh8R.js";import{o as lt,r as dt}from"./input-BYmR6HBp.js";import{a as ht,c as pt,i as mt,n as ut,o as bt,p as ft,r as gt,s as yt,u as St}from"./form-C4OqdpTP.js";import{a as _t,c as wt,l as xt,n as $t,r as kt,t as Ct}from"./dialog-B0ZQRYmN.js";import{n as At}from"./ImperativeModal-wjVZ4eQS.js";import{n as jt}from"./useAsyncData-CXeJJDVW.js";import{t as Rt}from"./links-L8RfsBvj.js";import{o as vt,t as It,u as i}from"./form-CYnSp8Xw.js";import{o as Tt}from"./focus-DkcYPAN0.js";import{t as qt}from"./request-registry-u0si-SpI.js";import{n as Dt,t as Lt}from"./write-secret-modal-CW3l4drx.js";var Pt=Ve(function(t,e,s){t[s?0:1].push(e)},function(){return[[],[]]});function $(){return d().optional().describe(i.of({label:"Password",inputType:"password",placeholder:"password",optionRegex:".*password.*"}))}function le(t,e){let s=d();return s=e?s.nonempty():s.optional(),s=s.describe(i.of({label:t||"Token",inputType:"password",placeholder:"token",optionRegex:".*token.*"})),s}function V(){return d().optional().describe(i.of({label:"Warehouse Name",placeholder:"warehouse",optionRegex:".*warehouse.*"}))}function de(t,e){let s=d();return s=e?s.nonempty():s.optional(),s.describe(i.of({label:t||"URI",optionRegex:".*uri.*"}))}function x(t){return d().nonempty().describe(i.of({label:t||"Host",placeholder:"localhost",optionRegex:".*host.*"}))}function w(){return d().describe(i.of({label:"Database",placeholder:"db name",optionRegex:".*database.*"}))}function Nt(){return d().describe(i.of({label:"Schema",placeholder:"schema name",optionRegex:".*schema.*"}))}function k(){return d().nonempty().describe(i.of({label:"Username",placeholder:"username",optionRegex:".*username.*"}))}function C(t){let e=Qe().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 ye(){return R().default(!1).describe(i.of({label:"Read Only"}))}const Se=m({type:p("postgres"),host:x(),port:C(5432).optional(),database:w(),username:k(),password:$(),ssl:R().default(!1).describe(i.of({label:"Use SSL"}))}).describe(i.of({direction:"two-columns"})),_e=m({type:p("mysql"),host:x(),port:C(3306),database:w(),username:k(),password:$(),ssl:R().default(!1).describe(i.of({label:"Use SSL"}))}).describe(i.of({direction:"two-columns"})),we=m({type:p("sqlite"),database:w().describe(i.of({label:"Database Path"}))}).describe(i.of({direction:"two-columns"})),xe=m({type:p("duckdb"),database:w().describe(i.of({label:"Database Path"})),read_only:ye()}).describe(i.of({direction:"two-columns"})),$e=m({type:p("motherduck"),database:w().default("my_db").describe(i.of({label:"Database Name"})),token:le()}).describe(i.of({direction:"two-columns"})),ke=m({type:p("snowflake"),account:d().nonempty().describe(i.of({label:"Account",optionRegex:".*snowflake.*"})),warehouse:d().optional().describe(i.of({label:"Warehouse",optionRegex:".*snowflake.*"})),database:w(),schema:d().optional().describe(i.of({label:"Schema",optionRegex:".*snowflake.*"})),username:k(),password:$(),role:d().optional().describe(i.of({label:"Role"}))}).describe(i.of({direction:"two-columns"})),Ce=m({type:p("bigquery"),project:d().nonempty().describe(i.of({label:"Project ID",optionRegex:".*bigquery.*"})),dataset:d().nonempty().describe(i.of({label:"Dataset",optionRegex:".*bigquery.*"})),credentials_json:d().describe(i.of({label:"Credentials JSON",inputType:"textarea"}))}).describe(i.of({direction:"two-columns"})),Ae=m({type:p("clickhouse_connect"),host:x(),port:C(8123).optional(),username:k(),password:$(),secure:R().default(!1).describe(i.of({label:"Use HTTPs"}))}).describe(i.of({direction:"two-columns"})),je=m({type:p("timeplus"),host:x().default("localhost"),port:C(8123).optional(),username:k().default("default"),password:$().default("")}).describe(i.of({direction:"two-columns"})),Re=m({type:p("chdb"),database:w().describe(i.of({label:"Database Path"})),read_only:ye()}).describe(i.of({direction:"two-columns"})),ve=m({type:p("trino"),host:x(),port:C(8080),database:w(),schema:Nt().optional(),username:k(),password:$(),async_support:R().default(!1).describe(i.of({label:"Async Support"}))}).describe(i.of({direction:"two-columns"})),Ie=m({type:p("iceberg"),name:d().describe(i.of({label:"Catalog Name"})),catalog:re("type",[m({type:p("REST"),warehouse:V(),uri:d().optional().describe(i.of({label:"URI",placeholder:"https://",optionRegex:".*uri.*"})),token:le()}),m({type:p("SQL"),warehouse:V(),uri:d().optional().describe(i.of({label:"URI",placeholder:"jdbc:iceberg://host:port/database",optionRegex:".*uri.*"}))}),m({type:p("Hive"),warehouse:V(),uri:de()}),m({type:p("Glue"),warehouse:V(),uri:de()}),m({type:p("DynamoDB"),"dynamodb.profile-name":d().optional().describe(i.of({label:"Profile Name"})),"dynamodb.region":d().optional().describe(i.of({label:"Region"})),"dynamodb.access-key-id":d().optional().describe(i.of({label:"Access Key ID"})),"dynamodb.secret-access-key":d().optional().describe(i.of({label:"Secret Access Key",inputType:"password"})),"dynamodb.session-token":d().optional().describe(i.of({label:"Session Token",inputType:"password"}))})]).default({type:"REST",token:void 0}).describe(i.of({special:"tabs"}))}),Te=m({type:p("datafusion"),sessionContext:R().optional().describe(i.of({label:"Use Session Context"}))}),qe=m({type:p("pyspark"),host:x().optional(),port:C().optional()}),De=m({type:p("redshift"),host:x(),port:C(5439),connectionType:re("type",[m({type:p("IAM credentials"),region:d().describe(i.of({label:"Region"})),aws_access_key_id:d().nonempty().describe(i.of({label:"AWS Access Key ID",inputType:"password",optionRegex:".*aws_access_key_id.*"})),aws_secret_access_key:d().nonempty().describe(i.of({label:"AWS Secret Access Key",inputType:"password",optionRegex:".*aws_secret_access_key.*"})),aws_session_token:d().optional().describe(i.of({label:"AWS Session Token",inputType:"password",optionRegex:".*aws_session_token.*"}))}),m({type:p("DB credentials"),user:k(),password:$()})]).default({type:"IAM credentials",aws_access_key_id:"",aws_secret_access_key:"",region:""}),database:w()}).describe(i.of({direction:"two-columns"})),Le=m({type:p("databricks"),access_token:le("Access Token",!0),server_hostname:x("Server Hostname"),http_path:de("HTTP Path",!0),catalog:d().optional().describe(i.of({label:"Catalog"})),schema:d().optional().describe(i.of({label:"Schema"}))}).describe(i.of({direction:"two-columns"})),Pe=m({type:p("supabase"),host:x(),port:C(5432).optional(),database:w(),username:k(),password:$(),disable_client_pooling:R().default(!1).describe(i.of({label:"Disable Client-Side Pooling"}))}).describe(i.of({direction:"two-columns"})),Ft=re("type",[Se,_e,we,xe,$e,ke,Ce,Ae,je,Re,ve,Ie,Te,qe,De,Le,Pe]);var he="env:";function Gs(t){return t}function z(t){return typeof t=="string"?t.startsWith(he):!1}function pe(t){return`${he}${t}`}function Et(t){return t.replace(he,"")}const Ne={sqlmodel:"SQLModel",sqlalchemy:"SQLAlchemy",duckdb:"DuckDB",clickhouse_connect:"ClickHouse Connect",chdb:"chDB",pyiceberg:"PyIceberg",ibis:"Ibis",motherduck:"MotherDuck",redshift:"Redshift",databricks:"Databricks"};var y=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}},Bt=t=>`_${t}`,Ot=class{constructor(){oe(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e,s){if(t=Bt(t),z(e)){let o=Et(e),n=s?`os.environ.get("${o}", "${s}")`:`os.environ.get("${o}")`;return this.secrets[t]=n,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"?U(e):e?`"${e}"`:""}printInFString(t,e,s){if(e===void 0)return"";if(typeof e=="number")return`${e}`;if(typeof e=="boolean")return U(e);let o=this.print(t,e,s);return o.startsWith('"')&&o.endsWith('"')?o.slice(1,-1):`{${o}}`}printPassword(t,e,s,o){let n=s?this.printInFString.bind(this):this.print.bind(this);return z(t)?n(o||"password",t):n(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(`
+`)}},Ut=class extends y{generateImports(){return[]}generateConnectionCode(){let t=this.connection.ssl?", connect_args={'sslmode': 'require'}":"",e=this.secrets.printPassword(this.connection.password,"POSTGRES_PASSWORD",!0);return h(`
+      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})
+    `)}},Kt=class extends y{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 h(`
+      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})
+    `)}},Wt=class extends y{generateImports(){return[]}generateConnectionCode(){let t=this.connection.database?this.secrets.printInFString("database",this.connection.database):"";return h(`
+      ${t.startsWith("{")&&t.endsWith("}")?`DATABASE_URL = f"sqlite:///${t}"`:`DATABASE_URL = "sqlite:///${t}"`}
+      engine = ${this.orm}.create_engine(DATABASE_URL)
+    `)}},Ht=class extends y{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 h(`
+      engine = ${this.orm}.create_engine(
+        URL(
+${J(e,s=>`          ${s}`)},
+        )
+      )
+    `)}},Mt=class extends y{generateImports(){return["import json"]}generateConnectionCode(){let t=this.secrets.printInFString("project",this.connection.project),e=this.secrets.printInFString("dataset",this.connection.dataset);return h(`
+      credentials = json.loads("""${this.connection.credentials_json}""")
+      engine = ${this.orm}.create_engine(f"bigquery://${t}/${e}", credentials_info=credentials)
+    `)}},Qt=class extends y{generateImports(){return[]}generateConnectionCode(){return h(`
+      DATABASE_URL = "${this.secrets.printInFString("database",this.connection.database||":memory:")}"
+      engine = ${this.orm}.connect(DATABASE_URL, read_only=${U(this.connection.read_only)})
+    `)}},Gt=class extends y{generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printInFString("database",this.connection.database);return this.connection.token?h(`
+      conn = duckdb.connect("md:${t}", config={"motherduck_token": ${this.secrets.printPassword(this.connection.token,"MOTHERDUCK_TOKEN",!1)}})
+    `):h(`
+        conn = duckdb.connect("md:${t}")
+        `)}},Vt=class extends y{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};return h(`
+      engine = ${this.orm}.get_client(
+${J(e,s=>`        ${s}`)},
+      )
+    `)}},zt=class extends y{generateImports(){return[]}generateConnectionCode(){let t=this.secrets.printPassword(this.connection.password,"TIMEPLUS_PASSWORD",!0);return h(`
+      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)
+    `)}},Jt=class extends y{generateImports(){return["import chdb"]}generateConnectionCode(){let t=this.secrets.print("database",this.connection.database)||'""';return h(`
+      engine = ${this.orm}.connect(${t}, read_only=${U(this.connection.read_only)})
+    `)}},Yt=class extends y{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),n=this.secrets.printInFString("port",this.connection.port),c=this.secrets.printInFString("database",this.connection.database),a=this.secrets.printPassword(this.connection.password,"TRINO_PASSWORD",!0);return h(`
+      engine = ${this.orm}.create_engine(f"${t}://${s}:${a}@${o}:${n}/${c}${e}")
+    `)}},Xt=class extends y{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:ce(this.connection.catalog)}}generateConnectionCode(){let t={...this.connection.catalog};t=Object.fromEntries(Object.entries(t).filter(([o,n])=>n!=null&&n!==""&&o!=="type"));for(let[o,n]of Object.entries(t))z(n)?t[o]=this.secrets.print(o,n):typeof n=="string"&&(t[o]=`"${n}"`);let e=is(t,o=>`              ${o}`),s=`"${this.connection.name}"`;switch(this.connection.catalog.type){case"REST":return h(`
+          catalog = RestCatalog(
+            ${s},
+            **{
+${e}
+            },
+          )
+        `);case"SQL":return h(`
+          catalog = SqlCatalog(
+            ${s},
+            **{
+${e}
+            },
+          )
+        `);case"Hive":return h(`
+          catalog = HiveCatalog(
+            ${s},
+            **{
+${e}
+            },
+          )
+        `);case"Glue":return h(`
+          catalog = GlueCatalog(
+            ${s},
+            **{
+${e}
+            },
+          )
+        `);case"DynamoDB":return h(`
+          catalog = DynamoDBCatalog(
+            ${s},
+            **{
+${e}
+            },
+          )
+        `);default:ce(this.connection.catalog)}}},Zt=class extends y{generateImports(){return["from datafusion import SessionContext"]}generateConnectionCode(){return this.connection.sessionContext?h(`
+        ctx = SessionContext()
+        # Sample table
+        _ = ctx.from_pydict({"a": [1, 2, 3]}, "my_table")
+
+        con = ibis.datafusion.connect(ctx)
+      `):h(`
+      con = ibis.datafusion.connect()
+    `)}},es=class extends y{generateImports(){return["from pyspark.sql import SparkSession"]}generateConnectionCode(){return this.connection.host||this.connection.port?h(`
+        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)
+      `):h(`
+      con = ibis.pyspark.connect()
+    `)}},ts=class extends y{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 c=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),l=this.connection.connectionType.aws_session_token?this.secrets.print("aws_session_token",this.connection.connectionType.aws_session_token):void 0;return h(`
+        con = redshift_connector.connect(
+${J({iam:!0,host:t,port:e,region:`"${this.connection.connectionType.region}"`,database:s,access_key_id:c,secret_access_key:a,...l&&{session_token:l}},b=>`          ${b}`)},
+        )
+      `)}let o=this.connection.connectionType.user?this.secrets.print("user",this.connection.connectionType.user):void 0,n=this.connection.connectionType.password?this.secrets.printPassword(this.connection.connectionType.password,"REDSHIFT_PASSWORD",!1):void 0;return h(`
+      con = redshift_connector.connect(
+${J({host:t,port:e,database:s,...o&&{user:o},...n&&{password:n}},c=>`          ${c}`)},
+      )
+    `)}},ss=class extends y{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"),n=this.connection.catalog?this.secrets.printInFString("catalog",this.connection.catalog):void 0,c=this.connection.schema?this.secrets.printInFString("schema",this.connection.schema):void 0,a=`databricks://token:${e}@${s}?http_path=${o}`;return n&&(a+=`&catalog=${n}`),c&&(a+=`&schema=${c}`),this.orm==="ibis"?h(`
+        engine = ibis.databricks.connect(
+          server_hostname=${s},
+          http_path=${o},${n?`
+          catalog=${n},`:""}${c?`
+          schema=${c},`:""}
+          access_token=${e}
+        )
+      `):h(`
+      DATABASE_URL = f"${a}"
+      engine = ${this.orm}.create_engine(DATABASE_URL)
+      `)}},ns=class extends y{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),n=this.secrets.printInFString("database",this.connection.database),c=this.connection.disable_client_pooling?", poolclass=NullPool":"";return h(`
+      DATABASE_URL = f"postgresql+psycopg2://${e}:${t}@${s}:${o}/${n}?sslmode=require"
+      engine = ${this.orm}.create_engine(DATABASE_URL${c})
+    `)}},os=class{constructor(){oe(this,"secrets",new Ot)}createGenerator(t,e){switch(t.type){case"postgres":return new Ut(t,e,this.secrets);case"mysql":return new Kt(t,e,this.secrets);case"sqlite":return new Wt(t,e,this.secrets);case"snowflake":return new Ht(t,e,this.secrets);case"bigquery":return new Mt(t,e,this.secrets);case"duckdb":return new Qt(t,e,this.secrets);case"motherduck":return new Gt(t,e,this.secrets);case"clickhouse_connect":return new Vt(t,e,this.secrets);case"timeplus":return new zt(t,e,this.secrets);case"chdb":return new Jt(t,e,this.secrets);case"trino":return new Yt(t,e,this.secrets);case"iceberg":return new Xt(t,e,this.secrets);case"datafusion":return new Zt(t,e,this.secrets);case"pyspark":return new es(t,e,this.secrets);case"redshift":return new ts(t,e,this.secrets);case"databricks":return new ss(t,e,this.secrets);case"supabase":return new ns(t,e,this.secrets);default:ce(t)}}};function rs(t,e){if(!(e in Ne))throw Error(`Unsupported library: ${e}`);Ft.parse(t);let s=new os,o=s.createGenerator(t,e),n=o.generateConnectionCode(),c=s.secrets,a=[...new Set([...c.imports,...o.imports])].sort();a.push("");let l=c.formatSecrets();return l&&a.push(l),a.push(n.trim()),a.join(`
+`)}function U(t){return t.toString().charAt(0).toUpperCase()+t.toString().slice(1)}function J(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,o])=>e(typeof o=="boolean"?`${s}=${U(o)}`:`${s}=${o}`)).join(`,
+`)}function is(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,o])=>{let n=`"${s}"`;return e(typeof o=="boolean"?`${n}: ${U(o)}`:`${n}: ${o}`)}).join(`,
+`)}var cs=G(be(),1),K=G(We(),1),r=G(Je(),1),Fe=(0,K.createContext)({providerNames:[],secretKeys:[],loading:!1,error:void 0,refreshSecrets:ze.NOOP});const as=()=>(0,K.use)(Fe),ls=t=>{let e=(0,cs.c)(14),{children:s}=t,o;e[0]===Symbol.for("react.memo_cache_sentinel")?(o=[],e[0]=o):o=e[0];let{data:n,isPending:c,error:a,refetch:l}=jt(us,o),b;e[1]===(n==null?void 0:n.secretKeys)?b=e[2]:(b=(n==null?void 0:n.secretKeys)||[],e[1]=n==null?void 0:n.secretKeys,e[2]=b);let u;e[3]===(n==null?void 0:n.providerNames)?u=e[4]:(u=(n==null?void 0:n.providerNames)||[],e[3]=n==null?void 0:n.providerNames,e[4]=u);let f;e[5]!==a||e[6]!==c||e[7]!==l||e[8]!==b||e[9]!==u?(f={secretKeys:b,providerNames:u,loading:c,error:a,refreshSecrets:l},e[5]=a,e[6]=c,e[7]=l,e[8]=b,e[9]=u,e[10]=f):f=e[10];let S;return e[11]!==s||e[12]!==f?(S=(0,r.jsx)(Fe,{value:f,children:s}),e[11]=s,e[12]=f,e[13]=S):S=e[13],S},ds={isMatch:t=>{if(t instanceof ue||t instanceof Ge){let{optionRegex:e}=i.parse(t.description||"");return!!e}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:o,providerNames:n,refreshSecrets:c}=as(),{openModal:a,closeModal:l}=At(),{label:b,description:u,optionRegex:f=""}=i.parse(t.description||""),[S,A]=Pt(o,g=>new RegExp(f,"i").test(g));return(0,r.jsx)(ht,{control:e.control,name:s,render:({field:g})=>(0,r.jsxs)(bt,{children:[(0,r.jsx)(yt,{children:b}),(0,r.jsx)(gt,{children:u}),(0,r.jsx)(ut,{children:(0,r.jsxs)("div",{className:"flex gap-2",children:[t instanceof ue?(0,r.jsx)(dt,{...g,value:g.value,onChange:g.onChange,className:fe("flex-1")}):(0,r.jsx)(lt,{...g,value:g.value,onChange:g.onChange,className:"flex-1"}),(0,r.jsxs)(at,{children:[(0,r.jsx)(it,{asChild:!0,children:(0,r.jsx)(ie,{variant:"outline",size:"icon",className:fe(z(g.value)&&"bg-accent"),children:(0,r.jsx)(Ye,{className:"h-3 w-3"})})}),(0,r.jsxs)(ct,{align:"end",className:"max-h-60 overflow-y-auto",children:[(0,r.jsxs)(ae,{onSelect:()=>{a((0,r.jsx)(Lt,{providerNames:n,onSuccess:_=>{c(),g.onChange(pe(_)),l()},onClose:l}))},children:[(0,r.jsx)(ot,{className:"mr-2 h-3.5 w-3.5"}),"Create a new secret"]}),S.length>0&&(0,r.jsxs)(r.Fragment,{children:[(0,r.jsx)(ge,{}),(0,r.jsx)(rt,{children:"Recommended"})]}),S.map(_=>(0,r.jsx)(ae,{onSelect:()=>g.onChange(pe(_)),children:_},_)),A.length>0&&(0,r.jsx)(ge,{}),A.map(_=>(0,r.jsx)(ae,{onSelect:()=>g.onChange(pe(_)),children:_},_))]})]})]})}),(0,r.jsx)(pt,{})]})})}};function hs(t){return t.provider!=="env"}function ps(t){return t.name}function ms(t){return t.keys}async function us(){let t=await qt.request({}),e=Dt(t.secrets).filter(hs).map(ps);return{secretKeys:t.secrets.flatMap(ms).sort(),providerNames:e}}var W=G(be(),1),Ee=[{name:"PostgreSQL",schema:Se,color:"#336791",logo:"postgres",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"MySQL",schema:_e,color:"#00758F",logo:"mysql",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"SQLite",schema:we,color:"#003B57",logo:"sqlite",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DuckDB",schema:xe,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:ke,color:"#29B5E8",logo:"snowflake",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse",schema:Ae,color:"#2C2C1D",logo:"clickhouse",connectionLibraries:{libraries:["clickhouse_connect"],preferred:"clickhouse_connect"}},{name:"Timeplus",schema:je,color:"#B83280",logo:"timeplus",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"BigQuery",schema:Ce,color:"#4285F4",logo:"bigquery",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse Embedded",schema:Re,color:"#f2b611",logo:"clickhouse",connectionLibraries:{libraries:["chdb"],preferred:"chdb"}},{name:"Trino",schema:ve,color:"#d466b6",logo:"trino",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DataFusion",schema:Te,color:"#202A37",logo:"datafusion",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"PySpark",schema:qe,color:"#1C5162",logo:"pyspark",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"Redshift",schema:De,color:"#522BAE",logo:"redshift",connectionLibraries:{libraries:["redshift"],preferred:"redshift"}},{name:"Databricks",schema:Le,color:"#c41e0c",logo:"databricks",connectionLibraries:{libraries:["sqlalchemy","sqlmodel","ibis"],preferred:"sqlalchemy"}},{name:"Supabase",schema:Pe,color:"#238F5F",logo:"supabase",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}}],Be=[{name:"Iceberg",schema:Ie,color:"#000000",logo:"iceberg",connectionLibraries:{libraries:["pyiceberg"],preferred:"pyiceberg"}}],bs=t=>{let e=(0,W.c)(14),{onSelect:s}=t,o;e[0]===s?o=e[1]:(o=S=>{let{name:A,schema:g,color:_,logo:v}=S;return(0,r.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:{backgroundColor:_},onClick:()=>s(g),children:[(0,r.jsx)(He,{name:v,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),(0,r.jsx)("span",{className:"text-white font-medium text-lg",children:A})]},A)},e[0]=s,e[1]=o);let n=o,c;e[2]===n?c=e[3]:(c=Ee.map(n),e[2]=n,e[3]=c);let a;e[4]===c?a=e[5]:(a=(0,r.jsx)("div",{className:"grid grid-cols-2 md:grid-cols-3 gap-4",children:c}),e[4]=c,e[5]=a);let l;e[6]===Symbol.for("react.memo_cache_sentinel")?(l=(0,r.jsxs)("h4",{className:"font-semibold text-muted-foreground text-lg flex items-center gap-4",children:["Data Catalogs",(0,r.jsx)("hr",{className:"flex-1"})]}),e[6]=l):l=e[6];let b;e[7]===n?b=e[8]:(b=Be.map(n),e[7]=n,e[8]=b);let u;e[9]===b?u=e[10]:(u=(0,r.jsx)("div",{className:"grid grid-cols-2 md:grid-cols-3 gap-4",children:b}),e[9]=b,e[10]=u);let f;return e[11]!==a||e[12]!==u?(f=(0,r.jsxs)(r.Fragment,{children:[a,l,u]}),e[11]=a,e[12]=u,e[13]=f):f=e[13],f},fs=[ds],gs=t=>{var me;let e=(0,W.c)(49),{schema:s,onSubmit:o,onBack:n}=t,c;e[0]===s?c=e[1]:(c=vt(s),e[0]=s,e[1]=c);let a=s,l;e[2]===a?l=e[3]:(l=St(a),e[2]=a,e[3]=l);let b;e[4]!==c||e[5]!==l?(b={defaultValues:c,resolver:l,reValidateMode:"onChange"},e[4]=c,e[5]=l,e[6]=b):b=e[6];let u=ft(b),f=(me=[...Ee,...Be].find(j=>j.schema===s))==null?void 0:me.connectionLibraries,[S,A]=(0,K.useState)((f==null?void 0:f.preferred)??"sqlalchemy"),{createNewCell:g}=Me(),_=Tt(),v;e[7]!==g||e[8]!==_?(v=j=>{g({code:j,before:!1,cellId:_??"__end__",skipIfCodeExists:!0})},e[7]=g,e[8]=_,e[9]=v):v=e[9];let Y=v,H;e[10]!==Y||e[11]!==o||e[12]!==S?(H=j=>{Y(rs(j,S)),o()},e[10]=Y,e[11]=o,e[12]=S,e[13]=H):H=e[13];let X=H,I;e[14]!==u||e[15]!==X?(I=u.handleSubmit(X),e[14]=u,e[15]=X,e[16]=I):I=e[16];let M;e[17]===Symbol.for("react.memo_cache_sentinel")?(M=(0,r.jsx)(mt,{}),e[17]=M):M=e[17];let T;e[18]!==u||e[19]!==s?(T=(0,r.jsx)(ls,{children:(0,r.jsx)(It,{schema:s,form:u,renderers:fs,children:M})}),e[18]=u,e[19]=s,e[20]=T):T=e[20];let q;e[21]===n?q=e[22]:(q=(0,r.jsx)(ie,{type:"button",variant:"outline",onClick:n,children:"Back"}),e[21]=n,e[22]=q);let Z=!u.formState.isValid,D;e[23]===Z?D=e[24]:(D=(0,r.jsx)(ie,{type:"submit",disabled:Z,children:"Add"}),e[23]=Z,e[24]=D);let L;e[25]!==q||e[26]!==D?(L=(0,r.jsxs)("div",{className:"flex gap-2",children:[q,D]}),e[25]=q,e[26]=D,e[27]=L):L=e[27];let ee=nt,P;e[28]===Symbol.for("react.memo_cache_sentinel")?(P=j=>A(j),e[28]=P):P=e[28];let N;e[29]===Symbol.for("react.memo_cache_sentinel")?(N=(0,r.jsxs)("div",{className:"flex flex-col gap-1 items-end",children:[(0,r.jsx)(st,{children:(0,r.jsx)(Xe,{placeholder:"Select a library"})}),(0,r.jsx)("span",{className:"text-xs text-muted-foreground",children:"Preferred connection library"})]}),e[29]=N):N=e[29];let te=et,se=tt,ne=f==null?void 0:f.libraries.map(_s),F;e[30]!==se||e[31]!==ne?(F=(0,r.jsx)(se,{children:ne}),e[30]=se,e[31]=ne,e[32]=F):F=e[32];let E;e[33]!==te||e[34]!==F?(E=(0,r.jsx)(te,{children:F}),e[33]=te,e[34]=F,e[35]=E):E=e[35];let B;e[36]!==ee||e[37]!==S||e[38]!==P||e[39]!==N||e[40]!==E?(B=(0,r.jsx)("div",{children:(0,r.jsxs)(ee,{value:S,onValueChange:P,children:[N,E]})}),e[36]=ee,e[37]=S,e[38]=P,e[39]=N,e[40]=E,e[41]=B):B=e[41];let O;e[42]!==L||e[43]!==B?(O=(0,r.jsxs)("div",{className:"flex gap-2 justify-between",children:[L,B]}),e[42]=L,e[43]=B,e[44]=O):O=e[44];let Q;return e[45]!==T||e[46]!==O||e[47]!==I?(Q=(0,r.jsxs)("form",{onSubmit:I,className:"space-y-4",children:[T,O]}),e[45]=T,e[46]=O,e[47]=I,e[48]=Q):Q=e[48],Q},ys=t=>{let e=(0,W.c)(5),{onSubmit:s}=t,[o,n]=(0,K.useState)(null);if(!o){let l;return e[0]===Symbol.for("react.memo_cache_sentinel")?(l=(0,r.jsx)(bs,{onSelect:n}),e[0]=l):l=e[0],l}let c;e[1]===Symbol.for("react.memo_cache_sentinel")?(c=()=>n(null),e[1]=c):c=e[1];let a;return e[2]!==s||e[3]!==o?(a=(0,r.jsx)(gs,{schema:o,onSubmit:s,onBack:c}),e[2]=s,e[3]=o,e[4]=a):a=e[4],a};const Ss=t=>{let e=(0,W.c)(6),{children:s}=t,[o,n]=(0,K.useState)(!1),c;e[0]===s?c=e[1]:(c=(0,r.jsx)(xt,{asChild:!0,children:s}),e[0]=s,e[1]=c);let a;e[2]===Symbol.for("react.memo_cache_sentinel")?(a=(0,r.jsx)(Oe,{onClose:()=>n(!1)}),e[2]=a):a=e[2];let l;return e[3]!==o||e[4]!==c?(l=(0,r.jsxs)(Ct,{open:o,onOpenChange:n,children:[c,a]}),e[3]=o,e[4]=c,e[5]=l):l=e[5],l},Oe=t=>{let e=(0,W.c)(4),{onClose:s}=t,o;e[0]===Symbol.for("react.memo_cache_sentinel")?(o=(0,r.jsx)(wt,{children:"Add Connection"}),e[0]=o):o=e[0];let n;e[1]===Symbol.for("react.memo_cache_sentinel")?(n=(0,r.jsxs)(_t,{className:"mb-4",children:[o,(0,r.jsxs)(kt,{children:["Connect to your database or data catalog to query data directly from your notebook. Learn more about how to connect to your database in our"," ",(0,r.jsx)(Rt,{href:"https://docs.marimo.io/guides/working_with_data/sql/#connecting-to-a-custom-database",children:"docs."})]})]}),e[1]=n):n=e[1];let c;return e[2]===s?c=e[3]:(c=(0,r.jsxs)($t,{className:"max-h-[75vh] overflow-y-auto",children:[n,(0,r.jsx)(ys,{onSubmit:()=>s()})]}),e[2]=s,e[3]=c),c};function _s(t){return(0,r.jsx)(Ze,{value:t,children:Ne[t]},t)}export{Oe as n,Ss as t};

package/dist/assets/add-missing-import-CMKdLyLj.js

@@ -0,0 +1 @@
+import{n as c}from"./jotai-C7U7c6dd.js";import{Fr as d,Yn as i,y as f}from"./cells-D2eha_-B.js";import{t as s}from"./requests-CftYh9v2.js";function m({moduleName:t,variableName:l,onAddImport:r,appStore:a=c}){if(l in a.get(i))return!1;let{cellData:o,cellIds:e}=a.get(f),n=RegExp(`import[ 	]+${t}[ 	]+as[ 	]+${l}`,"g");for(let u of e.inOrderIds)if(n.test(o[u].code))return!1;return r(`import ${t} as ${l}`),!0}function I({autoInstantiate:t,createNewCell:l,fromCellId:r,before:a}){let o=s(),e=null;return m({moduleName:"marimo",variableName:"mo",onAddImport:n=>{e=d.create(),l({cellId:r??"__end__",before:a??!1,code:n,lastCodeRun:t?n:void 0,newCellId:e,skipIfCodeExists:!0,autoFocus:!1}),t&&o.sendRun({cellIds:[e],codes:[n]})}})?e:null}function p({autoInstantiate:t,createNewCell:l,fromCellId:r}){let a=s(),o=null;return m({moduleName:"altair",variableName:"alt",onAddImport:e=>{o=d.create(),l({cellId:r??"__end__",before:!1,code:e,lastCodeRun:t?e:void 0,newCellId:o,skipIfCodeExists:!0,autoFocus:!1}),t&&a.sendRun({cellIds:[o],codes:[e]})}})?o:null}export{I as n,p as t};