@prisma/extension-optimize 0.0.0-dev.202412140052 → 0.0.0-dev.202412161942

package/dist/index.js

@@ -1,2 +1,2 @@
-"use strict";var Se=Object.create;var x=Object.defineProperty;var de=Object.getOwnPropertyDescriptor;var he=Object.getOwnPropertyNames;var fe=Object.getPrototypeOf,ge=Object.prototype.hasOwnProperty;var xe=(n,e)=>{for(var t in e)x(n,t,{get:e[t],enumerable:!0})},F=(n,e,t,s)=>{if(e&&typeof e=="object"||typeof e=="function")for(let r of he(e))!ge.call(n,r)&&r!==t&&x(n,r,{get:()=>e[r],enumerable:!(s=de(e,r))||s.enumerable});return n};var Te=(n,e,t)=>(t=n!=null?Se(fe(n)):{},F(e||!n||!n.__esModule?x(t,"default",{value:n,enumerable:!0}):t,n)),ye=n=>F(x({},"__esModule",{value:!0}),n);var Fe={};xe(Fe,{PROD_DASHBOARD_URL:()=>se,PROD_INGESTION_URL:()=>ae,withOptimize:()=>ze});module.exports=ye(Fe);var te=require("@opentelemetry/api"),re=require("@prisma/client/extension");function Ie(n,e){let t;return async function(s,r,i){let p=new URL(s,n),c=new Headers({"Content-Type":"application/json",Authorization:`Bearer ${e}`});t&&c.set("prisma-optimize-jwt",t);let l=await fetch(p,{method:r,headers:c,body:JSON.stringify(i)});if(!l.ok){let g=await l.clone().text().catch(()=>"<unreadable>");console.error(`[optimize] HTTP ${l.status} ${l.statusText}: ${g}`)}let f=l.headers.get("prisma-optimize-jwt");return f&&(t=f),l}}function k(n,e){let t=Ie(n,e);return{request:t,post:(s,r)=>t(s,"POST",r)}}var H=Te(require("kleur"));var{bold:be,underline:Re}=H.default;function M(n){let e="See your Optimize dashboard at:",t=e.length,s=`${Re(be(n))}`,r=n.length,i=Math.max(t,r)+2;console.log("\u250C"+"\u2500".repeat(i)+"\u2510"),console.log("\u2502 "+e+" ".repeat(i-t-2)+" \u2502"),console.log("\u2502 "+s+" ".repeat(i-r-2)+" \u2502"),console.log("\u2514"+"\u2500".repeat(i)+"\u2518")}var D=require("@opentelemetry/api"),N=require("@opentelemetry/context-async-hooks"),J=require("@opentelemetry/instrumentation"),V=require("@opentelemetry/resources"),G=require("@opentelemetry/sdk-trace-base"),R=require("@opentelemetry/semantic-conventions"),Z=require("@prisma/instrumentation");async function B(n,e){e.length>0&&await n.post("/ingest",e)}async function $(n,e){return await(await n.post("/schema",e)).text()}function u(n){return n[0]*1e3+n[1]/1e6}var a=require("zod");var m=require("zod"),T=class{spans={};rootSpans={};waitSpans={};treeShape;collectedSpans;collectedRootSpans;sendTimeout;sendInterval;pendingSend;constructor(e){this.treeShape=e.treeShape,this.sendTimeout=e.sendTimeout,this.sendInterval=e.sendInterval,this.collectedSpans=e.collectedSpans,this.collectedRootSpans=e.collectedRootSpans,this.onEnd=ve(()=>void this.forceFlush(),this.sendInterval)}onStart(e){let{parentSpanId:t,name:s}=e,{spanId:r}=e.spanContext();if(!this.collectedSpans.includes(s))return this.cleanSpan(this.spans[r]);this.spans[r]&&this.spans[r].kind==="future"?Object.assign(this.spans[r],{kind:"present",value:e}):this.spans[r]={spanId:r,kind:"present",children:[],value:e,retries:0},this.collectedRootSpans.includes(s)&&(this.rootSpans[r]={...this.spans[r],done:!1},this.waitSpans[r]=this.spans[r]),t!==void 0&&(this.spans[t]?this.spans[t].children.push(this.spans[r]):(this.spans[t]={spanId:t,kind:"future",children:[this.spans[r]],retries:0},this.waitSpans[t]=this.spans[t]))}cleanSpan(e){if(e===void 0)return;let t=e.spanId;for(let s of e.children)this.cleanSpan(s);delete this.spans[t],delete this.rootSpans[t],delete this.waitSpans[t],this.onClean(e)}async forceFlush(){await this.pendingSend;let e=Object.values(this.rootSpans).flatMap(t=>{if(t.done)return[];let s=this.treeShape.safeParse(t);return s.data?(t.done=!0,[{original:t,pruned:s.data}]):[]});e.length>0&&(this.pendingSend=this.onFlush(e).finally(()=>{e.forEach(({original:t})=>this.cleanSpan(t))}));for(let t of Object.values(this.waitSpans))++t.retries*this.sendInterval>this.sendTimeout&&this.cleanSpan(t)}async shutdown(){await this.forceFlush(),this.spans={},this.rootSpans={},this.waitSpans={}}onEnd(){}};function ve(n,e){let t=!1;return()=>{t===!1&&(t=!0,setTimeout(()=>(t=!1)||n(),e))}}var o={ended:m.z.literal(!0),_spanContext:m.z.custom(),spanContext:m.z.custom(),startTime:m.z.custom(),endTime:m.z.custom(),attributes:m.z.custom()};var L=a.z.object({value:a.z.object({name:a.z.literal("prisma:engine:db_query"),...o})}),qe=a.z.object({value:a.z.object({name:a.z.literal("prisma:engine:itx_execute_single"),...o}),children:S([L])}),y=a.z.object({value:a.z.object({name:a.z.union([a.z.literal("prisma:engine"),a.z.literal("prisma:engine:query")]),...o}),children:S([L])}),Ce=a.z.object({value:a.z.object({name:a.z.union([a.z.literal("prisma:engine"),a.z.literal("prisma:engine:query")]),...o}),children:S([qe])}),I=a.z.object({value:a.z.object({name:a.z.literal("prisma:client:connect"),...o})}),q=a.z.object({value:a.z.object({name:a.z.literal("prisma:client:operation"),...o}),children:a.z.union([a.z.tuple([y]),a.z.tuple([I,y]),a.z.tuple([y,I])])}),C=a.z.object({value:a.z.object({name:a.z.literal("prisma:client:operation"),...o}),children:S([Ce])}),we=a.z.object({value:a.z.object({name:a.z.union([a.z.literal("prisma:engine:start_transaction"),a.z.literal("prisma:engine:commit_transaction")]),...o})}),w=a.z.object({value:a.z.object({name:a.z.literal("prisma:client:operation"),...o}),children:a.z.array(I)}),O=a.z.object({value:a.z.object({name:a.z.literal("prisma:client:transaction"),...o}),children:S([we,C,I])}),P=a.z.object({value:a.z.object({name:a.z.literal("prisma:client:transaction"),...o}),children:S([w,y])}),U=a.z.union([q,O,P]);function S(n){return a.z.array(a.z.any()).nonempty().superRefine((e,t)=>{n.every(r=>e.some(i=>r.safeParse(i).success))||t.addIssue({code:"custom",message:"Array does not contain all the required items"})})}var b=class extends T{requests={};errors={};schemaHash;resolvedSchemaHash;apiClient;schemaRequest;constructor(e){super({treeShape:U,sendTimeout:e.sendTimeout,sendInterval:e.sendInterval,collectedSpans:["prisma:client:transaction","prisma:client:operation","prisma:client:connect","prisma:engine","prisma:engine:query","prisma:engine:db_query","prisma:engine:itx_execute_single","prisma:engine:start_transaction","prisma:engine:commit_transaction"],collectedRootSpans:["prisma:client:operation","prisma:client:transaction"]}),this.apiClient=e.apiClient,this.schemaRequest=e.schemaRequest}setRequest(e){this.requests[e.spanId]=e}setError(e){this.errors[e.spanId]=e}onClean({spanId:e}){delete this.requests[e],delete this.errors[e]}async onFlush(e){this.schemaHash??=$(this.apiClient,this.schemaRequest),this.resolvedSchemaHash??=await this.schemaHash;let t=e.flatMap(({pruned:s})=>Oe(s,this));await B(this.apiClient,t)}};function Oe(n,e){return O.safeParse(n).success?Pe(n,e):P.safeParse(n).success?Ee(n,e):q.safeParse(n).success?je(n,e):[]}function Pe(n,e){let t=[],s=n.children.filter(r=>C.safeParse(r).success);for(let r of s){if(j({span:r,ip:e})===!1)continue;let i=_e(r);t.push(d({ip:e,span:r,sql:i,connect:!1}))}if(t.length){let r=E(n);t.unshift(d({ip:e,span:n,sql:null,connect:r}))}return t}function Ee(n,e){let t=[],s=n.children.filter(i=>w.safeParse(i).success),r=!1;for(let i of s)j({span:i,ip:e})!==!1&&(r||=E(i),t.push(d({ip:e,span:i,sql:null,connect:!1})));return t.length&&t.unshift(d({ip:e,span:n,sql:null,connect:r})),t}function je(n,e){let t=E(n),s=Ae(n);return j({span:n,ip:e})===!1?[]:[d({ip:e,span:n,sql:s,connect:t})]}function d(n){let{span:e,sql:t,connect:s,ip:r}=n,i=e.value.spanContext().spanId,{value:p}=e;return{hash:r.resolvedSchemaHash,sql:t??null,connect:s??!1,error:r.errors[i]?.error??null,ts:u(p.startTime),model:r.requests[i]?.model??p.attributes.model??null,operation:r.requests[i]?.operation??p.attributes.method,args:r.requests[i]?.args??{},latency:u(p.endTime)-u(p.startTime)}}function Q(n){return n.sort((t,s)=>u(t.value.startTime)-u(s.value.startTime)).reduce((t,s)=>{let r=s.value.attributes["db.statement"];return typeof r=="string"?`${t}${r}
-`:t},"")}function Ae(n){let{children:e}=n.children.find(({value:s})=>s.name==="prisma:engine"||s.name==="prisma:engine:query"),t=e.filter(({value:s})=>s.name==="prisma:engine:db_query");return Q(t)}function _e(n){let{children:e}=n.children.find(({value:r})=>r.name==="prisma:engine"||r.name==="prisma:engine:query"),s=e.filter(({value:r})=>r.name==="prisma:engine:itx_execute_single").flatMap(({children:r})=>r.filter(({value:i})=>i.name==="prisma:engine:db_query"));return Q(s)}function E(n){let{value:e}=n.children.find(({value:t})=>t.name==="prisma:client:connect")??{};return e?u(e.endTime)-u(e.startTime):!1}function j(n){let{span:e,ip:t}=n;return!!t.requests[e.value.spanContext().spanId]}var h;function K(n){if(h)return h;let e=new N.AsyncHooksContextManager().enable();D.context.setGlobalContextManager(e);let t=new G.BasicTracerProvider({resource:new V.Resource({[R.SEMRESATTRS_SERVICE_NAME]:"extension-optimize",[R.SEMRESATTRS_SERVICE_VERSION]:"0.0.0"})});return h=new b(n),t.addSpanProcessor(h),(0,J.registerInstrumentations)({tracerProvider:t,instrumentations:[new Z.PrismaInstrumentation]}),t.register(),h}function W(n){return n._previewFeatures?.includes("tracing")}function X(n){return n._engineConfig.inlineSchema}function Y(n){if(n instanceof Error)return n.stack??n.message;switch(typeof n){case"undefined":return"undefined";case"object":{let e;return n!==null&&typeof n.toString=="function"&&(e=n.toString()),typeof e=="string"&&e!=="[object Object]"?e:JSON.stringify(n)}default:return String(n)}}var ee=require("@prisma/client/extension"),ne=ee.Prisma.defineExtension(n=>n.$extends({}));var ae="https://optimize-ingestion.datacdn.workers.dev/",se="https://optimize.prisma.io";function ze({enable:n=!0,ingestionUrl:e=ae,dashboardUrl:t=se,sendInterval:s=500,sendTimeout:r=1e4,showNotification:i=!0,apiKey:p}){return n?(i&&M(t),re.Prisma.defineExtension(c=>{let l=X(c),f=Reflect.get(c,"_activeProvider"),g=Reflect.get(c,"_accelerateEngineConfig"),{protocol:ie}=new URL(g.accelerateUtils.resolveDatasourceUrl(g)),oe={schema:l,provider:f,protocol:ie},pe=k(e,p),A=K({schemaRequest:oe,apiClient:pe,sendInterval:s,sendTimeout:r});if(!W(c))throw new Error('Please enable the "tracing" preview feature and regenerate the client.');return c.$extends({query:{async $allOperations({query:ce,model:le,operation:ue,args:_}){let v=te.trace.getActiveSpan()?.spanContext().spanId;if(!v)throw new Error("prisma:client:operation span is expected to be entered in the client extension when tracing is enabled");A.setRequest({spanId:v,model:le,operation:ue,args:_});try{return await ce(_)}catch(z){let me=Y(z);throw A.setError({spanId:v,error:me}),z}}}})})):ne}0&&(module.exports={PROD_DASHBOARD_URL,PROD_INGESTION_URL,withOptimize});
+"use strict";var te=Object.create;var S=Object.defineProperty;var ne=Object.getOwnPropertyDescriptor;var se=Object.getOwnPropertyNames;var re=Object.getPrototypeOf,ie=Object.prototype.hasOwnProperty;var oe=(t,e)=>{for(var n in e)S(t,n,{get:e[n],enumerable:!0})},v=(t,e,n,r)=>{if(e&&typeof e=="object"||typeof e=="function")for(let s of se(e))!ie.call(t,s)&&s!==n&&S(t,s,{get:()=>e[s],enumerable:!(r=ne(e,s))||r.enumerable});return t};var ae=(t,e,n)=>(n=t!=null?te(re(t)):{},v(e||!t||!t.__esModule?S(n,"default",{value:t,enumerable:!0}):n,t)),pe=t=>v(S({},"__esModule",{value:!0}),t);var be={};oe(be,{PROD_DASHBOARD_URL:()=>B,PROD_INGESTION_URL:()=>V,withOptimize:()=>ge});module.exports=pe(be);var J=require("@opentelemetry/api"),Q=require("@prisma/client/extension");function ce(t,e){let n;return async function(r,s,o){let a=new URL(r,t),p=new Headers({"Content-Type":"application/json",Authorization:`Bearer ${e}`});n&&p.set("prisma-optimize-jwt",n);let c=await fetch(a,{method:s,headers:p,body:JSON.stringify(o)});if(!c.ok){let f=await c.clone().text().catch(()=>"<unreadable>");console.error(`[optimize] HTTP ${c.status} ${c.statusText}: ${f}`)}let h=c.headers.get("prisma-optimize-jwt");return h&&(n=h),c}}function w(t,e){let n=ce(t,e);return{request:n,post:(r,s)=>n(r,"POST",s)}}var q=ae(require("kleur"));var{bold:le,underline:me}=q.default;function P(t){let e="See your Optimize dashboard at:",n=e.length,r=`${me(le(t))}`,s=t.length,o=Math.max(n,s)+2;console.log("\u250C"+"\u2500".repeat(o)+"\u2510"),console.log("\u2502 "+e+" ".repeat(o-n-2)+" \u2502"),console.log("\u2502 "+r+" ".repeat(o-s-2)+" \u2502"),console.log("\u2514"+"\u2500".repeat(o)+"\u2518")}var j=require("@opentelemetry/api"),_=require("@opentelemetry/context-async-hooks"),z=require("@opentelemetry/instrumentation"),k=require("@opentelemetry/resources"),F=require("@opentelemetry/sdk-trace-base"),R=require("@opentelemetry/semantic-conventions"),M=require("@prisma/instrumentation");var i=require("zod");async function A(t,e){e.length>0&&await t.post("/ingest",e)}async function E(t,e){return await(await t.post("/schema",e)).text()}function l(t){return t[0]*1e3+t[1]/1e6}var m=require("zod"),g=class{spans={};rootSpans={};waitSpans={};treeShape;collectedSpans;collectedRootSpans;sendTimeout;sendInterval;pendingSend;constructor(e){this.treeShape=e.treeShape,this.sendTimeout=e.sendTimeout,this.sendInterval=e.sendInterval,this.collectedSpans=e.collectedSpans,this.collectedRootSpans=e.collectedRootSpans,this.onEnd=ue(()=>void this.forceFlush(),this.sendInterval)}onStart(e){let{parentSpanId:n,name:r}=e,{spanId:s}=e.spanContext();if(!this.collectedSpans.includes(r))return this.cleanSpan(this.spans[s]);this.spans[s]&&this.spans[s].kind==="future"?Object.assign(this.spans[s],{kind:"present",value:e}):this.spans[s]={spanId:s,kind:"present",children:[],value:e,retries:0},this.collectedRootSpans.includes(r)&&(this.rootSpans[s]={...this.spans[s],done:!1},this.waitSpans[s]=this.spans[s]),n!==void 0&&(this.spans[n]?this.spans[n].children.push(this.spans[s]):(this.spans[n]={spanId:n,kind:"future",children:[this.spans[s]],retries:0},this.waitSpans[n]=this.spans[n]))}cleanSpan(e){if(e===void 0)return;let n=e.spanId;for(let r of e.children)this.cleanSpan(r);delete this.spans[n],delete this.rootSpans[n],delete this.waitSpans[n],this.onClean(e)}async forceFlush(){await this.pendingSend;let e=Object.values(this.rootSpans).flatMap(n=>{if(n.done)return[];let{data:r}=this.treeShape.safeParse(n);return r?(n.done=!0,[{original:n,pruned:r}]):[]});e.length>0&&(this.pendingSend=this.onFlush(e).finally(()=>{e.forEach(({original:n})=>this.cleanSpan(n))}));for(let n of Object.values(this.waitSpans))++n.retries*this.sendInterval>this.sendTimeout&&this.cleanSpan(n)}async shutdown(){await this.forceFlush(),this.spans={},this.rootSpans={},this.waitSpans={}}onEnd(){}};function ue(t,e){let n=!1;return()=>{n===!1&&(n=!0,setTimeout(()=>(n=!1)||t(),e))}}var u={ended:m.z.literal(!0),_spanContext:m.z.custom(),spanContext:m.z.custom(),startTime:m.z.custom(),endTime:m.z.custom(),attributes:m.z.custom()};var de=i.z.object({value:i.z.object({name:i.z.literal("prisma:engine:db_query"),...u})}),y=i.z.object({value:i.z.object({name:i.z.literal("prisma:engine"),...u}),children:i.z.array(de).nonempty()}),O=i.z.object({value:i.z.object({name:i.z.literal("prisma:client:connect"),...u})}),he=i.z.object({value:i.z.object({name:i.z.literal("prisma:client:operation"),...u}),children:i.z.union([i.z.tuple([y]),i.z.tuple([O,y]),i.z.tuple([y,O])])}),b=class extends g{requests={};errors={};schemaHash;apiClient;schemaRequest;constructor(e){super({treeShape:he,sendTimeout:e.sendTimeout,sendInterval:e.sendInterval,collectedSpans:["prisma:client:operation","prisma:client:connect","prisma:engine","prisma:engine:db_query"],collectedRootSpans:["prisma:client:operation"]}),this.apiClient=e.apiClient,this.schemaRequest=e.schemaRequest}setRequest(e){this.requests[e.spanId]=e}setError(e){this.errors[e.spanId]=e}onClean({spanId:e}){delete this.requests[e],delete this.errors[e]}async onFlush(e){this.schemaHash??=E(this.apiClient,this.schemaRequest);let n=await this.schemaHash,r=[];for(let{pruned:s}of e){let{value:o}=s,a=o.spanContext().spanId;this.requests[a]&&r.push({hash:n,sql:fe(s),ts:l(o.startTime),connect:Se(s),error:this.errors[a]?.error??null,model:this.requests[a].model??null,operation:this.requests[a].operation,args:this.requests[a].args??{},latency:l(o.endTime)-l(o.startTime)})}await A(this.apiClient,r)}};function fe(t){return t.children.find(({value:s})=>s.name==="prisma:engine").children.filter(({value:s})=>s.name==="prisma:engine:db_query").sort((s,o)=>l(s.value.startTime)-l(o.value.startTime)).reduce((s,o)=>{let a=o.value.attributes["db.statement"];return typeof a=="string"?`${s}${a}
+`:s},"")}function Se(t){let{value:e}=t.children.find(({value:n})=>n.name==="prisma:client:connect")??{};return e?l(e.endTime)-l(e.startTime):!1}var d;function H(t){if(d)return d;let e=new _.AsyncHooksContextManager().enable();j.context.setGlobalContextManager(e);let n=new F.BasicTracerProvider({resource:new k.Resource({[R.SEMRESATTRS_SERVICE_NAME]:"extension-optimize",[R.SEMRESATTRS_SERVICE_VERSION]:"0.0.0"})});return d=new b(t),n.addSpanProcessor(d),(0,z.registerInstrumentations)({tracerProvider:n,instrumentations:[new M.PrismaInstrumentation]}),n.register(),d}function L(t){return t._previewFeatures?.includes("tracing")}function U(t){return t._engineConfig.inlineSchema}function $(t){if(t instanceof Error)return t.stack??t.message;switch(typeof t){case"undefined":return"undefined";case"object":{let e;return t!==null&&typeof t.toString=="function"&&(e=t.toString()),typeof e=="string"&&e!=="[object Object]"?e:JSON.stringify(t)}default:return String(t)}}var D=require("@prisma/client/extension"),N=D.Prisma.defineExtension(t=>t.$extends({}));var V="https://optimize-ingestion.datacdn.workers.dev/",B="https://optimize.prisma.io";function ge({enable:t=!0,ingestionUrl:e=V,dashboardUrl:n=B,sendInterval:r=500,sendTimeout:s=1e4,showNotification:o=!0,apiKey:a}){return t?(o&&P(n),Q.Prisma.defineExtension(p=>{let c=U(p),h=Reflect.get(p,"_activeProvider"),f=Reflect.get(p,"_accelerateEngineConfig"),{protocol:G}=new URL(f.accelerateUtils.resolveDatasourceUrl(f)),K={schema:c,provider:h,protocol:G},Z=w(e,a),C=H({schemaRequest:K,apiClient:Z,sendInterval:r,sendTimeout:s});if(!L(p))throw new Error('Please enable the "tracing" preview feature and regenerate the client.');return p.$extends({query:{async $allOperations({query:W,model:X,operation:Y,args:I}){let T=J.trace.getActiveSpan()?.spanContext().spanId;if(!T)throw new Error("prisma:client:operation span is expected to be entered in the client extension when tracing is enabled");C.setRequest({spanId:T,model:X,operation:Y,args:I});try{return await W(I)}catch(x){let ee=$(x);throw C.setError({spanId:T,error:ee}),x}}}})})):N}0&&(module.exports={PROD_DASHBOARD_URL,PROD_INGESTION_URL,withOptimize});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prisma/extension-optimize",
-  "version": "0.0.0-dev.202412140052",
+  "version": "0.0.0-dev.202412161942",
   "sideEffects": false,
   "description": "",
   "main": "./dist/index.js",
@@ -25,7 +25,7 @@
     "hono": "4.4.6",
     "tsup": "8.0.2",
     "vitest": "1.4.0",
-    "common": "0.0.0-dev.202412140052"
+    "common": "0.0.0-dev.202412161942"
   },
   "peerDependencies": {
     "@prisma/client": "5.x"