Refactor label rendering and default label set generation

Signed-off-by: Dmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>

1 files changed, 18 insertions, 19 deletions

diff --git a/lib/gitlab/issues_labels.rb b/lib/gitlab/issues_labels.rb
index 9da1977a3d6..0d34976736f 100644
--- a/lib/gitlab/issues_labels.rb
+++ b/lib/gitlab/issues_labels.rb
@@ -1,27 +1,26 @@
 module Gitlab
   class IssuesLabels
     class << self
-      def important_labels
-        %w(bug critical confirmed)
-      end
-
-      def warning_labels
-        %w(documentation support)
-      end
-
-      def neutral_labels
-        %w(discussion suggestion)
-      end
-
-      def positive_labels
-        %w(feature enhancement)
-      end
-
       def generate(project)
-        label_names = important_labels + warning_labels + neutral_labels + positive_labels
+        red = '#d9534f'
+        yellow = '#f0ad4e'
+        blue = '#428bca'
+        green = '#5cb85c'
+
+        labels = [
+          { title: "bug", color: red },
+          { title: "critical", color: red },
+          { title: "confirmed", color: red },
+          { title: "documentation", color: yellow },
+          { title: "support", color: yellow },
+          { title: "discussion", color: blue },
+          { title: "suggestion", color: blue },
+          { title: "feature", color: green },
+          { title: "enhancement", color: green }
+        ]
 
-        label_names.each do |label_name|
-          project.labels.create(title: label_name)
+        labels.each do |label|
+          project.labels.create(label)
         end
       end
     end