rack-force_domain 0.1.0

data/README.markdown

@@ -0,0 +1,20 @@
+# Rack::ForceDomain
+
+Directs all traffic to a single domain via 301 redirects.
+
+## Example Usage (Heroku)
+
+### config.ru
+    use Rack::ForceDomain, ENV["DOMAIN"]
+    
+### environment.rb
+    config.middleware.use Rack::ForceDomain, ENV["DOMAIN"]
+
+### Heroku Config
+
+    heroku config add DOMAIN="foo.com"
+
+
+Now all requests to www.foo.com (or anything else pointed at the app) will 301 to foo.com.
+
+If the `$DOMAIN` environment variable is missing, no redirection will occur.

data/lib/rack-force_domain.rb

@@ -0,0 +1 @@
+require File.expand_path("rack/force_domain", __FILE__)

data/lib/rack/force_domain.rb

@@ -0,0 +1,18 @@
+require 'rack'
+
+class Rack::ForceDomain
+  def initialize(app, domain)
+    @app = app
+    @domain = domain
+  end
+
+  def call(env)
+    request = Rack::Request.new(env)
+    if @domain and request.host != @domain
+      fake_request = Rack::Request.new(env.merge("HTTP_HOST" => @domain))
+      Rack::Response.new([], 301, "Location" => fake_request.url).finish
+    else
+      @app.call(env)
+    end
+  end
+end

data/test/force_domain_test.rb

@@ -0,0 +1,30 @@
+$:.unshift(File.join(File.dirname(__FILE__), *%w[.. lib]))
+
+require 'rack/test'
+require 'test/unit'
+require 'rack/force_domain'
+
+class ForceDomainTest < Test::Unit::TestCase
+  include Rack::Test::Methods
+
+  def test_should_do_nothing_if_domain_is_null
+    app = Rack::ForceDomain.new(lambda{|env| @called = true; [200, [], {}] }, nil)
+    app.call(Rack::MockRequest.env_for("http://foo.com/baz"))
+    assert @called, "expected the request to be passed through"
+  end
+
+  def test_should_301_if_domain_is_wrong
+    app = Rack::ForceDomain.new(lambda{|env| @called = true; [200, [], {}] } , "bar.com")
+    status, body, headers = app.call(Rack::MockRequest.env_for("http://foo.com/baz"))
+    assert_equal 301, status
+    assert_equal "http://bar.com/baz", headers["Location"]
+    assert !@called, "should not have passed through"
+  end
+
+  def test_should_passthrough_for_correct_domain
+    app = Rack::ForceDomain.new(lambda{|env| @called = true; [200, [], {}] } , "foo.com")
+    status, body, headers = app.call(Rack::MockRequest.env_for("http://foo.com/baz"))
+    assert_equal 200, status
+    assert @called, "expected the request to be passed through"
+  end
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification 
+name: rack-force_domain
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Tom Lea
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-01-29 00:00:00 +00:00
+default_executable: 
+dependencies: []
+
+description: 
+email: contrib@tomlea.co.uk
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- README.markdown
+- test/force_domain_test.rb
+- lib/rack/force_domain.rb
+- lib/rack-force_domain.rb
+has_rdoc: true
+homepage: http://tomlea.co.uk/
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --main
+- README.markdown
+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: rack-force_domain
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Force all visitors onto a single domain.
+test_files: []
+