const { DataTypes } = require('sequelize');

module.exports = sequelize => {
  const Group = sequelize.define(
    'Group',
    {
      id: {
        type: DataTypes.INTEGER,
        autoIncrement: true,
        primaryKey: true,
      },
      name: {
        type: DataTypes.STRING,
        allowNull: false,
      },
    },
    {
      timestamps: false,
      tableName: 'groups',
    }
  );

  Group.prototype.toJSON = function () {
    const values = { ...this.get() };
    if (values.id !== undefined) values.id = String(values.id);
    return values;
  };

  return Group;
};