Class: Skylight::Probes::ActionController::Probe Private

Inherits:
Object
  • Object
show all
Defined in:
lib/skylight/probes/action_controller.rb

This class is part of a private API. You should avoid using this class if possible, as it may be removed or be changed in the future.

Instance Method Summary collapse

Instance Method Details

#installObject

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# File 'lib/skylight/probes/action_controller.rb', line 5

def install
  # Prepending doesn't work here since this a module that's already been included
  ::ActionController::Instrumentation.class_eval do
    private

    alias_method :append_info_to_payload_without_sk, :append_info_to_payload
    def append_info_to_payload(payload)
      append_info_to_payload_without_sk(payload)

      mime = sk_rendered_mime.try(:ref)

      payload[:sk_rendered_format] = if request&.headers&.[]("Turbo-Frame")
        [mime, "turbo-frame"].compact.join("+")
      else
        mime
      end

      payload[:sk_variant] = request.respond_to?(:variant) ? request.variant : nil
    end

    def sk_rendered_mime
      if respond_to?(:media_type)
        mt = media_type
        return mt && Mime::Type.lookup(mt)
      end

      if content_type.is_a?(Mime::Type)
        content_type
      elsif content_type.respond_to?(:to_s)
        type_str = content_type.to_s.split(";").first
        Mime::Type.lookup(type_str) unless type_str.blank?
      elsif respond_to?(:rendered_format) && rendered_format
        rendered_format
      end
    rescue StandardError
      # There are cases in which actionpack can return
      # a stringified representation of a Mime::NullType instance,
      # which is invalid for a number of reasons. This string raises
      # errors when piped through Mime::Type.lookup, so it's probably
      # best to just return nil in those cases.
      nil
    end
  end
end