Function downloadAndProcessImage

outdated/gpt/gpt.ts:269–336

(
  client: TelegramClient,
  message: Api.Message,
  infoMessage: Api.Message
)

// 下载并处理图片
async function downloadAndProcessImage(
  client: TelegramClient,
  message: Api.Message,
  infoMessage: Api.Message
): Promise<{ imagePath: string; imageSource: string }> {
  const tempDir = os.tmpdir();
  const imageName = `gpt_tmp_${Math.random()
    .toString(36)
    .substring(7)}_${Date.now()}.png`;
  const imagePath = path.join(tempDir, imageName);

  try {
    // 下载图片
    await infoMessage.edit({ text: "下载图片..." });

    let mediaMsg = message;
    const replyMsg = await message.getReplyMessage();
    if (!message.media && replyMsg?.media) {
      mediaMsg = replyMsg;
    }

    if (!mediaMsg.media) {
      throw new Error("未找到图片");
    }

    // 尝试下载图片
    const buffer = await client.downloadMedia(mediaMsg.media, {
      progressCallback: (received: any, total: any) => {
        const percent = (Number(received) * 100) / Number(total);
        infoMessage
          .edit({
            text: `下载图片 ${percent.toFixed(1)}%`,
          })
          .catch(() => {});
      },
    });

    if (!buffer) {
      throw new Error("图片下载失败");
    }

    // 保存图片
    await fs.promises.writeFile(imagePath, buffer as any);
    await infoMessage.edit({ text: "下载图片 100%" });

    // 检查是否需要上传图片
    const imageUploadEnabled =
      ConfigManager.get(CONFIG_KEYS.GPT_IMAGE_UPLOAD).toLowerCase() === "true";

    let imageSource: string;
    if (imageUploadEnabled) {
      const imageUrl = await uploadImage(imagePath);
      imageSource = imageUrl;
    } else {
      const imageBuffer = await fs.promises.readFile(imagePath);
      const base64 = imageBuffer.toString("base64");
      imageSource = `data:image/png;base64,${base64}`;
    }