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
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);
|
|
}
|
|
};
|