300 lines
7.9 KiB
JavaScript
300 lines
7.9 KiB
JavaScript
'use strict';
|
||
import FileApi from '@/sheep/api/infra/file';
|
||
|
||
const ERR_MSG_OK = 'chooseAndUploadFile:ok';
|
||
const ERR_MSG_FAIL = 'chooseAndUploadFile:fail';
|
||
|
||
function chooseImage(opts) {
|
||
const {
|
||
count,
|
||
sizeType = ['original', 'compressed'],
|
||
sourceType = ['album', 'camera'],
|
||
extension,
|
||
} = opts;
|
||
return new Promise((resolve, reject) => {
|
||
uni.chooseImage({
|
||
count,
|
||
sizeType,
|
||
sourceType,
|
||
extension,
|
||
success(res) {
|
||
resolve(normalizeChooseAndUploadFileRes(res, 'image'));
|
||
},
|
||
fail(res) {
|
||
reject({
|
||
errMsg: res.errMsg.replace('chooseImage:fail', ERR_MSG_FAIL),
|
||
});
|
||
},
|
||
});
|
||
});
|
||
}
|
||
|
||
function chooseVideo(opts) {
|
||
const { camera, compressed, maxDuration, sourceType = ['album', 'camera'], extension } = opts;
|
||
return new Promise((resolve, reject) => {
|
||
uni.chooseVideo({
|
||
camera,
|
||
compressed,
|
||
maxDuration,
|
||
sourceType,
|
||
extension,
|
||
success(res) {
|
||
const { tempFilePath, duration, size, height, width } = res;
|
||
resolve(
|
||
normalizeChooseAndUploadFileRes(
|
||
{
|
||
errMsg: 'chooseVideo:ok',
|
||
tempFilePaths: [tempFilePath],
|
||
tempFiles: [
|
||
{
|
||
name: (res.tempFile && res.tempFile.name) || '',
|
||
path: tempFilePath,
|
||
size,
|
||
type: (res.tempFile && res.tempFile.type) || '',
|
||
width,
|
||
height,
|
||
duration,
|
||
fileType: 'video',
|
||
cloudPath: '',
|
||
},
|
||
],
|
||
},
|
||
'video',
|
||
),
|
||
);
|
||
},
|
||
fail(res) {
|
||
reject({
|
||
errMsg: res.errMsg.replace('chooseVideo:fail', ERR_MSG_FAIL),
|
||
});
|
||
},
|
||
});
|
||
});
|
||
}
|
||
|
||
function chooseAll(opts) {
|
||
const { count, extension } = opts;
|
||
return new Promise((resolve, reject) => {
|
||
let chooseFile = uni.chooseFile;
|
||
if (typeof wx !== 'undefined' && typeof wx.chooseMessageFile === 'function') {
|
||
chooseFile = wx.chooseMessageFile;
|
||
}
|
||
if (typeof chooseFile !== 'function') {
|
||
return reject({
|
||
errMsg: ERR_MSG_FAIL + ' 请指定 type 类型，该平台仅支持选择 image 或 video。',
|
||
});
|
||
}
|
||
chooseFile({
|
||
type: 'all',
|
||
count,
|
||
extension,
|
||
success(res) {
|
||
resolve(normalizeChooseAndUploadFileRes(res));
|
||
},
|
||
fail(res) {
|
||
reject({
|
||
errMsg: res.errMsg.replace('chooseFile:fail', ERR_MSG_FAIL),
|
||
});
|
||
},
|
||
});
|
||
});
|
||
}
|
||
|
||
function normalizeChooseAndUploadFileRes(res, fileType) {
|
||
res.tempFiles.forEach((item, index) => {
|
||
if (!item.name) {
|
||
item.name = item.path.substring(item.path.lastIndexOf('/') + 1);
|
||
}
|
||
if (fileType) {
|
||
item.fileType = fileType;
|
||
}
|
||
item.cloudPath = Date.now() + '_' + index + item.name.substring(item.name.lastIndexOf('.'));
|
||
});
|
||
if (!res.tempFilePaths) {
|
||
res.tempFilePaths = res.tempFiles.map((file) => file.path);
|
||
}
|
||
return res;
|
||
}
|
||
|
||
function convertToArrayBuffer(uniFile) {
|
||
return new Promise((resolve, reject) => {
|
||
const fs = uni.getFileSystemManager();
|
||
|
||
fs.readFile({
|
||
filePath: uniFile.path, // 确保路径正确
|
||
success: (fileRes) => {
|
||
try {
|
||
// 将读取的内容转换为 ArrayBuffer
|
||
const arrayBuffer = new Uint8Array(fileRes.data).buffer;
|
||
resolve(arrayBuffer);
|
||
} catch (error) {
|
||
reject(new Error(`转换为 ArrayBuffer 失败: ${error.message}`));
|
||
}
|
||
},
|
||
fail: (error) => {
|
||
reject(new Error(`读取文件失败: ${error.errMsg}`));
|
||
},
|
||
});
|
||
});
|
||
}
|
||
|
||
function uploadCloudFiles(files, max = 5, onUploadProgress) {
|
||
files = JSON.parse(JSON.stringify(files));
|
||
const len = files.length;
|
||
let count = 0;
|
||
let self = this;
|
||
return new Promise((resolve) => {
|
||
while (count < max) {
|
||
next();
|
||
}
|
||
|
||
function next() {
|
||
let cur = count++;
|
||
if (cur >= len) {
|
||
!files.find((item) => !item.url && !item.errMsg) && resolve(files);
|
||
return;
|
||
}
|
||
const fileItem = files[cur];
|
||
const index = self.files.findIndex((v) => v.uuid === fileItem.uuid);
|
||
fileItem.url = '';
|
||
delete fileItem.errMsg;
|
||
|
||
uniCloud
|
||
.uploadFile({
|
||
filePath: fileItem.path,
|
||
cloudPath: fileItem.cloudPath,
|
||
fileType: fileItem.fileType,
|
||
onUploadProgress: (res) => {
|
||
res.index = index;
|
||
onUploadProgress && onUploadProgress(res);
|
||
},
|
||
})
|
||
.then((res) => {
|
||
fileItem.url = res.fileID;
|
||
fileItem.index = index;
|
||
if (cur < len) {
|
||
next();
|
||
}
|
||
})
|
||
.catch((res) => {
|
||
fileItem.errMsg = res.errMsg || res.message;
|
||
fileItem.index = index;
|
||
if (cur < len) {
|
||
next();
|
||
}
|
||
});
|
||
}
|
||
});
|
||
}
|
||
|
||
async function uploadFiles(choosePromise, { onChooseFile, onUploadProgress }) {
|
||
// 获取选择的文件
|
||
const res = await choosePromise;
|
||
// 处理文件选择回调
|
||
let files = res.tempFiles || [];
|
||
if (onChooseFile) {
|
||
const customChooseRes = onChooseFile(res);
|
||
if (typeof customChooseRes !== 'undefined') {
|
||
files = await Promise.resolve(customChooseRes);
|
||
if (typeof files === 'undefined') {
|
||
files = res.tempFiles || []; // Fallback
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果是前端直连上传
|
||
if (UPLOAD_TYPE.CLIENT === import.meta.env.SHOPRO_UPLOAD_TYPE) {
|
||
// 为上传创建一组 Promise
|
||
const uploadPromises = files.map(async (file) => {
|
||
try {
|
||
// 1.1 获取文件预签名地址
|
||
const { data: presignedInfo } = await FileApi.getFilePresignedUrl(file.name);
|
||
// 1.2 获取二进制文件对象
|
||
const fileBuffer = await convertToArrayBuffer(file);
|
||
|
||
// 返回上传的 Promise
|
||
return new Promise((resolve, reject) => {
|
||
uni.request({
|
||
url: presignedInfo.uploadUrl, // 预签名的上传 URL
|
||
method: 'PUT', // 使用 PUT 方法
|
||
header: {
|
||
'Content-Type':
|
||
file.fileType + '/' + file.name.substring(file.name.lastIndexOf('.') + 1), // 设置内容类型
|
||
},
|
||
data: fileBuffer, // 文件的路径，适用于小程序
|
||
success: (res) => {
|
||
// 1.4. 记录文件信息到后端（异步）
|
||
createFile(presignedInfo, file);
|
||
// 1.5. 重新赋值
|
||
file.url = presignedInfo.url;
|
||
console.log('上传成功:', res);
|
||
resolve(file);
|
||
},
|
||
fail: (err) => {
|
||
console.error('上传失败:', err);
|
||
reject(err);
|
||
},
|
||
});
|
||
});
|
||
} catch (error) {
|
||
console.error('上传失败：', error);
|
||
throw error;
|
||
}
|
||
});
|
||
|
||
// 等待所有上传完成
|
||
return await Promise.all(uploadPromises); // 返回已上传的文件列表
|
||
} else {
|
||
// 后端上传
|
||
for (let file of files) {
|
||
const { data } = await FileApi.uploadFile(file.path);
|
||
file.url = data;
|
||
}
|
||
|
||
return files;
|
||
}
|
||
}
|
||
|
||
function chooseAndUploadFile(
|
||
opts = {
|
||
type: 'all',
|
||
},
|
||
) {
|
||
if (opts.type === 'image') {
|
||
return uploadFiles(chooseImage(opts), opts);
|
||
} else if (opts.type === 'video') {
|
||
return uploadFiles(chooseVideo(opts), opts);
|
||
}
|
||
return uploadFiles(chooseAll(opts), opts);
|
||
}
|
||
|
||
/**
|
||
* 创建文件信息
|
||
* @param vo 文件预签名信息
|
||
* @param file 文件
|
||
*/
|
||
function createFile(vo, file) {
|
||
const fileVo = {
|
||
configId: vo.configId,
|
||
url: vo.url,
|
||
path: file.name,
|
||
name: file.name,
|
||
type: file.fileType,
|
||
size: file.size,
|
||
};
|
||
FileApi.createFile(fileVo);
|
||
return fileVo;
|
||
}
|
||
|
||
/**
|
||
* 上传类型
|
||
*/
|
||
const UPLOAD_TYPE = {
|
||
// 客户端直接上传（只支持S3服务）
|
||
CLIENT: 'client',
|
||
// 客户端发送到后端上传
|
||
SERVER: 'server',
|
||
};
|
||
|
||
export { chooseAndUploadFile, uploadCloudFiles };
|