Proper Rails 5 Mime type check

Author: straydogstudio

README.md

@@ -205,6 +205,8 @@ class UserMailer < ActionMailer::Base
     xlsx = render_to_string layout: false, handlers: [:axlsx], formats: [:xlsx], template: "users/export", locals: {users: users}
     attachment = Base64.encode64(xlsx)
     attachments["Users.xlsx"] = {mime_type: Mime[:xlsx], content: attachment, encoding: 'base64'}
+    # For rails 4 use Mime::XLSX
+    # attachments["Users.xlsx"] = {mime_type: Mime::XLSX, content: attachment, encoding: 'base64'}
     # self.instance_variable_set(:@_lookup_context, nil) # If attachments are rendered as content, try this and open an issue
     ...
   end
@@ -244,6 +246,8 @@ class UserMailer < ActionMailer::Base
   def export(users)
     xlsx = render_to_string handlers: [:axlsx], formats: [:xlsx], template: "users/export", locals: {users: users}
     attachments["Users.xlsx"] = {mime_type: Mime[:xlsx], content: xlsx, encoding: 'base64'}
+    # For Rails 4 use Mime::XLSX
+    # attachments["Users.xlsx"] = {mime_type: Mime::XLSX, content: xlsx, encoding: 'base64'}
     # self.instance_variable_set(:@_lookup_context, nil) # If attachments are rendered as content, try this and open an issue
     ...
   end

lib/axlsx_rails/action_controller.rb

@@ -1,6 +1,13 @@
 require 'action_controller'
-unless defined? Mime[:xlsx]
-	Mime::Type.register "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", :xlsx
+
+if Rails.version.to_f >= 5
+  unless Mime[:xlsx]
+  	Mime::Type.register "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", :xlsx
+  end
+else
+  unless defined? Mime::XLSX
+    Mime::Type.register "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", :xlsx
+  end
 end
 
 ActionController::Renderers.add :xlsx do |filename, options|
@@ -42,7 +49,7 @@
     options[:locals][:xlsx_use_shared_strings] = options.delete(:xlsx_use_shared_strings)
   end
 
-  mime = Mime[:xlsx]
+  mime = Rails.version.to_f >= 5 ? Mime[:xlsx] : Mime::XLSX
   send_data render_to_string(options), :filename => file_name, :type => mime, :disposition => disposition
 end
 

lib/axlsx_rails/template_handler.rb

@@ -3,7 +3,7 @@ module ActionView
   module Template::Handlers
     class AxlsxBuilder
       def default_format
-        Mime[:xlsx]
+        Rails.version.to_f >= 5 ? Mime[:xlsx] : Mime::XLSX
       end
 
       def self.call(template)

spec/dummy/app/mailers/notifier.rb

@@ -6,7 +6,7 @@ def instructions(user)
 
     # normal syntax
     xlsx = render_to_string handlers: [:axlsx], template: 'users/send_instructions', layout: false
-    attachments["user_#{user.id}.xlsx"] = {mime_type: Mime[:xlsx], content: xlsx}
+    attachments["user_#{user.id}.xlsx"] = {mime_type: Mime::XLSX, content: xlsx}
 
     mail :to => user.email, :subject => 'Instructions'
   end

spec/dummy_4/app/mailers/notifier.rb

@@ -6,7 +6,7 @@ def instructions(user)
 
     # normal syntax
     xlsx = render_to_string handlers: [:axlsx], template: 'users/send_instructions', layout: false, formats: [:xlsx]
-    attachments["user_#{user.id}.xlsx"] = {mime_type: Mime[:xlsx], content: xlsx}
+    attachments["user_#{user.id}.xlsx"] = {mime_type: Mime::XLSX, content: xlsx}
 
     mail :to => user.email, :subject => 'Instructions'
   end

spec/spec_helper.rb

@@ -40,5 +40,5 @@ class ExampleGroup
 end
 
 def mime_type
-  Mime[:xlsx]
+  Rails.version.to_f >= 5 ? Mime[:xlsx] : Mime::XLSX
 end