drizzle-cube 0.4.19 → 0.4.21

package/dist/adapters/hono/index.js

@@ -1,261 +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 = (C) => {
-  const y = {
+var re = (j) => {
+  const d = {
     ...{
       origin: "*",
       allowMethods: ["GET", "HEAD", "PUT", "POST", "DELETE", "PATCH"],
       allowHeaders: [],
       exposeHeaders: []
     },
-    ...C
-  }, H = /* @__PURE__ */ ((l) => typeof l == "string" ? l === "*" ? () => l : (i) => l === i ? i : null : typeof l == "function" ? l : (i) => l.includes(i) ? i : null)(y.origin), f = ((l) => typeof l == "function" ? l : Array.isArray(l) ? () => l : () => [])(y.allowMethods);
-  return async function(i, m) {
-    function g(q, a) {
-      i.res.headers.set(q, 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 v = await H(i.req.header("origin") || "", i);
-    if (v && g("Access-Control-Allow-Origin", v), y.credentials && g("Access-Control-Allow-Credentials", "true"), y.exposeHeaders?.length && g("Access-Control-Expose-Headers", y.exposeHeaders.join(",")), i.req.method === "OPTIONS") {
-      y.origin !== "*" && g("Vary", "Origin"), y.maxAge != null && g("Access-Control-Max-Age", y.maxAge.toString());
-      const q = await f(i.req.header("origin") || "", i);
-      q.length && g("Access-Control-Allow-Methods", q.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 && (g("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(C) {
+function te(j) {
   const {
-    cubes: j,
-    drizzle: y,
-    schema: H,
-    extractSecurityContext: f,
+    cubes: x,
+    drizzle: d,
+    schema: M,
+    extractSecurityContext: m,
     engineType: l,
     cors: i,
-    basePath: m = "/cubejs-api/v1",
-    cache: g,
-    mcp: v = { enabled: !0 },
-    agent: q
-  } = C;
-  if (!j || j.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,
+  const s = new K({
+    drizzle: d,
+    schema: M,
     engineType: l,
-    cache: g
+    cache: C
   });
-  if (j.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);
     }
-  }), q && a.post(`${m}/agent/chat`, async (r) => {
+  }), p && a.post(`${h}/agent/chat`, async (e) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-BV4JuWNW.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 = (q.apiKey || "").trim();
-      if (q.allowClientApiKey) {
-        const w = r.req.header("x-agent-api-key");
-        w && (u = w.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), x = new TextEncoder(), p = new ReadableStream({
-        async start(w) {
+      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 b = e({
+            const w = r({
               message: o,
               sessionId: t,
               history: c,
               semanticLayer: s,
-              securityContext: h,
-              agentConfig: q,
-              apiKey: u
+              securityContext: E,
+              agentConfig: p,
+              apiKey: u,
+              systemContext: O,
+              providerOverride: q,
+              modelOverride: A,
+              baseURLOverride: f
             });
-            for await (const d of b) {
-              const R = `data: ${JSON.stringify(d)}
+            for await (const v of w) {
+              const R = `data: ${JSON.stringify(v)}
 
 `;
-              w.enqueue(x.encode(R));
+              g.enqueue(y.encode(R));
             }
-          } catch (b) {
-            const d = {
+          } catch (w) {
+            const v = {
               type: "error",
-              data: { message: b instanceof Error ? b.message : "Stream failed" }
+              data: { message: w instanceof Error ? w.message : "Stream failed" }
             };
-            w.enqueue(x.encode(`data: ${JSON.stringify(d)}
+            g.enqueue(y.encode(`data: ${JSON.stringify(v)}
 
 `));
           } finally {
-            w.close();
+            g.close();
           }
         }
       });
-      return new Response(p, {
+      return new Response(S, {
         status: 200,
         headers: {
           "Content-Type": "text/event-stream",
@@ -263,86 +267,86 @@ function te(C) {
           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);
     }
-  }), v.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 = v.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"),
-        v.allowedOrigins ? { allowedOrigins: v.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 x = await t.req.json().catch(() => null), p = _(x);
-      if (!p)
-        return t.json(M(null, -32600, "Invalid JSON-RPC 2.0 request"), 400);
-      const w = U(u), b = 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 d = 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 R = W(p.id ?? null, d), P = b && d && typeof d == "object" && "sessionId" in d ? d.sessionId : void 0, S = {};
-        if (P && (S[Y] = P), w) {
-          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(R, 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(R, 200, S);
-      } catch (d) {
-        if (O(p))
-          return console.error("MCP notification processing error:", d), t.body(null, 202);
-        console.error("MCP RPC error:", d);
-        const R = d?.code ?? -32603, P = d?.data, S = d.message || "MCP request failed", E = M(p.id ?? null, R, S, P);
-        if (w) {
-          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, {
@@ -354,28 +358,28 @@ function te(C) {
             }
           });
         }
-        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 x = 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(x, {
+      return new Response(A, {
         status: 200,
         headers: {
           "Content-Type": "text/event-stream",
@@ -387,13 +391,13 @@ function te(C) {
   }
   return a;
 }
-function ne(C, j) {
-  const y = te(j);
-  return C.route("/", y), C;
+function ne(j, x) {
+  const d = te(x);
+  return j.route("/", d), j;
 }
-function ie(C) {
-  const j = new L();
-  return ne(j, C);
+function ie(j) {
+  const x = new L();
+  return ne(x, j);
 }
 export {
   ie as createCubeApp,