lib/express_translate.rb in express_translate-1.0.3 vs lib/express_translate.rb in express_translate-1.0.4

@@ -11,10 +11,11 @@
 require 'seeds/languages_seed'
 require 'seeds/accounts_seed'
 
 # Lib
 require 'redis'
+require 'rails'
 
 # Main module for my gem
 module ExpressTranslate
   
   class << self; attr_accessor :package, :language, :url end
@@ -22,21 +23,23 @@
   @package = ""
   @language = ""
   @url = ""
   
   # Change language locals for I18N
-  # lang: is a code of language want to change for backend
+  # @lang: is a code of language want to change for backend
   def language(lang)
     I18n.locale = "#{YAML.load_file(Rails.root.to_s + '/config/express_translate.yml')['package']['id']}#{lang}"
   end
   
   def self.root
     File.expand_path '../..', __FILE__
   end
   
   def self.config
-    YAML.load_file(Rails.root.to_s + "/config/express_translate.yml")
+    file_name = Rails.root.to_s + "/config/express_translate.yml"
+    file_name = File.exist?(file_name) ? file_name : self.root + "/lib/generators/express_translate/templates/config/express_translate.yml"
+    return YAML.load_file(file_name)
   end
   
   def self.seeds
     if Package.all.count == 0
       Package.seed
@@ -64,18 +67,9 @@
   end
   
   class Engine < Rails::Engine
     TRANSLATION_STORE = Redis.new
     I18n.backend = I18n::Backend::Chain.new(I18n::Backend::KeyValue.new(TRANSLATION_STORE), I18n.backend)
+    I18n.enforce_available_locales = false
     I18n.locale = "been"
-    
-    config.generators do |g|
-      g.test_framework :rspec, fixture: true
-      g.fixture_replacement :factory_girl, dir: 'spec/factories'
-      g.view_specs false
-      g.helper_specs false
-      g.stylesheets = false
-      g.javascripts = false
-      g.helper = false
-    end
   end
 end
\ No newline at end of file