module CurbFu
  module Response
    class Base
      attr_accessor :status, :body, :headers
      
      def initialize(status, headers, body)
        @status = status
        set_response_type(status)
        @body = body
        @headers = headers.is_a?(String) ? parse_headers(headers) : headers
      end
      
      def success?
        self.is_a?(CurbFu::Response::Success)
      end
      
      def redirect?
        self.is_a?(CurbFu::Response::Redirection)
      end
      
      def failure?
        !(success? || redirect?)
      end
      
      def server_fail?
        self.is_a?(CurbFu::Response::ServerError)
      end
      
      def client_fail?
        self.is_a?(CurbFu::Response::ClientError)
      end
      
      def parse_headers(header_string)
        header_lines = header_string.split($/)
        header_lines.shift
        header_lines.inject({}) do |hsh, line|
          whole_enchillada, key, value = /^(.*?):\s*(.*)$/.match(line.chomp).to_a
          hsh[key] = value unless whole_enchillada.nil?
          hsh
        end
      end
      
      def to_hash
        { :status => status, :body => body, :headers => headers }
      end
      
      def set_response_type(status)
        case status
        when 100..199 then
          self.extend CurbFu::Response::Information
          case self.status
          when 101 then self.extend CurbFu::Response::Continue
          when 102 then self.extend CurbFu::Response::SwitchProtocl
          end
        when 200..299 then
          self.extend CurbFu::Response::Success
          case self.status
          when 200 then self.extend CurbFu::Response::OK
          when 201 then self.extend CurbFu::Response::Created
          when 202 then self.extend CurbFu::Response::Accepted
          when 203 then self.extend CurbFu::Response::NonAuthoritativeInformation
          when 204 then self.extend CurbFu::Response::NoContent
          when 205 then self.extend CurbFu::Response::ResetContent
          when 206 then self.extend CurbFu::Response::PartialContent
          end
        when 300..399 then
          self.extend CurbFu::Response::Redirection
          case self.status
          when 300 then self.extend CurbFu::Response::MultipleChoice
          when 301 then self.extend CurbFu::Response::MovedPermanently
          when 302 then self.extend CurbFu::Response::Found
          when 303 then self.extend CurbFu::Response::SeeOther
          when 304 then self.extend CurbFu::Response::NotModified
          when 305 then self.extend CurbFu::Response::UseProxy
          when 307 then self.extend CurbFu::Response::TemporaryRedirect
          end
        when 400..499 then
          self.extend CurbFu::Response::ClientError
          case self.status
          when 400 then self.extend CurbFu::Response::BadRequest
          when 401 then self.extend CurbFu::Response::Unauthorized
          when 402 then self.extend CurbFu::Response::PaymentRequired
          when 403 then self.extend CurbFu::Response::Forbidden
          when 404 then self.extend CurbFu::Response::NotFound
          when 405 then self.extend CurbFu::Response::MethodNotAllowed
          when 406 then self.extend CurbFu::Response::NotAcceptable
          when 407 then self.extend CurbFu::Response::ProxyAuthenticationRequired
          when 408 then self.extend CurbFu::Response::RequestTimeOut
          when 409 then self.extend CurbFu::Response::Conflict
          when 410 then self.extend CurbFu::Response::Gone
          when 411 then self.extend CurbFu::Response::LengthRequired
          when 412 then self.extend CurbFu::Response::PreconditionFailed
          when 413 then self.extend CurbFu::Response::RequestEntityTooLarge
          when 414 then self.extend CurbFu::Response::RequestURITooLong
          when 415 then self.extend CurbFu::Response::UnsupportedMediaType
          when 416 then self.extend CurbFu::Response::UnsupportedMediaType
          when 417 then self.extend CurbFu::Response::ExpectationFailed
          end
        when 500..599 then
          self.extend CurbFu::Response::ServerError
          case self.status
          when 500 then self.extend CurbFu::Response::InternalServerError
          when 501 then self.extend CurbFu::Response::NotImplemented
          when 502 then self.extend CurbFu::Response::BadGateway
          when 503 then self.extend CurbFu::Response::ServiceUnavailable
          when 504 then self.extend CurbFu::Response::GatewayTimeOut
          when 505 then self.extend CurbFu::Response::VersionNotSupported
          end
        else
          self.extend CurbFu::Response::UnknownResponse
        end
      end
      
      class << self
        def from_rack_response(rack)
          raise ArgumentError.new("Rack response may not be nil") if rack.nil?
          response = self.new(rack.status, rack.headers, rack.body)
        end
      
        def from_curb_response(curb)
          response = self.new(curb.response_code, curb.header_str, curb.body_str)
          response
        end
        
        def from_hash(hash)
          return nil if hash.nil?
          self.new(hash[:status], hash[:headers], hash[:body])
        end
      end
    end
          
    module Information; end
      module Continue;                           def self.to_i; 100; end; end
      module SwitchProtocl; end
    module Success;                              def self.to_i; 200; end; end
      module OK;                                 def self.to_i; 200; end; end
      module Created;                            def self.to_i; 201; end; end
      module Accepted;                           def self.to_i; 202; end; end
      module NonAuthoritativeInformation; end
      module NoContent; end
      module ResetContent; end
      module PartialContent; end
    module Redirection; end                    # 3xx
      module MultipleChoice; end                 # 300
      module MovedPermanently; end               # 301
      module Found;                              def self.to_i; 302; end; end
      module SeeOther; end                       # 303
      module NotModified; end                    # 304
      module UseProxy; end                       # 305
      module TemporaryRedirect; end              # 307
    module ClientError; end                    # 4xx
      module BadRequest; end                     # 400
      module Unauthorized; end                   # 401
      module PaymentRequired; end                # 402
      module Forbidden; end                      # 403
      module NotFound;                           def self.to_i; 404; end; end
      module MethodNotAllowed; end               # 405
      module NotAcceptable; end                  # 406
      module ProxyAuthenticationRequired; end    # 407
      module RequestTimeOut; end                 # 408
      module Conflict; end                       # 409
      module Gone;                               def self.to_i; 410; end; end
      module LengthRequired; end                 # 411
      module PreconditionFailed; end             # 412
      module RequestEntityTooLarge; end          # 413
      module RequestURITooLong; end              # 414
      module UnsupportedMediaType; end           # 415
      module RequestedRangeNotSatisfiable; end   # 416
      module ExpectationFailed; end              # 417
    module ServerError; end                    # 5xx
      module InternalServerError;                def self.to_i; 500; end; end
      module NotImplemented; end                 # 501
      module BadGateway; end                     # 502
      module ServiceUnavailable;                 def self.to_i; 503; end; end
      module GatewayTimeOut; end                 # 504
      module VersionNotSupported; end            # 505
    module UnknownResponse; end
  end
end