wechaty-puppet-wechat/src/pure-function-helpers/message-raw-payload-parser.ts

Puppet WeChat for Wechaty

import type * as PUPPET from 'wechaty-puppet'

import type {
  WebMessageRawPayload,
}                         from '../web-schemas.js'

import {
  isRoomId,
}                 from './is-type.js'

import {
  messageFilename,
}                 from './message-filename.js'
import {
  webMessageType,
}                 from './web-message-type.js'

export function messageRawPayloadParser (
  rawPayload: WebMessageRawPayload,
): PUPPET.payloads.Message {
  const id                           = rawPayload.MsgId
  const talkerId                     = rawPayload.MMActualSender               // MMPeerUserName
  const text: string                 = rawPayload.MMActualContent              // Content has @id prefix added by wx
  const timestamp: number            = rawPayload.MMDisplayTime                // Javascript timestamp of milliseconds
  const msgFileName: undefined | string = messageFilename(rawPayload) || undefined

  let roomId     : undefined | string
  let listenerId : undefined | string

  // FIXME: has there any better method to know the room ID?
  if (rawPayload.MMIsChatRoom) {
    if (isRoomId(rawPayload.FromUserName)) {
      roomId = rawPayload.FromUserName // MMPeerUserName always eq FromUserName ?
    } else if (isRoomId(rawPayload.ToUserName)) {
      roomId = rawPayload.ToUserName
    } else {
      throw new Error('parse found a room message, but neither FromUserName nor ToUserName is a room(/^@@/)')
    }

    // console.log('rawPayload.FromUserName: ', rawPayload.FromUserName)
    // console.log('rawPayload.ToUserName: ', rawPayload.ToUserName)
    // console.log('rawPayload.MMPeerUserName: ', rawPayload.MMPeerUserName)
  }

  if (rawPayload.ToUserName) {
    if (!isRoomId(rawPayload.ToUserName)) {
      // if a message in room without any specific receiver, then it will set to be `undefined`
      listenerId = rawPayload.ToUserName
    }
  }

  const type: PUPPET.types.Message = webMessageType(rawPayload)

  const payloadBase = {
    filename: msgFileName,
    id,
    mentionIdList: [],
    talkerId,
    text,
    timestamp,
    type,
  }

  let payload: PUPPET.payloads.Message

  if (listenerId) {
    payload = {
      ...payloadBase,
      listenerId,
      roomId,
    }
  } else if (roomId) {
    payload = {
      ...payloadBase,
      listenerId,
      roomId,
    }
  } else {
    throw new Error('neither roomId nor listenerId')
  }

  return payload
}