116 lines
3.0 KiB
Ruby
116 lines
3.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module AutoMerge
|
|
class BaseService < ::BaseService
|
|
include Gitlab::Utils::StrongMemoize
|
|
include MergeRequests::AssignsMergeParams
|
|
|
|
def execute(merge_request)
|
|
ActiveRecord::Base.transaction do
|
|
register_auto_merge_parameters!(merge_request)
|
|
yield if block_given?
|
|
end
|
|
|
|
notify(merge_request)
|
|
AutoMergeProcessWorker.perform_async(merge_request.id)
|
|
|
|
strategy.to_sym
|
|
rescue => e
|
|
track_exception(e, merge_request)
|
|
:failed
|
|
end
|
|
|
|
def update(merge_request)
|
|
assign_allowed_merge_params(merge_request, params.merge(auto_merge_strategy: strategy))
|
|
|
|
return :failed unless merge_request.save
|
|
|
|
strategy.to_sym
|
|
end
|
|
|
|
def cancel(merge_request)
|
|
ActiveRecord::Base.transaction do
|
|
clear_auto_merge_parameters!(merge_request)
|
|
yield if block_given?
|
|
end
|
|
|
|
success
|
|
rescue => e
|
|
track_exception(e, merge_request)
|
|
error("Can't cancel the automatic merge", 406)
|
|
end
|
|
|
|
def abort(merge_request, reason)
|
|
ActiveRecord::Base.transaction do
|
|
clear_auto_merge_parameters!(merge_request)
|
|
yield if block_given?
|
|
end
|
|
|
|
success
|
|
rescue => e
|
|
track_exception(e, merge_request)
|
|
error("Can't abort the automatic merge", 406)
|
|
end
|
|
|
|
def available_for?(merge_request)
|
|
strong_memoize("available_for_#{merge_request.id}") do
|
|
merge_request.can_be_merged_by?(current_user) &&
|
|
merge_request.mergeable_state?(skip_ci_check: true) &&
|
|
yield
|
|
end
|
|
end
|
|
|
|
##
|
|
# NOTE: This method is to be removed when `disallow_to_create_merge_request_pipelines_in_target_project`
|
|
# feature flag is removed.
|
|
def self.can_add_to_merge_train?(merge_request)
|
|
if Gitlab::Ci::Features.disallow_to_create_merge_request_pipelines_in_target_project?(merge_request.target_project)
|
|
merge_request.for_same_project?
|
|
else
|
|
true
|
|
end
|
|
end
|
|
|
|
def can_add_to_merge_train?(merge_request)
|
|
self.class.can_add_to_merge_train?(merge_request)
|
|
end
|
|
|
|
private
|
|
|
|
# Overridden in child classes
|
|
def notify(merge_request)
|
|
end
|
|
|
|
def strategy
|
|
strong_memoize(:strategy) do
|
|
self.class.name.demodulize.remove('Service').underscore
|
|
end
|
|
end
|
|
|
|
def register_auto_merge_parameters!(merge_request)
|
|
assign_allowed_merge_params(merge_request, params.merge(auto_merge_strategy: strategy))
|
|
merge_request.auto_merge_enabled = true
|
|
merge_request.merge_user = current_user
|
|
merge_request.save!
|
|
end
|
|
|
|
def clear_auto_merge_parameters!(merge_request)
|
|
merge_request.auto_merge_enabled = false
|
|
merge_request.merge_user = nil
|
|
|
|
merge_request.merge_params&.except!(
|
|
'should_remove_source_branch',
|
|
'commit_message',
|
|
'squash_commit_message',
|
|
'auto_merge_strategy'
|
|
)
|
|
|
|
merge_request.save!
|
|
end
|
|
|
|
def track_exception(error, merge_request)
|
|
Gitlab::ErrorTracking.track_exception(error, merge_request_id: merge_request&.id)
|
|
end
|
|
end
|
|
end
|