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

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

    const db = mongoose.connection.db;
    const result = await db.collection('users').updateOne(
      { userId: '00000001' },
      { $set: { isVip: true, vipExpireAt: new Date('2026-12-31') } }
    );
    console.log('更新结果:', result.modifiedCount);

    const user = await db.collection('users').findOne({ userId: '00000001' });
    console.log('用户信息:', JSON.stringify({
      nickname: user.nickname,
      userId: user.userId,
      isVip: user.isVip,
      vipExpireAt: user.vipExpireAt
    }, null, 2));

    process.exit(0);
  } catch (err) {
    console.error('错误:', err);
    process.exit(1);
  }
}

setVip();