drizzle-cube 0.5.7 → 0.6.0

package/dist/adapters/mcp-tools.cjs

@@ -1 +1 @@
-Object.defineProperty(exports,Symbol.toStringTag,{value:`Module`});const e=require("./utils-CkEkU8Bo.cjs"),t=require("./mcp-transport-C4J4SAcn.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:o,getSecurityContext:s,toolPrefix:c=`drizzle_cube_`,tools:l=[`discover`,`validate`,`load`,`chart`],prompts:u=t.d(),resources:d,app:f=!1}=i,p=!!f,m=typeof f==`object`?f:void 0,h=d??t.f(),g=p?[...h,...a(m)]:h,_=t.s({appEnabled:p}),v=new Map(_.map(e=>[e.name,e])),y=l.filter(e=>v.has(e)).map(e=>{let t=v.get(e),n={name:`${c}${e}`,description:t.description,inputSchema:t.inputSchema},r=t._meta;return r&&(n._meta=r),n}),b=y.map(e=>e.name),x=new Set;for(let e of l)x.add(e),x.add(`${c}${e}`);function S(e){return x.has(e)}async function C(t,i,a){let u=t.startsWith(c)?t.slice(c.length):t;if(!l.includes(u))return r(`Unknown tool: ${t}`);try{switch(u){case`discover`:return n(await e.d(o,i||{}));case`validate`:{let t=i||{};if(!t.query)return r(`query is required`);let c;try{c=await s(a)}catch{}return n(await e.h(o,t,c))}case`load`:{let t=i||{};return t.query?n(await e.p(o,await s(a),t)):r(`query is required`)}case`chart`:{let t=i||{};return t.query?n(await e.p(o,await s(a),t)):r(`query is required`)}default:return r(`Unknown tool: ${t}`)}}catch(e){return r(e)}}return{definitions:y,handle:C,handles:S,prompts:u,resources:g,toolNames:b}}function a(e){let n=t.p(e);return n?[{uri:t.n,name:`Drizzle Cube Visualization`,description:`Interactive chart visualization for query results`,mimeType:t.t,text:n}]:[]}exports.getCubeTools=i;
+Object.defineProperty(exports,Symbol.toStringTag,{value:`Module`});const e=require("./utils-_whV4fXL.cjs"),t=require("./mcp-transport-Chiaj4ll.cjs");function n(t,n){return e.p(t.semanticLayer,n||{}).then(t.wrapContent)}async function r(t,n,r){let i=n||{};if(!i.query)return t.wrapError(`query is required`);let a;try{a=await t.getSecurityContext(r)}catch{}return t.wrapContent(await e.f(t.semanticLayer,i,a))}async function i(t,n,r){let i=n||{};if(!i.query)return t.wrapError(`query is required`);let a=await t.getSecurityContext(r);return t.wrapContent(await e.m(t.semanticLayer,a,i))}function a(e){return{content:[{type:`text`,text:typeof e==`string`?e:JSON.stringify(e)}],isError:!1}}function o(e){return{content:[{type:`text`,text:e instanceof Error?e.message:String(e)}],isError:!0}}function s(e){let{semanticLayer:s,getSecurityContext:l,toolPrefix:u=`drizzle_cube_`,tools:d=[`discover`,`validate`,`load`,`chart`],prompts:f=t.d(),resources:p,app:m=!1}=e,h=!!m,g=typeof m==`object`?m:void 0,_=p??t.f(),v=h?[..._,...c(g)]:_,y=t.s({appEnabled:h}),b=new Map(y.map(e=>[e.name,e])),x=d.filter(e=>b.has(e)).map(e=>{let t=b.get(e),n={name:`${u}${e}`,description:t.description,inputSchema:t.inputSchema},r=t._meta;return r&&(n._meta=r),n}),S=x.map(e=>e.name),C=new Set;for(let e of d)C.add(e),C.add(`${u}${e}`);function w(e){return C.has(e)}let T={semanticLayer:s,getSecurityContext:l,wrapContent:a,wrapError:o},E={discover:e=>n(T,e),validate:(e,t)=>r(T,e,t),load:(e,t)=>i(T,e,t),chart:(e,t)=>i(T,e,t)};async function D(e,t,n){let r=e.startsWith(u)?e.slice(u.length):e,i=d.includes(r)?E[r]:void 0;if(!i)return o(`Unknown tool: ${e}`);try{return await i(t,n)}catch(e){return o(e)}}return{definitions:x,handle:D,handles:w,prompts:f,resources:v,toolNames:S}}function c(e){let n=t.p(e);return n?[{uri:t.n,name:`Drizzle Cube Visualization`,description:`Interactive chart visualization for query results`,mimeType:t.t,text:n}]:[]}exports.getCubeTools=s;

package/dist/adapters/mcp-tools.js

@@ -1,7 +1,27 @@
-import { d as e, h as t, p as n } from "./utils-DosuoabY.js";
-import { d as r, f as i, n as a, p as o, s, t as c } from "./mcp-transport-CVUiifEc.js";
+import { f as e, m as t, p as n } from "./utils-BPWZsX7k.js";
+import { d as r, f as i, n as a, p as o, s, t as c } from "./mcp-transport-H96VG3WY.js";
+//#region src/adapters/mcp-tools-handlers.ts
+function l(e, t) {
+	return n(e.semanticLayer, t || {}).then(e.wrapContent);
+}
+async function u(t, n, r) {
+	let i = n || {};
+	if (!i.query) return t.wrapError("query is required");
+	let a;
+	try {
+		a = await t.getSecurityContext(r);
+	} catch {}
+	return t.wrapContent(await e(t.semanticLayer, i, a));
+}
+async function d(e, n, r) {
+	let i = n || {};
+	if (!i.query) return e.wrapError("query is required");
+	let a = await e.getSecurityContext(r);
+	return e.wrapContent(await t(e.semanticLayer, a, i));
+}
+//#endregion
 //#region src/adapters/mcp-tools.ts
-function l(e) {
+function f(e) {
 	return {
 		content: [{
 			type: "text",
@@ -10,7 +30,7 @@ function l(e) {
 		isError: !1
 	};
 }
-function u(e) {
+function p(e) {
 	return {
 		content: [{
 			type: "text",
@@ -19,63 +39,54 @@ function u(e) {
 		isError: !0
 	};
 }
-function d(a) {
-	let { semanticLayer: o, getSecurityContext: c, toolPrefix: d = "drizzle_cube_", tools: p = [
+function m(e) {
+	let { semanticLayer: t, getSecurityContext: n, toolPrefix: a = "drizzle_cube_", tools: o = [
 		"discover",
 		"validate",
 		"load",
 		"chart"
-	], prompts: m = r(), resources: h, app: g = !1 } = a, _ = !!g, v = typeof g == "object" ? g : void 0, y = h ?? i(), b = _ ? [...y, ...f(v)] : y, x = s({ appEnabled: _ }), S = new Map(x.map((e) => [e.name, e])), C = p.filter((e) => S.has(e)).map((e) => {
+	], prompts: c = r(), resources: m, app: g = !1 } = e, _ = !!g, v = typeof g == "object" ? g : void 0, y = m ?? i(), b = _ ? [...y, ...h(v)] : y, x = s({ appEnabled: _ }), S = new Map(x.map((e) => [e.name, e])), C = o.filter((e) => S.has(e)).map((e) => {
 		let t = S.get(e), n = {
-			name: `${d}${e}`,
+			name: `${a}${e}`,
 			description: t.description,
 			inputSchema: t.inputSchema
 		}, r = t._meta;
 		return r && (n._meta = r), n;
 	}), w = C.map((e) => e.name), T = /* @__PURE__ */ new Set();
-	for (let e of p) T.add(e), T.add(`${d}${e}`);
+	for (let e of o) T.add(e), T.add(`${a}${e}`);
 	function E(e) {
 		return T.has(e);
 	}
-	async function D(r, i, a) {
-		let s = r.startsWith(d) ? r.slice(d.length) : r;
-		if (!p.includes(s)) return u(`Unknown tool: ${r}`);
+	let D = {
+		semanticLayer: t,
+		getSecurityContext: n,
+		wrapContent: f,
+		wrapError: p
+	}, O = {
+		discover: (e) => l(D, e),
+		validate: (e, t) => u(D, e, t),
+		load: (e, t) => d(D, e, t),
+		chart: (e, t) => d(D, e, t)
+	};
+	async function k(e, t, n) {
+		let r = e.startsWith(a) ? e.slice(a.length) : e, i = o.includes(r) ? O[r] : void 0;
+		if (!i) return p(`Unknown tool: ${e}`);
 		try {
-			switch (s) {
-				case "discover": return l(await e(o, i || {}));
-				case "validate": {
-					let e = i || {};
-					if (!e.query) return u("query is required");
-					let n;
-					try {
-						n = await c(a);
-					} catch {}
-					return l(await t(o, e, n));
-				}
-				case "load": {
-					let e = i || {};
-					return e.query ? l(await n(o, await c(a), e)) : u("query is required");
-				}
-				case "chart": {
-					let e = i || {};
-					return e.query ? l(await n(o, await c(a), e)) : u("query is required");
-				}
-				default: return u(`Unknown tool: ${r}`);
-			}
+			return await i(t, n);
 		} catch (e) {
-			return u(e);
+			return p(e);
 		}
 	}
 	return {
 		definitions: C,
-		handle: D,
+		handle: k,
 		handles: E,
-		prompts: m,
+		prompts: c,
 		resources: b,
 		toolNames: w
 	};
 }
-function f(e) {
+function h(e) {
 	let t = o(e);
 	return t ? [{
 		uri: a,
@@ -86,4 +97,4 @@ function f(e) {
 	}] : [];
 }
 //#endregion
-export { d as getCubeTools };
+export { m as getCubeTools };