Contents

module HttpUtilities
  module Http
    class Response
      include HttpUtilities::Http::Logger

      attr_accessor :body, :parsed_body, :page, :format, :request, :force_encoding

      def initialize(response = nil, request = nil, options = {})
        self.body             =   (response && response.body) ? response.body : nil
        self.request          =   request

        self.parsed_body      =   nil

        self.format           =   options.fetch(:format, nil)
        self.force_encoding   =   options.fetch(:force_encoding, true)

        encode if (self.force_encoding)
        parse_response
      end

      def encode
        if self.body && self.body.is_a?(String)
          begin
            self.body = self.body.force_encoding('UTF-8').encode("UTF-8", :invalid => :replace, :undef => :replace, :replace => "")
          rescue Exception => e
            log(:error, "[HttpUtilities::Http::Format] - Failed to convert response with String#encode. Error: #{e.class.name}. Message: #{e.message}.")
          end
        end
      end

      def parse_response
        self.send("as_#{self.format}".to_sym) if (self.body && self.format)
      end
      
      def as_html
        self.parsed_body = (self.body && self.body != "") ? Nokogiri::HTML(self.body.to_s.force_encoding("utf-8"), nil, "utf-8") : nil
      end

      def as_xml
        self.parsed_body = (self.body && self.body != "") ? Nokogiri::XML(self.body.to_s.force_encoding("utf-8"), nil, "utf-8") : nil
      end

      def as_json
        self.parsed_body = (self.body && self.body != "") ? self.body.to_s.force_encoding("utf-8").to_json : nil
      end

      def set_page(page)
        self.page = page

        if (self.page && self.page.parser)
          self.body         =   self.page.parser.content
          self.parsed_body  =   self.page.parser
        end
      end

    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
http_utilities-1.2.5.3	lib/http_utilities/http/response.rb
http_utilities-1.2.5.2	lib/http_utilities/http/response.rb
http_utilities-1.2.5.1	lib/http_utilities/http/response.rb
http_utilities-1.2.5	lib/http_utilities/http/response.rb
http_utilities-1.2.4.1	lib/http_utilities/http/response.rb
http_utilities-1.2.4	lib/http_utilities/http/response.rb