diff --git a/app/api/chat-stream/route.ts b/app/api/chat-stream/route.ts index 41f135495..060cc1fe3 100644 --- a/app/api/chat-stream/route.ts +++ b/app/api/chat-stream/route.ts @@ -1,6 +1,7 @@ import { createParser } from "eventsource-parser"; import { NextRequest } from "next/server"; import { requestOpenai } from "../common"; +import clientPromise from "../../db/mongdb"; async function createStream(req: NextRequest) { const encoder = new TextEncoder(); @@ -48,7 +49,55 @@ async function createStream(req: NextRequest) { } export async function POST(req: NextRequest) { + const accessCode = req.headers.get("access-code"); try { + // 使用 clientPromise 连接到 MongoDB 数据库 + const client = await clientPromise; + // 选择数据库和集合 + const db = client.db("chat_db"); + const usersCollection = db.collection("users"); + // 查询用户数据 + const user = await usersCollection.findOne({ key: accessCode }); + const tips = + "您的链接授权已过期，为了避免恶意盗刷，\n 关注微信公众号【coder思维】\n回复关键词：`ai` 获取授权链接 \n ![](/wx.png)"; + if (!user) { + return new Response(tips); + } + if (user["expire"] < new Date().getTime()) { + // 判断用户是否过期 + return new Response(tips); + } + + // 创建查询条件 + // 计算24小时前的时间戳 + const currentTime = new Date(); + const startTime = new Date(currentTime.getTime() - 24 * 60 * 60 * 1000); + + // 创建查询条件 + const query = { + username: user["username"], + create_time: { $gte: startTime }, + }; + + const collection = db.collection("chat_logs"); + // 根据查询条件查询记录数 + const recordCount = await collection.countDocuments(query); + if (recordCount > 1000) { + // 判断用户是否超过1000条记录 + return new Response( + "您的聊天提问已超过1000条，为了避免恶意盗刷，请过稍后再试，24小时内只能提问1000条", + ); + } + // 创建聊天记录 + const chatLog = { + username: user["username"], + create_time: new Date().getTime(), + content: req.body, + }; + collection.insertOne(chatLog).then(() => { + client.close(); + }); + const stream = await createStream(req); return new Response(stream); } catch (error) { @@ -58,7 +107,3 @@ export async function POST(req: NextRequest) { ); } } - -export const config = { - runtime: "edge", -}; diff --git a/app/api/common.ts b/app/api/common.ts index 842eeacaf..abee6fd71 100644 --- a/app/api/common.ts +++ b/app/api/common.ts @@ -10,6 +10,7 @@ export async function requestOpenai(req: NextRequest) { const openaiPath = req.headers.get("path"); console.log("[Proxy] ", openaiPath); + console.log("[apiKey] ", apiKey); return fetch(`${PROTOCOL}://${BASE_URL}/${openaiPath}`, { headers: { diff --git a/app/components/chat.tsx b/app/components/chat.tsx index 03f905ecf..f39a304d5 100644 --- a/app/components/chat.tsx +++ b/app/components/chat.tsx @@ -4,7 +4,7 @@ import { memo, useState, useRef, useEffect, useLayoutEffect } from "react"; import SendWhiteIcon from "../icons/send-white.svg"; import BrainIcon from "../icons/brain.svg"; import ExportIcon from "../icons/share.svg"; -import ReturnIcon from "../icons/return.svg"; +import MenuIcon from "../icons/menu.svg"; import CopyIcon from "../icons/copy.svg"; import DownloadIcon from "../icons/download.svg"; import LoadingIcon from "../icons/three-dots.svg"; @@ -565,7 +565,7 @@ export function Chat(props: {
} + icon={} bordered title={Locale.Chat.Actions.ChatList} onClick={props?.showSideBar} diff --git a/app/components/home.tsx b/app/components/home.tsx index b9c6a8a3e..20e2a4ac7 100644 --- a/app/components/home.tsx +++ b/app/components/home.tsx @@ -34,9 +34,9 @@ export function Loading(props: { noLogo?: boolean }) { ); } -const Settings = dynamic(async () => (await import("./settings")).Settings, { - loading: () => , -}); +// const Settings = dynamic(async () => (await import("./settings")).Settings, { +// loading: () => , +// }); const ChatList = dynamic(async () => (await import("./chat-list")).ChatList, { loading: () => , @@ -142,7 +142,7 @@ function _Home() { ); const chatStore = useChatStore(); const loading = !useHasHydrated(); - const [showSideBar, setShowSideBar] = useState(true); + const [showSideBar, setShowSideBar] = useState(false); //移动端隐藏侧边栏 // setting const [openSettings, setOpenSettings] = useState(false); @@ -169,7 +169,7 @@ function _Home() { className={styles.sidebar + ` ${showSideBar && styles["sidebar-show"]}`} >
-
ChatGPT Next
+
ai大师助手
Build your own AI assistant.
@@ -196,7 +196,7 @@ function _Home() { onClick={chatStore.deleteSession} />
-
+ {/*
} onClick={() => { @@ -205,7 +205,7 @@ function _Home() { }} shadow /> -
+
*/}
- {openSettings ? ( - { - setOpenSettings(false); - setShowSideBar(true); - }} - /> - ) : ( - setShowSideBar(true)} - sideBarShowing={showSideBar} - /> - )} + setShowSideBar(true)} + sideBarShowing={showSideBar} + />
); diff --git a/app/components/settings.tsx b/app/components/settings.tsx index e418d4843..5bb435f20 100644 --- a/app/components/settings.tsx +++ b/app/components/settings.tsx @@ -223,8 +223,9 @@ export function Settings(props: { closeSettings: () => void }) { - void }) { onClick={() => checkUpdate(true)} /> )} - + */}