Move renderer call out of railtie into base file. Fix syntax issues i…

…n renderer.

lib/inertia.rb

@@ -0,0 +1,16 @@
+require 'inertia/renderer'
+require 'inertia/railtie'
+
+ActionController::Renderers.add :inertia do |component, options|
+  Inertia::Renderer.new(
+    component,
+    request,
+    response,
+    method(:render),
+    props: options[:props],
+    view_data: options[:view_data],
+  ).render
+end
+
+module Inertia
+end

lib/inertia/railtie.rb

@@ -14,13 +14,5 @@ class Railtie < Rails::Railtie
       app.config.autoload_paths << Inertia::Configuration.path
     end
 
-    config.after_initialize do
-      require 'inertia/renderer'
-      require 'action_controller/metal/renderers'
-
-      ActionController::Renderers.add :inertia do |component, options|
-        Inertia::Renderer.new(component, props: options[:props], view_data: options[:view_data]).render
-      end
-    end
   end
 end

lib/inertia/renderer.rb

@@ -1,31 +1,36 @@
+require_relative "inertia"
+
 module Inertia
   class Renderer
     attr_reader :component, :view_data
 
-    def initialize(component, props:, view_data:)
+    def initialize(component, request, response, render_method, props:, view_data:)
       @component = component
+      @request = request
+      @response = response
+      @render_method = render_method
       @props = props || {}
       @view_data = view_data || {}
     end
 
     def render
-      if request.headers['X-Inertia']
-        response.set_header('Vary', 'Accept')
-        response.set_header('X-Inertia', 'true')
-        render json: page, status: 200
+      if @request.headers['X-Inertia']
+        @response.set_header('Vary', 'Accept')
+        @response.set_header('X-Inertia', 'true')
+        @render_method.call json: page, status: 200
       else
-        render template: 'inertia', layout: Inertia.layout, locals: (view_data).merge({page: page}) 
+        @render_method.call template: 'inertia', layout: Inertia.layout, locals: (view_data).merge({page: page})
       end
     end
 
     private
 
     def props
-      only = (request.headers['X-Inertia-Partial-Data'] || '').split(',').compact.map(&:to_sym)
+      only = (@request.headers['X-Inertia-Partial-Data'] || '').split(',').compact.map(&:to_sym)
 
       _props = Inertia.instance.shared_data.merge(@props)
 
-      _props = (only.any? && request.headers['X-Inertia-Partial-Component'] == component) ?
+      _props = (only.any? && @request.headers['X-Inertia-Partial-Component'] == component) ?
         _props.select {|key| key.in? only} :
         _props
 
@@ -36,15 +41,15 @@ def page
       {
         component: component,
         props: props,
-        url: request.original_url,
+        url: @request.original_url,
         version: Inertia.version,
       }
     end
-  end
 
-  def deep_transform_values(hash, proc)
-    return proc.call(hash) unless hash.is_a? Hash
-
-    hash.transform_values {|value| deep_transform_values(value, proc)}
+    def deep_transform_values(hash, proc)
+      return proc.call(hash) unless hash.is_a? Hash
+
+      hash.transform_values {|value| deep_transform_values(value, proc)}
+    end
   end
 end