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

require 'contrast/components/logger'
require 'contrast/agent/worker_thread'

module Contrast
  module Agent
    # The ServiceHeartbeat functions to keep the Contrast Service alive and
    # ensure that it maintains this Agent's ApplicationContext.
    class ServiceHeartbeat < WorkerThread
      include Contrast::Components::Logger::InstanceMethods

      # Spec recommends 30 seconds, we're going with 15.
      REFRESH_INTERVAL_SEC = 15

      def start_thread!
        return if running?

        @_thread = Contrast::Agent::Thread.new do
          logger.info('Starting heartbeat thread.')
          loop do
            Contrast::Agent.messaging_queue.send_event_eventually(poll_message)

            sleep REFRESH_INTERVAL_SEC
          end
        end
      end

      def poll_message
        @_poll_message ||= Contrast::Api::Dtm::Poll.new
      end
    end
  end
end