Module AdWords::Extensions

# File lib/adwords4r/apiextensions.rb, line 124
    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 200
    def self.downloadReport(wrapper, report_definition_id)
      # Set URL parameters.
      parameters = {'__rd' => report_definition_id.to_s}

      # Set HTTP headers.
      headers = {}
      headers['Authorization'] = 'GoogleLogin auth=%s' %
          wrapper.api.credentials.auth_token
      creds = wrapper.api.credentials.credentials
      if creds['clientEmail']
        headers['clientEmail'] = creds['clientEmail']
      elsif creds['clientCustomerId']
        headers['clientCustomerId'] = creds['clientCustomerId']
      end

      # Get download URL.
      url = AdWords::Service.report_download_url(
          wrapper.api.credentials.environment, 201003)

      # Download report data.
      client = HTTPClient.new
      report_data = client.get_content(url, parameters, headers)

      return report_data
    end

# File lib/adwords4r/apiextensions.rb, line 177
    def self.downloadReportAsFile(wrapper, report_definition_id, path)
      report_data = downloadReport(wrapper, report_definition_id)

      # Write to file (if provided)
      if path
        open(path, 'w') { |file| file.puts(report_data) }
      end

      return nil
    end

# File lib/adwords4r/apiextensions.rb, line 77
    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 49
    def self.extensions
      return @@extensions
    end

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