import { WsProxyManager } from './manager.ts';
import { getLoginUserByToken } from '@/modules/auth.ts';
import { logger } from '../logger.ts';
export const wsProxyManager = new WsProxyManager();
import { WebSocketListenerFun } from '@kevisual/router/src/server/server-type.ts'
// 生成一个随机六位字符串作为注册 ID
const generateRegistryId = () => {
  return Math.random().toString(36).substring(2, 8);
}
export const wssFun: WebSocketListenerFun = async (req, res) => {
  // do nothing, just to enable ws upgrade event
  const { id, ws, token, data, emitter } = req;
  // console.log('req', req)
  const { type } = data || {};

  if (type === 'registryClient') {
    const loginUser = await getLoginUserByToken(token);
    let isLogin = false;
    let user = '';
    if (loginUser?.tokenUser) {
      isLogin = true;
      user = loginUser?.tokenUser?.username;
    } else {
      logger.debug('未登录，请求等待用户验证', data);
      user = data?.username || '';
    }
    if (!user) {
      logger.debug('未提供用户名，无法注册 ws 连接');
      ws.close();
      return;
    }
    let userApp = user + '--' + id;
    // TODO: 如果存在, 而且之前的那个关闭了,不需要验证,直接覆盖和复用.
    let wsConnect = await wsProxyManager.createNewConnection({ ws, user, userApp, isLogin });
    if (wsConnect.isNew) {
      logger.debug('新连接注册成功', userApp);
    }
    // @ts-ignore
    ws.data.userApp = wsConnect.id;
    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);
  }
}