@prisma/extension-optimize 0.0.0-dev.202408261306 → 0.0.0-dev.202408261452

package/dist/index.d.ts

@@ -28,10 +28,10 @@ type OptimizeOptions = {
      */
     minSendInterval?: number;
     /**
-     * Whether to show a toast notification when the dashboard is ready.
+     * Whether to show a CLI notification when the dashboard is ready.
      */
-    showToast?: boolean;
+    showNotification?: boolean;
 };
-declare function withOptimize({ enable, ingestionUrl, dashboardUrl, minSendInterval, showToast, token, }: OptimizeOptions): (client: any) => _prisma_client_extension.PrismaClientExtends<_prisma_client_runtime_library.InternalArgs<{}, {}, {}, {}> & _prisma_client_runtime_library.DefaultArgs>;
+declare function withOptimize({ enable, ingestionUrl, dashboardUrl, minSendInterval, showNotification, token, }: OptimizeOptions): (client: any) => _prisma_client_extension.PrismaClientExtends<_prisma_client_runtime_library.InternalArgs<{}, {}, {}, {}> & _prisma_client_runtime_library.DefaultArgs>;
 
 export { type OptimizeOptions, PROD_DASHBOARD_URL, PROD_INGESTION_URL, withOptimize };

package/dist/index.js

@@ -1,2 +1,2 @@
-"use strict";var se=Object.create;var h=Object.defineProperty;var re=Object.getOwnPropertyDescriptor;var oe=Object.getOwnPropertyNames;var ae=Object.getPrototypeOf,pe=Object.prototype.hasOwnProperty;var de=(n,e)=>{for(var t in e)h(n,t,{get:e[t],enumerable:!0})},x=(n,e,t,i)=>{if(e&&typeof e=="object"||typeof e=="function")for(let s of oe(e))!pe.call(n,s)&&s!==t&&h(n,s,{get:()=>e[s],enumerable:!(i=re(e,s))||i.enumerable});return n};var m=(n,e,t)=>(t=n!=null?se(ae(n)):{},x(e||!n||!n.__esModule?h(t,"default",{value:n,enumerable:!0}):t,n)),ce=n=>x(h({},"__esModule",{value:!0}),n);var Se={};de(Se,{PROD_DASHBOARD_URL:()=>ee,PROD_INGESTION_URL:()=>K,withOptimize:()=>ge});module.exports=ce(Se);var X=require("@opentelemetry/api"),Y=require("@prisma/client/extension"),Z=m(require("@prisma/debug"));function me(n,e){let t;return async function(i,s,o){let a=new URL(i,n),p=new Headers({"Content-Type":"application/json",Authorization:`Bearer ${e}`});t&&p.set("prisma-optimize-jwt",t);let r=await fetch(a,{method:s,headers:p,body:JSON.stringify(o)});if(!r.ok){let d=await r.text().catch(()=>"<unreadable>");console.error(`[optimize] HTTP ${r.status} ${r.statusText}: ${d}`)}let c=r.headers.get("prisma-optimize-jwt");return c&&(t=c),r}}function P(n,e){let t=me(n,e);return{request:t,post:(i,s)=>t(i,"POST",s)}}async function v(n,e){return await(await n.post("/schema",{schema:e})).text()}var l=m(require("fs")),q=m(require("path")),T=require("stream"),E=require("stream/promises"),A=m(require("kleur")),_=require("node-notifier"),O=m(require("xdg-app-paths")),le="https://avatars.githubusercontent.com/u/17219288?s=96",{bold:ue,underline:he}=A.default;async function k(n){let e=new O.default("prisma-optimize").config();l.default.mkdirSync(e,{recursive:!0});let t=q.default.resolve(e,"avatar.png");if(!l.default.existsSync(t)){let r=await fetch(le);if(r.ok&&r.body!=null){let c=l.default.createWriteStream(t,{flags:"wx"});await(0,E.finished)(T.Readable.fromWeb(r.body).pipe(c))}}let i="See your Optimize dashboard at:",s=i.length,o=`${he(ue(n))}`,a=n.length,p=Math.max(s,a)+2;console.log("\u250C"+"\u2500".repeat(p)+"\u2510"),console.log("\u2502 "+i+" ".repeat(p-s-2)+" \u2502"),console.log("\u2502 "+o+" ".repeat(p-a-2)+" \u2502"),console.log("\u2514"+"\u2500".repeat(p)+"\u2518"),(0,_.notify)({timeout:10,open:n,subtitle:"Your dashboard is ready! \u{1F680} ",message:"Click to open",title:"Prisma Optimize",icon:l.default.existsSync(t)?t:void 0,contentImage:l.default.existsSync(t)?t:void 0})}var L=require("@opentelemetry/api"),N=require("@opentelemetry/context-async-hooks"),U=require("@opentelemetry/instrumentation"),$=require("@opentelemetry/resources"),H=require("@opentelemetry/sdk-trace-base"),y=require("@opentelemetry/semantic-conventions"),D=require("@prisma/instrumentation");var F=m(require("@prisma/debug"));async function M(n,e){await n.post("/ingest",e)}function I(n,...e){setTimeout(()=>{n(...e)},0)}function u(n){return n[0]*1e3+n[1]/1e6}var g=(0,F.default)("prisma:extension:optimize"),f="0000000000000000",j=(o=>(o.ClientOperation="prisma:client:operation",o.ClientConnect="prisma:client:connect",o.Engine="prisma:engine",o.EngineConnection="prisma:engine:connection",o.EngineQuery="prisma:engine:db_query",o))(j||{}),w=class{spanId;timestamp;model;operation;args;statementSpans;duration;connect;error;completionFlags={clientSpanClosed:!1,engineSpanClosed:!1};constructor(e){this.spanId=e.spanId,this.timestamp=e.timestamp,this.model=e.model,this.operation=e.operation,this.args=e.args,this.statementSpans=[]}isCompleted(){return this.completionFlags.clientSpanClosed&&this.completionFlags.engineSpanClosed}statementSpansToString(){return this.statementSpans.sort((t,i)=>{let s=u(t.startTime),o=u(i.startTime);return s-o}).reduce((t,i)=>{let s=i.attributes["db.statement"];return typeof s=="string"?`${t}${s}
-`:(g("sql attribute is not a string: %o",s),t)},"")}toIngestRequestItem(e){if(this.duration===void 0)throw new Error("`duration` is `undefined`, this should not happen");return{ts:this.timestamp,model:this.model??null,operation:this.operation,args:this.args,latency:this.duration,connect:this.connect??!1,sql:this.statementSpansToString(),error:this.error??null,hash:e}}},S=class{#t=new Map;#n=new Map;#i=new Map;#o=new Set;#e=[];#a=Promise.resolve();#p=!1;#c;#m;#s=new Set;#l=Promise.resolve();#d=!1;#u;setSchemaHash=()=>{};__test_internalMetrics__={rescheduledCountTotal:0};constructor(e,t){this.#c=e,this.#m=t,this.#u=new Promise(i=>{this.setSchemaHash=i})}createRequest(e){let t=new w(e);this.#t.set(e.spanId,t)}setRequestError(e,t){let i=this.#t.get(e);if(!i)throw new Error(`Unknown request ${e}`);i.error=t}#S(e){for(let t=e;t!==f;t=this.#n.get(t)){if(t===void 0)return{type:"UndeliveredSpanInTree"};let i=this.#t.get(t);if(i)return{type:"Ok",request:i}}return{type:"NotInRequestTree"}}onStart(e,t){if(!z(e))return;let i=e.spanContext().spanId;if(!e.parentSpanId){this.#n.set(i,f);return}if(this.#n.set(i,e.parentSpanId),e.parentSpanId===f)return;let s=this.#i.get(e.parentSpanId);s?s.add(i):this.#i.set(e.parentSpanId,new Set([i])),e.name==="prisma:engine:connection"&&this.#o.add(i)}onEnd(e){z(e)?this.#h(e):(e.parentSpanId===void 0||e.parentSpanId===f)&&this.#r(e.spanContext().spanId)}async forceFlush(){await this.#a,await this.#g()}async shutdown(){await this.forceFlush()}#h(e){let t=e.spanContext().spanId,i,s=this.#S(t);switch(s.type){case"Ok":i=s.request;break;case"UndeliveredSpanInTree":this.#y(e);break;case"NotInRequestTree":this.#r(t);break}i&&(e.name==="prisma:client:connect"&&(i.connect=u(e.duration)),e.name==="prisma:engine"&&(i.completionFlags.engineSpanClosed=!0),e.name==="prisma:engine:db_query"&&!this.#o.has(e.parentSpanId)&&i.statementSpans.push(e),e.name==="prisma:client:operation"&&(i.duration=u(e.duration),i.completionFlags.clientSpanClosed=!0,g("latency otel: %d",i.duration)),i.isCompleted()&&this.#w(i))}#y(e){this.#e.push(e),this.__test_internalMetrics__.rescheduledCountTotal++;let{spanId:t,traceId:i}=e.spanContext();g("rescheduled ",e.name,t,e.parentSpanId,i),!this.#p&&(this.#p=!0,this.#a=Promise.all([this.#a,new Promise(I).then(()=>this.#C())]))}#C(){let e=this.#e;this.#e=[],this.#p=!1;for(let t of e)this.#h(t)}#r(e){this.#n.delete(e),this.#o.delete(e);let t=this.#i.get(e);if(this.#i.delete(e),this.#I(e),t)for(let i of t)this.#r(i)}#I(e){for(let t=this.#f(e);t!==-1;t=this.#f(e))this.#e.splice(t,1)}#f(e){return this.#e.findIndex(t=>t.spanContext().spanId===e)}#w(e){I(()=>{this.#t.delete(e.spanId),this.#r(e.spanId),this.#s.add(e),this.#R()})}async#R(){await this.#l,!this.#d&&(this.#d=!0,setTimeout(()=>{this.#l=this.#g().finally(()=>{this.#d=!1})},this.#m))}async#g(){if(this.#s.size===0)return;let e=await this.#u,t=[...this.#s].map(i=>i.toIngestRequestItem(e));this.#s.clear(),g("sending batch of %d requests",t.length);try{await M(this.#c,t)}catch(i){console.error(i.message??i)}}};function z(n){return Object.values(j).includes(n.name)}function Q(n,e){let t=new N.AsyncHooksContextManager().enable();L.context.setGlobalContextManager(t);let i=new H.BasicTracerProvider({resource:new $.Resource({[y.SEMRESATTRS_SERVICE_NAME]:"extension-optimize",[y.SEMRESATTRS_SERVICE_VERSION]:"0.0.0"})}),s=new S(n,e);return i.addSpanProcessor(s),(0,U.registerInstrumentations)({tracerProvider:i,instrumentations:[new D.PrismaInstrumentation]}),i.register(),s}function B(n){return n._previewFeatures?.includes("tracing")}function J(n){return n._engineConfig.inlineSchema}function V(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 G=require("@prisma/client/extension"),W=G.Prisma.defineExtension(n=>n.$extends({}));var K="https://optimize-ingestion.datacdn.workers.dev/",ee="https://optimize.prisma.io",fe=(0,Z.default)("prisma:extension:optimize");function ge({enable:n=!0,ingestionUrl:e=K,dashboardUrl:t=ee,minSendInterval:i=50,showToast:s=!0,token:o}){return n?(s&&k(t).then().catch(a=>{console.error("Failed to show toast",a)}),Y.Prisma.defineExtension(a=>{let p=P(e,o),r=Q(p,i);if(!B(a))throw new Error('Please enable the "tracing" preview feature and regenerate the client.');let c=J(a);return v(p,c).then(d=>{fe("uploaded schema with hash",d),r.setSchemaHash(d)},d=>{console.error(d),console.error("Failed to upload the schema, no data will be sent to Prisma Optimize")}),a.$extends({query:{async $allOperations({query:d,model:te,operation:ne,args:R}){let C=X.trace.getActiveSpan()?.spanContext().spanId;if(!C)throw new Error("prisma:client:operation span is expected to be entered in the client extension when tracing is enabled");r.createRequest({spanId:C,timestamp:Date.now(),model:te,operation:ne,args:R});try{return await d(R)}catch(b){let ie=V(b);throw r.setRequestError(C,ie),b}}}})})):W}0&&(module.exports={PROD_DASHBOARD_URL,PROD_INGESTION_URL,withOptimize});
+"use strict";var X=Object.create;var l=Object.defineProperty;var Y=Object.getOwnPropertyDescriptor;var ee=Object.getOwnPropertyNames;var te=Object.getPrototypeOf,ne=Object.prototype.hasOwnProperty;var se=(n,e)=>{for(var t in e)l(n,t,{get:e[t],enumerable:!0})},x=(n,e,t,s)=>{if(e&&typeof e=="object"||typeof e=="function")for(let i of ee(e))!ne.call(n,i)&&i!==t&&l(n,i,{get:()=>e[i],enumerable:!(s=Y(e,i))||s.enumerable});return n};var C=(n,e,t)=>(t=n!=null?X(te(n)):{},x(e||!n||!n.__esModule?l(t,"default",{value:n,enumerable:!0}):t,n)),ie=n=>x(l({},"__esModule",{value:!0}),n);var ce={};se(ce,{PROD_DASHBOARD_URL:()=>G,PROD_INGESTION_URL:()=>V,withOptimize:()=>de});module.exports=ie(ce);var Q=require("@opentelemetry/api"),B=require("@prisma/client/extension"),J=C(require("@prisma/debug"));function re(n,e){let t;return async function(s,i,r){let p=new URL(s,n),d=new Headers({"Content-Type":"application/json",Authorization:`Bearer ${e}`});t&&d.set("prisma-optimize-jwt",t);let o=await fetch(p,{method:i,headers:d,body:JSON.stringify(r)});if(!o.ok){let a=await o.text().catch(()=>"<unreadable>");console.error(`[optimize] HTTP ${o.status} ${o.statusText}: ${a}`)}let m=o.headers.get("prisma-optimize-jwt");return m&&(t=m),o}}function b(n,e){let t=re(n,e);return{request:t,post:(s,i)=>t(s,"POST",i)}}async function P(n,e){return await(await n.post("/schema",{schema:e})).text()}var q=C(require("kleur"));var{bold:oe,underline:ae}=q.default;function v(n){let e="See your Optimize dashboard at:",t=e.length,s=`${ae(oe(n))}`,i=n.length,r=Math.max(t,i)+2;console.log("\u250C"+"\u2500".repeat(r)+"\u2510"),console.log("\u2502 "+e+" ".repeat(r-t-2)+" \u2502"),console.log("\u2502 "+s+" ".repeat(r-i-2)+" \u2502"),console.log("\u2514"+"\u2500".repeat(r)+"\u2518")}var O=require("@opentelemetry/api"),k=require("@opentelemetry/context-async-hooks"),M=require("@opentelemetry/instrumentation"),F=require("@opentelemetry/resources"),z=require("@opentelemetry/sdk-trace-base"),g=require("@opentelemetry/semantic-conventions"),N=require("@prisma/instrumentation");var A=C(require("@prisma/debug"));async function E(n,e){await n.post("/ingest",e)}function I(n,...e){setTimeout(()=>{n(...e)},0)}function c(n){return n[0]*1e3+n[1]/1e6}var h=(0,A.default)("prisma:extension:optimize"),u="0000000000000000",_=(r=>(r.ClientOperation="prisma:client:operation",r.ClientConnect="prisma:client:connect",r.Engine="prisma:engine",r.EngineConnection="prisma:engine:connection",r.EngineQuery="prisma:engine:db_query",r))(_||{}),y=class{spanId;timestamp;model;operation;args;statementSpans;duration;connect;error;completionFlags={clientSpanClosed:!1,engineSpanClosed:!1};constructor(e){this.spanId=e.spanId,this.timestamp=e.timestamp,this.model=e.model,this.operation=e.operation,this.args=e.args,this.statementSpans=[]}isCompleted(){return this.completionFlags.clientSpanClosed&&this.completionFlags.engineSpanClosed}statementSpansToString(){return this.statementSpans.sort((t,s)=>{let i=c(t.startTime),r=c(s.startTime);return i-r}).reduce((t,s)=>{let i=s.attributes["db.statement"];return typeof i=="string"?`${t}${i}
+`:(h("sql attribute is not a string: %o",i),t)},"")}toIngestRequestItem(e){if(this.duration===void 0)throw new Error("`duration` is `undefined`, this should not happen");return{ts:this.timestamp,model:this.model??null,operation:this.operation,args:this.args,latency:this.duration,connect:this.connect??!1,sql:this.statementSpansToString(),error:this.error??null,hash:e}}},f=class{#t=new Map;#n=new Map;#s=new Map;#o=new Set;#e=[];#a=Promise.resolve();#p=!1;#c;#m;#i=new Set;#l=Promise.resolve();#d=!1;#u;setSchemaHash=()=>{};__test_internalMetrics__={rescheduledCountTotal:0};constructor(e,t){this.#c=e,this.#m=t,this.#u=new Promise(s=>{this.setSchemaHash=s})}createRequest(e){let t=new y(e);this.#t.set(e.spanId,t)}setRequestError(e,t){let s=this.#t.get(e);if(!s)throw new Error(`Unknown request ${e}`);s.error=t}#S(e){for(let t=e;t!==u;t=this.#n.get(t)){if(t===void 0)return{type:"UndeliveredSpanInTree"};let s=this.#t.get(t);if(s)return{type:"Ok",request:s}}return{type:"NotInRequestTree"}}onStart(e,t){if(!T(e))return;let s=e.spanContext().spanId;if(!e.parentSpanId){this.#n.set(s,u);return}if(this.#n.set(s,e.parentSpanId),e.parentSpanId===u)return;let i=this.#s.get(e.parentSpanId);i?i.add(s):this.#s.set(e.parentSpanId,new Set([s])),e.name==="prisma:engine:connection"&&this.#o.add(s)}onEnd(e){T(e)?this.#h(e):(e.parentSpanId===void 0||e.parentSpanId===u)&&this.#r(e.spanContext().spanId)}async forceFlush(){await this.#a,await this.#g()}async shutdown(){await this.forceFlush()}#h(e){let t=e.spanContext().spanId,s,i=this.#S(t);switch(i.type){case"Ok":s=i.request;break;case"UndeliveredSpanInTree":this.#C(e);break;case"NotInRequestTree":this.#r(t);break}s&&(e.name==="prisma:client:connect"&&(s.connect=c(e.duration)),e.name==="prisma:engine"&&(s.completionFlags.engineSpanClosed=!0),e.name==="prisma:engine:db_query"&&!this.#o.has(e.parentSpanId)&&s.statementSpans.push(e),e.name==="prisma:client:operation"&&(s.duration=c(e.duration),s.completionFlags.clientSpanClosed=!0,h("latency otel: %d",s.duration)),s.isCompleted()&&this.#w(s))}#C(e){this.#e.push(e),this.__test_internalMetrics__.rescheduledCountTotal++;let{spanId:t,traceId:s}=e.spanContext();h("rescheduled ",e.name,t,e.parentSpanId,s),!this.#p&&(this.#p=!0,this.#a=Promise.all([this.#a,new Promise(I).then(()=>this.#I())]))}#I(){let e=this.#e;this.#e=[],this.#p=!1;for(let t of e)this.#h(t)}#r(e){this.#n.delete(e),this.#o.delete(e);let t=this.#s.get(e);if(this.#s.delete(e),this.#y(e),t)for(let s of t)this.#r(s)}#y(e){for(let t=this.#f(e);t!==-1;t=this.#f(e))this.#e.splice(t,1)}#f(e){return this.#e.findIndex(t=>t.spanContext().spanId===e)}#w(e){I(()=>{this.#t.delete(e.spanId),this.#r(e.spanId),this.#i.add(e),this.#R()})}async#R(){await this.#l,!this.#d&&(this.#d=!0,setTimeout(()=>{this.#l=this.#g().finally(()=>{this.#d=!1})},this.#m))}async#g(){if(this.#i.size===0)return;let e=await this.#u,t=[...this.#i].map(s=>s.toIngestRequestItem(e));this.#i.clear(),h("sending batch of %d requests",t.length);try{await E(this.#c,t)}catch(s){console.error(s.message??s)}}};function T(n){return Object.values(_).includes(n.name)}function j(n,e){let t=new k.AsyncHooksContextManager().enable();O.context.setGlobalContextManager(t);let s=new z.BasicTracerProvider({resource:new F.Resource({[g.SEMRESATTRS_SERVICE_NAME]:"extension-optimize",[g.SEMRESATTRS_SERVICE_VERSION]:"0.0.0"})}),i=new f(n,e);return s.addSpanProcessor(i),(0,M.registerInstrumentations)({tracerProvider:s,instrumentations:[new N.PrismaInstrumentation]}),s.register(),i}function L(n){return n._previewFeatures?.includes("tracing")}function U(n){return n._engineConfig.inlineSchema}function $(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 H=require("@prisma/client/extension"),D=H.Prisma.defineExtension(n=>n.$extends({}));var V="https://optimize-ingestion.datacdn.workers.dev/",G="https://optimize.prisma.io",pe=(0,J.default)("prisma:extension:optimize");function de({enable:n=!0,ingestionUrl:e=V,dashboardUrl:t=G,minSendInterval:s=50,showNotification:i=!0,token:r}){return n?(i&&v(t),B.Prisma.defineExtension(p=>{let d=b(e,r),o=j(d,s);if(!L(p))throw new Error('Please enable the "tracing" preview feature and regenerate the client.');let m=U(p);return P(d,m).then(a=>{pe("uploaded schema with hash",a),o.setSchemaHash(a)},a=>{console.error(a),console.error("Failed to upload the schema, no data will be sent to Prisma Optimize")}),p.$extends({query:{async $allOperations({query:a,model:Z,operation:K,args:w}){let S=Q.trace.getActiveSpan()?.spanContext().spanId;if(!S)throw new Error("prisma:client:operation span is expected to be entered in the client extension when tracing is enabled");o.createRequest({spanId:S,timestamp:Date.now(),model:Z,operation:K,args:w});try{return await a(w)}catch(R){let W=$(R);throw o.setRequestError(S,W),R}}}})})):D}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.202408261306",
+  "version": "0.0.0-dev.202408261452",
   "sideEffects": false,
   "description": "",
   "main": "./dist/index.js",
@@ -17,9 +17,7 @@
     "@prisma/debug": "5.12.1",
     "@prisma/instrumentation": "5.18.0",
     "kleur": "4.1.5",
-    "node-notifier": "10.0.1",
-    "readline-sync": "1.4.10",
-    "xdg-app-paths": "8.3.0"
+    "readline-sync": "1.4.10"
   },
   "devDependencies": {
     "@hono/node-server": "1.9.1",
@@ -28,7 +26,7 @@
     "hono": "4.4.6",
     "tsup": "8.0.2",
     "vitest": "1.4.0",
-    "common": "0.0.0-dev.202408261306"
+    "common": "0.0.0-dev.202408261452"
   },
   "peerDependencies": {
     "@prisma/client": "5.x"