backup/backup

module Backup
  module Notifier
    class Zabbix < Base
      attr_accessor :zabbix_host

      attr_accessor :zabbix_port

      attr_accessor :service_name

      attr_accessor :service_host

      attr_accessor :item_key

      def initialize(model, &block)
        super
        instance_eval(&block) if block_given?

        @zabbix_host  ||= Config.hostname
        @zabbix_port  ||= 10_051
        @service_name ||= "Backup #{model.trigger}"
        @service_host ||= Config.hostname
        @item_key     ||= "backup_status"
      end

      private

      ##
      # Notify the user of the backup operation results.
      #
      # `status` indicates one of the following:
      #
      # `:success`
      # : The backup completed successfully.
      # : Notification will be sent if `on_success` is `true`.
      #
      # `:warning`
      # : The backup completed successfully, but warnings were logged.
      # : Notification will be sent if `on_warning` or `on_success` is `true`.
      #
      # `:failure`
      # : The backup operation failed.
      # : Notification will be sent if `on_warning` or `on_success` is `true`.
      #
      def notify!(status)
        send_message(message.call(model, status: status_data_for(status)))
      end

      def send_message(message)
        msg = [service_host, service_name, model.exit_status, message].join("\t")
        cmd = utility(:zabbix_sender).to_s +
          " -z '#{zabbix_host}'" \
          " -p '#{zabbix_port}'" \
          " -s #{service_host}"  \
          " -k #{item_key}"      \
          " -o '#{msg}'"
        run("echo '#{msg}' | #{cmd}")
      end
    end
  end
end