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

require 'contrast/components/logger'

module Contrast
  module Utils
    # A module that detects whether any job servers attached to
    # the application are running
    module JobServersRunning
      extend Contrast::Components::Logger::InstanceMethods

      class << self
        def job_servers_running?
          sidekiq_running? || rake_running?
        end

        private

        def sidekiq_running?
          return unless defined?(Sidekiq) && Sidekiq.cs__respond_to?(:server?) && Sidekiq.server?

          logger.trace('Detected the spawn of a Sidekiq process')
          true
        end

        def rake_running?
          unless defined?(Rake) &&
                Rake.cs__respond_to?(:application) &&
                Rake.application.cs__respond_to?(:top_level_tasks)

            return
          end

          disabled_rake_tasks = ::Contrast::APP_CONTEXT.disabled_agent_rake_tasks
          has_disabled_task = Rake.application.top_level_tasks.any? do |top_level_task|
            disabled_rake_tasks.include?(top_level_task)
          end
          return false unless has_disabled_task

          logger.trace('Detected startup within Rake task')
          true
        end
      end
    end
  end
end