add rbs

Author: ZStriker19

lib/datadog/tracing/distributed/baggage.rb

@@ -77,7 +77,7 @@ def extract(data)
           )
         end
 
-        private
+          private
 
         def encode_key(key)
           key.strip.chars.map do |char|
@@ -106,18 +106,14 @@ def parse_baggage_header(baggage_header)
             next unless key_value.include?('=')
 
             key, value = key_value.split('=', 2)
-            key = decode_and_preserve_safe_characters(key.strip, SAFE_CHARACTERS_KEY)
-            value = decode_and_preserve_safe_characters(value.strip, SAFE_CHARACTERS_VALUE)
+            key = URI.decode_www_form_component(key.strip)
+            value = URI.decode_www_form_component(value.strip)
             next if key.empty? || value.empty?
 
             baggage[key] = value
           end
           baggage
         end
-
-        def decode_and_preserve_safe_characters(str, _safe_characters)
-          URI.decode_www_form_component(str)
-        end
       end
     end
   end

sig/datadog/tracing/distributed/baggage.rbs

@@ -0,0 +1,28 @@
+module Datadog
+  module Tracing
+    module Distributed
+      class Baggage
+        BAGGAGE_KEY : 'baggage'
+        DD_TRACE_BAGGAGE_MAX_ITEMS : 64
+        DD_TRACE_BAGGAGE_MAX_BYTES : 8192
+        SAFE_CHARACTERS_VALUE : "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789$!#&'()*+-./:<>?@[]^_`{|}~"
+
+        def initialize: (fetcher: untyped, baggage_key: string) -> void
+
+        def inject!: (untyped digest, untyped data) -> (nil | untyped)
+
+        def extract: (untyped data) -> (nil | untyped)
+          
+
+        private
+
+        def encode_key: (string key) -> string
+
+        def encode_value: (string key) -> string
+
+        def parse_baggage_header: (string baggage_header) -> hash[string, string]
+        end
+      end
+    end
+  end
+end