@prisma/cli-init 0.0.0-dev.202512170049 → 0.0.0-dev.202512221324

package/dist/index.js

@@ -2461,8 +2461,16 @@ async function startDBServer(purpose, serverState) {
       console.debug(`[${purpose}][${channel}] ${payload}`);
     });
   }
-  const { PGLiteSocketServer } = await import("@electric-sql/pglite-socket");
-  const server = new PGLiteSocketServer({ db, debug: debug3, inspect: debug3, port });
+  const { PGLiteSocketHandler, PGLiteSocketServer } = await import("@electric-sql/pglite-socket");
+  const idleTimeout = purpose === "shadow_database" ? serverState.shadowDatabaseIdleTimeoutMillis : serverState.databaseIdleTimeoutMillis;
+  applyIdleTimeout(idleTimeout, db, PGLiteSocketHandler);
+  const server = new PGLiteSocketServer({
+    connectionQueueTimeout: purpose === "shadow_database" ? serverState.shadowDatabaseConnectTimeoutMillis : serverState.databaseConnectTimeoutMillis,
+    db,
+    debug: debug3,
+    inspect: debug3,
+    port
+  });
   if (debug3) {
     server.addEventListener("listening", (event) => {
       const { detail } = event;
@@ -2558,7 +2566,7 @@ async function getDB(dataDir, debug3) {
 }
 async function getShadowDB(_dataDir, debug3) {
   const { PGlite } = await import("@electric-sql/pglite");
-  return await PGlite.create({
+  const db = await PGlite.create({
     database: BASE_PGLITE_OPTIONS.database,
     dataDir: "memory://",
     debug: debug3 ? 5 : void 0,
@@ -2566,6 +2574,7 @@ async function getShadowDB(_dataDir, debug3) {
     relaxedDurability: false,
     username: BASE_PGLITE_OPTIONS.username
   });
+  return db;
 }
 async function getExtensions() {
   const importedExtensions = await Promise.all([
@@ -2619,6 +2628,46 @@ async function getExtensions() {
   ]);
   return Object.assign({}, ...importedExtensions);
 }
+function applyIdleTimeout(idleTimeout, db, PGLiteSocketHandlerConstructor) {
+  db.__IDLE_TIMEOUT__ = idleTimeout;
+  if (PGLiteSocketHandlerConstructor.prototype.__ATTACH_OVERRIDDEN__) {
+    return;
+  }
+  const originalAttach = PGLiteSocketHandlerConstructor.prototype.attach;
+  PGLiteSocketHandlerConstructor.prototype.__ATTACH_OVERRIDDEN__ = true;
+  PGLiteSocketHandlerConstructor.prototype.attach = async function(socket) {
+    const result = await originalAttach.call(this, socket);
+    if (!Number.isFinite(this.db.__IDLE_TIMEOUT__)) {
+      return result;
+    }
+    let idleTimeoutId = null;
+    const clearIdleTimeout = () => {
+      if (idleTimeoutId != null) {
+        clearTimeout(idleTimeoutId);
+        idleTimeoutId = null;
+      }
+    };
+    const resetIdleTimeout = () => {
+      clearIdleTimeout();
+      idleTimeoutId = setTimeout(
+        () => this.detach(true),
+        // @ts-expect-error shhhh!
+        this.db.__IDLE_TIMEOUT__
+      );
+    };
+    socket.on("error", () => {
+      clearIdleTimeout();
+    });
+    socket.on("data", () => {
+      resetIdleTimeout();
+    });
+    socket.on("close", () => {
+      clearIdleTimeout();
+    });
+    resetIdleTimeout();
+    return result;
+  };
+}
 async function dumpDB(options) {
   const { dataDir, db, debug: debug3, destinationPath } = options;
   const pg = db || await getDB(dataDir, debug3);
@@ -2767,20 +2816,28 @@ var PRIVATE_INITIALIZE_SYMBOL = Symbol("initialize");
 var DEFAULT_NAME = "default";
 var ServerState = class {
   _databasePort;
+  databaseConnectTimeoutMillis;
+  databaseIdleTimeoutMillis;
   debug;
   dryRun;
   name;
   persistenceMode;
   pid;
+  shadowDatabaseConnectTimeoutMillis;
+  shadowDatabaseIdleTimeoutMillis;
   _port;
   _shadowDatabasePort;
   constructor(options) {
     this._databasePort = options.databasePort ?? NO_PORT;
+    this.databaseConnectTimeoutMillis = options.databaseConnectTimeoutMillis ?? 6e4;
+    this.databaseIdleTimeoutMillis = options.databaseIdleTimeoutMillis ?? Infinity;
     this.debug = options.debug ?? false;
     this.dryRun = options.dryRun ?? false;
     this.name = options.name ?? DEFAULT_NAME;
     this.persistenceMode = options.persistenceMode;
     this.pid = options.pid ?? y.pid;
+    this.shadowDatabaseConnectTimeoutMillis = options.shadowDatabaseConnectTimeoutMillis ?? this.databaseConnectTimeoutMillis;
+    this.shadowDatabaseIdleTimeoutMillis = options.shadowDatabaseIdleTimeoutMillis ?? this.databaseIdleTimeoutMillis;
     this._port = options.port ?? NO_PORT;
     this._shadowDatabasePort = options.shadowDatabasePort ?? NO_PORT;
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prisma/cli-init",
-  "version": "0.0.0-dev.202512170049",
+  "version": "0.0.0-dev.202512221324",
   "description": "Init CLI for Prisma",
   "type": "module",
   "sideEffects": false,