OnlineTraining0622/components/trtc-room/controller/user-controller.js

import Event from '../utils/event.js'
import User from '../model/user.js'
import Stream from '../model/stream.js'
import {
  EVENT
} from '../common/constants.js'

const TAG_NAME = 'UserController'
/**
 * 通讯成员管理
 */
class UserController {
  constructor(componentContext) {
    // userMap 用于存储完整的数据结构
    this.userMap = new Map()
    // userList 用于存储简化的用户数据 Object，包括 {userID hasMainAudio hasMainVideo hasAuxAudio hasAuxVideo}
    this.userList = []
    // streamList 存储steam 对象列表，用于 trtc-room 渲染 player
    this.streamList = []
    this._emitter = new Event()
    this.componentContext = componentContext
    this.isNewVersion = componentContext.isNewVersion
  }
  userEventHandler(event) {
    const code = event.detail.code
    let data
    if (event.detail.message && typeof event.detail.message === 'string') {
      try {
        data = JSON.parse(event.detail.message)
      } catch (exception) {
        console.warn(TAG_NAME, 'userEventHandler 数据格式错误', exception)
        return false
      }
    } else {
      console.warn(TAG_NAME, 'userEventHandler 数据格式错误')
      return false
    }
    switch (code) {
      case 1020:
        // console.log(TAG_NAME, '远端用户全量列表更新：', code)
        if (!this.isNewVersion) {
          // TODO 旧版SDK处理逻辑，返回全量的用户列表,需要对userList 进行前后对比，筛选出新增用户，暂不实现
        }
        break
      case 1031:
        // console.log(TAG_NAME, '远端用户进房通知：', code)
        // 1031 有新用户
        // {
        //   "userlist":[
        //          {
        //              "userid":"webrtc11"
        //          }
        //      ]
        // }
        this.addUser(data)
        break
      case 1032:
        // console.log(TAG_NAME, '远端用户退房通知：', code)
        // 1032 有用户退出
        this.removeUser(data)
        break
      case 1033:
        // console.log(TAG_NAME, '远端用户视频状态位变化通知：', code)
        // 1033 用户视频状态变化，新增stream或者更新stream 状态
        // {
        //   "userlist":[
        //          {
        //              "userid":"webrtc11",
        //              "playurl":" room://rtc.tencent.com?userid=xxx&streamtype=main",
        //              "streamtype":"main",
        //              "hasvideo":true
        //          }
        //      ]
        // }
        this.updateUserVideo(data)
        break
      case 1034:
        // console.log(TAG_NAME, '远端用户音频状态位变化通知：', code)
        // 1034 用户音频状态变化
        // {
        //   "userlist":[
        //          {
        //              "userid":"webrtc11",
        //              "playurl":" room://rtc.tencent.com?userid=xxx&streamtype=main",
        //              "hasaudio":false
        //          }
        //      ]
        // }
        this.updateUserAudio(data)
        break
    }
  }
  /**
   * 处理用户进房事件
   * @param {Object} data pusher 下发的数据
   */
  addUser(data) {
    // console.log(TAG_NAME, 'addUser', data)
    const incomingUserList = data.userlist
    const userMap = this.userMap
    if (Array.isArray(incomingUserList) && incomingUserList.length > 0) {
      incomingUserList.forEach((item) => {
        const userID = item.userid
        // 已经在 map 中的用户
        let user = this.getUser(userID)
        if (!user) {
          // 新增用户
          user = new User({
            userID: userID
          })
          this.userList.push({
            userID: userID,
          })
        }
        userMap.set(userID, user)
        this._emitter.emit(EVENT.REMOTE_USER_JOIN, {
          userID: userID,
          userList: this.userList
        })
        // console.log(TAG_NAME, 'addUser', item, userMap.get(userID), this.userMap)
      })
    }
  }
  /**
   * 处理用户退房事件
   * @param {Object} data pusher 下发的数据 {userlist}
   */
  removeUser(data) {
    // console.log(TAG_NAME, 'removeUser', data)
    const incomingUserList = data.userlist
    if (Array.isArray(incomingUserList) && incomingUserList.length > 0) {
      incomingUserList.forEach((item) => {
        const userID = item.userid
        let user = this.getUser(userID)
        // 偶现SDK触发退房事件前没有触发进房事件
        if (!user || !user.streams) {
          return
        }
        // 从userList 里删除指定的用户和 stream
        this._removeUserAndStream(userID)
        // 重置
        user.streams['main'] && user.streams['main'].reset()
        user.streams['aux'] && user.streams['aux'].reset()
        // 用户退出，释放引用，外部调用该 user 所有stream 的 playerContext.stop() 方法停止播放
        // TODO 触发时机提前了，方便外部用户做出处理，时机仍需进一步验证
        this._emitter.emit(EVENT.REMOTE_USER_LEAVE, {
          userID: userID,
          userList: this.userList,
          streamList: this.streamList
        })
        user = undefined
        this.userMap.delete(userID)
        // console.log(TAG_NAME, 'removeUser', this.userMap)
      })
    }
  }
  /**
   * 处理用户视频通知事件
   * @param {Object} data pusher 下发的数据 {userlist}
   */
  updateUserVideo(data) {
    console.log(TAG_NAME, 'updateUserVideo', data)
    const incomingUserList = data.userlist
    if (Array.isArray(incomingUserList) && incomingUserList.length > 0) {
      incomingUserList.forEach((item) => {
        console.log(item, "5555555555555555555555555")
        // if (item.userid == "0" || item.userid == "share-userId") {
          const userID = item.userid
          const streamType = item.streamtype
          const streamID = userID + '_' + streamType
          const hasVideo = item.hasvideo
          const src = item.playurl
          const user = this.getUser(userID)
          // 更新指定用户的属性
          if (user) {
            // 查找对应的 stream
            let stream = user.streams[streamType]
            console.log(TAG_NAME, 'updateUserVideo start', user, streamType, stream)
            // 常规逻辑
            // 新来的stream，新增到 user.steams 和 streamList，streamList 包含所有用户(有音频或视频)的 stream
            if (!stream) {
              // 不在 user streams 里，需要新建
              user.streams[streamType] = stream = new Stream({
                userID,
                streamID,
                hasVideo,
                src,
                streamType
              })
              this._addStream(stream)
            } else {
              // 更新 stream 属性
              stream.setProperty({
                hasVideo
              })
              if (!hasVideo && !stream.hasAudio) {
                this._removeStream(stream)
              }
              // or
              // if (hasVideo) {
              //   stream.setProperty({ hasVideo })
              // } else if (!stream.hasAudio) {
              //   // hasVideo == false && hasAudio == false
              //   this._removeStream(stream)
              // }
            }
            // 特殊逻辑
            if (streamType === 'aux') {
              if (hasVideo) {
                // 辅流需要修改填充模式
                stream.objectFit = 'contain'
                this._addStream(stream)
              } else {
                // 如果是辅流要移除该 stream，否则需要移除 player
                this._removeStream(stream)
              }
            }
            // 更新所属user 的 hasXxx 值
            this.userList.find((item) => {
              if (item.userID === userID) {
                item[`has${streamType.replace(/^\S/, (s) => s.toUpperCase())}Video`] = hasVideo
                return true
              }
            })
            console.log(TAG_NAME, 'updateUserVideo end', user, streamType, stream)
            const eventName = hasVideo ? EVENT.REMOTE_VIDEO_ADD : EVENT.REMOTE_VIDEO_REMOVE
            this._emitter.emit(eventName, {
              stream: stream,
              streamList: this.streamList,
              userList: this.userList
            })
            // console.log(TAG_NAME, 'updateUserVideo', user, stream, this.userMap)
          }
        // }
      })
    }
  }
  /**
   * 处理用户音频通知事件
   * @param {Object} data pusher 下发的数据 {userlist}
   */
  updateUserAudio(data) {
    // console.log(TAG_NAME, 'updateUserAudio', data)
    const incomingUserList = data.userlist
    if (Array.isArray(incomingUserList) && incomingUserList.length > 0) {
      incomingUserList.forEach((item) => {
        // if (item.userid == "0" || item.userid == "share-userId") {
          const userID = item.userid
          // 音频只跟着 stream main ，这里只修改 main
          const streamType = 'main'
          const streamID = userID + '_' + streamType
          const hasAudio = item.hasaudio
          const src = item.playurl
          const user = this.getUser(userID)
          if (user) {
            let stream = user.streams[streamType]
            // if (!stream) {
            //   user.streams[streamType] = stream = new Stream({ streamType: streamType })
            //   this._addStream(stream)
            // }

            // 常规逻辑
            // 新来的stream，新增到 user.steams 和 streamList，streamList 包含所有用户的 stream
            if (!stream) {
              // 不在 user streams 里，需要新建
              user.streams[streamType] = stream = new Stream({
                userID,
                streamID,
                hasAudio,
                src,
                streamType
              })
              this._addStream(stream)
            } else {
              // 更新 stream 属性
              stream.setProperty({
                hasAudio
              })
              if (!hasAudio && !stream.hasVideo) {
                this._removeStream(stream)
              }
              // or
              // if (hasAudio) {
              //   stream.setProperty({ hasAudio })
              // } else if (!stream.hasVideo) {
              // // hasVideo == false && hasAudio == false
              //   this._removeStream(stream)
              // }
            }
            // stream.userID = userID
            // stream.streamID = userID + '_' + streamType
            // stream.hasAudio = hasAudio
            // stream.src = src
            // 更新所属 user 的 hasXxx 值
            this.userList.find((item) => {
              if (item.userID === userID) {
                item[`has${streamType.replace(/^\S/, (s) => s.toUpperCase())}Audio`] = hasAudio
                return true
              }
            })
            const eventName = hasAudio ? EVENT.REMOTE_AUDIO_ADD : EVENT.REMOTE_AUDIO_REMOVE
            this._emitter.emit(eventName, {
              stream: stream,
              streamList: this.streamList,
              userList: this.userList
            })
            // console.log(TAG_NAME, 'updateUserAudio', user, stream, this.userMap)
          }
        // }
      })
    }
  }
  /**
   *
   * @param {String} userID 用户ID
   * @returns {Object}
   */
  getUser(userID) {
    return this.userMap.get(userID)
  }
  getStream({
    userID,
    streamType
  }) {
    const user = this.userMap.get(userID)
    if (user) {
      return user.streams[streamType]
    }
    return undefined
  }
  getUserList() {
    return this.userList
  }
  getStreamList() {
    return this.streamList
  }
  /**
   * 重置所有user 和 steam
   * @returns {Object}
   */
  reset() {
    this.streamList.forEach((item) => {
      item.reset()
    })
    this.streamList = []
    this.userList = []
    this.userMap.clear()
    return {
      userList: this.userList,
      streamList: this.streamList,
    }
  }
  on(eventCode, handler, context) {
    this._emitter.on(eventCode, handler, context)
  }
  off(eventCode, handler) {
    this._emitter.off(eventCode, handler)
  }
  /**
   * 删除用户和所有的 stream
   * @param {String} userID 用户ID
   */
  _removeUserAndStream(userID) {
    this.streamList = this.streamList.filter((item) => {
      return item.userID !== userID && item.userID !== ''
    })
    this.userList = this.userList.filter((item) => {
      return item.userID !== userID
    })
  }
  _addStream(stream) {
    console.log(stream, "ioioioooioioioo")
    // if (stream.userID == "0" || stream.userID == "share-userId") {
      if (!this.streamList.includes(stream)) {
        this.streamList.push(stream)
        console.log(this.streamList, "我是过滤后的推送列表")
      }
    // }
  }
  _removeStream(stream) {
    this.streamList = this.streamList.filter((item) => {
      if (item.userID === stream.userID && item.streamType === stream.streamType) {
        return false
      }
      return true
    })
    const user = this.getUser(stream.userID)
    user.streams[stream.streamType] = undefined
  }
}
export default UserController