issuable_base_service.rb 5.9 KB
Newer Older
1 2 3 4
class IssuableBaseService < BaseService
  private

  def create_assignee_note(issuable)
5
    SystemNoteService.change_assignee(
6 7 8 9
      issuable, issuable.project, current_user, issuable.assignee)
  end

  def create_milestone_note(issuable)
10
    SystemNoteService.change_milestone(
11 12
      issuable, issuable.project, current_user, issuable.milestone)
  end
N
Nikita Verkhovin 已提交
13

14 15 16 17
  def create_labels_note(issuable, old_labels)
    added_labels = issuable.labels - old_labels
    removed_labels = old_labels - issuable.labels

18
    SystemNoteService.change_label(
N
Nikita Verkhovin 已提交
19 20
      issuable, issuable.project, current_user, added_labels, removed_labels)
  end
21 22 23 24 25

  def create_title_change_note(issuable, old_title)
    SystemNoteService.change_title(
      issuable, issuable.project, current_user, old_title)
  end
26 27 28 29 30 31

  def create_branch_change_note(issuable, branch_type, old_branch, new_branch)
    SystemNoteService.change_branch(
      issuable, issuable.project, current_user, branch_type,
      old_branch, new_branch)
  end
32

33 34 35 36 37 38
  def create_task_status_note(issuable)
    issuable.updated_tasks.each do |task|
      SystemNoteService.change_task_status(issuable, issuable.project, current_user, task)
    end
  end

39
  def filter_params(issuable_ability_name = :issue)
40 41 42
    filter_assignee
    filter_milestone
    filter_labels
43

44 45 46
    ability = :"admin_#{issuable_ability_name}"

    unless can?(current_user, ability, project)
47
      params.delete(:milestone_id)
48 49
      params.delete(:add_label_ids)
      params.delete(:remove_label_ids)
50 51 52 53
      params.delete(:label_ids)
      params.delete(:assignee_id)
    end
  end
54

55 56 57 58 59 60 61
  def filter_assignee
    if params[:assignee_id] == IssuableFinder::NONE
      params[:assignee_id] = ''
    end
  end

  def filter_milestone
62 63
    milestone_id = params[:milestone_id]
    return unless milestone_id
64

65 66
    if milestone_id == IssuableFinder::NONE ||
        project.milestones.find_by(id: milestone_id).nil?
67 68 69 70 71
      params[:milestone_id] = ''
    end
  end

  def filter_labels
72 73 74
    filter_labels_in_param(:add_label_ids)
    filter_labels_in_param(:remove_label_ids)
    filter_labels_in_param(:label_ids)
75 76
  end

77 78
  def filter_labels_in_param(key)
    return if params[key].to_a.empty?
79

80
    params[key] = project.labels.where(id: params[key]).pluck(:id)
81 82
  end

83
  def process_label_ids(attributes, base_label_ids: [], merge_all: false)
84 85 86
    label_ids = attributes.delete(:label_ids)
    add_label_ids = attributes.delete(:add_label_ids)
    remove_label_ids = attributes.delete(:remove_label_ids)
87 88

    new_label_ids = base_label_ids
R
Rémy Coutable 已提交
89
    new_label_ids = label_ids if label_ids && (merge_all || (add_label_ids.blank? && remove_label_ids.blank?))
90 91
    new_label_ids |= add_label_ids if add_label_ids
    new_label_ids -= remove_label_ids if remove_label_ids
92 93 94 95

    new_label_ids
  end

96
  def merge_slash_commands_into_params!(issuable)
97
    command_params = SlashCommands::InterpretService.new(project, current_user).
98
      execute(params[:description], issuable)
99 100 101 102 103

    params.merge!(command_params)
  end

  def create_issuable(issuable, attributes)
104
    issuable.with_transaction_returning_status do
105 106 107 108 109 110 111 112 113 114 115
      attributes.delete(:state_event)
      params[:author] ||= current_user
      label_ids = process_label_ids(attributes, merge_all: true)

      issuable.assign_attributes(attributes)

      if issuable.save
        issuable.update_attributes(label_ids: label_ids)
      end
    end
  end
116

117
  def create(issuable)
118
    merge_slash_commands_into_params!(issuable)
119
    filter_params
120

121 122 123 124 125 126 127 128
    if params.present? && create_issuable(issuable, params)
      handle_creation(issuable)
      issuable.create_cross_references!(current_user)
      execute_hooks(issuable)
    end

    issuable
  end
129

130 131 132 133 134
  def update_issuable(issuable, attributes)
    issuable.with_transaction_returning_status do
      attributes[:label_ids] = process_label_ids(attributes, base_label_ids: issuable.label_ids)

      issuable.update(attributes.merge(updated_by: current_user))
135
    end
136 137
  end

138 139
  def update(issuable)
    change_state(issuable)
140
    change_subscription(issuable)
141
    change_todo(issuable)
142 143 144
    filter_params
    old_labels = issuable.labels.to_a

145
    if params.present? && update_issuable(issuable, params)
146
      issuable.reset_events_cache
147
      handle_common_system_notes(issuable, old_labels: old_labels)
148
      handle_changes(issuable, old_labels: old_labels)
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163
      issuable.create_new_cross_references!(current_user)
      execute_hooks(issuable, 'update')
    end

    issuable
  end

  def change_state(issuable)
    case params.delete(:state_event)
    when 'reopen'
      reopen_service.new(project, current_user, {}).execute(issuable)
    when 'close'
      close_service.new(project, current_user, {}).execute(issuable)
    end
  end
164

165 166 167 168 169 170 171 172 173
  def change_subscription(issuable)
    case params.delete(:subscription_event)
    when 'subscribe'
      issuable.subscribe(current_user)
    when 'unsubscribe'
      issuable.unsubscribe(current_user)
    end
  end

174 175 176 177 178 179 180 181 182 183 184 185
  def change_todo(issuable)
    case params.delete(:todo_event)
    when 'mark'
      todo_service.mark_todo(issuable, current_user)
    when 'done'
      todo = TodosFinder.new(current_user).execute.find_by(target: issuable)
      if todo
        todo_service.mark_todos_as_done([todo], current_user)
      end
    end
  end

186
  def has_changes?(issuable, old_labels: [])
187 188 189 190 191 192
    valid_attrs = [:title, :description, :assignee_id, :milestone_id, :target_branch]

    attrs_changed = valid_attrs.any? do |attr|
      issuable.previous_changes.include?(attr.to_s)
    end

193
    labels_changed = issuable.labels != old_labels
194 195 196 197

    attrs_changed || labels_changed
  end

198
  def handle_common_system_notes(issuable, old_labels: [])
199 200 201 202 203 204 205
    if issuable.previous_changes.include?('title')
      create_title_change_note(issuable, issuable.previous_changes['title'].first)
    end

    if issuable.previous_changes.include?('description') && issuable.tasks?
      create_task_status_note(issuable)
    end
206

207
    create_labels_note(issuable, old_labels) if issuable.labels != old_labels
208
  end
209
end