lib/generators/ecm/contact/install/install_generator.rb in ecm_contact2-5.0.1 vs lib/generators/ecm/contact/install/install_generator.rb in ecm_contact2-5.1.0

@@ -4,17 +4,22 @@
       class InstallGenerator < Rails::Generators::Base
         desc 'Generates the intializer'
 
         source_root File.expand_path('../templates', __FILE__)
 
+        attr_reader :base_controller_class_name
+
+        def initialize(*args)
+          super
+          @base_controller_class_name = ENV.fetch('BASE_CONTROLLER_CLASS_NAME') { '::FrontendController' }
+        end
+
         def generate_initializer
-          copy_file 'initializer.rb', 'config/initializers/ecm_contact.rb'
+          template 'initializer.rb', 'config/initializers/ecm_contact.rb'
         end
 
         def generate_routes
-          inject_into_file 'config/routes.rb', before: "\nend" do
-            File.read(File.join(File.expand_path('../templates', __FILE__), 'routes.source'))
-          end
+          route File.read(File.join(File.expand_path('../templates', __FILE__), 'routes.source'))
         end
       end
     end
   end
 end