Class: Goliath::Rack::Render
- Inherits:
-
Object
- Object
- Goliath::Rack::Render
- Includes:
- Rack::RespondTo
- Defined in:
- lib/goliath/rack/render.rb
Overview
The render middleware will set the Content-Type of the response based on the provided HTTP_ACCEPT headers.
Instance Method Summary collapse
- #call(env) ⇒ Object
- #get_content_type(env) ⇒ Object
-
#initialize(app, types = nil) ⇒ Render
constructor
A new instance of Render.
- #post_process(env, status, headers, body) ⇒ Object
Constructor Details
#initialize(app, types = nil) ⇒ Render
Returns a new instance of Render.
15 16 17 18 |
# File 'lib/goliath/rack/render.rb', line 15 def initialize(app, types = nil) @app = app ::Rack::RespondTo.media_types = [types].flatten if types end |
Instance Method Details
#call(env) ⇒ Object
20 21 22 23 24 25 26 27 28 29 |
# File 'lib/goliath/rack/render.rb', line 20 def call(env) async_cb = env['async.callback'] env['async.callback'] = Proc.new do |status, headers, body| async_cb.call(post_process(env, status, headers, body)) end status, headers, body = @app.call(env) post_process(env, status, headers, body) end |
#get_content_type(env) ⇒ Object
52 53 54 55 56 57 58 59 60 61 62 63 |
# File 'lib/goliath/rack/render.rb', line 52 def get_content_type(env) fmt = env.params['format'] fmt = fmt.last if fmt.is_a?(Array) type = if fmt.nil? || fmt =~ /^\s*$/ ::Rack::RespondTo.selected_media_type else ::Rack::RespondTo::MediaType(fmt) end "#{type}; charset=utf-8" end |
#post_process(env, status, headers, body) ⇒ Object
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
# File 'lib/goliath/rack/render.rb', line 31 def post_process(env, status, headers, body) ::Rack::RespondTo.env = env # the respond_to block is what actually triggers the # setting of selected_media_type, so it's required respond_to do |format| format.json { body } format.html { body } format.xml { body } format.rss { body } format.js { body } end extra = { 'Content-Type' => get_content_type(env), 'Server' => 'PostRank Goliath API Server', 'Vary' => [headers.delete('Vary'), 'Accept'].compact.join(',') } [status, extra.merge(headers), body] end |