1 files changed, 64 insertions, 0 deletions
diff --git a/lib/gitlab/ci/badge/coverage/template.rb b/lib/gitlab/ci/badge/coverage/template.rb
new file mode 100644
index 00000000000..7589fa5ff8b
--- /dev/null
+++ b/lib/gitlab/ci/badge/coverage/template.rb
@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module Gitlab::Ci
+  module Badge
+    module Coverage
+      ##
+      # Class that represents a coverage badge template.
+      #
+      # Template object will be passed to badge.svg.erb template.
+      #
+      class Template < Badge::Template
+        STATUS_COLOR = {
+          good: '#4c1',
+          acceptable: '#a3c51c',
+          medium: '#dfb317',
+          low: '#e05d44',
+          unknown: '#9f9f9f'
+        }.freeze
+
+        def initialize(badge)
+          @entity = badge.entity
+          @status = badge.status
+          @key_text = badge.customization.dig(:key_text)
+          @key_width = badge.customization.dig(:key_width)
+        end
+
+        def key_text
+          if @key_text && @key_text.size <= MAX_KEY_TEXT_SIZE
+            @key_text
+          else
+            @entity.to_s
+          end
+        end
+
+        def value_text
+          @status ? ("%.2f%%" % @status) : 'unknown'
+        end
+
+        def key_width
+          if @key_width && @key_width.between?(1, MAX_KEY_WIDTH)
+            @key_width
+          else
+            62
+          end
+        end
+
+        def value_width
+          @status ? 54 : 58
+        end
+
+        def value_color
+          case @status
+          when 95..100 then STATUS_COLOR[:good]
+          when 90..95 then STATUS_COLOR[:acceptable]
+          when 75..90 then STATUS_COLOR[:medium]
+          when 0..75 then STATUS_COLOR[:low]
+          else
+            STATUS_COLOR[:unknown]
+          end
+        end
+      end
+    end
+  end
+end