lapin 0.1.2

data/.gitignore

@@ -0,0 +1,27 @@
+*.gem
+*.rbc
+.bundle
+.config
+coverage
+InstalledFiles
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+public
+example
+
+# YARD artifacts
+.yardoc
+_yardoc
+doc/
+.DS_Store
+config.ru.development
+.rbx
+log
+.DS_Store
+config.ru
+Gemfile.lock

data/.travis.yml

@@ -0,0 +1,14 @@
+language: ruby
+
+script: "bundle exec rake test"
+
+rvm:
+  - 1.8.7
+  - 1.9.2
+  - 1.9.3
+  - ree
+
+notifications:
+  email:
+    on_success: always
+    on_failure: always

data/Gemfile

@@ -0,0 +1,9 @@
+source "http://rubygems.org"
+
+# Load the gem's dependencies
+gemspec
+
+group :development do
+  gem "thin"
+  gem "passenger"
+end

data/README.md

@@ -0,0 +1,59 @@
+# Lapin
+
+[![Build Status](https://secure.travis-ci.org/giraffi/lapin.png?branch=master)](http://travis-ci.org/giraffi/lapin) [![Code Climate](https://codeclimate.com/badge.png)](https://codeclimate.com/github/giraffi/lapin)
+
+A sinatra-based app that provides a JSON endpoint over HTTP for interacting with an AMQP message broker.
+
+## Requirements
+
+* [RabbitMQ](http://www.rabbitmq.com/) server
+
+## Installation
+
+```bash
+$ gem install lapin
+```
+
+To use with bundler, drop in your Gemfile.
+
+```ruby
+gem "lapin", :git => "git://github.com/giraffi/lapin.git"
+```
+
+## Setup
+
+Create `config.ru` in the root folder of your app.
+
+```ruby
+require 'lapin'
+
+Lapin::Config.amqp_url = 'amqp://guest:guest@localhost/vhost'
+Lapin::Config.exchange = 'amq.direct'
+Lapin::Config.logging  = true if ENV['RACK_ENV'] == 'development'
+Lapin.options.quiet    = false
+
+run Lapin::Server
+```
+
+## Usage
+
+Just do the following to start `Lapin::Server` using [thin](https://github.com/macournoyer/thin/) for the web server.
+
+```bash
+$ thin start -R config.ru
+```
+
+Or, for [passenger](http://www.modrails.com/) (standalone version),
+
+```bash
+$ passenger start -R config.ru
+```
+
+To handle a POST request like below, you need a running RabbitMQ server according to the `Lapin::Config.amqp_url` option beforehand.
+
+```bash
+$ curl -v \
+> -H "Accept: application/json" -H "Content-type: application/json" -H "X-ROUTING-KEY: amqp.giraffi" \
+> -X POST -d '{"user":{"name":"foo", "email":"hoge@example.com"}}'  \
+> http://localhost:3000/publish.json
+```

data/Rakefile

@@ -0,0 +1,13 @@
+# encoding: utf-8
+require 'rake'
+require 'rake/testtask'
+
+desc "Default: run unit tests"
+task :default => :test
+
+desc "Test Lapin"
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end

data/examples/subscriber.rb

@@ -0,0 +1,26 @@
+#!/usr/bin/ruby
+# -*- encoding: utf-8 -*-
+
+require "amqp"
+
+AMQP.start(host: 'localhost') do |conn|
+  channel  = AMQP::Channel.new(conn)
+  exchange = channel.direct('amqp.direct')
+  queue    = channel.queue("", exclusive: true)
+  queue.bind(exchange, routing_key: 'amqp.giraffi')
+
+  Signal.trap("INT") do
+    conn.close do
+      EM.stop { exit }
+    end
+  end
+
+  puts " [*] Waiting for logs. To exit press CTRL+C"
+
+  i = 1
+  queue.subscribe do |header, payload|
+    #puts " [x] #{header}:#{payload}"
+    puts " [x] #{header} ----- #{i}"
+    i += 1
+  end
+end

data/lapin.gemspec

@@ -0,0 +1,34 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'lapin/version'
+
+Gem::Specification.new do |s|
+  s.name        = "lapin"
+  s.version     = Lapin::VERSION
+  s.authors     = ["azukiwasher"]
+  s.email       = ["azukiwasher@higanworks.com"]
+  s.homepage    = "https://github.com/giraffi/lapin.git"
+  s.summary     = %q{A sinatra-based app that redirects data received via HTTP to an AMQP message broker}
+  s.description = %q{A sinatra-based app that redirects data received via HTTP to an AMQP message broker.}
+
+  s.rubyforge_project = "lapin"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency "rack",        "~> 1.4.0"
+  s.add_dependency "bunny",       "~> 0.8.0"
+  s.add_dependency "multi_json",  "~> 1.3.0"
+  s.add_dependency "sinatra",     "~> 1.3.0"
+
+  s.add_development_dependency "rake"
+  s.add_development_dependency "yajl-ruby"
+  s.add_development_dependency "rack-test"
+  s.add_development_dependency "shoulda"
+  s.add_development_dependency "simplecov"
+  s.add_development_dependency "mocha"
+  s.add_development_dependency "thin"
+end

data/lib/lapin/config.rb

@@ -0,0 +1,43 @@
+# -*- encoding: utf-8 -*-
+require 'uri'
+
+module Lapin
+  class Config
+
+    class << self
+      attr_accessor :amqp_url, :vhost
+      attr_accessor :exchange, :routing_key
+      attr_accessor :logging
+
+      def amqp_config
+        uri = URI.parse(amqp_url)
+        {
+          :user    => uri.user,
+          :pass    => uri.password,
+          :host    => uri.host,
+          :port    => (uri.port || 5672),
+          :vhost   => @vhost || uri.path ,
+          :logging => @logging || false
+        }
+      rescue Object => e
+        raise "Invalid AMQP url: #{uri.inspect} (#{e})"
+      end
+
+      def amqp_url
+        @amqp_url ||= ENV["AMQP_URL"] || "amqp://guest:guest@localhost"
+      end
+
+      def vhost
+        @vhost
+      end
+
+      def exchange
+        @exchange ||= "amq.direct"
+      end
+
+      def routing_key
+        @routing_key ||= "*"
+      end
+    end
+  end
+end

data/lib/lapin/server.rb

@@ -0,0 +1,83 @@
+# -*- encoding: utf-8 -*-
+require 'sinatra/base'
+require 'multi_json'
+
+module Lapin
+  class Server < Sinatra::Base
+
+    # Reduce the amount of data transferred 
+    # from the app.
+    use Rack::Deflater
+
+    set :raise_error    => false
+    set :show_exception => false
+
+    helpers do
+      # define helper methods
+      def json_status(code, reason)
+        content_type "application/json"
+        status code
+        MultiJson.encode({:status => code, :reason => reason})
+      end
+
+      def validate_json(data)
+        begin
+          if MultiJson.respond_to?(:adapter)
+            MultiJson.load(data)
+          else
+            MultiJson.decode(data)
+          end
+          return true
+        rescue
+          return false
+        end
+      end
+    end
+
+    error do
+      e = request.env['sinatra.error']
+      Lapin.log "Error: #{e.class}: #{e.message}"
+      json_status 412, e.message
+    end
+
+    # Create an exchange from the current AMQP connection.
+    def exchange
+      @exchange ||= Lapin.client.exchange(Config.exchange, :type => 'direct')
+    end
+
+    # Recieve post data, validate and redirect it
+    # to an AMQP broker.
+    post '/publish.json' do
+      content_type :json
+
+      begin
+        # Set routing key from headers
+        Config.routing_key = request.env['HTTP_X_ROUTING_KEY']
+
+        # Set payload from bod
+        payload = request.body.read
+
+        if validate_json(payload)
+          exchange.publish(payload, :key => Config.routing_key)
+
+          # No Content
+          status 204
+        else
+          # Bad request
+          status 400
+        end
+      rescue Errno::ECONNRESET => e
+        Lapin.log "Error: #{e.class}: #{e.message}"
+        Lapin.client = nil
+
+        # Bad gateway
+        json_status 502, e.message
+      rescue Bunny::ConnectionError, Bunny::ServerDownError => e
+        Lapin.log "Error: #{e.class}: #{e.message}"
+
+        # Bad gateway
+        json_status 502, e.message
+      end
+    end
+  end
+end

data/lib/lapin/version.rb

@@ -0,0 +1,5 @@
+# -*- encoding: utf-8 -*-
+
+module Lapin
+  VERSION = "0.1.2"
+end

data/lib/lapin.rb

@@ -0,0 +1,38 @@
+# -*- encoding: utf-8 -*-
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+ENV['RACK_ENV'] ||= 'development'
+
+require 'bunny'
+require 'lapin/config'
+require 'lapin/server'
+
+module Lapin
+  class << self
+    attr_accessor :options
+    attr_accessor :client
+
+    def options
+      @options ||= Struct.new(:debug, :quiet).new
+      @options.debug ||= false
+      @options.quiet ||= false
+      @options
+    end
+
+    def client
+      unless @client
+        c = ::Bunny.new(Config.amqp_config)
+        c.start
+        @client = c
+      end
+      @client
+    end
+
+    def log(msg)
+      unless options.quiet
+        time = Time.now.strftime('%H:%M:%S %Y-%m-%d')
+        puts "** [#{time}] #$$: #{msg}"
+      end
+    end
+  end
+end

data/test/config_test.rb

@@ -0,0 +1,29 @@
+# encoding: utf-8
+require 'test_helper'
+
+class ConfigTest < Test::Unit::TestCase
+  context 'AMQP URL parsing' do
+    setup do
+      Lapin::Config.amqp_url = 'amqp://guest:guest@localhost:5671/my-vhost'
+    end
+
+    should 'handle all options properly' do
+      assert_equal Lapin::Config.amqp_config[:user], 'guest'
+      assert_equal Lapin::Config.amqp_config[:pass], 'guest'
+      assert_equal Lapin::Config.amqp_config[:host], 'localhost'
+      assert_equal Lapin::Config.amqp_config[:port], 5671
+      assert_equal Lapin::Config.amqp_config[:vhost], '/my-vhost'
+      assert_equal Lapin::Config.amqp_config[:logging], false
+    end
+
+    should 'overwrite vhost by the vhost attribute' do
+      Lapin::Config.vhost = 'your-vhost'
+      assert_equal Lapin::Config.amqp_config[:vhost], 'your-vhost'
+    end
+
+    should 'set routing key to * when x-routing-key is nil' do
+      Lapin::Config.routing_key = nil
+      assert_equal "*", Lapin::Config.routing_key
+    end
+  end
+end

data/test/server_test.rb

@@ -0,0 +1,83 @@
+# encoding: utf-8
+require 'test_helper'
+require 'mocha'
+
+class ServerTest < Test::Unit::TestCase
+  include Rack::Test::Methods
+
+  def app
+    Lapin::Server.new
+  end
+
+  context "POST /publish.json" do
+    setup do
+      @payload = "{\"data\": \"sample\"}"
+      Lapin::Config.amqp_url = 'amqp://guest:guest@localhost/'
+    end
+
+    teardown do
+      @payload = nil
+    end
+
+    should "return 204 with an appropriate request" do
+      m = mock()
+      Bunny.expects(:new).with(Lapin::Config.amqp_config).returns(m)
+      m.expects(:start)
+      e = mock("exchange")
+      m.stubs(:exchange).with(Lapin::Config.exchange, {:type => 'direct'}).returns(e)
+      e.stubs(:publish).with(@payload, {:key => 'giraffi.nagios'})
+
+      post "/publish.json", @payload, {"CONTENT_TYPE" => "application/json", "HTTP_X_ROUTING_KEY" => "giraffi.nagios"}
+      assert_equal last_request.body.string, @payload
+      assert_equal last_response.status, 204
+    end
+
+    should "return 400 with a malformed request body" do
+      post "/publish.json", "malformed_body", {"CONTENT_TYPE" => "text/plain"}
+      assert_equal last_response.status, 400
+    end
+  end
+
+  context "Error handling" do
+    setup do
+      Lapin.options.quiet = true
+    end
+
+    should "return 412 when the sinatra.error is catched" do
+      m = mock()
+      Bunny.expects(:new).with(Lapin::Config.amqp_config).returns(m)
+      m.stubs(:start).raises(Exception.new)
+
+      post "/publish.json", @payload, {"CONTENT_TYPE" => "application/json"}
+      assert_equal last_response.status, 412
+    end
+
+    should "return 502 when the connection is reset by peer" do
+      m = mock()
+      Bunny.expects(:new).with(Lapin::Config.amqp_config).returns(m)
+      m.stubs(:start).raises(Errno::ECONNRESET.new)
+
+      post "/publish.json", @payload, {"CONTENT_TYPE" => "application/json"}
+      assert_equal last_response.status, 502
+    end
+
+    should "return 502 when the no connection and no socket are found" do
+      m = mock()
+      Bunny.expects(:new).with(Lapin::Config.amqp_config).returns(m)
+      m.stubs(:start).raises(Bunny::ConnectionError.new)
+
+      post "/publish.json", @payload, {"CONTENT_TYPE" => "application/json"}
+      assert_equal last_response.status, 502
+    end
+
+    should "return 502 when the broken pipe is catched" do
+      m = mock()
+      Bunny.expects(:new).with(Lapin::Config.amqp_config).returns(m)
+      m.stubs(:start).raises(Bunny::ServerDownError.new)
+
+      post "/publish.json", @payload, {"CONTENT_TYPE" => "application/json"}
+      assert_equal last_response.status, 502
+    end
+
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+
+require 'rubygems'
+require 'bundler'
+
+# Configure Rack Environment
+ENV['RACK_ENV'] = "test"
+
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundleError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+
+require 'test/unit'
+require 'shoulda'
+require 'rack/test'
+require 'simplecov'
+SimpleCov.start
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'lapin'
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,241 @@
+--- !ruby/object:Gem::Specification
+name: lapin
+version: !ruby/object:Gem::Version
+  version: 0.1.2
+  prerelease: 
+platform: ruby
+authors:
+- azukiwasher
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rack
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.4.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.4.0
+- !ruby/object:Gem::Dependency
+  name: bunny
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: shoulda
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thin
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A sinatra-based app that redirects data received via HTTP to an AMQP
+  message broker.
+email:
+- azukiwasher@higanworks.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .travis.yml
+- Gemfile
+- README.md
+- Rakefile
+- examples/subscriber.rb
+- lapin.gemspec
+- lib/lapin.rb
+- lib/lapin/config.rb
+- lib/lapin/server.rb
+- lib/lapin/version.rb
+- test/config_test.rb
+- test/server_test.rb
+- test/test_helper.rb
+homepage: https://github.com/giraffi/lapin.git
+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: lapin
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: A sinatra-based app that redirects data received via HTTP to an AMQP message
+  broker
+test_files:
+- test/config_test.rb
+- test/server_test.rb
+- test/test_helper.rb
+has_rdoc: