rosa-build/app/models/advisory.rb

class Advisory < ActiveRecord::Base
  self.include_root_in_json = false

  has_and_belongs_to_many :platforms
  has_and_belongs_to_many :projects
  has_many :build_lists

  validates :description, :update_type, presence: true
  validates :update_type, inclusion: BuildList::RELEASE_UPDATE_TYPES

  after_create :generate_advisory_id
  before_save  :normalize_references, if: :references_changed?

  attr_accessible :description, :references

  ID_TEMPLATE        = 'ROSA-%<type>s-%<year>d:%<id>04d'
  ID_STRING_TEMPLATE = 'ROSA-%<type>s-%<year>04s:%<id>04s'
  TYPES = {'security' => 'SA', 'bugfix' => 'A'}

  scope :search, ->(q) {
    q = q.to_s.strip
    where("#{table_name}.advisory_id ILIKE :q OR #{table_name}.description ILIKE :q OR build_list_packages.fullname ILIKE :q", q: "%#{q}%").
      joins(build_lists: :packages) if q.present?
  }
  scope :search_by_id,   ->(aid) { where("#{table_name}.advisory_id ILIKE ?", "%#{aid.to_s.strip}%") }
  scope :by_update_type, ->(ut) { where(update_type: ut) }
  default_scope { order(created_at: :desc) }

  def to_param
    advisory_id
  end

  def attach_build_list(build_list)
    return false if update_type != build_list.update_type
    self.platforms  << build_list.save_to_platform unless platforms.include? build_list.save_to_platform
    self.projects   << build_list.project unless projects.include? build_list.project
    build_list.advisory = self
    save && build_list.save
  end

  # this method fetches and structurize packages attached to current advisory.
  def fetch_packages_info
    packages_info = Hash.new { |h, k| h[k] = {} } # maaagic, it's maaagic ;)
    build_lists.includes(:save_to_platform, :packages, :project).find_in_batches do |batch|
      batch.each do |build_list|
        tmp = build_list.packages.inject({srpm: nil, rpm: []}) do |h, p|
          p.package_type == 'binary' ? h[:rpm] << p.fullname : h[:srpm] = p.fullname
          h
        end
        h = { build_list.project => tmp }
        packages_info[build_list.save_to_platform].merge!(h) do |pr, old, new|
          {srpm: new[:srpm], rpm: old[:rpm].concat(new[:rpm]).uniq}
        end
      end
    end
    packages_info
  end

  protected

  def generate_advisory_id
    self.advisory_id = sprintf(ID_TEMPLATE, type: TYPES[self.update_type], year: Time.now.utc.year, id: self.id)
    self.save
  end

  def normalize_references
    self.references.gsub!(/\r| /, '')
    self.references = self.references.split('\n').map do |ref|
      ref = CGI::escapeHTML(ref)
      ref = "http://#{ref}" unless ref =~ %r[^http(s?)://*]
      ref
    end.join("\n")
  end

end