v4.5.2 (#439)

packages/service/core/plugin/controller.ts

@@ -0,0 +1,63 @@
+import { CreateOnePluginParams, UpdatePluginParams } from '@fastgpt/global/core/plugin/controller';
+import { MongoPlugin } from './schema';
+import { FlowModuleTemplateType } from '@fastgpt/global/core/module/type';
+import { FlowNodeTypeEnum } from '@fastgpt/global/core/module/node/constant';
+import { formatPluginIOModules } from '@fastgpt/global/core/module/utils';
+
+export async function createOnePlugin(data: CreateOnePluginParams & { userId: string }) {
+  const { _id } = await MongoPlugin.create(data);
+  return _id;
+}
+
+export async function updateOnePlugin({
+  id,
+  userId,
+  ...data
+}: UpdatePluginParams & { userId: string }) {
+  await MongoPlugin.findOneAndUpdate({ _id: id, userId }, data);
+}
+
+export async function deleteOnePlugin({ id, userId }: { id: string; userId: string }) {
+  await MongoPlugin.findOneAndDelete({ _id: id, userId });
+}
+export async function getUserPlugins({ userId }: { userId: string }) {
+  return MongoPlugin.find({ userId }, 'name avatar intro');
+}
+export async function getOnePluginDetail({ id, userId }: { userId: string; id: string }) {
+  return MongoPlugin.findOne({ _id: id, userId });
+}
+/* plugin templates */
+export async function getUserPlugins2Templates({
+  userId
+}: {
+  userId: string;
+}): Promise<FlowModuleTemplateType[]> {
+  const plugins = await MongoPlugin.find({ userId }).lean();
+
+  return plugins.map((plugin) => ({
+    id: String(plugin._id),
+    flowType: FlowNodeTypeEnum.pluginModule,
+    logo: plugin.avatar,
+    name: plugin.name,
+    description: plugin.intro,
+    intro: plugin.intro,
+    showStatus: false,
+    inputs: [],
+    outputs: []
+  }));
+}
+/* one plugin 2 module detail */
+export async function getPluginModuleDetail({ id, userId }: { userId: string; id: string }) {
+  const plugin = await getOnePluginDetail({ id, userId });
+  if (!plugin) return Promise.reject('plugin not found');
+  return {
+    id: String(plugin._id),
+    flowType: FlowNodeTypeEnum.pluginModule,
+    logo: plugin.avatar,
+    name: plugin.name,
+    description: plugin.intro,
+    intro: plugin.intro,
+    showStatus: false,
+    ...formatPluginIOModules(String(plugin._id), plugin.modules)
+  };
+}

packages/service/core/plugin/schema.ts

@@ -0,0 +1,42 @@
+import { connectionMongo, type Model } from '../../common/mongo';
+const { Schema, model, models } = connectionMongo;
+import type { PluginItemSchema } from '@fastgpt/global/core/plugin/type.d';
+
+export const ModuleCollectionName = 'plugins';
+
+const PluginSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  avatar: {
+    type: String,
+    default: '/icon/logo.svg'
+  },
+  intro: {
+    type: String,
+    default: ''
+  },
+  updateTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  modules: {
+    type: Array,
+    default: []
+  }
+});
+
+try {
+  PluginSchema.index({ userId: 1 });
+} catch (error) {
+  console.log(error);
+}
+
+export const MongoPlugin: Model<PluginItemSchema> =
+  models[ModuleCollectionName] || model(ModuleCollectionName, PluginSchema);