drizzle-cube 0.4.53 → 0.5.1

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-S6KHiOY6.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 { S as d, _ as f, a as p, b as m, c as h, g, h as _, i as v, l as y, m as b, o as x, r as S, u as C, v as w, x as T, y as E } from "../mcp-transport-C7VLf4T5.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 D = function(o, D, O) {
+	let { cubes: k, drizzle: A, schema: j, extractSecurityContext: M, engineType: N, cors: P, basePath: F = "/cubejs-api/v1", bodyLimit: I = 10485760, cache: L, mcp: R = { enabled: !0 }, agent: z } = D;
+	if (!k || k.length === 0) return O(/* @__PURE__ */ Error("At least one cube must be provided in the cubes array"));
+	let B = async (e) => l(await M(e), c((t) => e.headers[t.toLowerCase()]));
+	if (P) {
+		let e = {
+			...P,
+			allowedHeaders: u(P.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 V = new s({
+		drizzle: A,
+		schema: j,
+		engineType: N,
+		cache: L,
+		rlsSetup: D.rlsSetup
 	});
-	if (S.forEach((e) => {
-		N.registerCube(e);
-	}), e.post(`${O}/load`, {
-		bodyLimit: k,
+	if (k.forEach((e) => {
+		V.registerCube(e);
+	}), o.post(`${F}/load`, {
+		bodyLimit: I,
 		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 B(e), s = V.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 V.executeMultiCubeQuery(a, o, { skipCache: c }), V);
+		} 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(`${F}/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 B(e), s = V.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 V.executeMultiCubeQuery(a, o, { skipCache: c });
+			return i(a, l, V);
+		} 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(`${F}/batch`, {
+		bodyLimit: I,
 		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 B(e), V, { 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(`${F}/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(V.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(`${F}/sql`, {
+		bodyLimit: I,
 		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 B(e), o = V.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 V.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(`${F}/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 B(e), s = V.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 V.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(`${F}/dry-run`, {
+		bodyLimit: I,
 		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 B(e), V);
+		} 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(`${F}/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 B(e), V);
+		} 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(`${F}/explain`, {
+		bodyLimit: I,
 		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 B(e), o = V.validateQuery(r);
+			return o.isValid ? await V.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,
+	}), z && o.post(`${F}/agent/chat`, {
+		bodyLimit: I,
 		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 = (z.apiKey || "").trim();
+			if (z.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 = z.allowClientApiKey ? e.headers["x-agent-provider"] : void 0, c = z.allowClientApiKey ? e.headers["x-agent-model"] : void 0, l = z.allowClientApiKey ? e.headers["x-agent-provider-endpoint"] : void 0, u = await B(e), d = z.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: V,
 					securityContext: u,
-					agentConfig: M,
+					agentConfig: z,
 					apiKey: o,
 					systemContext: d,
 					providerOverride: s,
@@ -198,64 +206,70 @@ 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,
+	}), R.enabled !== !1) {
+		let e = x(V, R.resources), t = w(R.prompts), n = R.basePath ?? "/mcp";
+		o.post(`${n}`, {
+			bodyLimit: I,
 			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));
-			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 (!i.ok) return t.status(426).send({
+		}, async (n, r) => {
+			if (R.resourceMetadataUrl && !C(n.headers.authorization)) return r.header("WWW-Authenticate", h(R.resourceMetadataUrl)), r.status(401).send({ error: "Bearer token required" });
+			let i = T(n.headers.origin, R.allowedOrigins ? { allowedOrigins: R.allowedOrigins } : {});
+			if (!i.valid) return r.status(403).send(v(null, -32600, i.reason));
+			let a = n.headers.accept;
+			if (!m(a)) return r.status(400).send(v(null, -32600, "Accept header must include both application/json and text/event-stream"));
+			let o = _(n.headers);
+			if (!o.ok) return r.status(426).send({
 				error: "Unsupported MCP protocol version",
-				supported: i.supported
+				supported: o.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";
+			let s = g(n.body);
+			if (!s) return r.status(400).send(v(null, -32600, "Invalid JSON-RPC 2.0 request"));
+			let c = d(a), l = s.method === "initialize";
 			try {
-				let n = await v(a.method, a.params, {
-					semanticLayer: N,
-					extractSecurityContext: T,
-					rawRequest: e,
-					rawResponse: t,
-					negotiatedProtocol: i.negotiated,
-					appEnabled: !!j.app
+				let i = await y(s.method, s.params, {
+					semanticLayer: V,
+					extractSecurityContext: (e, t) => B(e),
+					rawRequest: n,
+					rawResponse: r,
+					negotiatedProtocol: o.negotiated,
+					resources: e,
+					prompts: t,
+					appEnabled: !!R.app,
+					appConfig: typeof R.app == "object" ? R.app : void 0,
+					serverName: R.serverName
 				});
-				if (u(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);
-				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)}`);
+				if (b(s)) return r.status(202).send();
+				let a = l && i && typeof i == "object" && "sessionId" in i ? i.sessionId : void 0;
+				a && r.header(S, a);
+				let u = p(s.id ?? null, i);
+				if (c) {
+					let e = f();
+					r.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${e}\n\n${E(u, 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();
-				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);
-				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)}`);
+				return r.send(u);
+			} catch (e) {
+				if (b(s)) return n.log.error({ err: String(e).replace(/\n|\r/g, "") }, "MCP notification processing error"), r.status(202).send();
+				n.log.error({ err: String(e).replace(/\n|\r/g, "") }, "MCP RPC error");
+				let t = e?.code ?? -32603, i = e?.data, a = e.message || "MCP request failed", o = v(s.id ?? null, t, a, i);
+				if (c) {
+					let e = f();
+					r.header("Content-Type", "text/event-stream").header("Cache-Control", "no-cache").header("Connection", "keep-alive").send(`id: ${e}\n\n${E(o, e)}`);
 					return;
 				}
-				return t.send(c);
+				return r.send(o);
 			}
-		}), e.get(`${t}`, async (e, t) => {
-			let n = h();
+		}), o.get(`${n}`, async (e, t) => {
+			if (R.resourceMetadataUrl && !C(e.headers.authorization)) return t.header("WWW-Authenticate", h(R.resourceMetadataUrl)), t.status(401).send({ error: "Bearer token required" });
+			let n = f();
 			t.raw.writeHead(200, {
 				"Content-Type": "text/event-stream",
 				"Cache-Control": "no-cache",
 				Connection: "keep-alive"
-			}), t.raw.write(p({
+			}), t.raw.write(E({
 				jsonrpc: "2.0",
 				method: "mcp/ready",
 				params: { protocol: "streamable-http" }
@@ -266,16 +280,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(`${n}`, async (e, t) => R.resourceMetadataUrl && !C(e.headers.authorization) ? (t.header("WWW-Authenticate", h(R.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))), O();
 };
-async function x(e, t) {
-	await e.register(b, t);
+async function O(e, t) {
+	await e.register(D, t);
 }
-function S(t) {
-	let n = e("fastify")({ logger: !0 });
-	return n.register(b, t), n;
+function k(e) {
+	let t = o("fastify")({ logger: !0 });
+	return t.register(D, e), t;
 }
 //#endregion
-export { S as createCubeApp, b as cubePlugin, x as registerCubeRoutes };
+export { k as createCubeApp, D as cubePlugin, O as registerCubeRoutes };