require('dotenv').config();
const pool = require('../src/config/database');

async function addLastLoginField() {
  let connection;
  
  try {
    connection = await pool.getConnection();
    console.log('开始添加 last_login 字段...\n');

    // 检查字段是否已存在
    const [columns] = await connection.query(`
      SHOW COLUMNS FROM users LIKE 'last_login'
    `);

    if (columns && columns.length > 0) {
      console.log('last_login 字段已存在，无需添加');
      return;
    }

    // 添加 last_login 字段
    await connection.query(`
      ALTER TABLE users 
      ADD COLUMN last_login TIMESTAMP NULL AFTER status
    `);
    
    console.log('✓ last_login 字段添加成功');
    console.log('\n数据库更新完成！');
  } catch (error) {
    console.error('添加字段失败:', error);
    process.exit(1);
  } finally {
    if (connection) {
      await connection.release();
    }
    process.exit(0);
  }
}

addLastLoginField();