lib/identity/engine.rb in anadea-identity-0.1.0 vs lib/identity/engine.rb in anadea-identity-0.2.0

@@ -1,21 +1,22 @@
 require "devise"
+require "identity/mixins/user"
 require_relative "./devise_config"
-
-require "rolify"
-
 require_relative "./view_helper"
 
-require "kaminari"
-Kaminari::Hooks.init
-
 module Identity
   class Engine < ::Rails::Engine
-    initializer 'identity-views' do |app|
+    initializer "identity-views" do |app|
       ActionView::Base.include Identity::ViewHelper
     end
 
-    initializer 'identity-admin' do |app|
-      ActiveAdmin.application.load_paths.unshift(File.dirname(__FILE__) + '/admin')
+    initializer "identity-admin" do |app|
+      ActiveAdmin.application.load_paths.unshift(File.dirname(__FILE__) + "/admin")
+    end
+
+    initializer "identity-model" do
+      unless Identity.user_class.is_a?(Identity::Mixins::User)
+        Identity.user_class.include(Identity::Mixins::User)
+      end
     end
   end
 end