@bitfab/sdk 0.18.1 → 0.18.2

package/dist/index.d.cts

@@ -335,6 +335,14 @@ declare class ReplayEnvironment {
     get active(): boolean;
     /** Non-throwing variant for callers that handle the inactive case. */
     snapshot(): ReplayEnvironmentSnapshot | null;
+    /**
+     * Record on the replay context that customer code obtained the branch
+     * URL. Only `databaseUrl` and `snapshot()` count — `active`, `readOnly`
+     * and friends inspect the lease without exposing the connection string,
+     * so they don't prove the replayed code could have connected to the
+     * branch.
+     */
+    private markAccessed;
     private read;
     private require;
 }
@@ -1055,7 +1063,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.18.1";
+declare const __version__ = "0.18.2";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -335,6 +335,14 @@ declare class ReplayEnvironment {
     get active(): boolean;
     /** Non-throwing variant for callers that handle the inactive case. */
     snapshot(): ReplayEnvironmentSnapshot | null;
+    /**
+     * Record on the replay context that customer code obtained the branch
+     * URL. Only `databaseUrl` and `snapshot()` count — `active`, `readOnly`
+     * and friends inspect the lease without exposing the connection string,
+     * so they don't prove the replayed code could have connected to the
+     * branch.
+     */
+    private markAccessed;
     private read;
     private require;
 }
@@ -1055,7 +1063,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.18.1";
+declare const __version__ = "0.18.2";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -11,10 +11,10 @@ import {
   flushTraces,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-ILIUTS5D.js";
+} from "./chunk-UTVFKV2Z.js";
 import {
   BitfabError
-} from "./chunk-QT7HWOKU.js";
+} from "./chunk-PY6V4FE3.js";
 export {
   Bitfab,
   BitfabClaudeAgentHandler,

package/dist/node.cjs

@@ -447,7 +447,7 @@ registerAsyncLocalStorageClass(
 );
 
 // src/version.generated.ts
-var __version__ = "0.18.1";
+var __version__ = "0.18.2";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -2162,7 +2162,9 @@ var ReplayEnvironment = class {
    * Throws if read outside a replay item.
    */
   get databaseUrl() {
-    return this.require().databaseUrl;
+    const snapshot = this.require();
+    this.markAccessed();
+    return snapshot.databaseUrl;
   }
   /** When the per-trace branch URL stops being valid. ISO-8601. */
   get expiresAt() {
@@ -2190,7 +2192,24 @@ var ReplayEnvironment = class {
   }
   /** Non-throwing variant for callers that handle the inactive case. */
   snapshot() {
-    return this.read();
+    const snapshot = this.read();
+    if (snapshot) {
+      this.markAccessed();
+    }
+    return snapshot;
+  }
+  /**
+   * Record on the replay context that customer code obtained the branch
+   * URL. Only `databaseUrl` and `snapshot()` count — `active`, `readOnly`
+   * and friends inspect the lease without exposing the connection string,
+   * so they don't prove the replayed code could have connected to the
+   * branch.
+   */
+  markAccessed() {
+    const ctx = getReplayContext();
+    if (ctx?.dbBranchLease) {
+      ctx.dbSnapshotAccessed = true;
+    }
   }
   read() {
     const ctx = getReplayContext();
@@ -3122,7 +3141,19 @@ var Bitfab = class {
               contexts: traceState?.contexts ?? [],
               testRunId: traceState?.testRunId,
               inputSourceTraceId: traceState?.inputSourceTraceId,
-              dbSnapshotRef: traceState?.dbSnapshotRef
+              dbSnapshotRef: traceState?.dbSnapshotRef,
+              // Built AFTER the wrapped fn finished, so `accessed` reflects
+              // whether customer code obtained the branch URL during this
+              // item. Omitted entirely when no lease was attached, so the
+              // server can distinguish "no branch" from "branch ignored".
+              ...replayCtx?.dbBranchLease && {
+                dbSnapshotUsage: {
+                  neonBranchId: replayCtx.dbBranchLease.neonBranchId,
+                  snapshotTimestamp: replayCtx.dbBranchLease.snapshotTimestamp,
+                  sourceTraceId: replayCtx.sourceBitfabTraceId,
+                  accessed: replayCtx.dbSnapshotAccessed === true
+                }
+              }
             });
             activeTraceStates.delete(traceId);
             if (persistenceCollector) {
@@ -3294,6 +3325,18 @@ var Bitfab = class {
     if (params.dbSnapshotRef) {
       rawTrace.db_snapshot_ref = params.dbSnapshotRef;
     }
+    if (params.dbSnapshotUsage) {
+      rawTrace.db_snapshot_usage = {
+        neon_branch_id: params.dbSnapshotUsage.neonBranchId,
+        ...params.dbSnapshotUsage.snapshotTimestamp && {
+          snapshot_timestamp: params.dbSnapshotUsage.snapshotTimestamp
+        },
+        ...params.dbSnapshotUsage.sourceTraceId && {
+          source_trace_id: params.dbSnapshotUsage.sourceTraceId
+        },
+        accessed: params.dbSnapshotUsage.accessed
+      };
+    }
     return this.httpClient.sendExternalTrace({
       type: "sdk-function",
       source: "typescript-sdk-function",