drizzle-cube 0.4.18 → 0.4.20

package/dist/adapters/nextjs/index.js

@@ -1,7 +1,7 @@
 import { NextResponse as t } from "next/server";
 import { e as T, s as A, v as D, b as P, a as _, n as Q, p as V, w as I, d as J, i as M, M as k, c as $, S as K } from "../mcp-transport-m1X1GtwG.js";
 import { formatErrorResponse as i, formatCubeResponse as z, formatMetaResponse as G, formatSqlResponse as U, handleDryRun as q, handleBatchRequest as B, handleDiscover as F, handleSuggest as X, handleValidate as W, handleLoad as Y } from "../utils.js";
-function E(n) {
+function x(n) {
   const { cubes: a, drizzle: s, schema: c, engineType: d, cache: o } = n;
   if (!a || a.length === 0)
     throw new Error("At least one cube must be provided in the cubes array");
@@ -15,13 +15,13 @@ function E(n) {
     r.registerCube(e);
   }), r;
 }
-function g(n, a) {
+function j(n, a) {
   const s = n.headers.get("origin"), c = {};
   return a.origin && (typeof a.origin == "string" ? c["Access-Control-Allow-Origin"] = a.origin : Array.isArray(a.origin) ? s && a.origin.includes(s) && (c["Access-Control-Allow-Origin"] = s) : typeof a.origin == "function" && s && a.origin(s) && (c["Access-Control-Allow-Origin"] = s)), a.methods && (c["Access-Control-Allow-Methods"] = a.methods.join(", ")), a.allowedHeaders && (c["Access-Control-Allow-Headers"] = a.allowedHeaders.join(", ")), a.credentials && (c["Access-Control-Allow-Credentials"] = "true"), c;
 }
 function ue(n) {
   return async function(s) {
-    const c = g(s, n);
+    const c = j(s, n);
     return new Response(null, {
       status: 200,
       headers: c
@@ -29,7 +29,7 @@ function ue(n) {
   };
 }
 function Z(n) {
-  const { extractSecurityContext: a, cors: s } = n, c = E(n);
+  const { extractSecurityContext: a, cors: s } = n, c = x(n);
   return async function(o, r) {
     try {
       let e;
@@ -56,15 +56,15 @@ function Z(n) {
           i("Method not allowed", 405),
           { status: 405 }
         );
-      const u = await a(o, r), y = c.validateQuery(e);
-      if (!y.isValid)
+      const u = await a(o, r), h = c.validateQuery(e);
+      if (!h.isValid)
         return t.json(
-          i(`Query validation failed: ${y.errors.join(", ")}`, 400),
+          i(`Query validation failed: ${h.errors.join(", ")}`, 400),
           { status: 400 }
         );
-      const l = o.headers.get("x-cache-control") === "no-cache", f = await c.executeMultiCubeQuery(e, u, { skipCache: l }), v = z(e, f, c);
-      return t.json(v, {
-        headers: s ? g(o, s) : {}
+      const l = o.headers.get("x-cache-control") === "no-cache", y = await c.executeMultiCubeQuery(e, u, { skipCache: l }), S = z(e, y, c);
+      return t.json(S, {
+        headers: s ? j(o, s) : {}
       });
     } catch (e) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js load handler error:", e), t.json(
@@ -78,12 +78,12 @@ function Z(n) {
   };
 }
 function ee(n) {
-  const { cors: a } = n, s = E(n);
+  const { cors: a } = n, s = x(n);
   return async function(d, o) {
     try {
       const r = s.getMetadata(), e = G(r);
       return t.json(e, {
-        headers: a ? g(d, a) : {}
+        headers: a ? j(d, a) : {}
       });
     } catch (r) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js meta handler error:", r), t.json(
@@ -97,22 +97,22 @@ function ee(n) {
   };
 }
 function te(n) {
-  const { extractSecurityContext: a, cors: s } = n, c = E(n);
+  const { extractSecurityContext: a, cors: s } = n, c = x(n);
   return async function(o, r) {
     try {
       let e;
       if (o.method === "POST") {
-        const C = await o.json();
-        e = C.query || C;
+        const E = await o.json();
+        e = E.query || E;
       } else if (o.method === "GET") {
-        const C = o.nextUrl.searchParams.get("query");
-        if (!C)
+        const E = o.nextUrl.searchParams.get("query");
+        if (!E)
           return t.json(
             i("Query parameter is required", 400),
             { status: 400 }
           );
         try {
-          e = JSON.parse(C);
+          e = JSON.parse(E);
         } catch {
           return t.json(
             i("Invalid JSON in query parameter", 400),
@@ -124,10 +124,10 @@ function te(n) {
           i("Method not allowed", 405),
           { status: 405 }
         );
-      const u = await a(o, r), y = c.validateQuery(e);
-      if (!y.isValid)
+      const u = await a(o, r), h = c.validateQuery(e);
+      if (!h.isValid)
         return t.json(
-          i(`Query validation failed: ${y.errors.join(", ")}`, 400),
+          i(`Query validation failed: ${h.errors.join(", ")}`, 400),
           { status: 400 }
         );
       const l = e.measures?.[0] || e.dimensions?.[0];
@@ -136,9 +136,9 @@ function te(n) {
           i("No measures or dimensions specified", 400),
           { status: 400 }
         );
-      const f = l.split(".")[0], v = await c.generateSQL(f, e, u), p = U(e, v);
+      const y = l.split(".")[0], S = await c.generateSQL(y, e, u), p = U(e, S);
       return t.json(p, {
-        headers: s ? g(o, s) : {}
+        headers: s ? j(o, s) : {}
       });
     } catch (e) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js SQL handler error:", e), t.json(
@@ -152,7 +152,7 @@ function te(n) {
   };
 }
 function re(n) {
-  const { extractSecurityContext: a, cors: s } = n, c = E(n);
+  const { extractSecurityContext: a, cors: s } = n, c = x(n);
   return async function(o, r) {
     try {
       let e;
@@ -179,9 +179,9 @@ function re(n) {
           { error: "Method not allowed", valid: !1 },
           { status: 405 }
         );
-      const u = await a(o, r), y = await q(e, u, c);
-      return t.json(y, {
-        headers: s ? g(o, s) : {}
+      const u = await a(o, r), h = await q(e, u, c);
+      return t.json(h, {
+        headers: s ? j(o, s) : {}
       });
     } catch (e) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js dry-run handler error:", e), t.json(
@@ -195,7 +195,7 @@ function re(n) {
   };
 }
 function ne(n) {
-  const { extractSecurityContext: a, cors: s } = n, c = E(n);
+  const { extractSecurityContext: a, cors: s } = n, c = x(n);
   return async function(o, r) {
     try {
       if (o.method !== "POST")
@@ -214,9 +214,9 @@ function ne(n) {
           i("Queries array cannot be empty", 400),
           { status: 400 }
         );
-      const y = await a(o, r), l = o.headers.get("x-cache-control") === "no-cache", f = await B(u, y, c, { skipCache: l });
-      return t.json(f, {
-        headers: s ? g(o, s) : {}
+      const h = await a(o, r), l = o.headers.get("x-cache-control") === "no-cache", y = await B(u, h, c, { skipCache: l });
+      return t.json(y, {
+        headers: s ? j(o, s) : {}
       });
     } catch (e) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js batch handler error:", e), t.json(
@@ -230,7 +230,7 @@ function ne(n) {
   };
 }
 function se(n) {
-  const { extractSecurityContext: a, cors: s } = n, c = E(n);
+  const { extractSecurityContext: a, cors: s } = n, c = x(n);
   return async function(o, r) {
     try {
       if (o.method !== "POST")
@@ -238,19 +238,19 @@ function se(n) {
           { error: "Method not allowed" },
           { status: 405 }
         );
-      const e = await o.json(), u = e.query || e, y = e.options || {}, l = await a(o, r), f = c.validateQuery(u);
-      if (!f.isValid)
+      const e = await o.json(), u = e.query || e, h = e.options || {}, l = await a(o, r), y = c.validateQuery(u);
+      if (!y.isValid)
         return t.json(
-          { error: `Query validation failed: ${f.errors.join(", ")}` },
+          { error: `Query validation failed: ${y.errors.join(", ")}` },
           { status: 400 }
         );
-      const v = await c.explainQuery(
+      const S = await c.explainQuery(
         u,
         l,
-        y
+        h
       );
-      return t.json(v, {
-        headers: s ? g(o, s) : {}
+      return t.json(S, {
+        headers: s ? j(o, s) : {}
       });
     } catch (e) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js explain handler error:", e), t.json(
@@ -261,7 +261,7 @@ function se(n) {
   };
 }
 function le(n) {
-  const { cors: a } = n, s = E(n);
+  const { cors: a } = n, s = x(n);
   return async function(d, o) {
     try {
       if (d.method !== "POST")
@@ -271,7 +271,7 @@ function le(n) {
         );
       const r = await d.json(), e = await F(s, r);
       return t.json(e, {
-        headers: a ? g(d, a) : {}
+        headers: a ? j(d, a) : {}
       });
     } catch (r) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js discover handler error:", r), t.json(
@@ -285,7 +285,7 @@ function le(n) {
   };
 }
 function fe(n) {
-  const { cors: a } = n, s = E(n);
+  const { cors: a } = n, s = x(n);
   return async function(d, o) {
     try {
       if (d.method !== "POST")
@@ -301,7 +301,7 @@ function fe(n) {
         );
       const e = await X(s, r);
       return t.json(e, {
-        headers: a ? g(d, a) : {}
+        headers: a ? j(d, a) : {}
       });
     } catch (r) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js suggest handler error:", r), t.json(
@@ -315,7 +315,7 @@ function fe(n) {
   };
 }
 function ye(n) {
-  const { cors: a } = n, s = E(n);
+  const { cors: a } = n, s = x(n);
   return async function(d, o) {
     try {
       if (d.method !== "POST")
@@ -331,7 +331,7 @@ function ye(n) {
         );
       const e = await W(s, r);
       return t.json(e, {
-        headers: a ? g(d, a) : {}
+        headers: a ? j(d, a) : {}
       });
     } catch (r) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js validate handler error:", r), t.json(
@@ -345,7 +345,7 @@ function ye(n) {
   };
 }
 function he(n) {
-  const { extractSecurityContext: a, cors: s } = n, c = E(n);
+  const { extractSecurityContext: a, cors: s } = n, c = x(n);
   return async function(o, r) {
     try {
       if (o.method !== "POST")
@@ -359,9 +359,9 @@ function he(n) {
           i("query field is required", 400),
           { status: 400 }
         );
-      const u = await a(o, r), y = await Y(c, u, e);
-      return t.json(y, {
-        headers: s ? g(o, s) : {}
+      const u = await a(o, r), h = await Y(c, u, e);
+      return t.json(h, {
+        headers: s ? j(o, s) : {}
       });
     } catch (e) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js MCP load handler error:", e), t.json(
@@ -375,7 +375,7 @@ function he(n) {
   };
 }
 function ae(n) {
-  const { extractSecurityContext: a, cors: s, mcp: c = { enabled: !0 } } = n, d = E(n);
+  const { extractSecurityContext: a, cors: s, mcp: c = { enabled: !0 } } = n, d = x(n);
   return async function(r) {
     if (r.method === "DELETE")
       return t.json(
@@ -383,24 +383,24 @@ function ae(n) {
         { status: 405 }
       );
     if (r.method === "GET") {
-      const h = new TextEncoder(), S = T(), x = new ReadableStream({
-        start(m) {
-          m.enqueue(h.encode(A({
+      const m = new TextEncoder(), v = T(), C = new ReadableStream({
+        start(f) {
+          f.enqueue(m.encode(A({
             jsonrpc: "2.0",
             method: "mcp/ready",
             params: { protocol: "streamable-http" }
-          }, S, 15e3)));
+          }, v, 15e3)));
         }
-      }), j = new Headers({
+      }), N = new Headers({
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
       });
       if (s) {
-        const m = g(r, s);
-        Object.entries(m).forEach(([w, N]) => j.set(w, N));
+        const f = j(r, s);
+        Object.entries(f).forEach(([g, w]) => N.set(g, w));
       }
-      return new t(x, { status: 200, headers: j });
+      return new t(C, { status: 200, headers: N });
     }
     if (r.method !== "POST")
       return t.json(
@@ -422,11 +422,11 @@ function ae(n) {
         P(null, -32600, "Accept header must include both application/json and text/event-stream"),
         { status: 400 }
       );
-    const y = Q(Object.fromEntries(r.headers.entries()));
-    if (!y.ok)
+    const h = Q(Object.fromEntries(r.headers.entries()));
+    if (!h.ok)
       return t.json({
         error: "Unsupported MCP protocol version",
-        supported: y.supported
+        supported: h.supported
       }, { status: 426 });
     let l;
     try {
@@ -434,63 +434,63 @@ function ae(n) {
     } catch {
       l = null;
     }
-    const f = V(l);
-    if (!f)
+    const y = V(l);
+    if (!y)
       return t.json(
         P(null, -32600, "Invalid JSON-RPC 2.0 request"),
         { status: 400 }
       );
-    const v = I(u), p = f.method === "initialize", C = (h, S = 200, x = {}) => t.json(h, {
-      status: S,
-      headers: { ...s ? g(r, s) : {}, ...x }
+    const S = I(u), p = y.method === "initialize", E = (m, v = 200, C = {}) => t.json(m, {
+      status: v,
+      headers: { ...s ? j(r, s) : {}, ...C }
     });
     try {
-      const h = await J(
-        f.method,
-        f.params,
+      const m = await J(
+        y.method,
+        y.params,
         {
           semanticLayer: d,
-          extractSecurityContext: (m) => a(m),
+          extractSecurityContext: (f) => a(f),
           rawRequest: r,
           rawResponse: null
         }
       );
-      if (M(f))
+      if (M(y))
         return new t(null, { status: 202 });
-      const S = p && h && typeof h == "object" && "sessionId" in h ? h.sessionId : void 0, x = {};
-      S && (x[k] = S);
-      const j = $(f.id ?? null, h);
-      if (v) {
-        const m = new TextEncoder(), w = T(), N = new ReadableStream({
+      const v = p && m && typeof m == "object" && "sessionId" in m ? m.sessionId : void 0, C = {};
+      v && (C[k] = v);
+      const N = $(y.id ?? null, m);
+      if (S) {
+        const f = new TextEncoder(), g = T(), w = new ReadableStream({
           start(b) {
-            b.enqueue(m.encode(`id: ${w}
+            b.enqueue(f.encode(`id: ${g}
 
-`)), b.enqueue(m.encode(A(j, w))), b.close();
+`)), b.enqueue(f.encode(A(N, g))), b.close();
           }
         }), O = new Headers({
           "Content-Type": "text/event-stream",
           "Cache-Control": "no-cache",
           Connection: "keep-alive",
-          ...x
+          ...C
         });
         if (s) {
-          const b = g(r, s);
+          const b = j(r, s);
           Object.entries(b).forEach(([H, R]) => O.set(H, R));
         }
-        return new t(N, { status: 200, headers: O });
+        return new t(w, { status: 200, headers: O });
       }
-      return C(j, 200, x);
-    } catch (h) {
-      if (M(f))
-        return process.env.NODE_ENV !== "test" && console.error("Next.js MCP notification processing error:", h), new t(null, { status: 202 });
-      process.env.NODE_ENV !== "test" && console.error("Next.js MCP RPC handler error:", h);
-      const S = h?.code ?? -32603, x = h?.data, j = h.message || "MCP request failed", m = P(f.id ?? null, S, j, x);
-      if (v) {
-        const w = new TextEncoder(), N = T(), O = new ReadableStream({
+      return E(N, 200, C);
+    } catch (m) {
+      if (M(y))
+        return process.env.NODE_ENV !== "test" && console.error("Next.js MCP notification processing error:", m), new t(null, { status: 202 });
+      process.env.NODE_ENV !== "test" && console.error("Next.js MCP RPC handler error:", m);
+      const v = m?.code ?? -32603, C = m?.data, N = m.message || "MCP request failed", f = P(y.id ?? null, v, N, C);
+      if (S) {
+        const g = new TextEncoder(), w = T(), O = new ReadableStream({
           start(H) {
-            H.enqueue(w.encode(`id: ${N}
+            H.enqueue(g.encode(`id: ${w}
 
-`)), H.enqueue(w.encode(A(m, N))), H.close();
+`)), H.enqueue(g.encode(A(f, w))), H.close();
           }
         }), b = new Headers({
           "Content-Type": "text/event-stream",
@@ -498,12 +498,12 @@ function ae(n) {
           Connection: "keep-alive"
         });
         if (s) {
-          const H = g(r, s);
+          const H = j(r, s);
           Object.entries(H).forEach(([R, L]) => b.set(R, L));
         }
         return new t(O, { status: 200, headers: b });
       }
-      return C(m, 200);
+      return E(f, 200);
     }
   };
 }
@@ -511,7 +511,7 @@ function oe(n) {
   const { extractSecurityContext: a, cors: s, agent: c } = n;
   if (!c)
     throw new Error("agent config is required for createAgentChatHandler");
-  const d = E(n);
+  const d = x(n);
   return async function(r, e) {
     try {
       if (r.method !== "POST")
@@ -519,7 +519,7 @@ function oe(n) {
           { error: "Method not allowed - use POST" },
           { status: 405 }
         );
-      const { handleAgentChat: u } = await import("../handler-DefTXJpi.js"), y = await r.json(), { message: l, sessionId: f, history: v } = y;
+      const { handleAgentChat: u } = await import("../handler-CbDMdSY5.js"), h = await r.json(), { message: l, sessionId: y, history: S } = h;
       if (!l || typeof l != "string")
         return t.json(
           { error: "message is required and must be a string" },
@@ -527,54 +527,55 @@ function oe(n) {
         );
       let p = (c.apiKey || "").trim();
       if (c.allowClientApiKey) {
-        const j = r.headers.get("x-agent-api-key");
-        j && (p = j.trim());
+        const f = r.headers.get("x-agent-api-key");
+        f && (p = f.trim());
       }
       if (!p)
         return t.json(
           { error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header." },
           { status: 401 }
         );
-      const C = await a(r, e), h = new TextEncoder(), S = new ReadableStream({
-        async start(j) {
+      const E = await a(r, e), m = c.buildSystemContext?.(E), v = new TextEncoder(), C = new ReadableStream({
+        async start(f) {
           try {
-            const m = u({
+            const g = u({
               message: l,
-              sessionId: f,
-              history: v,
+              sessionId: y,
+              history: S,
               semanticLayer: d,
-              securityContext: C,
+              securityContext: E,
               agentConfig: c,
-              apiKey: p
+              apiKey: p,
+              systemContext: m
             });
-            for await (const w of m) {
-              const N = `data: ${JSON.stringify(w)}
+            for await (const w of g) {
+              const O = `data: ${JSON.stringify(w)}
 
 `;
-              j.enqueue(h.encode(N));
+              f.enqueue(v.encode(O));
             }
-          } catch (m) {
+          } catch (g) {
             const w = {
               type: "error",
-              data: { message: m instanceof Error ? m.message : "Stream failed" }
+              data: { message: g instanceof Error ? g.message : "Stream failed" }
             };
-            j.enqueue(h.encode(`data: ${JSON.stringify(w)}
+            f.enqueue(v.encode(`data: ${JSON.stringify(w)}
 
 `));
           } finally {
-            j.close();
+            f.close();
           }
         }
-      }), x = new Headers({
+      }), N = new Headers({
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
       });
       if (s) {
-        const j = g(r, s);
-        Object.entries(j).forEach(([m, w]) => x.set(m, w));
+        const f = j(r, s);
+        Object.entries(f).forEach(([g, w]) => N.set(g, w));
       }
-      return new Response(S, { status: 200, headers: x });
+      return new Response(C, { status: 200, headers: N });
     } catch (u) {
       return process.env.NODE_ENV !== "test" && console.error("Next.js agent chat handler error:", u), t.json(
         { error: u instanceof Error ? u.message : "Agent chat failed" },