lib/bullet.rb in flyerhzm-bullet-1.4.1 vs lib/bullet.rb in flyerhzm-bullet-1.5.0

@@ -1,23 +1,71 @@
 require 'bulletware'
 
 module Bullet
-  @@enable = nil
-
   class <<self
+    attr_accessor :enable, :alert, :console, :growl, :growl_password, :rails_logger, :bullet_logger, :logger, :logger_file
+
     def enable=(enable)
-      @@enable = enable
+      @enable = enable
       if enable? 
         Bullet::ActiveRecord.enable
         ActionController::Dispatcher.middleware.use Bulletware
       end
     end
 
     def enable?
-      @@enable == true
+      @enable == true
     end
+
+    def growl=(growl)
+      if growl
+        begin
+          require 'ruby-growl'
+          growl = Growl.new('localhost', 'ruby-growl', ['Bullet Notification'], nil, @@growl_password)
+          growl.notify('Bullet Notification', 'Bullet Notification', 'Bullet Growl notifications have been turned on')
+        rescue MissingSourceFile
+          raise NotificationError.new('You must install the ruby-growl gem to use Growl notifications: `sudo gem install ruby-growl`')
+        end
+      end
+      @growl = growl
+    end
+
+    def bullet_logger=(bullet_logger)
+      if @bullet_logger = bullet_logger
+        @logger_file = File.open(Bullet::BulletLogger::LOG_FILE, 'a+')
+        @logger = Bullet::BulletLogger.new(@logger_file)
+      end
+    end
+
+    BULLETS = [Bullet::Association, Bullet::Counter]
+
+    def start_request
+      BULLETS.each {|bullet| bullet.start_request}
+    end
+
+    def end_request
+      BULLETS.each {|bullet| bullet.end_request}
+    end
+
+    def notification?
+      BULLETS.any? {|bullet| bullet.notification?}
+    end
+
+    def javascript_notification
+      BULLETS.collect {|bullet| bullet.javascript_notification if bullet.notification?}.join("\n")
+    end
+
+    def growl_notification
+      BULLETS.each {|bullet| bullet.growl_notification if bullet.notification?}
+    end
+
+    def log_notification(path)
+      BULLETS.each {|bullet| bullet.log_notification(path) if bullet.notification?}
+    end
   end
 
   autoload :ActiveRecord, 'bullet/active_record'
   autoload :Association, 'bullet/association'
+  autoload :Counter, 'bullet/counter'
   autoload :BulletLogger, 'bullet/logger'
+  autoload :Notification, 'bullet/notification'
 end