module Ci class ImageForBuildService def execute(project, params) image_name = if params[:sha] commit = project.commits.find_by(sha: params[:sha]) image_for_commit(commit) elsif params[:ref] commit = project.last_commit_for_ref(params[:ref]) image_for_commit(commit) else 'build-unknown.svg' end image_path = Rails.root.join('public/ci', image_name) OpenStruct.new( path: image_path, name: image_name ) end private def image_for_commit(commit) return 'build-unknown.svg' unless commit 'build-' + commit.status + ".svg" end end end