issuable_base_service.rb 8.1 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 40 41 42 43 44 45 46
  def create_time_estimate_note(issuable)
    SystemNoteService.change_time_estimate(issuable, issuable.project, current_user)
  end

  def create_time_spent_note(issuable)
    SystemNoteService.change_time_spent(issuable, issuable.project, current_user)
  end

47 48
  def filter_params(issuable)
    ability_name = :"admin_#{issuable.to_ability_name}"
49

50
    unless can?(current_user, ability_name, project)
51
      params.delete(:milestone_id)
52
      params.delete(:labels)
53 54
      params.delete(:add_label_ids)
      params.delete(:remove_label_ids)
55 56
      params.delete(:label_ids)
      params.delete(:assignee_id)
57
      params.delete(:due_date)
58
    end
59 60 61 62

    filter_assignee(issuable)
    filter_milestone
    filter_labels
63
  end
64

65 66 67 68 69 70 71 72 73
  def filter_assignee(issuable)
    return unless params[:assignee_id].present?

    assignee_id = params[:assignee_id]

    if assignee_id.to_s == IssuableFinder::NONE
      params[:assignee_id] = ""
    else
      params.delete(:assignee_id) unless assignee_can_read?(issuable, assignee_id)
74 75 76
    end
  end

77 78 79 80 81 82 83 84 85 86 87
  def assignee_can_read?(issuable, assignee_id)
    new_assignee = User.find_by_id(assignee_id)

    return false unless new_assignee.present?

    ability_name = :"read_#{issuable.to_ability_name}"
    resource     = issuable.persisted? ? issuable : project

    can?(new_assignee, ability_name, resource)
  end

88
  def filter_milestone
89 90
    milestone_id = params[:milestone_id]
    return unless milestone_id
91

92 93
    if milestone_id == IssuableFinder::NONE ||
        project.milestones.find_by(id: milestone_id).nil?
94 95 96 97 98
      params[:milestone_id] = ''
    end
  end

  def filter_labels
99 100 101
    filter_labels_in_param(:add_label_ids)
    filter_labels_in_param(:remove_label_ids)
    filter_labels_in_param(:label_ids)
102
    find_or_create_label_ids
103 104
  end

105 106
  def filter_labels_in_param(key)
    return if params[key].to_a.empty?
107

108
    params[key] = available_labels.where(id: params[key]).pluck(:id)
109 110
  end

111 112
  def find_or_create_label_ids
    labels = params.delete(:labels)
113

114 115
    return unless labels

116
    params[:label_ids] = labels.split(",").map do |label_name|
117
      service = Labels::FindOrCreateService.new(current_user, project, title: label_name.strip)
118
      label   = service.execute
119

120 121
      label.try(:id)
    end.compact
122 123
  end

124
  def process_label_ids(attributes, existing_label_ids: nil)
125 126 127
    label_ids = attributes.delete(:label_ids)
    add_label_ids = attributes.delete(:add_label_ids)
    remove_label_ids = attributes.delete(:remove_label_ids)
128

129
    new_label_ids = existing_label_ids || label_ids || []
130

131 132 133 134 135 136
    if add_label_ids.blank? && remove_label_ids.blank?
      new_label_ids = label_ids if label_ids
    else
      new_label_ids |= add_label_ids if add_label_ids
      new_label_ids -= remove_label_ids if remove_label_ids
    end
137 138 139 140

    new_label_ids
  end

141 142 143 144
  def available_labels
    LabelsFinder.new(current_user, project_id: @project.id).execute
  end

145
  def merge_slash_commands_into_params!(issuable)
D
Douwe Maan 已提交
146 147
    description, command_params =
      SlashCommands::InterpretService.new(project, current_user).
148
        execute(params[:description], issuable)
149

150 151
    # Avoid a description already set on an issuable to be overwritten by a nil
    params[:description] = description if params.has_key?(:description)
D
Douwe Maan 已提交
152 153

    params.merge!(command_params)
154 155
  end

156
  def create_issuable(issuable, attributes, label_ids:)
157
    issuable.with_transaction_returning_status do
158 159 160 161 162
      if issuable.save
        issuable.update_attributes(label_ids: label_ids)
      end
    end
  end
163

164
  def create(issuable)
165
    merge_slash_commands_into_params!(issuable)
166
    filter_params(issuable)
167

168 169
    params.delete(:state_event)
    params[:author] ||= current_user
170

171 172 173 174 175 176 177 178
    label_ids = process_label_ids(params)

    issuable.assign_attributes(params)

    before_create(issuable)

    if params.present? && create_issuable(issuable, params, label_ids: label_ids)
      after_create(issuable)
179 180 181 182 183 184
      issuable.create_cross_references!(current_user)
      execute_hooks(issuable)
    end

    issuable
  end
185

186 187 188 189 190 191 192
  def before_create(issuable)
    # To be overridden by subclasses
  end

  def after_create(issuable)
    # To be overridden by subclasses
  end
193

194 195 196 197
  def after_update(issuable)
    # To be overridden by subclasses
  end

198 199 200
  def update_issuable(issuable, attributes)
    issuable.with_transaction_returning_status do
      issuable.update(attributes.merge(updated_by: current_user))
201
    end
202 203
  end

204 205
  def update(issuable)
    change_state(issuable)
206
    change_subscription(issuable)
207
    change_todo(issuable)
208
    filter_params(issuable)
209
    old_labels = issuable.labels.to_a
210
    old_mentioned_users = issuable.mentioned_users.to_a
211

212 213
    label_ids = process_label_ids(params, existing_label_ids: issuable.label_ids)
    params[:label_ids] = label_ids if labels_changing?(issuable.label_ids, label_ids)
214

215
    if (issuable.changed? || params.present?) && update_issuable(issuable, params)
216 217 218 219 220
      # We do not touch as it will affect a update on updated_at field
      ActiveRecord::Base.no_touching do
        handle_common_system_notes(issuable, old_labels: old_labels)
      end

221
      handle_changes(issuable, old_labels: old_labels, old_mentioned_users: old_mentioned_users)
222
      after_update(issuable)
223 224 225 226 227 228 229
      issuable.create_new_cross_references!(current_user)
      execute_hooks(issuable, 'update')
    end

    issuable
  end

230 231 232 233
  def labels_changing?(old_label_ids, new_label_ids)
    old_label_ids.sort != new_label_ids.sort
  end

234 235 236 237 238 239 240 241
  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
242

243 244 245
  def change_subscription(issuable)
    case params.delete(:subscription_event)
    when 'subscribe'
246
      issuable.subscribe(current_user, project)
247
    when 'unsubscribe'
248
      issuable.unsubscribe(current_user, project)
249 250 251
    end
  end

252 253
  def change_todo(issuable)
    case params.delete(:todo_event)
254
    when 'add'
255 256 257
      todo_service.mark_todo(issuable, current_user)
    when 'done'
      todo = TodosFinder.new(current_user).execute.find_by(target: issuable)
D
Douwe Maan 已提交
258
      todo_service.mark_todos_as_done([todo], current_user) if todo
259 260 261
    end
  end

262
  def has_changes?(issuable, old_labels: [])
263 264 265 266 267 268
    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

269
    labels_changed = issuable.labels != old_labels
270 271 272 273

    attrs_changed || labels_changed
  end

274
  def handle_common_system_notes(issuable, old_labels: [])
275 276 277 278 279 280 281
    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
282

283 284 285 286 287 288 289 290
    if issuable.previous_changes.include?('time_estimate')
      create_time_estimate_note(issuable)
    end

    if issuable.time_spent?
      create_time_spent_note(issuable)
    end

291
    create_labels_note(issuable, old_labels) if issuable.labels != old_labels
292
  end
293
end