lib/generators/inky/install_generator.rb in inky-rb-1.3.7.0 vs lib/generators/inky/install_generator.rb in inky-rb-1.3.7.1

@@ -4,24 +4,25 @@
   module Generators
     class InstallGenerator < ::Rails::Generators::Base
       desc 'Install Foundation for Emails'
       source_root File.join(File.dirname(__FILE__), 'templates')
       argument :layout_name, type: :string, default: 'mailer', banner: 'layout_name'
+      argument :extension,   type: :string, default: 'erb',    banner: 'extension'
 
       def preserve_original_mailer_layout
         return unless layout_name == 'mailer'
 
-        original_mailer = File.join(layouts_base_dir, 'mailer.html.erb')
+        original_mailer = File.join(layouts_base_dir, "mailer.html.#{extension}")
         rename_filename = File.join(layouts_base_dir, "old_mailer_#{Time.now.to_i}.html.erb")
         File.rename(original_mailer, rename_filename) if File.exist? original_mailer
       end
 
       def create_mailer_stylesheet
         template 'foundation_emails.scss', File.join(stylesheets_base_dir, 'foundation_emails.scss')
       end
 
       def create_mailer_layout
-        template 'mailer_layout.html.erb', File.join(layouts_base_dir, "#{layout_name.underscore}.html.erb")
+        template "mailer_layout.html.#{extension}", File.join(layouts_base_dir, "#{layout_name.underscore}.html.#{extension}")
       end
 
       private
 
       def stylesheets_base_dir