You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

74 lines
2.1 KiB
TypeScript

import { consola } from 'consola';
import dotenv from 'dotenv';
import s3 from './s3';
import type { ImgInfo, S3UserConfig, UploadResult } from './s3/types';
import { formatPath } from './s3/utils';
dotenv.config();
if (!process.env.DOC_S3_ACCESS_KEY_ID) {
consola.error('请配置 Doc S3 存储的环境变量: DOC_S3_ACCESS_KEY_ID');
// eslint-disable-next-line unicorn/no-process-exit
process.exit(1);
}
if (!process.env.DOC_S3_SECRET_ACCESS_KEY) {
consola.error('请配置 Doc S3 存储的环境变量: DOC_S3_SECRET_ACCESS_KEY');
// eslint-disable-next-line unicorn/no-process-exit
process.exit(1);
}
if (!process.env.DOC_S3_PUBLIC_DOMAIN) {
consola.error('请配置 Doc S3 存储的环境变量: DOC_S3_PUBLIC_DOMAIN');
// eslint-disable-next-line unicorn/no-process-exit
process.exit(1);
}
export const BASE_PATH = 'blog/assets';
export const uploader = async (
file: File,
filename: string,
basePath: string = BASE_PATH,
uploadPath?: string,
) => {
const item: ImgInfo = {
buffer: Buffer.from(await file.arrayBuffer()),
extname: file.name.split('.').pop() as string,
fileName: file.name,
mimeType: file.type,
};
const userConfig: S3UserConfig = {
accessKeyId: process.env.DOC_S3_ACCESS_KEY_ID || '',
bucketName: 'hub-apac-1',
endpoint: 'https://d35842305b91be4b48e06ff9a9ad83f5.r2.cloudflarestorage.com',
pathPrefix: process.env.DOC_S3_PUBLIC_DOMAIN || '',
pathStyleAccess: true,
region: 'auto',
secretAccessKey: process.env.DOC_S3_SECRET_ACCESS_KEY || '',
uploadPath: uploadPath || `${basePath}${filename}.{extName}`,
};
const client = s3.createS3Client(userConfig);
let results: UploadResult;
try {
results = await s3.createUploadTask({
acl: 'public-read',
bucketName: userConfig.bucketName,
client,
item: item,
path: formatPath(item, userConfig.uploadPath),
urlPrefix: userConfig.pathPrefix,
});
return results.url;
} catch (error) {
consola.error('上传到 S3 存储发生错误,请检查网络连接和配置是否正确');
consola.error(error);
}
};