bitfab 0.36.0 → 0.36.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b5fdfbac8e62c59768ba554cd7c68b6fb7de3f1a37befd01a66cc6097821499
-  data.tar.gz: 31bb825d1fcde3c946c5467bb6e1ccbd56816815b38ba4f3eb5395bdb5061907
+  metadata.gz: e0b827ad6e9bca08b3025f9c2419290910999a5985705369e6375aee8091232c
+  data.tar.gz: e6e0a414013d3bee0efda22415430d9690fe537973a36e5f3dda540fcda3d5e4
 SHA512:
-  metadata.gz: efd78672cf98417507533b22bc0d4783ad7c29a18c4e9f419d95e328831275c26b4d15888642f07e19cd7a85b62babe55e223576d6a4885228fea4243ab84607
-  data.tar.gz: 83dc9683122eefbe92fc747d783fb81351acc6ccef94e1c31f5ae2b391adac9d775832d9f56818cde2541817a4b8cde475cda44021b93a5f23bcbac9a010904a
+  metadata.gz: ee72cf5ed5c81ad34dbd4d980091523c90c7cb1d11fc716499ab5f604f1829f5653851350942b2cea6660e20e67c065ae8807e0778153a0ccf08a1397cc08491
+  data.tar.gz: 297efa6c3e5d9b1086536dd5e560168e9f9d4c638638b562a67d69ccb1c37253534da6447778032f93e186532f0a563eb7b7f41148eb6b9e0e2ac91c4ed2196e

data/lib/bitfab/client.rb

@@ -407,6 +407,7 @@ module Bitfab
               {
                 neon_branch_id: lease["neonBranchId"],
                 snapshot_timestamp: lease["snapshotTimestamp"],
+                region: lease["region"],
                 original_trace_id: replay_ctx[:source_bitfab_trace_id],
                 # Deprecated wire alias, kept so this SDK still reports usage
                 # against servers that predate the rename.
@@ -593,6 +594,9 @@ module Bitfab
         if db_snapshot_usage[:snapshot_timestamp]
           usage["snapshot_timestamp"] = db_snapshot_usage[:snapshot_timestamp]
         end
+        if db_snapshot_usage[:region]
+          usage["region"] = db_snapshot_usage[:region]
+        end
         if db_snapshot_usage[:original_trace_id]
           usage["original_trace_id"] = db_snapshot_usage[:original_trace_id]
           # Deprecated wire alias, kept so this SDK still reports usage

data/lib/bitfab/replay.rb

@@ -62,10 +62,12 @@ module Bitfab
       # (via +database_url+). Reported on the trace
       # completion inside the +db_snapshot_usage+ record (its +accessed+
       # field) so the server can distinguish "branch was provisioned and
-      # exposed" from "branch URL was actually consumed". Any future
-      # consumption path that hands the URL to customer code by other means
-      # (e.g. a process-isolated runner writing an env overlay) must also
-      # set this.
+      # exposed" from "branch URL was actually consumed". Only an explicit
+      # +database_url+ read may set it. A path that hands the URL over by other
+      # means (e.g. a process-isolated runner writing an env overlay) must
+      # leave it alone: setting it there would make every such replay report
+      # accessed for free, and the flag would stop separating "branch was
+      # used" from "branch was offered".
       ctx[:db_branch_lease] = db_branch_lease if db_branch_lease
       ctx[:source_bitfab_trace_id] = source_bitfab_trace_id if source_bitfab_trace_id
       Thread.current[REPLAY_CONTEXT_KEY] = ctx

data/lib/bitfab/replay_branch.rb

@@ -17,9 +17,20 @@ module Bitfab
   # protocol term). Its useful fields are exposed directly here so customer code
   # never sees the word.
   class ReplayBranch
+    # The provider's own id for this branch, e.g. for correlating with its console.
+    attr_reader :neon_branch_id
+
+    # Env var name the customer's app reads, e.g. "DATABASE_URL".
+    attr_reader :env_key
+
     # When this branch's URL stops being valid. ISO-8601.
     attr_reader :expires_at
 
+    # The instant this branch is pinned to: the source trace's wall clock, read
+    # just before the traced method ran. Compare it against the trace you meant
+    # to replay to confirm the branch is the right point in history.
+    attr_reader :snapshot_timestamp
+
     # Deep link to the branch in the provider console, if available.
     attr_reader :provider_console_url
 
@@ -36,13 +47,28 @@ module Bitfab
 
     # Built by Bitfab.current_replay_branch; never constructed by callers.
     def initialize(lease, trace_id, context)
-      @expires_at = lease["expiresAt"]
-      @provider_console_url = lease["providerConsoleUrl"]
-      @read_only = lease["readOnly"]
-      @region = lease["region"]
+      # Copy the lease wholesale minus the connection string, so a field the
+      # server starts sending reaches customer code without an SDK release.
+      # Everything set here must stay plain data: #database_url is the only
+      # member allowed to mark the branch as accessed.
+      # The internals are underscore-prefixed because the loop below writes an
+      # ivar per lease key: snake_case never emits a leading underscore, so no
+      # field the server invents can collide with them. A plain @fields would
+      # be clobbered mid-loop by a key named "fields", corrupting #as_json.
+      fields = {}
+      lease.each do |key, value|
+        next if key == "databaseUrl"
+
+        name = snake_case(key)
+        fields[name] = value
+        instance_variable_set(:"@#{name}", value)
+        define_singleton_method(name) { value } unless respond_to?(name)
+      end
       @trace_id = trace_id
-      @url = lease["databaseUrl"]
-      @context = context
+      fields["trace_id"] = trace_id
+      @_fields = fields.freeze
+      @_url = lease["databaseUrl"]
+      @_context = context
       freeze
     end
 
@@ -54,15 +80,39 @@ module Bitfab
     # was actually used". The other readers inspect the lease without exposing
     # the connection string, so they deliberately do not record anything.
     def database_url
-      @context[:db_snapshot_accessed] = true
-      @url
+      @_context[:db_snapshot_accessed] = true
+      @_url
     end
 
     # Redact the connection string so a logged or inspected branch cannot leak it.
     def inspect
       "#<Bitfab::ReplayBranch trace_id=#{@trace_id.inspect} " \
         "expires_at=#{@expires_at.inspect} region=#{@region.inspect} " \
-        "read_only=#{@read_only.inspect}>"
+        "read_only=#{@read_only.inspect} " \
+        "snapshot_timestamp=#{@snapshot_timestamp.inspect}>"
+    end
+
+    # ActiveSupport gives every object a +to_json+ that serializes its instance
+    # variables, so under Rails a branch reaching a log line or an API response
+    # would carry the connection string and the whole replay context. Serialize
+    # the exposed fields only, the same set +inspect+ shows.
+    def as_json(_options = nil)
+      @_fields.dup
+    end
+
+    def to_json(*args)
+      as_json.to_json(*args)
+    end
+
+    private
+
+    # Two passes so a run of capitals stays one word: "dbURL" is "db_url", not
+    # "db_u_r_l". Must match the Python SDK's conversion exactly, or the same
+    # server field would arrive under different names in the two SDKs.
+    def snake_case(key)
+      key.gsub(/([A-Z\d]+)([A-Z][a-z])/, '\1_\2')
+        .gsub(/([a-z\d])([A-Z])/, '\1_\2')
+        .downcase
     end
   end
 end

data/lib/bitfab/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Bitfab
-  VERSION = "0.36.0"
+  VERSION = "0.36.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bitfab
 version: !ruby/object:Gem::Version
-  version: 0.36.0
+  version: 0.36.1
 platform: ruby
 authors:
 - Harvest Team