feat: 账单模块

This commit is contained in:
archer
2023-03-21 18:04:39 +08:00
parent 42c26bd155
commit 129f3a2a30
11 changed files with 190 additions and 37 deletions

View File

@@ -0,0 +1,51 @@
// 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, Bill } from '@/service/mongo';
import { authToken } from '@/service/utils/tools';
import type { BillSchema } from '@/types/mongoSchema';
export default async function handler(req: NextApiRequest, res: NextApiResponse) {
try {
const { authorization } = req.headers;
let { pageNum = 1, pageSize = 10 } = req.query as { pageNum: string; pageSize: string };
pageNum = +pageNum;
pageSize = +pageSize;
if (!authorization) {
throw new Error('缺少登录凭证');
}
const userId = await authToken(authorization);
await connectToDatabase();
// 根据 id 获取用户账单
const bills = await Bill.find<BillSchema>({
userId
})
.sort({ createdAt: -1 }) // 按照创建时间倒序排列
.skip((pageNum - 1) * pageSize)
.limit(pageSize);
// 获取total
const total = await Bill.countDocuments({
userId
});
jsonRes(res, {
data: {
pageNum,
pageSize,
data: bills,
total
}
});
} catch (err) {
jsonRes(res, {
code: 500,
error: err
});
}
}

View File

@@ -85,7 +85,7 @@ const SlideBar = ({
: {})}
onClick={() => {
if (item.chatId === chatId) return;
router.push(`/chat?chatId=${item.chatId}`);
router.replace(`/chat?chatId=${item.chatId}`);
onClose();
}}
>
@@ -187,7 +187,7 @@ const SlideBar = ({
: {})}
onClick={async () => {
if (item.name === name) return;
router.push(`/chat?chatId=${await getChatSiteId(item._id)}`);
router.replace(`/chat?chatId=${await getChatSiteId(item._id)}`);
onClose();
}}
>

View File

@@ -152,7 +152,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
const resetChat = useCallback(async () => {
if (!chatData) return;
try {
router.push(`/chat?chatId=${await getChatSiteId(chatData.modelId)}`);
router.replace(`/chat?chatId=${await getChatSiteId(chatData.modelId)}`);
} catch (error: any) {
toast({
title: error?.message || '生成新对话失败',

View File

@@ -18,11 +18,13 @@ import {
import { DeleteIcon } from '@chakra-ui/icons';
import { useForm, useFieldArray } from 'react-hook-form';
import { UserUpdateParams } from '@/types/user';
import { putUserInfo } from '@/api/user';
import { putUserInfo, getUserBills } from '@/api/user';
import { useToast } from '@/hooks/useToast';
import { useGlobalStore } from '@/store/global';
import { useUserStore } from '@/store/user';
import { UserType } from '@/types/user';
import { usePaging } from '@/hooks/usePaging';
import type { UserBillType } from '@/types/user';
const NumberSetting = () => {
const { userInfo, updateUserInfo } = useUserStore();
@@ -39,7 +41,11 @@ const NumberSetting = () => {
control,
name: 'accounts'
});
const { setPageNum, data: bills } = usePaging<UserBillType>({
api: getUserBills,
pageSize: 20
});
console.log(bills);
const onclickSave = useCallback(
async (data: UserUpdateParams) => {
setLoading(true);
@@ -156,41 +162,19 @@ const NumberSetting = () => {
<Table>
<Thead>
<Tr>
<Th></Th>
<Th></Th>
<Th></Th>
<Th></Th>
<Th></Th>
<Th></Th>
</Tr>
</Thead>
<Tbody>
{accounts.map((item, i) => (
<Tbody fontSize={'sm'}>
{bills.map((item) => (
<Tr key={item.id}>
<Td minW={'200px'}>
<Select
{...register(`accounts.${i}.type`, {
required: '类型不能为空'
})}
>
<option value="openai">openai</option>
</Select>
</Td>
<Td minW={'200px'}>{item.time}</Td>
<Td minW={'200px'} whiteSpace="pre-wrap" wordBreak={'break-all'}>
<Input
{...register(`accounts.${i}.value`, {
required: '账号不能为空'
})}
></Input>
</Td>
<Td>
<IconButton
aria-label="删除账号"
icon={<DeleteIcon />}
colorScheme={'red'}
onClick={() => {
removeAccount(i);
handleSubmit(onclickSave)();
}}
/>
{item.textLen}
</Td>
<Td>{item.price}</Td>
</Tr>
))}
</Tbody>