lib/mail_engine/engine.rb in mail_engine-0.0.1 vs lib/mail_engine/engine.rb in mail_engine-0.1.0

@@ -20,23 +20,23 @@
 
     initializer "mail_engine.add_acts_as_mail_receiver" do
       ActiveRecord::Base.send(:include, MailEngine::ActsAsMailReceiver)
     end
 
-    # initializer "mail_engine.register_mail_log" do
-    #   ActionMailer::Base.register_interceptor MailEngine::MailLog
-    # end
-
     rake_tasks do
       load "mail_engine/tasks/sendmail.rake"
     end
 
     generators do
+      require 'mail_engine/generators/install.rb'
       require 'mail_engine/generators/create_config.rb'
-      require 'mail_engine/generators/create_migration.rb'
+      require 'mail_engine/generators/copy_resources.rb'
+      require 'mail_engine/generators/copy_migrations.rb'
     end
 
     config.to_prepare do
       MailEngine::Configuration.load
+      raise "Please add :user_class_name config into mail_engine_config.yml." if MailEngine::Base.current_config["user_class_name"].blank?
+      MailEngine::USER_MODEL = MailEngine::Base.current_config["user_class_name"].constantize unless defined?(MailEngine::USER_MODEL)
     end
   end
 end
\ No newline at end of file