@retaw106/mmc-pi 2.4.0 → 2.6.0

package/dist/{index-cr4hzd76.js → index-vhs0y3s0.js}

@@ -166,6 +166,177 @@ function topLevelCommaSegments(text) {
 function getTagIdentityWriteGeneration() {
   return tagIdentityWriteGeneration;
 }
+var busyTimeoutStates = new WeakMap;
+var BUSY_TIMEOUT_QUERY = /^\s*PRAGMA\s+(?:(?:\w+|"[^"]+"|`[^`]+`|\[[^\]]+\])\s*\.\s*)?(?:busy_timeout|"busy_timeout"|`busy_timeout`|\[busy_timeout\])\s*;?\s*$/i;
+var BUSY_TIMEOUT_SETTER = /^\s*PRAGMA\s+(?:(?:\w+|"[^"]+"|`[^`]+`|\[[^\]]+\])\s*\.\s*)?(?:busy_timeout|"busy_timeout"|`busy_timeout`|\[busy_timeout\])\s*(?==|\()/i;
+function literalBusyTimeout(sql) {
+  const head = BUSY_TIMEOUT_SETTER.exec(sql);
+  if (!head)
+    return;
+  const match = /^(?:=\s*|\(\s*)(?:'(-?\d+)'|"(-?\d+)"|(-?\d+))\s*\)?\s*;?\s*$/.exec(sql.slice(head[0].length));
+  if (!match)
+    return;
+  const value = Number(match[1] ?? match[2] ?? match[3]);
+  return value < 0 ? 0 : value <= 2147483647 ? value : undefined;
+}
+function busyTimeoutStatements(sql) {
+  const statements = [];
+  let statement = "";
+  let quote = "";
+  for (let i = 0;i < sql.length; i++) {
+    const char = sql[i];
+    if (quote) {
+      statement += char;
+      if (char === quote) {
+        if (sql[i + 1] === quote)
+          statement += sql[++i];
+        else
+          quote = "";
+      }
+    } else if (char === "'" || char === '"' || char === "`" || char === "[") {
+      quote = char === "[" ? "]" : char;
+      statement += char;
+    } else if (char === "-" && sql[i + 1] === "-") {
+      while (i + 1 < sql.length && sql[i + 1] !== `
+`)
+        i++;
+      statement += " ";
+    } else if (char === "/" && sql[i + 1] === "*") {
+      i += 2;
+      while (i < sql.length && !(sql[i] === "*" && sql[i + 1] === "/"))
+        i++;
+      i++;
+      statement += " ";
+    } else if (char === ";") {
+      statements.push(statement);
+      statement = "";
+    } else
+      statement += char;
+  }
+  statements.push(statement);
+  return statements.filter((part) => part.trim().length > 0);
+}
+function installNativeDeadlineRouting(db) {
+  const rawExec = db.exec.bind(db);
+  const rawPrepare = db.prepare.bind(db);
+  const readTimeout = () => {
+    const statement = rawPrepare("PRAGMA busy_timeout");
+    try {
+      return Object.values(statement.get())[0];
+    } finally {
+      statement.finalize?.();
+    }
+  };
+  const timeoutState = { current: readTimeout() };
+  busyTimeoutStates.set(db, timeoutState);
+  const setBusyTimeout = (timeout) => {
+    rawExec(`PRAGMA busy_timeout=${timeout}`);
+    timeoutState.current = timeout;
+  };
+  const withinDeadline = (operation) => {
+    const deadline = nativeWaitScope?.getStore();
+    if (!deadline?.active || !deadline.remainingMs)
+      return operation();
+    const remaining = deadline.remainingMs();
+    if (remaining === Infinity)
+      return operation();
+    const previous = timeoutState.current;
+    const timeout = Math.min(previous, Math.max(0, Math.floor(remaining)));
+    if (timeout === previous)
+      return operation();
+    setBusyTimeout(timeout);
+    try {
+      return operation();
+    } finally {
+      setBusyTimeout(previous);
+    }
+  };
+  Object.defineProperty(db, "exec", {
+    configurable: true,
+    writable: true,
+    value: (sql) => {
+      const parts = /\bbusy_timeout\b/i.test(sql) ? busyTimeoutStatements(sql) : [];
+      if (parts.length === 1 && BUSY_TIMEOUT_QUERY.test(parts[0]))
+        return rawExec(sql);
+      const singleSetter = parts.length === 1 && BUSY_TIMEOUT_SETTER.test(parts[0]);
+      const setting = singleSetter ? literalBusyTimeout(parts[0]) : undefined;
+      if (singleSetter) {
+        const result = rawExec(sql);
+        timeoutState.current = setting ?? readTimeout();
+        return result;
+      }
+      if (!/\bbusy_timeout\b/i.test(sql))
+        return withinDeadline(() => rawExec(sql));
+      const setters = parts.filter((part) => BUSY_TIMEOUT_SETTER.test(part));
+      return withinDeadline(() => {
+        try {
+          const result = rawExec(sql);
+          if (setters.length)
+            timeoutState.current = literalBusyTimeout(setters[setters.length - 1]) ?? readTimeout();
+          return result;
+        } catch (error) {
+          if (setters.length)
+            timeoutState.current = readTimeout();
+          throw error;
+        }
+      });
+    }
+  });
+  const nativePrepare = (sql) => BUSY_TIMEOUT_QUERY.test(sql) || BUSY_TIMEOUT_SETTER.test(sql) ? rawPrepare(sql) : withinDeadline(() => rawPrepare(sql));
+  Object.defineProperty(db, "prepare", {
+    configurable: true,
+    writable: true,
+    value: (sql) => {
+      const statement = nativePrepare(sql);
+      const normalized = /\bbusy_timeout\b/i.test(sql) ? busyTimeoutStatements(sql)[0] ?? sql : sql;
+      const setting = literalBusyTimeout(normalized);
+      const timeoutSetter = BUSY_TIMEOUT_SETTER.test(normalized);
+      const timeoutQuery = BUSY_TIMEOUT_QUERY.test(normalized);
+      if (timeoutSetter)
+        timeoutState.current = setting ?? readTimeout();
+      for (const method of ["run", "get", "all", "values"]) {
+        const execute = statement[method]?.bind(statement);
+        if (!execute)
+          continue;
+        Object.defineProperty(statement, method, {
+          configurable: true,
+          value: (...args) => {
+            if (timeoutQuery)
+              return execute(...args);
+            if (!timeoutSetter)
+              return withinDeadline(() => execute(...args));
+            const result = execute(...args);
+            timeoutState.current = readTimeout();
+            return result;
+          }
+        });
+      }
+      if (typeof statement.iterate === "function") {
+        const iterate = statement.iterate.bind(statement);
+        Object.defineProperty(statement, "iterate", {
+          configurable: true,
+          value: (...args) => {
+            const iterator = !timeoutSetter && !timeoutQuery ? withinDeadline(() => iterate(...args)) : iterate(...args);
+            const next = iterator.next.bind(iterator);
+            if (timeoutSetter)
+              timeoutState.current = readTimeout();
+            iterator.next = () => {
+              if (timeoutQuery)
+                return next();
+              if (!timeoutSetter)
+                return withinDeadline(next);
+              const result = next();
+              timeoutState.current = readTimeout();
+              return result;
+            };
+            return iterator;
+          }
+        });
+      }
+      return statement;
+    }
+  });
+}
 function installTransactionRouting(db, readonly) {
   const nativeExec = db.exec.bind(db);
   Object.defineProperty(db, "exec", {
@@ -303,7 +474,7 @@ function wrapRemainingTagWriteForms(statement) {
     });
   }
 }
-function trackSqliteConnection(db, filename, options) {
+function trackSqliteConnection(db, filename, options, turnDeadlines = false) {
   const originalClose = db.close.bind(db);
   const sequence = nextSqliteConnectionSequence++;
   const metadata = {
@@ -311,6 +482,8 @@ function trackSqliteConnection(db, filename, options) {
     filename: typeof filename === "string" ? filename : Buffer.isBuffer(filename) ? "<buffer>" : ":memory:",
     readonly: Boolean(options) && typeof options === "object" && (options.readonly === true || options.readOnly === true)
   };
+  if (turnDeadlines)
+    installNativeDeadlineRouting(db);
   installTransactionRouting(db, metadata.readonly);
   Object.defineProperty(db, "close", {
     configurable: true,
@@ -363,6 +536,12 @@ function buildNodeSqliteDatabaseClass(DatabaseSync) {
   return NodeSqliteDatabase;
 }
 var Database = TrackedDatabase;
+function createSqliteTurnDatabase(filename, options) {
+  return trackSqliteConnection(new DatabaseImpl(filename, options), filename, options, true);
+}
+function hasSqliteTurnDeadlineRouting(db) {
+  return busyTimeoutStates.has(db);
+}
 function getSqliteDatabasePath(db) {
   for (const connection of trackedSqliteConnections.values()) {
     if (connection.reference.deref() === db) {
@@ -382,6 +561,11 @@ function pragmaValue(db, name) {
   return row[name] ?? Object.values(row)[0];
 }
 function pragmaNumber(db, name) {
+  if (name === "busy_timeout") {
+    const state = busyTimeoutStates.get(db);
+    if (state)
+      return state.current;
+  }
   try {
     const value = pragmaValue(db, name);
     return typeof value === "number" && Number.isFinite(value) ? value : null;
@@ -447,11 +631,34 @@ function prepareCachedStatement(db, sql) {
   return ownPreparedStatement(db, db.prepare(sql));
 }
 var privilegeDepth = new WeakMap;
+var nativeWaitScope;
+function releaseNativeWaitLease(lease) {
+  lease.active = false;
+  lease.remainingMs = undefined;
+}
+function withSqliteTurnDeadline(remainingMs, operation) {
+  const scope = nativeWaitScope ??= new AsyncLocalStorage;
+  const lease = { active: true, remainingMs };
+  return scope.run(lease, () => {
+    try {
+      const result = operation();
+      if (result && typeof result.then === "function")
+        return Promise.resolve(result).finally(releaseNativeWaitLease.bind(null, lease));
+      releaseNativeWaitLease(lease);
+      return result;
+    } catch (error) {
+      releaseNativeWaitLease(lease);
+      throw error;
+    }
+  });
+}
 var transformPassScope = new AsyncLocalStorage;
 var admissionScope = new AsyncLocalStorage;
 var backgroundWriterScope = new AsyncLocalStorage;
 function withSqliteBackgroundWriter(operation) {
-  return backgroundWriterScope.run(true, operation);
+  if (!nativeWaitScope?.getStore())
+    return backgroundWriterScope.run(true, operation);
+  return nativeWaitScope.run(undefined, () => backgroundWriterScope.run(true, operation));
 }
 function withSqliteTransformPass(operation) {
   const existing = transformPassScope.getStore();
@@ -482,7 +689,9 @@ function withSqliteTransformPass(operation) {
   return typeof reservation === "function" ? run(reservation) : reservation.then(run);
 }
 function withoutSqliteTransformPass(operation) {
-  return transformPassScope.run(undefined, operation);
+  if (!nativeWaitScope?.getStore())
+    return transformPassScope.run(undefined, operation);
+  return nativeWaitScope.run(undefined, () => transformPassScope.run(undefined, operation));
 }
 function guardSqliteTransformPass(guard) {
   const lease = transformPassScope.getStore();
@@ -656,7 +865,7 @@ async function withAsyncPrivilegedWriter(db, operation, options = {}) {
     } catch (error) {
       if (entered || !(error instanceof SqliteAcquisitionBusyError))
         throw error;
-      options.beforeRetry?.(error);
+      await options.beforeRetry?.(error);
       const elapsed = performance.now() - started;
       if (elapsed >= budgetMs) {
         reportSqliteDiagnostic(`sqlite writer site=privileged_writer lane=foreground acquire_ms=${Math.round(elapsed)} attempts=${attempts} outcome=busy`);
@@ -791,4 +1000,4 @@ async function withIndexChunk(input, operation) {
   }
 }
 
-export { inRootAsyncContext, registerSlowWriteReporter, registerSqliteDiagnosticSink, detectSqliteRuntime, getTagIdentityWriteGeneration, Database, getSqliteDatabasePath, prepareCachedStatement, withSqliteBackgroundWriter, withSqliteTransformPass, withoutSqliteTransformPass, guardSqliteTransformPass, isTransientSqliteError, withAsyncPrivilegedWriter, withPrivilegedWriter, indexCoordinationInput, withIndexChunk };
+export { inRootAsyncContext, registerSlowWriteReporter, registerSqliteDiagnosticSink, detectSqliteRuntime, getTagIdentityWriteGeneration, Database, createSqliteTurnDatabase, hasSqliteTurnDeadlineRouting, getSqliteDatabasePath, prepareCachedStatement, withSqliteTurnDeadline, withSqliteBackgroundWriter, withSqliteTransformPass, withoutSqliteTransformPass, guardSqliteTransformPass, isTransientSqliteError, withAsyncPrivilegedWriter, withPrivilegedWriter, indexCoordinationInput, withIndexChunk };