You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
138 lines
2.9 KiB
138 lines
2.9 KiB
// Native
|
|
const EventEmitter = require('events')
|
|
|
|
// Packages
|
|
const io = require('socket.io-client')
|
|
const chalk = require('chalk')
|
|
|
|
const { compare, deserialize } = require('./logs')
|
|
|
|
module.exports = class Logger extends EventEmitter {
|
|
constructor(host, token, { debug = false, quiet = false } = {}) {
|
|
super()
|
|
this.host = host
|
|
this.token = token
|
|
this.debug = debug
|
|
this.quiet = quiet
|
|
|
|
// ReadyState
|
|
this.building = false
|
|
|
|
this.socket = io(`https://io.now.sh/states?host=${host}&v=2`)
|
|
this.socket.once('error', this.onSocketError.bind(this))
|
|
this.socket.on('auth', this.onAuth.bind(this))
|
|
this.socket.on('state', this.onState.bind(this))
|
|
this.socket.on('logs', this.onLog.bind(this))
|
|
this.socket.on('backend', this.onComplete.bind(this))
|
|
|
|
// Log buffer
|
|
this.buf = []
|
|
this.printed = new Set()
|
|
}
|
|
|
|
onAuth(callback) {
|
|
if (this.debug) {
|
|
console.log('> [debug] authenticate')
|
|
}
|
|
callback(this.token)
|
|
}
|
|
|
|
onState(state) {
|
|
// Console.log(state)
|
|
if (!state.id) {
|
|
console.error('> Deployment not found')
|
|
this.emit('error')
|
|
return
|
|
}
|
|
|
|
if (state.error) {
|
|
this.emit('error', state)
|
|
return
|
|
}
|
|
|
|
if (state.backend) {
|
|
this.onComplete()
|
|
return
|
|
}
|
|
|
|
if (state.logs) {
|
|
state.logs.forEach(this.onLog, this)
|
|
}
|
|
}
|
|
|
|
onLog(log) {
|
|
if (!this.building) {
|
|
if (!this.quiet) {
|
|
console.log('> Building')
|
|
}
|
|
this.building = true
|
|
}
|
|
|
|
if (this.quiet) {
|
|
return
|
|
}
|
|
|
|
log = deserialize(log)
|
|
|
|
const timer = setTimeout(() => {
|
|
this.buf.sort((a, b) => compare(a.log, b.log))
|
|
const idx = this.buf.findIndex(b => b.log.id === log.id) + 1
|
|
for (const b of this.buf.slice(0, idx)) {
|
|
clearTimeout(b.timer)
|
|
this.printLog(b.log)
|
|
}
|
|
this.buf = this.buf.slice(idx)
|
|
}, 500)
|
|
|
|
this.buf.push({ log, timer })
|
|
}
|
|
|
|
onComplete() {
|
|
this.socket.disconnect()
|
|
|
|
if (this.building) {
|
|
this.building = false
|
|
}
|
|
|
|
this.buf.sort((a, b) => compare(a.log, b.log))
|
|
|
|
// Flush all buffer
|
|
for (const b of this.buf) {
|
|
clearTimeout(b.timer)
|
|
this.printLog(b.log)
|
|
}
|
|
this.buf = []
|
|
|
|
this.emit('close')
|
|
}
|
|
|
|
onSocketError(err) {
|
|
if (this.debug) {
|
|
console.log(`> [debug] Socket error ${err}\n${err.stack}`)
|
|
}
|
|
}
|
|
|
|
printLog(log) {
|
|
if (this.printed.has(log.id)) return
|
|
|
|
this.printed.add(log.id)
|
|
|
|
const data = log.object ? JSON.stringify(log.object) : log.text
|
|
|
|
if (log.type === 'command') {
|
|
console.log(`${chalk.gray('>')} ▲ ${data}`)
|
|
} else if (log.type === 'stderr') {
|
|
data.split('\n').forEach(v => {
|
|
if (v.length > 0) {
|
|
console.error(chalk.gray(`> ${v}`))
|
|
}
|
|
})
|
|
} else if (log.type === 'stdout') {
|
|
data.split('\n').forEach(v => {
|
|
if (v.length > 0) {
|
|
console.log(`${chalk.gray('>')} ${v}`)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|