4.6.4-alpha (#582)

packages/service/common/file/image/controller.ts

@@ -1,3 +1,4 @@
+import { UploadImgProps } from '@fastgpt/global/common/file/api';
 import { imageBaseUrl } from './constant';
 import { MongoImage } from './schema';
 
@@ -9,11 +10,10 @@ export const maxImgSize = 1024 * 1024 * 12;
 export async function uploadMongoImg({
   base64Img,
   teamId,
-  expiredTime
-}: {
-  base64Img: string;
+  expiredTime,
+  metadata
+}: UploadImgProps & {
   teamId: string;
-  expiredTime?: Date;
 }) {
   if (base64Img.length > maxImgSize) {
     return Promise.reject('Image too large');
@@ -24,7 +24,8 @@ export async function uploadMongoImg({
   const { _id } = await MongoImage.create({
     teamId,
     binary: Buffer.from(base64Data, 'base64'),
-    expiredTime
+    expiredTime: expiredTime,
+    metadata
   });
 
   return getMongoImgUrl(String(_id));
@@ -37,3 +38,9 @@ export async function readMongoImg({ id }: { id: string }) {
   }
   return data?.binary;
 }
+
+export async function delImgByFileIdList(fileIds: string[]) {
+  return MongoImage.deleteMany({
+    'metadata.fileId': { $in: fileIds.map((item) => String(item)) }
+  });
+}

packages/service/common/file/image/schema.ts

@@ -5,13 +5,17 @@ const { Schema, model, models } = connectionMongo;
 const ImageSchema = new Schema({
   teamId: {
     type: Schema.Types.ObjectId,
-    ref: TeamCollectionName
+    ref: TeamCollectionName,
+    required: true
   },
   binary: {
     type: Buffer
   },
   expiredTime: {
     type: Date
+  },
+  metadata: {
+    type: Object
   }
 });
 
@@ -21,7 +25,7 @@ try {
   console.log(error);
 }
 
-export const MongoImage: Model<{ teamId: string; binary: Buffer }> =
+export const MongoImage: Model<{ teamId: string; binary: Buffer; metadata?: Record<string, any> }> =
   models['image'] || model('image', ImageSchema);
 
 MongoImage.syncIndexes();