lib/svgshield.rb in svgshield-0.1.1 vs lib/svgshield.rb in svgshield-0.1.2

@@ -1,13 +1,16 @@
 require 'rasem'
+require 'color'
 require "svgshield/version"
 
 class Svgshield
+  attr_reader :shield
+
   def initialize(subject, status, color = '#aaa')
     color = Color::CSS[color] if color[0] == '#'
 
-    Rasem::SVGImage.new(width: 90, height: 20) do
+    @shield = Rasem::SVGImage.new(width: 90, height: 20) do
       linearGradient('a', x2: 0, y2: '100%') do
         stop 0, '#bbb', '.1'
         stop 1, nil, '.1'
       end
 
@@ -26,7 +29,11 @@
         text(19.5, 14, fill: '#fff') { raw subject }
         text(62.5, 15, fill: '#010101', 'fill-opacity' => '.3') { raw status }
         text(62.5, 14, fill: '#fff') { raw status }
       end
     end
+  end
+
+  def to_s
+    shield.to_s
   end
 end