35 lines
958 B
JavaScript
35 lines
958 B
JavaScript
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;
|
|
};
|