105 lines
2.5 KiB
TypeScript
105 lines
2.5 KiB
TypeScript
import { postUploadImg, postUploadFiles } from '@/web/common/file/api';
|
|
import { BucketNameEnum } from '@fastgpt/global/common/file/constants';
|
|
|
|
/**
|
|
* upload file to mongo gridfs
|
|
*/
|
|
export const uploadFiles = ({
|
|
files,
|
|
bucketName,
|
|
metadata = {},
|
|
percentListen
|
|
}: {
|
|
files: File[];
|
|
bucketName: `${BucketNameEnum}`;
|
|
metadata?: Record<string, any>;
|
|
percentListen?: (percent: number) => void;
|
|
}) => {
|
|
const form = new FormData();
|
|
form.append('metadata', JSON.stringify(metadata));
|
|
form.append('bucketName', bucketName);
|
|
files.forEach((file) => {
|
|
form.append('file', file, encodeURIComponent(file.name));
|
|
});
|
|
return postUploadFiles(form, (e) => {
|
|
if (!e.total) return;
|
|
|
|
const percent = Math.round((e.loaded / e.total) * 100);
|
|
percentListen && percentListen(percent);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* compress image. response base64
|
|
* @param maxSize The max size of the compressed image
|
|
*/
|
|
export const compressImgAndUpload = ({
|
|
file,
|
|
maxW = 200,
|
|
maxH = 200,
|
|
maxSize = 1024 * 100
|
|
}: {
|
|
file: File;
|
|
maxW?: number;
|
|
maxH?: number;
|
|
maxSize?: number;
|
|
}) =>
|
|
new Promise<string>((resolve, reject) => {
|
|
const reader = new FileReader();
|
|
reader.readAsDataURL(file);
|
|
reader.onload = async () => {
|
|
const img = new Image();
|
|
// @ts-ignore
|
|
img.src = reader.result;
|
|
img.onload = async () => {
|
|
let width = img.width;
|
|
let height = img.height;
|
|
|
|
if (width > height) {
|
|
if (width > maxW) {
|
|
height *= maxW / width;
|
|
width = maxW;
|
|
}
|
|
} else {
|
|
if (height > maxH) {
|
|
width *= maxH / height;
|
|
height = maxH;
|
|
}
|
|
}
|
|
|
|
const canvas = document.createElement('canvas');
|
|
canvas.width = width;
|
|
canvas.height = height;
|
|
const ctx = canvas.getContext('2d');
|
|
|
|
if (!ctx) {
|
|
return reject('压缩图片异常');
|
|
}
|
|
|
|
ctx.drawImage(img, 0, 0, width, height);
|
|
const compressedDataUrl = canvas.toDataURL(file.type, 0.8);
|
|
// 移除 canvas 元素
|
|
canvas.remove();
|
|
|
|
if (compressedDataUrl.length > maxSize) {
|
|
return reject('图片太大了');
|
|
}
|
|
|
|
const src = await (async () => {
|
|
try {
|
|
const src = await postUploadImg(compressedDataUrl);
|
|
return src;
|
|
} catch (error) {
|
|
return compressedDataUrl;
|
|
}
|
|
})();
|
|
|
|
resolve(src);
|
|
};
|
|
};
|
|
reader.onerror = (err) => {
|
|
console.log(err);
|
|
reject('压缩图片异常');
|
|
};
|
|
});
|