debian-mirror-gitlab/app/services/ci/image_for_build_service.rb

module Ci
  class ImageForBuildService
    def execute(project, opts)
      sha = opts[:sha] || ref_sha(project, opts[:ref])

      ci_commits = project.ci_commits.where(sha: sha)
      ci_commits = ci_commits.where(ref: opts[:ref]) if opts[:ref]
      image_name = image_for_status(ci_commits.status)

      image_path = Rails.root.join('public/ci', image_name)
      OpenStruct.new(path: image_path, name: image_name)
    end

    private

    def ref_sha(project, ref)
      project.commit(ref).try(:sha) if ref
    end

    def image_for_status(status)
      status ||= 'unknown'
      'build-' + status + ".svg"
    end
  end
end