module MyNagios
  class Check < ActiveRecord::Base
    belongs_to :group

    enum status:  [ :info, :success, :critical ]
    enum state:   [ :completed, :running ]

    scope :enabled,     -> { where(enabled: true) }
    scope :not_snoozed, -> { where('`my_nagios_checks`.`snooze_for` <= UTC_TIMESTAMP() OR `my_nagios_checks`.`snooze_for` IS NULL') }

    attr_accessor :additional_command_result

    INFO_STATES  = ['AuthenticationFailed message', 'No such file or directory']
    ERROR_STATES = ['!ruby/exception', 'packet_write_wait: Connection to UNKNOWN: Broken pipe']

    def run!
      begin
        self.update(state: :running)

        Net::SSH.start( self.host, self.user, MyNagios::Check.ssh_config(self.pem_key) ) do |ssh|
          result = ssh.exec! self.command
          status = self.determinate_status_by_response(result)

          self.additional_command_result = ssh.exec!(self.additional_command) if not self.additional_command.blank? and status.eql?(:critical)

          self.update(state: :completed, status: status, latest_state: result, latest_updated_at: Time.now, retry_count: 0)
        end
      rescue => e
        self.update(state: :completed, status: :info, latest_state: e, latest_updated_at: Time.now, retry_count: (self.retry_count + 1))
      end
    end

    def self.multiple_run!(check_ids, config)
      check_list = MyNagios::Check.where(id: check_ids)

      begin
        check_list.update_all(state: :running)

        Net::SSH.start( config['host'], config['user'], MyNagios::Check.ssh_config(config['pem_key']) ) do |ssh|
          check_list.each do |check|
            result = ssh.exec! check.command
            status = check.determinate_status_by_response(result)

            check.additional_command_result = ssh.exec!(check.additional_command) if not check.additional_command.blank? and status.eql?(:critical)

            check.update(state: :completed, status: status, latest_state: result, latest_updated_at: Time.now, retry_count: 0)
          end
        end

      rescue => e
        check_list.each { |check| check.update(state: :completed, status: :info, latest_state: e, latest_updated_at: Time.now, retry_count: (check.retry_count + 1)) }
      end
    end

    def determinate_status_by_response(response)
      return :critical if not regexp.blank? and response =~ /#{regexp}/
      return :critical if not response.scan('CRITICAL').blank?
      return :info     if response.nil? or MyNagios::Check::INFO_STATES.any? { |msg| response.include?(msg) }
      :success
    end

    def self.ssh_config(pem_key = nil)
      config = { config: true, non_interactive: true, auth_methods: ['publickey', 'hostbased'] }
      config[:keys] = [pem_key] unless pem_key.blank?

      config
    end

  end
end