lib/turbolinks-animate/railtie.rb in turbolinks-animate-2.0.1 vs lib/turbolinks-animate/railtie.rb in turbolinks-animate-2.0.2

@@ -1,28 +1,29 @@
+# frozen_string_literal: true
+
 require 'rails/railtie'
 require 'mozaic'
 
 module TurbolinksAnimate
-    class Railtie < Rails::Railtie
+  class Railtie < Rails::Railtie
+    initializer 'turbolinks-animate.mozaic' do
+      Mozaic.configure do |config|
+        config.define_component('turbolinks-animate',
+                                element: 'body') do |options|
+          animation = options.delete :animation
+          return options unless animation
 
-        initializer 'turbolinks-animate.mozaic' do
-            Mozaic.configure do |config|
-                config.define_component 'turbolinks-animate', element: 'body' do |options|
-                    animation = options.delete :animation
-                    if animation
-                        options[:class] += ( options[:class].length > 0 ? ' turbolinks-animate' : 'turbolinks-animate' )
-                        options[:data] ||= {}
-                        options[:data][:turbolinks_animate_animation] = animation
-                    end
-                    options
-                end
-            end
+          options[:class] = [options[:class], 'turbolinks-animate'].compact
+                                                                   .join(' ')
+          options[:data] ||= {}
+          options[:data][:turbolinks_animate_animation] = animation
         end
+      end
+    end
 
-        initializer 'turbolinks-animate.action_controller' do
-            ActiveSupport.on_load :action_controller do
-                include TurbolinksAnimateHelper
-            end
-        end
-
+    initializer 'turbolinks-animate.action_controller' do
+      ActiveSupport.on_load :action_controller do
+        include TurbolinksAnimateHelper
+      end
     end
+  end
 end