drizzle-cube 0.4.42 → 0.4.44

package/dist/adapters/hono/index.js

@@ -1,5 +1,6 @@
-import { a as e, f as t, o as n, r, u as i } from "../utils-_2o905Fi.js";
-import { a, c as o, d as s, f as c, h as l, i as u, l as d, m as f, n as p, o as m, p as h, r as g, s as _, t as v, u as y } from "../mcp-transport-tB5a7Het.js";
+import { a as e, f as t, o as n, r, u as i } from "../utils-C7Nrw9Wb.js";
+import { t as a } from "../compiler-O3T1u7jl.js";
+import { a as o, c as s, d as c, f as l, h as u, l as d, m as f, n as p, o as m, p as h, r as g, s as _, t as v, u as y } from "../mcp-transport--zhJJHJc.js";
 import { Hono as b } from "hono";
 //#region node_modules/hono/dist/middleware/cors/index.js
 var x = (e) => {
@@ -16,14 +17,14 @@ var x = (e) => {
 		allowHeaders: [],
 		exposeHeaders: [],
 		...e
-	}, n = ((e) => typeof e == "string" ? e === "*" ? () => e : (t) => e === t ? t : null : typeof e == "function" ? e : (t) => e.includes(t) ? t : null)(t.origin), r = ((e) => typeof e == "function" ? e : Array.isArray(e) ? () => e : () => [])(t.allowMethods);
+	}, n = ((e) => typeof e == "string" ? e === "*" ? t.credentials ? (e) => e || null : () => e : (t) => e === t ? t : null : typeof e == "function" ? e : (t) => e.includes(t) ? t : null)(t.origin), r = ((e) => typeof e == "function" ? e : Array.isArray(e) ? () => e : () => [])(t.allowMethods);
 	return async function(e, i) {
 		function a(t, n) {
 			e.res.headers.set(t, n);
 		}
 		let o = await n(e.req.header("origin") || "", e);
 		if (o && a("Access-Control-Allow-Origin", o), t.credentials && a("Access-Control-Allow-Credentials", "true"), t.exposeHeaders?.length && a("Access-Control-Expose-Headers", t.exposeHeaders.join(",")), e.req.method === "OPTIONS") {
-			t.origin !== "*" && a("Vary", "Origin"), t.maxAge != null && a("Access-Control-Max-Age", t.maxAge.toString());
+			(t.origin !== "*" || t.credentials) && a("Vary", "Origin"), t.maxAge != null && a("Access-Control-Max-Age", t.maxAge.toString());
 			let n = await r(e.req.header("origin") || "", e);
 			n.length && a("Access-Control-Allow-Methods", n.join(","));
 			let i = t.allowHeaders;
@@ -37,7 +38,7 @@ var x = (e) => {
 				statusText: "No Content"
 			});
 		}
-		await i(), t.origin !== "*" && e.header("Vary", "Origin", { append: !0 });
+		await i(), (t.origin !== "*" || t.credentials) && e.header("Vary", "Origin", { append: !0 });
 	};
 };
 //#endregion
@@ -47,7 +48,7 @@ function S(S) {
 	if (!S.semanticLayer && (!C || C.length === 0)) throw Error("Either semanticLayer or a non-empty cubes array must be provided");
 	let N = new b();
 	O && N.use("/*", x(O));
-	let P = S.semanticLayer ?? new l({
+	let P = S.semanticLayer ?? new a({
 		drizzle: w,
 		schema: T,
 		engineType: D,
@@ -159,7 +160,7 @@ function S(S) {
 		}
 	}), M && N.post(`${k}/agent/chat`, async (e) => {
 		try {
-			let { handleAgentChat: t } = await import("../handler-fto6TSVn.js"), { message: n, sessionId: r, history: i } = await e.req.json();
+			let { handleAgentChat: t } = await import("../handler-CjVc3ytc.js"), { message: n, sessionId: r, history: i } = await e.req.json();
 			if (!n || typeof n != "string") return e.json({ error: "message is required and must be a string" }, 400);
 			let a = (M.apiKey || "").trim();
 			if (M.allowClientApiKey) {
@@ -214,22 +215,22 @@ function S(S) {
 			description: "Current cube metadata as JSON",
 			mimeType: "application/json",
 			text: JSON.stringify(P.getMetadata(), null, 2)
-		}, t = [...m(), e], n = a(), r = j.basePath ?? "/mcp";
+		}, t = [..._(), e], n = m(), r = j.basePath ?? "/mcp";
 		N.post(`${r}`, async (e) => {
-			let r = h(e.req.header("origin"), j.allowedOrigins ? { allowedOrigins: j.allowedOrigins } : {});
+			let r = f(e.req.header("origin"), j.allowedOrigins ? { allowedOrigins: j.allowedOrigins } : {});
 			if (!r.valid) return e.json(p(null, -32600, r.reason), 403);
 			let i = e.req.header("accept");
-			if (!c(i)) return e.json(p(null, -32600, "Accept header must include both application/json and text/event-stream"), 400);
-			let a = o(e.req.header());
+			if (!h(i)) return e.json(p(null, -32600, "Accept header must include both application/json and text/event-stream"), 400);
+			let a = d(e.req.header());
 			if (!a.ok) return e.json({
 				error: "Unsupported MCP protocol version",
 				supported: a.supported
 			}, 426);
-			let l = d(await e.req.json().catch(() => null));
-			if (!l) return e.json(p(null, -32600, "Invalid JSON-RPC 2.0 request"), 400);
-			let m = f(i), b = l.method === "initialize";
+			let m = y(await e.req.json().catch(() => null));
+			if (!m) return e.json(p(null, -32600, "Invalid JSON-RPC 2.0 request"), 400);
+			let _ = u(i), b = m.method === "initialize";
 			try {
-				let r = await u(l.method, l.params, {
+				let r = await o(m.method, m.params, {
 					semanticLayer: P,
 					extractSecurityContext: E,
 					rawRequest: e,
@@ -238,11 +239,11 @@ function S(S) {
 					resources: t,
 					prompts: n
 				});
-				if (_(l)) return e.body(null, 202);
-				let i = g(l.id ?? null, r), o = b && r && typeof r == "object" && "sessionId" in r ? r.sessionId : void 0, c = {};
-				if (o && (c[v] = o), m) {
-					let e = new TextEncoder(), t = y(), n = new ReadableStream({ start(n) {
-						n.enqueue(e.encode(`id: ${t}\n\n`)), n.enqueue(e.encode(s(i, t))), n.close();
+				if (s(m)) return e.body(null, 202);
+				let i = g(m.id ?? null, r), u = b && r && typeof r == "object" && "sessionId" in r ? r.sessionId : void 0, d = {};
+				if (u && (d[v] = u), _) {
+					let e = new TextEncoder(), t = c(), n = new ReadableStream({ start(n) {
+						n.enqueue(e.encode(`id: ${t}\n\n`)), n.enqueue(e.encode(l(i, t))), n.close();
 					} });
 					return new Response(n, {
 						status: 200,
@@ -250,18 +251,18 @@ function S(S) {
 							"Content-Type": "text/event-stream",
 							"Cache-Control": "no-cache",
 							Connection: "keep-alive",
-							...c
+							...d
 						}
 					});
 				}
-				return e.json(i, 200, c);
+				return e.json(i, 200, d);
 			} catch (t) {
-				if (_(l)) return console.error("MCP notification processing error:", t), e.body(null, 202);
+				if (s(m)) return console.error("MCP notification processing error:", t), e.body(null, 202);
 				console.error("MCP RPC error:", t);
-				let n = t?.code ?? -32603, r = t?.data, i = t.message || "MCP request failed", a = p(l.id ?? null, n, i, r);
-				if (m) {
-					let e = new TextEncoder(), t = y(), n = new ReadableStream({ start(n) {
-						n.enqueue(e.encode(`id: ${t}\n\n`)), n.enqueue(e.encode(s(a, t))), n.close();
+				let n = t?.code ?? -32603, r = t?.data, i = t.message || "MCP request failed", a = p(m.id ?? null, n, i, r);
+				if (_) {
+					let e = new TextEncoder(), t = c(), n = new ReadableStream({ start(n) {
+						n.enqueue(e.encode(`id: ${t}\n\n`)), n.enqueue(e.encode(l(a, t))), n.close();
 					} });
 					return new Response(n, {
 						status: 200,
@@ -275,9 +276,9 @@ function S(S) {
 				return e.json(a, 200);
 			}
 		}), N.delete(`${r}`, (e) => e.json({ error: "Session termination not supported" }, 405)), N.get(`${r}`, (e) => {
-			let t = new TextEncoder(), n = y(), r, i = new ReadableStream({
+			let t = new TextEncoder(), n = c(), r, i = new ReadableStream({
 				start(e) {
-					e.enqueue(t.encode(s({
+					e.enqueue(t.encode(l({
 						jsonrpc: "2.0",
 						method: "mcp/ready",
 						params: { protocol: "streamable-http" }

package/dist/adapters/mcp-tools.cjs

@@ -0,0 +1 @@
+Object.defineProperty(exports,Symbol.toStringTag,{value:`Module`});const e=require(`./utils-tNZ6Cvzw.cjs`),t=require(`./mcp-transport-MOoCDu2M.cjs`);function n(e){return{content:[{type:`text`,text:typeof e==`string`?e:JSON.stringify(e)}],isError:!1}}function r(e){return{content:[{type:`text`,text:e instanceof Error?e.message:String(e)}],isError:!0}}function i(i){let{semanticLayer:a,getSecurityContext:o,toolPrefix:s=`drizzle_cube_`,tools:c=[`discover`,`validate`,`load`],prompts:l=t.o(),resources:u=t.s()}=i,d=t.i(),f=new Map(d.map(e=>[e.name,e])),p=c.filter(e=>f.has(e)).map(e=>{let t=f.get(e);return{name:`${s}${e}`,description:t.description,inputSchema:t.inputSchema}}),m=p.map(e=>e.name),h=new Set;for(let e of c)h.add(e),h.add(`${s}${e}`);function g(e){return h.has(e)}async function _(t,i,l){let u=t.startsWith(s)?t.slice(s.length):t;if(!c.includes(u))return r(`Unknown tool: ${t}`);try{switch(u){case`discover`:return n(await e.d(a,i||{}));case`validate`:{let t=i||{};return t.query?n(await e.h(a,t)):r(`query is required`)}case`load`:{let t=i||{};return t.query?n(await e.p(a,await o(l),t)):r(`query is required`)}default:return r(`Unknown tool: ${t}`)}}catch(e){return r(e)}}return{definitions:p,handle:_,handles:g,prompts:l,resources:u,toolNames:m}}exports.getCubeTools=i;

package/dist/adapters/mcp-tools.d.ts

@@ -0,0 +1,128 @@
+import { SemanticLayerCompiler, SecurityContext } from '../server';
+import { DiscoverRequest, ValidateRequest, LoadRequest } from './utils';
+import { MCPPrompt } from '../server/ai/mcp-prompts';
+import { MCPResource } from './mcp-transport';
+export type { MCPPrompt, MCPResource, DiscoverRequest, ValidateRequest, LoadRequest };
+export type { SecurityContext } from '../server';
+/**
+ * MCP tool content block (matches MCP spec)
+ */
+export interface MCPToolContent {
+    type: 'text';
+    text: string;
+}
+/**
+ * MCP tool result (matches MCP spec CallToolResult)
+ */
+export interface MCPToolResult {
+    content: MCPToolContent[];
+    isError: boolean;
+}
+/**
+ * MCP tool definition (matches MCP spec Tool)
+ */
+export interface MCPToolDefinition {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: 'object';
+        required?: string[];
+        properties: Record<string, unknown>;
+    };
+}
+/**
+ * Options for creating composable cube tools
+ */
+export interface GetCubeToolsOptions {
+    /** The semantic layer compiler instance with registered cubes */
+    semanticLayer: SemanticLayerCompiler;
+    /**
+     * Extract security context for query execution.
+     * Called when the `load` tool is invoked.
+     * Receives whatever metadata your MCP server provides (auth info, request context, etc.)
+     */
+    getSecurityContext: (meta?: unknown) => SecurityContext | Promise<SecurityContext>;
+    /**
+     * Optional tool name prefix. Defaults to 'drizzle_cube_'.
+     * Set to '' for no prefix, or provide your own like 'analytics_'.
+     */
+    toolPrefix?: string;
+    /**
+     * Which tools to expose. Defaults to all: ['discover', 'validate', 'load']
+     */
+    tools?: ('discover' | 'validate' | 'load')[];
+    /** Custom MCP prompts (defaults to built-in drizzle-cube prompts) */
+    prompts?: MCPPrompt[];
+    /** Custom MCP resources (defaults to built-in drizzle-cube resources) */
+    resources?: MCPResource[];
+}
+/**
+ * The composable tools object returned by getCubeTools()
+ */
+export interface CubeTools {
+    /** Tool definitions for MCP tools/list responses */
+    definitions: MCPToolDefinition[];
+    /**
+     * Execute a tool by name. Returns MCP-spec-compliant result.
+     * @param name - Tool name (with or without prefix)
+     * @param args - Tool arguments from the MCP client
+     * @param meta - Optional metadata passed to getSecurityContext
+     */
+    handle: (name: string, args: unknown, meta?: unknown) => Promise<MCPToolResult>;
+    /**
+     * Check if a tool name is handled by these cube tools
+     * @param name - Tool name to check (with or without prefix)
+     */
+    handles: (name: string) => boolean;
+    /** MCP prompts for prompts/list responses */
+    prompts: MCPPrompt[];
+    /** MCP resources for resources/list responses */
+    resources: MCPResource[];
+    /** The tool names this instance handles (with prefix applied) */
+    toolNames: string[];
+}
+/**
+ * Create composable MCP tools for your semantic layer.
+ *
+ * Returns tool definitions, a handler function, and prompts/resources
+ * that you can register on any MCP server.
+ *
+ * @example With @modelcontextprotocol/sdk
+ * ```typescript
+ * import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+ * import { ListToolsRequestSchema, CallToolRequestSchema } from '@modelcontextprotocol/sdk/types.js'
+ * import { getCubeTools } from 'drizzle-cube/mcp'
+ *
+ * const cubeTools = getCubeTools({
+ *   semanticLayer,
+ *   getSecurityContext: async (meta) => ({ orgId: meta.authInfo?.orgId })
+ * })
+ *
+ * const server = new Server({ name: 'my-server', version: '1.0.0' })
+ *
+ * server.setRequestHandler(ListToolsRequestSchema, async () => ({
+ *   tools: [...myOtherTools, ...cubeTools.definitions]
+ * }))
+ *
+ * server.setRequestHandler(CallToolRequestSchema, async (req) => {
+ *   if (cubeTools.handles(req.params.name)) {
+ *     return cubeTools.handle(req.params.name, req.params.arguments, req)
+ *   }
+ *   return handleOtherTools(req)
+ * })
+ * ```
+ *
+ * @example With Hono + postgrest-mcp style
+ * ```typescript
+ * import { getCubeTools } from 'drizzle-cube/mcp'
+ *
+ * const cubeTools = getCubeTools({
+ *   semanticLayer,
+ *   getSecurityContext: async () => ({ orgId: 'default' })
+ * })
+ *
+ * // Merge with your existing PostgREST tools
+ * const allTools = [...postgrestTools, ...cubeTools.definitions]
+ * ```
+ */
+export declare function getCubeTools(options: GetCubeToolsOptions): CubeTools;

package/dist/adapters/mcp-tools.js

@@ -0,0 +1,69 @@
+import { d as e, h as t, p as n } from "./utils-C7Nrw9Wb.js";
+import { i as r, o as i, s as a } from "./mcp-transport--zhJJHJc.js";
+//#region src/adapters/mcp-tools.ts
+function o(e) {
+	return {
+		content: [{
+			type: "text",
+			text: typeof e == "string" ? e : JSON.stringify(e)
+		}],
+		isError: !1
+	};
+}
+function s(e) {
+	return {
+		content: [{
+			type: "text",
+			text: e instanceof Error ? e.message : String(e)
+		}],
+		isError: !0
+	};
+}
+function c(c) {
+	let { semanticLayer: l, getSecurityContext: u, toolPrefix: d = "drizzle_cube_", tools: f = [
+		"discover",
+		"validate",
+		"load"
+	], prompts: p = i(), resources: m = a() } = c, h = r(), g = new Map(h.map((e) => [e.name, e])), _ = f.filter((e) => g.has(e)).map((e) => {
+		let t = g.get(e);
+		return {
+			name: `${d}${e}`,
+			description: t.description,
+			inputSchema: t.inputSchema
+		};
+	}), v = _.map((e) => e.name), y = /* @__PURE__ */ new Set();
+	for (let e of f) y.add(e), y.add(`${d}${e}`);
+	function b(e) {
+		return y.has(e);
+	}
+	async function x(r, i, a) {
+		let c = r.startsWith(d) ? r.slice(d.length) : r;
+		if (!f.includes(c)) return s(`Unknown tool: ${r}`);
+		try {
+			switch (c) {
+				case "discover": return o(await e(l, i || {}));
+				case "validate": {
+					let e = i || {};
+					return e.query ? o(await t(l, e)) : s("query is required");
+				}
+				case "load": {
+					let e = i || {};
+					return e.query ? o(await n(l, await u(a), e)) : s("query is required");
+				}
+				default: return s(`Unknown tool: ${r}`);
+			}
+		} catch (e) {
+			return s(e);
+		}
+	}
+	return {
+		definitions: _,
+		handle: x,
+		handles: b,
+		prompts: p,
+		resources: m,
+		toolNames: v
+	};
+}
+//#endregion
+export { c as getCubeTools };