ZZZ-Plugin/apps/manage/panel.js

import { char } from '../../lib/convert.js';
import { downloadFile } from '../../lib/download.js';
import { imageResourcesPath } from '../../lib/path.js';
import common from '../../../../lib/common/common.js';
import fs from 'fs';
import path from 'path';
export async function uploadCharacterImg() {
  if (!this.e.isMaster) {
    this.reply('只有主人才能添加');
    return false;
  }
  const reg = /(上传|添加)(.+)(角色|面板)图$/;
  const match = this.e.msg.match(reg);
  if (!match) {
    return false;
  }
  const charName = match[2].trim();
  const name = char.aliasToName(charName);
  if (!name) {
    this.reply('未找到对应角色');
    return false;
  }
  const images = [];
  // 下面方法来源于miao-plugin/apps/character/ImgUpload.js
  for (const val of this.e.message) {
    if (val.type === 'image') {
      images.push(val);
    }
  }
  if (images.length === 0) {
    let source;
    if (this.e.getReply) {
      source = await this.e.getReply();
    } else if (this.e.source) {
      if (this.e.group?.getChatHistory) {
        // 支持at图片添加，以及支持后发送
        source = (
          await this.e.group.getChatHistory(this.e.source?.seq, 1)
        ).pop();
      } else if (this.e.friend?.getChatHistory) {
        source = (
          await this.e.friend.getChatHistory(this.e.source?.time + 1, 1)
        ).pop();
      }
    }
    if (source) {
      for (const val of source.message) {
        if (val.type === 'image') {
          images.push(val);
        } else if (val.type === 'xml' || val.type === 'forward') {
          let resid;
          try {
            resid = val.data
              .match(/m_resid="(\d|\w|\/|\+)*"/)[0]
              .replace(/m_resid=|"/g, '');
          } catch (err) {
            resid = val.id;
          }
          if (!resid) break;
          let message = await this.e.bot.getForwardMsg(resid);
          for (const item of message) {
            for (const i of item.message) {
              if (i.type === 'image') {
                images.push(i);
              }
            }
          }
        }
      }
    }
  }
  if (images.length <= 0) {
    this.reply(
      '消息中未找到图片，请将要发送的图片与消息一同发送或引用要添加的图像。'
    );
    return false;
  }
  const resourcesImagesPath = imageResourcesPath;
  const panelImagesPath = path.join(resourcesImagesPath, `panel/${name}`);
  let success = 0;
  let failed = 0;
  for (const image of images) {
    let fileName = new Date().getTime().toString();
    let fileType = 'png';
    if (image.file) {
      fileName = image.file.substring(0, image.file.lastIndexOf('.'));
      fileType = image.file.substring(image.file.lastIndexOf('.') + 1);
    }
    const filePath = path.join(panelImagesPath, `${fileName}.${fileType}`);
    const result = await downloadFile(image.url, filePath);
    if (result) {
      success++;
    } else {
      failed++;
    }
  }
  this.reply(`成功上传${success}张图片，失败${failed}张图片。`);
  return false;
}

export async function getCharacterImages() {
  const reg = /(获取|查看)(.+)(角色|面板)图(\d+)?$/;
  const match = this.e.msg.match(reg);
  if (!match) {
    return false;
  }
  const charName = match[2].trim();
  const name = char.aliasToName(charName);
  let page = match[4];
  if (!name) {
    this.reply('未找到对应角色');
    return false;
  }
  const pageSize = 5;
  const resourcesImagesPath = imageResourcesPath;
  const panelImagesPath = path.join(resourcesImagesPath, `panel/${name}`);
  const files = fs.readdirSync(panelImagesPath);
  const images = [];
  for (const file of files) {
    images.push(path.join(panelImagesPath, file));
  }
  page = Number(page);
  if (!page || page < 1) {
    page = 1;
  }
  const start = (page - 1) * pageSize;
  const end = page * pageSize;
  if (start >= images.length) {
    this.reply('哪有这么多图片');
    return false;
  }
  const imagePaths = images.slice(start, end);
  const imageMsg = imagePaths.map(imagePath => {
    const id = String(images.findIndex(value => value === imagePath) + 1);
    const msg = [`ID：${id}`, segment.image(imagePath)];
    return msg;
  });
  imageMsg.unshift(
    `当前页数：${page}，总页数：${Math.ceil(
      images.length / pageSize
    )}，查询指定页数请在指令后面加上页码。`
  );
  imageMsg.push(
    '删除或者添加后会重新排序ID，此时若想删除，请重新获取图片列表，否则可能会删除错误的图片。'
  );
  if (imageMsg.length)
    await this.reply(await common.makeForwardMsg(this.e, imageMsg));

  return false;
}

export async function deleteCharacterImg() {
  if (!this.e.isMaster) {
    this.reply('只有主人才能删除');
    return false;
  }
  const reg = /(删除)(.+)(角色|面板)图(.+)$/;
  const match = this.e.msg.match(reg);
  if (!match) {
    return false;
  }
  const charName = match[2].trim();
  const name = char.aliasToName(charName);
  if (!name) {
    this.reply('未找到对应角色');
    return false;
  }
  const ids = match[4].split(/[,，、\s]+/);
  const resourcesImagesPath = imageResourcesPath;
  const panelImagesPath = path.join(resourcesImagesPath, `panel/${name}`);
  const files = fs.readdirSync(panelImagesPath);
  const images = [];
  for (const file of files) {
    images.push(path.join(panelImagesPath, file));
  }
  const success = [];
  const failed = [];
  for (const id of ids) {
    const index = Number(id) - 1;
    if (index < 0 || index >= images.length) {
      failed.push(id);
      continue;
    }
    const imagePath = images[index];
    fs.unlinkSync(imagePath);
    success.push(id);
  }
  const msgs = [
    `成功删除ID为${success.join(',')}的图片`,
    failed ? `删除失败ID为${failed.join(',')}` : '无失败ID',
    '删除后会重新排序ID，若想要再次删除，请重新获取图片列表，否则可能会删除错误的图片。',
  ];
  this.reply(common.makeForwardMsg(this.e, msgs));
  return false;
}