@cortexkit/pi-magic-context 0.45.0 → 0.46.0

package/dist/{index-2pbbk3en.js → index-wv28nczk.js}

@@ -168,6 +168,12 @@ function trackSqliteConnection(db, filename, options) {
     ...metadata,
     reference: new WeakRef(db)
   });
+  const prepare = db.prepare.bind(db);
+  Object.defineProperty(db, "prepare", {
+    configurable: true,
+    writable: true,
+    value: (sql) => ownPreparedStatement(db, prepare(sql))
+  });
   return db;
 }
 var TrackedDatabase = new Proxy(DatabaseImpl, {
@@ -213,6 +219,54 @@ function pragmaNumber(db, name) {
     return null;
   }
 }
+var statementOwners = new WeakMap;
+var FIRST_PRUNE_AT = 256;
+function ownPreparedStatement(db, statement) {
+  const finalizable = statement;
+  if (typeof finalizable.finalize !== "function")
+    return statement;
+  let owner = statementOwners.get(db);
+  if (!owner) {
+    const references = new Set;
+    owner = { references, registered: new WeakSet, pruneAt: FIRST_PRUNE_AT };
+    statementOwners.set(db, owner);
+    const close = db.close.bind(db);
+    Object.defineProperty(db, "close", {
+      configurable: true,
+      writable: true,
+      value: (...args) => {
+        let failure;
+        for (const reference of references) {
+          try {
+            reference.deref()?.finalize();
+          } catch (error) {
+            failure ??= error;
+          }
+        }
+        references.clear();
+        const result = close(...args);
+        if (failure)
+          throw failure;
+        return result;
+      }
+    });
+  }
+  if (!owner.registered.has(finalizable)) {
+    if (owner.references.size >= owner.pruneAt) {
+      for (const reference of owner.references) {
+        if (reference.deref() === undefined)
+          owner.references.delete(reference);
+      }
+      owner.pruneAt = Math.max(FIRST_PRUNE_AT, owner.references.size * 2);
+    }
+    owner.references.add(new WeakRef(finalizable));
+    owner.registered.add(finalizable);
+  }
+  return statement;
+}
+function prepareCachedStatement(db, sql) {
+  return ownPreparedStatement(db, db.prepare(sql));
+}
 var privilegeDepth = new WeakMap;
 var transformPassScope = new AsyncLocalStorage;
 var admissionScope = new AsyncLocalStorage;
@@ -407,4 +461,4 @@ function withPrivilegedWriter(db, operation) {
   }
 }
 
-export { registerSlowWriteReporter, registerSqliteDiagnosticSink, detectSqliteRuntime, Database, withSqliteBackgroundWriter, withSqliteTransformPass, withoutSqliteTransformPass, isTransientSqliteError, withPrivilegedWriter };
+export { registerSlowWriteReporter, registerSqliteDiagnosticSink, detectSqliteRuntime, Database, prepareCachedStatement, withSqliteBackgroundWriter, withSqliteTransformPass, withoutSqliteTransformPass, isTransientSqliteError, withPrivilegedWriter };