pager_tree-integrations 1.0.1 → 1.1.0

data/app/models/pager_tree/integrations/kapacitor/v3.rb

@@ -0,0 +1,65 @@
+module PagerTree::Integrations
+  class Kapacitor::V3 < Integration
+    OPTIONS = []
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      adapter_incoming_request_params.dig("id")
+    end
+
+    def adapter_action
+      adapter_incoming_request_params.dig("level") == "OK" ? :resolve : :create
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        description: _description,
+        urgency: _urgency,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums
+      )
+    end
+
+    private
+
+    def _title
+      adapter_incoming_request_params.dig("message")
+    end
+
+    def _description
+      adapter_incoming_request_params.dig("details")
+    end
+
+    def _urgency
+      case adapter_incoming_request_params.dig("level")
+      when "INFO"
+        "low"
+      when "WARN"
+        "medium"
+      when "CRITICAL"
+        "critical"
+      end
+    end
+
+    def _additional_datums
+      []
+    end
+  end
+end

data/app/models/pager_tree/integrations/logic_monitor/v3.rb

@@ -0,0 +1,79 @@
+module PagerTree::Integrations
+  class LogicMonitor::V3 < Integration
+    OPTIONS = []
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      adapter_incoming_request_params.dig("alertid")
+    end
+
+    def adapter_action
+      alert_status = adapter_incoming_request_params.dig("alertstatus")
+      case alert_status
+      when "active", "test"
+        :create
+        # when "ack"
+        # :acknowledge
+      when "clear"
+        :resolve
+      else
+        :other
+      end
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        description: _description,
+        urgency: _urgency,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums
+      )
+    end
+
+    private
+
+    def _title
+      adapter_incoming_request_params.dig("eventmsg")
+    end
+
+    def _description
+      adapter_incoming_request_params.dig("eventlogmsg")
+    end
+
+    def _urgency
+      level = adapter_incoming_request_params.dig("level").to_s.downcase
+      case level
+      when "warn", "warning"
+        "medium"
+      when "error"
+        "high"
+      when "critical"
+        "critical"
+      end
+    end
+
+    def _additional_datums
+      [
+        AdditionalDatum.new(format: "text", label: "Host", value: adapter_incoming_request_params.dig("host")),
+        AdditionalDatum.new(format: "text", label: "Service", value: adapter_incoming_request_params.dig("service"))
+      ]
+    end
+  end
+end

data/app/models/pager_tree/integrations/mattermost/outgoing_webhook/v3.rb

@@ -0,0 +1,77 @@
+module PagerTree::Integrations
+  class Mattermost::OutgoingWebhook::V3 < Integration
+    OPTIONS = [
+      {key: :token, type: :string, default: nil}
+    ]
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+      self.option_token ||= nil
+    end
+
+    def adapter_should_block_incoming?(request)
+      self.option_token.present? && (request.params["token"] != self.option_token)
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      "#{adapter_incoming_request_params.dig("channel_id")}.#{adapter_incoming_request_params.dig("timestamp")}"
+    end
+
+    def adapter_action
+      :create
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        urgency: _urgency,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums
+      )
+    end
+
+    private
+
+    def _title
+      adapter_incoming_request_params.dig("text")
+    end
+
+    def _urgency
+      text = adapter_incoming_request_params.dig("text").to_s
+      match = text.match(/low|medium|high|critical/i).try("[]", 0)
+      match&.downcase!
+
+      case match
+      when "low"
+        "low"
+      when "medium"
+        "medium"
+      when "high"
+        "high"
+      when "critical"
+        "critical"
+      end
+    end
+
+    def _additional_datums
+      [
+        AdditionalDatum.new(format: "text", label: "Trigger", value: adapter_incoming_request_params.dig("trigger_word")),
+        AdditionalDatum.new(format: "text", label: "Channel", value: adapter_incoming_request_params.dig("channel_name")),
+        AdditionalDatum.new(format: "text", label: "User", value: adapter_incoming_request_params.dig("user_name"))
+      ]
+    end
+  end
+end

data/app/models/pager_tree/integrations/new_relic/v3.rb

@@ -0,0 +1,67 @@
+module PagerTree::Integrations
+  class NewRelic::V3 < Integration
+    OPTIONS = []
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      adapter_incoming_request_params.dig("incident_id")
+    end
+
+    def adapter_action
+      event_type = adapter_incoming_request_params.dig("event_type")
+      current_state = adapter_incoming_request_params.dig("current_state")
+
+      if event_type == "INCIDENT_OPEN" || (event_type == "INCIDENT" && current_state == "open")
+        :create
+        # elsif event_type == "INCIDENT_ACKNOWLEDGED" || (event_type == "INCIDENT" && current_state == "acknowledged")
+        # :acknowledge
+      elsif event_type == "INCIDENT_RESOLVED" || (event_type == "INCIDENT" && current_state == "closed")
+        :resolve
+      else
+        :other
+      end
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        description: _description,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums
+      )
+    end
+
+    private
+
+    def _title
+      adapter_incoming_request_params.dig("condition_name")
+    end
+
+    def _description
+      adapter_incoming_request_params.dig("details")
+    end
+
+    def _additional_datums
+      [
+        AdditionalDatum.new(format: "text", label: "Account Name", value: adapter_incoming_request_params.dig("account_name")),
+        AdditionalDatum.new(format: "text", label: "Incident URL", value: adapter_incoming_request_params.dig("incident_url"))
+      ]
+    end
+  end
+end

data/app/models/pager_tree/integrations/outgoing_event.rb

@@ -7,6 +7,7 @@ module PagerTree::Integrations
     attr_accessor :event_name
     attr_accessor :item
     attr_accessor :changes
+    attr_accessor :outgoing_rules_data
 
     define_model_callbacks :initialize
 
@@ -20,6 +21,7 @@ module PagerTree::Integrations
       self.event_name ||= nil
       self.item ||= nil
       self.changes ||= nil
+      self.outgoing_rules_data ||= {}
     end
   end
 end

data/app/models/pager_tree/integrations/outgoing_webhook/v3.rb

@@ -13,7 +13,8 @@ module PagerTree::Integrations
       {key: :alert_dropped, type: :boolean, default: false},
       {key: :alert_handoff, type: :boolean, default: false},
       {key: :template, type: :string, default: nil},
-      {key: :send_linked, type: :boolean, default: false}
+      {key: :send_linked, type: :boolean, default: false},
+      {key: :outgoing_rules, type: :string, default: nil}
     ]
     store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
 
@@ -39,6 +40,7 @@ module PagerTree::Integrations
       self.option_alert_handoff ||= false
       self.option_send_linked ||= false
       self.option_template ||= ""
+      self.option_outgoing_rules ||= ""
     end
 
     def adapter_supports_outgoing?
@@ -49,20 +51,49 @@ module PagerTree::Integrations
       true
     end
 
+    def adapter_supports_title_template?
+      false
+    end
+
+    def adapter_supports_description_template?
+      false
+    end
+
     def adapter_outgoing_interest?(event_name)
       try("option_#{event_name}") || false
     end
 
     def adapter_process_outgoing
-      body = {
-        data: adapter_outgoing_event.item,
-        type: adapter_outgoing_event.event_name
+      body = nil
+      event_type = adapter_outgoing_event.event_name.to_s.tr("_", ".")
+
+      # Do the custom templating portion for outgoing hooks
+      if self.option_template.present?
+        begin
+          body = JSON.parse(handlebars(self.option_template, {
+            alert: adapter_outgoing_event.item.try(:v3_format) || adapter_outgoing_event.item,
+            event: {
+              type: event_type
+            }
+          }))
+        rescue JSON::ParserError => e
+          logs.create(message: "Error parsing JSON, abort custom format for option template. Error: #{e.message}")
+        rescue => e
+          Rails.logger.error "Error while processing option_template for #{id}: #{e.message}"
+        end
+      end
+
+      body ||= {
+        data: adapter_outgoing_event.item.try(:v3_format) || adapter_outgoing_event.item,
+        type: event_type
       }
 
-      # create the delivery, save it, and send it later
+      url = adapter_outgoing_event.outgoing_rules_data.dig("webhook_url") || option_webhook_url
+      body.merge!(adapter_outgoing_event.outgoing_rules_data.except("webhook_url"))
+
       outgoing_webhook_delivery = OutgoingWebhookDelivery.factory(
         resource: self,
-        url: option_webhook_url,
+        url: url,
         auth: {username: option_username, password: option_password},
         body: body
       )

data/app/models/pager_tree/integrations/outgoing_webhook_delivery/hook_relay.rb

@@ -40,8 +40,8 @@ module PagerTree::Integrations
           pagertree_options = {
             headers: {
               Accept: "*/*",
-              'User-Agent': "pagertree outgoing webhook service; ref: #{resource&.id}; report: support@pagertree.com",
-              'Content-Type': "application/json"
+              "User-Agent": "pagertree outgoing webhook service; ref: #{resource&.id}; report: support@pagertree.com",
+              "Content-Type": "application/json"
             }
           }
 
@@ -77,18 +77,20 @@ module PagerTree::Integrations
 
       options = {
         headers: {
-          'Content-Type': "application/json",
+          "Content-Type": "application/json",
           Authorization: "Bearer #{OutgoingWebhookDelivery::HookRelay.hook_relay_api_key}"
         },
         timeout: 15
       }
 
-      @delivery = ::HTTParty.get(hook_relay_delivery_url, **options)
-      @delivery
+      response = ::HTTParty.get(hook_relay_delivery_url, **options)
+      @delivery = response.parsed_response if response.success?
+
+      @delivery || {}
     end
 
     def request
-      delivery&.dig("request")
+      delivery&.dig("request") || {}
     end
 
     def responses

data/app/models/pager_tree/integrations/outgoing_webhook_delivery.rb

@@ -8,7 +8,7 @@ module PagerTree::Integrations
     store_accessor :data, *[:url, :body, :auth].map(&:to_s)
 
     HTTP_OPTIONS = {
-      headers: {'Content-Type': "application/json"},
+      headers: {"Content-Type": "application/json"},
       timeout: 15
     }
 

data/app/models/pager_tree/integrations/pingdom/v3.rb

@@ -0,0 +1,71 @@
+module PagerTree::Integrations
+  class Pingdom::V3 < Integration
+    OPTIONS = []
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      adapter_incoming_request_params.dig("check_id")
+    end
+
+    def adapter_action
+      previous_state = adapter_incoming_request_params.dig("previous_state")
+      current_state = adapter_incoming_request_params.dig("current_state")
+
+      if (previous_state == "UP" && current_state == "DOWN") || (previous_state == "SUCCESS" && current_state == "FAILING")
+        :create
+      elsif (previous_state == "DOWN" && current_state == "UP") || (previous_state == "FAILING" && current_state == "SUCCESS")
+        :resolve
+      else
+        :other
+      end
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        description: _description,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums,
+        tags: _tags
+      )
+    end
+
+    private
+
+    def _title
+      [adapter_incoming_request_params.dig("check_name"), adapter_incoming_request_params.dig("current_state")].join(" ")
+    end
+
+    def _description
+      adapter_incoming_request_params.dig("description")
+    end
+
+    def _tags
+      adapter_incoming_request_params.dig("tags") || []
+    end
+
+    def _additional_datums
+      [
+        AdditionalDatum.new(format: "link", label: "Full URL", value: adapter_incoming_request_params.dig("check_params", "full_url")),
+        AdditionalDatum.new(format: "text", label: "Importance Level", value: adapter_incoming_request_params.dig("importance_level")),
+        AdditionalDatum.new(format: "text", label: "Custom Message", value: adapter_incoming_request_params.dig("custom_message"))
+      ]
+    end
+  end
+end

data/app/models/pager_tree/integrations/prometheus/v3.rb

@@ -0,0 +1,79 @@
+module PagerTree::Integrations
+  class Prometheus::V3 < Integration
+    OPTIONS = []
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      adapter_incoming_request_params.dig("groupKey")
+    end
+
+    def adapter_action
+      case adapter_incoming_request_params.dig("status")
+      when "firing"
+        :create
+      when "resolved"
+        :resolve
+      else
+        :other
+      end
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        description: _description,
+        urgency: _urgency,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums
+      )
+    end
+
+    private
+
+    def _title
+      adapter_incoming_request_params.dig("commonLabels", "alertname") || ["Prometheus", adapter_incoming_request_params.dig("receiver"), "firing"].compact_blank.join(" ")
+    end
+
+    def _description
+      adapter_incoming_request_params.dig("commonAnnotations", "summary")
+    end
+
+    def _urgency
+      case adapter_incoming_request_params.dig("commonLabels", "severity")
+      when "low"
+        "low"
+      when "medium"
+        "medium"
+      when "high"
+        "high"
+      when "critical"
+        "critical"
+      end
+    end
+
+    def _additional_datums
+      [
+        AdditionalDatum.new(format: "link", label: "Alert Manager URL", value: adapter_incoming_request_params.dig("externalURL")),
+        AdditionalDatum.new(format: "text", label: "Receiver", value: adapter_incoming_request_params.dig("receiver"))
+      ] + adapter_incoming_request_params.dig("commonLabels").map do |key, value|
+        AdditionalDatum.new(format: "text", label: key, value: value.to_s)
+      end
+    end
+  end
+end

data/app/models/pager_tree/integrations/prtg/v3.rb

@@ -0,0 +1,58 @@
+module PagerTree::Integrations
+  class Prtg::V3 < Integration
+    OPTIONS = []
+    store_accessor :options, *OPTIONS.map { |x| x[:key] }.map(&:to_s), prefix: "option"
+
+    after_initialize do
+    end
+
+    def adapter_supports_incoming?
+      true
+    end
+
+    def adapter_supports_outgoing?
+      false
+    end
+
+    def adapter_incoming_can_defer?
+      true
+    end
+
+    def adapter_thirdparty_id
+      adapter_incoming_request_params.dig("sensorid")
+    end
+
+    def adapter_action
+      status = adapter_incoming_request_params.dig("status").to_s.downcase
+
+      if status == "down"
+        :create
+      elsif status.include?("now: up")
+        :resolve
+      else
+        :other
+      end
+    end
+
+    def adapter_process_create
+      Alert.new(
+        title: _title,
+        thirdparty_id: adapter_thirdparty_id,
+        dedup_keys: [adapter_thirdparty_id],
+        additional_data: _additional_datums
+      )
+    end
+
+    private
+
+    def _title
+      "Sensor #{adapter_incoming_request_params.dig("sensorid")} #{adapter_incoming_request_params.dig("sensor")} is DOWN"
+    end
+
+    def _additional_datums
+      adapter_incoming_request_params.map do |key, value|
+        AdditionalDatum.new(format: "text", label: key, value: value.to_s)
+      end
+    end
+  end
+end