# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

require 'contrast/components/interface'

module Contrast
  module Agent
    # This module adds an at_exit hook for us to send messages that may be lost at process exit
    module AtExitHook
      include Contrast::Components::Interface
      access_component :logging
      def self.exit_hook
        @_exit_hook ||= begin
          at_exit do
            on_exit
          end
          true
        end
      end

      # Actions to take when a process exits. Typically called from our
      # exit_hook, but exposed here for other process terminations, like those
      # in Kernel#exec
      def self.on_exit
        logger.debug('at_exit invoked, host application terminating',
                     p_id: @pid,
                     pp_id: @ppid,
                     process_pid: Process.pid,
                     process_pp_id: Process.ppid)

        context = Contrast::Agent::REQUEST_TRACKER.current
        return unless context

        Contrast::Agent.messaging_queue.send_event_immediately(context.activity)
      end
    end
  end
end