# frozen_string_literal: true

# Copyright The OpenTelemetry Authors
#
# SPDX-License-Identifier: Apache-2.0

# OpenTelemetry is an open source observability framework, providing a
# general-purpose API, SDK, and related tools required for the instrumentation
# of cloud-native software, frameworks, and libraries.
#
# The OpenTelemetry module provides global accessors for telemetry objects.
# See the documentation for the `opentelemetry-api` gem for details.
module OpenTelemetry
  # Namespace for OpenTelemetry propagator extension libraries
  module Propagator
    # Namespace for OpenTelemetry B3 propagation
    module B3
      # Namespace for OpenTelemetry b3 single header encoding
      module Single
        # Extracts context from carriers in the b3 single header format
        class TextMapExtractor
          B3_CONTEXT_REGEX = /\A(?<trace_id>(?:[0-9a-f]{16}){1,2})-(?<span_id>[0-9a-f]{16})(?:-(?<sampling_state>[01d](?![0-9a-f])))?(?:-(?<parent_span_id>[0-9a-f]{16}))?\z/.freeze
          SAMPLED_VALUES = %w[1 d].freeze

          # Returns a new TextMapExtractor that extracts b3 context using the
          # specified getter
          #
          # @param [optional Getter] default_getter The default getter used to read
          #   headers from a carrier during extract. Defaults to a
          #   {OpenTelemetry::Context:Propagation::TextMapGetter} instance.
          # @return [TextMapExtractor]
          def initialize(default_getter = Context::Propagation.text_map_getter)
            @default_getter = default_getter
          end

          # Extract b3 context from the supplied carrier and set the active span
          # in the given context. The original context will be returned if b3
          # cannot be extracted from the carrier.
          #
          # @param [Carrier] carrier The carrier to get the header from.
          # @param [Context] context The context to be updated with extracted context
          # @param [optional Getter] getter If the optional getter is provided, it
          #   will be used to read the header from the carrier, otherwise the default
          #   getter will be used.
          # @return [Context] Updated context with active span derived from the header, or the original
          #   context if parsing fails.
          def extract(carrier, context, getter = nil)
            getter ||= @default_getter
            header = getter.get(carrier, B3_CONTEXT_KEY)
            return context unless (match = header.match(B3_CONTEXT_REGEX))

            span_context = Trace::SpanContext.new(
              trace_id: B3.to_trace_id(match['trace_id']),
              span_id: B3.to_span_id(match['span_id']),
              trace_flags: to_trace_flags(match['sampling_state']),
              remote: true
            )

            span = Trace::Span.new(span_context: span_context)
            context = B3.context_with_debug(context) if match['sampling_state'] == 'd'
            Trace.context_with_span(span, parent_context: context)
          rescue OpenTelemetry::Error
            context
          end

          private

          def to_trace_flags(sampling_state)
            if SAMPLED_VALUES.include?(sampling_state)
              Trace::TraceFlags::SAMPLED
            else
              Trace::TraceFlags::DEFAULT
            end
          end
        end
      end
    end
  end
end