const { DataTypes } = require('sequelize');
const { sequelize } = require('../db');
const User = require('./User');

const InventoryPlan = sequelize.define('InventoryPlan', {
  planId: {
    type: DataTypes.STRING,
    primaryKey: true,
    allowNull: false,
    unique: true
  },
  name: {
    type: DataTypes.STRING,
    allowNull: false
  },
  type: {
    type: DataTypes.STRING,
    allowNull: false,
    defaultValue: 'full',
    comment: 'full:全面盘点, partial:局部盘点, sample:抽样盘点'
  },
  description: {
    type: DataTypes.TEXT,
    allowNull: true
  },
  status: {
    type: DataTypes.STRING,
    defaultValue: 'draft',
    comment: 'draft:草稿, pending:待执行, in_progress:进行中, completed:已完成, cancelled:已取消'
  },
  scheduledDate: {
    type: DataTypes.DATE,
    allowNull: true
  },
  completedDate: {
    type: DataTypes.DATE,
    allowNull: true
  },
  targetRooms: {
    type: DataTypes.JSON,
    defaultValue: [],
    comment: '目标机房ID列表'
  },
  targetRacks: {
    type: DataTypes.JSON,
    defaultValue: [],
    comment: '目标机柜ID列表'
  },
  totalDevices: {
    type: DataTypes.INTEGER,
    defaultValue: 0,
    comment: '盘点设备总数'
  },
  checkedDevices: {
    type: DataTypes.INTEGER,
    defaultValue: 0,
    comment: '已盘点设备数'
  },
  normalDevices: {
    type: DataTypes.INTEGER,
    defaultValue: 0,
    comment: '正常设备数'
  },
  abnormalDevices: {
    type: DataTypes.INTEGER,
    defaultValue: 0,
    comment: '异常设备数'
  },
  missedDevices: {
    type: DataTypes.INTEGER,
    defaultValue: 0,
    comment: '漏盘设备数'
  },
  extraDevices: {
    type: DataTypes.INTEGER,
    defaultValue: 0,
    comment: '多出设备数'
  },
  createdBy: {
    type: DataTypes.STRING,
    allowNull: true,
    references: {
      model: User,
      key: 'userId'
    }
  },
  remark: {
    type: DataTypes.TEXT,
    allowNull: true
  }
}, {
  tableName: 'inventory_plans',
  timestamps: true,
  indexes: [
    { fields: ['status'] },
    { fields: ['scheduledDate'] },
    { fields: ['createdAt'] }
  ]
});

InventoryPlan.belongsTo(User, { foreignKey: 'createdBy', as: 'Creator' });
User.hasMany(InventoryPlan, { foreignKey: 'createdBy' });

module.exports = InventoryPlan;