Adds missing Railtie

pehlert · Dec 28, 2013 · 54fe902 · 54fe902
1 parent a903f68
commit 54fe902
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/lib/promocode_generator.rb b/lib/promocode_generator.rb
@@ -1,5 +1,6 @@
 require "promocode_generator/version"
 require "promocode_generator/model_additions"
+require "promocode_generator/railtie" if defined? Rails
 
 module PromocodeGenerator
   # Generates an 8 character random code of uppercase letters and numbers

diff --git a/lib/promocode_generator/railtie.rb b/lib/promocode_generator/railtie.rb
@@ -0,0 +1,9 @@
+module PromocodeGenerator
+  class Railtie < Rails::Railtie
+    initializer 'promocode_generator.model_additions' do
+      ActiveSupport.on_load :active_record do
+        extend ModelAdditions
+      end
+    end
+  end
+end
diff --git a/lib/promocode_generator/version.rb b/lib/promocode_generator/version.rb
@@ -1,3 +1,3 @@
 module PromocodeGenerator
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end