export default function (sequelize, DataTypes) {
  const Malware = sequelize.define('malware', {
    fileId: { type: DataTypes.STRING(12), primaryKey: true }, // eslint-disable-line new-cap
    positives: DataTypes.INTEGER,
    virustotal: DataTypes.JSON,
  });

  Malware.associate = function associate(models) {
    Malware.belongsTo(models.file);
  };

  return Malware;
}