52 lines
1.7 KiB
TypeScript
52 lines
1.7 KiB
TypeScript
import { WsProxyManager } from './manager.ts';
|
|
import { getLoginUserByToken } from '@/modules/auth.ts';
|
|
import { logger } from '../logger.ts';
|
|
export const wsProxyManager = new WsProxyManager();
|
|
import { WebScoketListenerFun } from '@kevisual/router/src/server/server-type.ts'
|
|
export const wssFun: WebScoketListenerFun = async (req, res) => {
|
|
// do nothing, just to enable ws upgrade event
|
|
const { id, ws, token, data, emitter } = req;
|
|
logger.debug('ws proxy connected, id=', id, ' token=', token, ' data=', data);
|
|
// console.log('req', req)
|
|
const { type } = data || {};
|
|
if (type === 'registryClient') {
|
|
const loginUser = await getLoginUserByToken(token);
|
|
if (!loginUser?.tokenUser) {
|
|
logger.debug('未登录，断开连接');
|
|
ws.send(JSON.stringify({ code: 401, message: '未登录' }));
|
|
setTimeout(() => {
|
|
ws.close(401, 'Unauthorized');
|
|
}, 1000);
|
|
return;
|
|
}
|
|
const user = loginUser?.tokenUser?.username;
|
|
const userApp = user + '-' + id;
|
|
logger.debug('注册 ws 连接', userApp);
|
|
// @ts-ignore
|
|
wsProxyManager.register(userApp, { user, ws });
|
|
ws.send(
|
|
JSON.stringify({
|
|
type: 'connected',
|
|
user: user,
|
|
id,
|
|
}),
|
|
);
|
|
emitter.once('close--' + id, () => {
|
|
logger.debug('ws emitter closed');
|
|
wsProxyManager.unregister(userApp);
|
|
});
|
|
// @ts-ignore
|
|
ws.data.userApp = userApp;
|
|
return;
|
|
}
|
|
// @ts-ignore
|
|
const userApp = ws.data.userApp;
|
|
logger.debug('message', data, ' userApp=', userApp);
|
|
const wsMessage = wsProxyManager.get(userApp);
|
|
if (wsMessage) {
|
|
wsMessage.sendResponse(data);
|
|
} else {
|
|
// @ts-ignore
|
|
logger.debug('账号应用未注册，无法处理消息。未授权？', ws.data);
|
|
}
|
|
} |