adab5dba43
Signed-off-by: Rémy Coutable <remy@rymai.me>
236 lines
6.3 KiB
Ruby
236 lines
6.3 KiB
Ruby
module SlashCommands
|
|
class InterpretService < BaseService
|
|
include Gitlab::SlashCommands::Dsl
|
|
|
|
attr_reader :issuable
|
|
|
|
# Takes a text and interprets the commands that are extracted from it.
|
|
# Returns the content without commands, and hash of changes to be applied to a record.
|
|
def execute(content, issuable)
|
|
@issuable = issuable
|
|
@updates = {}
|
|
|
|
opts = {
|
|
issuable: issuable,
|
|
current_user: current_user,
|
|
project: project
|
|
}
|
|
|
|
content, commands = extractor.extract_commands(content, opts)
|
|
|
|
commands.each do |name, arg|
|
|
definition = self.class.command_definitions_by_name[name.to_sym]
|
|
next unless definition
|
|
|
|
definition.execute(self, opts, arg)
|
|
end
|
|
|
|
[content, @updates]
|
|
end
|
|
|
|
private
|
|
|
|
def extractor
|
|
Gitlab::SlashCommands::Extractor.new(self.class.command_definitions)
|
|
end
|
|
|
|
desc do
|
|
"Close this #{issuable.to_ability_name.humanize(capitalize: false)}"
|
|
end
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.open? &&
|
|
current_user.can?(:"update_#{issuable.to_ability_name}", issuable)
|
|
end
|
|
command :close do
|
|
@updates[:state_event] = 'close'
|
|
end
|
|
|
|
desc do
|
|
"Reopen this #{issuable.to_ability_name.humanize(capitalize: false)}"
|
|
end
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.closed? &&
|
|
current_user.can?(:"update_#{issuable.to_ability_name}", issuable)
|
|
end
|
|
command :reopen do
|
|
@updates[:state_event] = 'reopen'
|
|
end
|
|
|
|
desc 'Change title'
|
|
params '<New title>'
|
|
condition do
|
|
issuable.persisted? &&
|
|
current_user.can?(:"update_#{issuable.to_ability_name}", issuable)
|
|
end
|
|
command :title do |title_param|
|
|
@updates[:title] = title_param
|
|
end
|
|
|
|
desc 'Assign'
|
|
params '@user'
|
|
condition do
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :assign do |assignee_param|
|
|
user = extract_references(assignee_param, :user).first
|
|
user ||= User.find_by(username: assignee_param)
|
|
|
|
@updates[:assignee_id] = user.id if user
|
|
end
|
|
|
|
desc 'Remove assignee'
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.assignee_id? &&
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :unassign do
|
|
@updates[:assignee_id] = nil
|
|
end
|
|
|
|
desc 'Set milestone'
|
|
params '%"milestone"'
|
|
condition do
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project) &&
|
|
project.milestones.active.any?
|
|
end
|
|
command :milestone do |milestone_param|
|
|
milestone = extract_references(milestone_param, :milestone).first
|
|
milestone ||= project.milestones.find_by(title: milestone_param.strip)
|
|
|
|
@updates[:milestone_id] = milestone.id if milestone
|
|
end
|
|
|
|
desc 'Remove milestone'
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.milestone_id? &&
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :remove_milestone do
|
|
@updates[:milestone_id] = nil
|
|
end
|
|
|
|
desc 'Add label(s)'
|
|
params '~label1 ~"label 2"'
|
|
condition do
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project) &&
|
|
project.labels.any?
|
|
end
|
|
command :label do |labels_param|
|
|
label_ids = find_label_ids(labels_param)
|
|
|
|
@updates[:add_label_ids] = label_ids unless label_ids.empty?
|
|
end
|
|
|
|
desc 'Remove all or specific label(s)'
|
|
params '~label1 ~"label 2"'
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.labels.any? &&
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :unlabel do |labels_param = nil|
|
|
if labels_param.present?
|
|
label_ids = find_label_ids(labels_param)
|
|
|
|
@updates[:remove_label_ids] = label_ids unless label_ids.empty?
|
|
else
|
|
@updates[:label_ids] = []
|
|
end
|
|
end
|
|
|
|
desc 'Replace all label(s)'
|
|
params '~label1 ~"label 2"'
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.labels.any? &&
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :relabel do |labels_param|
|
|
label_ids = find_label_ids(labels_param)
|
|
|
|
@updates[:label_ids] = label_ids unless label_ids.empty?
|
|
end
|
|
|
|
desc 'Add a todo'
|
|
condition do
|
|
issuable.persisted? &&
|
|
!TodoService.new.todo_exist?(issuable, current_user)
|
|
end
|
|
command :todo do
|
|
@updates[:todo_event] = 'add'
|
|
end
|
|
|
|
desc 'Mark todo as done'
|
|
condition do
|
|
issuable.persisted? &&
|
|
TodoService.new.todo_exist?(issuable, current_user)
|
|
end
|
|
command :done do
|
|
@updates[:todo_event] = 'done'
|
|
end
|
|
|
|
desc 'Subscribe'
|
|
condition do
|
|
issuable.persisted? &&
|
|
!issuable.subscribed?(current_user)
|
|
end
|
|
command :subscribe do
|
|
@updates[:subscription_event] = 'subscribe'
|
|
end
|
|
|
|
desc 'Unsubscribe'
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.subscribed?(current_user)
|
|
end
|
|
command :unsubscribe do
|
|
@updates[:subscription_event] = 'unsubscribe'
|
|
end
|
|
|
|
desc 'Set due date'
|
|
params '<in 2 days | this Friday | December 31st>'
|
|
condition do
|
|
issuable.respond_to?(:due_date) &&
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :due do |due_date_param|
|
|
due_date = Chronic.parse(due_date_param).try(:to_date)
|
|
|
|
@updates[:due_date] = due_date if due_date
|
|
end
|
|
|
|
desc 'Remove due date'
|
|
condition do
|
|
issuable.persisted? &&
|
|
issuable.respond_to?(:due_date) &&
|
|
issuable.due_date? &&
|
|
current_user.can?(:"admin_#{issuable.to_ability_name}", project)
|
|
end
|
|
command :remove_due_date do
|
|
@updates[:due_date] = nil
|
|
end
|
|
|
|
# This is a dummy command, so that it appears in the autocomplete commands
|
|
desc 'CC'
|
|
params '@user'
|
|
command :cc
|
|
|
|
def find_label_ids(labels_param)
|
|
label_ids_by_reference = extract_references(labels_param, :label).map(&:id)
|
|
labels_ids_by_name = @project.labels.where(name: labels_param.split).select(:id)
|
|
|
|
label_ids_by_reference | labels_ids_by_name
|
|
end
|
|
|
|
def extract_references(arg, type)
|
|
ext = Gitlab::ReferenceExtractor.new(project, current_user)
|
|
ext.analyze(arg, author: current_user)
|
|
|
|
ext.references(type)
|
|
end
|
|
end
|
|
end
|