85 lines
2.1 KiB
TypeScript
85 lines
2.1 KiB
TypeScript
import type { NextApiRequest, NextApiResponse } from 'next';
|
|
import { jsonRes } from '@/service/response';
|
|
import { connectToDatabase, User } from '@/service/mongo';
|
|
import { authUser } from '@/service/utils/auth';
|
|
import { PgClient } from '@/service/pg';
|
|
import { PgTrainingTableName } from '@/constants/plugin';
|
|
|
|
export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
|
|
try {
|
|
let { kbId } = req.query as {
|
|
kbId: string;
|
|
};
|
|
|
|
if (!kbId) {
|
|
throw new Error('缺少参数');
|
|
}
|
|
|
|
await connectToDatabase();
|
|
|
|
// 凭证校验
|
|
const { userId } = await authUser({ req, authToken: true });
|
|
|
|
const thirtyMinutesAgo = new Date(Date.now() - 30 * 60 * 1000);
|
|
|
|
// auth export times
|
|
const authTimes = await User.findOne(
|
|
{
|
|
_id: userId,
|
|
$or: [
|
|
{ 'limit.exportKbTime': { $exists: false } },
|
|
{ 'limit.exportKbTime': { $lte: thirtyMinutesAgo } }
|
|
]
|
|
},
|
|
'_id limit'
|
|
);
|
|
|
|
if (!authTimes) {
|
|
throw new Error('上次导出未到半小时，每半小时仅可导出一次。');
|
|
}
|
|
|
|
// 统计数据
|
|
const count = await PgClient.count(PgTrainingTableName, {
|
|
where: [['kb_id', kbId], 'AND', ['user_id', userId]]
|
|
});
|
|
// 从 pg 中获取所有数据
|
|
const pgData = await PgClient.select<{ q: string; a: string; source: string }>(
|
|
PgTrainingTableName,
|
|
{
|
|
where: [['kb_id', kbId], 'AND', ['user_id', userId]],
|
|
fields: ['q', 'a', 'source'],
|
|
order: [{ field: 'id', mode: 'DESC' }],
|
|
limit: count
|
|
}
|
|
);
|
|
|
|
const data: [string, string, string][] = pgData.rows.map((item) => [
|
|
item.q.replace(/\n/g, '\\n'),
|
|
item.a.replace(/\n/g, '\\n'),
|
|
item.source
|
|
]);
|
|
|
|
// update export time
|
|
await User.findByIdAndUpdate(userId, {
|
|
'limit.exportKbTime': new Date()
|
|
});
|
|
|
|
jsonRes(res, {
|
|
data
|
|
});
|
|
} catch (err) {
|
|
jsonRes(res, {
|
|
code: 500,
|
|
error: err
|
|
});
|
|
}
|
|
}
|
|
|
|
export const config = {
|
|
api: {
|
|
bodyParser: {
|
|
sizeLimit: '100mb'
|
|
}
|
|
}
|
|
};
|