drizzle-cube 0.4.52 → 0.5.0

package/dist/adapters/fastify/index.js

@@ -1,58 +1,66 @@
-import { _ as e, a as t, f as n, i as r, o as i, r as a, u as o } from "../utils-IH1ePsBd.js";
-import { t as s } from "../compiler-D_o2IzHn.js";
-import { _ as c, a as l, d as u, f as d, g as f, h as p, i as m, m as h, p as g, r as _, s as v, v as y } from "../mcp-transport-C6bsIOSV.js";
+import { a as e, f as t, i as n, o as r, r as i, u as a, v as o } from "../utils-DOg9oGdt.js";
+import { i as s, n as c, r as l, t as u } from "../locale-DTnJrxm1.js";
+import { _ as d, a as f, b as p, c as m, g as h, h as g, i as _, l as v, m as y, p as b, r as x, s as S, v as C, y as w } from "../mcp-transport-Bxpc4mRy.js";
 //#region src/adapters/fastify/index.ts
-var b = function(e, b, x) {
-	let { cubes: S, drizzle: C, schema: w, extractSecurityContext: T, engineType: E, cors: D, basePath: O = "/cubejs-api/v1", bodyLimit: k = 10485760, cache: A, mcp: j = { enabled: !0 }, agent: M } = b;
-	if (!S || S.length === 0) return x(/* @__PURE__ */ Error("At least one cube must be provided in the cubes array"));
-	D && e.register(import("@fastify/cors"), D), e.addHook("onRequest", async (e, t) => {
+var T = function(o, T, E) {
+	let { cubes: D, drizzle: O, schema: k, extractSecurityContext: A, engineType: j, cors: M, basePath: N = "/cubejs-api/v1", bodyLimit: P = 10485760, cache: F, mcp: I = { enabled: !0 }, agent: L } = T;
+	if (!D || D.length === 0) return E(/* @__PURE__ */ Error("At least one cube must be provided in the cubes array"));
+	let R = async (e) => l(await A(e), c((t) => e.headers[t.toLowerCase()]));
+	if (M) {
+		let e = {
+			...M,
+			allowedHeaders: u(M.allowedHeaders)
+		};
+		o.register(import("@fastify/cors"), e);
+	}
+	o.addHook("onRequest", async (e, t) => {
 		e.method === "POST" && (e.body = void 0);
 	});
-	let N = new s({
-		drizzle: C,
-		schema: w,
-		engineType: E,
-		cache: A,
-		rlsSetup: b.rlsSetup
+	let z = new s({
+		drizzle: O,
+		schema: k,
+		engineType: j,
+		cache: F,
+		rlsSetup: T.rlsSetup
 	});
-	if (S.forEach((e) => {
-		N.registerCube(e);
-	}), e.post(`${O}/load`, {
-		bodyLimit: k,
+	if (D.forEach((e) => {
+		z.registerCube(e);
+	}), o.post(`${N}/load`, {
+		bodyLimit: P,
 		schema: { body: {
 			type: "object",
 			additionalProperties: !0
 		} }
 	}, async (e, t) => {
 		try {
-			let n = e.body, i = n.query || n, o = await T(e), s = N.validateQuery(i);
-			if (!s.isValid) return t.status(400).send(r(`Query validation failed: ${s.errors.join(", ")}`, 400));
+			let r = e.body, a = r.query || r, o = await R(e), s = z.validateQuery(a);
+			if (!s.isValid) return t.status(400).send(n(`Query validation failed: ${s.errors.join(", ")}`, 400));
 			let c = e.headers["x-cache-control"] === "no-cache";
-			return a(i, await N.executeMultiCubeQuery(i, o, { skipCache: c }), N);
-		} catch (n) {
-			return e.log.error(n, "Query execution error"), t.status(500).send(r(n instanceof Error ? n.message : "Query execution failed", 500));
+			return i(a, await z.executeMultiCubeQuery(a, o, { skipCache: c }), z);
+		} catch (r) {
+			return e.log.error(r, "Query execution error"), t.status(500).send(n(r instanceof Error ? r.message : "Query execution failed", 500));
 		}
-	}), e.get(`${O}/load`, { schema: { querystring: {
+	}), o.get(`${N}/load`, { schema: { querystring: {
 		type: "object",
 		properties: { query: { type: "string" } },
 		required: ["query"]
 	} } }, async (e, t) => {
 		try {
-			let { query: n } = e.query, i;
+			let { query: r } = e.query, a;
 			try {
-				i = JSON.parse(n);
+				a = JSON.parse(r);
 			} catch {
-				return t.status(400).send(r("Invalid JSON in query parameter", 400));
+				return t.status(400).send(n("Invalid JSON in query parameter", 400));
 			}
-			let o = await T(e), s = N.validateQuery(i);
-			if (!s.isValid) return t.status(400).send(r(`Query validation failed: ${s.errors.join(", ")}`, 400));
-			let c = e.headers["x-cache-control"] === "no-cache", l = await N.executeMultiCubeQuery(i, o, { skipCache: c });
-			return a(i, l, N);
-		} catch (n) {
-			return e.log.error(n, "Query execution error"), t.status(500).send(r(n instanceof Error ? n.message : "Query execution failed", 500));
+			let o = await R(e), s = z.validateQuery(a);
+			if (!s.isValid) return t.status(400).send(n(`Query validation failed: ${s.errors.join(", ")}`, 400));
+			let c = e.headers["x-cache-control"] === "no-cache", l = await z.executeMultiCubeQuery(a, o, { skipCache: c });
+			return i(a, l, z);
+		} catch (r) {
+			return e.log.error(r, "Query execution error"), t.status(500).send(n(r instanceof Error ? r.message : "Query execution failed", 500));
 		}
-	}), e.post(`${O}/batch`, {
-		bodyLimit: k,
+	}), o.post(`${N}/batch`, {
+		bodyLimit: P,
 		schema: { body: {
 			type: "object",
 			required: ["queries"],
@@ -63,109 +71,109 @@ var b = function(e, b, x) {
 		} }
 	}, async (e, t) => {
 		try {
-			let { queries: n } = e.body;
-			return !n || !Array.isArray(n) ? t.status(400).send(r("Request body must contain a \"queries\" array", 400)) : n.length === 0 ? t.status(400).send(r("Queries array cannot be empty", 400)) : await o(n, await T(e), N, { skipCache: e.headers["x-cache-control"] === "no-cache" });
-		} catch (n) {
-			return e.log.error(n, "Batch execution error"), t.status(500).send(r(n instanceof Error ? n.message : "Batch execution failed", 500));
+			let { queries: r } = e.body;
+			return !r || !Array.isArray(r) ? t.status(400).send(n("Request body must contain a \"queries\" array", 400)) : r.length === 0 ? t.status(400).send(n("Queries array cannot be empty", 400)) : await a(r, await R(e), z, { skipCache: e.headers["x-cache-control"] === "no-cache" });
+		} catch (r) {
+			return e.log.error(r, "Batch execution error"), t.status(500).send(n(r instanceof Error ? r.message : "Batch execution failed", 500));
 		}
-	}), e.get(`${O}/meta`, async (e, n) => {
+	}), o.get(`${N}/meta`, async (t, r) => {
 		try {
-			return t(N.getMetadata());
-		} catch (t) {
-			return e.log.error(t, "Metadata error"), n.status(500).send(r(t instanceof Error ? t.message : "Failed to fetch metadata", 500));
+			return e(z.getMetadata());
+		} catch (e) {
+			return t.log.error(e, "Metadata error"), r.status(500).send(n(e instanceof Error ? e.message : "Failed to fetch metadata", 500));
 		}
-	}), e.post(`${O}/sql`, {
-		bodyLimit: k,
+	}), o.post(`${N}/sql`, {
+		bodyLimit: P,
 		schema: { body: {
 			type: "object",
 			additionalProperties: !0
 		} }
 	}, async (e, t) => {
 		try {
-			let n = e.body, a = await T(e), o = N.validateQuery(n);
-			if (!o.isValid) return t.status(400).send(r(`Query validation failed: ${o.errors.join(", ")}`, 400));
-			let s = n.measures?.[0] || n.dimensions?.[0];
-			if (!s) return t.status(400).send(r("No measures or dimensions specified", 400));
+			let i = e.body, a = await R(e), o = z.validateQuery(i);
+			if (!o.isValid) return t.status(400).send(n(`Query validation failed: ${o.errors.join(", ")}`, 400));
+			let s = i.measures?.[0] || i.dimensions?.[0];
+			if (!s) return t.status(400).send(n("No measures or dimensions specified", 400));
 			let c = s.split(".")[0];
-			return i(n, await N.generateSQL(c, n, a));
-		} catch (n) {
-			return e.log.error({ err: String(n).replace(/\n|\r/g, "") }, "SQL generation error"), t.status(500).send(r(n instanceof Error ? n.message : "SQL generation failed", 500));
+			return r(i, await z.generateSQL(c, i, a));
+		} catch (r) {
+			return e.log.error({ err: String(r).replace(/\n|\r/g, "") }, "SQL generation error"), t.status(500).send(n(r instanceof Error ? r.message : "SQL generation failed", 500));
 		}
-	}), e.get(`${O}/sql`, { schema: { querystring: {
+	}), o.get(`${N}/sql`, { schema: { querystring: {
 		type: "object",
 		properties: { query: { type: "string" } },
 		required: ["query"]
 	} } }, async (e, t) => {
 		try {
-			let { query: n } = e.query, a = JSON.parse(n), o = await T(e), s = N.validateQuery(a);
-			if (!s.isValid) return t.status(400).send(r(`Query validation failed: ${s.errors.join(", ")}`, 400));
+			let { query: i } = e.query, a = JSON.parse(i), o = await R(e), s = z.validateQuery(a);
+			if (!s.isValid) return t.status(400).send(n(`Query validation failed: ${s.errors.join(", ")}`, 400));
 			let c = a.measures?.[0] || a.dimensions?.[0];
-			if (!c) return t.status(400).send(r("No measures or dimensions specified", 400));
+			if (!c) return t.status(400).send(n("No measures or dimensions specified", 400));
 			let l = c.split(".")[0];
-			return i(a, await N.generateSQL(l, a, o));
-		} catch (n) {
-			return e.log.error({ err: String(n).replace(/\n|\r/g, "") }, "SQL generation error"), t.status(500).send(r(n instanceof Error ? n.message : "SQL generation failed", 500));
+			return r(a, await z.generateSQL(l, a, o));
+		} catch (r) {
+			return e.log.error({ err: String(r).replace(/\n|\r/g, "") }, "SQL generation error"), t.status(500).send(n(r instanceof Error ? r.message : "SQL generation failed", 500));
 		}
-	}), e.post(`${O}/dry-run`, {
-		bodyLimit: k,
+	}), o.post(`${N}/dry-run`, {
+		bodyLimit: P,
 		schema: { body: {
 			type: "object",
 			additionalProperties: !0
 		} }
-	}, async (e, t) => {
+	}, async (e, n) => {
 		try {
-			let t = e.body;
-			return await n(t.query || t, await T(e), N);
-		} catch (n) {
-			return e.log.error(n, "Dry-run error"), t.status(400).send({
-				error: n instanceof Error ? n.message : "Dry-run validation failed",
+			let n = e.body;
+			return await t(n.query || n, await R(e), z);
+		} catch (t) {
+			return e.log.error(t, "Dry-run error"), n.status(400).send({
+				error: t instanceof Error ? t.message : "Dry-run validation failed",
 				valid: !1
 			});
 		}
-	}), e.get(`${O}/dry-run`, { schema: { querystring: {
+	}), o.get(`${N}/dry-run`, { schema: { querystring: {
 		type: "object",
 		properties: { query: { type: "string" } },
 		required: ["query"]
-	} } }, async (e, t) => {
+	} } }, async (e, n) => {
 		try {
-			let { query: t } = e.query;
-			return await n(JSON.parse(t), await T(e), N);
-		} catch (n) {
-			return e.log.error(n, "Dry-run error"), t.status(400).send({
-				error: n instanceof Error ? n.message : "Dry-run validation failed",
+			let { query: n } = e.query;
+			return await t(JSON.parse(n), await R(e), z);
+		} catch (t) {
+			return e.log.error(t, "Dry-run error"), n.status(400).send({
+				error: t instanceof Error ? t.message : "Dry-run validation failed",
 				valid: !1
 			});
 		}
-	}), e.post(`${O}/explain`, {
-		bodyLimit: k,
+	}), o.post(`${N}/explain`, {
+		bodyLimit: P,
 		schema: { body: {
 			type: "object",
 			additionalProperties: !0
 		} }
 	}, async (e, t) => {
 		try {
-			let n = e.body, r = n.query || n, i = n.options || {}, a = await T(e), o = N.validateQuery(r);
-			return o.isValid ? await N.explainQuery(r, a, i) : t.status(400).send({ error: `Query validation failed: ${o.errors.join(", ")}` });
+			let n = e.body, r = n.query || n, i = n.options || {}, a = await R(e), o = z.validateQuery(r);
+			return o.isValid ? await z.explainQuery(r, a, i) : t.status(400).send({ error: `Query validation failed: ${o.errors.join(", ")}` });
 		} catch (n) {
 			return e.log.error(n, "Explain error"), t.status(500).send({ error: n instanceof Error ? n.message : "Explain query failed" });
 		}
-	}), M && e.post(`${O}/agent/chat`, {
-		bodyLimit: k,
+	}), L && o.post(`${N}/agent/chat`, {
+		bodyLimit: P,
 		schema: { body: {
 			type: "object",
 			additionalProperties: !0
 		} }
 	}, async (e, t) => {
 		try {
-			let { handleAgentChat: n } = await import("../handler-RItnSaEl.js"), { message: r, sessionId: i, history: a } = e.body;
+			let { handleAgentChat: n } = await import("../handler-3LGcjLtr.js"), { message: r, sessionId: i, history: a } = e.body;
 			if (!r || typeof r != "string") return t.status(400).send({ error: "message is required and must be a string" });
-			let o = (M.apiKey || "").trim();
-			if (M.allowClientApiKey) {
+			let o = (L.apiKey || "").trim();
+			if (L.allowClientApiKey) {
 				let t = e.headers["x-agent-api-key"];
 				t && (o = t.trim());
 			}
 			if (!o) return t.status(401).send({ error: "No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header." });
-			let s = M.allowClientApiKey ? e.headers["x-agent-provider"] : void 0, c = M.allowClientApiKey ? e.headers["x-agent-model"] : void 0, l = M.allowClientApiKey ? e.headers["x-agent-provider-endpoint"] : void 0, u = await T(e), d = M.buildSystemContext?.(u);
+			let s = L.allowClientApiKey ? e.headers["x-agent-provider"] : void 0, c = L.allowClientApiKey ? e.headers["x-agent-model"] : void 0, l = L.allowClientApiKey ? e.headers["x-agent-provider-endpoint"] : void 0, u = await R(e), d = L.buildSystemContext?.(u);
 			t.raw.writeHead(200, {
 				"Content-Type": "text/event-stream",
 				"Cache-Control": "no-cache",
@@ -176,9 +184,9 @@ var b = function(e, b, x) {
 					message: r,
 					sessionId: i,
 					history: a,
-					semanticLayer: N,
+					semanticLayer: z,
 					securityContext: u,
-					agentConfig: M,
+					agentConfig: L,
 					apiKey: o,
 					systemContext: d,
 					providerOverride: s,
@@ -198,64 +206,66 @@ var b = function(e, b, x) {
 		} catch (n) {
 			if (e.log.error(n, "Agent chat error"), !t.raw.headersSent) return t.status(500).send({ error: n instanceof Error ? n.message : "Agent chat failed" });
 		}
-	}), j.enabled !== !1) {
-		let t = j.basePath ?? "/mcp";
-		e.post(`${t}`, {
-			bodyLimit: k,
+	}), I.enabled !== !1) {
+		let e = I.basePath ?? "/mcp";
+		o.post(`${e}`, {
+			bodyLimit: P,
 			schema: { body: {
 				type: "object",
 				additionalProperties: !0
 			} }
 		}, async (e, t) => {
-			let n = c(e.headers.origin, j.allowedOrigins ? { allowedOrigins: j.allowedOrigins } : {});
-			if (!n.valid) return t.status(403).send(m(null, -32600, n.reason));
+			if (I.resourceMetadataUrl && !v(e.headers.authorization)) return t.header("WWW-Authenticate", S(I.resourceMetadataUrl)), t.status(401).send({ error: "Bearer token required" });
+			let n = w(e.headers.origin, I.allowedOrigins ? { allowedOrigins: I.allowedOrigins } : {});
+			if (!n.valid) return t.status(403).send(_(null, -32600, n.reason));
 			let r = e.headers.accept;
-			if (!f(r)) return t.status(400).send(m(null, -32600, "Accept header must include both application/json and text/event-stream"));
-			let i = d(e.headers);
+			if (!C(r)) return t.status(400).send(_(null, -32600, "Accept header must include both application/json and text/event-stream"));
+			let i = y(e.headers);
 			if (!i.ok) return t.status(426).send({
 				error: "Unsupported MCP protocol version",
 				supported: i.supported
 			});
 			let a = g(e.body);
-			if (!a) return t.status(400).send(m(null, -32600, "Invalid JSON-RPC 2.0 request"));
-			let o = y(r), s = a.method === "initialize";
+			if (!a) return t.status(400).send(_(null, -32600, "Invalid JSON-RPC 2.0 request"));
+			let o = p(r), s = a.method === "initialize";
 			try {
-				let n = await v(a.method, a.params, {
-					semanticLayer: N,
-					extractSecurityContext: T,
+				let n = await m(a.method, a.params, {
+					semanticLayer: z,
+					extractSecurityContext: (e, t) => R(e),
 					rawRequest: e,
 					rawResponse: t,
 					negotiatedProtocol: i.negotiated,
-					appEnabled: !!j.app
+					appEnabled: !!I.app
 				});
-				if (u(a)) return t.status(202).send();
+				if (b(a)) return t.status(202).send();
 				let r = s && n && typeof n == "object" && "sessionId" in n ? n.sessionId : void 0;
-				r && t.header(_, r);
-				let c = l(a.id ?? null, n);
+				r && t.header(x, r);
+				let c = f(a.id ?? null, n);
 				if (o) {
 					let e = h();
-					t.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${e}\n\n${p(c, e)}`);
+					t.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${e}\n\n${d(c, e)}`);
 					return;
 				}
 				return t.send(c);
 			} catch (n) {
-				if (u(a)) return e.log.error({ err: String(n).replace(/\n|\r/g, "") }, "MCP notification processing error"), t.status(202).send();
+				if (b(a)) return e.log.error({ err: String(n).replace(/\n|\r/g, "") }, "MCP notification processing error"), t.status(202).send();
 				e.log.error({ err: String(n).replace(/\n|\r/g, "") }, "MCP RPC error");
-				let r = n?.code ?? -32603, i = n?.data, s = n.message || "MCP request failed", c = m(a.id ?? null, r, s, i);
+				let r = n?.code ?? -32603, i = n?.data, s = n.message || "MCP request failed", c = _(a.id ?? null, r, s, i);
 				if (o) {
 					let e = h();
-					t.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${e}\n\n${p(c, e)}`);
+					t.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${e}\n\n${d(c, e)}`);
 					return;
 				}
 				return t.send(c);
 			}
-		}), e.get(`${t}`, async (e, t) => {
+		}), o.get(`${e}`, async (e, t) => {
+			if (I.resourceMetadataUrl && !v(e.headers.authorization)) return t.header("WWW-Authenticate", S(I.resourceMetadataUrl)), t.status(401).send({ error: "Bearer token required" });
 			let n = h();
 			t.raw.writeHead(200, {
 				"Content-Type": "text/event-stream",
 				"Cache-Control": "no-cache",
 				Connection: "keep-alive"
-			}), t.raw.write(p({
+			}), t.raw.write(d({
 				jsonrpc: "2.0",
 				method: "mcp/ready",
 				params: { protocol: "streamable-http" }
@@ -266,16 +276,16 @@ var b = function(e, b, x) {
 			e.raw.on("close", () => {
 				clearInterval(r);
 			});
-		}), e.delete(`${t}`, async (e, t) => t.status(405).send({ error: "Session termination not supported" }));
+		}), o.delete(`${e}`, async (e, t) => I.resourceMetadataUrl && !v(e.headers.authorization) ? (t.header("WWW-Authenticate", S(I.resourceMetadataUrl)), t.status(401).send({ error: "Bearer token required" })) : t.status(405).send({ error: "Session termination not supported" }));
 	}
-	e.setErrorHandler(async (e, t, n) => (t.log.error(e, "Fastify cube adapter error"), n.statusCode < 400 && n.status(500), r(e instanceof Error ? e : String(e), n.statusCode))), x();
+	o.setErrorHandler(async (e, t, r) => (t.log.error(e, "Fastify cube adapter error"), r.statusCode < 400 && r.status(500), n(e instanceof Error ? e : String(e), r.statusCode))), E();
 };
-async function x(e, t) {
-	await e.register(b, t);
+async function E(e, t) {
+	await e.register(T, t);
 }
-function S(t) {
-	let n = e("fastify")({ logger: !0 });
-	return n.register(b, t), n;
+function D(e) {
+	let t = o("fastify")({ logger: !0 });
+	return t.register(T, e), t;
 }
 //#endregion
-export { S as createCubeApp, b as cubePlugin, x as registerCubeRoutes };
+export { D as createCubeApp, T as cubePlugin, E as registerCubeRoutes };