feat: support auto delete plugin inactivated in 10mins

Author: FinleyGe

lib/mongo/models/plugins.ts

@@ -7,7 +7,8 @@ export const pluginTypeEnum = z.enum(['tool']);
 export const PluginZodSchema = z.object({
   type: z.literal('tool'),
   toolId: z.string(),
-  status: z.enum(['active', 'pending'])
+  status: z.enum(['active', 'pending']),
+  ttl: z.date().optional()
 });
 
 export type MongoPluginSchemaType = z.infer<typeof PluginZodSchema>;
@@ -19,10 +20,19 @@ const pluginMongooseSchema = new Schema({
     type: String,
     required: true,
     enum: Object.values(PluginZodSchema.shape.status.enum)
+  },
+  ttl: {
+    type: Date
   }
 });
 
 pluginMongooseSchema.index({ toolId: 1 }, { unique: true, sparse: true });
 pluginMongooseSchema.index({ type: 1 });
+pluginMongooseSchema.index(
+  { ttl: 1 },
+  {
+    expireAfterSeconds: 0
+  }
+);
 
 export const MongoPlugin = getMongoModel('fastgpt_plugins', pluginMongooseSchema);

modules/tool/api/upload/confirmUpload.ts

@@ -36,6 +36,9 @@ export default s.route(contract.tool.upload.confirmUpload, async ({ body }) => {
       {
         $set: {
           status: PluginZodSchema.shape.status.Enum.active
+        },
+        $unset: {
+          ttl: 1
         }
       },
       {

modules/tool/api/upload/install.ts

@@ -35,7 +35,9 @@ export default s.route(contract.tool.upload.install, async ({ body }) => {
       },
       type: 'tool'
     },
-    {},
+    {
+      status: 'active'
+    },
     {
       upsert: true
     }

modules/tool/utils.ts

@@ -11,6 +11,7 @@ import { ToolDetailSchema } from './type/api';
 import { catchError } from '@/utils/catch';
 import { mimeMap } from '@/s3/const';
 import { MongoPlugin } from '@/mongo/models/plugins';
+import { addHours, addMinutes } from 'date-fns';
 
 /**
  * Move files from unzipped structure to dist directory
@@ -227,7 +228,8 @@ export const parsePkg = async (filepath: string, setTTL: boolean = true) => {
       type: 'tool'
     },
     {
-      status: 'pending'
+      status: 'pending',
+      ttl: addMinutes(new Date(), 10)
     },
     {
       upsert: true