module Pardot
  module Objects
    module Visitors
      
      def visitors
        @visitors ||= Visitors.new self
      end
      
      class Visitors
        
        def initialize client
          @client = client
        end
        
        def query params
          result = get "/do/query", params, "result"
          result["total_results"] = result["total_results"].to_i if result["total_results"]
          result
        end
        
        def assign id, params = {}
          post "/do/assign/id/#{id}", params
        end
        
        def read id, params = {}
          post "/do/read/id/#{id}", params
        end
        
        protected
        
        def get path, params = {}, result = "visitor"
          response = @client.get "visitor", path, params
          result ? response[result] : response
        end
        
        def post path, params = {}, result = "visitor"
          response = @client.post "visitor", path, params
          result ? response[result] : response
        end
        
      end
      
    end
  end
end