drizzle-cube 0.4.20 → 0.4.21

package/dist/adapters/hono/index.js

@@ -1,262 +1,265 @@
 import { Hono as L } from "hono";
-import { S as V, v as B, b as M, a as z, n as K, p as _, w as U, d as F, g as G, f as X, i as O, c as W, M as Y, e as T, s as I } from "../mcp-transport-m1X1GtwG.js";
+import { S as K, v as V, b as $, a as B, n as z, p as _, w as U, d as F, g as G, f as X, i as I, c as W, M as Y, e as N, s as T } from "../mcp-transport-m1X1GtwG.js";
 import { formatCubeResponse as k, handleBatchRequest as Z, formatMetaResponse as ee, formatSqlResponse as D, handleDryRun as J } from "../utils.js";
-var re = (w) => {
-  const y = {
+var re = (j) => {
+  const d = {
     ...{
       origin: "*",
       allowMethods: ["GET", "HEAD", "PUT", "POST", "DELETE", "PATCH"],
       allowHeaders: [],
       exposeHeaders: []
     },
-    ...w
-  }, H = /* @__PURE__ */ ((d) => typeof d == "string" ? d === "*" ? () => d : (i) => d === i ? i : null : typeof d == "function" ? d : (i) => d.includes(i) ? i : null)(y.origin), f = ((d) => typeof d == "function" ? d : Array.isArray(d) ? () => d : () => [])(y.allowMethods);
-  return async function(i, m) {
-    function q(g, a) {
-      i.res.headers.set(g, a);
+    ...j
+  }, M = /* @__PURE__ */ ((l) => typeof l == "string" ? l === "*" ? () => l : (i) => l === i ? i : null : typeof l == "function" ? l : (i) => l.includes(i) ? i : null)(d.origin), m = ((l) => typeof l == "function" ? l : Array.isArray(l) ? () => l : () => [])(d.allowMethods);
+  return async function(i, h) {
+    function C(p, a) {
+      i.res.headers.set(p, a);
     }
-    const x = await H(i.req.header("origin") || "", i);
-    if (x && q("Access-Control-Allow-Origin", x), y.credentials && q("Access-Control-Allow-Credentials", "true"), y.exposeHeaders?.length && q("Access-Control-Expose-Headers", y.exposeHeaders.join(",")), i.req.method === "OPTIONS") {
-      y.origin !== "*" && q("Vary", "Origin"), y.maxAge != null && q("Access-Control-Max-Age", y.maxAge.toString());
-      const g = await f(i.req.header("origin") || "", i);
-      g.length && q("Access-Control-Allow-Methods", g.join(","));
-      let a = y.allowHeaders;
+    const b = await M(i.req.header("origin") || "", i);
+    if (b && C("Access-Control-Allow-Origin", b), d.credentials && C("Access-Control-Allow-Credentials", "true"), d.exposeHeaders?.length && C("Access-Control-Expose-Headers", d.exposeHeaders.join(",")), i.req.method === "OPTIONS") {
+      d.origin !== "*" && C("Vary", "Origin"), d.maxAge != null && C("Access-Control-Max-Age", d.maxAge.toString());
+      const p = await m(i.req.header("origin") || "", i);
+      p.length && C("Access-Control-Allow-Methods", p.join(","));
+      let a = d.allowHeaders;
       if (!a?.length) {
         const s = i.req.header("Access-Control-Request-Headers");
         s && (a = s.split(/\s*,\s*/));
       }
-      return a?.length && (q("Access-Control-Allow-Headers", a.join(",")), i.res.headers.append("Vary", "Access-Control-Request-Headers")), i.res.headers.delete("Content-Length"), i.res.headers.delete("Content-Type"), new Response(null, {
+      return a?.length && (C("Access-Control-Allow-Headers", a.join(",")), i.res.headers.append("Vary", "Access-Control-Request-Headers")), i.res.headers.delete("Content-Length"), i.res.headers.delete("Content-Type"), new Response(null, {
         headers: i.res.headers,
         status: 204,
         statusText: "No Content"
       });
     }
-    await m(), y.origin !== "*" && i.header("Vary", "Origin", { append: !0 });
+    await h(), d.origin !== "*" && i.header("Vary", "Origin", { append: !0 });
   };
 };
-function te(w) {
+function te(j) {
   const {
-    cubes: C,
-    drizzle: y,
-    schema: H,
-    extractSecurityContext: f,
-    engineType: d,
+    cubes: x,
+    drizzle: d,
+    schema: M,
+    extractSecurityContext: m,
+    engineType: l,
     cors: i,
-    basePath: m = "/cubejs-api/v1",
-    cache: q,
-    mcp: x = { enabled: !0 },
-    agent: g
-  } = w;
-  if (!C || C.length === 0)
+    basePath: h = "/cubejs-api/v1",
+    cache: C,
+    mcp: b = { enabled: !0 },
+    agent: p
+  } = j;
+  if (!x || x.length === 0)
     throw new Error("At least one cube must be provided in the cubes array");
   const a = new L();
   i && a.use("/*", re(i));
-  const s = new V({
-    drizzle: y,
-    schema: H,
-    engineType: d,
-    cache: q
+  const s = new K({
+    drizzle: d,
+    schema: M,
+    engineType: l,
+    cache: C
   });
-  if (C.forEach((r) => {
-    s.registerCube(r);
-  }), a.post(`${m}/load`, async (r) => {
+  if (x.forEach((e) => {
+    s.registerCube(e);
+  }), a.post(`${h}/load`, async (e) => {
     try {
-      const e = await r.req.json(), n = e.query || e, o = await f(r), t = s.validateQuery(n);
+      const r = await e.req.json(), n = r.query || r, o = await m(e), t = s.validateQuery(n);
       if (!t.isValid)
-        return r.json({
+        return e.json({
           error: `Query validation failed: ${t.errors.join(", ")}`
         }, 400);
-      const c = r.req.header("x-cache-control") === "no-cache", u = await s.executeMultiCubeQuery(n, o, { skipCache: c });
-      return r.json(k(n, u, s));
-    } catch (e) {
-      return console.error("Query execution error:", e), r.json({
-        error: e instanceof Error ? e.message : "Query execution failed"
+      const c = e.req.header("x-cache-control") === "no-cache", u = await s.executeMultiCubeQuery(n, o, { skipCache: c });
+      return e.json(k(n, u, s));
+    } catch (r) {
+      return console.error("Query execution error:", r), e.json({
+        error: r instanceof Error ? r.message : "Query execution failed"
       }, 500);
     }
-  }), a.get(`${m}/load`, async (r) => {
+  }), a.get(`${h}/load`, async (e) => {
     try {
-      const e = r.req.query("query");
-      if (!e)
-        return r.json({
+      const r = e.req.query("query");
+      if (!r)
+        return e.json({
           error: "Query parameter is required"
         }, 400);
       let n;
       try {
-        n = JSON.parse(e);
+        n = JSON.parse(r);
       } catch {
-        return r.json({
+        return e.json({
           error: "Invalid JSON in query parameter"
         }, 400);
       }
-      const o = await f(r), t = s.validateQuery(n);
+      const o = await m(e), t = s.validateQuery(n);
       if (!t.isValid)
-        return r.json({
+        return e.json({
           error: `Query validation failed: ${t.errors.join(", ")}`
         }, 400);
-      const c = r.req.header("x-cache-control") === "no-cache", u = await s.executeMultiCubeQuery(n, o, { skipCache: c });
-      return r.json(k(n, u, s));
-    } catch (e) {
-      return console.error("Query execution error:", e), r.json({
-        error: e instanceof Error ? e.message : "Query execution failed"
+      const c = e.req.header("x-cache-control") === "no-cache", u = await s.executeMultiCubeQuery(n, o, { skipCache: c });
+      return e.json(k(n, u, s));
+    } catch (r) {
+      return console.error("Query execution error:", r), e.json({
+        error: r instanceof Error ? r.message : "Query execution failed"
       }, 500);
     }
-  }), a.post(`${m}/batch`, async (r) => {
+  }), a.post(`${h}/batch`, async (e) => {
     try {
-      const e = await r.req.json(), { queries: n } = e;
+      const r = await e.req.json(), { queries: n } = r;
       if (!n || !Array.isArray(n))
-        return r.json({
+        return e.json({
           error: 'Request body must contain a "queries" array'
         }, 400);
       if (n.length === 0)
-        return r.json({
+        return e.json({
           error: "Queries array cannot be empty"
         }, 400);
-      const o = await f(r), t = r.req.header("x-cache-control") === "no-cache", c = await Z(n, o, s, { skipCache: t });
-      return r.json(c);
-    } catch (e) {
-      return console.error("Batch execution error:", e), r.json({
-        error: e instanceof Error ? e.message : "Batch execution failed"
+      const o = await m(e), t = e.req.header("x-cache-control") === "no-cache", c = await Z(n, o, s, { skipCache: t });
+      return e.json(c);
+    } catch (r) {
+      return console.error("Batch execution error:", r), e.json({
+        error: r instanceof Error ? r.message : "Batch execution failed"
       }, 500);
     }
-  }), a.get(`${m}/meta`, (r) => {
+  }), a.get(`${h}/meta`, (e) => {
     try {
-      const e = s.getMetadata();
-      return r.json(ee(e));
-    } catch (e) {
-      return console.error("Metadata error:", e), r.json({
-        error: e instanceof Error ? e.message : "Failed to fetch metadata"
+      const r = s.getMetadata();
+      return e.json(ee(r));
+    } catch (r) {
+      return console.error("Metadata error:", r), e.json({
+        error: r instanceof Error ? r.message : "Failed to fetch metadata"
       }, 500);
     }
-  }), a.post(`${m}/sql`, async (r) => {
+  }), a.post(`${h}/sql`, async (e) => {
     try {
-      const e = await r.req.json(), n = await f(r), o = s.validateQuery(e);
+      const r = await e.req.json(), n = await m(e), o = s.validateQuery(r);
       if (!o.isValid)
-        return r.json({
+        return e.json({
           error: `Query validation failed: ${o.errors.join(", ")}`
         }, 400);
-      const t = e.measures?.[0] || e.dimensions?.[0];
+      const t = r.measures?.[0] || r.dimensions?.[0];
       if (!t)
-        return r.json({
+        return e.json({
           error: "No measures or dimensions specified"
         }, 400);
-      const c = t.split(".")[0], u = await s.generateSQL(c, e, n);
-      return r.json(D(e, u));
-    } catch (e) {
-      return console.error("SQL generation error:", e), r.json({
-        error: e instanceof Error ? e.message : "SQL generation failed"
+      const c = t.split(".")[0], u = await s.generateSQL(c, r, n);
+      return e.json(D(r, u));
+    } catch (r) {
+      return console.error("SQL generation error:", r), e.json({
+        error: r instanceof Error ? r.message : "SQL generation failed"
       }, 500);
     }
-  }), a.get(`${m}/sql`, async (r) => {
+  }), a.get(`${h}/sql`, async (e) => {
     try {
-      const e = r.req.query("query");
-      if (!e)
-        return r.json({
+      const r = e.req.query("query");
+      if (!r)
+        return e.json({
           error: "Query parameter is required"
         }, 400);
-      const n = JSON.parse(e), o = await f(r), t = s.validateQuery(n);
+      const n = JSON.parse(r), o = await m(e), t = s.validateQuery(n);
       if (!t.isValid)
-        return r.json({
+        return e.json({
           error: `Query validation failed: ${t.errors.join(", ")}`
         }, 400);
       const c = n.measures?.[0] || n.dimensions?.[0];
       if (!c)
-        return r.json({
+        return e.json({
           error: "No measures or dimensions specified"
         }, 400);
-      const u = c.split(".")[0], h = await s.generateSQL(u, n, o);
-      return r.json(D(n, h));
-    } catch (e) {
-      return console.error("SQL generation error:", e), r.json({
-        error: e instanceof Error ? e.message : "SQL generation failed"
+      const u = c.split(".")[0], q = await s.generateSQL(u, n, o);
+      return e.json(D(n, q));
+    } catch (r) {
+      return console.error("SQL generation error:", r), e.json({
+        error: r instanceof Error ? r.message : "SQL generation failed"
       }, 500);
     }
-  }), a.post(`${m}/dry-run`, async (r) => {
+  }), a.post(`${h}/dry-run`, async (e) => {
     try {
-      const e = await r.req.json(), n = e.query || e, o = await f(r), t = await J(n, o, s);
-      return r.json(t);
-    } catch (e) {
-      return console.error("Dry-run error:", e), r.json({
-        error: e instanceof Error ? e.message : "Dry-run validation failed",
+      const r = await e.req.json(), n = r.query || r, o = await m(e), t = await J(n, o, s);
+      return e.json(t);
+    } catch (r) {
+      return console.error("Dry-run error:", r), e.json({
+        error: r instanceof Error ? r.message : "Dry-run validation failed",
         valid: !1
       }, 400);
     }
-  }), a.get(`${m}/dry-run`, async (r) => {
+  }), a.get(`${h}/dry-run`, async (e) => {
     try {
-      const e = r.req.query("query");
-      if (!e)
-        return r.json({
+      const r = e.req.query("query");
+      if (!r)
+        return e.json({
           error: "Query parameter is required",
           valid: !1
         }, 400);
-      const n = JSON.parse(e), o = await f(r), t = await J(n, o, s);
-      return r.json(t);
-    } catch (e) {
-      return console.error("Dry-run error:", e), r.json({
-        error: e instanceof Error ? e.message : "Dry-run validation failed",
+      const n = JSON.parse(r), o = await m(e), t = await J(n, o, s);
+      return e.json(t);
+    } catch (r) {
+      return console.error("Dry-run error:", r), e.json({
+        error: r instanceof Error ? r.message : "Dry-run validation failed",
         valid: !1
       }, 400);
     }
-  }), a.post(`${m}/explain`, async (r) => {
+  }), a.post(`${h}/explain`, async (e) => {
     try {
-      const e = await r.req.json(), n = e.query || e, o = e.options || {}, t = await f(r), c = s.validateQuery(n);
+      const r = await e.req.json(), n = r.query || r, o = r.options || {}, t = await m(e), c = s.validateQuery(n);
       if (!c.isValid)
-        return r.json({
+        return e.json({
           error: `Query validation failed: ${c.errors.join(", ")}`
         }, 400);
       const u = await s.explainQuery(n, t, o);
-      return r.json(u);
-    } catch (e) {
-      return console.error("Explain error:", e), r.json({
-        error: e instanceof Error ? e.message : "Explain query failed"
+      return e.json(u);
+    } catch (r) {
+      return console.error("Explain error:", r), e.json({
+        error: r instanceof Error ? r.message : "Explain query failed"
       }, 500);
     }
-  }), g && a.post(`${m}/agent/chat`, async (r) => {
+  }), p && a.post(`${h}/agent/chat`, async (e) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-CbDMdSY5.js"), n = await r.req.json(), { message: o, sessionId: t, history: c } = n;
+      const { handleAgentChat: r } = await import("../handler-9Rdn7zM2.js"), n = await e.req.json(), { message: o, sessionId: t, history: c } = n;
       if (!o || typeof o != "string")
-        return r.json({ error: "message is required and must be a string" }, 400);
-      let u = (g.apiKey || "").trim();
-      if (g.allowClientApiKey) {
-        const j = r.req.header("x-agent-api-key");
-        j && (u = j.trim());
+        return e.json({ error: "message is required and must be a string" }, 400);
+      let u = (p.apiKey || "").trim();
+      if (p.allowClientApiKey) {
+        const g = e.req.header("x-agent-api-key");
+        g && (u = g.trim());
       }
       if (!u)
-        return r.json({
+        return e.json({
           error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."
         }, 401);
-      const h = await f(r), R = g.buildSystemContext?.(h), p = new TextEncoder(), P = new ReadableStream({
-        async start(j) {
+      const q = p.allowClientApiKey ? e.req.header("x-agent-provider") : void 0, A = p.allowClientApiKey ? e.req.header("x-agent-model") : void 0, f = p.allowClientApiKey ? e.req.header("x-agent-provider-endpoint") : void 0, E = await m(e), O = p.buildSystemContext?.(E), y = new TextEncoder(), S = new ReadableStream({
+        async start(g) {
           try {
-            const l = e({
+            const w = r({
               message: o,
               sessionId: t,
               history: c,
               semanticLayer: s,
-              securityContext: h,
-              agentConfig: g,
+              securityContext: E,
+              agentConfig: p,
               apiKey: u,
-              systemContext: R
+              systemContext: O,
+              providerOverride: q,
+              modelOverride: A,
+              baseURLOverride: f
             });
-            for await (const v of l) {
-              const b = `data: ${JSON.stringify(v)}
+            for await (const v of w) {
+              const R = `data: ${JSON.stringify(v)}
 
 `;
-              j.enqueue(p.encode(b));
+              g.enqueue(y.encode(R));
             }
-          } catch (l) {
+          } catch (w) {
             const v = {
               type: "error",
-              data: { message: l instanceof Error ? l.message : "Stream failed" }
+              data: { message: w instanceof Error ? w.message : "Stream failed" }
             };
-            j.enqueue(p.encode(`data: ${JSON.stringify(v)}
+            g.enqueue(y.encode(`data: ${JSON.stringify(v)}
 
 `));
           } finally {
-            j.close();
+            g.close();
           }
         }
       });
-      return new Response(P, {
+      return new Response(S, {
         status: 200,
         headers: {
           "Content-Type": "text/event-stream",
@@ -264,86 +267,86 @@ function te(w) {
           Connection: "keep-alive"
         }
       });
-    } catch (e) {
-      return console.error("Agent chat error:", e), r.json({
-        error: e instanceof Error ? e.message : "Agent chat failed"
+    } catch (r) {
+      return console.error("Agent chat error:", r), e.json({
+        error: r instanceof Error ? r.message : "Agent chat failed"
       }, 500);
     }
-  }), x.enabled !== !1) {
-    const r = {
+  }), b.enabled !== !1) {
+    const e = {
       uri: "drizzle-cube://schema",
       name: "Cube Schema",
       description: "Current cube metadata as JSON",
       mimeType: "application/json",
       text: JSON.stringify(s.getMetadata(), null, 2)
-    }, e = [...X(), r], n = G(), o = x.basePath ?? "/mcp";
+    }, r = [...X(), e], n = G(), o = b.basePath ?? "/mcp";
     a.post(`${o}`, async (t) => {
-      const c = B(
+      const c = V(
         t.req.header("origin"),
-        x.allowedOrigins ? { allowedOrigins: x.allowedOrigins } : {}
+        b.allowedOrigins ? { allowedOrigins: b.allowedOrigins } : {}
       );
       if (!c.valid)
-        return t.json(M(null, -32600, c.reason), 403);
+        return t.json($(null, -32600, c.reason), 403);
       const u = t.req.header("accept");
-      if (!z(u))
-        return t.json(M(null, -32600, "Accept header must include both application/json and text/event-stream"), 400);
-      const h = K(t.req.header());
-      if (!h.ok)
+      if (!B(u))
+        return t.json($(null, -32600, "Accept header must include both application/json and text/event-stream"), 400);
+      const q = z(t.req.header());
+      if (!q.ok)
         return t.json({
           error: "Unsupported MCP protocol version",
-          supported: h.supported
+          supported: q.supported
         }, 426);
-      const R = await t.req.json().catch(() => null), p = _(R);
-      if (!p)
-        return t.json(M(null, -32600, "Invalid JSON-RPC 2.0 request"), 400);
-      const P = U(u), j = p.method === "initialize";
+      const A = await t.req.json().catch(() => null), f = _(A);
+      if (!f)
+        return t.json($(null, -32600, "Invalid JSON-RPC 2.0 request"), 400);
+      const E = U(u), O = f.method === "initialize";
       try {
-        const l = await F(
-          p.method,
-          p.params,
+        const y = await F(
+          f.method,
+          f.params,
           {
             semanticLayer: s,
-            extractSecurityContext: f,
+            extractSecurityContext: m,
             rawRequest: t,
             rawResponse: null,
-            negotiatedProtocol: h.negotiated,
-            resources: e,
+            negotiatedProtocol: q.negotiated,
+            resources: r,
             prompts: n
           }
         );
-        if (O(p))
+        if (I(f))
           return t.body(null, 202);
-        const v = W(p.id ?? null, l), b = j && l && typeof l == "object" && "sessionId" in l ? l.sessionId : void 0, S = {};
-        if (b && (S[Y] = b), P) {
-          const E = new TextEncoder(), A = T(), $ = new ReadableStream({
+        const S = W(f.id ?? null, y), g = O && y && typeof y == "object" && "sessionId" in y ? y.sessionId : void 0, w = {};
+        if (g && (w[Y] = g), E) {
+          const v = new TextEncoder(), R = N(), P = new ReadableStream({
             start(Q) {
-              Q.enqueue(E.encode(`id: ${A}
+              Q.enqueue(v.encode(`id: ${R}
 
-`)), Q.enqueue(E.encode(I(v, A))), Q.close();
+`)), Q.enqueue(v.encode(T(S, R))), Q.close();
             }
           });
-          return new Response($, {
+          return new Response(P, {
             status: 200,
             headers: {
               "Content-Type": "text/event-stream",
               "Cache-Control": "no-cache",
               Connection: "keep-alive",
-              ...S
+              ...w
             }
           });
         }
-        return t.json(v, 200, S);
-      } catch (l) {
-        if (O(p))
-          return console.error("MCP notification processing error:", l), t.body(null, 202);
-        console.error("MCP RPC error:", l);
-        const v = l?.code ?? -32603, b = l?.data, S = l.message || "MCP request failed", E = M(p.id ?? null, v, S, b);
-        if (P) {
-          const A = new TextEncoder(), $ = T(), Q = new ReadableStream({
-            start(N) {
-              N.enqueue(A.encode(`id: ${$}
+        return t.json(S, 200, w);
+      } catch (y) {
+        if (I(f))
+          return console.error("MCP notification processing error:", y), t.body(null, 202);
+        console.error("MCP RPC error:", y);
+        const S = y?.code ?? -32603, g = y?.data, w = y.message || "MCP request failed", v = $(f.id ?? null, S, w, g);
+        if (E) {
+          const R = new TextEncoder(), P = N(), Q = new ReadableStream({
+            start(H) {
+              H.enqueue(R.encode(`id: ${P}
 
-`)), N.enqueue(A.encode(I(E, $))), N.close();
+`)), H.enqueue(R.encode(T(v, P))), H.close();
             }
           });
           return new Response(Q, {
@@ -355,28 +358,28 @@ function te(w) {
             }
           });
         }
-        return t.json(E, 200);
+        return t.json(v, 200);
       }
     }), a.delete(`${o}`, (t) => t.json({ error: "Session termination not supported" }, 405)), a.get(`${o}`, (t) => {
-      const c = new TextEncoder(), u = T();
-      let h;
-      const R = new ReadableStream({
-        start(p) {
-          p.enqueue(c.encode(I({
+      const c = new TextEncoder(), u = N();
+      let q;
+      const A = new ReadableStream({
+        start(f) {
+          f.enqueue(c.encode(T({
             jsonrpc: "2.0",
             method: "mcp/ready",
             params: { protocol: "streamable-http" }
-          }, u, 15e3))), h = setInterval(() => {
-            p.enqueue(c.encode(`: keep-alive
+          }, u, 15e3))), q = setInterval(() => {
+            f.enqueue(c.encode(`: keep-alive
 
 `));
           }, 15e3);
         },
         cancel() {
-          clearInterval(h);
+          clearInterval(q);
         }
       });
-      return new Response(R, {
+      return new Response(A, {
         status: 200,
         headers: {
           "Content-Type": "text/event-stream",
@@ -388,13 +391,13 @@ function te(w) {
   }
   return a;
 }
-function ne(w, C) {
-  const y = te(C);
-  return w.route("/", y), w;
+function ne(j, x) {
+  const d = te(x);
+  return j.route("/", d), j;
 }
-function ie(w) {
-  const C = new L();
-  return ne(C, w);
+function ie(j) {
+  const x = new L();
+  return ne(x, j);
 }
 export {
   ie as createCubeApp,