Contents

require 'httparty'
require 'addressable/uri'
require 'wp/api/endpoints'
require 'active_support/hash_with_indifferent_access'

module WP::API
  class Client
    include HTTParty
    include Endpoints

    attr_accessor :host

    DIRECT_PARAMS = %w(type context filter)

    def initialize(host:, scheme: 'http', user: nil, password: nil)
      @scheme = scheme
      @host = host
      @user = user
      @password = password

      fail ':host is required' unless host.is_a?(String) && host.length > 0
    end

    def inspect
      to_s.sub(/>$/, '') + " @scheme=\"#{@scheme}\" @host=\"#{@host}\" @user=\"#{@user}\" @password=#{@password.present?}>"
    end

    protected

    def get(resource, query = {})
      should_raise_on_empty = query.delete(:should_raise_on_empty) || true
      query = ActiveSupport::HashWithIndifferentAccess.new(query)
      path = url_for(resource, query)
      
      response = if authenticate?
        Client.get(path, basic_auth: { username: @user, password: @password })
      else
        Client.get(path)
      end

      if response.code != 200
        raise WP::API::ResourceNotFoundError
      elsif response.parsed_response.empty? && should_raise_on_empty
        raise WP::API::ResourceNotFoundError
      else
        [ response.parsed_response, response.headers ] # Already parsed.
      end
    end

    private

    def authenticate?
      @user && @password
    end

    def url_for(fragment, query)
      url = "#{@scheme}://#{@host}/wp-json/#{fragment}"
      url << ("?" + params(query)) unless query.empty?

      url
    end

    def params(query)
      uri = Addressable::URI.new
      filter_hash = { page: query.delete('page') || 1 }
      query.each do |key, value|
        filter_hash[key] = value if DIRECT_PARAMS.include?(key) || key.include?('[')
        filter_hash["filter[#{key}]"] = value
      end
      uri.query_values = filter_hash

      uri.query
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
wp-api-0.1.6	lib/wp/api/client.rb
wp-api-0.1.5	lib/wp/api/client.rb
wp-api-0.1.4	lib/wp/api/client.rb