@juzi/wechaty-puppet-whatsapp/src/whatsapp/event-handler/group-event-handler.ts

Wechaty Puppet for WhatsApp

/* eslint-disable no-case-declarations */
import {
  GroupNotificationTypes,
} from '../../schema/whatsapp-interface.js'
import WhatsAppBase from '../whatsapp-base.js'

import * as PUPPET from '@juzi/wechaty-puppet'
import type {
  GroupNotification,
} from '../../schema/whatsapp-type.js'
import { log } from '../../config.js'
import {
  genRoomTopicEvent,
  genRoomJoinEvent,
} from '../../helper/pure-function/room-event-generator.js'

const PRE = 'GroupEventHandler'

export default class GroupEventHandler extends WhatsAppBase {

  public async onRoomJoin (notification: GroupNotification) {
    log.info(PRE, `onRoomJoin(${JSON.stringify(notification)})`)
    const roomId = notification.id.remote
    const roomJoinPayload: PUPPET.payloads.EventRoomJoin = {
      inviteeIdList: notification.recipientIds,
      inviterId: notification.author,
      roomId,
      timestamp: notification.timestamp,
    }
    await this.manager.syncRoomMemberList(roomId)
    this.emit('room-join', roomJoinPayload)
    this.emit('dirty', {
      payloadId: roomId,
      payloadType: PUPPET.types.Dirty.Room,
    })
  }

  public async onRoomLeave (notification: GroupNotification) {
    log.info(PRE, `onRoomLeave(${JSON.stringify(notification)})`)
    const { id, recipientIds } = notification
    const roomId = id.remote
    const isLeaveSelf = id.fromMe && recipientIds.length === 1 &&  recipientIds[0] === this.getBotId()
    const roomLeavePayload: PUPPET.payloads.EventRoomLeave = {
      removeeIdList: notification.recipientIds,
      removerId: notification.author || isLeaveSelf ? this.getBotId() : '',
      roomId,
      timestamp: notification.timestamp,
    }
    const cacheManager = await this.manager.getCacheManager()
    await cacheManager.removeRoomMemberFromList(roomId, notification.recipientIds)
    this.emit('room-leave', roomLeavePayload)
    this.emit('dirty', {
      payloadId: roomId,
      payloadType: PUPPET.types.Dirty.Room,
    })
  }

  public async onRoomUpdate (notification: GroupNotification) {
    log.info(PRE, `onRoomUpdate(${JSON.stringify(notification)})`)
    const roomId = notification.id.remote
    const cacheManager = await this.manager.getCacheManager()
    let roomPayload = await cacheManager.getContactOrRoomRawPayload(roomId)

    if (!roomPayload) {
      const rawRoom = await this.manager.requestManager.getContactById(roomId)
      const avatar = await rawRoom.getProfilePicUrl()
      roomPayload = Object.assign(rawRoom, { avatar })
      await cacheManager.setContactOrRoomRawPayload(roomId, roomPayload)
    }
    const type = notification.type
    switch (type) {
      case GroupNotificationTypes.SUBJECT:
        const roomTopicPayload = genRoomTopicEvent(notification, roomPayload)
        roomPayload.name = notification.body
        await cacheManager.setContactOrRoomRawPayload(roomId, roomPayload)
        this.emit('room-topic', roomTopicPayload)
        break
      case GroupNotificationTypes.DESCRIPTION:
        this.emit('room-announce', {
          changerId: notification.author,
          newAnnounce: notification.body,
          oldAnnounce: '',
          roomId,
          timestamp: notification.timestamp,
        })
        break
      case GroupNotificationTypes.CREATE:
        const members = await this.manager.syncRoomMemberList(roomId)
        const roomJoinPayload = genRoomJoinEvent(notification, members)
        this.emit('room-join', roomJoinPayload)
        break
      case GroupNotificationTypes.PICTURE:
        const rawRoom = await this.manager.requestManager.getContactById(roomId)
        const avatar = await rawRoom.getProfilePicUrl() || ''
        const roomPayloadInCache = await cacheManager.getContactOrRoomRawPayload(roomId)
        if (roomPayloadInCache) {
          roomPayloadInCache.avatar = avatar
          await cacheManager.setContactOrRoomRawPayload(roomId, roomPayloadInCache)
        }
        this.emit('dirty', {
          payloadId: roomId,
          payloadType: PUPPET.types.Dirty.Room,
        })
        break
    }
  }

}