@prisma/dev 0.0.3 → 0.1.1

package/dist/index.d.ts

@@ -1,8 +1,19 @@
+declare const DEFAULT_ACCELERATE_PORT = 51213;
+declare const DEFAULT_DATABASE_PORT = 51214;
+declare const DEFAULT_SHADOW_DATABASE_PORT = 51215;
+type PortAssignableService = "accelerate" | "database" | "shadow_database";
+declare class PortNotAvailableError extends Error {
+    port: number;
+    service: PortAssignableService;
+    name: string;
+    constructor(port: number, service: PortAssignableService);
+}
+
 interface StartServerOptions {
     /**
      * The port the Accelerate server will listen on.
      *
-     * Defaults to `5433`.
+     * Defaults to `51213`.
      *
      * An error is thrown if the port is already in use.
      */
@@ -10,7 +21,7 @@ interface StartServerOptions {
     /**
      * The port the database server will listen on.
      *
-     * Defaults to `5432`.
+     * Defaults to `51214`.
      *
      * An error is thrown if the port is already in use.
      */
@@ -21,6 +32,14 @@ interface StartServerOptions {
      * Defaults to `false`.
      */
     debug?: boolean;
+    /**
+     * The port the shadow database server will listen on.
+     *
+     * Defaults to `51215`.
+     *
+     * An error is thrown if the port is already in use.
+     */
+    shadowDatabasePort?: number;
 }
 interface Server {
     accelerate: {
@@ -33,7 +52,10 @@ interface Server {
     ppg: {
         url: string;
     };
+    shadowDatabase: {
+        connectionString: string;
+    };
 }
 declare function unstable_startServer(options?: StartServerOptions): Promise<Server>;
 
-export { type Server, type StartServerOptions, unstable_startServer };
+export { DEFAULT_ACCELERATE_PORT, DEFAULT_DATABASE_PORT, DEFAULT_SHADOW_DATABASE_PORT, type PortAssignableService, PortNotAvailableError, type Server, type StartServerOptions, unstable_startServer };

package/dist/index.js

@@ -1,2 +1,7 @@
 import { createRequire } from 'module'; const require = createRequire(import.meta.url);
-import{createHash as g}from"crypto";import{createServer as S}from"http2";import{promisify as h}from"util";import{serve as w}from"@hono/node-server";import{Hono as p}from"hono/tiny";function d(){let e,a,o=new Promise((r,t)=>{e=r,a=t});return{fail:a,promise:o,succeed:e}}async function v(e,a,o){let r=new p;r.use(async(n,i)=>{n.set("db",a),n.set("debug",!!o),await i()}),r.route("/",f);let{fail:t,promise:s,succeed:l}=d(),c=w({createServer:S,fetch:r.fetch,port:e},()=>{l()});return c.on("error",n=>{if(typeof n=="object"&&"code"in n&&n.code==="EADDRINUSE")return t({portNotAvailable:e});console.error(n)}),await s,{async close(){await h(c.close.bind(c))()},port:e,url:`http://localhost:${e}`}}var u=new Map,f=new p;f.put("/:clientVersion/:schemaHash/schema",async e=>{let{req:a}=e,o=await a.text(),r=a.param("schemaHash");return o===u.get(r)?e.text(r):r!==y(o)?e.text("hashes do not match",500):(u.set(r,o),e.text(r))});function y(e){return g("sha256").update(e).digest("hex")}import{promisify as P}from"util";import{PGlite as D}from"@electric-sql/pglite";import{createServer as A,LogLevel as x}from"pglite-server";async function b(e,a){let o="default",r="postgres",t=new D({database:o,dataDir:"memory://",debug:a?5:void 0,username:r});await t.waitReady;let s=A(t,{keepAlive:!0,logLevel:a?x.Debug:void 0}),{fail:l,promise:c,succeed:n}=d();return s.on("error",i=>{if("code"in i&&i.code==="EADDRINUSE")return l(new Error(`Port ${e} is already in use. Provide a different \`dbport\` number and try again.`));console.error(i)}),s.listen(e,()=>n()),await c,{async close(){try{await P(s.close.bind(s))()}finally{await t.close()}},connectionString:`postgres://${r}@localhost:${e}/${o}`,database:o,port:e,username:r}}async function F(e){let{acceleratePort:a=5433,databasePort:o=5432,debug:r}=e||{},t=await b(o,r),s=await v(a,t);return{accelerate:{url:s.url},close:()=>l([s,t]),database:{connectionString:t.connectionString},ppg:{url:`prisma+postgres://localhost:${s.port}/?dbname=${t.database}&dbport=${t.port}&dbuser=${t.username}`}};async function l(c){let i=(await Promise.allSettled(c.map(m=>m.close()))).filter(m=>m.status==="rejected").map(m=>new Error(m.reason));if(i.length>0)throw new AggregateError(i,"Failed to close some servers")}}export{F as unstable_startServer};
+import{createServer as $e}from"http";import{promisify as ke}from"util";import{serve as Be}from"@hono/node-server";import Le from"@prisma/get-platform";import{logger as je}from"hono/logger";import{Hono as ee}from"hono/tiny";import{validator as y}from"hono/validator";import{HTTPException as P}from"hono/http-exception";import{object as ne,optional as re,parseJson as oe,pipe as x,regex as _,safeParse as se,string as O,url as $}from"valibot";var k=/^(postgres|postgresql):\/\//,ae=x(O(),oe(),ne({databaseUrl:x(O(),$(),_(k)),shadowDatabaseUrl:re(x(O(),$(),_(k)))}));function B(t){return Buffer.from(JSON.stringify(t),"utf8").toString("base64url")}function ie(t){let e=Buffer.from(t,"base64url").toString("utf8"),{issues:n,output:r,success:o}=se(ae,e,{abortEarly:!0});return o?[null,r]:[n]}var u=(t,e)=>{let{authorization:n}=t;if(!n)throw new P(401,{message:"Missing API Key"});let[r,o="",s]=n.split(" ");if(r!=="Bearer"||s)throw new P(401,{message:"Invalid API Key"});let[i,a]=ie(o);if(i)throw new P(401,{message:"Invalid API Key",cause:i.join(", ")});let{databaseUrl:c}=a;if(c!==e.var.db.connectionString)throw new P(401,{message:"Unauthorized"});return{decodedAPIKey:a}};import{array as ce,literal as le,minLength as de,object as ue,pipe as me,safeParse as pe,string as ge,union as fe}from"valibot";var he=ue({tags:fe([me(ce(ge()),de(1)),le("all")])});async function L(t){let{output:e,success:n}=pe(he,await t.req.json(),{abortEarly:!0});return n?e:t.text("Invalid input",400)}import{spawn as Ee}from"child_process";import{once as Ae}from"events";import{mkdir as Te}from"fs/promises";import{join as Re}from"path";import{chmod as ye,stat as we,writeFile as Se}from"fs/promises";import be from"env-paths";import{inflate as Pe}from"pako";var ve=be("prisma-dev").cache;function j(t,e){return`${ve}/${t}/${e}`}async function U(t){try{return await we(t),!0}catch(e){if(e!=null&&typeof e=="object"&&"code"in e&&e.code==="ENOENT")return!1;throw e}}async function q(t,e){let n=Pe(t);await Se(e,n),await ye(e,"755")}function v(){let t,e,n=new Promise((r,o)=>{t=r,e=o});return{fail:e,promise:n,succeed:t}}var f=class t{static#n=new Map;#e;#t;constructor(e){this.#e=e,this.#t=null}static async get(e){let n=`${e.schemaHash}:${e.clientVersion}`;try{let r=t.#n.get(n);if(r)return r;let o=new t(e);return t.#n.set(n,o),e.debug&&console.debug("starting engine...",e),await o.start(),e.debug&&console.debug("engine started!"),o}finally{t.stopAll(n)}}static async stopAll(e){let r=(await Promise.allSettled(Array.from(t.#n.entries()).filter(([o])=>o!==e).map(async([o,s])=>{try{await s.stop()}finally{t.#n.delete(o)}}))).filter(o=>o.status==="rejected").map(o=>o.reason);if(r.length>0)throw new AggregateError(r,"Failed to stop engines")}async commitTransaction(e,n){return await this.#o(e,n,"commit")}async request(e,n){let{url:r}=await this.#t,o=this.#r(n),s=await fetch(r,{body:typeof e=="string"?e:JSON.stringify(e),headers:{...o,"Content-Type":"application/json"},method:"POST"});if(!s.ok)throw await m.fromResponse(s);return await s.text()}async rollbackTransaction(e,n){return await this.#o(e,n,"rollback")}async startTransaction(e,n){let{url:r}=await this.#t,o=this.#r(n),s=await fetch(`${r}/transaction/start`,{body:JSON.stringify(e),headers:{...o,"Content-Type":"application/json"},method:"POST"});if(!s.ok)throw await m.fromResponse(s);return await s.json()}async start(){if(this.#t!=null)return;let e=await this.#s();this.#e.debug&&console.debug("spinning up engine at path...",e);let n=Ee(e,["--enable-raw-queries","--enable-telemetry-in-response","--port","0"],{env:{LOG_QUERIES:"y",PRISMA_DML:this.#e.base64Schema,QE_LOG_LEVEL:"TRACE",RUST_BACKTRACE:"1",RUST_LOG:"info"},stdio:["ignore","pipe","pipe"],windowsHide:!0});n.stderr.setEncoding("utf8"),n.stdout.setEncoding("utf8");let{fail:r,promise:o,succeed:s}=v();this.#t=o;let i=a=>{let c=a.split(`
+`).find(b=>b.includes("Started query engine http server"));if(!c)return;n.stdout.removeListener("data",i);let{fields:l}=JSON.parse(c);if(l==null)return r(new Error(`Unexpected data during initialization, "fields" are missing: ${a}`));let{ip:d,port:R}=l;if(d==null||R==null)return r(new Error(`This version of query-engine is not compatible with minippg, "ip" and "port" are missing in the startup log entry.
+Received data: ${a}`));s({childProcess:n,url:`http://${d}:${R}`})};n.once("error",a=>{this.#t=null,r(new h(String(a)))}),n.once("exit",(a,c)=>{this.#t=null,r(new h(`Query Engine exited with code ${a} and signal ${c}`))}),n.stdout.on("data",i),n.stderr.on("data",a=>{console.error(a)}),await this.#t}async stop(){if(this.#t==null)return;let{childProcess:e}=await this.#t;e.exitCode==null&&e.signalCode==null&&(e.kill(),await Ae(e,"exit"))}async#s(){this.#e.debug&&console.debug("getting engine commit hash...");let e=await this.#a();this.#e.debug&&console.debug("got engine commit hash",e);let n=j(this.#e.clientVersion,e);this.#e.debug&&console.debug("cache directory path",n),await Te(n,{recursive:!0});let{platform:r}=this.#e.platform,o=r==="windows"?".exe":"",s=Re(n,`query-engine-${r}${o}`);return this.#e.debug&&console.debug("engine binary path",s),(process.env.PRISMA_DEV_FORCE_ENGINE_DOWNLOAD==="1"||await U(s)===!1)&&await this.#i({commitHash:e,extension:o,engineBinaryPath:s}),s}async#a(){let e=await fetch(`https://registry.npmjs.org/@prisma/client/${this.#e.clientVersion}`);if(!e.ok)throw new Error(`Couldn't fetch package.json from npm registry, status code: ${e.status}`);let r=(await e.json()).devDependencies?.["@prisma/engines-version"];if(!r)throw new Error("Couldn't find engines version in package.json");let o=r.split(".").at(-1);if(!o)throw new Error("Couldn't find commit hash in engines version");return o}async#i(e){let{commitHash:n,extension:r,engineBinaryPath:o}=e,{binaryTarget:s}=this.#e.platform,i=`https://binaries.prisma.sh/all_commits/${n}/${s}/query-engine${r}.gz`;this.#e.debug&&console.debug("downloading engine from url",i);let a=await fetch(i);if(!a.ok)throw new Error(`Couldn't download engine. URL: ${i}, status code: ${a.status}`);await q(await a.arrayBuffer(),o),this.#e.debug&&console.debug("engine downloaded and saved at",o)}#r(e){let n={};for(let[r,o]of Object.entries(e))o!=null&&(n[r]=o);return n}async#o(e,n,r){let{url:o}=await this.#t,s=this.#r(n),i=await fetch(`${o}/transaction/${e}/${r}`,{headers:{...s,"Content-Type":"application/json"},method:"POST"});if(!i.ok)throw await m.fromResponse(i);try{return await i.json()}catch{return{}}}};function E(t,e){return console.error(t),t instanceof h?e.json({EngineNotStarted:{reason:{EngineStartupError:{logs:[],msg:t.message}}}},500):t instanceof m?e.text(t.responseBody,t.statusCode):e.body(null,500)}var h=class extends Error{name="EngineStartError"},m=class t extends Error{constructor(n,r,o){super(`${n}: Query Engine response status ${r}, body: ${o}`);this.action=n;this.statusCode=r;this.responseBody=o}name="EngineHttpError";static async fromResponse(n){let r=new URL(n.url),o=await n.text();return new t(r.pathname,n.status,o)}};var F=51213,V=51214,M=51215,p=class extends Error{constructor(n,r){super(`Port number \`${n}\` is not available for service ${r}.`);this.port=n;this.service=r}name="PortNotAvailableError"};import{Buffer as N}from"buffer";var A=new Map;async function H(t){let n=new TextEncoder().encode(t),r=await crypto.subtle.digest("SHA-256",n);return Array.from(new Uint8Array(r)).map(i=>i.toString(16).padStart(2,"0")).join("")}function K(t){let e=t.req.param("schemaHash"),n=A.get(e);return n==null?t.json({EngineNotStarted:{reason:"SchemaMissing"}},404):{schemaHash:e,schemas:n}}var xe=/datasource\s+db\s+\{\s*provider\s*=\s*"postgres(!?ql)?"\s+url\s*=\s*.+\s*\}/;async function z(t,e){let n=N.from(t,"base64").toString("utf8"),r=`datasource db {
+	provider = "postgresql"
+	url = "${e}"
+}`,o=n.replace(xe,r),s=await H(o);return{base64Override:N.from(o,"utf8").toString("base64"),overrideHash:s}}function T(t){let{req:e}=t;return{traceparent:e.header("traceparent"),"X-capture-telemetry":e.header("X-capture-telemetry")}}import{integer as G,looseObject as Oe,minValue as J,number as I,object as He,optional as Ie,pipe as Q,safeParse as X,string as W,union as Ce}from"valibot";var De=He({isolation_level:Ie(W()),max_wait:Q(I(),G(),J(0)),timeout:Q(I(),G(),J(0))});async function Y(t){let{issues:e,output:n,success:r}=X(De,await t.req.json(),{abortEarly:!0});return r?n:t.json({EngineNotStarted:{reason:"InvalidRequest",issues:e}},400)}var _e=Oe({id:Ce([W(),I()])});function Z(t,e){let{output:n,success:r}=X(_e,t);return r?n:e.json({EngineMalfunction:{}},500)}async function te(t,e,n){let r=await Ue(t,e,n),{fail:o,promise:s,succeed:i}=v(),a=Be({createServer:$e,fetch:r.fetch,port:t},()=>{i()});return a.on("error",c=>{if(typeof c=="object"&&"code"in c&&c.code==="EADDRINUSE")return o(new p(t,"accelerate"));console.error(c)}),await s,{async close(){await Promise.allSettled([ke(a.close.bind(a))(),f.stopAll()])},port:t,url:`http://localhost:${t}`}}async function Ue(t,e,n){let r=new ee,o=await Le.getPlatformInfo();return n&&console.debug("platform info: %s",JSON.stringify(o)),n&&r.use("*",je()),r.use("*",async(s,i)=>(s.set("db",e),s.set("debug",!!n),s.set("platform",o),s.set("port",t),s.set("protocol","http"),await i())),r.route("/",w),r}var w=new ee;w.get("/health",t=>t.text("ok"));w.post("/invalidate",y("header",u),async t=>{let e=await L(t);return e instanceof Response?e:t.body(null)});var qe="/:clientVersion/:schemaHash",S=w.basePath(qe);w.route("/",S);var Fe=["/graphql","/itx/:transactionId/graphql"];S.on("POST",[...Fe],y("header",u),async t=>{let{req:e}=t;try{let n=await C(t);if(n instanceof Response)return n;let r=await e.text(),o=e.param("transactionId"),s=await n.request(r,{...T(t),"X-transaction-id":o});return t.text(s)}catch(n){return E(n,t)}});S.basePath("/itx/:transactionId").on("POST",["/commit","/rollback"],y("header",u),async t=>{let{req:e}=t;try{let n=await C(t);if(n instanceof Response)return n;let o=`${e.routePath.split("/").filter(Boolean).at(-1)}Transaction`,s=e.param("transactionId"),i=await n[o](s,T(t));return t.json(i)}catch(n){return E(n,t)}});S.put("/schema",y("header",u),async t=>{let{req:e}=t,n=await e.text();if(!n)return t.text("Missing schema",400);let r=e.param("schemaHash"),o=A.get(r);if(o==null){if(r!==await H(n))return t.text("Schema hash mismatch",400);let s=await z(n,t.get("db").connectionString);return A.set(r,{base64Original:n,...s}),t.text(r)}return n!==o.base64Original?t.text("Schema mismatch",400):t.text(r)});S.post("/transaction/start",y("header",u),async t=>{let{req:e}=t,n=await Y(t);if(n instanceof Response)return n;try{let r=await C(t);if(r instanceof Response)return r;let o=await r.startTransaction(n,T(t)),s=Z(o,t);if(s instanceof Response)return s;let{id:i}=s,a=e.param("clientVersion"),c=t.get("port"),l=t.get("protocol"),d=e.param("schemaHash");return t.json({...o,"data-proxy":{endpoint:`${l}://localhost:${c}/${a}/${d}/itx/${i}`}})}catch(r){return E(r,t)}});async function C(t){let{req:e}=t,n=K(t);if(n instanceof Response)return n;let{base64Override:r,overrideHash:o}=n.schemas;return await f.get({base64Schema:r,clientVersion:process.env.PRISMA_DEV_FORCE_CLIENT_VERSION||e.param("clientVersion"),debug:t.get("debug"),platform:t.get("platform"),schemaHash:o})}import{PGlite as Ve}from"@electric-sql/pglite";import{PGLiteSocketServer as Me}from"@electric-sql/pglite-socket";async function D(t,e,n){let r={connectionLimit:1,connectTimeout:0,database:"postgres",maxIdleConnectionLifetime:0,password:"postgres",poolTimeout:0,port:t,socketTimeout:0,sslMode:"disable",username:"postgres"},o=await Ve.create({database:r.database,dataDir:"memory://",debug:n?5:void 0,defaultDataTransferContainer:"file",username:r.username});await o.waitReady;let s=new Me({db:o,inspect:n,port:t});try{await s.start()}catch(a){throw a instanceof Error&&"code"in a&&a.code==="EADDRINUSE"?new p(t,e):a}n&&console.debug(`${e} server started on port %i`,t);let i=`postgres://${r.username}:${r.password}@localhost:${t}/${r.database}?${new URLSearchParams({connection_limit:String(r.connectionLimit),connect_timeout:String(r.connectTimeout),max_idle_connection_lifetime:String(r.maxIdleConnectionLifetime),pool_timeout:String(r.poolTimeout),socket_timeout:String(r.socketTimeout),sslmode:r.sslMode}).toString()}`;return{...r,async close(){try{await s.stop()}finally{await o.close()}},connectionString:i}}async function jt(t){let{acceleratePort:e=51213,databasePort:n=51214,debug:r=!1,shadowDatabasePort:o=51215}=t||{},[s,i]=await Promise.all([D(n,"database",r),D(o,"shadow_database",r)]),a=await te(e,s,r),c=`prisma+postgres://localhost:${a.port}/?${new URLSearchParams({api_key:B({databaseUrl:s.connectionString,shadowDatabaseUrl:i.connectionString})}).toString()}`;return{accelerate:{url:a.url},close:()=>l([a,s,i]),database:{connectionString:s.connectionString},ppg:{url:c},shadowDatabase:{connectionString:i.connectionString}};async function l(d){let b=(await Promise.allSettled(d.map(g=>g.close()))).filter(g=>g.status==="rejected").map(g=>new Error(g.reason));if(b.length>0)throw new AggregateError(b,"Failed to close some servers")}}export{F as DEFAULT_ACCELERATE_PORT,V as DEFAULT_DATABASE_PORT,M as DEFAULT_SHADOW_DATABASE_PORT,p as PortNotAvailableError,jt as unstable_startServer};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prisma/dev",
-  "version": "0.0.3",
+  "version": "0.1.1",
   "description": "A local Prisma Postgres server for development and testing",
   "type": "module",
   "main": "./dist/index.js",
@@ -29,15 +29,21 @@
   "devDependencies": {
     "@arethetypeswrong/cli": "0.17.4",
     "@types/node": "22.3.0",
+    "@types/pako": "2.0.3",
+    "pkg-types": "2.1.0",
     "tsup": "8.0.2",
     "typescript": "5.8.3",
     "vitest": "3.1.3"
   },
   "dependencies": {
-    "@electric-sql/pglite": "0.2.17",
+    "@electric-sql/pglite": "0.3.1",
+    "@electric-sql/pglite-socket": "0.0.4",
     "@hono/node-server": "1.14.1",
+    "@prisma/get-platform": "6.7.0",
+    "env-paths": "3.0.0",
     "hono": "4.7.8",
-    "pglite-server": "0.1.4"
+    "pako": "2.1.0",
+    "valibot": "1.1.0"
   },
   "scripts": {
     "build": "tsup",