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'
    }
  }
};