drizzle-cube 0.4.28 → 0.4.30

package/dist/adapters/hono/index.cjs

@@ -1,11 +1,3 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const M=require("hono"),a=require("../mcp-transport-DCiSGtp1.cjs"),R=require("../utils.cjs");var L=g=>{const y={...{origin:"*",allowMethods:["GET","HEAD","PUT","POST","DELETE","PATCH"],allowHeaders:[],exposeHeaders:[]},...g},T=(d=>typeof d=="string"?d==="*"?()=>d:c=>d===c?c:null:typeof d=="function"?d:c=>d.includes(c)?c:null)(y.origin),h=(d=>typeof d=="function"?d:Array.isArray(d)?()=>d:()=>[])(y.allowMethods);return async function(c,q){function b(f,i){c.res.headers.set(f,i)}const E=await T(c.req.header("origin")||"",c);if(E&&b("Access-Control-Allow-Origin",E),y.credentials&&b("Access-Control-Allow-Credentials","true"),y.exposeHeaders?.length&&b("Access-Control-Expose-Headers",y.exposeHeaders.join(",")),c.req.method==="OPTIONS"){y.origin!=="*"&&b("Vary","Origin"),y.maxAge!=null&&b("Access-Control-Max-Age",y.maxAge.toString());const f=await h(c.req.header("origin")||"",c);f.length&&b("Access-Control-Allow-Methods",f.join(","));let i=y.allowHeaders;if(!i?.length){const s=c.req.header("Access-Control-Request-Headers");s&&(i=s.split(/\s*,\s*/))}return i?.length&&(b("Access-Control-Allow-Headers",i.join(",")),c.res.headers.append("Vary","Access-Control-Request-Headers")),c.res.headers.delete("Content-Length"),c.res.headers.delete("Content-Type"),new Response(null,{headers:c.res.headers,status:204,statusText:"No Content"})}await q(),y.origin!=="*"&&c.header("Vary","Origin",{append:!0})}};function N(g){const{cubes:j,drizzle:y,schema:T,extractSecurityContext:h,engineType:d,cors:c,basePath:q="/cubejs-api/v1",cache:b,mcp:E={enabled:!0},agent:f}=g;if(!g.semanticLayer&&(!j||j.length===0))throw new Error("Either semanticLayer or a non-empty cubes array must be provided");const i=new M.Hono;c&&i.use("/*",L(c));const s=g.semanticLayer??new a.SemanticLayerCompiler({drizzle:y,schema:T,engineType:d,cache:b});if(!g.semanticLayer&&j&&j.forEach(e=>{s.registerCube(e)}),i.post(`${q}/load`,async e=>{try{const r=await e.req.json(),n=r.query||r,o=await h(e),t=s.validateQuery(n);if(!t.isValid)return e.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const u=e.req.header("x-cache-control")==="no-cache",l=await s.executeMultiCubeQuery(n,o,{skipCache:u});return e.json(R.formatCubeResponse(n,l,s))}catch(r){return console.error("Query execution error:",r),e.json({error:r instanceof Error?r.message:"Query execution failed"},500)}}),i.get(`${q}/load`,async e=>{try{const r=e.req.query("query");if(!r)return e.json({error:"Query parameter is required"},400);let n;try{n=JSON.parse(r)}catch{return e.json({error:"Invalid JSON in query parameter"},400)}const o=await h(e),t=s.validateQuery(n);if(!t.isValid)return e.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const u=e.req.header("x-cache-control")==="no-cache",l=await s.executeMultiCubeQuery(n,o,{skipCache:u});return e.json(R.formatCubeResponse(n,l,s))}catch(r){return console.error("Query execution error:",r),e.json({error:r instanceof Error?r.message:"Query execution failed"},500)}}),i.post(`${q}/batch`,async e=>{try{const r=await e.req.json(),{queries:n}=r;if(!n||!Array.isArray(n))return e.json({error:'Request body must contain a "queries" array'},400);if(n.length===0)return e.json({error:"Queries array cannot be empty"},400);const o=await h(e),t=e.req.header("x-cache-control")==="no-cache",u=await R.handleBatchRequest(n,o,s,{skipCache:t});return e.json(u)}catch(r){return console.error("Batch execution error:",r),e.json({error:r instanceof Error?r.message:"Batch execution failed"},500)}}),i.get(`${q}/meta`,e=>{try{const r=s.getMetadata();return e.json(R.formatMetaResponse(r))}catch(r){return console.error("Metadata error:",r),e.json({error:r instanceof Error?r.message:"Failed to fetch metadata"},500)}}),i.post(`${q}/sql`,async e=>{try{const r=await e.req.json(),n=await h(e),o=s.validateQuery(r);if(!o.isValid)return e.json({error:`Query validation failed: ${o.errors.join(", ")}`},400);const t=r.measures?.[0]||r.dimensions?.[0];if(!t)return e.json({error:"No measures or dimensions specified"},400);const u=t.split(".")[0],l=await s.generateSQL(u,r,n);return e.json(R.formatSqlResponse(r,l))}catch(r){return console.error("SQL generation error:",r),e.json({error:r instanceof Error?r.message:"SQL generation failed"},500)}}),i.get(`${q}/sql`,async e=>{try{const r=e.req.query("query");if(!r)return e.json({error:"Query parameter is required"},400);const n=JSON.parse(r),o=await h(e),t=s.validateQuery(n);if(!t.isValid)return e.json({error:`Query validation failed: ${t.errors.join(", ")}`},400);const u=n.measures?.[0]||n.dimensions?.[0];if(!u)return e.json({error:"No measures or dimensions specified"},400);const l=u.split(".")[0],v=await s.generateSQL(l,n,o);return e.json(R.formatSqlResponse(n,v))}catch(r){return console.error("SQL generation error:",r),e.json({error:r instanceof Error?r.message:"SQL generation failed"},500)}}),i.post(`${q}/dry-run`,async e=>{try{const r=await e.req.json(),n=r.query||r,o=await h(e),t=await R.handleDryRun(n,o,s);return e.json(t)}catch(r){return console.error("Dry-run error:",r),e.json({error:r instanceof Error?r.message:"Dry-run validation failed",valid:!1},400)}}),i.get(`${q}/dry-run`,async e=>{try{const r=e.req.query("query");if(!r)return e.json({error:"Query parameter is required",valid:!1},400);const n=JSON.parse(r),o=await h(e),t=await R.handleDryRun(n,o,s);return e.json(t)}catch(r){return console.error("Dry-run error:",r),e.json({error:r instanceof Error?r.message:"Dry-run validation failed",valid:!1},400)}}),i.post(`${q}/explain`,async e=>{try{const r=await e.req.json(),n=r.query||r,o=r.options||{},t=await h(e),u=s.validateQuery(n);if(!u.isValid)return e.json({error:`Query validation failed: ${u.errors.join(", ")}`},400);const l=await s.explainQuery(n,t,o);return e.json(l)}catch(r){return console.error("Explain error:",r),e.json({error:r instanceof Error?r.message:"Explain query failed"},500)}}),f&&i.post(`${q}/agent/chat`,async e=>{try{const{handleAgentChat:r}=await Promise.resolve().then(()=>require("../handler-LMRPeTNJ.cjs")),n=await e.req.json(),{message:o,sessionId:t,history:u}=n;if(!o||typeof o!="string")return e.json({error:"message is required and must be a string"},400);let l=(f.apiKey||"").trim();if(f.allowClientApiKey){const C=e.req.header("x-agent-api-key");C&&(l=C.trim())}if(!l)return e.json({error:"No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header."},401);const v=f.allowClientApiKey?e.req.header("x-agent-provider"):void 0,Q=f.allowClientApiKey?e.req.header("x-agent-model"):void 0,m=f.allowClientApiKey?e.req.header("x-agent-provider-endpoint"):void 0,P=await h(e),H=f.buildSystemContext?.(P),p=new TextEncoder,A=new ReadableStream({async start(C){try{const w=r({message:o,sessionId:t,history:u,semanticLayer:s,securityContext:P,agentConfig:f,apiKey:l,systemContext:H,providerOverride:v,modelOverride:Q,baseURLOverride:m});for await(const x of w){const S=`data: ${JSON.stringify(x)}
+Object.defineProperty(exports,Symbol.toStringTag,{value:`Module`});const e=require(`../utils-CwJplXR5.cjs`),t=require(`../mcp-transport-Dpp6hdZe.cjs`);let n=require(`hono`);var r=e=>{let t={origin:`*`,allowMethods:[`GET`,`HEAD`,`PUT`,`POST`,`DELETE`,`PATCH`],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);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());let n=await r(e.req.header(`origin`)||``,e);n.length&&a(`Access-Control-Allow-Methods`,n.join(`,`));let i=t.allowHeaders;if(!i?.length){let t=e.req.header(`Access-Control-Request-Headers`);t&&(i=t.split(/\s*,\s*/))}return i?.length&&(a(`Access-Control-Allow-Headers`,i.join(`,`)),e.res.headers.append(`Vary`,`Access-Control-Request-Headers`)),e.res.headers.delete(`Content-Length`),e.res.headers.delete(`Content-Type`),new Response(null,{headers:e.res.headers,status:204,statusText:`No Content`})}await i(),t.origin!==`*`&&e.header(`Vary`,`Origin`,{append:!0})}};function i(i){let{cubes:a,drizzle:o,schema:s,extractSecurityContext:c,engineType:l,cors:u,basePath:d=`/cubejs-api/v1`,cache:f,mcp:p={enabled:!0},agent:m}=i;if(!i.semanticLayer&&(!a||a.length===0))throw Error(`Either semanticLayer or a non-empty cubes array must be provided`);let h=new n.Hono;u&&h.use(`/*`,r(u));let g=i.semanticLayer??new t.h({drizzle:o,schema:s,engineType:l,cache:f});if(!i.semanticLayer&&a&&a.forEach(e=>{g.registerCube(e)}),h.post(`${d}/load`,async t=>{try{let n=await t.req.json(),r=n.query||n,i=await c(t),a=g.validateQuery(r);if(!a.isValid)return t.json({error:`Query validation failed: ${a.errors.join(`, `)}`},400);let o=t.req.header(`x-cache-control`)===`no-cache`,s=await g.executeMultiCubeQuery(r,i,{skipCache:o});return t.json(e.r(r,s,g))}catch(e){return console.error(`Query execution error:`,e),t.json({error:e instanceof Error?e.message:`Query execution failed`},500)}}),h.get(`${d}/load`,async t=>{try{let n=t.req.query(`query`);if(!n)return t.json({error:`Query parameter is required`},400);let r;try{r=JSON.parse(n)}catch{return t.json({error:`Invalid JSON in query parameter`},400)}let i=await c(t),a=g.validateQuery(r);if(!a.isValid)return t.json({error:`Query validation failed: ${a.errors.join(`, `)}`},400);let o=t.req.header(`x-cache-control`)===`no-cache`,s=await g.executeMultiCubeQuery(r,i,{skipCache:o});return t.json(e.r(r,s,g))}catch(e){return console.error(`Query execution error:`,e),t.json({error:e instanceof Error?e.message:`Query execution failed`},500)}}),h.post(`${d}/batch`,async t=>{try{let{queries:n}=await t.req.json();if(!n||!Array.isArray(n))return t.json({error:`Request body must contain a "queries" array`},400);if(n.length===0)return t.json({error:`Queries array cannot be empty`},400);let r=await e.u(n,await c(t),g,{skipCache:t.req.header(`x-cache-control`)===`no-cache`});return t.json(r)}catch(e){return console.error(`Batch execution error:`,e),t.json({error:e instanceof Error?e.message:`Batch execution failed`},500)}}),h.get(`${d}/meta`,t=>{try{let n=g.getMetadata();return t.json(e.a(n))}catch(e){return console.error(`Metadata error:`,e),t.json({error:e instanceof Error?e.message:`Failed to fetch metadata`},500)}}),h.post(`${d}/sql`,async t=>{try{let n=await t.req.json(),r=await c(t),i=g.validateQuery(n);if(!i.isValid)return t.json({error:`Query validation failed: ${i.errors.join(`, `)}`},400);let a=n.measures?.[0]||n.dimensions?.[0];if(!a)return t.json({error:`No measures or dimensions specified`},400);let o=a.split(`.`)[0],s=await g.generateSQL(o,n,r);return t.json(e.o(n,s))}catch(e){return console.error(`SQL generation error:`,e),t.json({error:e instanceof Error?e.message:`SQL generation failed`},500)}}),h.get(`${d}/sql`,async t=>{try{let n=t.req.query(`query`);if(!n)return t.json({error:`Query parameter is required`},400);let r=JSON.parse(n),i=await c(t),a=g.validateQuery(r);if(!a.isValid)return t.json({error:`Query validation failed: ${a.errors.join(`, `)}`},400);let o=r.measures?.[0]||r.dimensions?.[0];if(!o)return t.json({error:`No measures or dimensions specified`},400);let s=o.split(`.`)[0],l=await g.generateSQL(s,r,i);return t.json(e.o(r,l))}catch(e){return console.error(`SQL generation error:`,e),t.json({error:e instanceof Error?e.message:`SQL generation failed`},500)}}),h.post(`${d}/dry-run`,async t=>{try{let n=await t.req.json(),r=await e.f(n.query||n,await c(t),g);return t.json(r)}catch(e){return console.error(`Dry-run error:`,e),t.json({error:e instanceof Error?e.message:`Dry-run validation failed`,valid:!1},400)}}),h.get(`${d}/dry-run`,async t=>{try{let n=t.req.query(`query`);if(!n)return t.json({error:`Query parameter is required`,valid:!1},400);let r=await e.f(JSON.parse(n),await c(t),g);return t.json(r)}catch(e){return console.error(`Dry-run error:`,e),t.json({error:e instanceof Error?e.message:`Dry-run validation failed`,valid:!1},400)}}),h.post(`${d}/explain`,async e=>{try{let t=await e.req.json(),n=t.query||t,r=t.options||{},i=await c(e),a=g.validateQuery(n);if(!a.isValid)return e.json({error:`Query validation failed: ${a.errors.join(`, `)}`},400);let o=await g.explainQuery(n,i,r);return e.json(o)}catch(t){return console.error(`Explain error:`,t),e.json({error:t instanceof Error?t.message:`Explain query failed`},500)}}),m&&h.post(`${d}/agent/chat`,async e=>{try{let{handleAgentChat:t}=await Promise.resolve().then(()=>require(`../handler-B8vuFQYP.cjs`)),{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){let t=e.req.header(`x-agent-api-key`);t&&(a=t.trim())}if(!a)return e.json({error:`No API key configured. Set agent.apiKey in server config or send X-Agent-Api-Key header.`},401);let o=m.allowClientApiKey?e.req.header(`x-agent-provider`):void 0,s=m.allowClientApiKey?e.req.header(`x-agent-model`):void 0,l=m.allowClientApiKey?e.req.header(`x-agent-provider-endpoint`):void 0,u=await c(e),d=m.buildSystemContext?.(u),f=new TextEncoder,p=new ReadableStream({async start(e){try{let c=t({message:n,sessionId:r,history:i,semanticLayer:g,securityContext:u,agentConfig:m,apiKey:a,systemContext:d,providerOverride:o,modelOverride:s,baseURLOverride:l});for await(let t of c){let n=`data: ${JSON.stringify(t)}\n\n`;e.enqueue(f.encode(n))}}catch(t){let n={type:`error`,data:{message:t instanceof Error?t.message:`Stream failed`}};e.enqueue(f.encode(`data: ${JSON.stringify(n)}\n\n`))}finally{e.close()}}});return new Response(p,{status:200,headers:{"Content-Type":`text/event-stream`,"Cache-Control":`no-cache`,Connection:`keep-alive`}})}catch(t){return console.error(`Agent chat error:`,t),e.json({error:t instanceof Error?t.message:`Agent chat failed`},500)}}),p.enabled!==!1){let e={uri:`drizzle-cube://schema`,name:`Cube Schema`,description:`Current cube metadata as JSON`,mimeType:`application/json`,text:JSON.stringify(g.getMetadata(),null,2)},n=[...t.o(),e],r=t.a(),i=p.basePath??`/mcp`;h.post(`${i}`,async e=>{let i=t.p(e.req.header(`origin`),p.allowedOrigins?{allowedOrigins:p.allowedOrigins}:{});if(!i.valid)return e.json(t.n(null,-32600,i.reason),403);let a=e.req.header(`accept`);if(!t.f(a))return e.json(t.n(null,-32600,`Accept header must include both application/json and text/event-stream`),400);let o=t.c(e.req.header());if(!o.ok)return e.json({error:`Unsupported MCP protocol version`,supported:o.supported},426);let s=t.l(await e.req.json().catch(()=>null));if(!s)return e.json(t.n(null,-32600,`Invalid JSON-RPC 2.0 request`),400);let l=t.m(a),u=s.method===`initialize`;try{let i=await t.i(s.method,s.params,{semanticLayer:g,extractSecurityContext:c,rawRequest:e,rawResponse:null,negotiatedProtocol:o.negotiated,resources:n,prompts:r});if(t.s(s))return e.body(null,202);let a=t.r(s.id??null,i),d=u&&i&&typeof i==`object`&&`sessionId`in i?i.sessionId:void 0,f={};if(d&&(f[t.t]=d),l){let e=new TextEncoder,n=t.u(),r=new ReadableStream({start(r){r.enqueue(e.encode(`id: ${n}\n\n`)),r.enqueue(e.encode(t.d(a,n))),r.close()}});return new Response(r,{status:200,headers:{"Content-Type":`text/event-stream`,"Cache-Control":`no-cache`,Connection:`keep-alive`,...f}})}return e.json(a,200,f)}catch(n){if(t.s(s))return console.error(`MCP notification processing error:`,n),e.body(null,202);console.error(`MCP RPC error:`,n);let r=n?.code??-32603,i=n?.data,a=n.message||`MCP request failed`,o=t.n(s.id??null,r,a,i);if(l){let e=new TextEncoder,n=t.u(),r=new ReadableStream({start(r){r.enqueue(e.encode(`id: ${n}\n\n`)),r.enqueue(e.encode(t.d(o,n))),r.close()}});return new Response(r,{status:200,headers:{"Content-Type":`text/event-stream`,"Cache-Control":`no-cache`,Connection:`keep-alive`}})}return e.json(o,200)}}),h.delete(`${i}`,e=>e.json({error:`Session termination not supported`},405)),h.get(`${i}`,e=>{let n=new TextEncoder,r=t.u(),i,a=new ReadableStream({start(e){e.enqueue(n.encode(t.d({jsonrpc:`2.0`,method:`mcp/ready`,params:{protocol:`streamable-http`}},r,15e3))),i=setInterval(()=>{e.enqueue(n.encode(`: keep-alive
 
-`;C.enqueue(p.encode(S))}}catch(w){const x={type:"error",data:{message:w instanceof Error?w.message:"Stream failed"}};C.enqueue(p.encode(`data: ${JSON.stringify(x)}
-
-`))}finally{C.close()}}});return new Response(A,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})}catch(r){return console.error("Agent chat error:",r),e.json({error:r instanceof Error?r.message:"Agent chat failed"},500)}}),E.enabled!==!1){const e={uri:"drizzle-cube://schema",name:"Cube Schema",description:"Current cube metadata as JSON",mimeType:"application/json",text:JSON.stringify(s.getMetadata(),null,2)},r=[...a.getDefaultResources(),e],n=a.getDefaultPrompts(),o=E.basePath??"/mcp";i.post(`${o}`,async t=>{const u=a.validateOriginHeader(t.req.header("origin"),E.allowedOrigins?{allowedOrigins:E.allowedOrigins}:{});if(!u.valid)return t.json(a.buildJsonRpcError(null,-32600,u.reason),403);const l=t.req.header("accept");if(!a.validateAcceptHeader(l))return t.json(a.buildJsonRpcError(null,-32600,"Accept header must include both application/json and text/event-stream"),400);const v=a.negotiateProtocol(t.req.header());if(!v.ok)return t.json({error:"Unsupported MCP protocol version",supported:v.supported},426);const Q=await t.req.json().catch(()=>null),m=a.parseJsonRpc(Q);if(!m)return t.json(a.buildJsonRpcError(null,-32600,"Invalid JSON-RPC 2.0 request"),400);const P=a.wantsEventStream(l),H=m.method==="initialize";try{const p=await a.dispatchMcpMethod(m.method,m.params,{semanticLayer:s,extractSecurityContext:h,rawRequest:t,rawResponse:null,negotiatedProtocol:v.negotiated,resources:r,prompts:n});if(a.isNotification(m))return t.body(null,202);const A=a.buildJsonRpcResult(m.id??null,p),C=H&&p&&typeof p=="object"&&"sessionId"in p?p.sessionId:void 0,w={};if(C&&(w[a.MCP_SESSION_ID_HEADER]=C),P){const x=new TextEncoder,S=a.primeEventId(),O=new ReadableStream({start($){$.enqueue(x.encode(`id: ${S}
-
-`)),$.enqueue(x.encode(a.serializeSseEvent(A,S))),$.close()}});return new Response(O,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive",...w}})}return t.json(A,200,w)}catch(p){if(a.isNotification(m))return console.error("MCP notification processing error:",p),t.body(null,202);console.error("MCP RPC error:",p);const A=p?.code??-32603,C=p?.data,w=p.message||"MCP request failed",x=a.buildJsonRpcError(m.id??null,A,w,C);if(P){const S=new TextEncoder,O=a.primeEventId(),$=new ReadableStream({start(I){I.enqueue(S.encode(`id: ${O}
-
-`)),I.enqueue(S.encode(a.serializeSseEvent(x,O))),I.close()}});return new Response($,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})}return t.json(x,200)}}),i.delete(`${o}`,t=>t.json({error:"Session termination not supported"},405)),i.get(`${o}`,t=>{const u=new TextEncoder,l=a.primeEventId();let v;const Q=new ReadableStream({start(m){m.enqueue(u.encode(a.serializeSseEvent({jsonrpc:"2.0",method:"mcp/ready",params:{protocol:"streamable-http"}},l,15e3))),v=setInterval(()=>{m.enqueue(u.encode(`: keep-alive
-
-`))},15e3)},cancel(){clearInterval(v)}});return new Response(Q,{status:200,headers:{"Content-Type":"text/event-stream","Cache-Control":"no-cache",Connection:"keep-alive"}})})}return i}function J(g,j){const y=N(j);return g.route("/",y),g}function D(g){const j=new M.Hono;return J(j,g)}exports.createCubeApp=D;exports.createCubeRoutes=N;exports.mountCubeRoutes=J;
+`))},15e3)},cancel(){clearInterval(i)}});return new Response(a,{status:200,headers:{"Content-Type":`text/event-stream`,"Cache-Control":`no-cache`,Connection:`keep-alive`}})})}return h}function a(e,t){let n=i(t);return e.route(`/`,n),e}function o(e){return a(new n.Hono,e)}exports.createCubeApp=o,exports.createCubeRoutes=i,exports.mountCubeRoutes=a;