import pathModule from 'path';
import { fileTypeFromBuffer } from 'file-type';
import moment from 'moment';
const _path = process.cwd();
const path = _path + "/temp/tp-bq";

// 没文件夹就创建一个
if (!fs.existsSync(path)) {
  fs.mkdirSync(path, { recursive: true })
}
if (!fs.existsSync(pathModule.join(path, 'pictures'))) {
  fs.mkdirSync(pathModule.join(path, 'pictures'), { recursive: true })
}

/**
 * 
 * @param {*} e - 输入的消息
 * @param {*} tag - 表情包标签
 * @returns 
 */
export async function getToimg(e, tag) {
    const picturesPath = pathModule.join(path, 'pictures');
    const fileImgList = await fs.promises.readdir(picturesPath);

    try {
        const sanitizedTag = tag
            .replace(/[\u200B-\u200D\uFEFF]/g, '')
            .replace(/[\[\]]/g, '')
            .trim()
            .replace(/[^a-zA-Z0-9\u4e00-\u9fa5-_.]/g, '-');
        let matchedFiles = fileImgList.filter(file => file === sanitizedTag);
        if (matchedFiles.length === 0) {
            matchedFiles = fileImgList.filter(file => file.startsWith(sanitizedTag));
        }
        if (matchedFiles.length === 0) {
            matchedFiles = fileImgList.filter(file => file.includes(sanitizedTag));
        }
        if (matchedFiles.length === 0) {
            logger.warn(`未找到匹配的表情包: ${sanitizedTag}`);
            return;
        }
        const selectedFile = matchedFiles[0];
        const picPath = pathModule.join(picturesPath, selectedFile);
        try {
            await fs.promises.access(picPath);
        } catch {
            logger.warn(`找不到指定的表情包文件: ${picPath}`);
            return;
        }
        e.reply(segment.image('file:///' + picPath));

        logger.info(`发送表情包: ${picPath}`);
        return false;
    } catch (error) {
        logger.error('Error in getToimg:', error);
    }
}

/**
 * 
 * @param {*} e - 输入的消息
 * @param {*} image - 图片Base64
 * @returns 
 */
export async function downImg(e, image) {
    try {
      let reply;
      if (e.source) {
        if (e.isGroup) {
          reply = (await e.group.getChatHistory(e.source.seq, 1)).pop()?.message;
        } else {
          reply = (await e.friend.getChatHistory(e.source.time, 1)).pop()?.message;
        }
        if (reply) {
          for (let val of reply) {
            if (val.type === "image") {
              e.img = [val.url];
              break;
            }
          }
        }
      }
      if (!e.img && !image) {
        return false;
      }
      let kWordReg = /^#?(DOWNIMG:)\s*(.*)/i;
      t = t.replace(/[\u200B-\u200D\uFEFF]/g, '').trim();
      const match = kWordReg.exec(t);
      if (!match) {
        logger.error('DOWNIMG command format invalid:', t);
        return;
      }
      let rawmsg = match[2] || "defaultTag";
      let kWord = rawmsg.replace(/，|,|、| |。/g, "-").replace(/--+/g, "-").replace(/^-|-$|--/g, "").trim() || "defaultTag";
      if (image) {
        const imageBuffer = Buffer.from(image, 'base64');
        const type = await fileTypeFromBuffer(imageBuffer);
        let picType = 'png';
        if (type && type.ext) {
          picType = type.ext;
        }
        const currentTime = moment().format("YYMMDDHHmmss");
        const safeTag = kWord.replace(/[^a-zA-Z0-9\u4e00-\u9fa5-_]/g, '-');
        const picPath = pathModule.join(path, 'pictures', `${currentTime}-${safeTag.substring(0, 200)}.${picType}`);
        logger.mark("DOWNIMG：", picPath);
        if (!fs.existsSync(pathModule.join(path, 'pictures'))) {
          fs.mkdirSync(pathModule.join(path, 'pictures'), { recursive: true });
        }
        fs.writeFileSync(picPath, imageBuffer);
        logger.info(`图片已保存，标签为：${kWord}`);
      }
    } catch (error) {
      logger.error('Error in downImg:', error);
      logger.error("保存图片时发生错误");
    }
  }