NapCatQQ/src/common/file.ts

import fs from 'fs';
import { stat } from 'fs/promises';
import crypto, { randomUUID } from 'crypto';
import path from 'node:path';
import { solveProblem } from '@/common/helper';

export interface HttpDownloadOptions {
    url: string;
    headers?: Record<string, string> | string;
}

type Uri2LocalRes = {
    success: boolean,
    errMsg: string,
    fileName: string,
    path: string
}

// 定义一个异步函数来检查文件是否存在
export function checkFileExist(path: string, timeout: number = 3000): Promise<void> {
    return new Promise((resolve, reject) => {
        const startTime = Date.now();

        function check() {
            if (fs.existsSync(path)) {
                resolve();
            } else if (Date.now() - startTime > timeout) {
                reject(new Error(`文件不存在: ${path}`));
            } else {
                setTimeout(check, 100);
            }
        }

        check();
    });
}

// 定义一个异步函数来检查文件是否存在
export async function checkFileExistV2(path: string, timeout: number = 3000): Promise<void> {
    // 使用 Promise.race 来同时进行文件状态检查和超时计时
    // Promise.race 会返回第一个解决（resolve）或拒绝（reject）的 Promise
    await Promise.race([
        checkFile(path),
        timeoutPromise(timeout, `文件不存在: ${path}`),
    ]);
}

// 转换超时时间至 Promise
function timeoutPromise(timeout: number, errorMsg: string): Promise<void> {
    return new Promise((_, reject) => {
        setTimeout(() => {
            reject(new Error(errorMsg));
        }, timeout);
    });
}

// 异步检查文件是否存在
async function checkFile(path: string): Promise<void> {
    try {
        await stat(path);
    } catch (error: any) {
        if (error.code === 'ENOENT') {
            // 如果文件不存在，则抛出一个错误
            throw new Error(`文件不存在: ${path}`);
        } else {
            // 对于 stat 调用的其他错误，重新抛出
            throw error;
        }
    }
    // 如果文件存在，则无需做任何事情，Promise 解决（resolve）自身
}

export function calculateFileMD5(filePath: string): Promise<string> {
    return new Promise((resolve, reject) => {
        // 创建一个流式读取器
        const stream = fs.createReadStream(filePath);
        const hash = crypto.createHash('md5');

        stream.on('data', (data: Buffer) => {
            // 当读取到数据时，更新哈希对象的状态
            hash.update(data);
        });

        stream.on('end', () => {
            // 文件读取完成，计算哈希
            const md5 = hash.digest('hex');
            resolve(md5);
        });

        stream.on('error', (err: Error) => {
            // 处理可能的读取错误
            reject(err);
        });
    });
}

async function tryDownload(options: string | HttpDownloadOptions, useReferer: boolean = false): Promise<Response> {
    let url: string;
    let headers: Record<string, string> = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.71 Safari/537.36',
    };
    if (typeof options === 'string') {
        url = options;
        headers['Host'] = new URL(url).hostname;
    } else {
        url = options.url;
        if (options.headers) {
            if (typeof options.headers === 'string') {
                headers = JSON.parse(options.headers);
            } else {
                headers = options.headers;
            }
        }
    }
    if (useReferer && !headers['Referer']) {
        headers['Referer'] = url;
    }
    const fetchRes = await fetch(url, { headers }).catch((err) => {
        if (err.cause) {
            throw err.cause;
        }
        throw err;
    });
    return fetchRes;
}

export async function httpDownload(options: string | HttpDownloadOptions): Promise<Buffer> {
    const useReferer = typeof options === 'string';
    let resp = await tryDownload(options);
    if (resp.status === 403 && useReferer) {
        resp = await tryDownload(options, true);
    }
    if (!resp.ok) throw new Error(`下载文件失败: ${resp.statusText}`);
    const blob = await resp.blob();
    const buffer = await blob.arrayBuffer();
    return Buffer.from(buffer);
}

export enum FileUriType {
    Unknown = 0,
    Local = 1,
    Remote = 2,
    Base64 = 3
}

export async function checkUriType(Uri: string) {
    const LocalFileRet = await solveProblem((uri: string) => {
        if (fs.existsSync(uri)) {
            return { Uri: uri, Type: FileUriType.Local };
        }
        return undefined;
    }, Uri);
    if (LocalFileRet) return LocalFileRet;
    const OtherFileRet = await solveProblem((uri: string) => {
        // 再判断是否是Http
        if (uri.startsWith('http:') || uri.startsWith('https:')) {
            return { Uri: uri, Type: FileUriType.Remote };
        }
        // 再判断是否是Base64
        if (uri.startsWith('base64:')) {
            return { Uri: uri, Type: FileUriType.Base64 };
        }
        // 默认file://
        if (uri.startsWith('file:')) {
            const filePath: string = decodeURIComponent(uri.startsWith('file:///') && process.platform === 'win32' ? uri.slice(8) : uri.slice(7));
            return { Uri: filePath, Type: FileUriType.Local };
        }
        if (uri.startsWith('data:')) {
            const data = uri.split(',')[1];
            if (data) return { Uri: data, Type: FileUriType.Base64 };
        }
    }, Uri);
    if (OtherFileRet) return OtherFileRet;

    return { Uri: Uri, Type: FileUriType.Unknown };
}

export async function uriToLocalFile(dir: string, uri: string): Promise<Uri2LocalRes> {
    const { Uri: HandledUri, Type: UriType } = await checkUriType(uri);

    const filename = randomUUID();
    const filePath = path.join(dir, filename);

    switch (UriType) {
        case FileUriType.Local:
            const fileExt = path.extname(HandledUri);
            const localFileName = path.basename(HandledUri, fileExt) + fileExt;
            const tempFilePath = path.join(dir, filename + fileExt);
            fs.copyFileSync(HandledUri, tempFilePath);
            return { success: true, errMsg: '', fileName: localFileName, path: tempFilePath };

        case FileUriType.Remote:
            const buffer = await httpDownload(HandledUri);
            fs.writeFileSync(filePath, buffer, { flag: 'wx' });
            return { success: true, errMsg: '', fileName: filename, path: filePath };

        case FileUriType.Base64:
            const base64 = HandledUri.replace(/^base64:\/\//, '');
            const base64Buffer = Buffer.from(base64, 'base64');
            fs.writeFileSync(filePath, base64Buffer, { flag: 'wx' });
            return { success: true, errMsg: '', fileName: filename, path: filePath };

        default:
            return { success: false, errMsg: `识别URL失败, uri= ${uri}`, fileName: '', path: '' };
    }
}