bitfab 0.10.4 → 0.10.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e5d6df92ce06e91ff963c8dc0e78af3a0f591c51b53330c1c8630bb5d66e7663
-  data.tar.gz: 199d4883cfad921563b3b5786d2db5b017d4b618e000c7d259ed9f69be3eb7ba
+  metadata.gz: f20353e1bb03affda1786ed5f8566ab1b3a401f0f9635023bae2fd350bcc7de6
+  data.tar.gz: 8e10f199e9d81cfdbfa2bc9172327131301dc3959307340bbe6c55e1ef2bc235
 SHA512:
-  metadata.gz: 02c87809df7f4039dd608d3e60ddc0ff85b791d834ea7dc5557a368987d5dba1a174705074a26a9df80d4a8bb7c4ff7493f4c8ac03beb457629d896768a70567
-  data.tar.gz: 579f94120498658ceeb5129b3a21a9d1477d816be9216963c168e0c8747aef0687298f632539f33ace2960db53a19a9ab7847f776d27d971e1b006ad1ebbc909
+  metadata.gz: ade88c2b4d5878a3f6e0cee6f70c4a2d63b1a2f0359f425e9da6ab787806c8ef9396ca0e0a71f5ec7c75647e658419f0e9eb099fbbb4ce22a56c38adf26632e8
+  data.tar.gz: 9aa7b82e532a84893533b97d67d413cd4012729624d6bde8433bfdecf73c1d929269e6a1d779e50ed7d13e6dcc84a050f22fdf41282f2de4469bd0c5a6a35b30

data/lib/bitfab/client.rb

@@ -36,9 +36,18 @@ module Bitfab
     # @param limit [Integer] maximum number of traces to replay (default: 5)
     # @param trace_ids [Array<String>, nil] optional list of trace IDs to filter
     # @param max_concurrency [Integer, nil] max threads for parallel replay (default: 10)
+    # @param code_change_description [String, nil] optional rationale for the
+    #   code change being tested in this replay (stored on the experiment)
+    # @param code_change_files [Array<Hash>, nil] optional list of edited files,
+    #   each as { path:, before:, after: } (use "" for new/deleted files)
     # @return [Hash] with :items, :test_run_id, :test_run_url
-    def replay(receiver, method_name, trace_function_key:, limit: 5, trace_ids: nil, max_concurrency: 10)
-      Replay.run(self, receiver, method_name, trace_function_key:, limit:, trace_ids:, max_concurrency:)
+    def replay(receiver, method_name, trace_function_key:, limit: 5, trace_ids: nil, max_concurrency: 10,
+      code_change_description: nil, code_change_files: nil)
+      Replay.run(
+        self, receiver, method_name,
+        trace_function_key:, limit:, trace_ids:, max_concurrency:,
+        code_change_description:, code_change_files:
+      )
     end
 
     # Execute a block inside a span context, sending trace data on completion.
@@ -56,10 +65,15 @@ module Bitfab
       replay_ctx = ReplayContext.current
       resolved_test_run_id = replay_ctx&.dig(:test_run_id)
       resolved_input_source_span_id = replay_ctx&.dig(:input_source_span_id)
+      resolved_input_source_trace_id = replay_ctx&.dig(:input_source_trace_id)
 
       # Register trace state for root spans
       if is_root_span && !TraceState.get(trace_id)
-        TraceState.create(trace_id, test_run_id: resolved_test_run_id)
+        TraceState.create(
+          trace_id,
+          test_run_id: resolved_test_run_id,
+          input_source_trace_id: resolved_input_source_trace_id
+        )
       end
 
       if is_root_span
@@ -219,6 +233,9 @@ module Bitfab
       if trace_state&.dig(:contexts)
         raw_trace["contexts"] = trace_state[:contexts]
       end
+      if trace_state&.dig(:input_source_trace_id)
+        raw_trace["input_source_trace_id"] = trace_state[:input_source_trace_id]
+      end
 
       payload = {
         "type" => "sdk-function",

data/lib/bitfab/http_client.rb

@@ -98,12 +98,19 @@ module Bitfab
 
     # Start a replay session by fetching historical traces.
     # Blocking call. Returns hash with testRunId, testRunUrl, and items array.
-    def start_replay(trace_function_key, limit, trace_ids: nil)
+    #
+    # @param code_change_description [String, nil] optional rationale for the
+    #   code change being tested in this replay
+    # @param code_change_files [Array<Hash>, nil] optional list of edited files,
+    #   each as { path:, before:, after: } (use "" for new/deleted files)
+    def start_replay(trace_function_key, limit, trace_ids: nil, code_change_description: nil, code_change_files: nil)
       payload = {
         "traceFunctionKey" => trace_function_key,
         "limit" => limit
       }
       payload["traceIds"] = trace_ids if trace_ids
+      payload["codeChangeDescription"] = code_change_description unless code_change_description.nil?
+      payload["codeChangeFiles"] = normalize_code_change_files(code_change_files) unless code_change_files.nil?
 
       request("/api/sdk/replay/start", payload, timeout: 30)
     end
@@ -129,6 +136,18 @@ module Bitfab
 
     private
 
+    # Normalize each entry to a hash with stable string keys, accepting either
+    # symbol-keyed or string-keyed hashes from callers.
+    def normalize_code_change_files(files)
+      files.map do |file|
+        {
+          "path" => file[:path] || file["path"],
+          "before" => file[:before] || file["before"],
+          "after" => file[:after] || file["after"]
+        }
+      end
+    end
+
     def headers
       {
         "Content-Type" => "application/json",

data/lib/bitfab/replay.rb

@@ -14,11 +14,12 @@ module Bitfab
 
     # Execute a block with replay context set on the current thread.
     # The context is automatically cleared when the block completes.
-    def with_context(test_run_id:, input_source_span_id: nil)
+    def with_context(test_run_id:, input_source_span_id: nil, input_source_trace_id: nil)
       previous = Thread.current[REPLAY_CONTEXT_KEY]
       Thread.current[REPLAY_CONTEXT_KEY] = {
         test_run_id:,
-        input_source_span_id:
+        input_source_span_id:,
+        input_source_trace_id:
       }
       yield
     ensure
@@ -42,11 +43,22 @@ module Bitfab
     # @param limit [Integer] maximum number of traces to replay (default: 5)
     # @param trace_ids [Array<String>, nil] optional list of trace IDs to filter
     # @param max_concurrency [Integer, nil] max threads for parallel replay (default: 10)
+    # @param code_change_description [String, nil] optional rationale for the
+    #   code change being tested in this replay (stored on the experiment)
+    # @param code_change_files [Array<Hash>, nil] optional list of edited files,
+    #   each as { path:, before:, after: } (empty string for new/deleted files)
     # @return [Hash] with :items, :test_run_id, :test_run_url
-    def run(client, receiver, method_name, trace_function_key:, limit: 5, trace_ids: nil, max_concurrency: 10)
+    def run(client, receiver, method_name, trace_function_key:, limit: 5, trace_ids: nil, max_concurrency: 10,
+      code_change_description: nil, code_change_files: nil)
       http_client = client.instance_variable_get(:@http_client)
 
-      replay_data = http_client.start_replay(trace_function_key, limit, trace_ids:)
+      replay_data = http_client.start_replay(
+        trace_function_key,
+        limit,
+        trace_ids:,
+        code_change_description:,
+        code_change_files:
+      )
       test_run_id = replay_data["testRunId"]
       test_run_url = replay_data["testRunUrl"]
       server_items = replay_data["items"] || []
@@ -106,7 +118,15 @@ module Bitfab
       span = http_client.get_external_span(server_item["externalSpanId"])
       item_data = extract_span_data(span)
       metrics = extract_server_item_metrics(server_item)
-      execute_item(item_data, receiver, method_name, test_run_id, span["id"], metrics)
+      execute_item(
+        item_data,
+        receiver,
+        method_name,
+        test_run_id,
+        span["id"],
+        metrics,
+        input_source_trace_id: span["externalTraceId"]
+      )
     end
 
     # Extract input/output data from an external span's rawData.
@@ -144,13 +164,14 @@ module Bitfab
     end
 
     # Execute a single replay item: deserialize inputs, call method with replay context.
-    def execute_item(item, receiver, method_name, test_run_id, input_source_span_id = nil, metrics = {})
+    def execute_item(item, receiver, method_name, test_run_id, input_source_span_id = nil, metrics = {},
+      input_source_trace_id: nil)
       args, kwargs = Serialize.deserialize_inputs(item)
 
       fn_result = nil
       fn_error = nil
 
-      ReplayContext.with_context(test_run_id:, input_source_span_id:) do
+      ReplayContext.with_context(test_run_id:, input_source_span_id:, input_source_trace_id:) do
         fn_result = if kwargs.empty?
           receiver.send(method_name, *args)
         else

data/lib/bitfab/span_context.rb

@@ -131,12 +131,13 @@ module Bitfab
       @states_mutex.synchronize { @states[trace_id] }
     end
 
-    def create(trace_id, test_run_id: nil)
+    def create(trace_id, test_run_id: nil, input_source_trace_id: nil)
       @states_mutex.synchronize do
         @states[trace_id] ||= {
           trace_id:,
           started_at: Time.now.utc.strftime("%Y-%m-%dT%H:%M:%S.%3NZ"),
-          test_run_id:
+          test_run_id:,
+          input_source_trace_id:
         }.compact
       end
     end

data/lib/bitfab/version.rb

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

metadata

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