new framwork

client/src/pages/api/model/create.ts

@@ -0,0 +1,48 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { ModelStatusEnum } from '@/constants/model';
+import { Model } from '@/service/models/model';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const { name } = req.body as {
+      name: string;
+    };
+
+    if (!name) {
+      throw new Error('缺少参数');
+    }
+
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    // 上限校验
+    const authCount = await Model.countDocuments({
+      userId
+    });
+    if (authCount >= 30) {
+      throw new Error('上限 30 个应用');
+    }
+
+    // 创建模型
+    const response = await Model.create({
+      name,
+      userId,
+      status: ModelStatusEnum.running
+    });
+
+    jsonRes(res, {
+      data: response._id
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/model/del.ts

@@ -0,0 +1,55 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { Chat, Model, connectToDatabase, Collection, ShareChat } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { authModel } from '@/service/utils/auth';
+
+/* 获取我的模型 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const { modelId } = req.query as { modelId: string };
+
+    if (!modelId) {
+      throw new Error('参数错误');
+    }
+
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    // 验证是否是该用户的 model
+    await authModel({
+      modelId,
+      userId
+    });
+
+    // 删除对应的聊天
+    await Chat.deleteMany({
+      modelId
+    });
+
+    // 删除收藏列表
+    await Collection.deleteMany({
+      modelId
+    });
+
+    // 删除分享链接
+    await ShareChat.deleteMany({
+      modelId
+    });
+
+    // 删除模型
+    await Model.deleteOne({
+      _id: modelId,
+      userId
+    });
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/model/detail.tsx

@@ -0,0 +1,36 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { authModel } from '@/service/utils/auth';
+
+/* 获取我的模型 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const { modelId } = req.query as { modelId: string };
+
+    if (!modelId) {
+      throw new Error('参数错误');
+    }
+
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    const { model } = await authModel({
+      modelId,
+      userId,
+      authOwner: false
+    });
+
+    jsonRes(res, {
+      data: model
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/model/list.ts

@@ -0,0 +1,58 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Collection, Model } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import type { ModelListResponse } from '@/api/response/model';
+
+/* 获取模型列表 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    // 根据 userId 获取模型信息
+    const [myModels, myCollections] = await Promise.all([
+      Model.find(
+        {
+          userId
+        },
+        '_id avatar name chat.systemPrompt'
+      ).sort({
+        updateTime: -1
+      }),
+      Collection.find({ userId })
+        .populate({
+          path: 'modelId',
+          select: '_id avatar name chat.systemPrompt',
+          match: { 'share.isShare': true }
+        })
+        .then((res) => res.filter((item) => item.modelId))
+    ]);
+
+    jsonRes<ModelListResponse>(res, {
+      data: {
+        myModels: myModels.map((item) => ({
+          _id: item._id,
+          name: item.name,
+          avatar: item.avatar,
+          systemPrompt: item.chat.systemPrompt
+        })),
+        myCollectionModels: myCollections
+          .map((item: any) => ({
+            _id: item.modelId?._id,
+            name: item.modelId?.name,
+            avatar: item.modelId?.avatar,
+            systemPrompt: item.modelId?.chat.systemPrompt
+          }))
+          .filter((item) => !myModels.find((model) => String(model._id) === String(item._id))) // 去重
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/model/share/collection.ts

@@ -0,0 +1,44 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Collection, Model } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+
+/* 模型收藏切换 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const { modelId } = req.query as { modelId: string };
+
+    if (!modelId) {
+      throw new Error('缺少参数');
+    }
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    const collectionRecord = await Collection.findOne({
+      userId,
+      modelId
+    });
+
+    if (collectionRecord) {
+      await Collection.findByIdAndRemove(collectionRecord._id);
+    } else {
+      await Collection.create({
+        userId,
+        modelId
+      });
+    }
+
+    await Model.findByIdAndUpdate(modelId, {
+      'share.collection': await Collection.countDocuments({ modelId })
+    });
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/model/share/getModels.ts

@@ -0,0 +1,111 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Model } from '@/service/mongo';
+import type { PagingData } from '@/types';
+import type { ShareModelItem } from '@/types/model';
+import { parseCookie } from '@/service/utils/auth';
+import { Types } from 'mongoose';
+
+/* 获取模型列表 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const {
+      searchText = '',
+      pageNum = 1,
+      pageSize = 20
+    } = req.body as { searchText: string; pageNum: number; pageSize: number };
+
+    await connectToDatabase();
+
+    let userId = '';
+
+    try {
+      userId = await parseCookie(req.headers.cookie);
+    } catch (error) {
+      error;
+    }
+
+    const regex = new RegExp(searchText, 'i');
+
+    const where = {
+      $and: [
+        { 'share.isShare': true },
+        {
+          $or: [{ name: { $regex: regex } }, { 'share.intro': { $regex: regex } }]
+        }
+      ]
+    };
+    const pipeline = [
+      {
+        $match: where
+      },
+      {
+        $lookup: {
+          from: 'collections',
+          let: { modelId: '$_id' },
+          pipeline: [
+            {
+              $match: {
+                $expr: {
+                  $and: [
+                    { $eq: ['$modelId', '$$modelId'] },
+                    {
+                      $eq: ['$userId', userId ? new Types.ObjectId(userId) : new Types.ObjectId()]
+                    }
+                  ]
+                }
+              }
+            }
+          ],
+          as: 'collections'
+        }
+      },
+      {
+        $project: {
+          _id: 1,
+          avatar: { $ifNull: ['$avatar', '/icon/logo.png'] },
+          name: 1,
+          userId: 1,
+          share: 1,
+          isCollection: {
+            $cond: {
+              if: { $gt: [{ $size: '$collections' }, 0] },
+              then: true,
+              else: false
+            }
+          }
+        }
+      },
+      {
+        $sort: { 'share.collection': -1 }
+      },
+      {
+        $skip: (pageNum - 1) * pageSize
+      },
+      {
+        $limit: pageSize
+      }
+    ];
+
+    // 获取被分享的模型
+    const [models, total] = await Promise.all([
+      // @ts-ignore
+      Model.aggregate(pipeline),
+      Model.countDocuments(where)
+    ]);
+
+    jsonRes<PagingData<ShareModelItem>>(res, {
+      data: {
+        pageNum,
+        pageSize,
+        data: models,
+        total
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

client/src/pages/api/model/update.ts

@@ -0,0 +1,52 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authUser } from '@/service/utils/auth';
+import { Model } from '@/service/models/model';
+import type { ModelUpdateParams } from '@/types/model';
+import { authModel } from '@/service/utils/auth';
+
+/* 获取我的模型 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const { name, avatar, chat, share } = req.body as ModelUpdateParams;
+    const { modelId } = req.query as { modelId: string };
+
+    if (!name || !chat || !modelId) {
+      throw new Error('参数错误');
+    }
+
+    // 凭证校验
+    const { userId } = await authUser({ req, authToken: true });
+
+    await connectToDatabase();
+
+    await authModel({
+      modelId,
+      userId
+    });
+
+    // 更新模型
+    await Model.updateOne(
+      {
+        _id: modelId,
+        userId
+      },
+      {
+        name,
+        avatar,
+        chat,
+        'share.isShare': share.isShare,
+        'share.isShareDetail': share.isShareDetail,
+        'share.intro': share.intro
+      }
+    );
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}