lib/middleman-robots/extension.rb in middleman-robots-0.1.0 vs lib/middleman-robots/extension.rb in middleman-robots-1.0.0

@@ -23,12 +23,14 @@
         return '' if rules.empty?
 
         data = []
         rules.each do |rule|
           row = []
-          if (rule["user-agent"])
-            row << "User-Agent: #{rule["user-agent"]}"
+
+          if (rule["user-agent"] || rule[:user_agent])
+            user_agent = rule[:user_agent] || rule["user-agent"]
+            row << "User-Agent: #{user_agent}"
           end
 
           if (rule[:disallow])
             rule[:disallow].each do |path|
               path = "/" + path unless /^\// =~ path
@@ -53,5 +55,6 @@
         path ? "Sitemap: #{path}" : ""
       end
     end
   end
 end
+