lib/cans/historian.rb in cans-0.1.1 vs lib/cans/historian.rb in cans-0.1.2

@@ -1,24 +1,49 @@
 module Cans
   class Historian
     attr_accessor :enabled
+    attr_accessor :history
     def initialize
+      self.history = Set.new
+      self.enabled = false
       try_history
+      try_bugging_rails
     end
 
     def delve
-      reload_history if enabled
+      return unless enabled
+      merge_history
+      reload_history
     end
 
+    def record
+      merge_history
+    end
+
     private
+    def merge_history
+      history.merge ActiveSupport::Dependencies.history
+    end
+
     def reload_history
-      ActiveSupport::Dependencies.history.each { |f| load f }
+      history.each { |f| require_or_load f }
     end
 
     def try_history
       ActiveSupport::Dependencies.history
-      enabled = true
+      self.enabled = true
     rescue
-      enabled = false
+    end
+
+    def try_bugging_rails
+      this_historian = self
+      ApplicationController.instance_eval do
+        after_filter :save_history
+        define_method :save_history do
+          this_historian.record
+        end
+      end
+      self.enabled = true
+    rescue => e
     end
   end
 end