80 lines
3.0 KiB
JavaScript
80 lines
3.0 KiB
JavaScript
import { assert } from 'oak-domain/lib/utils/assert';
|
|
import { OakUploadException } from '../../types/Exception';
|
|
import { OakNetworkException } from 'oak-domain/lib/types/Exception';
|
|
export default class S3 {
|
|
name = 's3';
|
|
autoInform() {
|
|
return false;
|
|
}
|
|
getConfig(application) {
|
|
const { system } = application;
|
|
const { config } = system;
|
|
const s3Config = config.Cos?.s3;
|
|
assert(s3Config);
|
|
const { accessKey, endpoint, defaultBucket } = s3Config;
|
|
const account = config.Account?.s3?.find((ele) => ele.accessKey === accessKey);
|
|
assert(account);
|
|
return {
|
|
config: s3Config,
|
|
account,
|
|
endpoint,
|
|
defaultBucket,
|
|
};
|
|
}
|
|
formKey(extraFile) {
|
|
const { id, extension, objectId } = extraFile;
|
|
assert(objectId);
|
|
return `extraFile/${objectId}${extension ? '.' + extension : ''}`;
|
|
}
|
|
async upload(extraFile, uploadFn, file, uploadToAspect, getPercent) {
|
|
const uploadMeta = extraFile.uploadMeta;
|
|
let response;
|
|
try {
|
|
// S3 使用预签名 URL 直接上传，不需要额外的 formData
|
|
response = await uploadFn(file, 'file', uploadMeta.uploadUrl, {}, true, getPercent, extraFile.id, "PUT");
|
|
}
|
|
catch (err) {
|
|
throw new OakNetworkException('网络异常,请求失败');
|
|
}
|
|
let isSuccess = false;
|
|
if (process.env.OAK_PLATFORM === 'wechatMp') {
|
|
// 小程序端上传
|
|
if (response.errMsg === 'uploadFile:ok') {
|
|
const statusCode = response.statusCode;
|
|
isSuccess = statusCode === 200 || statusCode === 204;
|
|
}
|
|
}
|
|
else {
|
|
isSuccess = response.status === 200 || response.status === 204;
|
|
}
|
|
if (isSuccess) {
|
|
return;
|
|
}
|
|
throw new OakUploadException('文件上传S3失败');
|
|
}
|
|
composeFileUrl(application, extraFile, style) {
|
|
const { config: s3CosConfig, endpoint } = this.getConfig(application);
|
|
if (s3CosConfig) {
|
|
let bucket = s3CosConfig.buckets.find((ele) => ele.name === extraFile.bucket);
|
|
if (bucket) {
|
|
const { domain, protocol, pathStyle } = bucket;
|
|
let protocol2 = protocol;
|
|
if (protocol instanceof Array) {
|
|
const index = protocol.includes('https:')
|
|
? protocol.findIndex((ele) => ele === 'https:')
|
|
: 0;
|
|
protocol2 = protocol[index];
|
|
}
|
|
const key = this.formKey(extraFile);
|
|
// 如果使用 pathStyle (Minio 常用)
|
|
if (pathStyle && endpoint) {
|
|
return `${protocol2}//${domain}/${bucket.name}/${key}${style ? style : ''}`;
|
|
}
|
|
// 否则使用虚拟主机风格 (AWS S3 默认)
|
|
return `${protocol2}//${domain}/${key}${style ? style : ''}`;
|
|
}
|
|
}
|
|
return '';
|
|
}
|
|
}
|