1 files changed, 146 insertions, 0 deletions
diff --git a/app/models/alert_management/alert.rb b/app/models/alert_management/alert.rb
new file mode 100644
index 00000000000..acaf474ecc2
--- /dev/null
+++ b/app/models/alert_management/alert.rb
@@ -0,0 +1,146 @@
+# frozen_string_literal: true
+
+module AlertManagement
+  class Alert < ApplicationRecord
+    include AtomicInternalId
+    include ShaAttribute
+    include Sortable
+    include Gitlab::SQL::Pattern
+
+    STATUSES = {
+      triggered: 0,
+      acknowledged: 1,
+      resolved: 2,
+      ignored: 3
+    }.freeze
+
+    STATUS_EVENTS = {
+      triggered: :trigger,
+      acknowledged: :acknowledge,
+      resolved: :resolve,
+      ignored: :ignore
+    }.freeze
+
+    belongs_to :project
+    belongs_to :issue, optional: true
+    has_internal_id :iid, scope: :project, init: ->(s) { s.project.alert_management_alerts.maximum(:iid) }
+
+    self.table_name = 'alert_management_alerts'
+
+    sha_attribute :fingerprint
+
+    HOSTS_MAX_LENGTH = 255
+
+    validates :title,           length: { maximum: 200 }, presence: true
+    validates :description,     length: { maximum: 1_000 }
+    validates :service,         length: { maximum: 100 }
+    validates :monitoring_tool, length: { maximum: 100 }
+    validates :project,         presence: true
+    validates :events,          presence: true
+    validates :severity,        presence: true
+    validates :status,          presence: true
+    validates :started_at,      presence: true
+    validates :fingerprint,     uniqueness: { scope: :project }, allow_blank: true
+    validate  :hosts_length
+
+    enum severity: {
+      critical: 0,
+      high: 1,
+      medium: 2,
+      low: 3,
+      info: 4,
+      unknown: 5
+    }
+
+    state_machine :status, initial: :triggered do
+      state :triggered, value: STATUSES[:triggered]
+
+      state :acknowledged, value: STATUSES[:acknowledged]
+
+      state :resolved, value: STATUSES[:resolved] do
+        validates :ended_at, presence: true
+      end
+
+      state :ignored, value: STATUSES[:ignored]
+
+      state :triggered, :acknowledged, :ignored do
+        validates :ended_at, absence: true
+      end
+
+      event :trigger do
+        transition any => :triggered
+      end
+
+      event :acknowledge do
+        transition any => :acknowledged
+      end
+
+      event :resolve do
+        transition any => :resolved
+      end
+
+      event :ignore do
+        transition any => :ignored
+      end
+
+      before_transition to: [:triggered, :acknowledged, :ignored] do |alert, _transition|
+        alert.ended_at = nil
+      end
+
+      before_transition to: :resolved do |alert, transition|
+        ended_at = transition.args.first
+        alert.ended_at = ended_at || Time.current
+      end
+    end
+
+    delegate :iid, to: :issue, prefix: true, allow_nil: true
+
+    scope :for_iid, -> (iid) { where(iid: iid) }
+    scope :for_status, -> (status) { where(status: status) }
+    scope :for_fingerprint, -> (project, fingerprint) { where(project: project, fingerprint: fingerprint) }
+    scope :search, -> (query) { fuzzy_search(query, [:title, :description, :monitoring_tool, :service]) }
+
+    scope :order_start_time,    -> (sort_order) { order(started_at: sort_order) }
+    scope :order_end_time,      -> (sort_order) { order(ended_at: sort_order) }
+    scope :order_events_count,  -> (sort_order) { order(events: sort_order) }
+    scope :order_severity,      -> (sort_order) { order(severity: sort_order) }
+    scope :order_status,        -> (sort_order) { order(status: sort_order) }
+
+    scope :counts_by_status, -> { group(:status).count }
+
+    def self.sort_by_attribute(method)
+      case method.to_s
+      when 'start_time_asc'     then order_start_time(:asc)
+      when 'start_time_desc'    then order_start_time(:desc)
+      when 'end_time_asc'       then order_end_time(:asc)
+      when 'end_time_desc'      then order_end_time(:desc)
+      when 'events_count_asc'   then order_events_count(:asc)
+      when 'events_count_desc'  then order_events_count(:desc)
+      when 'severity_asc'       then order_severity(:asc)
+      when 'severity_desc'      then order_severity(:desc)
+      when 'status_asc'         then order_status(:asc)
+      when 'status_desc'        then order_status(:desc)
+      else
+        order_by(method)
+      end
+    end
+
+    def details
+      details_payload = payload.except(*attributes.keys)
+
+      Gitlab::Utils::InlineHash.merge_keys(details_payload)
+    end
+
+    def prometheus?
+      monitoring_tool == Gitlab::AlertManagement::AlertParams::MONITORING_TOOLS[:prometheus]
+    end
+
+    private
+
+    def hosts_length
+      return unless hosts
+
+      errors.add(:hosts, "hosts array is over #{HOSTS_MAX_LENGTH} chars") if hosts.join.length > HOSTS_MAX_LENGTH
+    end
+  end
+end