尝试解决图片第一次下载失败后无法自动解决图片裂开问题&驱动盘序号错位问题

lib/download.js

@@ -1,10 +1,10 @@
 import path from 'path';
 import fs from 'fs';
-import fetch from 'node-fetch';
 import { ZZZ_SQUARE_AVATAR, ZZZ_SQUARE_BANGBOO } from './mysapi/api.js';
 import { imageResourcesPath } from './path.js';
 import { char, equip, weapon } from './convert.js';
 import { getResourceRemotePath } from './assets.js';
+import request from '../utils/request.js';
 
 const ZZZ_SQUARE_AVATAR_PATH = path.join(imageResourcesPath, 'square_avatar');
 const ZZZ_SMALL_SQUARE_AVATAR_PATH = path.join(
@@ -26,32 +26,24 @@ const ZZZ_SUIT_PATH = path.join(imageResourcesPath, 'suit');
  * 下载文件
  * @param {string} url 下载地址
  * @param {string} savePath 保存路径
- * @returns
+ * @returns {Promise<string | null>} 保存路径
  */
 export const downloadFile = async (url, savePath) => {
-  const _download = async (url, savePath, retry = 0) => {
-    // 重试次数超过 5 次则返回 null
-    if (retry > 5) {
-      return null;
+  // 下载文件
+  try {
+    const download = await request(url, {}, 5);
+    const arrayBuffer = await download.arrayBuffer();
+    const buffer = Buffer.from(arrayBuffer);
+    // 保存文件
+    if (!fs.existsSync(path.dirname(savePath))) {
+      fs.mkdirSync(path.dirname(savePath), { recursive: true });
     }
-    // 下载文件
-    try {
-      const download = await fetch(url);
-      const arrayBuffer = await download.arrayBuffer();
-      const buffer = Buffer.from(arrayBuffer);
-      // 保存文件
-      if (!fs.existsSync(path.dirname(savePath))) {
-        fs.mkdirSync(path.dirname(savePath), { recursive: true });
-      }
-      fs.writeFileSync(savePath, buffer);
-      // 返回保存路径
-      return savePath;
-    } catch (error) {
-      // 下载失败，重试
-      return await _download(url, savePath, retry + 1);
-    }
-  };
-  return await _download(url, savePath);
+    fs.writeFileSync(savePath, buffer);
+    // 返回保存路径
+    return savePath;
+  } catch (error) {
+    return null;
+  }
 };
 
 /**
@@ -62,7 +54,12 @@ export const downloadFile = async (url, savePath) => {
 export const getSquareAvatar = async charID => {
   const filename = `role_square_avatar_${charID}.png`;
   const avatarPath = path.join(ZZZ_SQUARE_AVATAR_PATH, filename);
-  if (fs.existsSync(avatarPath)) return avatarPath;
+  if (fs.existsSync(avatarPath)) {
+    const stats = fs.statSync(avatarPath);
+    if (stats.size > 0) {
+      return avatarPath;
+    }
+  }
   const url = `${ZZZ_SQUARE_AVATAR}/${filename}`;
   const savePath = avatarPath;
   const download = await downloadFile(url, savePath);
@@ -80,7 +77,12 @@ export const getSmallSquareAvatar = async charID => {
   if (!sprite) return null;
   const filename = `IconRoleGeneral${sprite}.png`;
   const avatarPath = path.join(ZZZ_SMALL_SQUARE_AVATAR_PATH, filename);
-  if (fs.existsSync(avatarPath)) return avatarPath;
+  if (fs.existsSync(avatarPath)) {
+    const stats = fs.statSync(avatarPath);
+    if (stats.size > 0) {
+      return avatarPath;
+    }
+  }
   const url = await getResourceRemotePath('role_general', filename);
   const savePath = avatarPath;
   const download = await downloadFile(url, savePath);
@@ -95,7 +97,12 @@ export const getSmallSquareAvatar = async charID => {
 export const getSquareBangboo = async bangbooId => {
   const filename = `bangboo_rectangle_avatar_${bangbooId}.png`;
   const bangbooPath = path.join(ZZZ_SQUARE_BANGBOO_PATH, filename);
-  if (fs.existsSync(bangbooPath)) return bangbooPath;
+  if (fs.existsSync(bangbooPath)) {
+    const stats = fs.statSync(bangbooPath);
+    if (stats.size > 0) {
+      return bangbooPath;
+    }
+  }
   const url = `${ZZZ_SQUARE_BANGBOO}/${filename}`;
   const savePath = bangbooPath;
   const download = await downloadFile(url, savePath);
@@ -111,7 +118,12 @@ export const getWeaponImage = async id => {
   const name = weapon.IDToWeaponFileName(id);
   const filename = `${name}.png`;
   const weaponPath = path.join(ZZZ_WEAPON_PATH, filename);
-  if (fs.existsSync(weaponPath)) return weaponPath;
+  if (fs.existsSync(weaponPath)) {
+    const stats = fs.statSync(weaponPath);
+    if (stats.size > 0) {
+      return weaponPath;
+    }
+  }
   const url = await getResourceRemotePath('weapon', filename);
   const savePath = weaponPath;
   const download = await downloadFile(url, savePath);
@@ -128,7 +140,12 @@ export const getRoleImage = async id => {
   if (!sprite) return null;
   const filename = `IconRole${sprite}.png`;
   const rolePath = path.join(ZZZ_ROLE_PATH, filename);
-  if (fs.existsSync(rolePath)) return rolePath;
+  if (fs.existsSync(rolePath)) {
+    const stats = fs.statSync(rolePath);
+    if (stats.size > 0) {
+      return rolePath;
+    }
+  }
   const url = await getResourceRemotePath('role', filename);
   const savePath = rolePath;
   const download = await downloadFile(url, savePath);
@@ -145,7 +162,12 @@ export const getRoleCircleImage = async id => {
   if (!sprite) return null;
   const filename = `IconRoleCircle${sprite}.png`;
   const roleCirclePath = path.join(ZZZ_ROLE_CIRCLE_PATH, filename);
-  if (fs.existsSync(roleCirclePath)) return roleCirclePath;
+  if (fs.existsSync(roleCirclePath)) {
+    const stats = fs.statSync(roleCirclePath);
+    if (stats.size > 0) {
+      return roleCirclePath;
+    }
+  }
   const url = await getResourceRemotePath('role_circle', filename);
   const savePath = roleCirclePath;
   const download = await downloadFile(url, savePath);
@@ -161,7 +183,12 @@ export const getSuitImage = async suitId => {
   const suitName = equip.equipIdToSprite(suitId);
   const filename = `${suitName}.png`;
   const suitPath = path.join(ZZZ_SUIT_PATH, filename);
-  if (fs.existsSync(suitPath)) return suitPath;
+  if (fs.existsSync(suitPath)) {
+    const stats = fs.statSync(suitPath);
+    if (stats.size > 0) {
+      return suitPath;
+    }
+  }
   const url = await getResourceRemotePath('suit', filename);
   const savePath = suitPath;
   const download = await downloadFile(url, savePath);
@@ -177,7 +204,12 @@ export const getSuit3DImage = async suitId => {
   const suitName = equip.equipIdToSprite(suitId);
   const filename = `${suitName}_3d.png`;
   const suitPath = path.join(ZZZ_SUIT_3D_PATH, filename);
-  if (fs.existsSync(suitPath)) return suitPath;
+  if (fs.existsSync(suitPath)) {
+    const stats = fs.statSync(suitPath);
+    if (stats.size > 0) {
+      return suitPath;
+    }
+  }
   const url = await getResourceRemotePath('suit_3d', filename);
   const savePath = suitPath;
   const download = await downloadFile(url, savePath);