@replayio-app-building/netlify-recorder 0.12.0 → 0.13.0

package/dist/index.d.ts

@@ -442,13 +442,11 @@ declare function databaseAuditMonitorTable(sql: SqlFunction, tableName: string):
  */
 declare function databaseAuditDumpLogTable(sql: SqlFunction): Promise<Record<string, unknown>[]>;
 /**
- * Wraps a Neon SQL tagged-template function so that after each query,
- * any new `audit_log` rows (with NULL `replay_request_id`) are tagged
- * with the current request ID and an incrementing call index.
- *
- * The request ID is read from module-level state set by
- * `createRecordingRequestHandler`. If no request is active, the
- * wrapper passes calls through without tagging.
+ * Wraps a Neon SQL tagged-template function so that before each query,
+ * `SET LOCAL` is used to inject the current request ID and call index
+ * into the PostgreSQL session. The `audit_trigger_function` reads these
+ * via `current_setting()` and stamps audit rows atomically at INSERT
+ * time inside the trigger — no separate UPDATE required.
  *
  * @example
  * ```ts
@@ -456,7 +454,7 @@ declare function databaseAuditDumpLogTable(sql: SqlFunction): Promise<Record<str
  *
  * const sql = createAuditedSql(getSql());
  * await sql`INSERT INTO haikus (text) VALUES (${haiku})`;
- * // audit_log entries from this INSERT are now tagged with the request ID
+ * // audit_log row created by the trigger already has the request ID
  * ```
  */
 declare function createAuditedSql(sql: SqlFunction): SqlFunction;

package/dist/index.js

@@ -1130,10 +1130,22 @@ async function databaseAuditEnsureLogTable(sql) {
     RETURNS TRIGGER AS $$
     DECLARE
       changed_cols TEXT[];
+      req_id TEXT;
+      call_idx INTEGER;
     BEGIN
+      -- Read application context set via SET LOCAL by createAuditedSql
+      req_id := COALESCE(current_setting('app.replay_request_id', true), '');
+      IF req_id = '' THEN req_id := NULL; END IF;
+
+      BEGIN
+        call_idx := current_setting('app.replay_call_index', true)::INTEGER;
+      EXCEPTION WHEN OTHERS THEN
+        call_idx := NULL;
+      END;
+
       IF TG_OP = 'INSERT' THEN
-        INSERT INTO audit_log (table_name, record_id, action, new_data)
-        VALUES (TG_TABLE_NAME, NEW.id::TEXT, 'INSERT', to_jsonb(NEW));
+        INSERT INTO audit_log (table_name, record_id, action, new_data, replay_request_id, replay_request_call_index)
+        VALUES (TG_TABLE_NAME, NEW.id::TEXT, 'INSERT', to_jsonb(NEW), req_id, call_idx);
         RETURN NEW;
       ELSIF TG_OP = 'UPDATE' THEN
         SELECT ARRAY_AGG(n.key) INTO changed_cols
@@ -1141,12 +1153,12 @@ async function databaseAuditEnsureLogTable(sql) {
         LEFT JOIN jsonb_each_text(to_jsonb(OLD)) o ON n.key = o.key
         WHERE o.value IS DISTINCT FROM n.value;
 
-        INSERT INTO audit_log (table_name, record_id, action, old_data, new_data, changed_fields)
-        VALUES (TG_TABLE_NAME, NEW.id::TEXT, 'UPDATE', to_jsonb(OLD), to_jsonb(NEW), COALESCE(changed_cols, ARRAY[]::TEXT[]));
+        INSERT INTO audit_log (table_name, record_id, action, old_data, new_data, changed_fields, replay_request_id, replay_request_call_index)
+        VALUES (TG_TABLE_NAME, NEW.id::TEXT, 'UPDATE', to_jsonb(OLD), to_jsonb(NEW), COALESCE(changed_cols, ARRAY[]::TEXT[]), req_id, call_idx);
         RETURN NEW;
       ELSIF TG_OP = 'DELETE' THEN
-        INSERT INTO audit_log (table_name, record_id, action, old_data)
-        VALUES (TG_TABLE_NAME, OLD.id::TEXT, 'DELETE', to_jsonb(OLD));
+        INSERT INTO audit_log (table_name, record_id, action, old_data, replay_request_id, replay_request_call_index)
+        VALUES (TG_TABLE_NAME, OLD.id::TEXT, 'DELETE', to_jsonb(OLD), req_id, call_idx);
         RETURN OLD;
       END IF;
       RETURN NULL;
@@ -1177,22 +1189,19 @@ async function databaseAuditDumpLogTable(sql) {
 }
 function createAuditedSql(sql) {
   let callIndex = 0;
-  const requestId = getCurrentRequestId();
+  const rawSql = sql;
   const auditedSql = async (strings, ...values) => {
-    const result = await sql(strings, ...values);
+    const requestId = getCurrentRequestId();
     callIndex++;
     if (requestId) {
       try {
-        await sql`
-          UPDATE audit_log
-          SET replay_request_id = ${requestId}, replay_request_call_index = ${callIndex}
-          WHERE replay_request_id IS NULL
-        `;
+        await rawSql(`SET LOCAL app.replay_request_id = '${requestId.replace(/'/g, "''")}'`);
+        await rawSql(`SET LOCAL app.replay_call_index = '${callIndex}'`);
       } catch (err) {
-        console.warn("netlify-recorder: failed to update audit log:", err);
+        console.warn("netlify-recorder: failed to SET LOCAL audit context:", err);
       }
     }
-    return result;
+    return await sql(strings, ...values);
   };
   return auditedSql;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@replayio-app-building/netlify-recorder",
-  "version": "0.12.0",
+  "version": "0.13.0",
   "description": "Capture and replay Netlify function executions as Replay recordings",
   "type": "module",
   "exports": {