Class: Angelo::Responder

Inherits:
Object
  • Object
show all
Includes:
Celluloid::Logger
Defined in:
lib/angelo/responder.rb

Direct Known Subclasses

WebsocketResponder

Class Attribute Summary collapse

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(&block) ⇒ Responder

Returns a new instance of Responder.



38
39
40
 
# File 'lib/angelo/responder.rb', line 38

def initialize &block
  @response_handler = Base.compile! :request_handler, &block
end

Class Attribute Details

.default_headers ⇒ Object 



24
25
26
27
 
# File 'lib/angelo/responder.rb', line 24

def default_headers
  @default_headers ||= DEFAULT_RESPONSE_HEADERS
  @default_headers
end

Instance Attribute Details

#connection=(value) ⇒ Object (writeonly)

Sets the attribute connection

Parameters:

  • value —

    the value to set the attribute connection to.



35
36
37
 
# File 'lib/angelo/responder.rb', line 35

def connection=(value)
  @connection = value
end

#request ⇒ Object

Returns the value of attribute request.



36
37
38
 
# File 'lib/angelo/responder.rb', line 36

def request
  @request
end

Class Method Details

.content_type(type) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/angelo/responder.rb', line 12

def content_type type
  dhs = self.default_headers
  case type
  when :json
    self.default_headers = dhs.merge CONTENT_TYPE_HEADER_KEY => JSON_TYPE
  when :html
    self.default_headers = dhs.merge CONTENT_TYPE_HEADER_KEY => HTML_TYPE
  else
    raise ArgumentError.new "invalid content_type: #{type}"
  end
end

.symhash ⇒ Object 



29
30
31
 
# File 'lib/angelo/responder.rb', line 29

def symhash
  Hash.new {|hash,key| hash[key.to_s] if Symbol === key }
end

Instance Method Details

#base=(base) ⇒ Object 



42
43
44
45
 
# File 'lib/angelo/responder.rb', line 42

def base= base
  @base = base
  @base.responder = self
end

#content_type(type) ⇒ Object 



105
106
107
108
109
110
111
112
113
114
 
# File 'lib/angelo/responder.rb', line 105

def content_type type
  case type
  when :json
    headers CONTENT_TYPE_HEADER_KEY => JSON_TYPE
  when :html
    headers CONTENT_TYPE_HEADER_KEY => HTML_TYPE
  else
    raise ArgumentError.new "invalid content_type: #{type}"
  end
end

#error_message(_error) ⇒ Object 



85
86
87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/angelo/responder.rb', line 85

def error_message _error
  case
  when respond_with?(:json)
    { error: _error.message }.to_json
  else
    case _error.message
    when Hash
      _error.message.to_s
    else
      _error.message
    end
  end
end

#handle_error(_error, type = :internal_server_error, report = true) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
 
# File 'lib/angelo/responder.rb', line 74

def handle_error _error, type = :internal_server_error, report = true
  err_msg = error_message _error
  Angelo.log @connection, @request, nil, type, err_msg.size
  @connection.respond type, headers, err_msg
  @connection.close
  if report
    error "#{_error.class} - #{_error.message}"
    ::STDERR.puts _error.backtrace
  end
end

#handle_request ⇒ Object 



55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/angelo/responder.rb', line 55

def handle_request
  if @response_handler
    @base.before if @base.respond_to? :before
    @body = catch(:halt) { @response_handler.bind(@base).call || '' }
    @base.after if @base.respond_to? :after
    respond
  else
    raise NotImplementedError
  end
rescue JSON::ParserError => jpe
  handle_error jpe, :bad_request, false
rescue FormEncodingError => fee
  handle_error fee, :bad_request, false
rescue RequestError => re
  handle_error re, re.type, false
rescue => e
  handle_error e
end

#headers(hs = nil) ⇒ Object 



99
100
101
102
103
 
# File 'lib/angelo/responder.rb', line 99

def headers hs = nil
  @headers ||= self.class.default_headers.dup
  @headers.merge! hs if hs
  @headers
end

#redirect(url) ⇒ Object 



156
157
158
 
# File 'lib/angelo/responder.rb', line 156

def redirect url
  @redirect = url
end

#respond ⇒ Object 



125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
 
# File 'lib/angelo/responder.rb', line 125

def respond
  status = nil
  case @body
  when HALT_STRUCT
    status = @body.status
    @body = @body.body
    if Hash === @body
      @body = {error: @body} if status != :ok or status < 200 && status >= 300
      @body = @body.to_json if respond_with? :json
    end

  when String
    JSON.parse @body if respond_with? :json # for the raises

  when Hash
    raise 'html response requires String' if respond_with? :html
    @body = @body.to_json if respond_with? :json

  when NilClass
    @body = EMPTY_STRING
  end

  status ||= @redirect.nil? ? :ok : :moved_permanently
  headers LOCATION_HEADER_KEY => @redirect if @redirect

  Angelo.log @connection, @request, nil, status, @body.size
  @connection.respond status, headers, @body
rescue => e
  handle_error e, :internal_server_error, false
end

#respond_with?(type) ⇒ Boolean

Returns:

  • (Boolean) 


116
117
118
119
120
121
122
123
 
# File 'lib/angelo/responder.rb', line 116

def respond_with? type
  case headers[CONTENT_TYPE_HEADER_KEY]
  when JSON_TYPE
    type == :json
  else
    type == :html
  end
end