Class: Goat::ReqHandler

Inherits:
Object show all
Defined in:
lib/goat.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ ReqHandler

Returns a new instance of ReqHandler.



250
251
252
 
# File 'lib/goat.rb', line 250

def initialize
  @around_handler_bindings = {}
end

Instance Attribute Details

#app_class ⇒ Object

we can’t know this at initialize time because we start using the req handler in context of Goat::App, which would lead to app = Goat::App, which is wrong



248
249
250
 
# File 'lib/goat.rb', line 248

def app_class
  @app_class
end

Instance Method Details

#add_mapping(opts) ⇒ Object 



264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
 
# File 'lib/goat.rb', line 264

def add_mapping(opts)
  opts = {
    :metal => false,
    :method => :get
  }.merge(opts)

  %w{method path hook}.each do |key|
    raise "#{key} must be set" unless opts[key.to_sym]
  end

  method = opts[:method]
  path = opts[:path]
  hook = opts[:hook]

  if hook.kind_of?(Proc) && !opts[:metal]
    hook = App.bind(opts[:hook])
  end

  mappings[method][path] = hook
end

#around_handler_binding(handler, type) ⇒ Object 



285
286
287
288
289
290
291
292
293
 
# File 'lib/goat.rb', line 285

def around_handler_binding(handler, type)
  if type == :before
    @around_handler_bindings[handler] ||= App.bind(handler)
  elsif type == :after
    @around_handler_bindings[handler] ||= App.bind(handler, nil, false)
  else
    raise 'bad handler type'
  end
end

#find_hook(meth, path) ⇒ Object 



334
335
336
337
338
339
340
341
342
343
344
345
346
347
 
# File 'lib/goat.rb', line 334

def find_hook(meth, path)
  hooks = mappings[meth.downcase.to_sym]
  if h = hooks[path]
    return h
  else
    hooks.each do |pathspec, hook|
      if pathspec.kind_of?(Regexp) && pathspec =~ path
        return hook
      end
    end
  end

  nil
end

#handle_request(env) ⇒ Object 



349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
 
# File 'lib/goat.rb', line 349

def handle_request(env)
  path = env['PATH_INFO']
  meth = env['REQUEST_METHOD']
  hook = find_hook(meth, path)
  hdrs = {}
  resp = nil
  req = Rack::Request.new(env)

  Dynamic.let(:current_request => req) do
    begin

      app = @app_class.new(req)

      begin
        run_before_handlers(app)
      rescue Halt => halt
        return halt.response.to_a
      end

      if hook
        resp = hook.handle_request(app)
      else
        raise NotFoundError.new(path)
      end
    rescue Halt => halt
      resp = halt.response
    rescue Exception => e
      resp = resp_for_error(e, app)
    end

    run_after_handlers(app)
  end

  resp.to_a
end

#mappings ⇒ Object 



260
261
262
 
# File 'lib/goat.rb', line 260

def mappings
  @map ||= Hash.new {|h, k| h[k] = Hash.new}
end

#resp_for_error(e, app) ⇒ Object 



304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
 
# File 'lib/goat.rb', line 304

def resp_for_error(e, app)
  resp = nil

  if e.kind_of?(NotFoundError)
    # 404
    if app.class.not_found_handler
      @not_found_binding ||= App.bind(app.class.not_found_handler)
      resp = @not_found_binding.call(app, e)
    else
      resp = [404, {}, 'not found']
    end
  else
    # not a 404 -- an actual problem
    if app.class.error_handler
      @error_handler_binding ||= App.bind(app.class.error_handler)
      resp = @error_handler_binding.call(app, e)
    else
      Goat.loge e.inspect
      Goat.loge e.backtrace.join("\n")

      resp = Rack::Response.new
      resp.status = 500
      resp['Content-Type'] = 'text/plain'
      resp.body = e.inspect + "\n" + e.backtrace.join("\n")
    end
  end

  resp
end

#run_after_handlers(app) ⇒ Object 



302
 
# File 'lib/goat.rb', line 302

def run_after_handlers(app); run_around_handlers(app, :after); end

#run_around_handlers(app, type) ⇒ Object 



295
296
297
298
299
 
# File 'lib/goat.rb', line 295

def run_around_handlers(app, type)
  app.class.around_handlers.select{|h| h.first == type}.each do |handler|
    around_handler_binding(handler[1], type).call(app)
  end
end

#run_before_handlers(app) ⇒ Object 



301
 
# File 'lib/goat.rb', line 301

def run_before_handlers(app); run_around_handlers(app, :before); end