diff options
| -rw-r--r-- | app/services/issuable_base_service.rb | 2 | ||||
| -rw-r--r-- | app/services/slash_commands/interpret_service.rb | 14 | ||||
| -rw-r--r-- | doc/workflow/slash_commands.md | 6 | ||||
| -rw-r--r-- | lib/gitlab/slash_commands/extractor.rb | 51 | ||||
| -rw-r--r-- | spec/lib/gitlab/slash_commands/extractor_spec.rb | 27 | ||||
| -rw-r--r-- | spec/services/slash_commands/interpret_service_spec.rb | 14 |
6 files changed, 98 insertions, 16 deletions
diff --git a/app/services/issuable_base_service.rb b/app/services/issuable_base_service.rb index 1ef7a2433dc..c14bda811c2 100644 --- a/app/services/issuable_base_service.rb +++ b/app/services/issuable_base_service.rb @@ -173,7 +173,7 @@ class IssuableBaseService < BaseService def change_todo(issuable) case params.delete(:todo_event) - when 'mark' + when 'add' todo_service.mark_todo(issuable, current_user) when 'done' todo = TodosFinder.new(current_user).execute.find_by(target: issuable) diff --git a/app/services/slash_commands/interpret_service.rb b/app/services/slash_commands/interpret_service.rb index 74825f30868..3030af05999 100644 --- a/app/services/slash_commands/interpret_service.rb +++ b/app/services/slash_commands/interpret_service.rb @@ -40,7 +40,7 @@ module SlashCommands @updates[:title] = title_param end - desc 'Reassign' + desc 'Assign' params '@user' command :assign, :reassign do |assignee_param| user = extract_references(assignee_param, :user).first @@ -54,7 +54,7 @@ module SlashCommands @updates[:assignee_id] = nil end - desc 'Change milestone' + desc 'Set milestone' params '%"milestone"' command :milestone do |milestone_param| milestone = extract_references(milestone_param, :milestone).first @@ -93,7 +93,7 @@ module SlashCommands desc 'Add a todo' command :todo do - @updates[:todo_event] = 'mark' + @updates[:todo_event] = 'add' end desc 'Mark todo as done' @@ -113,11 +113,15 @@ module SlashCommands desc 'Set a due date' params '<YYYY-MM-DD> | <N days>' - command :due_date do |due_date_param| + command :due_date, :due do |due_date_param| return unless noteable.respond_to?(:due_date) due_date = begin - Time.now + ChronicDuration.parse(due_date_param) + if due_date_param.downcase == 'tomorrow' + Date.tomorrow + else + Time.now + ChronicDuration.parse(due_date_param) + end rescue ChronicDuration::DurationParseError Date.parse(due_date_param) rescue nil end diff --git a/doc/workflow/slash_commands.md b/doc/workflow/slash_commands.md index bf5b8ebe1c8..46f291561d7 100644 --- a/doc/workflow/slash_commands.md +++ b/doc/workflow/slash_commands.md @@ -14,9 +14,9 @@ do. | `/close` | None | Close the issue or merge request | | `/open` | `/reopen` | Reopen the issue or merge request | | `/title <New title>` | None | Change title | -| `/assign @username` | `/reassign` | Reassign | +| `/assign @username` | `/reassign` | Assign | | `/unassign` | `/remove_assignee` | Remove assignee | -| `/milestone %milestone` | None | Change milestone | +| `/milestone %milestone` | None | Set milestone | | `/clear_milestone` | `/remove_milestone` | Remove milestone | | `/label ~foo ~"bar baz"` | `/labels` | Add label(s) | | `/unlabel ~foo ~"bar baz"` | `/remove_label`, `remove_labels` | Remove label(s) | @@ -25,5 +25,5 @@ do. | `/done` | None | Mark todo as done | | `/subscribe` | None | Subscribe | | `/unsubscribe` | None | Unsubscribe | -| `/due_date <YYYY-MM-DD> | <N days>` | None | Set a due date | +| `/due_date <YYYY-MM-DD> | <N days>` | `/due` | Set a due date | | `/clear_due_date` | None | Remove due date | diff --git a/lib/gitlab/slash_commands/extractor.rb b/lib/gitlab/slash_commands/extractor.rb index 1a854b81aca..ce0a2eba535 100644 --- a/lib/gitlab/slash_commands/extractor.rb +++ b/lib/gitlab/slash_commands/extractor.rb @@ -34,9 +34,14 @@ module Gitlab commands = [] + content.delete!("\r") content.gsub!(commands_regex) do - commands << [$1, $2].flatten.reject(&:blank?) - '' + if $~[:cmd] + commands << [$~[:cmd], $~[:args]].reject(&:blank?) + '' + else + $~[0] + end end commands @@ -52,7 +57,47 @@ module Gitlab # # /^\/(?<cmd>close|reopen|...)(?:( |$))(?<args>[^\/\n]*)(?:\n|$)/ def commands_regex - /^\/(?<cmd>#{command_names.join('|')})(?:( |$))(?<args>[^\/\n]*)(?:\n|$)/ + @commands_regex ||= %r{ + (?<code> + # Code blocks: + # ``` + # Anything, including `/cmd args` which are ignored by this filter + # ``` + + ^``` + .+? + \n```$ + ) + | + (?<html> + # HTML block: + # <tag> + # Anything, including `/cmd args` which are ignored by this filter + # </tag> + + ^<[^>]+?>\n + .+? + \n<\/[^>]+?>$ + ) + | + (?<html> + # Quote block: + # >>> + # Anything, including `/cmd args` which are ignored by this filter + # >>> + + ^>>> + .+? + \n>>>$ + ) + | + (?: + # Command not in a blockquote, blockcode, or HTML tag: + # /close + + ^\/(?<cmd>#{command_names.join('|')})(?:(\ |$))(?<args>[^\/\n]*)(?:\n|$) + ) + }mx end end end diff --git a/spec/lib/gitlab/slash_commands/extractor_spec.rb b/spec/lib/gitlab/slash_commands/extractor_spec.rb index fd1b30052ed..11836b10204 100644 --- a/spec/lib/gitlab/slash_commands/extractor_spec.rb +++ b/spec/lib/gitlab/slash_commands/extractor_spec.rb @@ -173,5 +173,32 @@ describe Gitlab::SlashCommands::Extractor do expect(commands).to be_empty expect(msg).to eq 'Fixes #123' end + + it 'does not extract commands inside a blockcode' do + msg = msg = "Hello\r\n```\r\nThis is some text\r\n/close\r\n/assign @user\r\n```\r\n\r\nWorld" + expected = msg.delete("\r") + commands = extractor.extract_commands!(msg) + + expect(commands).to be_empty + expect(msg).to eq expected + end + + it 'does not extract commands inside a blockquote' do + msg = "Hello\r\n>>>\r\nThis is some text\r\n/close\r\n/assign @user\r\n>>>\r\n\r\nWorld" + expected = msg.delete("\r") + commands = extractor.extract_commands!(msg) + + expect(commands).to be_empty + expect(msg).to eq expected + end + + it 'does not extract commands inside a HTML tag' do + msg = msg = "Hello\r\n<div>\r\nThis is some text\r\n/close\r\n/assign @user\r\n</div>\r\n\r\nWorld" + expected = msg.delete("\r") + commands = extractor.extract_commands!(msg) + + expect(commands).to be_empty + expect(msg).to eq expected + end end end diff --git a/spec/services/slash_commands/interpret_service_spec.rb b/spec/services/slash_commands/interpret_service_spec.rb index 66ebe091893..620687e3212 100644 --- a/spec/services/slash_commands/interpret_service_spec.rb +++ b/spec/services/slash_commands/interpret_service_spec.rb @@ -27,7 +27,7 @@ describe SlashCommands::InterpretService, services: true do :done, :subscribe, :unsubscribe, - :due_date, + :due_date, :due, :clear_due_date ]) end @@ -119,10 +119,10 @@ describe SlashCommands::InterpretService, services: true do end shared_examples 'todo command' do - it 'populates todo_event: "mark" if content contains /todo' do + it 'populates todo_event: "add" if content contains /todo' do changes = service.execute(content, issuable) - expect(changes).to eq(todo_event: 'mark') + expect(changes).to eq(todo_event: 'add') end end @@ -154,7 +154,7 @@ describe SlashCommands::InterpretService, services: true do it 'populates due_date: Date.new(2016, 8, 28) if content contains /due_date 2016-08-28' do changes = service.execute(content, issuable) - expect(changes).to eq(due_date: Date.new(2016, 8, 28)) + expect(changes).to eq(due_date: defined?(expected_date) ? expected_date : Date.new(2016, 8, 28)) end end @@ -369,6 +369,12 @@ describe SlashCommands::InterpretService, services: true do let(:issuable) { issue } end + it_behaves_like 'due_date command' do + let(:content) { '/due tomorrow' } + let(:issuable) { issue } + let(:expected_date) { Date.tomorrow } + end + it_behaves_like 'empty command' do let(:content) { '/due_date foo bar' } let(:issuable) { issue } |