lib/grape-rabl/formatter.rb in grape-rabl-0.3.1 vs lib/grape-rabl/formatter.rb in grape-rabl-0.4.0

@@ -1,87 +1,89 @@
 require 'json'
 
 module Grape
-  module Formatter
-    module Rabl
+  module Rabl
+    class Formatter
       class << self
-        attr_reader :env
-        attr_reader :endpoint
+        def tilt_cache
+          @tilt_cache ||= ::Tilt::Cache.new
+        end
+      end
 
-        def call(object, env)
-          @env = env
-          @endpoint = env['api.endpoint']
+      attr_reader :env, :endpoint, :object
 
-          if rablable?
-            rabl do |template|
-              engine = tilt_template(template)
-              output = engine.render endpoint, locals
-              if layout_template
-                layout_template.render(endpoint) { output }
-              else
-                output
-              end
+      def initialize(object, env)
+        @env      = env
+        @endpoint = env['api.endpoint']
+        @object   = object
+      end
+
+      def render
+        if rablable?
+          rabl do |template|
+            engine = tilt_template(template)
+            output = engine.render endpoint, locals
+            if layout_template
+              layout_template.render(endpoint) { output }
+            else
+              output
             end
-          else
-            Grape::Formatter::Json.call object, env
           end
+        else
+          Grape::Formatter::Json.call object, env
         end
+      end
 
-        private
+      private
 
-        def view_path(template)
-          if template.split('.')[-1] == 'rabl'
-            File.join(env['api.tilt.root'], template)
-          else
-            File.join(env['api.tilt.root'], (template + '.rabl'))
-          end
+      def view_path(template)
+        if template.split('.')[-1] == 'rabl'
+          File.join(env['api.tilt.root'], template)
+        else
+          File.join(env['api.tilt.root'], (template + '.rabl'))
         end
+      end
 
-        def rablable?
-          !!rabl_template
-        end
+      def rablable?
+        !!rabl_template
+      end
 
-        def rabl
-          fail 'missing rabl template' unless rabl_template
-          set_view_root unless env['api.tilt.root']
-          yield rabl_template
-        end
+      def rabl
+        fail 'missing rabl template' unless rabl_template
+        set_view_root unless env['api.tilt.root']
+        yield rabl_template
+      end
 
-        def locals
-          env['api.tilt.rabl_locals'] || endpoint.options[:route_options][:rabl_locals] || {}
-        end
+      def locals
+        env['api.tilt.rabl_locals'] || endpoint.options[:route_options][:rabl_locals] || {}
+      end
 
-        def rabl_template
-          env['api.tilt.rabl'] || endpoint.options[:route_options][:rabl]
-        end
+      def rabl_template
+        env['api.tilt.rabl'] || endpoint.options[:route_options][:rabl]
+      end
 
-        def set_view_root
-          fail "Use Rack::Config to set 'api.tilt.root' in config.ru"
-        end
+      def set_view_root
+        fail "Use Rack::Config to set 'api.tilt.root' in config.ru"
+      end
 
-        def tilt_template(template)
-          if Grape::Rabl.configuration.cache_template_loading
-            tilt_cache.fetch(template) { ::Tilt.new(view_path(template), tilt_options) }
-          else
-            ::Tilt.new(view_path(template), tilt_options)
-          end
+      def tilt_template(template)
+        if Grape::Rabl.configuration.cache_template_loading
+          Grape::Rabl::Formatter.tilt_cache.fetch(template) { ::Tilt.new(view_path(template), tilt_options) }
+        else
+          ::Tilt.new(view_path(template), tilt_options)
         end
+      end
 
-        def tilt_cache
-          @tilt_cache ||= ::Tilt::Cache.new
-        end
+      def tilt_options
+        { format: env['api.format'], view_path: env['api.tilt.root'] }
+      end
 
-        def tilt_options
-          { format: env['api.format'], view_path: env['api.tilt.root'] }
-        end
-
-        def layout_template
-          layout_path = view_path(env['api.tilt.layout'] || 'layouts/application')
-          if Grape::Rabl.configuration.cache_template_loading
-            tilt_cache.fetch(layout_path) { ::Tilt.new(layout_path, tilt_options) if File.exist?(layout_path) }
-          else
-            ::Tilt.new(layout_path, tilt_options) if File.exist?(layout_path)
-          end
+      def layout_template
+        layout_path = view_path(env['api.tilt.layout'] || 'layouts/application')
+        if Grape::Rabl.configuration.cache_template_loading
+          Grape::Rabl::Formatter.tilt_cache.fetch(layout_path) { ::Tilt.new(layout_path, tilt_options) if File.exist?(layout_path) }
+        else
+          ::Tilt.new(layout_path, tilt_options) if File.exist?(layout_path)
         end
       end
     end
   end
 end