-
Notifications
You must be signed in to change notification settings - Fork 47
/
server.js
69 lines (60 loc) · 1.45 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import express from 'express'
import path from 'path'
import {
createServer
} from 'http'
import {
Server
} from 'socket.io'
import {
toBuffer
} from 'qrcode'
import fetch from 'node-fetch'
import Helper from './lib/helper.js'
function connect(conn, PORT) {
var app = global.app = express()
var server = global.server = createServer(app)
var _qr = 'invalid'
conn.ev.on('connection.update', function appQR({
qr
}) {
if (qr) _qr = qr
})
app.use(async (req, res) => {
res.setHeader('content-type', 'image/png')
res.end(await toBuffer(_qr))
})
app.use(express.static(path.join(Helper.__dirname(
import.meta.url), 'views')))
var io = new Server(server)
io.on('connection', socket => {
var {
unpipeEmit
} = pipeEmit(conn, socket, 'conn-')
socket.once('disconnect', unpipeEmit)
})
server.listen(PORT, () => {
console.log('App listened on port', PORT)
if (opts['keepalive']) keepAlive()
})
}
function pipeEmit(event, event2, prefix = '') {
var old = event.emit
event.emit = function(event, ...args) {
old.emit(event, ...args)
event2.emit(prefix + event, ...args)
}
return {
unpipeEmit() {
event.emit = old
}
}
}
function keepAlive() {
var url = `https://${process.env.SPACE_HOST}`
if (/(\/\/|\.)undefined\./.test(url)) return
setInterval(() => {
fetch(url, {headers: {'Authorization': 'Bearer ' + process.env.HF_TOKEN}}).then(v => console.log(v.status)).catch(console.error)
}, 5 * 1000 * 60)
}
export default connect