re-work callout tag conditional

Output a warning if the type is not supported and default to info
twbs · Apr 24, 2015 · 58c208e · 58c208e
1 parent 4c3c402
commit 58c208e
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