/**
 * User 模型定义
 * @file User.js
 * @description 定义用户模型，用于数据库操作
 */
const { DataTypes } = require('sequelize');
const bcrypt = require('bcrypt');
const BaseModel = require('./BaseModel');
const { sequelize } = require('../config/database-pool');

class User extends BaseModel {
  /**
   * 验证密码
   * @param {String} password 待验证的密码
   * @returns {Promise<Boolean>} 验证结果
   */
  async validPassword(password) {
    return await bcrypt.compare(password, this.password);
  }

  /**
   * 获取用户角色
   * @returns {Promise<Array>} 用户角色列表
   */
  async getRoles() {
    return await this.getRoles();
  }

  /**
   * 检查用户是否具有指定角色
   * @param {String|Array} roleName 角色名称或角色名称数组
   * @returns {Promise<Boolean>} 检查结果
   */
  async hasRole(roleName) {
    const roles = await this.getRoles();
    const roleNames = roles.map(role => role.name);
    
    if (Array.isArray(roleName)) {
      return roleName.some(name => roleNames.includes(name));
    }
    
    return roleNames.includes(roleName);
  }

  /**
   * 为用户分配角色
   * @param {Number|Array} roleId 角色ID或角色ID数组
   * @returns {Promise<Boolean>} 分配结果
   */
  async assignRole(roleId) {
    try {
      if (Array.isArray(roleId)) {
        await this.addRoles(roleId);
      } else {
        await this.addRole(roleId);
      }
      return true;
    } catch (error) {
      console.error('分配角色失败:', error);
      return false;
    }
  }

  /**
   * 移除用户角色
   * @param {Number|Array} roleId 角色ID或角色ID数组
   * @returns {Promise<Boolean>} 移除结果
   */
  async removeRole(roleId) {
    try {
      if (Array.isArray(roleId)) {
        await this.removeRoles(roleId);
      } else {
        await this.removeRole(roleId);
      }
      return true;
    } catch (error) {
      console.error('移除角色失败:', error);
      return false;
    }
  }

  /**
   * 获取用户安全信息（不包含密码）
   * @returns {Object} 用户安全信息
   */
  getSafeInfo() {
    const { password, ...safeInfo } = this.get({ plain: true });
    return safeInfo;
  }
}

// 初始化User模型
User.init({
  id: {
    type: DataTypes.INTEGER,
    primaryKey: true,
    autoIncrement: true
  },
  username: {
    type: DataTypes.STRING(50),
    allowNull: false,
    unique: true
  },
  email: {
    type: DataTypes.STRING(100),
    allowNull: false,
    unique: true,
    validate: {
      isEmail: true
    }
  },
  password: {
    type: DataTypes.STRING(255),
    allowNull: false
  },
  phone: {
    type: DataTypes.STRING(20),
    allowNull: true
  },
  avatar: {
    type: DataTypes.STRING(255),
    allowNull: true
  },
  status: {
    type: DataTypes.ENUM('active', 'inactive', 'suspended'),
    defaultValue: 'active'
  }
}, {
  sequelize,
  tableName: 'users',
  modelName: 'User',
  hooks: {
    beforeCreate: async (user) => {
      if (user.password) {
        user.password = await bcrypt.hash(user.password, 10);
      }
    },
    beforeUpdate: async (user) => {
      if (user.changed('password')) {
        user.password = await bcrypt.hash(user.password, 10);
      }
    }
  }
});

module.exports = User;