Module AdWords::Extensions

# File lib/adwords4r/apiextensions.rb, line 118
    def self.downloadCsvReport(wrapper, job_id, report_xml=nil)
      # Get XML report data.
      report_xml = downloadXmlReport(wrapper, job_id) if report_xml.nil?

      begin
        # Construct DOM object.
        doc = REXML::Document.new(report_xml)

        # Get data columns.
        columns = []
        doc.elements.each('report/table/columns/column') do |column_elem|
          name = column_elem.attributes['name']
          columns << name unless name.nil?
        end

        # Get data rows.
        rows = []
        doc.elements.each('report/table/rows/row') do |row_elem|
          rows << row_elem.attributes unless row_elem.attributes.nil?
        end

        # Build CSV
        csv = ''
        CSV::Writer.generate(csv) do |writer|
          writer << columns
          rows.each do |row|
            row_values = []
            columns.each { |column| row_values << row[column] }
            writer << row_values
          end
        end

        return csv
      rescue REXML::ParseException => e
        # Error parsing XML
        raise AdWords::Error::Error,
            "Error parsing report XML: %s\nSource: %s" % [e, e.backtrace.first]
      end
    end

# File lib/adwords4r/apiextensions.rb, line 71
    def self.downloadXmlReport(wrapper, job_id)
      sleep_interval = 30

      # Repeatedly check the report status until it is finished.
      # 'Pending' and 'InProgress' statuses indicate the job is still being run.
      status = wrapper.getReportJobStatus(job_id).getReportJobStatusReturn
      while status != 'Completed' && status != 'Failed'
        sleep(sleep_interval)
        status = wrapper.getReportJobStatus(job_id).getReportJobStatusReturn
      end

      if status == 'Completed'
        report_url =
            wrapper.getReportDownloadUrl(job_id).getReportDownloadUrlReturn

        # Download the report via the HTTPClient library and return its
        # contents. The report is an XML document; the actual element names vary
        # depending on the type of report run and columns requested.
        begin
          client = HTTPClient.new
          return client.get_content(report_url)
        rescue Errno::ECONNRESET, SOAP::HTTPStreamError, SocketError => e
          # This exception indicates a connection-level error.
          # In general, it is likely to be transitory.
          raise AdWords::Error::Error, "Connection Error: %s\nSource: %s" %
              [e, e.backtrace.first]
        end
      else
        # Reports that pass validation will normally not fail, but if there is
        # an error in the report generation service it can sometimes happen.
        raise AdWords::Error::Error, 'Report generation failed.'
      end
    end

# File lib/adwords4r/apiextensions.rb, line 43
    def self.extensions
      return @@extensions
    end

# File lib/adwords4r/apiextensions.rb, line 209
    def self.getClientUnitsUsage(wrapper, start_date, end_date)
      # Create a new AdWords::API object to ensure thread-safety (we'll need to
      # change the clientEmail)
      adwords = AdWords::API.new(wrapper.api.credentials.dup)
      adwords.credentials.set_header('clientEmail', '')
      # Call unit_adder on the main user
      unit_map = client_unit_adder(adwords, start_date, end_date)
      # Pass back the spent unit information to the main AdWords::API object
      wrapper.api.mutex.synchronize do
        wrapper.api.last_units = adwords.total_units
        wrapper.api.total_units += adwords.total_units
      end
      return unit_map
    end

# File lib/adwords4r/apiextensions.rb, line 176
    def self.getMethodUsage(wrapper, start_date, end_date)
      op_rates = AdWords::Utils.get_operation_rates
      usage = {}

      op_rates.each do |op|
        service, method = op
        usage[service + '.' + method] = wrapper.getUnitCountForMethod(service,
            method, start_date, end_date).getUnitCountForMethodReturn
      end

      return usage
    end

# File lib/adwords4r/apiextensions.rb, line 48
    def self.methods
      return @@methods
    end