ManageIQ/manageiq

class NotificationType < ApplicationRecord
  AUDIENCE_USER = 'user'.freeze
  AUDIENCE_GROUP = 'group'.freeze
  AUDIENCE_TENANT = 'tenant'.freeze
  AUDIENCE_GLOBAL = 'global'.freeze
  AUDIENCE_SUPERADMIN = 'superadmin'.freeze
  # don't send out notifications, but keep the template around
  AUDIENCE_NONE = 'none'.freeze
  has_many :notifications
  validates :message, :presence => true
  validates :level, :inclusion => {:in => %w[success error warning info]}
  validates :link_to, :inclusion => {:in => %w[subject initiator cause]}, :allow_blank => true
  validates :audience, :inclusion => {
    :in => [AUDIENCE_USER, AUDIENCE_GROUP, AUDIENCE_TENANT, AUDIENCE_GLOBAL, AUDIENCE_SUPERADMIN, AUDIENCE_NONE]
  }

  def subscriber_ids(subject, initiator)
    case audience
    when AUDIENCE_GLOBAL
      User.pluck(:id)
    when AUDIENCE_USER
      [initiator.id]
    when AUDIENCE_GROUP
      subject.try(:requester).try(:current_group).try(:user_ids)
    when AUDIENCE_TENANT
      if subject.respond_to?(:tenant)
        subject.tenant
      elsif initiator.kind_of?(User)
        initiator.current_tenant
      end.try(:user_ids).try(:uniq)
    when AUDIENCE_SUPERADMIN
      User.superadmins.pluck(:id)
    when AUDIENCE_NONE
      []
    end || []
  end

  # this disables notifications, but allows the notification to still exist
  # this notification template can be used for emails
  def enabled?
    audience != AUDIENCE_NONE
  end

  def self.names
    @names ||= Set.new(pluck(:name))
  end

  def self.seed
    seed_data.each do |t|
      t[:expires_in] = t[:expires_in].to_i_with_method
      rec = find_by(:name => t[:name]) || new
      rec.update!(t)
    end
  end

  def self.seed_data
    fixture_files = Vmdb::Plugins.flat_map { |plugin| plugin.root.join("content/notification_types.yml") }
    fixture_files << FIXTURE_DIR.join('notification_types.yml')

    fixture_files.select(&:exist?).flat_map { |f| YAML.load_file(f) }
  end
end