miao-plugin/apps/character/character-img-upload.js

import fs from "fs";
import { promisify } from "util";
import { pipeline } from "stream";
import { segment } from "oicq";
import MD5 from "md5";
import fetch from "node-fetch";
import lodash from "lodash";
import Data from "../../components/Data.js";
import { Character } from "../../components/models.js";


const _res_path = process.cwd() + "/plugins/miao-plugin/resources/";
let regex = /^#?\s*(?:喵喵)?(?:上传|添加)(.+)(?:照片|写真|图片|图像)\s*$/;

export const rule = {
  uploadCharacterImage: {
    hashMark: true,
    reg: "^#*喵喵(上传|添加)(.+)写真.*$",
    describe: "喵喵上传角色写真",
  },
};

export async function uploadCharacterImg(e) {
  let promise = await isAllowedToUploadCharacterImage(e);
  if (!promise) {
    return;
  }

  let imageMessages = [];
  let msg = e.msg;
  let regRet = regex.exec(msg);
  //通过解析正则获取消息中的角色名
  if (!regRet || !regRet[1]) {
    return;
  }
  let char = Character.get(regRet[1]);
  if (!char || !char.name) {
    return;
  }
  let name = char.name;
  for (let val of e.message) {
    if ("image" === val.type) {
      imageMessages.push(val);
    }
  }
  if (imageMessages.length <= 0) {
    // TODO 支持at图片添加，以及支持后发送
    e.reply("消息中未找到图片，请将要发送的图片与消息一同发送..");
    return true;
  }
  await saveImages(e, name, imageMessages);
  return true;
}

async function saveImages(e, name, imageMessages) {
  let imgMaxSize = e.groupConfig.imgMaxSize || 1;
  let pathSuffix = `character-img/${name}/upload`;
  let path = _res_path + pathSuffix;

  if (!fs.existsSync(path)) {
    Data.createDir(_res_path, pathSuffix);
  }
  let senderName = lodash.truncate(e.sender.card, { length: 8 });
  let imgCount = 0;
  for (let val of imageMessages) {
    const response = await fetch(val.url);
    if (!response.ok) {
      e.reply("图片下载失败。");
      return true;
    }
    if (response.headers.get("size") > 1024 * 1024 * imgMaxSize) {
      e.reply([segment.at(e.user_id, senderName), "添加失败：图片太大了。"]);
      return true;
    }
    let fileName = val.file.substring(0, val.file.lastIndexOf("."));
    let fileType = val.file.substring(val.file.lastIndexOf(".") + 1);
    if (response.headers.get("content-type") === "image/gif") {
      fileType = "gif";
    }
    let imgPath = `${path}/${fileName}.${fileType}`;
    const streamPipeline = promisify(pipeline);
    await streamPipeline(response.body, fs.createWriteStream(imgPath));

    // 使用md5作为文件名
    let buffers = fs.readFileSync(imgPath);
    let base64 = new Buffer.from(buffers, 'base64').toString();
    let md5 = MD5(base64);
    let newImgPath = `${path}/${md5}.${fileType}`
    if (fs.existsSync(newImgPath)) {
      fs.unlink(newImgPath, (err) => {
        console.log('unlink', err);
      });
    }
    fs.rename(imgPath, newImgPath, (err) => {
      console.log('rename', err);
    })
    imgCount++;
    Bot.logger.mark(`添加成功: ${path}/${fileName}`);
  }
  e.reply([segment.at(e.user_id, senderName), `\n成功添加${imgCount}张${name}图片。`]);
  return true;
}

async function isAllowedToUploadCharacterImage(e) {
  if (!e.message) {
    return false;
  }
  if (!e.msg) {
    return false;
  }
  if (!e.isMaster) {
    return false;
  }

  // 由于添加角色图是全局，暂时屏蔽非管理员的添加
  if (e.isPrivate) {
    if (!e.isMaster) {
      e.reply(`只有主人才能添加。`);
      return false;
    }
    return true;
  }

  let group_id = e.group_id;
  if (!group_id) {
    return false;
  }
  if (e.groupConfig.imgAddLimit === 2) {
    if (!e.isMaster) {
      e.reply(`只有主人才能添加。`);
      return false;
    }
  }
  if (e.groupConfig.imgAddLimit === 1 && !e.isMaster) {
    if (!(e.sender.role === "owner" || e.sender.role === "admin")) {
      e.reply(`只有管理员才能添加。`);
      return false;
    }
  }
  return true;
}