re-work callout tag conditional

Output a warning if the type is not supported and default to info

1 files changed, 5 insertions, 6 deletions

diff --git a/docs/_plugins/callout.rb b/docs/_plugins/callout.rb
index 29ecbc99d..536ca588e 100644
--- a/docs/_plugins/callout.rb
+++ b/docs/_plugins/callout.rb
@@ -6,12 +6,11 @@ module Jekyll
 
       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