owobot/plugins/chat.js

const settings = require('../settings.json')
const parsePlain = require('../util/chatparse_plain.js')
const parseConsole = require('../util/chatparse_console.js')
const parse1204 = require('../util/parseNBT.js')
const { getMessage } = require('../util/lang.js')
const fs = require('fs')
const convertChatStyleItem = (item) => {
  const output = {}
  for (const i in item) {
    output[i] = item[i].value
  }
  return output
}
const convertChatTypeItem = (item) => {
  if (item.style) {
    return {
      translation_key: item.translation_key.value,
      parameters: item.parameters.value.value,
      style: convertChatStyleItem(item.style.value)
    }
  } else {
    return {
      translation_key: item.translation_key.value,
      parameters: item.parameters.value.value,
      style: {}
    }
  }
}

// Level 0: highly specific parsers for certain players
// Level 1: server chat format parsers
// Level 2: generic parsers
const parsers = [[], [], []]
const bpl = fs.readdirSync('plugins/chatParsers')
for (const plugin of bpl) {
  if (!plugin.endsWith('.js')) {
    continue
  }
  try {
    const parser = require(`./chatParsers/${plugin}`)
    parsers[parser.priority].push(parser.parse)
  } catch (e) { console.log(e) }
}

module.exports = {
  load: (b) => {
    b.messageCount = 0
    b.chatDisabledUntil = 0
    b.interval.antiSpam = setInterval(() => {
      b.messageCount = 0
    }, 4000)
    b.messageTypes = []
    b._client.on('registry_data', (data) => {
      if (data.codec.value['minecraft:chat_type']) {
        b.messageTypes = data.codec.value['minecraft:chat_type']
        const nbtItems = data.codec.value['minecraft:chat_type'].value.value.value.value
        nbtItems.forEach((item, i) => {
          b.messageTypes[i] = convertChatTypeItem(item.element.value.chat.value)
        })
      }
    })
    b._client.on('profileless_chat', (data) => {
      let messageType = b.messageTypes[data.type]
      if (messageType === undefined) messageType = { translation_key: '%s', parameters: ['content'] }
      const json = { translate: messageType.translation_key, with: [] }
      messageType.parameters.forEach((item, i) => {
        if (item === 'content') {
          json.with[i] = parse1204(data.message)
        } else if (item === 'sender') {
          json.with[i] = parse1204(data.name)
        } else if (item === 'target') {
          json.with[i] = parse1204(data.target)
        }
      })
      for (const i in messageType.style) {
        json[i] = messageType.style[i]
      }
      const message = parsePlain(parse1204(data.message))
      const uuid = b.findUUID(parsePlain(parse1204(data.name)))
      const nickname = b.findDisplayName(uuid)
      const username = parsePlain(parse1204(data.name))
      b.emit('chat_unparsed', {
        json,
        type: 'profileless',
        uuid,
        message,
        nickname,
        username,
        playerChatType: messageType
      })
    })

    b._client.on('player_chat', (data) => {
      let messageType = b.messageTypes[data.type]
      if (messageType === undefined) messageType = { translation_key: '%s', parameters: ['content'] }
      const json = { translate: messageType.translation_key, with: [] }
      messageType.parameters.forEach((item, i) => {
        if (item === 'content') {
          if (messageType.translation_key === '%s') {
            json.with[i] = parse1204(data.unsignedChatContent)
          } else {
            json.with[i] = data.plainMessage
          }
        } else if (item === 'sender') {
          json.with[i] = parse1204(data.networkName)
        } else if (item === 'target') {
          json.with[i] = parse1204(data.networkTargetName)
        }
      })
      for (const i in messageType.style) {
        json[i] = messageType.style[i]
      }
      b.emit('chat_unparsed', {
        json,
        type: 'player',
        uuid: data.senderUuid,
        message: data.plainMessage,
        nickname: parsePlain(parse1204(data.networkName)),
        username: b.findRealNameFromUUID(data.senderUuid),
        playerChatType: messageType
      })
    })

    b._client.on('system_chat', (data) => {
      const json = parse1204(data.content)
      b.emit('chat_unparsed', {
        json,
        type: 'system',
        uuid: '00000000-0000-0000-0000-000000000000',
        message: '',
        nickname: '',
        username: '',
        playerChatType: {}
      })
    })

    b._client.on('chat', (data) => { // Legacy chat for versions <1.19
      const json = parse1204(data.message)
      let nickname
      let username
      let message
      let uuid
      if (data.uuid) uuid = data.uuid
      b.emit('chat_unparsed', {
        json,
        type: 'legacy',
        uuid,
        message,
        nickname,
        username,
        playerChatType: {}
      })
    })

    b.on('chat_unparsed', (data) => {
      for (const lvl of parsers) {
        for (const item of lvl) {
          const output = item(data, b)
          if (output.parsed) {
            b.emit('chat', output)
            return
          }
        }
      }
      b.emit('chat', {
        parsed: true,
        json: data.json,
        type: data.type,
        subtype: 'fallback',
        uuid: '00000000-0000-0000-0000-000000000000',
        message: '',
        nickname: '',
        username: ''
      })
    })

    b.on('chat', (data) => {
      b.messageCount++
      if (Date.now() < b.chatDisabledUntil) return
      if (b.messageCount >= 100) {
        b.info(getMessage(settings.defaultLang, 'chat.antiSpamTriggered'))
        b.chatDisabledUntil = Date.now() + 30000
        return
      }
      const msgConsole = parseConsole(data.json)
      const msgPlain = parsePlain(data.json)
      if (settings.logJSONmessages) console.log(data.json)
      if (msgPlain.endsWith('\n\n\n\n\nThe chat has been cleared')) return
      if (msgPlain.startsWith('Command set: ')) return
      b.emit('plainchat', msgPlain, data.type, data.subtype)
      b.displayChat(data.type, data.subtype, `${msgConsole}\x1b[0m`)
    })
  }
}