hatetepe 0.0.1

data/Gemfile

@@ -0,0 +1,6 @@
+source :rubygems
+
+gemspec
+
+gem "rake"
+gem "awesome_print"

data/README.md

@@ -0,0 +1,13 @@
+Builds and parses HTTP messages
+===============================
+
+Hatetepe combines its own builder with http_parser.rb to make dealing with HTTP
+messages as comfortable as possible.
+
+TODO
+----
+
+- Fix http_parser.rb's parsing of chunked bodies
+- Does http_parser.rb recognize trailing headers?
+- Support for pausing and resuming parsing/building
+- Encoding support (see https://github.com/tmm1/http_parser.rb/pull/1)

data/Rakefile

@@ -0,0 +1,11 @@
+require "bundler"
+Bundler.setup :default
+
+task :default => :test
+
+require "rake/testtask"
+Rake::TestTask.new :test do |t|
+  t.test_files = FileList["test/*_test.rb"]
+end
+
+Bundler::GemHelper.install_tasks

data/example.rb

@@ -0,0 +1,29 @@
+require "bundler"
+Bundler.setup
+
+require "hatetepe"
+require "awesome_print"
+
+Hatetepe::Parser.parse do
+  [:request, :response, :header, :body_chunk, :complete, :error].each do |hook|
+    send :"on_#{hook}" do |*args|
+      puts "on_#{hook}: #{args.inspect}"
+    end
+  end
+  
+  self << "HTTP/1.1 200 OK\r\nContent-Length: 11\r\n\r\nHallo Welt!"
+end
+
+Hatetepe::Builder.build do
+  on_write {|data| p data }
+  on_complete {|bytes_written| puts "Wrote #{bytes_written} bytes" }
+
+  response 200
+  header "Content-Type", "text/html", "utf-8"
+  raw_header "Content-Length: 25"
+  body "<p>Hallo Welt!</p>"
+  
+  response 201
+  header "Location", "/new_entity"
+  complete
+end

data/hatetepe.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "hatetepe/version"
+
+Gem::Specification.new do |s|
+  s.name        = "hatetepe"
+  s.version     = Hatetepe::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Lars Gierth"]
+  s.email       = ["lars.gierth@gmail.com"]
+  s.homepage    = "https://github.com/lgierth/hatetepe"
+  s.summary     = %q{Builds and parses HTTP messages}
+  s.description = %q{Hatetepe combines its own builder with http_parser.rb to make dealing with HTTP messages as comfortable as possible.}
+
+  s.add_dependency "http_parser.rb"
+  
+  s.add_development_dependency "test-unit"
+
+  s.files         = `git ls-files`.split("\n") - [".gitignore"]
+  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"]
+end

data/lib/hatetepe.rb

@@ -0,0 +1,3 @@
+require "hatetepe/builder"
+require "hatetepe/parser"
+require "hatetepe/version"

data/lib/hatetepe/builder.rb

@@ -0,0 +1,142 @@
+require "hatetepe/status"
+
+module Hatetepe
+  class BuilderError < StandardError; end
+  
+  class Builder
+    def self.build(&block)
+      message = ""
+      builder = new do |b|
+        b.on_write {|data| message << data }
+      end
+      
+      block.arity == 0 ? builder.instance_eval(&block) : block.call(builder)
+      
+      builder.complete
+      return message.empty? ? nil : message
+    end
+    
+    attr_reader :state, :bytes_written
+    
+    def initialize(&block)
+      reset
+      @on_write, @on_complete, @on_error = [], [], []
+      
+      if block
+        block.arity == 0 ? instance_eval(&block) : block.call(self)
+      end
+    end
+    
+    def reset
+      @state = :ready
+      @chunked = true
+      @bytes_written = 0
+    end
+    
+    [:write, :complete, :error].each do |hook|
+      define_method :"on_#{hook}" do |&block|
+        store = instance_variable_get(:"@on_#{hook}")
+        return store unless block
+        store << block
+      end
+    end
+    
+    def ready?
+      state == :ready
+    end
+    
+    def writing_headers?
+      state == :writing_headers
+    end
+    
+    def writing_body?
+      state == :writing_body
+    end
+    
+    def writing_trailing_headers?
+      state == :writing_trailing_headers
+    end
+    
+    def chunked?
+      @chunked
+    end
+    
+    def request(verb, uri, version = "1.1")
+      complete unless ready?
+      write "#{verb.upcase} #{uri} HTTP/#{version}\r\n"
+      @state = :writing_headers
+    end
+    
+    def response(code, version = "1.1")
+      complete unless ready?
+      unless status = STATUS_CODES[code]
+        error "Unknown status code: #{code}"
+      end
+      write "HTTP/#{version} #{code} #{status}\r\n"
+      @state = :writing_headers
+    end
+    
+    def header(name, value, charset = nil)
+      charset = charset ? "; charset=#{charset}" : ""
+      raw_header "#{name}: #{value}#{charset}"
+    end
+    
+    def raw_header(header)
+      if ready?
+        error "A request or response line is required before writing headers"
+      elsif writing_body?
+        error "Trailing headers require chunked transfer encoding" unless chunked?
+        write "0\r\n"
+        @state = :writing_trailing_headers
+      end
+      
+      if header[0..13] == "Content-Length"
+        @chunked = false
+      end
+      
+      write "#{header}\r\n"
+    end
+    
+    def body(chunk)
+      if ready?
+        error "A request or response line and headers are required before writing body"
+      elsif writing_trailing_headers?
+        error "Cannot write body after trailing headers"
+      elsif writing_headers?
+        write "\r\n"
+        @state = :writing_body
+      end
+      
+      if chunked?
+        write "#{chunk.length.to_s(16)}\r\n#{chunk}\r\n"
+      else
+        write chunk
+      end
+    end
+    
+    def complete
+      return if ready?
+      
+      if writing_body? && chunked?
+        write "0\r\n\r\n"
+      elsif writing_headers? || writing_trailing_headers?
+        write "\r\n"
+      end
+      
+      on_complete.each {|blk| blk.call(bytes_written) }
+      reset
+    end
+    
+    def write(chunk)
+      @bytes_written += chunk.length
+      on_write.each {|blk| blk.call(chunk) }
+    end
+    
+    def error(message)
+      exception = BuilderError.new(message)
+      exception.set_backtrace(caller[1..-1])
+      on_error.each {|blk| blk.call(exception) }
+      raise(exception)
+    end
+  end
+end

data/lib/hatetepe/parser.rb

@@ -0,0 +1,96 @@
+require "http/parser"
+
+module Hatetepe
+  class ParserError < StandardError; end
+  
+  class Parser
+    def self.parse(data = [], &block)
+      message = {}
+      parser = new do |p|
+        p.on_request do |*args|
+          message[:http_method] = args[0]
+          message[:request_url] = args[1]
+          message[:http_version] = args[2]
+        end
+        p.on_response do |*args|
+          message[:status] = args[0]
+          message[:http_version] = args[1]
+        end
+        p.on_header do |name, value|
+          (message[:headers] ||= {})[name] = value
+        end
+        p.on_body_chunk do |chunk|
+          (message[:body] ||= "") << chunk
+        end
+      end
+      
+      if block
+        block.arity == 0 ? parser.instance_eval(&block) : block.call(parser)
+      end
+      
+      Array(data).each {|chunk| parser << chunk }
+      message
+    end
+    
+    attr_reader :bytes_read
+    
+    def initialize(&block)
+      @on_request, @on_response, @on_header = [], [], []
+      @on_body_chunk, @on_complete, @on_error = [], [], []
+      @parser = HTTP::Parser.new
+      
+      @parser.on_headers_complete = proc do
+        if @parser.http_method
+          on_request.each do |r|
+            r.call(@parser.http_method, @parser.request_url, @parser.http_version.join("."))
+          end
+        else
+          on_response.each do |r|
+            r.call(@parser.status_code, @parser.http_version.join("."))
+          end
+        end
+        
+        @parser.headers.each do |header|
+          on_header.each {|h| h.call(*header) }
+        end
+      end
+      
+      @parser.on_body = proc do |chunk|
+        on_body_chunk.each {|b| b.call(chunk) }
+      end
+      
+      @parser.on_message_complete = proc do
+        on_complete.each {|f| f.call(bytes_read) }
+      end
+      
+      reset
+      
+      if block
+        block.arity == 0 ? instance_eval(&block) : block.call(self)
+      end
+    end
+    
+    def reset
+      @parser.reset!
+      @bytes_read = 0
+    end
+    
+    [:request, :response, :header, :body_chunk, :complete, :error].each do |hook|
+      define_method :"on_#{hook}" do |&block|
+        store = instance_variable_get(:"@on_#{hook}")
+        return store unless block
+        store << block
+      end
+    end
+    
+    def <<(data)
+      @bytes_read += data.length
+      @parser << data
+    rescue HTTP::Parser::Error => original_error
+      error = ParserError.new(original_error.message)
+      error.set_backtrace(original_error.backtrace)
+      on_error.each {|e| e.call(error) }
+      raise(error)
+    end
+  end
+end

data/lib/hatetepe/status.rb

@@ -0,0 +1,42 @@
+module Hatetepe
+  # @author Mongrel
+  STATUS_CODES = {
+    100 => "Continue",
+    101 => "Switching Protocols",
+    200 => "OK",
+    201 => "Created",
+    202 => "Accepted",
+    203 => "Non-Authoritative Information",
+    204 => "No Content",
+    205 => "Reset Content",
+    206 => "Partial Content",
+    300 => "Multiple Choices",
+    301 => "Moved Permanently",
+    302 => "Moved Temporarily",
+    303 => "See Other",
+    304 => "Not Modified",
+    305 => "Use Proxy",
+    400 => "Bad Request",
+    401 => "Unauthorized",
+    402 => "Payment Required",
+    403 => "Forbidden",
+    404 => "Not Found",
+    405 => "Method Not Allowed",
+    406 => "Not Acceptable",
+    407 => "Proxy Authentication Required",
+    408 => "Request Time-out",
+    409 => "Conflict",
+    410 => "Gone",
+    411 => "Length Required",
+    412 => "Precondition Failed",
+    413 => "Request Entity Too Large",
+    414 => "Request-URI Too Large",
+    415 => "Unsupported Media Type",
+    500 => "Internal Server Error",
+    501 => "Not Implemented",
+    502 => "Bad Gateway",
+    503 => "Service Unavailable",
+    504 => "Gateway Time-out",
+    505 => "HTTP Version Not Supported"
+  }
+end

data/lib/hatetepe/version.rb

@@ -0,0 +1,3 @@
+module Hatetepe
+  VERSION = "0.0.1"
+end

data/test/builder_test.rb

@@ -0,0 +1,7 @@
+require File.expand_path("..", __FILE__) + "/test_helper"
+
+class BuilderTest < Test::Unit::TestCase
+  def test_bar
+    assert true
+  end
+end

data/test/parser_test.rb

@@ -0,0 +1,7 @@
+require File.expand_path("..", __FILE__) + "/test_helper"
+
+class ParserTest < Test::Unit::TestCase
+  def test_foo
+    assert true
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,7 @@
+require "bundler"
+Bundler.setup :default, :development
+
+require "hatetepe"
+
+require "test/unit"
+require "awesome_print"

metadata

@@ -0,0 +1,94 @@
+--- !ruby/object:Gem::Specification 
+name: hatetepe
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Lars Gierth
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-03 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: http_parser.rb
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  prerelease: false
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: test-unit
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  prerelease: false
+  version_requirements: *id002
+description: Hatetepe combines its own builder with http_parser.rb to make dealing with HTTP messages as comfortable as possible.
+email: 
+- lars.gierth@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Gemfile
+- README.md
+- Rakefile
+- example.rb
+- hatetepe.gemspec
+- lib/hatetepe.rb
+- lib/hatetepe/builder.rb
+- lib/hatetepe/parser.rb
+- lib/hatetepe/status.rb
+- lib/hatetepe/version.rb
+- test/builder_test.rb
+- test/parser_test.rb
+- test/test_helper.rb
+homepage: https://github.com/lgierth/hatetepe
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 678111023
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 678111023
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.2
+signing_key: 
+specification_version: 3
+summary: Builds and parses HTTP messages
+test_files: []
+