drizzle-cube 0.4.18 → 0.4.20

package/dist/adapters/fastify/index.js

@@ -1,19 +1,19 @@
 import { S as D, v as K, b as x, a as T, n as V, p as z, w as _, d as B, i as j, M as F, c as U, e as E, s as P } from "../mcp-transport-m1X1GtwG.js";
-import { formatErrorResponse as u, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as I, handleDryRun as M } from "../utils.js";
+import { formatErrorResponse as d, formatCubeResponse as A, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as I, handleDryRun as M } from "../utils.js";
 const N = function(i, k, R) {
   const {
     cubes: S,
     drizzle: O,
     schema: J,
-    extractSecurityContext: y,
+    extractSecurityContext: p,
     engineType: L,
     cors: q,
-    basePath: p = "/cubejs-api/v1",
-    bodyLimit: m = 10485760,
+    basePath: m = "/cubejs-api/v1",
+    bodyLimit: g = 10485760,
     // 10MB
     cache: H,
-    mcp: f = { enabled: !0 },
-    agent: v
+    mcp: C = { enabled: !0 },
+    agent: f
   } = k;
   if (!S || S.length === 0)
     return R(new Error("At least one cube must be provided in the cubes array"));
@@ -28,8 +28,8 @@ const N = function(i, k, R) {
   });
   if (S.forEach((t) => {
     c.registerCube(t);
-  }), i.post(`${p}/load`, {
-    bodyLimit: m,
+  }), i.post(`${m}/load`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -38,21 +38,21 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await y(t), s = c.validateQuery(a);
+      const e = t.body, a = e.query || e, n = await p(t), s = c.validateQuery(a);
       if (!s.isValid)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${s.errors.join(", ")}`,
           400
         ));
-      const o = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: o });
-      return A(a, l, c);
+      const o = t.headers["x-cache-control"] === "no-cache", u = await c.executeMultiCubeQuery(a, n, { skipCache: o });
+      return A(a, u, c);
     } catch (e) {
-      return t.log.error(e, "Query execution error"), r.status(500).send(u(
+      return t.log.error(e, "Query execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), i.get(`${p}/load`, {
+  }), i.get(`${m}/load`, {
     schema: {
       querystring: {
         type: "object",
@@ -69,27 +69,27 @@ const N = function(i, k, R) {
       try {
         a = JSON.parse(e);
       } catch {
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           "Invalid JSON in query parameter",
           400
         ));
       }
-      const n = await y(t), s = c.validateQuery(a);
+      const n = await p(t), s = c.validateQuery(a);
       if (!s.isValid)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${s.errors.join(", ")}`,
           400
         ));
-      const o = t.headers["x-cache-control"] === "no-cache", l = await c.executeMultiCubeQuery(a, n, { skipCache: o });
-      return A(a, l, c);
+      const o = t.headers["x-cache-control"] === "no-cache", u = await c.executeMultiCubeQuery(a, n, { skipCache: o });
+      return A(a, u, c);
     } catch (e) {
-      return t.log.error(e, "Query execution error"), r.status(500).send(u(
+      return t.log.error(e, "Query execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Query execution failed",
         500
       ));
     }
-  }), i.post(`${p}/batch`, {
-    bodyLimit: m,
+  }), i.post(`${m}/batch`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -106,35 +106,35 @@ const N = function(i, k, R) {
     try {
       const { queries: e } = t.body;
       if (!e || !Array.isArray(e))
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           'Request body must contain a "queries" array',
           400
         ));
       if (e.length === 0)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           "Queries array cannot be empty",
           400
         ));
-      const a = await y(t), n = t.headers["x-cache-control"] === "no-cache";
+      const a = await p(t), n = t.headers["x-cache-control"] === "no-cache";
       return await X(e, a, c, { skipCache: n });
     } catch (e) {
-      return t.log.error(e, "Batch execution error"), r.status(500).send(u(
+      return t.log.error(e, "Batch execution error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Batch execution failed",
         500
       ));
     }
-  }), i.get(`${p}/meta`, async (t, r) => {
+  }), i.get(`${m}/meta`, async (t, r) => {
     try {
       const e = c.getMetadata();
       return G(e);
     } catch (e) {
-      return t.log.error(e, "Metadata error"), r.status(500).send(u(
+      return t.log.error(e, "Metadata error"), r.status(500).send(d(
         e instanceof Error ? e.message : "Failed to fetch metadata",
         500
       ));
     }
-  }), i.post(`${p}/sql`, {
-    bodyLimit: m,
+  }), i.post(`${m}/sql`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -143,27 +143,27 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = await y(t), n = c.validateQuery(e);
+      const e = t.body, a = await p(t), n = c.validateQuery(e);
       if (!n.isValid)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${n.errors.join(", ")}`,
           400
         ));
       const s = e.measures?.[0] || e.dimensions?.[0];
       if (!s)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const o = s.split(".")[0], l = await c.generateSQL(o, e, a);
-      return I(e, l);
+      const o = s.split(".")[0], u = await c.generateSQL(o, e, a);
+      return I(e, u);
     } catch (e) {
-      return t.log.error(e, "SQL generation error"), r.status(500).send(u(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), i.get(`${p}/sql`, {
+  }), i.get(`${m}/sql`, {
     schema: {
       querystring: {
         type: "object",
@@ -175,28 +175,28 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await y(t), s = c.validateQuery(a);
+      const { query: e } = t.query, a = JSON.parse(e), n = await p(t), s = c.validateQuery(a);
       if (!s.isValid)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           `Query validation failed: ${s.errors.join(", ")}`,
           400
         ));
       const o = a.measures?.[0] || a.dimensions?.[0];
       if (!o)
-        return r.status(400).send(u(
+        return r.status(400).send(d(
           "No measures or dimensions specified",
           400
         ));
-      const l = o.split(".")[0], g = await c.generateSQL(l, a, n);
-      return I(a, g);
+      const u = o.split(".")[0], b = await c.generateSQL(u, a, n);
+      return I(a, b);
     } catch (e) {
-      return t.log.error(e, "SQL generation error"), r.status(500).send(u(
+      return t.log.error(e, "SQL generation error"), r.status(500).send(d(
         e instanceof Error ? e.message : "SQL generation failed",
         500
       ));
     }
-  }), i.post(`${p}/dry-run`, {
-    bodyLimit: m,
+  }), i.post(`${m}/dry-run`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -205,7 +205,7 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = await y(t);
+      const e = t.body, a = e.query || e, n = await p(t);
       return await M(a, n, c);
     } catch (e) {
       return t.log.error(e, "Dry-run error"), r.status(400).send({
@@ -213,7 +213,7 @@ const N = function(i, k, R) {
         valid: !1
       });
     }
-  }), i.get(`${p}/dry-run`, {
+  }), i.get(`${m}/dry-run`, {
     schema: {
       querystring: {
         type: "object",
@@ -225,7 +225,7 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { query: e } = t.query, a = JSON.parse(e), n = await y(t);
+      const { query: e } = t.query, a = JSON.parse(e), n = await p(t);
       return await M(a, n, c);
     } catch (e) {
       return t.log.error(e, "Dry-run error"), r.status(400).send({
@@ -233,8 +233,8 @@ const N = function(i, k, R) {
         valid: !1
       });
     }
-  }), i.post(`${p}/explain`, {
-    bodyLimit: m,
+  }), i.post(`${m}/explain`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -243,7 +243,7 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const e = t.body, a = e.query || e, n = e.options || {}, s = await y(t), o = c.validateQuery(a);
+      const e = t.body, a = e.query || e, n = e.options || {}, s = await p(t), o = c.validateQuery(a);
       return o.isValid ? await c.explainQuery(a, s, n) : r.status(400).send({
         error: `Query validation failed: ${o.errors.join(", ")}`
       });
@@ -252,8 +252,8 @@ const N = function(i, k, R) {
         error: e instanceof Error ? e.message : "Explain query failed"
       });
     }
-  }), v && i.post(`${p}/agent/chat`, {
-    bodyLimit: m,
+  }), f && i.post(`${m}/agent/chat`, {
+    bodyLimit: g,
     schema: {
       body: {
         type: "object",
@@ -262,42 +262,43 @@ const N = function(i, k, R) {
     }
   }, async (t, r) => {
     try {
-      const { handleAgentChat: e } = await import("../handler-DefTXJpi.js"), a = t.body, { message: n, sessionId: s, history: o } = a;
+      const { handleAgentChat: e } = await import("../handler-CbDMdSY5.js"), a = t.body, { message: n, sessionId: s, history: o } = a;
       if (!n || typeof n != "string")
         return r.status(400).send({ error: "message is required and must be a string" });
-      let l = (v.apiKey || "").trim();
-      if (v.allowClientApiKey) {
-        const d = t.headers["x-agent-api-key"];
-        d && (l = d.trim());
+      let u = (f.apiKey || "").trim();
+      if (f.allowClientApiKey) {
+        const y = t.headers["x-agent-api-key"];
+        y && (u = y.trim());
       }
-      if (!l)
+      if (!u)
         return r.status(401).send({
           error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."
         });
-      const g = await y(t);
+      const b = await p(t), l = f.buildSystemContext?.(b);
       r.raw.writeHead(200, {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
         Connection: "keep-alive"
       });
       try {
-        const d = e({
+        const y = e({
           message: n,
           sessionId: s,
           history: o,
           semanticLayer: c,
-          securityContext: g,
-          agentConfig: v,
-          apiKey: l
+          securityContext: b,
+          agentConfig: f,
+          apiKey: u,
+          systemContext: l
         });
-        for await (const h of d)
+        for await (const h of y)
           r.raw.write(`data: ${JSON.stringify(h)}
 
 `);
-      } catch (d) {
+      } catch (y) {
         const h = {
           type: "error",
-          data: { message: d instanceof Error ? d.message : "Stream failed" }
+          data: { message: y instanceof Error ? y.message : "Stream failed" }
         };
         r.raw.write(`data: ${JSON.stringify(h)}
 
@@ -311,10 +312,10 @@ const N = function(i, k, R) {
           error: e instanceof Error ? e.message : "Agent chat failed"
         });
     }
-  }), f.enabled !== !1) {
-    const t = f.basePath ?? "/mcp";
+  }), C.enabled !== !1) {
+    const t = C.basePath ?? "/mcp";
     i.post(`${t}`, {
-      bodyLimit: m,
+      bodyLimit: g,
       schema: {
         body: {
           type: "object",
@@ -324,7 +325,7 @@ const N = function(i, k, R) {
     }, async (r, e) => {
       const a = K(
         r.headers.origin,
-        f.allowedOrigins ? { allowedOrigins: f.allowedOrigins } : {}
+        C.allowedOrigins ? { allowedOrigins: C.allowedOrigins } : {}
       );
       if (!a.valid)
         return e.status(403).send(x(null, -32600, a.reason));
@@ -340,14 +341,14 @@ const N = function(i, k, R) {
       const o = z(r.body);
       if (!o)
         return e.status(400).send(x(null, -32600, "Invalid JSON-RPC 2.0 request"));
-      const l = _(n), g = o.method === "initialize";
+      const u = _(n), b = o.method === "initialize";
       try {
-        const d = await B(
+        const l = await B(
           o.method,
           o.params,
           {
             semanticLayer: c,
-            extractSecurityContext: y,
+            extractSecurityContext: p,
             rawRequest: r,
             rawResponse: e,
             negotiatedProtocol: s.negotiated
@@ -355,23 +356,23 @@ const N = function(i, k, R) {
         );
         if (j(o))
           return e.status(202).send();
-        const h = g && d && typeof d == "object" && "sessionId" in d ? d.sessionId : void 0;
-        h && e.header(F, h);
-        const C = U(o.id ?? null, d);
-        if (l) {
+        const y = b && l && typeof l == "object" && "sessionId" in l ? l.sessionId : void 0;
+        y && e.header(F, y);
+        const h = U(o.id ?? null, l);
+        if (u) {
           const w = E();
           e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${w}
 
-${P(C, w)}`);
+${P(h, w)}`);
           return;
         }
-        return e.send(C);
-      } catch (d) {
+        return e.send(h);
+      } catch (l) {
         if (j(o))
-          return r.log.error(d, "MCP notification processing error"), e.status(202).send();
-        r.log.error(d, "MCP RPC error");
-        const h = d?.code ?? -32603, C = d?.data, w = d.message || "MCP request failed", Q = x(o.id ?? null, h, w, C);
-        if (l) {
+          return r.log.error(l, "MCP notification processing error"), e.status(202).send();
+        r.log.error(l, "MCP RPC error");
+        const y = l?.code ?? -32603, h = l?.data, w = l.message || "MCP request failed", Q = x(o.id ?? null, y, w, h);
+        if (u) {
           const $ = E();
           e.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${$}
 
@@ -404,17 +405,17 @@ ${P(Q, $)}`);
   i.setErrorHandler(async (t, r, e) => {
     r.log.error(t, "Fastify cube adapter error"), e.statusCode < 400 && e.status(500);
     const a = t instanceof Error ? t : String(t);
-    return u(a, e.statusCode);
+    return d(a, e.statusCode);
   }), R();
 };
-async function Z(b, i) {
-  await b.register(N, i);
+async function Z(v, i) {
+  await v.register(N, i);
 }
-function ee(b) {
+function ee(v) {
   const i = require("fastify")({
     logger: !0
   });
-  return i.register(N, b), i;
+  return i.register(N, v), i;
 }
 export {
   ee as createCubeApp,