BigW Consortium Gitlab

Commit 123669a5 by Robert Speicher

Merge branch 'simplify-cross-references' into 'master'

Simplify code around (cross)-references See merge request !1568
parents a3a80eac 276b3a7b
...@@ -55,7 +55,7 @@ class Projects::IssuesController < Projects::ApplicationController ...@@ -55,7 +55,7 @@ class Projects::IssuesController < Projects::ApplicationController
end end
def show def show
@participants = @issue.participants(current_user, @project) @participants = @issue.participants(current_user)
@note = @project.notes.new(noteable: @issue) @note = @project.notes.new(noteable: @issue)
@notes = @issue.notes.inc_author.fresh @notes = @issue.notes.inc_author.fresh
@noteable = @issue @noteable = @issue
......
...@@ -246,7 +246,7 @@ class Projects::MergeRequestsController < Projects::ApplicationController ...@@ -246,7 +246,7 @@ class Projects::MergeRequestsController < Projects::ApplicationController
end end
def define_show_vars def define_show_vars
@participants = @merge_request.participants(current_user, @project) @participants = @merge_request.participants(current_user)
# Build a note object for comment form # Build a note object for comment form
@note = @project.notes.new(noteable: @merge_request) @note = @project.notes.new(noteable: @merge_request)
......
...@@ -41,55 +41,49 @@ module Mentionable ...@@ -41,55 +41,49 @@ module Mentionable
self self
end end
# Determine whether or not a cross-reference Note has already been created between this Mentionable and def all_references(current_user = self.author, text = self.mentionable_text)
# the specified target. ext = Gitlab::ReferenceExtractor.new(self.project, current_user)
def has_mentioned?(target) ext.analyze(text)
SystemNoteService.cross_reference_exists?(target, local_reference) ext
end end
def mentioned_users(current_user = nil) def mentioned_users(current_user = nil)
return [] if mentionable_text.blank? all_references(current_user).users.uniq
ext = Gitlab::ReferenceExtractor.new(self.project, current_user)
ext.analyze(mentionable_text)
ext.users.uniq
end end
# Extract GFM references to other Mentionables from this Mentionable. Always excludes its #local_reference. # Extract GFM references to other Mentionables from this Mentionable. Always excludes its #local_reference.
def references(p = project, current_user = self.author, text = mentionable_text) def referenced_mentionables(current_user = self.author, text = self.mentionable_text)
return [] if text.blank? return [] if text.blank?
ext = Gitlab::ReferenceExtractor.new(p, current_user) refs = all_references(current_user, text)
ext.analyze(text) (refs.issues + refs.merge_requests + refs.commits).uniq - [local_reference]
(ext.issues + ext.merge_requests + ext.commits).uniq - [local_reference]
end end
# Create a cross-reference Note for each GFM reference to another Mentionable found in +mentionable_text+. # Create a cross-reference Note for each GFM reference to another Mentionable found in +mentionable_text+.
def create_cross_references!(p = project, a = author, without = []) def create_cross_references!(author = self.author, without = [], text = self.mentionable_text)
refs = references(p) refs = referenced_mentionables(author, text)
# We're using this method instead of Array diffing because that requires # We're using this method instead of Array diffing because that requires
# both of the object's `hash` values to be the same, which may not be the # both of the object's `hash` values to be the same, which may not be the
# case for otherwise identical Commit objects. # case for otherwise identical Commit objects.
refs.reject! { |ref| without.include?(ref) } refs.reject! { |ref| without.include?(ref) || cross_reference_exists?(ref) }
refs.each do |ref| refs.each do |ref|
SystemNoteService.cross_reference(ref, local_reference, a) SystemNoteService.cross_reference(ref, local_reference, author)
end end
end end
# When a mentionable field is changed, creates cross-reference notes that # When a mentionable field is changed, creates cross-reference notes that
# don't already exist # don't already exist
def create_new_cross_references!(p = project, a = author) def create_new_cross_references!(author = self.author)
changes = detect_mentionable_changes changes = detect_mentionable_changes
return if changes.empty? return if changes.empty?
original_text = changes.collect { |_, vals| vals.first }.join(' ') original_text = changes.collect { |_, vals| vals.first }.join(' ')
preexisting = references(p, self.author, original_text) preexisting = referenced_mentionables(author, original_text)
create_cross_references!(p, a, preexisting) create_cross_references!(author, preexisting)
end end
private private
...@@ -111,4 +105,10 @@ module Mentionable ...@@ -111,4 +105,10 @@ module Mentionable
# Only include changed fields that are mentionable # Only include changed fields that are mentionable
source.select { |key, val| mentionable.include?(key) } source.select { |key, val| mentionable.include?(key) }
end end
# Determine whether or not a cross-reference Note has already been created between this Mentionable and
# the specified target.
def cross_reference_exists?(target)
SystemNoteService.cross_reference_exists?(target, local_reference)
end
end end
...@@ -37,8 +37,8 @@ module Participable ...@@ -37,8 +37,8 @@ module Participable
# Be aware that this method makes a lot of sql queries. # Be aware that this method makes a lot of sql queries.
# Save result into variable if you are going to reuse it inside same request # Save result into variable if you are going to reuse it inside same request
def participants(current_user = self.author, project = self.project) def participants(current_user = self.author)
participants = self.class.participant_attrs.flat_map do |attr| self.class.participant_attrs.flat_map do |attr|
meth = method(attr) meth = method(attr)
value = value =
...@@ -48,28 +48,22 @@ module Participable ...@@ -48,28 +48,22 @@ module Participable
meth.call meth.call
end end
participants_for(value, current_user, project) participants_for(value, current_user)
end.compact.uniq end.compact.uniq.select do |user|
user.can?(:read_project, self.project)
if project
participants.select! do |user|
user.can?(:read_project, project)
end
end end
participants
end end
private private
def participants_for(value, current_user = nil, project = nil) def participants_for(value, current_user = nil)
case value case value
when User when User
[value] [value]
when Enumerable, ActiveRecord::Relation when Enumerable, ActiveRecord::Relation
value.flat_map { |v| participants_for(v, current_user, project) } value.flat_map { |v| participants_for(v, current_user) }
when Participable when Participable
value.participants(current_user, project) value.participants(current_user)
end end
end end
end end
...@@ -62,7 +62,6 @@ class Note < ActiveRecord::Base ...@@ -62,7 +62,6 @@ class Note < ActiveRecord::Base
serialize :st_diff serialize :st_diff
before_create :set_diff, if: ->(n) { n.line_code.present? } before_create :set_diff, if: ->(n) { n.line_code.present? }
after_update :set_references
class << self class << self
def discussions_from_notes(notes) def discussions_from_notes(notes)
...@@ -333,15 +332,13 @@ class Note < ActiveRecord::Base ...@@ -333,15 +332,13 @@ class Note < ActiveRecord::Base
end end
def noteable_type_name def noteable_type_name
if noteable_type.present? noteable_type.downcase if noteable_type.present?
noteable_type.downcase
end
end end
# FIXME: Hack for polymorphic associations with STI # FIXME: Hack for polymorphic associations with STI
# For more information visit http://api.rubyonrails.org/classes/ActiveRecord/Associations/ClassMethods.html#label-Polymorphic+Associations # For more information visit http://api.rubyonrails.org/classes/ActiveRecord/Associations/ClassMethods.html#label-Polymorphic+Associations
def noteable_type=(sType) def noteable_type=(noteable_type)
super(sType.to_s.classify.constantize.base_class.to_s) super(noteable_type.to_s.classify.constantize.base_class.to_s)
end end
# Reset notes events cache # Reset notes events cache
...@@ -357,10 +354,6 @@ class Note < ActiveRecord::Base ...@@ -357,10 +354,6 @@ class Note < ActiveRecord::Base
Event.reset_event_cache_for(self) Event.reset_event_cache_for(self)
end end
def set_references
create_new_cross_references!(project, author)
end
def system? def system?
read_attribute(:system) read_attribute(:system)
end end
......
...@@ -74,48 +74,30 @@ class GitPushService ...@@ -74,48 +74,30 @@ class GitPushService
def process_commit_messages(ref) def process_commit_messages(ref)
is_default_branch = is_default_branch?(ref) is_default_branch = is_default_branch?(ref)
@push_commits.each do |commit| authors = Hash.new do |hash, commit|
# Close issues if these commits were pushed to the project's default branch and the commit message matches the email = commit.author_email
# closing regex. Exclude any mentioned Issues from cross-referencing even if the commits are being pushed to return hash[email] if hash.has_key?(email)
# a different branch.
issues_to_close = commit.closes_issues(user)
# Load commit author only if needed. hash[email] = commit_user(commit)
# For push with 1k commits it prevents 900+ requests in database end
author = nil
@push_commits.each do |commit|
# Keep track of the issues that will be actually closed because they are on a default branch. # Keep track of the issues that will be actually closed because they are on a default branch.
# Hence, when creating cross-reference notes, the not-closed issues (on non-default branches) # Hence, when creating cross-reference notes, the not-closed issues (on non-default branches)
# will also have cross-reference. # will also have cross-reference.
actually_closed_issues = [] closed_issues = []
if issues_to_close.present? && is_default_branch if is_default_branch
author ||= commit_user(commit) # Close issues if these commits were pushed to the project's default branch and the commit message matches the
actually_closed_issues = issues_to_close # closing regex. Exclude any mentioned Issues from cross-referencing even if the commits are being pushed to
issues_to_close.each do |issue| # a different branch.
Issues::CloseService.new(project, author, {}).execute(issue, commit) closed_issues = commit.closes_issues(user)
closed_issues.each do |issue|
Issues::CloseService.new(project, authors[commit], {}).execute(issue, commit)
end end
end end
if project.default_issues_tracker? commit.create_cross_references!(authors[commit], closed_issues)
create_cross_reference_notes(commit, actually_closed_issues)
end
end
end
def create_cross_reference_notes(commit, issues_to_close)
# Create cross-reference notes for any other references than those given in issues_to_close.
# Omit any issues that were referenced in an issue-closing phrase, or have already been
# mentioned from this commit (probably from this commit being pushed to a different branch).
refs = commit.references(project, user) - issues_to_close
refs.reject! { |r| commit.has_mentioned?(r) }
if refs.present?
author ||= commit_user(commit)
refs.each do |r|
SystemNoteService.cross_reference(r, commit, author)
end
end end
end end
......
...@@ -10,7 +10,7 @@ module Issues ...@@ -10,7 +10,7 @@ module Issues
issue.update_attributes(label_ids: label_params) issue.update_attributes(label_ids: label_params)
notification_service.new_issue(issue, current_user) notification_service.new_issue(issue, current_user)
event_service.open_issue(issue, current_user) event_service.open_issue(issue, current_user)
issue.create_cross_references!(issue.project, current_user) issue.create_cross_references!(current_user)
execute_hooks(issue, 'open') execute_hooks(issue, 'open')
end end
......
...@@ -35,7 +35,7 @@ module Issues ...@@ -35,7 +35,7 @@ module Issues
create_title_change_note(issue, issue.previous_changes['title'].first) create_title_change_note(issue, issue.previous_changes['title'].first)
end end
issue.create_new_cross_references!(issue.project, current_user) issue.create_new_cross_references!
execute_hooks(issue, 'update') execute_hooks(issue, 'update')
end end
......
...@@ -18,7 +18,7 @@ module MergeRequests ...@@ -18,7 +18,7 @@ module MergeRequests
merge_request.update_attributes(label_ids: label_params) merge_request.update_attributes(label_ids: label_params)
event_service.open_mr(merge_request, current_user) event_service.open_mr(merge_request, current_user)
notification_service.new_merge_request(merge_request, current_user) notification_service.new_merge_request(merge_request, current_user)
merge_request.create_cross_references!(merge_request.project, current_user) merge_request.create_cross_references!(current_user)
execute_hooks(merge_request) execute_hooks(merge_request)
end end
......
...@@ -59,7 +59,7 @@ module MergeRequests ...@@ -59,7 +59,7 @@ module MergeRequests
merge_request.mark_as_unchecked merge_request.mark_as_unchecked
end end
merge_request.create_new_cross_references!(merge_request.project, current_user) merge_request.create_new_cross_references!
execute_hooks(merge_request, 'update') execute_hooks(merge_request, 'update')
end end
......
...@@ -11,13 +11,7 @@ module Notes ...@@ -11,13 +11,7 @@ module Notes
# Skip system notes, like status changes and cross-references. # Skip system notes, like status changes and cross-references.
unless note.system unless note.system
event_service.leave_note(note, note.author) event_service.leave_note(note, note.author)
note.create_cross_references!
# Create a cross-reference note if this Note contains GFM that names an
# issue, merge request, or commit.
note.references.each do |mentioned|
SystemNoteService.cross_reference(mentioned, note.noteable, note.author)
end
execute_hooks(note) execute_hooks(note)
end end
end end
......
...@@ -4,7 +4,7 @@ module Notes ...@@ -4,7 +4,7 @@ module Notes
return note unless note.editable? return note unless note.editable?
note.update_attributes(params.merge(updated_by: current_user)) note.update_attributes(params.merge(updated_by: current_user))
note.create_new_cross_references!
note.reset_events_cache note.reset_events_cache
note note
......
...@@ -39,6 +39,8 @@ module Gitlab ...@@ -39,6 +39,8 @@ module Gitlab
# #
# Returns the results Array for the requested filter type # Returns the results Array for the requested filter type
def pipeline_result(filter_type) def pipeline_result(filter_type)
return [] if @text.blank?
klass = filter_type.to_s.camelize + 'ReferenceFilter' klass = filter_type.to_s.camelize + 'ReferenceFilter'
filter = Gitlab::Markdown.const_get(klass) filter = Gitlab::Markdown.const_get(klass)
......
...@@ -89,9 +89,9 @@ eos ...@@ -89,9 +89,9 @@ eos
end end
it_behaves_like 'a mentionable' do it_behaves_like 'a mentionable' do
subject { commit } subject { create(:project).commit }
let(:author) { create(:user, email: commit.author_email) } let(:author) { create(:user, email: subject.author_email) }
let(:backref_text) { "commit #{subject.id}" } let(:backref_text) { "commit #{subject.id}" }
let(:set_mentionable_text) do let(:set_mentionable_text) do
->(txt) { allow(subject).to receive(:safe_message).and_return(txt) } ->(txt) { allow(subject).to receive(:safe_message).and_return(txt) }
......
...@@ -25,7 +25,7 @@ describe Issue, "Mentionable" do ...@@ -25,7 +25,7 @@ describe Issue, "Mentionable" do
it 'correctly removes already-mentioned Commits' do it 'correctly removes already-mentioned Commits' do
expect(SystemNoteService).not_to receive(:cross_reference) expect(SystemNoteService).not_to receive(:cross_reference)
issue.create_cross_references!(project, author, [commit2]) issue.create_cross_references!(author, [commit2])
end end
end end
......
...@@ -69,7 +69,7 @@ describe Issue do ...@@ -69,7 +69,7 @@ describe Issue do
end end
it_behaves_like 'an editable mentionable' do it_behaves_like 'an editable mentionable' do
subject { create(:issue, project: project) } subject { create(:issue) }
let(:backref_text) { "issue #{subject.to_reference}" } let(:backref_text) { "issue #{subject.to_reference}" }
let(:set_mentionable_text) { ->(txt){ subject.description = txt } } let(:set_mentionable_text) { ->(txt){ subject.description = txt } }
......
...@@ -192,10 +192,9 @@ describe Note do ...@@ -192,10 +192,9 @@ describe Note do
end end
it_behaves_like 'an editable mentionable' do it_behaves_like 'an editable mentionable' do
subject { create :note, noteable: issue, project: project } subject { create :note, noteable: issue, project: issue.project }
let(:project) { create(:project) } let(:issue) { create :issue }
let(:issue) { create :issue, project: project }
let(:backref_text) { issue.gfm_reference } let(:backref_text) { issue.gfm_reference }
let(:set_mentionable_text) { ->(txt) { subject.note = txt } } let(:set_mentionable_text) { ->(txt) { subject.note = txt } }
end end
......
...@@ -155,7 +155,7 @@ describe GitPushService do ...@@ -155,7 +155,7 @@ describe GitPushService do
before do before do
allow(commit).to receive_messages( allow(commit).to receive_messages(
safe_message: "this commit \n mentions ##{issue.id}", safe_message: "this commit \n mentions #{issue.to_reference}",
references: [issue], references: [issue],
author_name: commit_author.name, author_name: commit_author.name,
author_email: commit_author.email author_email: commit_author.email
......
...@@ -5,7 +5,7 @@ ...@@ -5,7 +5,7 @@
# - let(:set_mentionable_text) { lambda { |txt| "block that assigns txt to the subject's mentionable_text" } } # - let(:set_mentionable_text) { lambda { |txt| "block that assigns txt to the subject's mentionable_text" } }
def common_mentionable_setup def common_mentionable_setup
let(:project) { create :project } let(:project) { subject.project }
let(:author) { subject.author } let(:author) { subject.author }
let(:mentioned_issue) { create(:issue, project: project) } let(:mentioned_issue) { create(:issue, project: project) }
...@@ -65,7 +65,7 @@ shared_examples 'a mentionable' do ...@@ -65,7 +65,7 @@ shared_examples 'a mentionable' do
it "extracts references from its reference property" do it "extracts references from its reference property" do
# De-duplicate and omit itself # De-duplicate and omit itself
refs = subject.references(project) refs = subject.referenced_mentionables
expect(refs.size).to eq(6) expect(refs.size).to eq(6)
expect(refs).to include(mentioned_issue) expect(refs).to include(mentioned_issue)
expect(refs).to include(mentioned_mr) expect(refs).to include(mentioned_mr)
...@@ -84,14 +84,7 @@ shared_examples 'a mentionable' do ...@@ -84,14 +84,7 @@ shared_examples 'a mentionable' do
with(referenced, subject.local_reference, author) with(referenced, subject.local_reference, author)
end end
subject.create_cross_references!(project, author) subject.create_cross_references!
end
it 'detects existing cross-references' do
SystemNoteService.cross_reference(mentioned_issue, subject.local_reference, author)
expect(subject).to have_mentioned(mentioned_issue)
expect(subject).not_to have_mentioned(mentioned_mr)
end end
end end
...@@ -143,6 +136,6 @@ shared_examples 'an editable mentionable' do ...@@ -143,6 +136,6 @@ shared_examples 'an editable mentionable' do
end end
set_mentionable_text.call(new_text) set_mentionable_text.call(new_text)
subject.create_new_cross_references!(project, author) subject.create_new_cross_references!(author)
end end
end end
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment