lib/a_b/boot.rb in a_b-0.1.0 vs lib/a_b/boot.rb in a_b-0.1.1

@@ -23,8 +23,13 @@
   
   # Rack flash
   use Rack::Flash, :accessorize => %w(error notice success)
   
   # Generate token
-  require File.dirname(__FILE__) + '/model/token'
+  require "#{root}/lib/a_b/model/token"
   Token.generate!
+  
+  # Hoptoad notifier
+  if File.exists?(hoptoad = "#{root}/config/hoptoad.txt")
+    use Rack::HoptoadNotifier, File.read(hoptoad)
+  end
 end