module Yelp4r
  class Response
    
    attr_accessor :body
    
    def initialize(response)
      @body = rash_response(response)
    end
    
    def response_code
      @body.message.code
    end
    
    def success?
      response_code == 0
    end
    
    def error_message
      @body.message.text unless success?
    end
    
    def data
      if !@body.businesses.blank?
        @body.businesses
      elsif !@body.neighborhoods.blank?
        @body.neighborhoods
      end
    end
    
    private
    
    def rash_response(response)
      if response.is_a?(Array)
        @body = []
        response.each do |b|
          if b.is_a?(Hash)
            @body << Hashie::Rash.new(b)
          else
            @body << b
          end
        end
      elsif response.is_a?(Hash)
        @body = Hashie::Rash.new(response)
      else
        @body = response
      end
    end
    
  end
end