Class: Angelo::Responder

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

Direct Known Subclasses

Eventsource, Websocket

Defined Under Namespace

Classes: Eventsource, Websocket

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.



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

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

#base=(value) ⇒ Object (writeonly)

Sets the attribute base

Parameters:

  • value —

    the value to set the attribute base to.



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

def base=(value)
  @base = value
end

#connection ⇒ Object

Returns the value of attribute connection.



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

def connection
  @connection
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

#content_type(type) ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/angelo/responder.rb', line 111

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

#error_message(_error) ⇒ Object 



91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/angelo/responder.rb', line 91

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 = @base.report_errors?) ⇒ Object 



80
81
82
83
84
85
86
87
88
89
 
# File 'lib/angelo/responder.rb', line 80

def handle_error _error, type = :internal_server_error, report = @base.report_errors?
  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 



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
 
# File 'lib/angelo/responder.rb', line 51

def handle_request
  if @response_handler
    @base.before if @base.respond_to? :before
    @body = catch(:halt) { @response_handler.bind(@base).call || EMPTY_STRING }

    # TODO any real reason not to run afters with SSE?
    case @body
    when HALT_STRUCT
      if @body.body != :sse and @base.respond_to? :after
        @base.after
      end
    else
      @base.after if @base.respond_to? :after
    end

    respond
  else
    raise NotImplementedError
  end
rescue JSON::ParserError => jpe
  handle_error jpe, :bad_request
rescue FormEncodingError => fee
  handle_error fee, :bad_request
rescue RequestError => re
  handle_error re, re.type
rescue => e
  handle_error e
end

#headers(hs = nil) ⇒ Object 



105
106
107
108
109
 
# File 'lib/angelo/responder.rb', line 105

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

#redirect(url) ⇒ Object 



166
167
168
 
# File 'lib/angelo/responder.rb', line 166

def redirect url
  @redirect = url
end

#respond ⇒ Object 



133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
 
# File 'lib/angelo/responder.rb', line 133

def respond
  status = nil
  case @body
  when HALT_STRUCT
    status = @body.status
    @body = @body.body
    @body = nil if @body == :sse
    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
  size = @body.nil? ? 0 : @body.size

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

#respond_with?(type) ⇒ Boolean

Returns:

  • (Boolean) 


124
125
126
127
128
129
130
131
 
# File 'lib/angelo/responder.rb', line 124

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