forked from postrank-labs/goliath
-
Notifications
You must be signed in to change notification settings - Fork 1
/
builder.rb
80 lines (71 loc) · 2.24 KB
/
builder.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
require 'http_router'
module Goliath
module Rack
class Builder < ::Rack::Builder
attr_accessor :params
attr_reader :inner_app
include Params::Parser
alias_method :original_run, :run
def run(app)
@inner_app = app
original_run(app)
end
module MappingHandlers
include Constants
attr_reader :last_app
def defer
defer = EM::DefaultDeferrable.new
Thread.current[GOLIATH_ENV].defer_stack << defer
defer
end
def on_body(env, body)
defer.callback do |api|
api.on_body(env, body) if api.respond_to?(:on_body)
end
end
def on_headers(env, headers)
defer.callback do |api|
api.on_headers(env, headers) if api.respond_to?(:on_headers)
end
end
def on_close(env)
defer.callback do |api|
api.on_close(env) if api.respond_to?(:on_close)
end
end
end
# Builds the rack middleware chain for the given API
#
# @param klass [Class] The API class to build the middlewares for
# @param api [Object] The instantiated API
# @return [Object] The Rack middleware chain
def self.build(klass, api)
Builder.app do
klass.middlewares.each do |mw_klass, args, blk|
use(mw_klass, *args, &blk)
end
if klass.maps?
klass.instance_eval "include MappingHandlers", __FILE__, __LINE__
klass.maps.each do |path, route_klass, opts, blk|
blk ||= Proc.new {
run Builder.build(route_klass, route_klass.new)
}
klass.router.add(path, opts.dup).to {|env|
builder = Builder.new
env['params'] ||= {}
env['params'].merge!(env['router.params']) if env['router.params']
builder.params = builder.retrieve_params(env)
builder.instance_eval(&blk)
env.defer_stack.each{ |defer| defer.succeed(builder.inner_app)}
builder.to_app.call(env)
}
end
run klass.router
else
run api
end
end
end
end
end
end