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)
+  };
+}