feat: new ui

src/pages/api/chat/delChatRecordByContentId.ts

@@ -6,11 +6,7 @@ import { authToken } from '@/service/utils/auth';
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
     const { chatId, contentId } = req.query as { chatId: string; contentId: string };
-    const { authorization } = req.headers;
 
-    if (!authorization) {
-      throw new Error('无权操作');
-    }
     if (!chatId || !contentId) {
       throw new Error('缺少参数');
     }
@@ -18,7 +14,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 凭证校验
-    const userId = await authToken(authorization);
+    const userId = await authToken(req);
 
     const chatRecord = await Chat.findById(chatId);
 

src/pages/api/chat/getHistory.ts

@@ -6,7 +6,7 @@ import { authToken } from '@/service/utils/auth';
 /* 获取历史记录 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const userId = await authToken(req.headers.authorization);
+    const userId = await authToken(req);
 
     await connectToDatabase();
 
@@ -14,7 +14,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       {
         userId
       },
-      '_id title modelId'
+      '_id title modelId updateTime latestChat'
     )
       .sort({ updateTime: -1 })
       .limit(20);

src/pages/api/chat/init.ts

@@ -1,33 +1,57 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Chat } from '@/service/mongo';
+import { connectToDatabase, Chat, Model } from '@/service/mongo';
 import type { InitChatResponse } from '@/api/response/chat';
 import { authToken } from '@/service/utils/auth';
 import { ChatItemType } from '@/types/chat';
 import { authModel } from '@/service/utils/auth';
 import mongoose from 'mongoose';
+import { ModelStatusEnum } from '@/constants/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 
 /* 初始化我的聊天框，需要身份验证 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { authorization } = req.headers;
-    const userId = await authToken(authorization);
+    const userId = await authToken(req);
 
-    const { modelId, chatId } = req.query as { modelId: string; chatId: '' | string };
-
-    if (!modelId) {
-      throw new Error('缺少参数');
-    }
+    let { modelId, chatId } = req.query as { modelId: '' | string; chatId: '' | string };
 
     await connectToDatabase();
 
-    // 获取 model 数据
-    const { model } = await authModel({ modelId, userId, authUser: false, authOwner: false });
+    let model: ModelSchema;
+
+    // 没有 modelId 时，直接获取用户的第一个id
+    if (!modelId) {
+      const myModel = await Model.findOne({ userId });
+      if (!myModel) {
+        const { _id } = await Model.create({
+          name: 'AI助手1',
+          userId,
+          status: ModelStatusEnum.running
+        });
+        model = (await Model.findById(_id)) as ModelSchema;
+      } else {
+        model = myModel;
+      }
+      modelId = model._id;
+    } else {
+      // 校验使用权限
+      const authRes = await authModel({ modelId, userId, authUser: false, authOwner: false });
+      model = authRes.model;
+    }
 
     // 历史记录
     let history: ChatItemType[] = [];
 
     if (chatId) {
+      // auth chatId
+      const chat = await Chat.countDocuments({
+        _id: chatId,
+        userId
+      });
+      if (chat === 0) {
+        throw new Error('聊天框不存在');
+      }
       // 获取 chat.content 数据
       history = await Chat.aggregate([
         {
@@ -59,9 +83,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       data: {
         chatId: chatId || '',
         modelId: modelId,
-        name: model.name,
-        avatar: model.avatar,
-        intro: model.share.intro,
+        model: {
+          name: model.name,
+          avatar: model.avatar,
+          intro: model.share.intro,
+          canUse: model.share.isShare || String(model.userId) === userId
+        },
         chatModel: model.chat.chatModel,
         history
       }

src/pages/api/chat/removeHistory.ts

@@ -1,6 +1,5 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { ChatItemType } from '@/types/chat';
 import { connectToDatabase, Chat } from '@/service/mongo';
 import { authToken } from '@/service/utils/auth';
 
@@ -8,7 +7,7 @@ import { authToken } from '@/service/utils/auth';
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
     const { id } = req.query;
-    const userId = await authToken(req.headers.authorization);
+    const userId = await authToken(req);
 
     await connectToDatabase();
 

src/pages/api/chat/saveChat.ts

@@ -20,7 +20,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       throw new Error('缺少参数');
     }
 
-    const userId = await authToken(req.headers.authorization);
+    const userId = await authToken(req);
 
     await connectToDatabase();
 
@@ -40,7 +40,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         userId,
         modelId,
         content,
-        title: content[0].value.slice(0, 20)
+        title: content[0].value.slice(0, 20),
+        latestChat: content[content.length - 1].value
       });
       return jsonRes(res, {
         data: _id
@@ -53,7 +54,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
             $each: content
           }
         },
-        updateTime: new Date()
+        updateTime: new Date(),
+        latestChat: content[content.length - 1].value
       });
     }
     jsonRes(res);