module Appsignal
  module Probes
    class MriProbe
      include Helpers

      # @api private
      def self.dependencies_present?
        defined?(::RubyVM) && ::RubyVM.respond_to?(:stat)
      end

      def initialize(appsignal: Appsignal, gc_profiler: Appsignal::GarbageCollectionProfiler.new)
        Appsignal.logger.debug("Initializing VM probe")
        @appsignal = appsignal
        @gc_profiler = gc_profiler
      end

      # @api private
      def call
        stat = RubyVM.stat

        set_gauge(
          "ruby_vm",
          stat[:class_serial],
          :metric => :class_serial
        )

        set_gauge(
          "ruby_vm",
          stat[:constant_cache] ? stat[:constant_cache].values.sum : stat[:global_constant_state],
          :metric => :global_constant_state
        )

        set_gauge("thread_count", Thread.list.size)
        gauge_delta(:gc_time, @gc_profiler.total_time) do |gc_time|
          set_gauge("gc_time", gc_time) if gc_time > 0
        end

        gc_stats = GC.stat
        gauge_delta(
          :allocated_objects,
          gc_stats[:total_allocated_objects] || gc_stats[:total_allocated_object]
        ) do |allocated_objects|
          set_gauge("allocated_objects", allocated_objects)
        end

        gauge_delta(:gc_count, GC.count) do |gc_count|
          set_gauge("gc_count", gc_count, :metric => :gc_count)
        end
        gauge_delta(:minor_gc_count, gc_stats[:minor_gc_count]) do |minor_gc_count|
          set_gauge("gc_count", minor_gc_count, :metric => :minor_gc_count)
        end
        gauge_delta(:major_gc_count, gc_stats[:major_gc_count]) do |major_gc_count|
          set_gauge("gc_count", major_gc_count, :metric => :major_gc_count)
        end

        set_gauge("heap_slots", gc_stats[:heap_live_slots] || gc_stats[:heap_live_slot], :metric => :heap_live)
        set_gauge("heap_slots", gc_stats[:heap_free_slots] || gc_stats[:heap_free_slot], :metric => :heap_free)
      end

      private

      def set_gauge(metric, value, tags = {})
        @appsignal.set_gauge(metric, value, { :hostname => hostname }.merge(tags))
      end

      def hostname
        return @hostname if defined?(@hostname)

        config = @appsignal.config
        @hostname =
          if config[:hostname]
            config[:hostname]
          else
            # Auto detect hostname as fallback. May be inaccurate.
            Socket.gethostname
          end
        Appsignal.logger.debug "MRI probe: Using hostname config " \
          "option '#{@hostname.inspect}' as hostname"

        @hostname
      end
    end
  end
end