kastner-rack_gateway 0.1

data/MIT-LICENSE

@@ -0,0 +1,21 @@
+Author: Erik Kastner
+Copyright (c) 2009.
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,3 @@
+= Rack::Gateway
+
+A Rack middleware for Failover Proxying.

data/Rakefile

@@ -0,0 +1,97 @@
+# Rakefile for Rack::Gateway.  -*-ruby-*-
+require 'rake/rdoctask'
+require 'rake/testtask'
+ 
+desc "Run all the tests"
+task :default => [:test]
+ 
+desc "Generate RDox"
+task "RDOX" do
+  sh "specrb -Ilib:test -a --rdox >RDOX"
+end
+ 
+desc "Run specs with test/unit style output"
+task :test do
+  sh "specrb -Ilib:test -w #{ENV['TEST'] || '-a'} #{ENV['TESTOPTS']}"
+end
+ 
+desc "Run specs with specdoc style output"
+task :spec do
+  sh "specrb -Ilib:test -s -w #{ENV['TEST'] || '-a'} #{ENV['TESTOPTS']}"
+end
+ 
+desc "Run all the tests"
+task :fulltest do
+  sh "specrb -Ilib:test -w #{ENV['TEST'] || '-a'} #{ENV['TESTOPTS']}"
+end
+ 
+desc "Generate RDoc documentation"
+Rake::RDocTask.new(:rdoc) do |rdoc|
+  rdoc.options << '--line-numbers' << '--inline-source' <<
+    '--main' << 'README' <<
+    '--title' << 'Rack Gateway Documentation' <<
+    '--charset' << 'utf-8'
+  rdoc.rdoc_dir = "doc"
+  rdoc.rdoc_files.include 'README.rdoc'
+  rdoc.rdoc_files.include 'RDOX'
+  rdoc.rdoc_files.include('lib/rack/*.rb')
+  rdoc.rdoc_files.include('lib/rack/*/*.rb')
+end
+task :rdoc => ["RDOX"]
+ 
+ 
+# PACKAGING =================================================================
+ 
+# load gemspec like github's gem builder to surface any SAFE issues.
+require 'rubygems/specification'
+$spec = eval(File.read('rack_gateway.gemspec'))
+ 
+def package(ext='')
+  "pkg/rack_gateway-#{$spec.version}" + ext
+end
+ 
+desc 'Build packages'
+task :package => %w[.gem .tar.gz].map {|e| package(e)}
+ 
+desc 'Build and install as local gem'
+task :install => package('.gem') do
+  sh "gem install #{package('.gem')}"
+end
+ 
+directory 'pkg/'
+ 
+file package('.gem') => %w[pkg/ rack_gateway.gemspec] + $spec.files do |f|
+  sh "gem build rack_gateway.gemspec"
+  mv File.basename(f.name), f.name
+end
+ 
+file package('.tar.gz') => %w[pkg/] + $spec.files do |f|
+  sh "git archive --format=tar HEAD | gzip > #{f.name}"
+end
+ 
+# desc 'Publish gem and tarball to rubyforge'
+# task 'publish:gem' => [package('.gem'), package('.tar.gz')] do |t|
+#   sh <<-end
+#     rubyforge add_release rack rack_gateway #{$spec.version} #{package('.gem')} &&
+#     rubyforge add_file    rack rack_gateway #{$spec.version} #{package('.tar.gz')}
+#   end
+# end
+ 
+# GEMSPEC ===================================================================
+ 
+file 'rack_gateway.gemspec' => FileList['{lib,test}/**','Rakefile', 'README.rdoc'] do |f|
+  # read spec file and split out manifest section
+  spec = File.read(f.name)
+  parts = spec.split("  # = MANIFEST =\n")
+  fail 'bad spec' if parts.length != 3
+  # determine file list from git ls-files
+  files = `git ls-files`.
+    split("\n").sort.reject{ |file| file =~ /^\./ }.
+    map{ |file| "    #{file}" }.join("\n")
+  # piece file back together and write...
+  parts[1] = "  s.files = %w[\n#{files}\n  ]\n"
+  spec = parts.join("  # = MANIFEST =\n")
+  spec.sub!(/s.date = '.*'/, "s.date = '#{Time.now.strftime("%Y-%m-%d")}'")
+  File.open(f.name, 'w') { |io| io.write(spec) }
+  puts "updated #{f.name}"
+end

data/lib/rack_gateway.rb

@@ -0,0 +1,60 @@
+require 'curb'
+
+module Rack
+  class Gateway
+    attr_accessor :env
+    
+    def initialize(app)
+      @app = app
+    end
+    
+    def call(env)
+      self.dup._call(env)
+    end
+    
+    def upstream_url
+      # raise env.inspect
+      "%s://%s%s%s" % [
+        @env["rack.url_scheme"],
+        ENV["upstream_server"],
+        @env["PATH_INFO"],
+        @env["QUERY_STRING"].empty? ? "" : "?#{@env["QUERY_STRING"]}"
+      ]
+    end
+    
+    def request_headers
+      headers = @env.select {|k,v| k.match(/^HTTP_/)}.map do |key, value|
+        my_key = key.gsub(/^HTTP_/, '').gsub(/_/, '-').downcase.capitalize
+        [my_key, value]
+      end
+      headers = Hash[*headers.flatten]
+    end
+    
+    def response_headers(curl)
+      headers = curl.header_str.split("\r\n")[1..-1].map{|h| h.split(": ")}
+      headers
+      Hash[*headers.flatten]
+    end
+    
+    def _call(env)
+      @env = env
+      @status, @headers, @response = @app.call(@env)
+      
+      if @status == 404
+        c = Curl::Easy.new(upstream_url)
+        c.headers = request_headers
+        case @env["REQUEST_METHOD"]
+        when /get/i
+          c.perform
+        when /post/i
+          c.http_post(@env["rack.request.form_vars"])
+        end
+        @response = c.body_str
+        @status = c.response_code
+        @headers = response_headers(c)
+      end
+      
+      [@status, @headers, @response]
+    end
+  end
+end

data/rack_gateway.gemspec

@@ -0,0 +1,36 @@
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+ 
+  s.name = 'rack_gateway'
+  s.version = '0.1'
+  s.date = '2009-04-02'
+ 
+  s.description = "Rack Gateway Proxy"
+  s.summary     = "Rack Gateway Proxy"
+ 
+  s.authors = ["Erik Kastner"]
+  s.email = "kastner@gmail.com"
+ 
+  # = MANIFEST =
+  s.files = %w[
+    Rakefile
+    lib/rack_gateway.rb
+    rack_gateway.gemspec
+    test/spec_rack_gateway.rb
+  ]
+  # = MANIFEST =
+ 
+  s.test_files = s.files.select {|path| path =~ /^test\/spec_.*\.rb/}
+ 
+  s.extra_rdoc_files = %w[README.rdoc MIT-LICENSE]
+  s.add_dependency 'rack', '~> 0.9.1'
+  s.add_dependency 'test-spec', '~> 0.9.0'
+  s.add_development_dependency 'json', '>= 1.1'
+ 
+  s.has_rdoc = true
+  s.homepage = "http://github.com/kastner/rack_gateway/"
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "rack-contrib", "--main", "README"]
+  s.require_paths = %w[lib]
+  s.rubygems_version = '1.1.1'
+end

data/test/spec_rack_gateway.rb

@@ -0,0 +1,39 @@
+require 'test/spec'
+require 'rack/mock'
+require 'rack_gateway'
+ 
+context 'Rack::Gateway' do
+  before(:each) do
+    ENV["upstream_server"] = "localhost"
+    @test_headers = {'Content-Type' => 'text/html'}
+    @test_body = "Not Found"
+    @app = lambda { |env| [404, @test_headers, [@test_body]]}
+    @gateway = Rack::Gateway.new(@app)
+  end
+  
+  specify "request url should be set" do
+    request = Rack::MockRequest.env_for("/bob", :lint => true, :fatal => true)
+    @gateway.env = request
+    @gateway.upstream_url.should.equal "http://localhost/bob"
+  end
+
+  specify "request url should be set even with a query string" do
+    request = Rack::MockRequest.env_for("/bob?a=1", :lint => true, :fatal => true)
+    @gateway.env = request
+    @gateway.upstream_url.should.equal "http://localhost/bob?a=1"
+  end
+  
+  specify "forwarding request should pass-through headers" do
+    request = Rack::MockRequest.env_for("/bob", {
+      :lint => true, :fatal => true, 
+      "HTTP_ACCEPT" => "application/xml,text/html;q=0.9,*/*;q=0.5",
+      "HTTP_USER_AGENT" => "Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_5_6; en-us) AppleWebKit/528.16 (KHTML, like Gecko) Version/4.0 Safari/528.16"
+    })
+    @gateway.env = request
+    @gateway.request_headers.should.include?("User-agent")
+    @gateway.request_headers['User-agent'].should.match /Safari/
+  end
+  
+  # specify "A POST should forward the post data" do
+  # end
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification 
+name: kastner-rack_gateway
+version: !ruby/object:Gem::Version 
+  version: "0.1"
+platform: ruby
+authors: 
+- Erik Kastner
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-04-02 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rack
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.9.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: test-spec
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.9.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: json
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "1.1"
+    version: 
+description: Rack Gateway Proxy
+email: kastner@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+- MIT-LICENSE
+files: 
+- Rakefile
+- lib/rack_gateway.rb
+- rack_gateway.gemspec
+- test/spec_rack_gateway.rb
+- README.rdoc
+- MIT-LICENSE
+has_rdoc: true
+homepage: http://github.com/kastner/rack_gateway/
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- rack-contrib
+- --main
+- README
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Rack Gateway Proxy
+test_files: 
+- test/spec_rack_gateway.rb