arsduo/koala

require 'koala/api'
require 'koala/api/graph_batch_api'

module Koala
  module Facebook
    class GraphBatchAPI < API
      # @private
      class BatchOperation
        attr_reader :access_token, :http_options, :post_processing, :files, :batch_api, :identifier

        @identifier = 0

        def self.next_identifier
          @identifier += 1
        end

        def initialize(options = {})
          @identifier = self.class.next_identifier
          @args = (options[:args] || {}).dup # because we modify it below
          @access_token = options[:access_token]
          @http_options = (options[:http_options] || {}).dup # dup because we modify it below
          @batch_args = @http_options.delete(:batch_args) || {}
          @url = options[:url]
          @method = options[:method].to_sym
          @post_processing = options[:post_processing]

          process_binary_args

          raise AuthenticationError.new(nil, nil, "Batch operations require an access token, none provided.") unless @access_token
        end

        def to_batch_params(main_access_token, app_secret)
          # set up the arguments
          if @access_token != main_access_token
            @args[:access_token] = @access_token
            if app_secret
              @args[:appsecret_proof] = OpenSSL::HMAC.hexdigest(OpenSSL::Digest.new("sha256"), app_secret, @access_token)
            end
          end
          args_string = Koala.http_service.encode_params(@args)

          response = {
            :method => @method.to_s,
            :relative_url => @url,
          }

          # handle batch-level arguments, such as name, depends_on, and attached_files
          @batch_args[:attached_files] = @files.keys.join(",") if @files
          response.merge!(@batch_args) if @batch_args

          # for get and delete, we append args to the URL string
          # otherwise, they go in the body
          if args_string.length > 0
            if args_in_url?
              response[:relative_url] += (@url =~ /\?/ ? "&" : "?") + args_string if args_string.length > 0
            else
              response[:body] = args_string if args_string.length > 0
            end
          end

          response
        end

        protected

        def process_binary_args
          # collect binary files
          @args.each_pair do |key, value|
            if HTTPService::UploadableIO.binary_content?(value)
              @files ||= {}
              # we use a class-level counter to ensure unique file identifiers across multiple batch operations
              # (this is thread safe, since we just care about uniqueness)
              # so remove the file from the original hash and add it to the file store
              id = "op#{identifier}_file#{@files.keys.length}"
              @files[id] = @args.delete(key).is_a?(HTTPService::UploadableIO) ? value : HTTPService::UploadableIO.new(value)
            end
          end
        end

        def args_in_url?
          @method == :get || @method == :delete
        end
      end
    end
  end
end