Sha256: 0776151ce2a7535209bfdb6ef2e2ef100737e79a7224e629d7d649c9f095b542
Contents?: true
Size: 1.9 KB
Versions: 4
Compression:
Stored size: 1.9 KB
Contents
# This file is part of the "Utopia Framework" project, and is released under the MIT license. # Copyright 2010 Samuel Williams. All rights reserved. # See <utopia.rb> for licensing details. require 'utopia/middleware' require 'set' module Utopia module Middleware class Requester TTL_KEY = "utopia.requestor.ttl" REQUESTOR_KEY = "utopia.requestor" PATH_VARIABLES = Set.new MAXIMUM_DEPTH = 5 class Response def initialize(status, headers, body) @status = status @headers = headers @body = body end attr :status attr :headers def body unless @body_string buffer = StringIO.new @body.each do |string| buffer.write(string) end @body_string = buffer.string end return @body_string end def [](key) return @headers[key] end def okay? @status == 200 end end class NoRequesterError < ArgumentError end def self.[](env) requestor = env[REQUESTOR_KEY] if requestor return requestor else raise NoRequesterError end end def initialize(app, env = {}) @app = app @env = env @env[TTL_KEY] = 0 end attr :env, true def call(env) requestor = dup env[REQUESTOR_KEY] = requestor requestor.env = env.merge(@env) requestor.env.delete("rack.request") @app.call(env) end class RecursiveRequestError < StandardError end def request(env) env = @env.merge(env) env[TTL_KEY] += 1 if env[TTL_KEY].to_i > MAXIMUM_DEPTH raise RecursiveRequestError, env["PATH_INFO"] end return Response.new(*@app.call(env)) end def [](path, method = "GET", env = {}) path_env = { "REQUEST_METHOD" => method, "PATH_INFO" => path.to_s } request(env.merge(path_env)) end # Avoid a huge amount of junk being printed when inspecting +env+. def inspect to_s end end end end
Version data entries
4 entries across 4 versions & 1 rubygems