const mongoose = require('mongoose');
require('dotenv').config();

async function assignUserIds() {
  try {
    await mongoose.connect(process.env.MONGODB_URI);
    console.log('数据库已连接');

    const db = mongoose.connection.db;
    const users = await db.collection('users').find({}).toArray();
    console.log(`找到 ${users.length} 个用户`);

    for (let i = 0; i < users.length; i++) {
      const user = users[i];
      const userId = (i + 1).toString().padStart(8, '0');
      
      await db.collection('users').updateOne(
        { _id: user._id },
        { $set: { userId } }
      );
      
      console.log(`已分配: ${user.nickname || '(空昵称)'} -> ${userId}`);
    }

    console.log('全部分配完成！');
    process.exit(0);
  } catch (err) {
    console.error('错误:', err);
    process.exit(1);
  }
}

assignUserIds();