Show badges in Milestone tabs and padding in Milestone list (both to match Issues page)

1 files changed, 13 insertions, 0 deletions

diff --git a/app/helpers/milestones_helper.rb b/app/helpers/milestones_helper.rb
index b3e6e468ecd..9a71761a55c 100644
--- a/app/helpers/milestones_helper.rb
+++ b/app/helpers/milestones_helper.rb
@@ -35,6 +35,19 @@ module MilestonesHelper
     milestone.issues.with_label(label.title).send(state).size
   end
 
+  def milestone_count_by_state(project, state)
+    if project.nil?
+      nil
+    else
+      case state
+      when 'all' then @project.milestones.size
+      when 'closed' then @project.milestones.closed.size
+      when 'opened' then @project.milestones.active.size
+      else nil
+      end
+    end
+  end
+
   def milestone_progress_bar(milestone)
     options = {
       class: 'progress-bar progress-bar-success',