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

const Device = sequelize.define('Device', {
  deviceId: {
    type: DataTypes.STRING,
    primaryKey: true,
    allowNull: false,
    unique: true
  },
  name: {
    type: DataTypes.STRING,
    allowNull: true
  },
  type: {
    type: DataTypes.STRING,
    allowNull: true
  },
  model: {
    type: DataTypes.STRING,
    allowNull: true
  },
  serialNumber: {
    type: DataTypes.STRING,
    allowNull: true,
    unique: true
  },
  rackId: {
    type: DataTypes.STRING,
    allowNull: true
  },
  position: {
    type: DataTypes.INTEGER,
    allowNull: true
  },
  height: {
    type: DataTypes.INTEGER,
    allowNull: true,
    defaultValue: 1
  },
  powerConsumption: {
    type: DataTypes.FLOAT,
    allowNull: true,
    defaultValue: 0
  },
  status: {
    type: DataTypes.STRING,
    defaultValue: 'offline'
  },
  isIdle: {
    type: DataTypes.BOOLEAN,
    defaultValue: false
  },
  idleDate: {
    type: DataTypes.DATE,
    allowNull: true
  },
  idleReason: {
    type: DataTypes.STRING,
    allowNull: true
  },
  warehouseId: {
    type: DataTypes.STRING,
    allowNull: true
  },
  sourceType: {
    type: DataTypes.ENUM('rack', 'warehouse'),
    defaultValue: 'rack'
  },
  purchaseDate: {
    type: DataTypes.DATE,
    allowNull: true
  },
  warrantyExpiry: {
    type: DataTypes.DATE,
    allowNull: true
  },
  ipAddress: {
    type: DataTypes.STRING,
    allowNull: true
  },
  description: {
    type: DataTypes.TEXT,
    allowNull: true
  },
  customFields: {
    type: DataTypes.JSON,
    defaultValue: {},
    allowNull: true
  }
}, {
  tableName: 'devices',
  timestamps: true,
  indexes: [
    { fields: ['status'] },
    { fields: ['type'] },
    { fields: ['rackId'] },
    { fields: ['createdAt'] },
    { fields: ['status', 'type'] },
    { fields: ['name'] }
  ]
});

Device.belongsTo(Rack, { foreignKey: 'rackId' });
Device.belongsTo(Warehouse, { foreignKey: 'warehouseId' });

module.exports = Device;