update_service.rb 2.4 KB
Newer Older
1 2 3 4
require_relative 'base_service'
require_relative 'reopen_service'
require_relative 'close_service'

5 6 7
module MergeRequests
  class UpdateService < MergeRequests::BaseService
    def execute(merge_request)
8
      # We don't allow change of source/target projects and source branch
9
      # after merge request was created
10 11
      params.except!(:source_project_id)
      params.except!(:target_project_id)
12
      params.except!(:source_branch)
13

14
      case params.delete(:state_event)
15 16 17 18 19 20
      when 'reopen'
        MergeRequests::ReopenService.new(project, current_user, {}).execute(merge_request)
      when 'close'
        MergeRequests::CloseService.new(project, current_user, {}).execute(merge_request)
      end

21 22
      params[:assignee_id]  = "" if params[:assignee_id] == IssuableFinder::NONE
      params[:milestone_id] = "" if params[:milestone_id] == IssuableFinder::NONE
23

24
      filter_params
N
Nikita Verkhovin 已提交
25 26
      old_labels = merge_request.labels.to_a

27
      if params.present? && merge_request.update_attributes(params.merge(updated_by: current_user))
28 29
        merge_request.reset_events_cache

N
Nikita Verkhovin 已提交
30 31 32 33 34 35 36 37
        if merge_request.labels != old_labels
          create_labels_note(
            merge_request,
            merge_request.labels - old_labels,
            old_labels - merge_request.labels
          )
        end

38 39 40 41
        handle_changes(merge_request)
        merge_request.create_new_cross_references!(current_user)
        execute_hooks(merge_request, 'update')
      end
42

43 44
      merge_request
    end
45

46 47 48 49 50 51
    def handle_changes(merge_request)
      if merge_request.previous_changes.include?('target_branch')
        create_branch_change_note(merge_request, 'target',
                                  merge_request.previous_changes['target_branch'].first,
                                  merge_request.target_branch)
      end
52

53 54 55
      if merge_request.previous_changes.include?('milestone_id')
        create_milestone_note(merge_request)
      end
56

57 58 59 60
      if merge_request.previous_changes.include?('assignee_id')
        create_assignee_note(merge_request)
        notification_service.reassigned_merge_request(merge_request, current_user)
      end
61

62 63
      if merge_request.previous_changes.include?('title')
        create_title_change_note(merge_request, merge_request.previous_changes['title'].first)
64 65
      end

66 67 68 69
      if merge_request.previous_changes.include?('target_branch') ||
          merge_request.previous_changes.include?('source_branch')
        merge_request.mark_as_unchecked
      end
70 71 72
    end
  end
end