82 lines
2.6 KiB
TypeScript
82 lines
2.6 KiB
TypeScript
import type { WebSocketEvent } from '@proj-airi/server-shared/types'
|
|
import type { Peer } from 'crossws'
|
|
import type { AuthenticatedPeer } from './types'
|
|
import { env } from 'node:process'
|
|
import { Format, LogLevel, setGlobalFormat, setGlobalLogLevel, useLogg } from '@guiiai/logg'
|
|
import { createApp, createRouter, defineWebSocketHandler } from 'h3'
|
|
|
|
setGlobalFormat(Format.Pretty)
|
|
setGlobalLogLevel(LogLevel.Log)
|
|
|
|
function send(peer: Peer, event: WebSocketEvent) {
|
|
peer.send(JSON.stringify(event))
|
|
}
|
|
|
|
function main() {
|
|
const appLogger = useLogg('App').useGlobalConfig()
|
|
const websocketLogger = useLogg('WebSocket').useGlobalConfig()
|
|
|
|
const app = createApp({
|
|
onError: error => appLogger.withError(error).error('an error occurred'),
|
|
})
|
|
|
|
const router = createRouter()
|
|
app.use(router)
|
|
|
|
const peers = new Map<string, AuthenticatedPeer>()
|
|
|
|
router.get('/ws', defineWebSocketHandler({
|
|
open: (peer) => {
|
|
const token = env.AUTHENTICATION_TOKEN || ''
|
|
if (token) {
|
|
peers.set(peer.id, { peer, authenticated: false })
|
|
}
|
|
else {
|
|
send(peer, { type: 'module:authenticated', data: { authenticated: true } })
|
|
peers.set(peer.id, { peer, authenticated: true })
|
|
}
|
|
|
|
websocketLogger.withFields({ peer: peer.id, activePeers: peers.size }).log('connected')
|
|
},
|
|
message: (peer, message) => {
|
|
const token = env.AUTHENTICATION_TOKEN || ''
|
|
const event = message.json() as WebSocketEvent
|
|
|
|
switch (event.type) {
|
|
case 'module:authenticate':
|
|
if (!!token && event.data.token !== token) {
|
|
websocketLogger.withFields({ peer: peer.id }).debug('authentication failed')
|
|
send(peer, { type: 'error', data: { message: 'invalid token' } })
|
|
return
|
|
}
|
|
|
|
send(peer, { type: 'module:authenticated', data: { authenticated: true } })
|
|
peers.set(peer.id, { peer, authenticated: true })
|
|
return
|
|
}
|
|
if (!peers.get(peer.id)?.authenticated) {
|
|
websocketLogger.withFields({ peer: peer.id }).debug('not authenticated')
|
|
send(peer, { type: 'error', data: { message: 'not authenticated' } })
|
|
return
|
|
}
|
|
|
|
for (const [id, p] of peers.entries()) {
|
|
if (id !== peer.id) {
|
|
p.peer.send(JSON.stringify(event))
|
|
}
|
|
}
|
|
},
|
|
error: (peer, error) => {
|
|
websocketLogger.withFields({ peer: peer.id }).withError(error).error('an error occurred')
|
|
},
|
|
close: (peer, details) => {
|
|
websocketLogger.withFields({ peer: peer.id, details, activePeers: peers.size }).log('closed')
|
|
peers.delete(peer.id)
|
|
},
|
|
}))
|
|
|
|
return app
|
|
}
|
|
|
|
export const app = main()
|