d8d-admin-mobile-starter-public/server/router_io.ts

import { Socket, Server } from "socket.io";
import { Auth } from '@d8d-appcontainer/auth';
import type { User as AuthUser } from '@d8d-appcontainer/auth';
import { APIClient } from '@d8d-appcontainer/api';
import { setupMessageEvents } from './routes_io_messages.ts';
import { setupExamEvents } from './routes_io_exam.ts';
import debug from "debug";

const log = debug('socketio:auth');

interface SetupSocketIOProps {
  io: Server, auth: Auth, apiClient: APIClient
}

export interface SocketWithUser extends Socket {
  user?: AuthUser;
}

// 定义自定义上下文类型
export interface Variables {
  socket: SocketWithUser
  auth: Auth
  user: AuthUser
  apiClient: APIClient
  // moduleDir: string
  // systemSettings?: SystemSettingRecord
}

export function setupSocketIO({ io, auth, apiClient }:SetupSocketIOProps) {
  // Socket.IO认证中间件
  io.use(async (socket: SocketWithUser) => {
    try {
      const token = socket.handshake.query.get('socket_token');
      if (!token) {
        log(`未提供token，拒绝连接: ${socket.id}`);
        throw new Error('未授权')
      }

      const userData = await auth.verifyToken(token);
      if (!userData) {
        log(`无效token，拒绝连接: ${socket.id}`);
        throw new Error('无效凭证')
      }

      socket.user = userData;
      log(`认证成功: ${socket.id} 用户: ${userData.username}`);
    } catch (error) {
      log(`认证错误: ${socket.id}`, error);
    }
  });

  io.on("connection", (socket: SocketWithUser) => {
    if (!socket.user) {
      socket.disconnect(true);
      return;
    }

    console.log(`socket ${socket.id} 已连接，用户: ${socket.user.username}`);

    socket.on("disconnect", (reason) => {
      console.log(`socket ${socket.id} 断开连接，原因: ${reason}`);
    });

    const context: Variables = {
      socket,
      auth,
      apiClient,
      user: socket.user,
    }

    // 初始化消息路由
    setupMessageEvents(context);
    setupExamEvents(context);
  });
}