blivechat/frontend/src/views/Room.vue

<template>
  <chat-renderer :paidMessages="paidMessages" :messages="messages" :css="config.css"></chat-renderer>
</template>

<script>
import config from '@/api/config'
import ChatRenderer from '@/components/ChatRenderer'
import * as constants from '@/components/ChatRenderer/constants'

const COMMAND_HEARTBEAT = 0
const COMMAND_JOIN_ROOM = 1
const COMMAND_ADD_TEXT = 2
const COMMAND_ADD_GIFT = 3
const COMMAND_ADD_MEMBER = 4
const COMMAND_ADD_SUPER_CHAT = 5
const COMMAND_DEL_SUPER_CHAT = 6

export default {
  name: 'Room',
  components: {
    ChatRenderer
  },
  data() {
    let cfg = {...config.DEFAULT_CONFIG}
    cfg.blockKeywords = cfg.blockKeywords.split('\n').filter(val => val)
    cfg.blockUsers = cfg.blockUsers.split('\n').filter(val => val)
    cfg.maxSpeed = 0
    return {
      config: cfg,

      websocket: null,
      retryCount: 0,
      isDestroying: false,
      heartbeatTimerId: null,

      messagesBufferTimerId: null,
      nextId: 0,
      messagesBuffer: [], // 暂时不显示的消息，可能会丢弃
      messages: [], // 正在显示的消息
      paidMessages: []
    }
  },
  created() {
    this.wsConnect()
    if (this.$route.query.config_id) {
      this.updateConfig(this.$route.query.config_id)
    }
  },
  beforeDestroy() {
    if (this.messagesBufferTimerId) {
      window.clearInterval(this.messagesBufferTimerId)
    }
    this.isDestroying = true
    this.websocket.close()
  },
  watch: {
    config(val) {
      if (this.messagesBufferTimerId) {
        window.clearInterval(this.messagesBufferTimerId)
        this.messagesBufferTimerId = null
      }
      if (val.maxSpeed > 0) {
        this.messagesBufferTimerId = window.setInterval(this.handleMessagesBuffer, 1000 / val.maxSpeed)
      }
    }
  },
  methods: {
    async updateConfig(configId) {
      try {
        let cfg = await config.getRemoteConfig(configId)
        cfg.blockKeywords = cfg.blockKeywords.split('\n').filter(val => val)
        cfg.blockUsers = cfg.blockUsers.split('\n').filter(val => val)
        this.config = cfg
      } catch (e) {
        this.$message.error('获取配置失败：' + e)
      }
    },
    wsConnect() {
      const protocol = window.location.protocol === 'https:' ? 'wss' : 'ws'
      // 开发时使用localhost:12450
      const host = process.env.NODE_ENV === 'development' ? 'localhost:12450' : window.location.host
      const url = `${protocol}://${host}/chat`
      this.websocket = new WebSocket(url)
      this.websocket.onopen = this.onWsOpen
      this.websocket.onclose = this.onWsClose
      this.websocket.onmessage = this.onWsMessage
      this.heartbeatTimerId = window.setInterval(this.sendHeartbeat, 10 * 1000)
    },
    sendHeartbeat() {
      this.websocket.send(JSON.stringify({
        cmd: COMMAND_HEARTBEAT
      }))
    },
    onWsOpen() {
      this.retryCount = 0
      this.websocket.send(JSON.stringify({
        cmd: COMMAND_JOIN_ROOM,
        data: {
          roomId: parseInt(this.$route.params.roomId)
        }
      }))
    },
    onWsClose() {
      if (this.heartbeatTimerId) {
        window.clearInterval(this.heartbeatTimerId)
        this.heartbeatTimerId = null
      }
      if (this.isDestroying) {
        return
      }
      window.console.log(`掉线重连中${++this.retryCount}`)
      this.wsConnect()
    },
    onWsMessage(event) {
      let {cmd, data} = JSON.parse(event.data)
      let message = null
      let time = data.timestamp ? new Date(data.timestamp * 1000) : new Date()
      switch (cmd) {
      case COMMAND_ADD_TEXT:
        if (!this.config.showDanmaku || !this.filterTextMessage(data) || this.mergeSimilar(data.content)) {
          break
        }
        message = {
          id: `text_${this.nextId++}`,
          type: constants.MESSAGE_TYPE_TEXT,
          avatarUrl: data.avatarUrl,
          time: time,
          authorName: data.authorName,
          authorType: data.authorType,
          content: data.content,
          privilegeType: data.privilegeType,
          repeated: 1
        }
        break
      case COMMAND_ADD_GIFT: {
        if (!this.config.showGift) {
          break
        }
        let price = data.totalCoin / 1000
        if (price < this.config.minGiftPrice) { // 丢人
          break
        }
        message = {
          id: `gift_${this.nextId++}`,
          type: constants.MESSAGE_TYPE_SUPER_CHAT,
          avatarUrl: data.avatarUrl,
          authorName: data.authorName,
          price: price,
          time: time,
          content: '' // 有了SC，礼物不需要内容了
        }
        break
      }
      case COMMAND_ADD_MEMBER:
        if (!this.config.showGift || !this.filterSuperChatMessage(data)) {
          break
        }
        message = {
          id: `member_${this.nextId++}`,
          type: constants.MESSAGE_TYPE_MEMBER,
          avatarUrl: data.avatarUrl,
          time: time,
          authorName: data.authorName,
          title: 'NEW MEMBER!',
          content: `Welcome ${data.authorName}!`
        }
        break
      case COMMAND_ADD_SUPER_CHAT:
        if (!this.config.showGift) {
          break
        }
        if (data.price < this.config.minGiftPrice) { // 丢人
          break
        }
        message = {
          id: `sc_${data.id}`,
          type: constants.MESSAGE_TYPE_SUPER_CHAT,
          avatarUrl: data.avatarUrl,
          authorName: data.authorName,
          price: data.price,
          time: time,
          content: data.content
        }
        break
      case COMMAND_DEL_SUPER_CHAT: {
        let arrays = [this.messages, this.messagesBuffer, this.paidMessages]
        for (let id of data.ids) {
          id = `sc_${id}`
          for (let arr of arrays) {
            for (let i = 0; i < arr.length; i++) {
              if (arr[i].id === id) {
                arr.splice(i, 1)
              }
            }
          }
        }
        break
      }
      }
      if (message) {
        this.addMessageBuffer(message)
      }
    },
    filterTextMessage(data) {
      if (this.config.blockGiftDanmaku && data.isGiftDanmaku) {
        return false
      } else if (this.config.blockLevel > 0 && data.authorLevel < this.config.blockLevel) {
        return false
      } else if (this.config.blockNewbie && data.isNewbie) {
        return false
      } else if (this.config.blockNotMobileVerified && !data.isMobileVerified) {
        return false
      } else if (this.config.blockMedalLevel > 0 && data.medalLevel < this.config.blockMedalLevel) {
        return false
      }
      for (let keyword of this.config.blockKeywords) {
        if (data.content.indexOf(keyword) !== -1) {
          return false
        }
      }
      for (let user of this.config.blockUsers) {
        if (data.authorName === user) {
          return false
        }
      }
      return true
    },
    filterSuperChatMessage(data) {
      for (let keyword of this.config.blockKeywords) {
        if (data.content.indexOf(keyword) !== -1) {
          return false
        }
      }
      for (let user of this.config.blockUsers) {
        if (data.authorName === user) {
          return false
        }
      }
      return true
    },
    mergeSimilar(content) {
      if (!this.config.mergeSimilarDanmaku) {
        return false
      }
      for (let i = this.messages.length - 1; i >= 0 && i >= this.messages.length - 5; i--) {
        let message = this.messages[i]
        if (
          (message.content.indexOf(content) !== -1 || content.indexOf(message.content) !== -1) // 包含对方
          && Math.abs(message.content.length - content.length) < Math.min(message.content.length, content.length) // 长度差比两者长度都小
        ) {
          message.repeated++
          return true
        }
      }
      return false
    },
    addMessageBuffer(message) {
      if (this.config.maxSpeed > 0 && message.type === constants.MESSAGE_TYPE_TEXT) {
        message.addTime = new Date()
        this.messagesBuffer.push(message)
      } else {
        // 无速度限制或者是礼物
        this.addMessageShow(message)
      }
    },
    addMessageShow(message) {
      message.addTime = new Date()
      this.messages.push(message)
      if (message.type !== constants.MESSAGE_TYPE_TEXT) {
        this.paidMessages.unshift(message)
      }
      // 防止同时添加和删除项目时所有的项目重新渲染 https://github.com/vuejs/vue/issues/6857
      this.$nextTick(() => {
        if (this.messages.length > this.config.maxNumber) {
          this.messages.splice(0, this.messages.length - this.config.maxNumber)
        }
      })
    },
    handleMessagesBuffer() {
      // 3秒内未发出则丢弃
      let i
      let curTime = new Date()
      for (i = 0; i < this.messagesBuffer.length; i++) {
        if (curTime - this.messagesBuffer[i].addTime <= 3 * 1000) {
          break
        }
      }
      this.messagesBuffer.splice(0, i)
      if (this.messagesBuffer.length <= 0) {
        return
      }
      let message = this.messagesBuffer.shift()
      // 防止短时间发送多条时不能合并
      if (this.mergeSimilar(message.content)) {
        this.handleMessagesBuffer()
        return
      }
      this.addMessageShow(message)
    }
  }
}
</script>