mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-23 21:13:50 +00:00

* update: Add type * fix: update import statement for NextApiRequest type * fix: update imports to use type for LexicalEditor and EditorState * Refactor imports to use 'import type' for type-only imports across multiple files - Updated imports in various components and API files to use 'import type' for better clarity and to optimize TypeScript's type checking. - Ensured consistent usage of type imports in files related to chat, dataset, workflow, and user management. - Improved code readability and maintainability by distinguishing between value and type imports. * refactor: remove old ESLint configuration and add new rules - Deleted the old ESLint configuration file from the app project. - Added a new ESLint configuration file with updated rules and settings. - Changed imports to use type-only imports in various files for better clarity and performance. - Updated TypeScript configuration to remove unnecessary options. - Added an ESLint ignore file to exclude build and dependency directories from linting. * fix: update imports to use 'import type' for type-only imports in schema files
136 lines
3.5 KiB
TypeScript
136 lines
3.5 KiB
TypeScript
import { type UploadImgProps } from '@fastgpt/global/common/file/api';
|
|
import { imageBaseUrl } from '@fastgpt/global/common/file/image/constants';
|
|
import { MongoImage } from './schema';
|
|
import { type ClientSession, Types } from '../../../common/mongo';
|
|
import { guessBase64ImageType } from '../utils';
|
|
import { readFromSecondary } from '../../mongo/utils';
|
|
import { addHours } from 'date-fns';
|
|
import { imageFileType } from '@fastgpt/global/common/file/constants';
|
|
import { retryFn } from '@fastgpt/global/common/system/utils';
|
|
|
|
export const maxImgSize = 1024 * 1024 * 12;
|
|
const base64MimeRegex = /data:image\/([^\)]+);base64/;
|
|
export async function uploadMongoImg({
|
|
base64Img,
|
|
teamId,
|
|
metadata,
|
|
shareId,
|
|
forever = false
|
|
}: UploadImgProps & {
|
|
teamId: string;
|
|
forever?: Boolean;
|
|
}) {
|
|
if (base64Img.length > maxImgSize) {
|
|
return Promise.reject('Image too large');
|
|
}
|
|
|
|
const [base64Mime, base64Data] = base64Img.split(',');
|
|
// Check if mime type is valid
|
|
if (!base64MimeRegex.test(base64Mime)) {
|
|
return Promise.reject('Invalid image base64');
|
|
}
|
|
|
|
const mime = `image/${base64Mime.match(base64MimeRegex)?.[1] ?? 'image/jpeg'}`;
|
|
const binary = Buffer.from(base64Data, 'base64');
|
|
let extension = mime.split('/')[1];
|
|
if (extension.startsWith('x-')) {
|
|
extension = extension.substring(2); // Remove 'x-' prefix
|
|
}
|
|
|
|
if (!extension || !imageFileType.includes(`.${extension}`)) {
|
|
return Promise.reject(`Invalid image file type: ${mime}`);
|
|
}
|
|
|
|
const { _id } = await retryFn(() =>
|
|
MongoImage.create({
|
|
teamId,
|
|
binary,
|
|
metadata: Object.assign({ mime }, metadata),
|
|
shareId,
|
|
expiredTime: forever ? undefined : addHours(new Date(), 1)
|
|
})
|
|
);
|
|
|
|
return `${process.env.NEXT_PUBLIC_BASE_URL || ''}${imageBaseUrl}${String(_id)}.${extension}`;
|
|
}
|
|
|
|
const getIdFromPath = (path?: string) => {
|
|
if (!path) return;
|
|
|
|
const paths = path.split('/');
|
|
const name = paths[paths.length - 1];
|
|
|
|
if (!name) return;
|
|
|
|
const id = name.split('.')[0];
|
|
if (!id || !Types.ObjectId.isValid(id)) return;
|
|
|
|
return id;
|
|
};
|
|
// 删除旧的头像,新的头像去除过期时间
|
|
export const refreshSourceAvatar = async (
|
|
path?: string,
|
|
oldPath?: string,
|
|
session?: ClientSession
|
|
) => {
|
|
const newId = getIdFromPath(path);
|
|
const oldId = getIdFromPath(oldPath);
|
|
|
|
if (!newId || newId === oldId) return;
|
|
|
|
await MongoImage.updateOne({ _id: newId }, { $unset: { expiredTime: 1 } }, { session });
|
|
|
|
if (oldId) {
|
|
await MongoImage.deleteOne({ _id: oldId }, { session });
|
|
}
|
|
};
|
|
export const removeImageByPath = (path?: string, session?: ClientSession) => {
|
|
if (!path) return;
|
|
|
|
const paths = path.split('/');
|
|
const name = paths[paths.length - 1];
|
|
|
|
if (!name) return;
|
|
|
|
const id = name.split('.')[0];
|
|
if (!id || !Types.ObjectId.isValid(id)) return;
|
|
|
|
return MongoImage.deleteOne({ _id: id }, { session });
|
|
};
|
|
|
|
export async function readMongoImg({ id }: { id: string }) {
|
|
const formatId = id.replace(/\.[^/.]+$/, '');
|
|
|
|
const data = await MongoImage.findById(formatId, undefined, {
|
|
...readFromSecondary
|
|
});
|
|
if (!data) {
|
|
return Promise.reject('Image not found');
|
|
}
|
|
|
|
return {
|
|
binary: data.binary,
|
|
mime: data.metadata?.mime ?? guessBase64ImageType(data.binary.toString('base64'))
|
|
};
|
|
}
|
|
|
|
export async function delImgByRelatedId({
|
|
teamId,
|
|
relateIds,
|
|
session
|
|
}: {
|
|
teamId: string;
|
|
relateIds: string[];
|
|
session?: ClientSession;
|
|
}) {
|
|
if (relateIds.length === 0) return;
|
|
|
|
return MongoImage.deleteMany(
|
|
{
|
|
teamId,
|
|
'metadata.relatedId': { $in: relateIds.map((id) => String(id)) }
|
|
},
|
|
{ session }
|
|
);
|
|
}
|