rrproxy 0.0.1

data/.gitignore

@@ -0,0 +1,3 @@
+*.gem
+.bundle
+Gemfile.lock

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in rrproxy.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Caleb Spare
+
+MIT License
+
+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.md

@@ -0,0 +1,50 @@
+rrproxy
+=======
+
+rrproxy ("routing reverse proxy" or "rockin' ruby proxy" -- you pick) is a very simple reverse proxy server
+that can route requests to different servers based on simple pattern-matching rules against the requests. It
+is intended for use in a development environment as a substitute for 'real' proxies (nginx/HAProxy/whatever)
+that you use in production.
+
+Installation
+------------
+
+    $ gem install rrproxy
+
+rrproxy requires Ruby 1.9 (because it's based on Goliath).
+
+Usage
+-----
+
+rrproxy takes a very simple Ruby configuration file.
+
+    [
+      ["/styles/screen.css", "localhost:8000"],
+      ["", "example.com"]
+    ]
+
+As you can probably guess, this will redirect requests for example.com's css file to a local server (while
+passing all other requests through normally).
+
+Here's a more complicated example (it's also in `examples/example_config.rb`):
+
+    [
+      ["/foo/bar?special=true", "localhost:1234"], # Route this specific request to a local server
+      [%r{^(/foo).*}, "google.com", ""],           # You can use a regex, and also replace the match with a third arg
+      ["", "amazon.com"]                           # Empty string matches everything
+    ]
+
+You will provide an array of tuples, where each one is of the form
+
+    [pattern, target_host, <optional replacement string>]
+
+For each request, rrproxy will examine each routing rule in order, checking for a prefix match (if the pattern
+is a string) or a regex match. The first match found will be used, and the host for the request will be
+changed to the corresponding target host. If a replacement is provided, it will be used as a substitute for
+the first regex matching group (if the pattern is a regex) or the whole pattern (if the pattern is a string).
+
+Credits
+-------
+
+* @tobert for the idea to use Goliath
+* The PostRank folks for a kickass server framework

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/bin/rrproxy

@@ -0,0 +1,68 @@
+#!/usr/bin/env ruby
+
+require "trollop"
+require "dedent"
+require "colorize"
+
+require "rrproxy"
+
+# The goliath folks really need to take a page out of Sinatra's book and provide an easy way to start up a
+# server manually without using all their application runner and option parsing shit (in Sinatra terms,
+# 'sinatra' vs 'sinatra/base'). As it is I had to poke around the source code a while to figure out all the
+# little things necessary to make it work.
+
+# See https://github.com/postrank-labs/goliath/blob/master/lib/goliath/runner.rb
+def create_logger
+  logger = Log4r::Logger.new("rrproxy")
+  log_format = Log4r::PatternFormatter.new(:pattern => "[%d : %l] %m")
+  logger.add(Log4r::StdoutOutputter.new("console", :formatter => log_format))
+  logger.level = Log4r::INFO
+  logger
+end
+
+options = Trollop.options do
+  banner <<-EOS.dedent
+    rrproxy is a very simple reverse http proxy that can route requests based on pattern-matching the
+    URI. The configuration file is Ruby, and it must have the following form:
+
+      [
+        ["/foo/bar?special=true", "localhost:1234"], # Route this specific request to a local server
+        [%r{^(/foo).*}, "google.com", ""],           # You can use a regex, and also replace the match
+        ["", "amazon.com"]                           # Empty string matches everything
+      ]
+
+    Usage:
+
+      $ rrproxy [options] CONFIG_FILE
+
+    where [options] are:
+  EOS
+  opt :address, "Bind address", :default => "0.0.0.0"
+  opt :port, "Port", :default => 9876
+end
+
+Trollop.die "You must provide a configuration file" unless ARGV[0]
+Trollop.die "Invalid config file: #{ARGV[0]}" unless File.file? ARGV[0]
+begin
+  RRProxy.routes = eval(File.read(ARGV[0]))
+rescue
+  Trollop.die "Error reading or parsing the config file #{ARGV[0]}"
+end
+
+logger = create_logger
+logger.info "Starting rrproxy (powered by Goliath) on #{options[:address]}:#{options[:port]}".green
+RRProxy.routes.each do |route|
+  pattern, target, replacement = route
+  message = "Proxying `#{pattern}` to #{target}"
+  message << " and replacing pattern matches with `#{replacement}`" if replacement
+  logger.info message.green
+end
+
+Goliath.env = :development
+api = RRProxy.new
+server = Goliath::Server.new(options[:address], options[:port])
+server.logger = logger
+server.api = api
+server.app = Goliath::Rack::Builder.build(RRProxy, api)
+server.plugins = []
+server.start

data/example/example_config.rb

@@ -0,0 +1,8 @@
+# General form: An array of
+# [pattern, target_host, <optional replacement string>]
+
+[
+  ["/foo/bar?special=true", "localhost:1234"], # Route this specific request to a local server
+  [%r{^(/foo).*}, "google.com", ""],           # You can use a regex, and also replace the match with a third arg
+  ["", "amazon.com"]                           # Empty string matches everything
+]

data/lib/rrproxy.rb

@@ -0,0 +1,54 @@
+require "goliath/api"
+require "goliath/goliath"
+require "goliath/rack"
+require "goliath/server"
+require "log4r"
+require "em-synchrony/em-http"
+
+class RRProxy < Goliath::API
+  class << self
+    attr_accessor :routes
+  end
+
+  use Goliath::Rack::Params
+
+  def options_parser(option_parser, options)
+    option_parser.on("-c", "--config CONFIG_FILE", "Ruby routes configuration file") do |filename|
+      options[:config] = filename
+    end
+    abort "asdf" unless File.file?(options[:config])
+  end
+
+  def on_headers(env, headers) env["untampered_headers"] = headers end
+
+  def response(env)
+    uri = env["REQUEST_URI"]
+    method = env["REQUEST_METHOD"]
+    logger.info "Proxying #{method} #{uri}"
+    method = method.downcase.to_sym
+    RRProxy.routes.each do |pattern, target, replacement|
+      if pattern.is_a?(Regexp)
+        next unless uri =~ pattern
+        uri.sub!($1, replacement) if replacement
+      else
+        next unless uri.start_with? pattern
+        uri.sub!(pattern, replacement) if replacement
+      end
+
+      target_url = "http://#{target}#{uri}"
+      headers = env["untampered_headers"]
+      headers["HOST"] = target
+      params = { :head => headers }
+      params[:body] = env["params"] if [:put, :post, :patch].include? method
+
+      http = EM::HttpRequest.new(target_url).send(method, params)
+      response_headers = {}
+      http.response_header.each do |k, v|
+        response_headers[k.downcase.split("_").map(&:capitalize).join("-")] = v
+      end
+
+      return [http.response_header.status, response_headers, [http.response]]
+    end
+    [503, {}, "No target specified for this path."]
+  end
+end

data/rrproxy.gemspec

@@ -0,0 +1,21 @@
+Gem::Specification.new do |gem|
+  gem.authors       = ["Caleb Spare"]
+  gem.email         = ["cespare@gmail.com"]
+  gem.description   = %q{rrproxy is a very simple routing reverse proxy written in Ruby.}
+  gem.summary       = %q{A routing reverse proxy.}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "rrproxy"
+  gem.version       = "0.0.1"
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency "goliath"
+  gem.add_dependency "log4r"
+  gem.add_dependency "em-http-request"
+  gem.add_dependency "trollop"
+  gem.add_dependency "dedent"
+  gem.add_dependency "colorize"
+end

metadata

@@ -0,0 +1,121 @@
+--- !ruby/object:Gem::Specification
+name: rrproxy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Caleb Spare
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-16 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: goliath
+  requirement: &24235400 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *24235400
+- !ruby/object:Gem::Dependency
+  name: log4r
+  requirement: &24200780 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *24200780
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: &24199760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *24199760
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: &24197520 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *24197520
+- !ruby/object:Gem::Dependency
+  name: dedent
+  requirement: &24196100 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *24196100
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: &24194600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *24194600
+description: rrproxy is a very simple routing reverse proxy written in Ruby.
+email:
+- cespare@gmail.com
+executables:
+- rrproxy
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- bin/rrproxy
+- example/example_config.rb
+- lib/rrproxy.rb
+- rrproxy.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: A routing reverse proxy.
+test_files: []