crystelf-plugin/lib/yunzai/utils.js

export default class YunzaiUtils {
  /**
   * 获取消息中的图片
   * @param e
   * @param limit 限制
   * @returns {Promise<*[]>}
   */
  static async getImages(e, limit = 1) {
    let imgUrls = [];
    const me = `https://q1.qlogo.cn/g?b=qq&s=640&nk=${e.user_id}`;
    //消息中的图片
    if (e.source || e.reply_id) {
      let reply;
      if (e.getReply) reply = await e.getReply();
      else {
        const history = await (e.isGroup ? e.group : e.friend).getChatHistory(
          e.isGroup ? e.source.seq : e.source.time,
          1
        );
        reply = history?.pop()?.message;
      }
      if (reply) imgUrls = reply.filter((m) => m.type === 'image').map((m) => m.url);
    }

    //当前消息中的图片
    if (!imgUrls.length && e.message) {
      imgUrls = e.message.filter((m) => m.type === 'image').map((m) => m.url);
    }

    //没图时用头像
    if (!imgUrls.length) imgUrls = [me];
    imgUrls = imgUrls.slice(0, limit);
    return imgUrls;
  }
}