class Travis::Client::Artifact

def body(stream = block_given?)
  return current_body unless block_given? || stream
  return yield(current_body) unless stream && job.pending?
  number = 0
  session.listen(self) do |listener|
    listener.on 'job:log' do |event|
      next unless event.payload['number'] > number
      number = event.payload['number']
      yield event.payload['_log']
      listener.disconnect if event.payload['final']
    end
    listener.on 'job:finished' do |_event|
      listener.disconnect
    end
    listener.on_connect do
      data = session.get_raw("/logs/#{id}", nil, 'Accept' => CHUNKED)['log']
      if data['parts']
        data['parts'].each { |p| yield p['content'] }
        number = data['parts'].last['number'] if data['parts'].any?
      else
        yield data['body']
        listener.disconnect
      end
    end
  end
end

def clean_body
  attributes['clean_body'] ||= Tools::SafeString.clean(body)
end

def colorized_body
  attributes['colorized_body'] ||= Tools::SafeString.colorized(body)
end

def current_body
  attributes['current_body'] ||= begin
    body = load_attribute('body')
    if body.to_s.empty?
      log  = session.get_raw("jobs/#{job_id}/log", nil, 'Accept' => TEXT)
      body = log.is_a?(String) ? log : log['log']['body']
    end
    body
  end
end

def delete_body(reason = {})
  reason = { reason: } unless reason.is_a? Hash
  session.patch_raw("jobs/#{job_id}/log", reason)
  reload
rescue Travis::Client::Error => e
  raise unless e.message == '409'
  self
end

def encoded_body
  Tools::SafeString.encoded(body)
end

def pusher_entity
  job
end