add mechanism for patching Rails

davejacobs · Sep 30, 2012 · 110c8aa · 110c8aa
1 parent 2c58c81
commit 110c8aa
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/letters/patch/object.rb b/lib/letters/patch/object.rb
@@ -1,4 +1,3 @@
 require "letters/patch"
 
 Letters.patch! Object
-Letters.patch! NilClass
diff --git a/lib/letters/patch/rails.rb b/lib/letters/patch/rails.rb
@@ -0,0 +1,9 @@
+require "letters/core"
+
+begin
+  Letters.patch! ActiveRecord::Base
+  Letters.patch! ActionController::Base
+  Letters.patch! ActionMailer::Base
+rescue
+  $stderr.puts "[warning] tried to patch Rails without Rails being loaded"
+end