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

@@ -1,9 +1,11 @@
 module MailEngine
   class Engine < Rails::Engine
     require 'mail_engine'
-
+    require 'carrierwave'
+    require 'kaminari'
+    
     initializer "mail_engine" do
       ActionMailer::Base.send(:include, MailEngine::Sendgrid::Base)
       require 'mail_engine/action_mailer_patch'
     end
 
@@ -15,11 +17,11 @@
 
     initializer "mail_engine.register_database_template" do
       ActionMailer::Base.layout "layouts/mail_engine/mail_template_layouts/none"
       ActionMailer::Base.send(:prepend_view_path, MailEngine::MailTemplate::Resolver.instance)
     end
-
+    
     initializer "mail_engine.add_acts_as_mail_receiver" do
       ActiveRecord::Base.send(:include, MailEngine::ActsAsMailReceiver)
     end
 
     rake_tasks do
@@ -32,11 +34,18 @@
       require 'mail_engine/generators/copy_resources.rb'
       require 'mail_engine/generators/copy_migrations.rb'
     end
 
     config.to_prepare do
+      # load config
       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)
+
+      # don't process config file if not fine it.
+      if MailEngine::Base.current_config.present?
+        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)
+      else
+        puts "\e[1;31;40m[Mail Engine Warning]\e[0m Not found mail_engine_config.yml, so mail_engine won't be able to work."
+      end
     end
   end
 end
\ No newline at end of file