@marimo-team/frontend 0.20.5-dev3 → 0.20.5-dev35

package/src/plugins/impl/mpl-interactive/__tests__/mpl-websocket-shim.test.ts

@@ -0,0 +1,110 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+import { describe, expect, it, vi } from "vitest";
+import { MplCommWebSocket } from "../mpl-websocket-shim";
+
+describe("MplCommWebSocket", () => {
+  it("starts in OPEN state", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    expect(ws.readyState).toBe(WebSocket.OPEN);
+  });
+
+  it("send() parses JSON and calls sendFn with parsed object", () => {
+    const sendFn = vi.fn();
+    const ws = new MplCommWebSocket(sendFn);
+
+    ws.send(JSON.stringify({ type: "resize", width: 640, height: 480 }));
+
+    expect(sendFn).toHaveBeenCalledOnce();
+    expect(sendFn).toHaveBeenCalledWith({
+      type: "resize",
+      width: 640,
+      height: 480,
+    });
+  });
+
+  it("receiveJson() dispatches MessageEvent with JSON string data", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    const handler = vi.fn();
+    ws.onmessage = handler;
+
+    ws.receiveJson({ type: "figure_size", size: [640, 480] });
+
+    expect(handler).toHaveBeenCalledOnce();
+    const event: MessageEvent = handler.mock.calls[0][0];
+    expect(event).toBeInstanceOf(MessageEvent);
+    expect(event.type).toBe("message");
+    expect(JSON.parse(event.data as string)).toEqual({
+      type: "figure_size",
+      size: [640, 480],
+    });
+  });
+
+  it("receiveJson() is a no-op if onmessage is not set", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    ws.receiveJson({ type: "test" });
+    expect(ws.onmessage).toBeNull();
+  });
+
+  it("receiveBinary() dispatches MessageEvent with Blob data", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    const handler = vi.fn();
+    ws.onmessage = handler;
+
+    // Simulate a PNG-like binary buffer
+    const bytes = new Uint8Array([0x89, 0x50, 0x4e, 0x47, 0, 1, 2, 3]);
+    const dv = new DataView(bytes.buffer);
+
+    ws.receiveBinary(dv);
+
+    expect(handler).toHaveBeenCalledOnce();
+    const event: MessageEvent = handler.mock.calls[0][0];
+    expect(event).toBeInstanceOf(MessageEvent);
+    expect(event.data).toBeInstanceOf(Blob);
+    expect((event.data as Blob).size).toBe(8);
+  });
+
+  it("receiveBinary() handles DataView with offset", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    const handler = vi.fn();
+    ws.onmessage = handler;
+
+    // Create a DataView that's a slice of a larger buffer
+    const fullBuffer = new ArrayBuffer(16);
+    const fullView = new Uint8Array(fullBuffer);
+    fullView.set([0, 0, 0, 0, 0x89, 0x50, 0x4e, 0x47, 0, 1, 2, 3, 0, 0, 0, 0]);
+    const dv = new DataView(fullBuffer, 4, 8);
+
+    ws.receiveBinary(dv);
+
+    expect(handler).toHaveBeenCalledOnce();
+    const blob = handler.mock.calls[0][0].data as Blob;
+    expect(blob.size).toBe(8);
+  });
+
+  it("close() sets readyState to CLOSED and fires onclose", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    const closeHandler = vi.fn();
+    ws.onclose = closeHandler;
+
+    ws.close();
+
+    expect(ws.readyState).toBe(WebSocket.CLOSED);
+    expect(closeHandler).toHaveBeenCalledOnce();
+  });
+
+  it("close() does not throw if onclose is not set", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    ws.close();
+    expect(ws.readyState).toBe(WebSocket.CLOSED);
+  });
+
+  it("onopen can be triggered externally", () => {
+    const ws = new MplCommWebSocket(vi.fn());
+    const openHandler = vi.fn();
+    ws.onopen = openHandler;
+
+    ws.onopen?.();
+
+    expect(openHandler).toHaveBeenCalledOnce();
+  });
+});

package/src/plugins/impl/mpl-interactive/mpl-websocket-shim.ts

@@ -0,0 +1,57 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+/**
+ * Fake WebSocket that routes messages through MarimoComm / MODEL_MANAGER
+ * instead of a real network WebSocket.
+ *
+ * mpl.js expects a WebSocket-like object with:
+ *   - readyState
+ *   - send(data: string)
+ *   - onopen / onmessage / onclose callbacks
+ */
+export class MplCommWebSocket {
+  readyState: number = WebSocket.OPEN;
+  private sendFn: (msg: unknown) => void;
+
+  onopen: (() => void) | null = null;
+  onmessage: ((evt: MessageEvent) => void) | null = null;
+  onclose: (() => void) | null = null;
+
+  constructor(sendFn: (msg: unknown) => void) {
+    this.sendFn = sendFn;
+  }
+
+  /**
+   * Called by mpl.js to send a message to the backend.
+   * mpl.js always sends JSON strings.
+   */
+  send(data: string): void {
+    this.sendFn(JSON.parse(data));
+  }
+
+  /**
+   * Called when the backend pushes a JSON message via the model custom event.
+   */
+  receiveJson(data: unknown): void {
+    this.onmessage?.(
+      new MessageEvent("message", { data: JSON.stringify(data) }),
+    );
+  }
+
+  /**
+   * Called when the backend pushes binary data (PNG render) via model custom event.
+   */
+  receiveBinary(buffer: DataView): void {
+    const ab = buffer.buffer.slice(
+      buffer.byteOffset,
+      buffer.byteOffset + buffer.byteLength,
+    ) as ArrayBuffer;
+    const blob = new Blob([ab]);
+    this.onmessage?.(new MessageEvent("message", { data: blob }));
+  }
+
+  close(): void {
+    this.readyState = WebSocket.CLOSED;
+    this.onclose?.();
+  }
+}

package/src/plugins/plugins.ts

@@ -26,6 +26,7 @@ import { MatrixPlugin } from "./impl/MatrixPlugin";
 import { MicrophonePlugin } from "./impl/MicrophonePlugin";
 import { MultiselectPlugin } from "./impl/MultiselectPlugin";
 import { MatplotlibPlugin } from "./impl/matplotlib/MatplotlibPlugin";
+import { MplInteractivePlugin } from "./impl/mpl-interactive/MplInteractivePlugin";
 import { NumberPlugin } from "./impl/NumberPlugin";
 import { PanelPlugin } from "./impl/panel/PanelPlugin";
 import { PlotlyPlugin } from "./impl/plotly/PlotlyPlugin";
@@ -92,6 +93,7 @@ export const UI_PLUGINS: IPlugin<any, unknown>[] = [
   AnyWidgetPlugin,
   DataEditorPlugin,
   PanelPlugin,
+  MplInteractivePlugin,
 ];
 
 // List of output / layout plugins

package/src/utils/__tests__/filenames.test.ts

@@ -23,6 +23,12 @@ describe("Filenames", () => {
     expect(Filenames.toPNG("test.foo.py")).toEqual("test.foo.png");
   });
 
+  it("should convert filename to ipynb", () => {
+    expect(Filenames.toIPYNB("test")).toEqual("test.ipynb");
+    expect(Filenames.toIPYNB("test.txt")).toEqual("test.ipynb");
+    expect(Filenames.toIPYNB("test.foo.py")).toEqual("test.foo.ipynb");
+  });
+
   it("should remove extension from filename", () => {
     expect(Filenames.withoutExtension("test")).toEqual("test");
     expect(Filenames.withoutExtension("test.txt")).toEqual("test");
@@ -39,6 +45,7 @@ describe("Filenames", () => {
     expect(Filenames.toHTML(filename)).toEqual(`${withoutExt}.html`);
     expect(Filenames.toPNG(filename)).toEqual(`${withoutExt}.png`);
     expect(Filenames.toPY(filename)).toEqual(`${withoutExt}.py`);
+    expect(Filenames.toIPYNB(filename)).toEqual(`${withoutExt}.ipynb`);
 
     // Ensure operations preserve unicode and special characters in base name
     expect(withoutExt).not.toEqual("");

package/src/utils/__tests__/smartMatch.test.ts

@@ -0,0 +1,61 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+import { describe, expect, it } from "vitest";
+import { smartMatch, smartMatchFilter } from "../smartMatch";
+
+describe("smartMatch", () => {
+  it("matches exact word", () => {
+    expect(smartMatch("run", "Run cell")).toBe(true);
+  });
+
+  it("matches word prefix", () => {
+    expect(smartMatch("exe", "execute start")).toBe(true);
+  });
+
+  it("matches across array of haystacks", () => {
+    expect(smartMatch("exe", ["Run", "execute start"])).toBe(true);
+  });
+
+  it("does not match unrelated terms", () => {
+    expect(smartMatch("xyz", "Run cell")).toBe(false);
+  });
+
+  it("empty search matches everything", () => {
+    expect(smartMatch("", "anything")).toBe(true);
+  });
+
+  it("multi-word needle requires all words to match", () => {
+    expect(smartMatch("run cell", "Run cell")).toBe(true);
+    expect(smartMatch("run xyz", "Run cell")).toBe(false);
+  });
+
+  it("is case-insensitive", () => {
+    expect(smartMatch("RUN", "run cell")).toBe(true);
+  });
+
+  it("skips null/undefined haystacks", () => {
+    expect(smartMatch("run", [null, undefined, "Run cell"])).toBe(true);
+    expect(smartMatch("run", [null, undefined])).toBe(false);
+  });
+});
+
+describe("smartMatchFilter", () => {
+  it("returns 1 for value match", () => {
+    expect(smartMatchFilter("Run cell", "run")).toBe(1);
+  });
+
+  it("returns 0 for no match", () => {
+    expect(smartMatchFilter("Run cell", "xyz")).toBe(0);
+  });
+
+  it("returns 0.8 for keyword-only match", () => {
+    expect(smartMatchFilter("Run", "execute", ["execute", "start"])).toBe(0.8);
+  });
+
+  it("returns 1 when value matches even if keywords also match", () => {
+    expect(smartMatchFilter("Run", "run", ["execute", "start"])).toBe(1);
+  });
+
+  it("does not match without relevant keywords", () => {
+    expect(smartMatchFilter("Run", "preferences", ["execute"])).toBe(0);
+  });
+});

package/src/utils/filenames.ts

@@ -15,6 +15,9 @@ export const Filenames = {
   toPY: (filename: string): string => {
     return Filenames.replace(filename, "py");
   },
+  toIPYNB: (filename: string): string => {
+    return Filenames.replace(filename, "ipynb");
+  },
   withoutExtension: (filename: string): string => {
     // Just remove the last extension
     const parts = filename.split(".");

package/src/utils/smartMatch.ts

@@ -0,0 +1,62 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+/**
+ * Split a string into lowercase words (letters/digits only).
+ */
+function words(s: string): string[] {
+  return s.toLowerCase().match(/[\da-z]+/g) || [];
+}
+
+/**
+ * Returns true when every word in `needle` is a prefix of at least one word
+ * in one of the `haystacks`.
+ *
+ * Examples:
+ *   smartMatch("run", "Run cell")               // true  – "run" prefixes "run"
+ *   smartMatch("exe", ["Run", "execute start"])  // true  – "exe" prefixes "execute"
+ *   smartMatch("xyz", "Run cell")                // false
+ */
+export function smartMatch(
+  needle: string,
+  haystackOrHaystacks: string | (string | null | undefined)[],
+): boolean {
+  const needleWords = words(needle);
+  if (needleWords.length === 0) {
+    return true; // empty search matches everything
+  }
+
+  const haystacks = Array.isArray(haystackOrHaystacks)
+    ? haystackOrHaystacks
+    : [haystackOrHaystacks];
+
+  // Collect all words from all haystacks
+  const haystackWords: string[] = [];
+  for (const h of haystacks) {
+    if (h) {
+      haystackWords.push(...words(h));
+    }
+  }
+
+  // Every needle word must be a prefix of at least one haystack word
+  return needleWords.every((nw) => {
+    return haystackWords.some((hw) => hw.startsWith(nw));
+  });
+}
+
+/**
+ * cmdk-compatible filter function.
+ * Returns 1 for a value match, 0.8 for a keyword-only match, 0 for no match.
+ */
+export function smartMatchFilter(
+  value: string,
+  search: string,
+  keywords?: string[],
+): number {
+  if (smartMatch(search, value)) {
+    return 1;
+  }
+  if (keywords && smartMatch(search, keywords)) {
+    return 0.8;
+  }
+  return 0;
+}

package/dist/assets/add-connection-dialog-9CzjZlY9.js

@@ -1,138 +0,0 @@
-var Ve=Object.defineProperty;var Ze=(t,e,s)=>e in t?Ve(t,e,{enumerable:!0,configurable:!0,writable:!0,value:s}):t[e]=s;var z=(t,e,s)=>Ze(t,typeof e!="symbol"?e+"":e,s);import{s as ne}from"./chunk-LvLJmgfZ.js";import{t as Qe}from"./react-Bj1aDYRI.js";import{Rn as g,T as Je,Tt as ce,rr as Ye}from"./cells-CCtxWKxf.js";import{D as K,P as y,R as u,T as N,_ as ae,i as Xe,k as _,m as et}from"./zod-H_cgTO0M.js";import{t as P}from"./compiler-runtime-B3qBwwSJ.js";import{t as H}from"./assertNever-Dw9aFN4h.js";import{t as tt}from"./_createAggregator-BjsxkJZj.js";import{h as st}from"./useEventListener-CjXriKgk.js";import{t as rt}from"./jsx-runtime-CP2fAiZX.js";import{t as Z}from"./button-BmOH9agK.js";import{t as W}from"./cn-d_A4rA-x.js";import{t as Q}from"./createLucideIcon-BvP5LBIs.js";import{t as ot}from"./circle-plus-DfyUnSPr.js";import{a as J,c as ie,o as nt,p as ct,r as at,t as it}from"./dropdown-menu-CKhrfxsx.js";import{t as lt}from"./database-zap-B3efE3T2.js";import{t as dt}from"./file-code-D9wtWSiR.js";import{t as pt}from"./file-DVwc11C3.js";import{t as ht}from"./github-BrsjF9dp.js";import{m as mt,o as ut,t as bt,u as a}from"./form-B7_NRa34.js";import{i as ft}from"./html-to-image-b9BimK8l.js";import{n as gt}from"./DeferredRequestRegistry-CMZsNx4R.js";import{o as yt,r as _t}from"./input-BPldEXfy.js";import{c as wt,d as xt,f as St,h as kt,l as vt,n as $t,p as jt,s as Ct,t as At,u as Rt,v as Tt}from"./azure-Ddr1pZpW.js";import{r as Lt}from"./useTheme-CwdqVOYu.js";import{c as Dt,i as It,n as Nt,r as qt,s as Ft,t as Pt}from"./select-DygbgNTD.js";import{a as Bt,c as Mt,l as Et,n as Ot,r as Ut,t as zt}from"./dialog-BXzvlof-.js";import{c as le,l as Kt,s as Ht,u as de}from"./popover-Cvl9xY6R.js";import{n as Wt}from"./ImperativeModal-CRNJOvqT.js";import{t as pe}from"./links-BKV6zi_7.js";import{n as Gt}from"./useAsyncData-DVm0BCFt.js";import{o as Vt}from"./focus-ShrwcKIv.js";import{t as Zt}from"./request-registry-CN4oD1xn.js";import{n as Qt,t as Jt}from"./write-secret-modal-DuxmGjDn.js";var Yt=tt(function(t,e,s){t[s?0:1].push(e)},function(){return[[],[]]}),Xt=Q("file-play",[["path",{d:"M6 22a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h8a2.4 2.4 0 0 1 1.704.706l3.588 3.588A2.4 2.4 0 0 1 20 8v12a2 2 0 0 1-2 2z",key:"1oefj6"}],["path",{d:"M14 2v5a1 1 0 0 0 1 1h5",key:"wfsgrz"}],["path",{d:"M15.033 13.44a.647.647 0 0 1 0 1.12l-4.065 2.352a.645.645 0 0 1-.968-.56v-4.704a.645.645 0 0 1 .967-.56z",key:"1tzo1f"}]]),es=Q("file-spreadsheet",[["path",{d:"M6 22a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h8a2.4 2.4 0 0 1 1.704.706l3.588 3.588A2.4 2.4 0 0 1 20 8v12a2 2 0 0 1-2 2z",key:"1oefj6"}],["path",{d:"M14 2v5a1 1 0 0 0 1 1h5",key:"wfsgrz"}],["path",{d:"M8 13h2",key:"yr2amv"}],["path",{d:"M14 13h2",key:"un5t4a"}],["path",{d:"M8 17h2",key:"2yhykz"}],["path",{d:"M14 17h2",key:"10kma7"}]]),Y=Q("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"}]]),X="env:";function zr(t){return t}function E(t){return typeof t=="string"?t.startsWith(X):!1}function ee(t){return`${X}${t}`}function he(t){return t.replace(X,"")}var ts=P(),B=ne(Qe(),1),n=ne(rt(),1),me=(0,B.createContext)({providerNames:[],secretKeys:[],loading:!1,error:void 0,refreshSecrets:st.NOOP});const ss=()=>(0,B.use)(me),rs=t=>{let e=(0,ts.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:i,refetch:c}=Gt(is,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]!==i||e[6]!==l||e[7]!==c||e[8]!==d||e[9]!==m?(p={secretKeys:d,providerNames:m,loading:l,error:i,refreshSecrets:c},e[5]=i,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)(me,{value:p,children:s}),e[11]=s,e[12]=p,e[13]=f):f=e[13],f},os={isMatch:t=>{if(t instanceof ae||t instanceof et){let{optionRegex:e}=a.parse(t.description||"");return!!e}return!1},Component:({schema:t,form:e,path:s})=>{let{secretKeys:o,providerNames:r,refreshSecrets:l}=ss(),{openModal:i,closeModal:c}=Wt(),{label:d,description:m,optionRegex:p=""}=a.parse(t.description||""),[f,h]=Yt(o,b=>new RegExp(p,"i").test(b));return(0,n.jsx)(Rt,{control:e.control,name:s,render:({field:b})=>(0,n.jsxs)(xt,{children:[(0,n.jsx)(St,{children:d}),(0,n.jsx)(wt,{children:m}),(0,n.jsx)(Ct,{children:(0,n.jsxs)("div",{className:"flex gap-2",children:[t instanceof ae?(0,n.jsx)(_t,{...b,value:b.value,onChange:b.onChange,className:W("flex-1")}):(0,n.jsx)(yt,{...b,value:b.value,onChange:b.onChange,className:"flex-1"}),(0,n.jsxs)(it,{children:[(0,n.jsx)(ct,{asChild:!0,children:(0,n.jsx)(Z,{variant:"outline",size:"icon",className:W(E(b.value)&&"bg-accent"),children:(0,n.jsx)(gt,{className:"h-3 w-3"})})}),(0,n.jsxs)(at,{align:"end",className:"max-h-60 overflow-y-auto",children:[(0,n.jsxs)(J,{onSelect:()=>{i((0,n.jsx)(Jt,{providerNames:r,onSuccess:w=>{l(),b.onChange(ee(w)),c()},onClose:c}))},children:[(0,n.jsx)(ot,{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)(nt,{children:"Recommended"})]}),f.map(w=>(0,n.jsx)(J,{onSelect:()=>b.onChange(ee(w)),children:w},w)),h.length>0&&(0,n.jsx)(ie,{}),h.map(w=>(0,n.jsx)(J,{onSelect:()=>b.onChange(ee(w)),children:w},w))]})]})]})}),(0,n.jsx)(jt,{})]})})}};function ns(t){return t.provider!=="env"}function cs(t){return t.name}function as(t){return t.keys}async function is(){let t=await Zt.request({}),e=Qt(t.secrets).filter(ns).map(cs);return{secretKeys:t.secrets.flatMap(as).sort(),providerNames:e}}var O=P(),ls=[os];const te=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},se=t=>{let e=(0,O.c)(9),{name:s,color:o,icon:r,onSelect:l}=t,i;e[0]===o?i=e[1]:(i={backgroundColor:o},e[0]=o,e[1]=i);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]!==i||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:i,onClick:l,children:[r,c]}),e[4]=r,e[5]=l,e[6]=i,e[7]=c,e[8]=d):d=e[8],d},ds=t=>{let e=(0,O.c)(25),{onBack:s,isValid:o,libraries:r,preferredLibrary:l,onLibraryChange:i,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)(Ft,{children:(0,n.jsx)(Dt,{placeholder:"Select a library"})}),e[7]=w):w=e[7];let S;e[8]===m?S=e[9]:(S=(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]=S);let R;if(e[10]!==c||e[11]!==r){let v;e[13]===c?v=e[14]:(v=C=>(0,n.jsx)(It,{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)(Nt,{children:(0,n.jsx)(qt,{children:R})}),e[15]=R,e[16]=$);let k;e[17]!==i||e[18]!==l||e[19]!==S||e[20]!==$?(k=(0,n.jsx)("div",{children:(0,n.jsxs)(Pt,{value:l,onValueChange:i,children:[S,$]})}),e[17]=i,e[18]=l,e[19]=S,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 ps(){let t=(0,O.c)(3),{createNewCell:e}=Je(),s=Vt(),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 ue=t=>{let e=(0,O.c)(30),{schema:s,libraries:o,preferredLibrary:r,displayNames:l,libraryLabel:i,generateCode:c,onSubmit:d,onBack:m}=t,p;e[0]===s?p=e[1]:(p=ut(s),e[0]=s,e[1]=p);let f=p,h;e[2]===s?h=e[3]:(h=kt(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=Tt(b),[S,R]=B.useState(r),$=ps(),k;e[7]!==c||e[8]!==$||e[9]!==d||e[10]!==S?(k=Ge=>{$(c(Ge,S)),d()},e[7]=c,e[8]=$,e[9]=d,e[10]=S,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)(vt,{}),e[15]=C):C=e[15];let q;e[16]!==w||e[17]!==s?(q=(0,n.jsx)(rs,{children:(0,n.jsx)(bt,{schema:s,form:w,renderers:ls,children:C})}),e[16]=w,e[17]=s,e[18]=q):q=e[18];let F;e[19]!==l||e[20]!==w.formState.isValid||e[21]!==o||e[22]!==i||e[23]!==m||e[24]!==S?(F=(0,n.jsx)(ds,{onBack:m,isValid:w.formState.isValid,libraries:o,preferredLibrary:S,onLibraryChange:R,displayNames:l,libraryLabel:i}),e[19]=l,e[20]=w.formState.isValid,e[21]=o,e[22]=i,e[23]=m,e[24]=S,e[25]=F):F=e[25];let U;return e[26]!==v||e[27]!==q||e[28]!==F?(U=(0,n.jsxs)("form",{onSubmit:v,className:"space-y-4",children:[q,F]}),e[26]=v,e[27]=q,e[28]=F,e[29]=U):U=e[29],U};function L(){return u().optional().describe(a.of({label:"Password",inputType:"password",placeholder:"password",optionRegex:".*password.*"}))}function re(t,e){let s=u();return s=e?s.nonempty():s.optional(),s=s.describe(a.of({label:t||"Token",inputType:"password",placeholder:"token",optionRegex:".*token.*"})),s}function G(){return u().optional().describe(a.of({label:"Warehouse Name",placeholder:"warehouse",optionRegex:".*warehouse.*"}))}function oe(t,e){let s=u();return s=e?s.nonempty():s.optional(),s.describe(a.of({label:t||"URI",optionRegex:".*uri.*"}))}function A(t){return u().nonempty().describe(a.of({label:t||"Host",placeholder:"localhost",optionRegex:".*host.*"}))}function j(){return u().describe(a.of({label:"Database",placeholder:"db name",optionRegex:".*database.*"}))}function hs(){return u().describe(a.of({label:"Schema",placeholder:"schema name",optionRegex:".*schema.*"}))}function D(){return u().nonempty().describe(a.of({label:"Username",placeholder:"username",optionRegex:".*username.*"}))}function I(t){let e=Xe().describe(a.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 be(){return N().default(!1).describe(a.of({label:"Read Only"}))}const fe=y({type:_("postgres"),host:A(),port:I(5432).optional(),database:j(),username:D(),password:L(),ssl:N().default(!1).describe(a.of({label:"Use SSL"}))}).describe(a.of({direction:"two-columns"})),ge=y({type:_("mysql"),host:A(),port:I(3306),database:j(),username:D(),password:L(),ssl:N().default(!1).describe(a.of({label:"Use SSL"}))}).describe(a.of({direction:"two-columns"})),ye=y({type:_("sqlite"),database:j().describe(a.of({label:"Database Path"}))}).describe(a.of({direction:"two-columns"})),_e=y({type:_("duckdb"),database:j().describe(a.of({label:"Database Path"})),read_only:be()}).describe(a.of({direction:"two-columns"})),we=y({type:_("motherduck"),database:j().default("my_db").describe(a.of({label:"Database Name"})),token:re()}).describe(a.of({direction:"two-columns"})),xe=y({type:_("snowflake"),account:u().nonempty().describe(a.of({label:"Account",optionRegex:".*snowflake.*"})),warehouse:u().optional().describe(a.of({label:"Warehouse",optionRegex:".*snowflake.*"})),database:j(),schema:u().optional().describe(a.of({label:"Schema",optionRegex:".*snowflake.*"})),username:D(),password:L(),role:u().optional().describe(a.of({label:"Role"}))}).describe(a.of({direction:"two-columns"})),Se=y({type:_("bigquery"),project:u().nonempty().describe(a.of({label:"Project ID",optionRegex:".*bigquery.*"})),dataset:u().nonempty().describe(a.of({label:"Dataset",optionRegex:".*bigquery.*"})),credentials_json:u().describe(a.of({label:"Credentials JSON",inputType:"textarea"}))}).describe(a.of({direction:"two-columns"})),ke=y({type:_("clickhouse_connect"),host:A(),port:I(8123).optional(),username:D(),password:L(),secure:N().default(!1).describe(a.of({label:"Use HTTPs"})),proxy_path:u().optional().describe(a.of({label:"Proxy Path",placeholder:"/clickhouse"}))}).describe(a.of({direction:"two-columns"})),ve=y({type:_("timeplus"),host:A().default("localhost"),port:I(8123).optional(),username:D().default("default"),password:L().default("")}).describe(a.of({direction:"two-columns"})),$e=y({type:_("chdb"),database:j().describe(a.of({label:"Database Path"})),read_only:be()}).describe(a.of({direction:"two-columns"})),je=y({type:_("trino"),host:A(),port:I(8080),database:j(),schema:hs().optional(),username:D(),password:L(),async_support:N().default(!1).describe(a.of({label:"Async Support"}))}).describe(a.of({direction:"two-columns"})),Ce=y({type:_("iceberg"),name:u().describe(a.of({label:"Catalog Name"})),catalog:K("type",[y({type:_("REST"),warehouse:G(),uri:u().optional().describe(a.of({label:"URI",placeholder:"https://",optionRegex:".*uri.*"})),token:re()}),y({type:_("SQL"),warehouse:G(),uri:u().optional().describe(a.of({label:"URI",placeholder:"jdbc:iceberg://host:port/database",optionRegex:".*uri.*"}))}),y({type:_("Hive"),warehouse:G(),uri:oe()}),y({type:_("Glue"),warehouse:G(),uri:oe()}),y({type:_("DynamoDB"),"dynamodb.profile-name":u().optional().describe(a.of({label:"Profile Name"})),"dynamodb.region":u().optional().describe(a.of({label:"Region"})),"dynamodb.access-key-id":u().optional().describe(a.of({label:"Access Key ID"})),"dynamodb.secret-access-key":u().optional().describe(a.of({label:"Secret Access Key",inputType:"password"})),"dynamodb.session-token":u().optional().describe(a.of({label:"Session Token",inputType:"password"}))})]).default({type:"REST",token:void 0}).describe(a.of({special:"tabs"}))}),Ae=y({type:_("datafusion"),sessionContext:N().optional().describe(a.of({label:"Use Session Context"}))}),Re=y({type:_("pyspark"),host:A().optional(),port:I().optional()}),Te=y({type:_("redshift"),host:A(),port:I(5439),connectionType:K("type",[y({type:_("IAM credentials"),region:u().describe(a.of({label:"Region"})),aws_access_key_id:u().nonempty().describe(a.of({label:"AWS Access Key ID",inputType:"password",optionRegex:".*aws_access_key_id.*"})),aws_secret_access_key:u().nonempty().describe(a.of({label:"AWS Secret Access Key",inputType:"password",optionRegex:".*aws_secret_access_key.*"})),aws_session_token:u().optional().describe(a.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(a.of({direction:"two-columns"})),Le=y({type:_("databricks"),access_token:re("Access Token",!0),server_hostname:A("Server Hostname"),http_path:oe("HTTP Path",!0),catalog:u().optional().describe(a.of({label:"Catalog"})),schema:u().optional().describe(a.of({label:"Schema"}))}).describe(a.of({direction:"two-columns"})),De=y({type:_("supabase"),host:A(),port:I(5432).optional(),database:j(),username:D(),password:L(),disable_client_pooling:N().default(!1).describe(a.of({label:"Disable Client-Side Pooling"}))}).describe(a.of({direction:"two-columns"})),ms=K("type",[fe,ge,ye,_e,we,xe,Se,ke,ve,$e,je,Ce,Ae,Re,Te,Le,De]),Ie={sqlmodel:"SQLModel",sqlalchemy:"SQLAlchemy",duckdb:"DuckDB",clickhouse_connect:"ClickHouse Connect",chdb:"chDB",pyiceberg:"PyIceberg",ibis:"Ibis",motherduck:"MotherDuck",redshift:"Redshift",databricks:"Databricks"};var x=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}},us=t=>`_${t}`,bs=class{constructor(){z(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e,s){if(t=us(t),E(e)){let o=he(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"?M(e):e?`"${e}"`:""}printInFString(t,e,s){if(e===void 0)return"";if(typeof e=="number")return`${e}`;if(typeof e=="boolean")return M(e);let 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 E(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(`
-`)}},fs=class extends x{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})
-    `)}},gs=class extends x{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})
-    `)}},ys=class extends x{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)
-    `)}},_s=class extends x{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}`)},
-        )
-      )
-    `)}},ws=class extends x{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)
-    `)}},xs=class extends x{generateImports(){return[]}generateConnectionCode(){return g(`
-      DATABASE_URL = "${this.secrets.printInFString("database",this.connection.database||":memory:")}"
-      engine = ${this.orm}.connect(DATABASE_URL, read_only=${M(this.connection.read_only)})
-    `)}},Ss=class extends x{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}")
-        `)}},ks=class extends x{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}`)},
-      )
-    `)}},vs=class extends x{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 x{generateImports(){return["import chdb"]}generateConnectionCode(){let t=this.secrets.print("database",this.connection.database)||'""';return g(`
-      engine = ${this.orm}.connect(${t}, read_only=${M(this.connection.read_only)})
-    `)}},js=class extends x{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),i=this.secrets.printPassword(this.connection.password,"TRINO_PASSWORD",!0);return g(`
-      engine = ${this.orm}.create_engine(f"${t}://${s}:${i}@${o}:${r}/${l}${e}")
-    `)}},Cs=class extends x{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:H(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))E(r)?t[o]=this.secrets.print(o,r):typeof r=="string"&&(t[o]=`"${r}"`);let e=qs(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:H(this.connection.catalog)}}},As=class extends x{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()
-    `)}},Rs=class extends x{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()
-    `)}},Ts=class extends x{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),i=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:i,...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}`)},
-      )
-    `)}},Ls=class extends x{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,i=`databricks://token:${e}@${s}?http_path=${o}`;return r&&(i+=`&catalog=${r}`),l&&(i+=`&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"${i}"
-      engine = ${this.orm}.create_engine(DATABASE_URL)
-      `)}},Ds=class extends x{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})
-    `)}},Is=class{constructor(){z(this,"secrets",new bs)}createGenerator(t,e){switch(t.type){case"postgres":return new fs(t,e,this.secrets);case"mysql":return new gs(t,e,this.secrets);case"sqlite":return new ys(t,e,this.secrets);case"snowflake":return new _s(t,e,this.secrets);case"bigquery":return new ws(t,e,this.secrets);case"duckdb":return new xs(t,e,this.secrets);case"motherduck":return new Ss(t,e,this.secrets);case"clickhouse_connect":return new ks(t,e,this.secrets);case"timeplus":return new vs(t,e,this.secrets);case"chdb":return new $s(t,e,this.secrets);case"trino":return new js(t,e,this.secrets);case"iceberg":return new Cs(t,e,this.secrets);case"datafusion":return new As(t,e,this.secrets);case"pyspark":return new Rs(t,e,this.secrets);case"redshift":return new Ts(t,e,this.secrets);case"databricks":return new Ls(t,e,this.secrets);case"supabase":return new Ds(t,e,this.secrets);default:H(t)}}};function Ns(t,e){if(!(e in Ie))throw Error(`Unsupported library: ${e}`);ms.parse(t);let s=new Is,o=s.createGenerator(t,e),r=o.generateConnectionCode(),l=s.secrets,i=[...new Set([...l.imports,...o.imports])].sort();i.push("");let c=l.formatSecrets();return c&&i.push(c),i.push(r.trim()),i.join(`
-`)}function M(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}=${M(o)}`:`${s}=${o}`)).join(`,
-`)}function qs(t,e){return Object.entries(t).filter(([,s])=>s!=null&&s!=="").map(([s,o])=>{let r=`"${s}"`;return e(typeof o=="boolean"?`${r}: ${M(o)}`:`${r}: ${o}`)}).join(`,
-`)}var Ne=P(),qe=[{name:"PostgreSQL",schema:fe,color:"#336791",logo:"postgres",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"MySQL",schema:ge,color:"#00758F",logo:"mysql",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"SQLite",schema:ye,color:"#003B57",logo:"sqlite",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DuckDB",schema:_e,color:"#FFD700",logo:"duckdb",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"MotherDuck",schema:we,color:"#ff9538",logo:"motherduck",connectionLibraries:{libraries:["duckdb"],preferred:"duckdb"}},{name:"Snowflake",schema:xe,color:"#29B5E8",logo:"snowflake",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"ClickHouse",schema:ke,color:"#2C2C1D",logo:"clickhouse",connectionLibraries:{libraries:["clickhouse_connect"],preferred:"clickhouse_connect"}},{name:"Timeplus",schema:ve,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:$e,color:"#f2b611",logo:"clickhouse",connectionLibraries:{libraries:["chdb"],preferred:"chdb"}},{name:"Trino",schema:je,color:"#d466b6",logo:"trino",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}},{name:"DataFusion",schema:Ae,color:"#202A37",logo:"datafusion",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"PySpark",schema:Re,color:"#1C5162",logo:"pyspark",connectionLibraries:{libraries:["ibis"],preferred:"ibis"}},{name:"Redshift",schema:Te,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:De,color:"#238F5F",logo:"supabase",connectionLibraries:{libraries:["sqlalchemy","sqlmodel"],preferred:"sqlalchemy"}}],Fe=[{name:"Iceberg",schema:Ce,color:"#000000",logo:"iceberg",connectionLibraries:{libraries:["pyiceberg"],preferred:"pyiceberg"}}],Fs=[...qe,...Fe],Ps=t=>{let e=(0,Ne.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)(se,{name:p,color:h,icon:(0,n.jsx)(ce,{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)(te,{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 i;e[5]===s?i=e[6]:(i=Fe.map(m=>{let{name:p,schema:f,color:h,logo:b}=m;return(0,n.jsx)(se,{name:p,color:h,icon:(0,n.jsx)(ce,{name:b,className:"w-8 h-8 text-white brightness-0 invert dark:invert"}),onSelect:()=>s(f)},p)}),e[5]=s,e[6]=i);let c;e[7]===i?c=e[8]:(c=(0,n.jsx)(te,{children:i}),e[7]=i,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 Bs=t=>{let e=(0,Ne.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)(Ps,{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 i;e[3]===r?i=e[4]:(i=Fs.find(h=>h.schema===r),e[3]=r,e[4]=i);let c=i==null?void 0:i.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)(ue,{schema:r,libraries:d,preferredLibrary:m,displayNames:Ie,libraryLabel:"Preferred connection library",generateCode:Ms,onSubmit:s,onBack:p}),e[8]=s,e[9]=r,e[10]=d,e[11]=m,e[12]=f):f=e[12],f};function Ms(t,e){return Ns(t,e)}var Es="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",Os="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",Us="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",zs="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",Ks="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",Hs="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",Ws=P();function Gs(t){var e;switch((e=t.split(".").pop())==null?void 0:e.toLowerCase()){case"png":case"jpg":case"jpeg":case"gif":case"svg":case"webp":return(0,n.jsx)(ft,{className:"h-3.5 w-3.5 text-purple-500"});case"csv":case"parquet":case"arrow":case"xlsx":return(0,n.jsx)(es,{className:"h-3.5 w-3.5 text-green-500"});case"py":case"js":case"ts":case"json":return(0,n.jsx)(dt,{className:"h-3.5 w-3.5 text-blue-500"});case"mp4":case"mpeg":return(0,n.jsx)(Xt,{className:"h-3.5 w-3.5 text-orange-500"});case"txt":case"md":case"log":return(0,n.jsx)(Ye,{className:"h-3.5 w-3.5 text-muted-foreground"});default:return(0,n.jsx)(pt,{className:"h-3.5 w-3.5 text-muted-foreground"})}}var Vs={s3:{src:$t,dark:Es},coreweave:{src:Us,dark:zs},cloudflare:{src:Os},azure:{src:At},gcs:{src:Ks},http:mt,file:Y,"in-memory":lt,gdrive:{src:Hs},github:ht};const Pe=t=>{let e=(0,Ws.c)(11),{protocol:s,forceDark:o,className:r}=t,{theme:l}=Lt(),i=Vs[s.toLowerCase()]??Y;if("src"in i){let p=i.src;i.dark&&(o||l==="dark")&&(p=i.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=i,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},Be=y({type:_("s3"),bucket:u().nonempty().describe(a.of({label:"Bucket",placeholder:"my-bucket"})),region:u().optional().describe(a.of({label:"Region",placeholder:"us-east-1"})),access_key_id:u().optional().describe(a.of({label:"Access Key ID",inputType:"password",optionRegex:".*access_key.*"})),secret_access_key:u().optional().describe(a.of({label:"Secret Access Key",inputType:"password",optionRegex:".*secret.*access.*"})),endpoint_url:u().optional().describe(a.of({label:"Endpoint URL",placeholder:"https://s3.amazonaws.com"}))}).describe(a.of({direction:"two-columns"})),Me=y({type:_("gcs"),bucket:u().nonempty().describe(a.of({label:"Bucket",placeholder:"my-bucket"})),service_account_key:u().optional().describe(a.of({label:"Service Account Key (JSON)",inputType:"textarea"}))}).describe(a.of({direction:"two-columns"})),Ee=y({type:_("azure"),container:u().nonempty().describe(a.of({label:"Container",placeholder:"my-container"})),account_name:u().nonempty().describe(a.of({label:"Account Name",placeholder:"storageaccount",optionRegex:".*account.*"})),account_key:u().optional().describe(a.of({label:"Account Key",inputType:"password",optionRegex:".*azure.*key.*"}))}).describe(a.of({direction:"two-columns"})),Oe=y({type:_("coreweave"),bucket:u().nonempty().describe(a.of({label:"Bucket",placeholder:"bucket-name"})),region:u().nonempty().describe(a.of({label:"Region",placeholder:"US-EAST-04A"})),access_key_id:u().optional().describe(a.of({label:"Access Key ID",inputType:"password",optionRegex:".*object_storage_key.*"})),secret_access_key:u().optional().describe(a.of({label:"Secret Access Key",inputType:"password",optionRegex:".*object_storage_secret.*"}))}).describe(a.of({direction:"two-columns"})),Ue=y({type:_("gdrive"),credentials_json:u().optional().describe(a.of({label:"Service Account JSON",description:"Leave empty to use browser-based authentication",inputType:"textarea"}))}).describe(a.of({direction:"two-columns"})),Zs=K("type",[Be,Me,Ee,Oe,Ue]),Qs={obstore:"Obstore",fsspec:"fsspec"};var Js=class{constructor(){z(this,"secrets",{})}get imports(){return Object.keys(this.secrets).length===0?new Set:new Set(["import os"])}print(t,e){if(e===void 0||e==="")return"";let s=`_${t}`;if(E(e)){let o=he(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 Ys(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 Xs(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 er(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(i=>`    ${i}`).join(`
-`)}
-)`}}function tr(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 sr(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 rr(t,e){Zs.parse(t);let s=new Js,o;switch(t.type){case"s3":o=Ys(t,s);break;case"gcs":o=Xs(t,s);break;case"azure":o=er(t,s);break;case"coreweave":o=tr(t,s);break;case"gdrive":o=sr(t,s);break;default:H(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 ze=P(),or="#232F3E",Ke=[{name:"Amazon S3",schema:Be,protocol:"s3",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Cloud Storage",schema:Me,protocol:"gcs",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Azure Blob Storage",schema:Ee,protocol:"azure",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"CoreWeave",schema:Oe,protocol:"coreweave",storageLibraries:{libraries:["obstore"],preferred:"obstore"}},{name:"Google Drive",schema:Ue,protocol:"gdrive",storageLibraries:{libraries:["fsspec"],preferred:"fsspec"}}],nr=t=>{let e=(0,ze.c)(4),{onSelect:s}=t,o;e[0]===s?o=e[1]:(o=Ke.map(l=>{let{name:i,schema:c,protocol:d}=l;return(0,n.jsx)(se,{name:i,color:or,icon:(0,n.jsx)(Pe,{protocol:d,forceDark:!0,className:"w-7.5 h-7.5"}),onSelect:()=>s(c)},i)}),e[0]=s,e[1]=o);let r;return e[2]===o?r=e[3]:(r=(0,n.jsx)(te,{children:o}),e[2]=o,e[3]=r),r};const cr=t=>{let e=(0,ze.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)(nr,{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 i;e[3]===r?i=e[4]:(i=Ke.find(h=>h.schema===r),e[3]=r,e[4]=i);let c=i==null?void 0:i.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)(ue,{schema:r,libraries:d,preferredLibrary:m,displayNames:Qs,libraryLabel:"Preferred storage library",generateCode:ar,onSubmit:s,onBack:p}),e[8]=s,e[9]=r,e[10]=d,e[11]=m,e[12]=f):f=e[12],f};function ar(t,e){return rr(t,e)}var He=P();const ir=t=>{let e=(0,He.c)(9),{children:s,defaultTab:o}=t,r=o===void 0?"databases":o,[l,i]=(0,B.useState)(!1),c;e[0]===s?c=e[1]:(c=(0,n.jsx)(Et,{asChild:!0,children:s}),e[0]=s,e[1]=c);let d;e[2]===Symbol.for("react.memo_cache_sentinel")?(d=()=>i(!1),e[2]=d):d=e[2];let m;e[3]===r?m=e[4]:(m=(0,n.jsx)(We,{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)(zt,{open:l,onOpenChange:i,children:[c,m]}),e[5]=l,e[6]=c,e[7]=m,e[8]=p):p=e[8],p},We=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)(Kt,{className:"w-full mb-4",children:[(0,n.jsx)(de,{value:"databases",className:"flex-1",children:"Databases & Catalogs"}),(0,n.jsx)(de,{value:"storage",className:"flex-1",children:"Remote Storages"})]}),e[0]=l):l=e[0];let i=l,c;e[1]===Symbol.for("react.memo_cache_sentinel")?(c=(0,n.jsx)(Mt,{children:"Add Connection"}),e[1]=c):c=e[1];let d;e[2]===Symbol.for("react.memo_cache_sentinel")?(d=(0,n.jsx)(pe,{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)(Bt,{children:[c,(0,n.jsxs)(Ut,{children:["Connect to a"," ",d," ","or"," ",(0,n.jsx)(pe,{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)(le,{value:"databases",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,n.jsx)(Bs,{onSubmit:o,header:i})}),e[4]=o,e[5]=p);let f;e[6]===o?f=e[7]:(f=(0,n.jsx)(le,{value:"storage",className:"mt-0 focus-visible:ring-0 focus-visible:ring-offset-0",children:(0,n.jsx)(cr,{onSubmit:o,header:i})}),e[6]=o,e[7]=f);let h;return e[8]!==r||e[9]!==p||e[10]!==f?(h=(0,n.jsxs)(Ot,{className:"max-h-[75vh] overflow-y-auto",children:[m,(0,n.jsxs)(Ht,{defaultValue:r,children:[p,f]})]}),e[8]=r,e[9]=p,e[10]=f,e[11]=h):h=e[11],h};export{Y as a,Gs as i,We as n,Pe as r,ir as t};