class Travis::CLI::Logs

def check_websocket
  require 'websocket-native'
rescue LoadError => e
  raise e if e.respond_to?(:path) and e.path != 'websocket-native'
  info "speed up log streaming by installing the websocket-native gem"
end

def run(number = last_build.number)
  error "##{number} is not a job, try #{number}.1" unless job = job(number)
  job.log.body { |part| print interactive? ? encoded(part) : clean(part) }
end

def setup
  super
  check_websocket
end