mirror of
https://github.com/ZZZure/ZZZ-Plugin.git
synced 2025-12-16 21:27:47 +00:00
85 lines
2.7 KiB
JavaScript
85 lines
2.7 KiB
JavaScript
import path from 'path';
|
||
import fs from 'fs';
|
||
import { ZZZ_SQUARE_AVATAR, ZZZ_SQUARE_BANGBOO } from './mysapi/api.js';
|
||
import { imageResourcesPath } from './path.js';
|
||
import { weapon } from './convert.js';
|
||
import { getResourceRemotePath } from './assets.js';
|
||
|
||
const ZZZ_SQUARE_AVATAR_PATH = path.join(imageResourcesPath, 'square_avatar');
|
||
const ZZZ_SQUARE_BANGBOO_PATH = path.join(
|
||
imageResourcesPath,
|
||
'bangboo_square_avatar'
|
||
);
|
||
const ZZZ_WEAPON_PATH = path.join(imageResourcesPath, 'weapon');
|
||
const ZZZ_GUIDES_PATH = path.join(imageResourcesPath, 'guides');
|
||
|
||
// 将下面的下载封装起来，支持错误重试5次
|
||
const downloadFile = async (url, savePath) => {
|
||
const _download = async (url, savePath, retry = 0) => {
|
||
if (retry > 5) {
|
||
return null;
|
||
}
|
||
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);
|
||
};
|
||
|
||
/**
|
||
*
|
||
* @param {string | number} charID
|
||
* @returns Promise<string>
|
||
*/
|
||
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;
|
||
const url = `${ZZZ_SQUARE_AVATAR}/${filename}`;
|
||
const savePath = avatarPath;
|
||
const download = await downloadFile(url, savePath);
|
||
return download;
|
||
};
|
||
|
||
/**
|
||
*
|
||
* @param {string | number} bangbooId
|
||
* @returns Promise<string>
|
||
*/
|
||
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;
|
||
const url = `${ZZZ_SQUARE_BANGBOO}/${filename}`;
|
||
const savePath = bangbooPath;
|
||
const download = await downloadFile(url, savePath);
|
||
return download;
|
||
};
|
||
|
||
/**
|
||
* Get weapon image path
|
||
* @param {string} id
|
||
* @returns Promise<string>
|
||
*/
|
||
export const getWeaponImage = async id => {
|
||
logger.debug('getWeaponImage', id);
|
||
const name = weapon.IDToWeaponFileName(id);
|
||
logger.debug('getWeaponImage', name);
|
||
const filename = `${name}.png`;
|
||
const weaponPath = path.join(ZZZ_WEAPON_PATH, filename);
|
||
if (fs.existsSync(weaponPath)) return weaponPath;
|
||
const url = await getResourceRemotePath('weapon', filename);
|
||
const savePath = weaponPath;
|
||
const download = await downloadFile(url, savePath);
|
||
logger.mark('getWeaponImage', download);
|
||
return download;
|
||
};
|