122 lines
4.1 KiB
Ruby
122 lines
4.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Ci
|
|
# A state object to centralize logic related to merge request pipelines
|
|
class PipelinesForMergeRequestFinder
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
def initialize(merge_request, current_user)
|
|
@merge_request = merge_request
|
|
@current_user = current_user
|
|
end
|
|
|
|
attr_reader :merge_request, :current_user
|
|
|
|
delegate :commit_shas, :target_project, :source_project, :source_branch, to: :merge_request
|
|
|
|
# Fetch all pipelines that the user can read.
|
|
def execute
|
|
if can_read_pipeline_in_target_project? && can_read_pipeline_in_source_project?
|
|
all
|
|
elsif can_read_pipeline_in_source_project?
|
|
all.for_project(merge_request.source_project)
|
|
elsif can_read_pipeline_in_target_project?
|
|
all.for_project(merge_request.target_project)
|
|
else
|
|
Ci::Pipeline.none
|
|
end
|
|
end
|
|
|
|
# Fetch all pipelines without permission check.
|
|
def all
|
|
strong_memoize(:all_pipelines) do
|
|
next Ci::Pipeline.none unless source_project
|
|
|
|
pipelines =
|
|
if merge_request.persisted?
|
|
if Feature.enabled?(:ci_pipelines_for_merge_request_finder_new_cte, target_project)
|
|
pipelines_using_cte
|
|
else
|
|
pipelines_using_legacy_cte
|
|
end
|
|
else
|
|
triggered_for_branch.for_sha(commit_shas)
|
|
end
|
|
|
|
sort(pipelines)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def pipelines_using_legacy_cte
|
|
cte = Gitlab::SQL::CTE.new(:shas, merge_request.all_commits.select(:sha))
|
|
|
|
source_sha_join = cte.table[:sha].eq(Ci::Pipeline.arel_table[:source_sha])
|
|
merged_result_pipelines = filter_by(triggered_by_merge_request, cte, source_sha_join)
|
|
detached_merge_request_pipelines = filter_by_sha(triggered_by_merge_request, cte)
|
|
pipelines_for_branch = filter_by_sha(triggered_for_branch, cte)
|
|
|
|
Ci::Pipeline.with(cte.to_arel) # rubocop: disable CodeReuse/ActiveRecord
|
|
.from_union([merged_result_pipelines, detached_merge_request_pipelines, pipelines_for_branch])
|
|
end
|
|
|
|
def pipelines_using_cte
|
|
cte = Gitlab::SQL::CTE.new(:shas, merge_request.all_commits.select(:sha))
|
|
|
|
pipelines_for_merge_requests = triggered_by_merge_request
|
|
pipelines_for_branch = filter_by_sha(triggered_for_branch, cte)
|
|
|
|
Ci::Pipeline.with(cte.to_arel) # rubocop: disable CodeReuse/ActiveRecord
|
|
.from_union([pipelines_for_merge_requests, pipelines_for_branch])
|
|
end
|
|
|
|
def filter_by_sha(pipelines, cte)
|
|
hex = Arel::Nodes::SqlLiteral.new("'hex'")
|
|
string_sha = Arel::Nodes::NamedFunction.new('encode', [cte.table[:sha], hex])
|
|
join_condition = string_sha.eq(Ci::Pipeline.arel_table[:sha])
|
|
|
|
filter_by(pipelines, cte, join_condition)
|
|
end
|
|
|
|
def filter_by(pipelines, cte, join_condition)
|
|
shas_table =
|
|
Ci::Pipeline.arel_table
|
|
.join(cte.table, Arel::Nodes::InnerJoin)
|
|
.on(join_condition)
|
|
.join_sources
|
|
|
|
pipelines.joins(shas_table) # rubocop: disable CodeReuse/ActiveRecord
|
|
end
|
|
|
|
# NOTE: this method returns only parent merge request pipelines.
|
|
# Child merge request pipelines have a different source.
|
|
def triggered_by_merge_request
|
|
Ci::Pipeline.triggered_by_merge_request(merge_request)
|
|
end
|
|
|
|
def triggered_for_branch
|
|
source_project.all_pipelines.ci_branch_sources.for_branch(source_branch)
|
|
end
|
|
|
|
def sort(pipelines)
|
|
sql = 'CASE ci_pipelines.source WHEN (?) THEN 0 ELSE 1 END, ci_pipelines.id DESC'
|
|
query = ApplicationRecord.send(:sanitize_sql_array, [sql, Ci::Pipeline.sources[:merge_request_event]]) # rubocop:disable GitlabSecurity/PublicSend
|
|
|
|
pipelines.order(Arel.sql(query)) # rubocop: disable CodeReuse/ActiveRecord
|
|
end
|
|
|
|
def can_read_pipeline_in_target_project?
|
|
strong_memoize(:can_read_pipeline_in_target_project) do
|
|
Ability.allowed?(current_user, :read_pipeline, target_project)
|
|
end
|
|
end
|
|
|
|
def can_read_pipeline_in_source_project?
|
|
strong_memoize(:can_read_pipeline_in_source_project) do
|
|
Ability.allowed?(current_user, :read_pipeline, source_project)
|
|
end
|
|
end
|
|
end
|
|
end
|