turnout_proxy 0.0.1

data/.gitignore

@@ -0,0 +1,7 @@
+tmp/
+.DS_Store
+.tags*
+.rvmrc
+.exrc
+pkg/
+Gemfile.lock

data/.travis.yml

@@ -0,0 +1,9 @@
+language: ruby
+rvm:
+  - ree-1.8.7
+  - 1.9.2
+  - 1.9.3
+  - 2.0.0
+branches:
+  only:
+    - master

data/Gemfile

@@ -0,0 +1,8 @@
+source 'https://rubygems.org'
+
+gemspec
+
+group :test do
+  gem 'rake'
+  gem 'posix-spawn'
+end

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2013 Vladimir Yarotsky
+
+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,18 @@
+Turnout Proxy
+=============
+
+[![Build Status](https://secure.travis-ci.org/v-yarotsky/turnout_proxy.png)](http://travis-ci.org/v-yarotsky/turnout_proxy)
+[![Gem Version](https://badge.fury.io/rb/turnout_proxy.png)](http://badge.fury.io/rb/turnout_proxy)
+
+Proxy server which allows to switch between two destinations using lock file.
+
+Installation:
+-------------
+
+    $ gem install turnout_proxy
+
+Usage:
+------
+
+    $ turnout_proxy --help
+

data/Rakefile

@@ -0,0 +1,24 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'rake'
+require 'rake/testtask'
+require 'rake/clean'
+require 'bundler/gem_tasks'
+
+begin
+  Bundler.setup
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+
+CLOBBER.include('doc/**', 'pkg/**', 'coverage/**')
+
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.pattern = "test/{lib,acceptance}/**/test*.rb"
+end
+
+task :default => [:test]
+

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/bin/turnout_proxy

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'optparse'
+require 'turnout_proxy'
+
+options = {
+  :lock_file => "/tmp/turnout.lock",
+  :default => { :host => "127.0.0.1" },
+  :alternate => { :host => "127.0.0.1" },
+  :host => "0.0.0.0",
+  :port => 5678,
+  :debug => false
+}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: turnout_proxy [options]"
+  opts.version = TurnoutProxy::VERSION
+
+  opts.separator "Proxy options:"
+  opts.on("--host HOST", "Proxy host (Default: localhost)") { |v| options[:host] = v }
+  opts.on("--port PORT", Integer, "Proxy port (Default: 5678)") { |v| options[:port] = v }
+  opts.on("--lock-file FILE", "File, which existence determines destination server (Default: /tmp/turnout.lock)") { |v| options[:lock_file] = v }
+
+  opts.separator "Default destination server options:"
+  opts.on("--default-host HOST", "(Default: localhost)") { |v| options[:default][:host] = v }
+  opts.on("--default-port PORT", Integer) { |v| options[:default][:port] = v }
+
+  opts.separator "Alternate destination server options:"
+  opts.on("--alternate-host HOST", "(Default: localhost)") { |v| options[:alternate][:host] = v }
+  opts.on("--alternate-port PORT", Integer) { |v| options[:alternate][:port] = v }
+
+  opts.separator "Special options:"
+  opts.on("--debug") { |*| options[:debug] = true }
+end.parse!
+
+Signal.trap("SIGINT") do
+  puts "Terminating"
+  exit
+end
+
+TurnoutProxy.run(options)

data/lib/turnout_proxy/host_chooser.rb

@@ -0,0 +1,37 @@
+module TurnoutProxy
+
+  class HostChooser
+    attr_accessor :file_checker
+
+    def initialize(connection, options = {})
+      @connection = connection
+      @file_checker = File
+
+      @default = validate_host(options[:default])
+      @alternate = validate_host(options[:alternate])
+
+      @lock_file = options[:lock_file]
+    end
+
+    def on_data(data)
+      use_alternate? ? @connection.server(:alternate, @alternate) : @connection.server(:default, @default)
+      data
+    end
+
+    private
+
+    def use_alternate?
+      @file_checker.exists?(@lock_file)
+    end
+
+    private
+
+    def validate_host(host_config)
+      if host_config.nil? || String(host_config[:host]).empty?  || host_config[:port].nil?
+        raise ArgumentError, "Bad host config, both host and port must be present."
+      end
+      host_config
+    end
+  end
+
+end

data/lib/turnout_proxy.rb

@@ -0,0 +1,15 @@
+require 'em-proxy'
+require 'turnout_proxy/host_chooser'
+
+module TurnoutProxy
+  version_file = File.expand_path('../VERSION', File.dirname(__FILE__))
+  VERSION = File.read(version_file).freeze
+
+  def self.run(options = {})
+    Proxy.start(options) do |conn|
+      callbacks = HostChooser.new(conn, options)
+      conn.on_data &callbacks.method(:on_data)
+    end
+  end
+end
+

data/test/acceptance/test_turnout_proxy.rb

@@ -0,0 +1,63 @@
+require 'test_helper'
+require 'fileutils'
+require 'net/http'
+require 'posix/spawn'
+require 'support/hello_server'
+require 'turnout_proxy'
+
+class TestTurnoutProxy < TurnoutProxyTestCase
+  include POSIX::Spawn
+  include TurnoutProxy
+
+  TEST_LOCK_FILE = PROJECT_ROOT.join("lock")
+
+  def setup
+    @@servers ||= begin
+      HelloServer.new(8083, "S1")
+      HelloServer.new(8084, "S2")
+    end
+    FileUtils.rm_rf TEST_LOCK_FILE
+  end
+
+  def teardown
+    FileUtils.rm_rf TEST_LOCK_FILE
+  end
+
+  def with_running_proxy(default_port = 8083, alternate_port = 8084)
+    router = PROJECT_ROOT.join("bin/turnout_proxy")
+    pid = spawn("ruby -Ilib #{router} --default-port=#{default_port} --alternate-port=#{alternate_port} --lock-file=#{TEST_LOCK_FILE}")
+    sleep 0.5
+    yield pid
+    sleep 0.5
+  ensure
+    Process.kill(:SIGQUIT, pid)
+  end
+
+  test "proxies to default server when lock does not exist" do
+    called = false
+    with_running_proxy do
+      called = Net::HTTP.get(URI("http://127.0.0.1:5678")) == "S1"
+    end
+    assert called, "Expected to proxy to server on 8083"
+  end
+
+  test "proxies to alternate server when lock file does exist" do
+    FileUtils.touch TEST_LOCK_FILE
+    called = false
+    with_running_proxy do
+      called = Net::HTTP.get(URI("http://127.0.0.1:5678")) == "S2"
+    end
+    assert called, "Expected to proxy to server on 8084"
+  end
+
+  test "doesn't fail if destination server isn't responding" do
+    @pid = nil
+    with_running_proxy(8085) do |pid|
+      @pid = pid
+      Net::HTTP.get(URI("http://127.0.0.1:5678")) rescue nil
+    end
+    Process.kill 0, @pid
+  end
+
+end
+

data/test/lib/turnout_proxy/test_host_chooser.rb

@@ -0,0 +1,66 @@
+require 'test_helper'
+require 'turnout_proxy/host_chooser'
+
+class TestHostChooser < TurnoutProxyTestCase
+  include TurnoutProxy
+
+  class FakeConnection
+    attr_reader :forwards
+
+    def initialize
+      @forwards = []
+    end
+
+    def server(*data)
+      @forwards << data
+    end
+  end
+
+  class FakeFile
+    def exists!(file)
+      @existing_file = file
+    end
+
+    def exists?(file)
+      !!@existing_file
+    end
+  end
+
+  def setup
+    @connection = FakeConnection.new
+    @file_checker = FakeFile.new
+  end
+
+  ALTERNATE_CONFIG = { :host => "127.0.0.1", :port => 19999 }
+  DEFAULT_CONFIG = { :host => "127.0.0.1", :port => 9012 }
+
+  def host_chooser
+    host_chooser = HostChooser.new(@connection,
+                                   :default => DEFAULT_CONFIG,
+                                   :alternate => ALTERNATE_CONFIG)
+    host_chooser.file_checker = @file_checker
+    host_chooser
+  end
+
+  test "it routes to developer tunnel if lock file exists" do
+    @file_checker.exists!("/tmp/betsoft.lock")
+    host_chooser.on_data("Hello")
+    assert_equal [[:alternate, ALTERNATE_CONFIG]], @connection.forwards
+  end
+
+  test "it routes to staging if lock file does not exist" do
+    host_chooser.on_data("Hello")
+    assert_equal [[:default, DEFAULT_CONFIG]], @connection.forwards
+  end
+
+  test "it raises if alternate host not specified" do
+    assert_raise(ArgumentError) do
+      HostChooser.new(@connection, :default => DEFAULT_CONFIG)
+    end
+
+    assert_raise(ArgumentError) do
+      HostChooser.new(@connection, :default => DEFAULT_CONFIG, :alternate => {})
+    end
+  end
+end
+

data/test/support/hello_server.rb

@@ -0,0 +1,36 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'posix/spawn'
+require 'eventmachine'
+
+class HelloServer
+  include POSIX::Spawn
+
+  def initialize(port, message)
+    @pid = spawn("ruby #{__FILE__} #{port} #{message}")
+    at_exit { die }
+  end
+
+  def pid
+    @pid
+  end
+
+  def die
+    Process.kill(:SIGQUIT, pid)
+  end
+end
+
+def HelloServer(response)
+  Module.new do
+   define_method :receive_data do |data|
+     send_data "HTTP/1.1 200 OK\r\nContent-Length: #{response.length}\r\n\r\n#{response}"
+     close_connection_after_writing
+   end
+  end
+end
+
+if $0 == __FILE__
+  EventMachine.run do
+    EventMachine.start_server "0.0.0.0", ARGV[0].to_i, HelloServer(ARGV[1] || "OK")
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,18 @@
+require 'test/unit'
+require 'pathname'
+
+$:.unshift File.expand_path('../../lib', __FILE__)
+
+class TurnoutProxyTestCase < Test::Unit::TestCase
+  PROJECT_ROOT = Pathname.new(File.expand_path("../../", __FILE__))
+
+  def default_test
+    # Make Test::Unit happy...
+  end
+
+  def self.test(name, &block)
+    raise ArgumentError, "Example name can't be empty" if String(name).empty?
+    define_method "test #{name}", &block
+  end
+end
+

data/turnout_proxy.gemspec

@@ -0,0 +1,35 @@
+# -*- encoding: utf-8 -*-
+
+$:.unshift File.expand_path('../lib/', __FILE__)
+
+require 'turnout_proxy'
+require 'date'
+
+Gem::Specification.new do |s|
+  s.name = "turnout_proxy"
+  s.summary = "Proxy server which allows to switch between two destinations using lock file"
+
+  s.version = TurnoutProxy::VERSION.dup
+  s.authors = ["Vladimir Yarotsky"]
+  s.date = Date.today.to_s
+  s.email = "vladimir.yarotksy@gmail.com"
+  s.homepage = "http://github.com/v-yarotsky/turnout_proxy"
+  s.licenses = ["MIT"]
+
+  s.rubygems_version = "1.8.21"
+  s.required_rubygems_version = ">= 1.3.6"
+  s.specification_version = 3
+
+  s.files            = `git ls-files`.split($/)
+  s.executables      = s.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  s.test_files       = s.files.grep(%r{^(test)/})
+  s.require_paths    = ["lib"]
+  s.extra_rdoc_files = %w[LICENSE.txt README.md]
+
+  s.add_dependency("em-proxy", "~> 0.1")
+
+  s.add_development_dependency("rake")
+  s.add_development_dependency("posix-spawn")
+end
+
+

metadata

@@ -0,0 +1,127 @@
+--- !ruby/object:Gem::Specification 
+name: turnout_proxy
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Vladimir Yarotsky
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-04-04 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: em-proxy
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 0
+        - 1
+        version: "0.1"
+  type: :runtime
+  requirement: *id001
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: rake
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  requirement: *id002
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: posix-spawn
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  requirement: *id003
+  prerelease: false
+description: 
+email: vladimir.yarotksy@gmail.com
+executables: 
+- turnout_proxy
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.md
+files: 
+- .gitignore
+- .travis.yml
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- bin/turnout_proxy
+- lib/turnout_proxy.rb
+- lib/turnout_proxy/host_chooser.rb
+- test/acceptance/test_turnout_proxy.rb
+- test/lib/turnout_proxy/test_host_chooser.rb
+- test/support/hello_server.rb
+- test/test_helper.rb
+- turnout_proxy.gemspec
+homepage: http://github.com/v-yarotsky/turnout_proxy
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 23
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Proxy server which allows to switch between two destinations using lock file
+test_files: 
+- test/acceptance/test_turnout_proxy.rb
+- test/lib/turnout_proxy/test_host_chooser.rb
+- test/support/hello_server.rb
+- test/test_helper.rb