Commit 375e83bb authored by Alejandro Rodríguez's avatar Alejandro Rodríguez

Consistently using iid when treating milestones as referrables

Also, addint a suffix to the reference text when the milestone is in another project
parent 077f9a4e
...@@ -83,7 +83,7 @@ class Milestone < ActiveRecord::Base ...@@ -83,7 +83,7 @@ class Milestone < ActiveRecord::Base
(#{Project.reference_pattern})? (#{Project.reference_pattern})?
#{Regexp.escape(reference_prefix)} #{Regexp.escape(reference_prefix)}
(?: (?:
(?<milestone_id>\d+) | # Integer-based milestone ID, or (?<milestone_iid>\d+) | # Integer-based milestone iid, or
(?<milestone_name> (?<milestone_name>
[A-Za-z0-9_-]+ | # String-based single-word milestone title, or [A-Za-z0-9_-]+ | # String-based single-word milestone title, or
"[^"]+" # String-based multi-word milestone surrounded in quotes "[^"]+" # String-based multi-word milestone surrounded in quotes
...@@ -100,7 +100,18 @@ class Milestone < ActiveRecord::Base ...@@ -100,7 +100,18 @@ class Milestone < ActiveRecord::Base
self.where('due_date > ?', Time.now).reorder(due_date: :asc).first self.where('due_date > ?', Time.now).reorder(due_date: :asc).first
end end
def to_reference(from_project = nil, format: :id) ##
# Returns the String necessary to reference this Milestone in Markdown
#
# format - Symbol format to use (default: :iid, optional: :name)
#
# Examples:
#
# Milestone.first.to_reference # => "%1"
# Milestone.first.to_reference(format: :name) # => "%\"goal\""
# Milestone.first.to_reference(project) # => "gitlab-org/gitlab-ce%1"
#
def to_reference(from_project = nil, format: :iid)
format_reference = milestone_format_reference(format) format_reference = milestone_format_reference(format)
reference = "#{self.class.reference_prefix}#{format_reference}" reference = "#{self.class.reference_prefix}#{format_reference}"
...@@ -179,13 +190,13 @@ class Milestone < ActiveRecord::Base ...@@ -179,13 +190,13 @@ class Milestone < ActiveRecord::Base
private private
def milestone_format_reference(format = :id) def milestone_format_reference(format = :iid)
raise StandardError, 'Unknown format' unless [:id, :name].include?(format) raise StandardError, 'Unknown format' unless [:iid, :name].include?(format)
if format == :name && !name.include?('"') if format == :name && !name.include?('"')
%("#{name}") %("#{name}")
else else
id iid
end end
end end
end end
...@@ -7,17 +7,17 @@ module Banzai ...@@ -7,17 +7,17 @@ module Banzai
end end
def find_object(project, id) def find_object(project, id)
project.milestones.find(id) project.milestones.find_by(iid: id)
end end
def references_in(text, pattern = Milestone.reference_pattern) def references_in(text, pattern = Milestone.reference_pattern)
text.gsub(pattern) do |match| text.gsub(pattern) do |match|
project = project_from_ref($~[:project]) project = project_from_ref($~[:project])
params = milestone_params($~[:milestone_id].to_i, $~[:milestone_name]) params = milestone_params($~[:milestone_iid].to_i, $~[:milestone_name])
milestone = project.milestones.find_by(params) milestone = project.milestones.find_by(params)
if milestone if milestone
yield match, milestone.id, $~[:project], $~ yield match, milestone.iid, $~[:project], $~
else else
match match
end end
...@@ -30,11 +30,20 @@ module Banzai ...@@ -30,11 +30,20 @@ module Banzai
only_path: context[:only_path]) only_path: context[:only_path])
end end
def milestone_params(id, name) def object_link_text(object, matches)
if context[:project] == object.project
super
else
"#{super} <i>in #{escape_once(object.project.name_with_namespace)}</i>".
html_safe
end
end
def milestone_params(iid, name)
if name if name
{ name: name.tr('"', '') } { name: name.tr('"', '') }
else else
{ id: id } { iid: iid }
end end
end end
end end
......
...@@ -5,6 +5,7 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do ...@@ -5,6 +5,7 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do
let(:project) { create(:project, :public) } let(:project) { create(:project, :public) }
let(:milestone) { create(:milestone, project: project) } let(:milestone) { create(:milestone, project: project) }
let(:reference) { milestone.to_reference }
it 'requires project context' do it 'requires project context' do
expect { described_class.call('') }.to raise_error(ArgumentError, /:project/) expect { described_class.call('') }.to raise_error(ArgumentError, /:project/)
...@@ -17,11 +18,42 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do ...@@ -17,11 +18,42 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do
end end
end end
context 'internal reference' do it 'includes default classes' do
# Convert the Markdown link to only the URL, since these tests aren't run through the regular Markdown pipeline. doc = reference_filter("Milestone #{reference}")
# Milestone reference behavior in the full Markdown pipeline is tested elsewhere. expect(doc.css('a').first.attr('class')).to eq 'gfm gfm-milestone'
let(:reference) { milestone.to_reference.gsub(/\[([^\]]+)\]\(([^)]+)\)/, '\2') } end
it 'includes a data-project attribute' do
doc = reference_filter("Milestone #{reference}")
link = doc.css('a').first
expect(link).to have_attribute('data-project')
expect(link.attr('data-project')).to eq project.id.to_s
end
it 'includes a data-milestone attribute' do
doc = reference_filter("See #{reference}")
link = doc.css('a').first
expect(link).to have_attribute('data-milestone')
expect(link.attr('data-milestone')).to eq milestone.id.to_s
end
it 'supports an :only_path context' do
doc = reference_filter("Milestone #{reference}", only_path: true)
link = doc.css('a').first.attr('href')
expect(link).not_to match %r(https?://)
expect(link).to eq urls.
namespace_project_milestone_path(project.namespace, project, milestone)
end
it 'adds to the results hash' do
result = reference_pipeline_result("Milestone #{reference}")
expect(result[:references][:milestone]).to eq [milestone]
end
context 'Integer-based references' do
it 'links to a valid reference' do it 'links to a valid reference' do
doc = reference_filter("See #{reference}") doc = reference_filter("See #{reference}")
...@@ -30,29 +62,82 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do ...@@ -30,29 +62,82 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do
end end
it 'links with adjacent text' do it 'links with adjacent text' do
doc = reference_filter("milestone (#{reference}.)") doc = reference_filter("Milestone (#{reference}.)")
expect(doc.to_html).to match(/\(<a.+>#{Regexp.escape(milestone.title)}<\/a>\.\)/) expect(doc.to_html).to match(%r(\(<a.+>#{milestone.name}</a>\.\)))
end end
it 'includes a title attribute' do it 'ignores invalid milestone IIDs' do
doc = reference_filter("milestone #{reference}") exp = act = "Milestone #{invalidate_reference(reference)}"
expect(doc.css('a').first.attr('title')).to eq "Milestone: #{milestone.title}"
expect(reference_filter(act).to_html).to eq exp
end
end end
it 'escapes the title attribute' do context 'String-based single-word references' do
milestone.update_attribute(:title, %{"></a>whatever<a title="}) let(:milestone) { create(:milestone, name: 'gfm', project: project) }
let(:reference) { "#{Milestone.reference_prefix}#{milestone.name}" }
doc = reference_filter("milestone #{reference}") it 'links to a valid reference' do
expect(doc.text).to eq "milestone #{milestone.title}" doc = reference_filter("See #{reference}")
expect(doc.css('a').first.attr('href')).to eq urls.
namespace_project_milestone_url(project.namespace, project, milestone)
expect(doc.text).to eq 'See gfm'
end end
it 'includes default classes' do it 'links with adjacent text' do
doc = reference_filter("milestone #{reference}") doc = reference_filter("Milestone (#{reference}.)")
expect(doc.css('a').first.attr('class')).to eq 'gfm gfm-milestone' expect(doc.to_html).to match(%r(\(<a.+>#{milestone.name}</a>\.\)))
end
it 'ignores invalid milestone names' do
exp = act = "Milestone #{Milestone.reference_prefix}#{milestone.name.reverse}"
expect(reference_filter(act).to_html).to eq exp
end
end
context 'String-based multi-word references in quotes' do
let(:milestone) { create(:milestone, name: 'gfm references', project: project) }
let(:reference) { milestone.to_reference(format: :name) }
it 'links to a valid reference' do
doc = reference_filter("See #{reference}")
expect(doc.css('a').first.attr('href')).to eq urls.
namespace_project_milestone_url(project.namespace, project, milestone)
expect(doc.text).to eq 'See gfm references'
end
it 'links with adjacent text' do
doc = reference_filter("Milestone (#{reference}.)")
expect(doc.to_html).to match(%r(\(<a.+>#{milestone.name}</a>\.\)))
end
it 'ignores invalid milestone names' do
exp = act = %(Milestone #{Milestone.reference_prefix}"#{milestone.name.reverse}")
expect(reference_filter(act).to_html).to eq exp
end
end
describe 'referencing a milestone in a link href' do
let(:reference) { %Q{<a href="#{milestone.to_reference}">Milestone</a>} }
it 'links to a valid reference' do
doc = reference_filter("See #{reference}")
expect(doc.css('a').first.attr('href')).to eq urls.
namespace_project_milestone_url(project.namespace, project, milestone)
end
it 'links with adjacent text' do
doc = reference_filter("Milestone (#{reference}.)")
expect(doc.to_html).to match(%r(\(<a.+>Milestone</a>\.\)))
end end
it 'includes a data-project attribute' do it 'includes a data-project attribute' do
doc = reference_filter("milestone #{reference}") doc = reference_filter("Milestone #{reference}")
link = doc.css('a').first link = doc.css('a').first
expect(link).to have_attribute('data-project') expect(link).to have_attribute('data-project')
...@@ -68,8 +153,28 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do ...@@ -68,8 +153,28 @@ describe Banzai::Filter::MilestoneReferenceFilter, lib: true do
end end
it 'adds to the results hash' do it 'adds to the results hash' do
result = reference_pipeline_result("milestone #{reference}") result = reference_pipeline_result("Milestone #{reference}")
expect(result[:references][:milestone]).to eq [milestone] expect(result[:references][:milestone]).to eq [milestone]
end end
end end
describe 'cross project milestone references' do
let(:another_project) { create(:empty_project, :public) }
let(:project_name) { another_project.name_with_namespace }
let(:milestone) { create(:milestone, project: another_project) }
let(:reference) { milestone.to_reference(project) }
let!(:result) { reference_filter("See #{reference}") }
it 'points to referenced project milestone page' do
expect(result.css('a').first.attr('href')).to eq urls.
namespace_project_milestone_url(another_project.namespace,
another_project,
milestone)
end
it 'contains cross project content' do
expect(result.css('a').first.text).to eq "#{milestone.name} in #{project_name}"
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