Render milestone links as references

1 files changed, 18 insertions, 0 deletions

diff --git a/app/models/milestone.rb b/app/models/milestone.rb
index d8c7536cd31..e47b6440746 100644
--- a/app/models/milestone.rb
+++ b/app/models/milestone.rb
@@ -22,6 +22,7 @@ class Milestone < ActiveRecord::Base
 
   include InternalId
   include Sortable
+  include Referable
   include StripAttribute
 
   belongs_to :project
@@ -61,6 +62,23 @@ class Milestone < ActiveRecord::Base
     end
   end
 
+  def self.reference_pattern
+    nil
+  end
+
+  def self.link_reference_pattern
+    super("milestones", /(?<milestone>\d+)/)
+  end
+
+  def to_reference(from_project = nil)
+    h = Gitlab::Application.routes.url_helpers
+    h.namespace_project_milestone_url(self.project.namespace, self.project, self)
+  end
+
+  def reference_link_text(from_project = nil)
+    %Q{<i class="fa fa-clock-o"></i> }.html_safe + self.title
+  end
+
   def expired?
     if due_date
       due_date.past?