debian-mirror-gitlab/app/services/issuable/clone/attributes_rewriter.rb

102 lines
3.2 KiB
Ruby
Raw Normal View History

2019-02-15 15:39:39 +05:30
# frozen_string_literal: true
module Issuable
module Clone
class AttributesRewriter < ::Issuable::Clone::BaseService
def initialize(current_user, original_entity, new_entity)
@current_user = current_user
@original_entity = original_entity
@new_entity = new_entity
end
def execute
2020-01-01 13:55:28 +05:30
update_attributes = { labels: cloneable_labels }
2020-03-09 13:42:32 +05:30
milestone = matching_milestone(original_entity.milestone&.title)
2020-01-01 13:55:28 +05:30
update_attributes[:milestone] = milestone if milestone.present?
new_entity.update(update_attributes)
2019-02-15 15:39:39 +05:30
copy_resource_label_events
2020-03-09 13:42:32 +05:30
copy_resource_weight_events
2020-04-08 14:13:33 +05:30
copy_resource_milestone_events
2019-02-15 15:39:39 +05:30
end
private
2020-03-09 13:42:32 +05:30
def matching_milestone(title)
return if title.blank? || !new_entity.supports_milestone?
2019-02-15 15:39:39 +05:30
params = { title: title, project_ids: new_entity.project&.id, group_ids: group&.id }
milestones = MilestonesFinder.new(params).execute
milestones.first
end
def cloneable_labels
params = {
project_id: new_entity.project&.id,
group_id: group&.id,
title: original_entity.labels.select(:title),
include_ancestor_groups: true
}
params[:only_group_labels] = true if new_parent.is_a?(Group)
LabelsFinder.new(current_user, params).execute
end
def copy_resource_label_events
2020-03-09 13:42:32 +05:30
entity_key = new_entity.class.name.underscore.foreign_key
copy_events(ResourceLabelEvent.table_name, original_entity.resource_label_events) do |event|
event.attributes
.except('id', 'reference', 'reference_html')
.merge(entity_key => new_entity.id, 'action' => ResourceLabelEvent.actions[event.action])
end
end
def copy_resource_weight_events
return unless original_entity.respond_to?(:resource_weight_events)
copy_events(ResourceWeightEvent.table_name, original_entity.resource_weight_events) do |event|
event.attributes
.except('id', 'reference', 'reference_html')
.merge('issue_id' => new_entity.id)
end
end
2020-04-08 14:13:33 +05:30
def copy_resource_milestone_events
entity_key = new_entity.class.name.underscore.foreign_key
copy_events(ResourceMilestoneEvent.table_name, original_entity.resource_milestone_events) do |event|
matching_destination_milestone = matching_milestone(event.milestone.title)
if matching_destination_milestone.present?
event.attributes
.except('id', 'reference', 'reference_html')
.merge(entity_key => new_entity.id,
'milestone_id' => matching_destination_milestone.id,
'action' => ResourceMilestoneEvent.actions[event.action],
'state' => ResourceMilestoneEvent.states[event.state])
end
end
end
2020-03-09 13:42:32 +05:30
def copy_events(table_name, events_to_copy)
events_to_copy.find_in_batches do |batch|
2019-02-15 15:39:39 +05:30
events = batch.map do |event|
2020-03-09 13:42:32 +05:30
yield(event)
end.compact
2019-02-15 15:39:39 +05:30
2020-03-09 13:42:32 +05:30
Gitlab::Database.bulk_insert(table_name, events)
2019-02-15 15:39:39 +05:30
end
end
def entity_key
new_entity.class.name.parameterize('_').foreign_key
end
end
end
end