Box View API Client

box_view_client.rb

module BoxView
  
  class SessionError < StandardError; end
  class GetError < StandardError; end
  class RetryError < StandardError
    attr_reader :retry_after

    def initialize(retry_after)
      @retry_after = retry_after
    end
  end

  class Client
    include HTTParty
    base_uri  'https://view-api.box.com/1'
    headers   'Authorization' => "Token #{SETTINGS['box']['api_key']}",
              'Content-Type' => 'application/json'

    class << self
      def create_session(url)
        response = post('/documents', body: { url: url }.to_json)
        if response['type'] == 'document' && response['id'].present?
          document_id = response['id']
        else
          raise SessionError, "No document_id in response: #{response.inspect}"
        end
        document_id
      end

      def get_pdf_file(document_id)
        response = get("/documents/#{document_id}/content.pdf")
        retry_after = response.headers['retry-after']
        if response.code == 200
          file = Tempfile.new([document_id, '.pdf'])
          file.binmode
          file.write(response.parsed_response)
          file
        elsif retry_after
          raise RetryError.new(retry_after.to_i)
        else
          raise GetError, "Invalid response: #{response.inspect}"
        end
      end
    end
  end
end