wechaty-puppet-wechat4u/src/wechat4u/events/event.ts

Wechat4u Puppet for Wechaty

import { Puppet, log } from 'wechaty-puppet'
import type * as PUPPET from 'wechaty-puppet'
import type { WebMessageRawPayload } from '../../web-schemas.js'

export enum EventType {
  Message,
  Friendship,
  RoomInvite,
  RoomJoin,
  RoomLeave,
  RoomTopic,
}

export interface EventPayloadSpec {
  [EventType.Message]: WebMessageRawPayload;
  [EventType.Friendship]: PUPPET.payloads.Friendship;
  [EventType.RoomInvite]: PUPPET.payloads.RoomInvitation;
  [EventType.RoomJoin]: PUPPET.payloads.EventRoomJoin;
  [EventType.RoomLeave]: PUPPET.payloads.EventRoomLeave;
  [EventType.RoomTopic]: PUPPET.payloads.EventRoomTopic;
}

export interface Event<T extends keyof EventPayloadSpec> {
  type: T;
  payload: EventPayloadSpec[T];
}

export type EventPayload = EventPayloadSpec[keyof EventPayloadSpec] | null;
export type EventParserHandler = (puppet: Puppet, message: WebMessageRawPayload) => Promise<EventPayload>;
type EventParser = { type: EventType, handler: EventParserHandler, };

const EventParserList: Array<EventParser> = []
export function addEventParser (eventType: EventType, parser: EventParserHandler): void {
  EventParserList.push({
    handler: parser,
    type: eventType,
  })
}

export async function parseEvent (puppet: Puppet, message: WebMessageRawPayload): Promise<Event<any>> {
  for (const parser of EventParserList) {
    try {
      const parsedPayload = await parser.handler(puppet, message)
      if (parsedPayload) {
        return {
          payload: parsedPayload,
          type: parser.type,
        }
      }
    } catch (e) {
      log.error('[Event]', `parse message error: ${(e as Error).stack}`)
    }
  }

  // return normal as message bvy default
  return {
    payload: message,
    type: EventType.Message,
  }
}