rack 0.1.0

data/lib/rack/urlmap.rb

@@ -0,0 +1,50 @@
+module Rack
+  # Rack::URLMap takes a hash mapping urls or paths to apps, and
+  # dispatches accordingly.  Support for HTTP/1.1 host names exists if
+  # the URLs start with <tt>http://</tt> or <tt>https://</tt>.
+  #
+  # URLMap modifies the SCRIPT_NAME and PATH_INFO such that the part
+  # relevant for dispatch is in the SCRIPT_NAME, and the rest in the
+  # PATH_INFO.  This should be taken care of when you need to
+  # reconstruct the URL in order to create links.
+  #
+  # URLMap dispatches in such a way that the longest paths are tried
+  # first, since they are most specific.
+  
+  class URLMap
+    def initialize(map)
+      @mapping = map.map { |location, app|
+        if location =~ %r{\Ahttps?://(.*?)(/.*)}
+          host, location = $1, $2
+        else
+          host = nil
+        end
+
+        location = ""  if location == "/"
+
+        [host, location, app]
+      }.sort_by { |(h, l, a)| -l.size } # Longest path first
+    end
+
+    def call(env)
+      path = env["PATH_INFO"].to_s.squeeze("/")
+      @mapping.each { |host, location, app|
+        if (env["HTTP_HOST"] == host ||
+            env["SERVER_NAME"] == host ||
+            (host == nil && (env["HTTP_HOST"] == env["SERVER_NAME"] ||
+                             env["HTTP_HOST"] ==
+                             "#{env["SERVER_NAME"]}:#{env["SERVER_PORT"]}"))) &&
+           location == path[0, location.size] && (path[location.size] == nil ||
+                                                  path[location.size] == ?/)
+          env["SCRIPT_NAME"] = location.dup
+          env["PATH_INFO"] = path[location.size..-1]
+          env["PATH_INFO"].gsub!(/\/\z/, '')
+          env["PATH_INFO"] = "/"  if env["PATH_INFO"].empty?
+          return app.call(env)
+        end
+      }
+      [404, {"Content-Type" => "text/plain"}, ["Not Found: #{path}"]]
+    end
+  end
+end
+

data/lib/rack/utils.rb

@@ -0,0 +1,176 @@
+require 'tempfile'
+
+module Rack
+  # Rack::Utils contains a grab-bag of useful methods for writing web
+  # applications adopted from all kinds of Ruby libraries.
+  
+  module Utils
+    # Performs URI escaping so that you can construct proper
+    # query strings faster.  Use this rather than the cgi.rb
+    # version since it's faster.  (Stolen from Camping).
+    def escape(s)
+      s.to_s.gsub(/([^ a-zA-Z0-9_.-]+)/n) {
+        '%'+$1.unpack('H2'*$1.size).join('%').upcase
+      }.tr(' ', '+')
+    end
+    module_function :escape
+
+    # Unescapes a URI escaped string. (Stolen from Camping).
+    def unescape(s)
+      s.tr('+', ' ').gsub(/((?:%[0-9a-fA-F]{2})+)/n){
+        [$1.delete('%')].pack('H*')
+      }
+    end
+    module_function :unescape
+
+    # Stolen from Mongrel:
+    # Parses a query string by breaking it up at the '&'
+    # and ';' characters.  You can also use this to parse
+    # cookies by changing the characters used in the second
+    # parameter (which defaults to '&;'.
+
+    def parse_query(qs, d = '&;')
+      params = {}
+      (qs||'').split(/[#{d}] */n).inject(params) { |h,p|
+        k, v=unescape(p).split('=',2)
+        if cur = params[k]
+          if cur.class == Array
+            params[k] << v
+          else
+            params[k] = [cur, v]
+          end
+        else
+          params[k] = v
+        end
+      }
+
+      return params
+    end
+    module_function :parse_query
+
+    # Escape ampersands, brackets and quotes to their HTML/XML entities.
+    def escape_html(string)
+      string.to_s.gsub("&", "&amp;").
+        gsub("<", "&lt;").
+        gsub(">", "&gt;").
+        gsub("'", "&#39;").
+        gsub('"', "&quot;")
+    end
+    module_function :escape_html
+
+    # A case-normalizing Hash, adjusting on [] and []=.
+    class HeaderHash < Hash
+      def initialize(hash={})
+        hash.each { |k, v| self[k] = v }
+      end
+
+      def to_hash
+        {}.replace(self)
+      end
+
+      def [](k)
+        super capitalize(k)
+      end
+
+      def []=(k, v)
+        super capitalize(k), v
+      end
+
+      def capitalize(k)
+        k.to_s.downcase.gsub(/^.|[-_\s]./) { |x| x.upcase }
+      end
+    end
+
+    # A multipart form data parser, adapted from IOWA.
+    #
+    # Usually, Rack::Request#POST takes care of calling this.
+    
+    module Multipart
+      EOL = "\r\n"
+
+      def self.parse_multipart(env)
+        unless env['CONTENT_TYPE'] =~
+            %r|\Amultipart/form-data.*boundary=\"?([^\";,]+)\"?|n
+          nil
+        else
+          boundary = "--#{$1}"
+
+          params = {}
+          buf = ""
+          content_length = env['CONTENT_LENGTH'].to_i
+          input = env['rack.input']
+
+          boundary_size = boundary.size + EOL.size
+          bufsize = 16384
+
+          content_length -= boundary_size
+
+          status = input.read(boundary_size)
+          raise EOFError, "bad content body"  unless status == boundary + EOL
+
+          rx = /(?:#{EOL})?#{Regexp.quote boundary}(#{EOL}|--)/
+
+          loop {
+            head = nil
+            body = ''
+            filename = content_type = name = nil
+
+            until head && buf =~ rx
+              if !head && i = buf.index("\r\n\r\n")
+                head = buf.slice!(0, i+2) # First \r\n
+                buf.slice!(0, 2)          # Second \r\n
+
+                filename = head[/Content-Disposition:.* filename="?([^\";]*)"?/ni, 1]
+                content_type = head[/Content-Type: (.*)\r\n/ni, 1]
+                name = head[/Content-Disposition:.* name="?([^\";]*)"?/ni, 1]
+
+                body = Tempfile.new("RackMultipart")  if filename
+
+                next
+              end
+
+              # Save the read body part.
+              if head && (boundary_size+4 < buf.size)
+                body << buf.slice!(0, buf.size - (boundary_size+4))
+              end
+
+              c = input.read(bufsize < content_length ? bufsize : content_length)
+              raise EOFError, "bad content body"  if c.nil? || c.empty?
+              buf << c
+              content_length -= c.size
+            end
+
+            # Save the rest.
+            if i = buf.index(rx)
+              body << buf.slice!(0, i)
+              buf.slice!(0, boundary_size+2)
+
+              content_length = -1  if $1 == "--"
+            end
+
+            if filename
+              body.rewind
+              data = {:filename => filename, :type => content_type,
+                      :name => name, :tempfile => body, :head => head}
+            else
+              data = body
+            end
+
+            if name
+              if name =~ /\[\]\z/
+                params[name] ||= []
+                params[name] << data
+              else
+                params[name] = data
+              end
+            end
+
+            break  if buf.empty? || content_length == -1
+          }
+
+          params
+        end
+      end
+    end
+  end
+end

data/test/cgi/lighttpd.conf

@@ -0,0 +1,20 @@
+server.modules = ("mod_fastcgi", "mod_cgi")
+server.document-root = "."
+server.errorlog = "lighttpd.errors"
+server.port = 9203
+
+server.event-handler = "freebsd-kqueue"
+
+cgi.assign = ("/test" => "/usr/local/bin/ruby",
+#              ".ru" => ""
+             )
+
+fastcgi.server = ("test.fcgi" => ("localhost" =>
+                    ("min-procs" => 1,
+                     "socket" => "/tmp/rack-test-fcgi",
+                     "bin-path" => "test.fcgi")),
+                  "test.ru" => ("localhost" =>
+                    ("min-procs" => 1,
+                     "socket" => "/tmp/rack-test-ru-fcgi",
+                     "bin-path" => "test.ru")),
+                 )

data/test/cgi/test

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+# -*- ruby -*-
+
+$: << File.join(File.dirname(__FILE__), "..", "..", "lib")
+
+require 'rack'
+require '../testrequest'
+
+Rack::Handler::CGI.run(Rack::Lint.new(TestRequest.new))

data/test/cgi/test.fcgi

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+# -*- ruby -*-
+
+$: << File.join(File.dirname(__FILE__), "..", "..", "lib")
+
+require 'rack'
+require '../testrequest'
+
+Rack::Handler::FastCGI.run(Rack::Lint.new(TestRequest.new))

data/test/cgi/test.ru

@@ -0,0 +1,7 @@
+#!/usr/local/bin/ruby ../../bin/rackup
+#\ -E deployment -I ~/projects/rack/lib
+# -*- ruby -*-
+
+require '../testrequest'
+
+run TestRequest.new

data/test/spec_rack_camping.rb

@@ -0,0 +1,44 @@
+require 'test/spec'
+require 'stringio'
+
+require 'rack/mock'
+
+$-w, w = nil, $-w               # yuck
+require 'camping'
+require 'rack/adapter/camping'
+
+Camping.goes :CampApp
+module CampApp
+  module Controllers
+    class HW < R('/')
+      def get
+        "Camping works!"
+      end
+
+      def post
+        "Data: #{input.foo}"
+      end
+    end
+  end
+end
+$-w = w
+
+context "Rack::Adapter::Camping" do
+  specify "works with GET" do
+    res = Rack::MockRequest.new(Rack::Adapter::Camping.new(CampApp)).
+      get("/")
+
+    res.should.be.ok
+    res["Content-Type"].should.equal "text/html"
+
+    res.body.should.equal "Camping works!"
+  end
+
+  specify "works with POST" do
+    res = Rack::MockRequest.new(Rack::Adapter::Camping.new(CampApp)).
+      post("/", :input => "foo=bar")
+
+    res.should.be.ok
+    res.body.should.equal "Data: bar"
+  end
+end

data/test/spec_rack_cascade.rb

@@ -0,0 +1,35 @@
+require 'test/spec'
+
+require 'rack/cascade'
+require 'rack/mock'
+
+require 'rack/urlmap'
+require 'rack/file'
+
+context "Rack::Cascade" do
+  docroot = File.expand_path(File.dirname(__FILE__))
+  app1 = Rack::File.new(docroot)
+
+  app2 = Rack::URLMap.new("/crash" => lambda { |env| raise "boom" })
+
+  app3 = Rack::URLMap.new("/foo" => lambda { |env|
+                            [200, { "Content-Type" => "text/plain"}, [""]]})
+
+  specify "should dispatch onward on 404 by default" do
+    cascade = Rack::Cascade.new([app1, app2, app3])
+    Rack::MockRequest.new(cascade).get("/cgi/test").should.be.ok
+    Rack::MockRequest.new(cascade).get("/foo").should.be.ok
+    Rack::MockRequest.new(cascade).get("/toobad").should.be.not_found
+    Rack::MockRequest.new(cascade).get("/cgi/../bla").should.be.forbidden
+  end
+
+  specify "should dispatch onward on whatever is passed" do
+    cascade = Rack::Cascade.new([app1, app2, app3], [404, 403])
+    Rack::MockRequest.new(cascade).get("/cgi/../bla").should.be.not_found
+  end
+
+  specify "should fail if empty" do
+    lambda { Rack::MockRequest.new(Rack::Cascade.new([])).get("/") }.
+      should.raise(ArgumentError)
+  end
+end

data/test/spec_rack_cgi.rb

@@ -0,0 +1,82 @@
+require 'test/spec'
+require 'testrequest'
+
+pid = fork {
+  exec "cd #{File.join(File.dirname(__FILE__), 'cgi')} && lighttpd -D -f lighttpd.conf"
+}
+
+at_exit {
+  Process.kill 15, pid
+}
+
+context "Rack::Handler::CGI" do
+  include TestRequest::Helpers
+  
+  setup do
+    @host = '0.0.0.0'
+    @port = 9203
+  end
+
+  specify "should respond" do
+    lambda {
+      GET("/test")
+    }.should.not.raise
+  end
+
+  specify "should be a lighttpd" do
+    GET("/test")
+    status.should.be 200
+    response["SERVER_SOFTWARE"].should =~ /lighttpd/
+    response["HTTP_VERSION"].should.equal "HTTP/1.1"
+    response["SERVER_PROTOCOL"].should.equal "HTTP/1.1"
+    response["SERVER_PORT"].should.equal "9203"
+    response["SERVER_NAME"].should =~ "0.0.0.0"
+  end
+
+  specify "should have rack headers" do
+    GET("/test")
+    response["rack.version"].should.equal [0,1]
+    response["rack.multithread"].should.be false
+    response["rack.multiprocess"].should.be true
+    response["rack.run_once"].should.be true
+  end
+
+  specify "should have CGI headers on GET" do
+    GET("/test")
+    response["REQUEST_METHOD"].should.equal "GET"
+    response["SCRIPT_NAME"].should.equal "/test"
+    response["REQUEST_PATH"].should.equal "/"
+    response["PATH_INFO"].should.be.nil
+    response["QUERY_STRING"].should.equal ""
+    response["test.postdata"].should.equal ""
+
+    GET("/test/foo?quux=1")
+    response["REQUEST_METHOD"].should.equal "GET"
+    response["SCRIPT_NAME"].should.equal "/test"
+    response["REQUEST_PATH"].should.equal "/"
+    response["PATH_INFO"].should.equal "/foo"
+    response["QUERY_STRING"].should.equal "quux=1"
+  end
+
+  specify "should have CGI headers on POST" do
+    POST("/test", {"rack-form-data" => "23"}, {'X-test-header' => '42'})
+    status.should.equal 200
+    response["REQUEST_METHOD"].should.equal "POST"
+    response["SCRIPT_NAME"].should.equal "/test"
+    response["REQUEST_PATH"].should.equal "/"
+    response["QUERY_STRING"].should.equal ""
+    response["HTTP_X_TEST_HEADER"].should.equal "42"
+    response["test.postdata"].should.equal "rack-form-data=23"
+  end
+
+  specify "should support HTTP auth" do
+    GET("/test", {:user => "ruth", :passwd => "secret"})
+    response["HTTP_AUTHORIZATION"].should.equal "Basic cnV0aDpzZWNyZXQ="
+  end
+
+  specify "should set status" do
+    GET("/test?secret")
+    status.should.equal 403
+    response["rack.url_scheme"].should.equal "http"
+  end
+end

data/test/spec_rack_commonlogger.rb

@@ -0,0 +1,32 @@
+require 'test/spec'
+require 'stringio'
+
+require 'rack/commonlogger'
+require 'rack/lobster'
+require 'rack/mock'
+
+context "Rack::CommonLogger" do
+  app = lambda { |env|
+    [200,
+     {"Content-Type" => "text/html"},
+     ["foo"]]}
+
+  specify "should log to rack.errors by default" do
+    log = StringIO.new
+    res = Rack::MockRequest.new(Rack::CommonLogger.new(app)).get("/")
+
+    res.errors.should.not.be.empty
+    res.errors.should =~ /GET /
+    res.errors.should =~ / 200 / # status
+    res.errors.should =~ / 3 /   # length
+  end
+
+  specify "should log to anything with <<" do
+    log = ""
+    res = Rack::MockRequest.new(Rack::CommonLogger.new(app, log)).get("/")
+
+    log.should =~ /GET /
+    log.should =~ / 200 / # status
+    log.should =~ / 3 / # length
+  end
+end