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

module.exports = sequelize => {
  const FriendRequest = sequelize.define(
    'FriendRequest',
    {
      id: {
        type: DataTypes.INTEGER,
        autoIncrement: true,
        primaryKey: true,
      },
      status: {
        type: DataTypes.STRING,
        defaultValue: 'PENDING',
        allowNull: false,
      },
      // Foreign keys 'fromUserId' and 'toUserId' are added via associations
    },
    {
      // Uses default `createdAt` and `updatedAt` timestamps from Sequelize
      tableName: 'friend_requests',
    }
  );

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

  return FriendRequest;
};