Concat plugin to app (#1799)

projects/app/src/pages/api/admin/initv485.ts

@@ -0,0 +1,163 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { authCert } from '@fastgpt/service/support/permission/auth/common';
+import { MongoPlugin } from '@fastgpt/service/core/plugin/schema';
+import { MongoApp } from '@fastgpt/service/core/app/schema';
+import { NextAPI } from '@/service/middleware/entry';
+import { PluginTypeEnum } from '@fastgpt/global/core/plugin/constants';
+import { mongoSessionRun } from '@fastgpt/service/common/mongo/sessionRun';
+import { AppTypeEnum } from '@fastgpt/global/core/app/constants';
+
+/* 
+  1. 先读取 HTTP plugin 内容,并找到所有的子plugin,然后事务批量创建,最后修改 inited
+  2. 读取剩下未 inited 的plugin,逐一创建
+*/
+
+let success = 0;
+async function handler(req: NextApiRequest, res: NextApiResponse) {
+  await authCert({ req, authRoot: true });
+
+  const total = await MongoPlugin.countDocuments({
+    inited: { $ne: true }
+  });
+
+  console.log('Total plugin', total);
+
+  await initHttp();
+  await initPlugin();
+}
+
+async function initHttp(): Promise<any> {
+  /* 读取http插件和他的children */
+  const plugin = await MongoPlugin.findOne({
+    type: PluginTypeEnum.folder,
+    inited: { $ne: true }
+  }).lean();
+  if (!plugin) return;
+
+  const children = await MongoPlugin.find({
+    teamId: plugin.teamId,
+    parentId: plugin._id,
+    inited: { $ne: true }
+  }).lean();
+
+  await mongoSessionRun(async (session) => {
+    /* 创建HTTP插件作为目录 */
+    const [{ _id }] = await MongoApp.create(
+      [
+        {
+          teamId: plugin.teamId,
+          tmbId: plugin.tmbId,
+          type: AppTypeEnum.httpPlugin,
+          name: plugin.name,
+          avatar: plugin.avatar,
+          intro: plugin.intro,
+          metadata: plugin.metadata,
+          version: 'v2',
+          pluginData: {
+            apiSchemaStr: plugin.metadata?.apiSchemaStr,
+            customHeaders: plugin.metadata?.customHeaders
+          }
+        }
+      ],
+      { session }
+    );
+
+    /* 批量创建子插件 */
+    for await (const item of children) {
+      await MongoApp.create(
+        [
+          {
+            parentId: _id,
+            teamId: item.teamId,
+            tmbId: item.tmbId,
+            type: AppTypeEnum.plugin,
+            name: item.name,
+            avatar: item.avatar,
+            intro: item.intro,
+            version: 'v2',
+            modules: item.modules,
+            edges: item.edges,
+            pluginData: {
+              nodeVersion: item.nodeVersion,
+              pluginUniId: plugin.metadata?.pluginUid
+            }
+          }
+        ],
+        { session }
+      );
+    }
+
+    /* 更新插件信息 */
+    await MongoPlugin.findOneAndUpdate(
+      {
+        _id: plugin._id
+      },
+      {
+        $set: { inited: true }
+      },
+      { session }
+    );
+    await MongoPlugin.updateMany(
+      {
+        teamId: plugin.teamId,
+        parentId: plugin._id
+      },
+      {
+        $set: { inited: true }
+      },
+      { session }
+    );
+
+    success += children.length + 1;
+    console.log(success);
+  });
+
+  return initHttp();
+}
+
+async function initPlugin(): Promise<any> {
+  const plugin = await MongoPlugin.findOne({
+    type: PluginTypeEnum.custom,
+    inited: { $ne: true }
+  }).lean();
+  if (!plugin) return;
+
+  await mongoSessionRun(async (session) => {
+    await MongoApp.create(
+      [
+        {
+          teamId: plugin.teamId,
+          tmbId: plugin.tmbId,
+          type: AppTypeEnum.plugin,
+          name: plugin.name,
+          avatar: plugin.avatar,
+          intro: plugin.intro,
+          version: 'v2',
+          modules: plugin.modules,
+          edges: plugin.edges,
+          pluginData: {
+            nodeVersion: plugin.nodeVersion
+          }
+        }
+      ],
+      { session }
+    );
+
+    await MongoPlugin.findOneAndUpdate(
+      {
+        _id: plugin._id
+      },
+      {
+        $set: { inited: true }
+      },
+      { session }
+    );
+
+    success++;
+    console.log(success);
+  });
+
+  return initPlugin();
+}
+
+export default NextAPI(handler);