lib/human_power.rb in human_power-0.0.2 vs lib/human_power.rb in human_power-0.0.3

@@ -1,9 +1,8 @@
 require "human_power/version"
 require "human_power/generator"
 require "human_power/rule"
-require "human_power/user_agents"
 require "human_power/rails" if defined?(Rails)
 
 module HumanPower
   class << self
     # Yields a configuration block.
@@ -18,10 +17,19 @@
     # Registers a user agent.
     def register_user_agent(key, user_agent_string)
       user_agents[key] = user_agent_string
     end
 
+    # Hash of registered user agents.
     def user_agents
-      @user_agents ||= DEFAULT_USER_AGENTS
+      @user_agents ||= load_user_agents
     end
+
+  private
+
+      # Loads the built-in user agents from crawlers.yml.
+      def load_user_agents
+        path = File.expand_path("../../user_agents.yml", __FILE__)
+        Hash[YAML.load(open(path).read).map { |k, v| [k.to_sym, v] }]
+      end
   end
 end
\ No newline at end of file