drizzle-cube 0.4.29 → 0.4.30

package/dist/adapters/express/index.js

@@ -1,345 +1,235 @@
-import R, { Router as J } from "express";
-import L from "cors";
-import { S as K, v as D, b as x, a as _, n as V, p as z, w as T, d as B, i as I, M as U, c as F, e as Q, s as E } from "../mcp-transport-CyeHMDPl.js";
-import { formatErrorResponse as u, formatCubeResponse as P, handleBatchRequest as X, formatMetaResponse as G, formatSqlResponse as H, handleDryRun as M } from "../utils.js";
-function W(v) {
-  const {
-    cubes: h,
-    drizzle: S,
-    schema: O,
-    extractSecurityContext: y,
-    engineType: N,
-    cors: $,
-    basePath: p = "/cubejs-api/v1",
-    jsonLimit: A = "10mb",
-    cache: k,
-    mcp: w = { enabled: !0 },
-    agent: m
-  } = v;
-  if (!h || h.length === 0)
-    throw new Error("At least one cube must be provided in the cubes array");
-  const c = J();
-  $ && c.use(L($)), c.use(R.json({ limit: A })), c.use(R.urlencoded({ extended: !0, limit: A }));
-  const i = new K({
-    drizzle: S,
-    schema: O,
-    engineType: N,
-    cache: k
-  });
-  if (h.forEach((r) => {
-    i.registerCube(r);
-  }), c.post(`${p}/load`, async (r, t) => {
-    try {
-      const e = r.body.query || r.body, o = await y(r, t), a = i.validateQuery(e);
-      if (!a.isValid)
-        return t.status(400).json(u(
-          `Query validation failed: ${a.errors.join(", ")}`,
-          400
-        ));
-      const n = r.headers["x-cache-control"] === "no-cache", s = await i.executeMultiCubeQuery(e, o, { skipCache: n });
-      t.json(P(e, s, i));
-    } catch (e) {
-      console.error("Query execution error:", e), t.status(500).json(u(
-        e instanceof Error ? e.message : "Query execution failed",
-        500
-      ));
-    }
-  }), c.get(`${p}/load`, async (r, t) => {
-    try {
-      const e = r.query.query;
-      if (!e)
-        return t.status(400).json(u(
-          "Query parameter is required",
-          400
-        ));
-      let o;
-      try {
-        o = JSON.parse(e);
-      } catch {
-        return t.status(400).json(u(
-          "Invalid JSON in query parameter",
-          400
-        ));
-      }
-      const a = await y(r, t), n = i.validateQuery(o);
-      if (!n.isValid)
-        return t.status(400).json(u(
-          `Query validation failed: ${n.errors.join(", ")}`,
-          400
-        ));
-      const s = r.headers["x-cache-control"] === "no-cache", f = await i.executeMultiCubeQuery(o, a, { skipCache: s });
-      t.json(P(o, f, i));
-    } catch (e) {
-      console.error("Query execution error:", e), t.status(500).json(u(
-        e instanceof Error ? e.message : "Query execution failed",
-        500
-      ));
-    }
-  }), c.post(`${p}/batch`, async (r, t) => {
-    try {
-      const { queries: e } = r.body;
-      if (!e || !Array.isArray(e))
-        return t.status(400).json(u(
-          'Request body must contain a "queries" array',
-          400
-        ));
-      if (e.length === 0)
-        return t.status(400).json(u(
-          "Queries array cannot be empty",
-          400
-        ));
-      const o = await y(r, t), a = r.headers["x-cache-control"] === "no-cache", n = await X(e, o, i, { skipCache: a });
-      t.json(n);
-    } catch (e) {
-      console.error("Batch execution error:", e), t.status(500).json(u(
-        e instanceof Error ? e.message : "Batch execution failed",
-        500
-      ));
-    }
-  }), c.get(`${p}/meta`, (r, t) => {
-    try {
-      const e = i.getMetadata();
-      t.json(G(e));
-    } catch (e) {
-      console.error("Metadata error:", e), t.status(500).json(u(
-        e instanceof Error ? e.message : "Failed to fetch metadata",
-        500
-      ));
-    }
-  }), c.post(`${p}/sql`, async (r, t) => {
-    try {
-      const e = r.body, o = await y(r, t), a = i.validateQuery(e);
-      if (!a.isValid)
-        return t.status(400).json(u(
-          `Query validation failed: ${a.errors.join(", ")}`,
-          400
-        ));
-      const n = e.measures?.[0] || e.dimensions?.[0];
-      if (!n)
-        return t.status(400).json(u(
-          "No measures or dimensions specified",
-          400
-        ));
-      const s = n.split(".")[0], f = await i.generateSQL(s, e, o);
-      t.json(H(e, f));
-    } catch (e) {
-      console.error("SQL generation error:", e), t.status(500).json(u(
-        e instanceof Error ? e.message : "SQL generation failed",
-        500
-      ));
-    }
-  }), c.get(`${p}/sql`, async (r, t) => {
-    try {
-      const e = r.query.query;
-      if (!e)
-        return t.status(400).json(u(
-          "Query parameter is required",
-          400
-        ));
-      const o = JSON.parse(e), a = await y(r, t), n = i.validateQuery(o);
-      if (!n.isValid)
-        return t.status(400).json(u(
-          `Query validation failed: ${n.errors.join(", ")}`,
-          400
-        ));
-      const s = o.measures?.[0] || o.dimensions?.[0];
-      if (!s)
-        return t.status(400).json(u(
-          "No measures or dimensions specified",
-          400
-        ));
-      const f = s.split(".")[0], b = await i.generateSQL(f, o, a);
-      t.json(H(o, b));
-    } catch (e) {
-      console.error("SQL generation error:", e), t.status(500).json(u(
-        e instanceof Error ? e.message : "SQL generation failed",
-        500
-      ));
-    }
-  }), c.post(`${p}/dry-run`, async (r, t) => {
-    try {
-      const e = r.body.query || r.body, o = await y(r, t), a = await M(e, o, i);
-      t.json(a);
-    } catch (e) {
-      console.error("Dry-run error:", e), t.status(400).json({
-        error: e instanceof Error ? e.message : "Dry-run validation failed",
-        valid: !1
-      });
-    }
-  }), c.get(`${p}/dry-run`, async (r, t) => {
-    try {
-      const e = r.query.query;
-      if (!e)
-        return t.status(400).json({
-          error: "Query parameter is required",
-          valid: !1
-        });
-      const o = JSON.parse(e), a = await y(r, t), n = await M(o, a, i);
-      t.json(n);
-    } catch (e) {
-      console.error("Dry-run error:", e), t.status(400).json({
-        error: e instanceof Error ? e.message : "Dry-run validation failed",
-        valid: !1
-      });
-    }
-  }), c.post(`${p}/explain`, async (r, t) => {
-    try {
-      const e = r.body.query || r.body, o = r.body.options || {}, a = await y(r, t), n = i.validateQuery(e);
-      if (!n.isValid)
-        return t.status(400).json({
-          error: `Query validation failed: ${n.errors.join(", ")}`
-        });
-      const s = await i.explainQuery(e, a, o);
-      t.json(s);
-    } catch (e) {
-      console.error("Explain error:", e), t.status(500).json({
-        error: e instanceof Error ? e.message : "Explain query failed"
-      });
-    }
-  }), m && c.post(`${p}/agent/chat`, async (r, t) => {
-    try {
-      const { handleAgentChat: e } = await import("../handler-BHguLZOY.js"), { message: o, sessionId: a, history: n } = r.body;
-      if (!o || typeof o != "string")
-        return t.status(400).json({ error: "message is required and must be a string" });
-      let s = (m.apiKey || "").trim();
-      if (m.allowClientApiKey) {
-        const l = r.headers["x-agent-api-key"];
-        l && (s = l.trim());
-      }
-      if (!s)
-        return t.status(401).json({
-          error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."
-        });
-      const f = m.allowClientApiKey ? r.headers["x-agent-provider"] : void 0, b = m.allowClientApiKey ? r.headers["x-agent-model"] : void 0, d = m.allowClientApiKey ? r.headers["x-agent-provider-endpoint"] : void 0, g = await y(r, t), j = m.buildSystemContext?.(g);
-      t.writeHead(200, {
-        "Content-Type": "text/event-stream",
-        "Cache-Control": "no-cache",
-        Connection: "keep-alive"
-      });
-      try {
-        const l = e({
-          message: o,
-          sessionId: a,
-          history: n,
-          semanticLayer: i,
-          securityContext: g,
-          agentConfig: m,
-          apiKey: s,
-          systemContext: j,
-          providerOverride: f,
-          modelOverride: b,
-          baseURLOverride: d
-        });
-        for await (const C of l)
-          t.write(`data: ${JSON.stringify(C)}
-
-`);
-      } catch (l) {
-        const C = {
-          type: "error",
-          data: { message: l instanceof Error ? l.message : "Stream failed" }
-        };
-        t.write(`data: ${JSON.stringify(C)}
-
-`);
-      } finally {
-        t.end();
-      }
-    } catch (e) {
-      console.error("Agent chat error:", e), t.headersSent || t.status(500).json({
-        error: e instanceof Error ? e.message : "Agent chat failed"
-      });
-    }
-  }), w.enabled !== !1) {
-    const r = w.basePath ?? "/mcp";
-    c.post(`${r}`, async (t, e) => {
-      const o = D(
-        t.headers.origin,
-        w.allowedOrigins ? { allowedOrigins: w.allowedOrigins } : {}
-      );
-      if (!o.valid)
-        return e.status(403).json(x(null, -32600, o.reason));
-      const a = t.headers.accept;
-      if (!_(a))
-        return e.status(400).json(x(null, -32600, "Accept header must include both application/json and text/event-stream"));
-      const n = V(t.headers);
-      if (!n.ok)
-        return e.status(426).json({
-          error: "Unsupported MCP protocol version",
-          supported: n.supported
-        });
-      const s = z(t.body);
-      if (!s)
-        return e.status(400).json(x(null, -32600, "Invalid JSON-RPC 2.0 request"));
-      const f = T(a), b = s.method === "initialize";
-      try {
-        const d = await B(
-          s.method,
-          s.params,
-          {
-            semanticLayer: i,
-            extractSecurityContext: y,
-            rawRequest: t,
-            rawResponse: e,
-            negotiatedProtocol: n.negotiated
-          }
-        );
-        if (I(s))
-          return e.status(202).end();
-        const g = b && d && typeof d == "object" && "sessionId" in d ? d.sessionId : void 0;
-        g && e.setHeader(U, g);
-        const j = F(s.id ?? null, d);
-        if (f) {
-          const l = Q();
-          return e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(`id: ${l}
-
-`), e.write(E(j, l)), e.end();
-        }
-        return e.json(j);
-      } catch (d) {
-        if (I(s))
-          return console.error("MCP notification processing error:", d), e.status(202).end();
-        console.error("MCP RPC error:", d);
-        const g = d?.code ?? -32603, j = d?.data, l = d.message || "MCP request failed", C = x(s.id ?? null, g, l, j);
-        if (f) {
-          const q = Q();
-          return e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(`id: ${q}
-
-`), e.write(E(C, q)), e.end();
-        }
-        return e.status(200).json(C);
-      }
-    }), c.get(`${r}`, async (t, e) => {
-      const o = Q();
-      e.status(200), e.setHeader("Content-Type", "text/event-stream"), e.setHeader("Cache-Control", "no-cache"), e.setHeader("Connection", "keep-alive"), e.write(E({
-        jsonrpc: "2.0",
-        method: "mcp/ready",
-        params: { protocol: "streamable-http" }
-      }, o, 15e3));
-      const a = setInterval(() => {
-        e.write(`: keep-alive
-
-`);
-      }, 15e3);
-      t.on("close", () => {
-        clearInterval(a);
-      });
-    }), c.delete(`${r}`, (t, e) => e.status(405).json({ error: "Session termination not supported" }));
-  }
-  return c.use((r, t, e, o) => {
-    console.error("Express adapter error:", r), e.headersSent || e.status(500).json(u(r, 500));
-  }), c;
+import { a as e, f as t, i as n, o as r, r as i, u as a } from "../utils-ChhNGUOF.js";
+import { c as o, d as s, f as c, h as l, i as u, l as d, m as f, n as p, p as m, r as h, s as g, t as _, u as v } from "../mcp-transport-CuugoG8t.js";
+import y, { Router as b } from "express";
+import x from "cors";
+//#region src/adapters/express/index.ts
+function S(S) {
+	let { cubes: C, drizzle: w, schema: T, extractSecurityContext: E, engineType: D, cors: O, basePath: k = "/cubejs-api/v1", jsonLimit: A = "10mb", cache: j, mcp: M = { enabled: !0 }, agent: N } = S;
+	if (!C || C.length === 0) throw Error("At least one cube must be provided in the cubes array");
+	let P = b();
+	O && P.use(x(O)), P.use(y.json({ limit: A })), P.use(y.urlencoded({
+		extended: !0,
+		limit: A
+	}));
+	let F = new l({
+		drizzle: w,
+		schema: T,
+		engineType: D,
+		cache: j
+	});
+	if (C.forEach((e) => {
+		F.registerCube(e);
+	}), P.post(`${k}/load`, async (e, t) => {
+		try {
+			let r = e.body.query || e.body, a = await E(e, t), o = F.validateQuery(r);
+			if (!o.isValid) return t.status(400).json(n(`Query validation failed: ${o.errors.join(", ")}`, 400));
+			let s = e.headers["x-cache-control"] === "no-cache", c = await F.executeMultiCubeQuery(r, a, { skipCache: s });
+			t.json(i(r, c, F));
+		} catch (e) {
+			console.error("Query execution error:", e), t.status(500).json(n(e instanceof Error ? e.message : "Query execution failed", 500));
+		}
+	}), P.get(`${k}/load`, async (e, t) => {
+		try {
+			let r = e.query.query;
+			if (!r) return t.status(400).json(n("Query parameter is required", 400));
+			let a;
+			try {
+				a = JSON.parse(r);
+			} catch {
+				return t.status(400).json(n("Invalid JSON in query parameter", 400));
+			}
+			let o = await E(e, t), s = F.validateQuery(a);
+			if (!s.isValid) return t.status(400).json(n(`Query validation failed: ${s.errors.join(", ")}`, 400));
+			let c = e.headers["x-cache-control"] === "no-cache", l = await F.executeMultiCubeQuery(a, o, { skipCache: c });
+			t.json(i(a, l, F));
+		} catch (e) {
+			console.error("Query execution error:", e), t.status(500).json(n(e instanceof Error ? e.message : "Query execution failed", 500));
+		}
+	}), P.post(`${k}/batch`, async (e, t) => {
+		try {
+			let { queries: r } = e.body;
+			if (!r || !Array.isArray(r)) return t.status(400).json(n("Request body must contain a \"queries\" array", 400));
+			if (r.length === 0) return t.status(400).json(n("Queries array cannot be empty", 400));
+			let i = await a(r, await E(e, t), F, { skipCache: e.headers["x-cache-control"] === "no-cache" });
+			t.json(i);
+		} catch (e) {
+			console.error("Batch execution error:", e), t.status(500).json(n(e instanceof Error ? e.message : "Batch execution failed", 500));
+		}
+	}), P.get(`${k}/meta`, (t, r) => {
+		try {
+			let t = F.getMetadata();
+			r.json(e(t));
+		} catch (e) {
+			console.error("Metadata error:", e), r.status(500).json(n(e instanceof Error ? e.message : "Failed to fetch metadata", 500));
+		}
+	}), P.post(`${k}/sql`, async (e, t) => {
+		try {
+			let i = e.body, a = await E(e, t), o = F.validateQuery(i);
+			if (!o.isValid) return t.status(400).json(n(`Query validation failed: ${o.errors.join(", ")}`, 400));
+			let s = i.measures?.[0] || i.dimensions?.[0];
+			if (!s) return t.status(400).json(n("No measures or dimensions specified", 400));
+			let c = s.split(".")[0], l = await F.generateSQL(c, i, a);
+			t.json(r(i, l));
+		} catch (e) {
+			console.error("SQL generation error:", e), t.status(500).json(n(e instanceof Error ? e.message : "SQL generation failed", 500));
+		}
+	}), P.get(`${k}/sql`, async (e, t) => {
+		try {
+			let i = e.query.query;
+			if (!i) return t.status(400).json(n("Query parameter is required", 400));
+			let a = JSON.parse(i), o = await E(e, t), s = F.validateQuery(a);
+			if (!s.isValid) return t.status(400).json(n(`Query validation failed: ${s.errors.join(", ")}`, 400));
+			let c = a.measures?.[0] || a.dimensions?.[0];
+			if (!c) return t.status(400).json(n("No measures or dimensions specified", 400));
+			let l = c.split(".")[0], u = await F.generateSQL(l, a, o);
+			t.json(r(a, u));
+		} catch (e) {
+			console.error("SQL generation error:", e), t.status(500).json(n(e instanceof Error ? e.message : "SQL generation failed", 500));
+		}
+	}), P.post(`${k}/dry-run`, async (e, n) => {
+		try {
+			let r = await t(e.body.query || e.body, await E(e, n), F);
+			n.json(r);
+		} catch (e) {
+			console.error("Dry-run error:", e), n.status(400).json({
+				error: e instanceof Error ? e.message : "Dry-run validation failed",
+				valid: !1
+			});
+		}
+	}), P.get(`${k}/dry-run`, async (e, n) => {
+		try {
+			let r = e.query.query;
+			if (!r) return n.status(400).json({
+				error: "Query parameter is required",
+				valid: !1
+			});
+			let i = await t(JSON.parse(r), await E(e, n), F);
+			n.json(i);
+		} catch (e) {
+			console.error("Dry-run error:", e), n.status(400).json({
+				error: e instanceof Error ? e.message : "Dry-run validation failed",
+				valid: !1
+			});
+		}
+	}), P.post(`${k}/explain`, async (e, t) => {
+		try {
+			let n = e.body.query || e.body, r = e.body.options || {}, i = await E(e, t), a = F.validateQuery(n);
+			if (!a.isValid) return t.status(400).json({ error: `Query validation failed: ${a.errors.join(", ")}` });
+			let o = await F.explainQuery(n, i, r);
+			t.json(o);
+		} catch (e) {
+			console.error("Explain error:", e), t.status(500).json({ error: e instanceof Error ? e.message : "Explain query failed" });
+		}
+	}), N && P.post(`${k}/agent/chat`, async (e, t) => {
+		try {
+			let { handleAgentChat: n } = await import("../handler-D-2-6uLM.js"), { message: r, sessionId: i, history: a } = e.body;
+			if (!r || typeof r != "string") return t.status(400).json({ error: "message is required and must be a string" });
+			let o = (N.apiKey || "").trim();
+			if (N.allowClientApiKey) {
+				let t = e.headers["x-agent-api-key"];
+				t && (o = t.trim());
+			}
+			if (!o) return t.status(401).json({ error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header." });
+			let s = N.allowClientApiKey ? e.headers["x-agent-provider"] : void 0, c = N.allowClientApiKey ? e.headers["x-agent-model"] : void 0, l = N.allowClientApiKey ? e.headers["x-agent-provider-endpoint"] : void 0, u = await E(e, t), d = N.buildSystemContext?.(u);
+			t.writeHead(200, {
+				"Content-Type": "text/event-stream",
+				"Cache-Control": "no-cache",
+				Connection: "keep-alive"
+			});
+			try {
+				let e = n({
+					message: r,
+					sessionId: i,
+					history: a,
+					semanticLayer: F,
+					securityContext: u,
+					agentConfig: N,
+					apiKey: o,
+					systemContext: d,
+					providerOverride: s,
+					modelOverride: c,
+					baseURLOverride: l
+				});
+				for await (let n of e) t.write(`data: ${JSON.stringify(n)}\n\n`);
+			} catch (e) {
+				let n = {
+					type: "error",
+					data: { message: e instanceof Error ? e.message : "Stream failed" }
+				};
+				t.write(`data: ${JSON.stringify(n)}\n\n`);
+			} finally {
+				t.end();
+			}
+		} catch (e) {
+			console.error("Agent chat error:", e), t.headersSent || t.status(500).json({ error: e instanceof Error ? e.message : "Agent chat failed" });
+		}
+	}), M.enabled !== !1) {
+		let e = M.basePath ?? "/mcp";
+		P.post(`${e}`, async (e, t) => {
+			let n = m(e.headers.origin, M.allowedOrigins ? { allowedOrigins: M.allowedOrigins } : {});
+			if (!n.valid) return t.status(403).json(p(null, -32600, n.reason));
+			let r = e.headers.accept;
+			if (!c(r)) return t.status(400).json(p(null, -32600, "Accept header must include both application/json and text/event-stream"));
+			let i = o(e.headers);
+			if (!i.ok) return t.status(426).json({
+				error: "Unsupported MCP protocol version",
+				supported: i.supported
+			});
+			let a = d(e.body);
+			if (!a) return t.status(400).json(p(null, -32600, "Invalid JSON-RPC 2.0 request"));
+			let l = f(r), y = a.method === "initialize";
+			try {
+				let n = await u(a.method, a.params, {
+					semanticLayer: F,
+					extractSecurityContext: E,
+					rawRequest: e,
+					rawResponse: t,
+					negotiatedProtocol: i.negotiated
+				});
+				if (g(a)) return t.status(202).end();
+				let r = y && n && typeof n == "object" && "sessionId" in n ? n.sessionId : void 0;
+				r && t.setHeader(_, r);
+				let o = h(a.id ?? null, n);
+				if (l) {
+					let e = v();
+					return t.status(200), t.setHeader("Content-Type", "text/event-stream"), t.setHeader("Cache-Control", "no-cache"), t.setHeader("Connection", "keep-alive"), t.write(`id: ${e}\n\n`), t.write(s(o, e)), t.end();
+				}
+				return t.json(o);
+			} catch (e) {
+				if (g(a)) return console.error("MCP notification processing error:", e), t.status(202).end();
+				console.error("MCP RPC error:", e);
+				let n = e?.code ?? -32603, r = e?.data, i = e.message || "MCP request failed", o = p(a.id ?? null, n, i, r);
+				if (l) {
+					let e = v();
+					return t.status(200), t.setHeader("Content-Type", "text/event-stream"), t.setHeader("Cache-Control", "no-cache"), t.setHeader("Connection", "keep-alive"), t.write(`id: ${e}\n\n`), t.write(s(o, e)), t.end();
+				}
+				return t.status(200).json(o);
+			}
+		}), P.get(`${e}`, async (e, t) => {
+			let n = v();
+			t.status(200), t.setHeader("Content-Type", "text/event-stream"), t.setHeader("Cache-Control", "no-cache"), t.setHeader("Connection", "keep-alive"), t.write(s({
+				jsonrpc: "2.0",
+				method: "mcp/ready",
+				params: { protocol: "streamable-http" }
+			}, n, 15e3));
+			let r = setInterval(() => {
+				t.write(": keep-alive\n\n");
+			}, 15e3);
+			e.on("close", () => {
+				clearInterval(r);
+			});
+		}), P.delete(`${e}`, (e, t) => t.status(405).json({ error: "Session termination not supported" }));
+	}
+	return P.use((e, t, r, i) => {
+		console.error("Express adapter error:", e), r.headersSent || r.status(500).json(n(e, 500));
+	}), P;
 }
-function Y(v, h) {
-  const S = W(h);
-  return v.use("/", S), v;
+function C(e, t) {
+	let n = S(t);
+	return e.use("/", n), e;
 }
-function oe(v) {
-  const h = R();
-  return Y(h, v);
+function w(e) {
+	return C(y(), e);
 }
-export {
-  oe as createCubeApp,
-  W as createCubeRouter,
-  Y as mountCubeRoutes
-};
+//#endregion
+export { w as createCubeApp, S as createCubeRouter, C as mountCubeRoutes };