import { NextAPI } from '@/service/middleware/entry';
import { mongoSessionRun } from '@fastgpt/service/common/mongo/sessionRun';
import { authCert } from '@fastgpt/service/support/permission/auth/common';
import { MongoUser } from '@fastgpt/service/support/user/schema';
import { MongoTeamMember } from '@fastgpt/service/support/user/team/teamMemberSchema';
import { NextApiRequest, NextApiResponse } from 'next';

/* 
  简单版迁移：直接升级到最新镜像，会去除 MongoDatasetData 里的索引。直接执行这个脚本。
  无缝迁移：
    1. 移动 User 表中的 avatar 字段到 TeamMember 表中。
*/
async function handler(req: NextApiRequest, res: NextApiResponse) {
  await authCert({ req, authRoot: true });
  await moveUserAvatar();
  return { success: true };
}

export default NextAPI(handler);

const moveUserAvatar = async () => {
  try {
    const users = await MongoUser.find({}, '_id avatar');
    console.log('Total users:', users.length);
    let success = 0;
    for await (const user of users) {
      // @ts-ignore
      if (!user.avatar) continue;
      try {
        await mongoSessionRun(async (session) => {
          await MongoTeamMember.updateOne(
            {
              userId: user._id
            },
            {
              $set: {
                avatar: (user as any).avatar // 删除 avatar 字段, 因为 Type 改了，所以这里不能直接写 user.avatar
              }
            },
            { session }
          );
          // @ts-ignore
          user.avatar = undefined;
          await user.save({ session });
        });
        success++;
        console.log('Move avatar success:', success);
      } catch (error) {
        console.error(error);
      }
    }
  } catch (error) {
    console.error(error);
  }
};