Merge pull request #109 from twbs/callout-if

re-work callout tag conditional
twbs · Apr 24, 2015 · 20543d5 · 20543d5
2 parents 4c3c402 + 58c208e
commit 20543d5
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/docs/_plugins/callout.rb b/docs/_plugins/callout.rb
@@ -6,12 +6,11 @@ class CalloutTag < Liquid::Block
 
       def initialize(tag_name, type, tokens)
         super
-        @type = type
-        if type == "danger"
-          @type = "danger"
-        elsif type == "warning"
-          @type = "warning"
-        elsif type == "info"
+        type.strip!
+        if %w(info danger warning).include?(type)
+          @type = type
+        else
+          puts "#{type} callout not supported. Defaulting to info"
           @type = "info"
         end
       end