unicorn-cuba-base 1.0.0

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -0,0 +1,18 @@
+source "http://rubygems.org"
+
+ruby "1.9.3"
+
+gem "cuba", "~> 3.0"
+gem "unicorn", ">= 4.6.2"
+gem "raindrops", "~> 0.11"
+gem "cli", "~> 1.1.0"
+gem "facter", "~> 1.6.11"
+gem "ruby-ip", "~> 0.9"
+
+# Add dependencies to develop your gem here.
+# Include everything needed to run rake, tests, features, etc.
+group :development do
+  gem "rspec", "~> 2.13"
+  gem "rdoc", "~> 3.9"
+  gem "jeweler", "~> 1.8.4"
+end

data/Gemfile.lock

@@ -0,0 +1,48 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    cli (1.1.1)
+    cuba (3.1.0)
+      rack
+    diff-lcs (1.2.4)
+    facter (1.6.18)
+    git (1.2.5)
+    jeweler (1.8.4)
+      bundler (~> 1.0)
+      git (>= 1.2.5)
+      rake
+      rdoc
+    json (1.7.7)
+    kgio (2.8.0)
+    rack (1.5.2)
+    raindrops (0.11.0)
+    rake (10.0.4)
+    rdoc (3.12.2)
+      json (~> 1.4)
+    rspec (2.13.0)
+      rspec-core (~> 2.13.0)
+      rspec-expectations (~> 2.13.0)
+      rspec-mocks (~> 2.13.0)
+    rspec-core (2.13.1)
+    rspec-expectations (2.13.0)
+      diff-lcs (>= 1.1.3, < 2.0)
+    rspec-mocks (2.13.1)
+    ruby-ip (0.9.1)
+    unicorn (4.6.2)
+      kgio (~> 2.6)
+      rack
+      raindrops (~> 0.7)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  cli (~> 1.1.0)
+  cuba (~> 3.0)
+  facter (~> 1.6.11)
+  jeweler (~> 1.8.4)
+  raindrops (~> 0.11)
+  rdoc (~> 3.9)
+  rspec (~> 2.13)
+  ruby-ip (~> 0.9)
+  unicorn (>= 4.6.2)

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2013 Jakub Pastuszek
+
+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,29 @@
+# unicorn-cuba-base
+
+It is web application base powered by Unicorn HTTP server and based on Cuba framework extended with additional Rack middleware and Cuba plugins.
+
+This gem was created to deduplicate code from [httpthumbnailer](http://github.com/jpastuszek/httpthumbnailer) and [httpimagestore](http://github.com/jpastuszek/httpimagestore).
+
+## Installing
+
+    gem install unicorn-cuba-base
+
+## Usage
+
+Please see [httpthumbnailer main executable](https://github.com/jpastuszek/httpthumbnailer/blob/master/bin/httpthumbnailer) and [httpimagestore main executable](https://github.com/jpastuszek/httpimagestore/blob/master/bin/httpimagestore) for usage example.
+
+## Contributing to unicorn-cuba-base
+ 
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it
+* Fork the project
+* Start a feature/bugfix branch
+* Commit and push until you are happy with your contribution
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+## Copyright
+
+Copyright (c) 2013 Jakub Pastuszek. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,49 @@
+# encoding: utf-8
+
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'rake'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.name = "unicorn-cuba-base"
+  gem.homepage = "http://github.com/jpastuszek/unicorn-cuba-base"
+  gem.license = "MIT"
+  gem.summary = "web appliaction base powered by Unicorn and Cuba"
+  gem.description = "web application base powered by Unicorn HTTP server and based on Cuba framework extended with additional Rack middleware and Cuba plugins"
+  gem.email = "jpastuszek@gmail.com"
+  gem.authors = ["Jakub Pastuszek"]
+  # dependencies defined in Gemfile
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+end
+
+RSpec::Core::RakeTask.new(:rcov) do |spec|
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :default => :spec
+
+require 'rdoc/task'
+RDoc::Task.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "unicorn-cuba-base #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+1.0.0

data/lib/unicorn-cuba-base.rb

@@ -0,0 +1,161 @@
+require 'cli'
+require 'cuba'
+require 'unicorn'
+require 'unicorn/launcher'
+require 'facter'
+require 'pathname'
+require 'ip'
+
+require_relative 'unicorn-cuba-base/stats'
+require_relative 'unicorn-cuba-base/root_logger'
+require_relative 'unicorn-cuba-base/plugin/error_matcher'
+require_relative 'unicorn-cuba-base/plugin/logging'
+require_relative 'unicorn-cuba-base/plugin/response_helpers'
+require_relative 'unicorn-cuba-base/plugin/memory_limit'
+require_relative 'unicorn-cuba-base/rack/error_handling'
+require_relative 'unicorn-cuba-base/rack/unhandled_request'
+require_relative 'unicorn-cuba-base/rack/memory_limit'
+
+class Controler < Cuba
+	include ClassLogging
+end
+
+require_relative 'unicorn-cuba-base/stats_reporter'
+
+class Application
+	def cli(&block)
+		@cli_setup = block
+	end
+
+	def settings(&block)
+		@settings_setup = block
+	end
+
+	def main(&block)
+		@main_setup = block
+	end
+
+	def after_fork(&block)
+		@after_fork = block
+	end
+
+	def initialize(program_name, defaults = {}, &block)
+		instance_eval &block
+
+		@cli = setup_cli(program_name, defaults, @cli_setup) or fail 'no cli defined'
+		@settings = @settings_setup ? setup_settings(@settings_setup) : @cli.parse!
+
+		root_logger = RootLogger.new(STDERR)
+		root_logger.level = RootLogger::WARN
+		root_logger.level = RootLogger::INFO if @settings.verbose
+		root_logger.level = RootLogger::DEBUG if @settings.debug
+		Controler.logger = root_logger
+		MemoryLimit.logger = Controler.logger_for(MemoryLimit)
+
+		unicorn_settings = {}
+		unicorn_settings[:logger] = root_logger.logger_for(Unicorn::HttpServer)
+		unicorn_settings[:pid] = @settings.pid_file.to_s
+		unicorn_settings[:worker_processes] = @settings.worker_processes
+		unicorn_settings[:timeout] = @settings.worker_timeout
+		unicorn_settings[:listeners] = @settings.listener
+		unicorn_settings[:user] = @settings.user if @settings.user
+		unicorn_settings[:rewindable_input] = false # don't keep the upload data in memory or on disk (tmp)
+		unicorn_settings[:after_fork] = @after_fork if @after_fork
+
+		unless @settings.foreground
+			unicorn_settings[:stderr_path] = @settings.log_file.to_s
+			unicorn_settings[:stdout_path] = @settings.log_file.to_s
+
+			Unicorn::Launcher.daemonize!(unicorn_settings) 
+
+			# capture startup messages
+			@settings.log_file.open('ab') do |log|
+				log.sync = true
+				STDERR.reopen log
+				STDOUT.reopen log
+			end
+		end
+
+		Controler.settings[:listeners] = @settings.listener
+		Controler.settings[:access_log_file] = @settings.access_log_file
+
+		Controler.plugin Plugin::ErrorMatcher
+		Controler.plugin Plugin::Logging
+		Controler.plugin Plugin::ResponseHelpers
+		Controler.plugin Plugin::MemoryLimit
+
+		@main_setup or fail 'no main controler provided'
+		main_controler = setup_main(@main_setup) or fail 'no main controler class returned'
+
+		access_log_file = @settings.access_log_file.open('a+')
+		access_log_file.sync = true
+		main_controler.use Rack::CommonLogger, access_log_file
+		main_controler.use Rack::ErrorHandling
+		main_controler.use Rack::UnhandledRequest
+		main_controler.use Rack::MemoryLimit, @settings.limit_memory * 1024 ** 2
+
+		Unicorn::HttpServer.new(main_controler, unicorn_settings).start.join
+	end
+
+	private
+
+	def setup_cli(program_name, defaults, block)
+		CLI.new do
+			instance_eval &block
+			option :log_file,
+				short: :l,
+				cast: Pathname,
+				description: 'log file location',
+				default: "#{program_name}.log"
+			option :access_log_file,
+				short: :a,
+				cast: Pathname,
+				description: 'NCSA access log file location',
+				default: "#{program_name}_access.log"
+			option :pid_file,
+				short: :p,
+				cast: Pathname,
+				description: 'PID file location',
+				default: "#{program_name}.pid"
+			switch :foreground,
+				short: :f,
+				description: 'stay in foreground'
+			options :listener,
+				short: :L,
+				description: 'HTTP server listener (bind) address in format <ip>:<port> or unix:<file> or ~[<username>]/<file> for UNIX sockets',
+				default: '127.0.0.1:' + (defaults[:port] || 8080).to_s
+			option :user,
+				short: :u,
+				description: 'run worker processes as given user'
+			option :worker_processes,
+				short: :w,
+				cast: Integer,
+				description: 'start given number of worker processes',
+				default: (Facter.processorcount.to_i + 1) * (defaults[:processor_count_factor] || 1)
+			option :worker_timeout,
+				short: :t,
+				cast: Integer,
+				description: 'workers handling the request taking longer than this time period will be forcibly killed',
+				default: 300
+			option :limit_memory,
+				cast: Integer,
+				description: 'memory usage limit in MiB',
+				default: 128
+			switch :verbose,
+				short: :v,
+				description: 'enable verbose logging (INFO)'
+			switch :debug,
+				short: :d,
+				description: 'enable verbose and debug logging (DEBUG)'
+		end
+	end
+
+	def setup_settings(block)
+		@cli.parse!(&block)
+	end
+
+	def setup_main(block)
+		block.call @settings
+	end
+end
+

data/lib/unicorn-cuba-base/memory_limit.rb

@@ -0,0 +1,62 @@
+require 'unicorn-cuba-base/root_logger'
+
+class MemoryLimit
+	include ClassLogging
+
+	class MemoryLimitedExceededError < RuntimeError
+		def initialize
+			super "memory limit exceeded"
+		end
+	end
+
+	module IO
+		def root_limit(ml)
+			@root_limit = ml
+		end
+		
+		def read(bytes = nil)
+			data = @root_limit.get do |max_read_bytes|
+				if not bytes or bytes > max_read_bytes
+					data = super max_read_bytes
+					raise MemoryLimitedExceededError unless eof?
+					data or '' # read() always returns '' on EOF
+				else
+					super bytes or ''
+				end
+			end
+		end
+	end	
+
+	def initialize(bytes = 256 * 1024 ** 2)
+		log.info "using memory limit of #{bytes} bytes" if bytes
+		@limit = bytes
+	end
+
+	attr_reader :limit
+
+	def get
+		yield(@limit).tap do |data|
+			borrow data.bytesize if data
+		end
+	end
+
+	def borrow(bytes)
+		log.debug "borrowing #{bytes} from #{@limit} bytes of limit"
+		bytes > @limit and raise MemoryLimitedExceededError
+		@limit -= bytes
+		bytes
+	end
+
+	def return(bytes)
+		log.debug "returning #{bytes} to #{@limit} bytes of limit"
+		@limit += bytes
+		bytes
+	end
+
+	def io(io)
+		io.extend MemoryLimit::IO
+		io.root_limit self
+		io
+	end
+end
+

data/lib/unicorn-cuba-base/plugin/error_matcher.rb

@@ -0,0 +1,12 @@
+module Plugin
+  module ErrorMatcher
+    def error(*klass)
+      lambda {klass.any?{|k| env["app.error"].is_a? k} and captures.push(env["app.error"])}
+    end
+
+    def error?
+      env.has_key? "app.error"
+    end
+  end
+end
+

data/lib/unicorn-cuba-base/plugin/logging.rb

@@ -0,0 +1,9 @@
+module Plugin
+	module Logging
+		def log
+			return @logger if defined? @logger
+			@logger = self.class.logger_for(self.class)
+		end
+	end
+end
+

data/lib/unicorn-cuba-base/plugin/memory_limit.rb

@@ -0,0 +1,8 @@
+module Plugin
+	module MemoryLimit
+		def memory_limit
+			env["app.memory_limit"] or fail 'Rack::MemoryLimit middleware not used!'
+		end
+	end
+end
+

data/lib/unicorn-cuba-base/plugin/response_helpers.rb

@@ -0,0 +1,83 @@
+require 'securerandom'
+require 'unicorn-cuba-base/stats'
+
+module Plugin
+	module ResponseHelpers
+		extend Stats
+		def_stats(
+			:total_write_multipart,
+			:total_write, 
+			:total_write_part, 
+			:total_write_error, 
+			:total_write_error_part
+		)
+
+		def write(code, content_type, body)
+			req.body.read # read all remaining upload before we send response so that client will read it
+			res.status = code
+			res["Content-Type"] = content_type
+			ResponseHelpers.stats.incr_total_write
+			res.write body
+		end
+
+		def write_plain(code, msg)
+			msg = msg.join("\r\n") if msg.is_a? Array
+			write code, 'text/plain', msg.gsub(/(?<!\r)\n/, "\r\n") + "\r\n"
+		end
+
+		def write_url_list(code, msg)
+			msg = msg.join("\r\n") if msg.is_a? Array
+			write code, 'text/uri-list', msg.gsub(/(?<!\r)\n/, "\r\n") + "\r\n"
+		end
+
+		def write_error(code, error)
+			msg = error.message
+			log.warn "sending #{code} error response: #{msg}"
+			ResponseHelpers.stats.incr_total_write_error
+			write_plain code, msg
+		end
+
+		def write_url_list(code, urls)
+			write code, 'text/uri-list', urls.join("\r\n") + "\r\n"
+		end
+
+		# Multipart
+		def write_preamble(code, headers = {})
+			res.status = code
+			@boundary = SecureRandom.uuid
+			res["Content-Type"] = "multipart/mixed; boundary=\"#{@boundary}\""
+			headers.each do |key, value|
+				res[key] = value
+			end
+			ResponseHelpers.stats.incr_total_write_multipart
+		end
+
+		def write_part(content_type, body, headers = {})
+			res.write "--#{@boundary}\r\n"
+			res.write "Content-Type: #{content_type}\r\n"
+			headers.each_pair do |name, value|
+				res.write "#{name}: #{value}\r\n"
+			end
+			res.write "\r\n"
+			ResponseHelpers.stats.incr_total_write_part
+			res.write body
+			res.write "\r\n"
+		end
+
+		def write_plain_part(msg, headers = {})
+			write_part 'text/plain', msg.to_s.gsub("\n", "\r\n"), headers
+		end
+
+		def write_error_part(code, error)
+			msg = error.message
+			log.warn "sending error in multipart response part: #{msg}"
+			ResponseHelpers.stats.incr_total_write_error_part
+			write_plain_part msg, 'Status' => code
+		end
+
+		def write_epilogue
+			res.write "--#{@boundary}--\r\n"
+		end
+	end
+end
+

data/lib/unicorn-cuba-base/rack/error_handling.rb

@@ -0,0 +1,31 @@
+require 'pp'
+
+module Rack
+	class ErrorHandling
+		def initialize(app, &block)
+			@app = app
+		end
+
+		def call(env)
+			# save original env
+			orig_env = env.dup
+			begin
+				return @app.call(env)
+			rescue => error
+				begin
+					# reset env to original since it could have been changed
+					env.clear
+					env.merge!(orig_env)
+
+					# set error so app can handle it
+					env["app.error"] = error
+
+					return @app.call(env)
+				rescue
+					raise 
+				end
+			end
+		end
+	end
+end
+

data/lib/unicorn-cuba-base/rack/memory_limit.rb

@@ -0,0 +1,20 @@
+require_relative '../memory_limit'
+
+module Rack
+	class MemoryLimit
+		def initialize(app, memory_limit)
+			@app = app
+			@memory_limit = memory_limit
+		end
+
+		def call(env)
+			memory_limit = ::MemoryLimit.new(@memory_limit)
+			env["app.memory_limit"] = memory_limit
+
+			# use up limit when reading request data
+			memory_limit.io env["rack.input"]
+			return @app.call(env)
+		end
+	end
+end
+

data/lib/unicorn-cuba-base/rack/unhandled_request.rb

@@ -0,0 +1,22 @@
+module Rack
+	class UnhandledRequest
+		class UnhandledRequestError < ArgumentError
+      attr_reader :uri
+			def initialize(uri)
+        @uri = uri
+				super "request for URI '#{uri}' was not handled by the server"
+			end
+		end
+
+		def initialize(app)
+			@app = app
+		end
+
+		def call(env)
+			status, headers, body = @app.call(env)
+			raise UnhandledRequestError, env['SCRIPT_NAME'] + env['PATH_INFO'] if body == [] and (status == 200 or status == 404)
+			[status, headers, body]
+		end
+	end
+end
+

data/lib/unicorn-cuba-base/root_logger.rb

@@ -0,0 +1,110 @@
+require 'logger'
+
+class RootLogger < Logger
+	class ClassLogger
+		@@levels = [:debug, :info, :warn, :error, :fatal, :unknown]
+
+		def initialize(root_logger, class_obj)
+			@root_logger = root_logger
+			@progname = class_obj.name
+			@root_logger.formatter = proc do |severity, datetime, progname, msg|
+				"[#{datetime.utc.strftime "%Y-%m-%d %H:%M:%S.%6N %Z"}] [#{$$} #{progname}] #{severity}: #{msg}\n"
+			end
+		end
+
+		def respond_to?(method)
+			super or @root_logger.respond_to? method
+		end
+
+		def method_missing(name, *args, &block)
+			if @@levels.include? name
+				message = if block_given?
+					self.progname
+				else
+					args.map do |arg|
+						if arg.is_a? Exception
+							"#{arg.class.name}: #{arg.message}\n#{arg.backtrace.join("\n")}"
+						else
+							arg.to_s
+						end
+					end.join(': ')
+				end
+
+				# set program name to current class
+				@root_logger.progname = @progname
+				@root_logger.send(name, message.chomp, &block)
+			else
+				@root_logger.send(name, *args, &block)
+			end
+		end
+
+		attr_reader :root_logger
+
+		def inspect
+			"#<ClassLogger[#{@progname}] #{"0x%X" % object_id} root_logger=#{@root_logger.inspect}>"
+		end
+	end
+
+	def logger_for(class_obj)
+		ClassLogger.new(self, class_obj)
+	end
+
+	def root_logger
+		self
+	end
+
+	def inspect
+		"#<RootLogger #{"0x%X" % object_id}>"
+	end
+end
+
+module ClassLogging
+	module ClassMethods
+		def init_logger
+			@@logger = {} unless defined? @@logger
+		end
+
+		def logger=(logger)
+			@@logger[self] = logger
+		end
+
+		def log
+			unless @@logger[self]
+				new_root_logger = false
+				# use root logger from ancestor or create new one
+				root_logger = 
+					if logging_class = ancestors.find{|an| an != self and an.respond_to? :log}
+						logging_class.log.respond_to?(:root_logger) ? logging_class.log.root_logger : logging_class.log
+					else
+						new_root_logger = true
+						Logger.new(STDERR)
+					end
+
+				root_logger.kind_of? RootLogger::ClassLogger and fail "got ClassLogger root logger: #{self}"
+
+				logger = RootLogger::ClassLogger.new(root_logger, self)
+				logger.warn "new default logger crated" if new_root_logger
+				@@logger[self] = logger
+			end
+			@@logger[self]
+		end
+
+		def logger_for(class_obj)
+			RootLogger::ClassLogger.new(log.root_logger, class_obj)
+		end
+	end
+
+	def log
+		self.class.log
+	end
+
+	def logger_for(class_obj)
+		self.class.logger_for(class_obj)
+	end
+
+	def self.included(class_obj)
+		class_obj.extend ClassMethods
+		class_obj.init_logger
+	end
+end
+

data/lib/unicorn-cuba-base/stats.rb

@@ -0,0 +1,30 @@
+require 'raindrops'
+
+module Stats
+	class MyStruct < Raindrops::Struct
+		def self.new(*members)
+			klass = super(*members)
+
+			str = ''
+			# add support to increment by more than 1
+			members.map { |x| x.to_sym }.each_with_index do |member, i|
+				str << "def incr_#{member}(v = 1); @raindrops.incr(#{i}, v); end; "
+				str << "def decr_#{member}(v = 1); @raindrops.decr(#{i}, v); end; "
+			end
+
+			klass.class_eval(str)
+			klass
+		end
+	end
+
+	def def_stats(*stat_names)
+		@@local_stats ||= {}
+		stats_class = eval "MyStruct.new(#{stat_names.map{|s| ":#{s.to_s}"}.join(', ')})"
+		@@local_stats[self] = stats_class.new
+	end
+
+	def stats
+		@@local_stats[self]
+	end
+end
+

data/lib/unicorn-cuba-base/stats_reporter.rb

@@ -0,0 +1,23 @@
+class StatsReporter < Controler
+	def self.<<(stats)
+		(@@stats ||= []) << stats
+	end
+	
+	self.define do
+		all_stats = {}
+		@@stats.each do |stats|
+			stats.class::MEMBERS.each.with_index.map do |stat, index|
+				all_stats[stat] = stats[index]
+			end
+		end
+
+		on :stat do |stat|
+			write_plain 200, all_stats[stat.to_sym].to_s || raise(ArgumentError, "unknown stat #{stat}")
+		end
+
+		on true do
+			write_plain 200, all_stats.map{|stat, value| "#{stat}: #{value}"}.join("\n")
+		end
+	end
+end
+

data/spec/memory_limit_spec.rb

@@ -0,0 +1,66 @@
+require_relative 'spec_helper'
+require 'unicorn-cuba-base/memory_limit'
+MemoryLimit.logger = Logger.new('/dev/null')
+
+describe MemoryLimit do
+	subject do
+		MemoryLimit.new(10)
+	end
+
+	it 'should raise MemoryLimitedExceededError when too much memory is borrowed' do
+		subject.borrow 8
+		subject.return 3
+		subject.borrow 4
+		subject.borrow 1
+		expect {
+			subject.borrow 1
+		}.to raise_error MemoryLimit::MemoryLimitedExceededError, 'memory limit exceeded'
+	end
+
+	describe '#get' do
+		it 'should yield limit left and borrow as much as returned string bytesize' do
+			subject.get do |limit|
+				limit.should == 10
+				'123'
+			end.should == '123'
+			subject.limit.should == 7
+		end
+
+		it 'should raise MemoryLimit::MemoryLimitedExceededError if retruned string is longer than the limit' do
+			expect {
+				subject.get do |limit|
+					'12345678901'
+				end
+			}.to raise_error MemoryLimit::MemoryLimitedExceededError
+		end
+	end
+
+	describe MemoryLimit::IO do
+		it 'should limit reading from extended IO like object' do
+			test_file = Pathname.new('/tmp/memlimtest')
+			test_file.open('w+') { |io|
+				io.write '12345'
+				io.seek 0
+
+				subject.io io
+				io.read.should == '12345'
+
+				io.seek 0
+				io.truncate 0
+				io.write '678'
+				io.seek 0
+				io.read.should == '678'
+
+				io.seek 0
+				io.write '09123'
+				io.seek 0
+
+				expect {
+					io.read
+				}.to raise_error MemoryLimit::MemoryLimitedExceededError, 'memory limit exceeded'
+			}
+			test_file.unlink
+		end
+	end
+end
+

data/spec/root_logger_spec.rb

@@ -0,0 +1,58 @@
+require_relative 'spec_helper'
+
+require 'unicorn-cuba-base/root_logger'
+require 'stringio'
+
+describe  do
+	let! :log_out do
+		StringIO.new
+	end
+
+	subject do
+		RootLogger.new(log_out)
+	end
+
+	it 'should log to given logger' do
+		subject.info 'hello world'
+		log_out.string.should include 'INFO'
+		log_out.string.should include 'hello world'
+	end
+
+	describe 'support for different logging levels' do
+		it 'should log info' do
+			subject.info 'hello world'
+			log_out.string.should include 'INFO'
+		end
+
+		it 'should log warn' do
+			subject.warn 'hello world'
+			log_out.string.should include 'WARN'
+		end
+
+		it 'should log error' do
+			subject.error 'hello world'
+			log_out.string.should include 'ERROR'
+		end
+	end
+
+	describe 'class logger' do
+		it 'should report class name' do
+			TestApp = Class.new
+			subject.logger_for(TestApp).info 'hello world'
+			log_out.string.should include 'TestApp'
+
+			subject.logger_for(String).info 'hello world'
+			log_out.string.should include 'String'
+		end
+
+		it 'should log exceptions' do
+			begin
+				raise RuntimeError, 'bad luck'
+			rescue => error
+				subject.logger_for(String).error 'hello world', error
+			end
+			log_out.string.should include 'hello world: RuntimeError: bad luck'
+		end
+	end
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,12 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'rspec'
+require 'unicorn-cuba-base'
+
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+  
+end

data/unicorn-cuba-base.gemspec

@@ -0,0 +1,87 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "unicorn-cuba-base"
+  s.version = "1.0.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Jakub Pastuszek"]
+  s.date = "2013-07-16"
+  s.description = "web application base powered by Unicorn HTTP server and based on Cuba framework extended with additional Rack middleware and Cuba plugins"
+  s.email = "jpastuszek@gmail.com"
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.md"
+  ]
+  s.files = [
+    ".document",
+    ".rspec",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.md",
+    "Rakefile",
+    "VERSION",
+    "lib/unicorn-cuba-base.rb",
+    "lib/unicorn-cuba-base/memory_limit.rb",
+    "lib/unicorn-cuba-base/plugin/error_matcher.rb",
+    "lib/unicorn-cuba-base/plugin/logging.rb",
+    "lib/unicorn-cuba-base/plugin/memory_limit.rb",
+    "lib/unicorn-cuba-base/plugin/response_helpers.rb",
+    "lib/unicorn-cuba-base/rack/error_handling.rb",
+    "lib/unicorn-cuba-base/rack/memory_limit.rb",
+    "lib/unicorn-cuba-base/rack/unhandled_request.rb",
+    "lib/unicorn-cuba-base/root_logger.rb",
+    "lib/unicorn-cuba-base/stats.rb",
+    "lib/unicorn-cuba-base/stats_reporter.rb",
+    "spec/memory_limit_spec.rb",
+    "spec/root_logger_spec.rb",
+    "spec/spec_helper.rb",
+    "unicorn-cuba-base.gemspec"
+  ]
+  s.homepage = "http://github.com/jpastuszek/unicorn-cuba-base"
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.25"
+  s.summary = "web appliaction base powered by Unicorn and Cuba"
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<cuba>, ["~> 3.0"])
+      s.add_runtime_dependency(%q<unicorn>, [">= 4.6.2"])
+      s.add_runtime_dependency(%q<raindrops>, ["~> 0.11"])
+      s.add_runtime_dependency(%q<cli>, ["~> 1.1.0"])
+      s.add_runtime_dependency(%q<facter>, ["~> 1.6.11"])
+      s.add_runtime_dependency(%q<ruby-ip>, ["~> 0.9"])
+      s.add_development_dependency(%q<rspec>, ["~> 2.13"])
+      s.add_development_dependency(%q<rdoc>, ["~> 3.9"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.8.4"])
+    else
+      s.add_dependency(%q<cuba>, ["~> 3.0"])
+      s.add_dependency(%q<unicorn>, [">= 4.6.2"])
+      s.add_dependency(%q<raindrops>, ["~> 0.11"])
+      s.add_dependency(%q<cli>, ["~> 1.1.0"])
+      s.add_dependency(%q<facter>, ["~> 1.6.11"])
+      s.add_dependency(%q<ruby-ip>, ["~> 0.9"])
+      s.add_dependency(%q<rspec>, ["~> 2.13"])
+      s.add_dependency(%q<rdoc>, ["~> 3.9"])
+      s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+    end
+  else
+    s.add_dependency(%q<cuba>, ["~> 3.0"])
+    s.add_dependency(%q<unicorn>, [">= 4.6.2"])
+    s.add_dependency(%q<raindrops>, ["~> 0.11"])
+    s.add_dependency(%q<cli>, ["~> 1.1.0"])
+    s.add_dependency(%q<facter>, ["~> 1.6.11"])
+    s.add_dependency(%q<ruby-ip>, ["~> 0.9"])
+    s.add_dependency(%q<rspec>, ["~> 2.13"])
+    s.add_dependency(%q<rdoc>, ["~> 3.9"])
+    s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+  end
+end
+

metadata

@@ -0,0 +1,219 @@
+--- !ruby/object:Gem::Specification
+name: unicorn-cuba-base
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Jakub Pastuszek
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-07-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: cuba
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: unicorn
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 4.6.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 4.6.2
+- !ruby/object:Gem::Dependency
+  name: raindrops
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.11'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.11'
+- !ruby/object:Gem::Dependency
+  name: cli
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: facter
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.11
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.11
+- !ruby/object:Gem::Dependency
+  name: ruby-ip
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.13'
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.9'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.8.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.8.4
+description: web application base powered by Unicorn HTTP server and based on Cuba
+  framework extended with additional Rack middleware and Cuba plugins
+email: jpastuszek@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.md
+files:
+- .document
+- .rspec
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- lib/unicorn-cuba-base.rb
+- lib/unicorn-cuba-base/memory_limit.rb
+- lib/unicorn-cuba-base/plugin/error_matcher.rb
+- lib/unicorn-cuba-base/plugin/logging.rb
+- lib/unicorn-cuba-base/plugin/memory_limit.rb
+- lib/unicorn-cuba-base/plugin/response_helpers.rb
+- lib/unicorn-cuba-base/rack/error_handling.rb
+- lib/unicorn-cuba-base/rack/memory_limit.rb
+- lib/unicorn-cuba-base/rack/unhandled_request.rb
+- lib/unicorn-cuba-base/root_logger.rb
+- lib/unicorn-cuba-base/stats.rb
+- lib/unicorn-cuba-base/stats_reporter.rb
+- spec/memory_limit_spec.rb
+- spec/root_logger_spec.rb
+- spec/spec_helper.rb
+- unicorn-cuba-base.gemspec
+homepage: http://github.com/jpastuszek/unicorn-cuba-base
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -3548490658984311397
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: web appliaction base powered by Unicorn and Cuba
+test_files: []