2026-04-12 17:17:18 +00:00

354 lines
6.0 KiB
JavaScript

const config = require('../../config');
const providers = config.providers;
const crypto = require('crypto');
const bcrypt = require('bcrypt');
const moment = require('moment');
module.exports = function(sequelize, DataTypes) {
const users = sequelize.define(
'users',
{
id: {
type: DataTypes.UUID,
defaultValue: DataTypes.UUIDV4,
primaryKey: true,
},
firstName: {
type: DataTypes.TEXT,
},
lastName: {
type: DataTypes.TEXT,
},
phoneNumber: {
type: DataTypes.TEXT,
},
email: {
type: DataTypes.TEXT,
},
disabled: {
type: DataTypes.BOOLEAN,
allowNull: false,
defaultValue: false,
},
password: {
type: DataTypes.TEXT,
},
emailVerified: {
type: DataTypes.BOOLEAN,
allowNull: false,
defaultValue: false,
},
emailVerificationToken: {
type: DataTypes.TEXT,
},
emailVerificationTokenExpiresAt: {
type: DataTypes.DATE,
},
passwordResetToken: {
type: DataTypes.TEXT,
},
passwordResetTokenExpiresAt: {
type: DataTypes.DATE,
},
provider: {
type: DataTypes.TEXT,
},
importHash: {
type: DataTypes.STRING(255),
allowNull: true,
unique: true,
},
},
{
timestamps: true,
paranoid: true,
freezeTableName: true,
},
);
users.associate = (db) => {
db.users.belongsToMany(db.permissions, {
as: 'custom_permissions',
foreignKey: {
name: 'users_custom_permissionsId',
},
constraints: false,
through: 'usersCustom_permissionsPermissions',
});
db.users.belongsToMany(db.permissions, {
as: 'custom_permissions_filter',
foreignKey: {
name: 'users_custom_permissionsId',
},
constraints: false,
through: 'usersCustom_permissionsPermissions',
});
/// loop through entities and it's fields, and if ref === current e[name] and create relation has many on parent entity
db.users.hasMany(db.workspaces, {
as: 'workspaces_owner_user',
foreignKey: {
name: 'owner_userId',
},
constraints: false,
});
db.users.hasMany(db.workspace_memberships, {
as: 'workspace_memberships_member_user',
foreignKey: {
name: 'member_userId',
},
constraints: false,
});
db.users.hasMany(db.projects, {
as: 'projects_created_user',
foreignKey: {
name: 'created_userId',
},
constraints: false,
});
db.users.hasMany(db.project_collaborations, {
as: 'project_collaborations_collaborator_user',
foreignKey: {
name: 'collaborator_userId',
},
constraints: false,
});
db.users.hasMany(db.assets, {
as: 'assets_uploaded_user',
foreignKey: {
name: 'uploaded_userId',
},
constraints: false,
});
db.users.hasMany(db.recording_sessions, {
as: 'recording_sessions_started_user',
foreignKey: {
name: 'started_userId',
},
constraints: false,
});
db.users.hasMany(db.generation_requests, {
as: 'generation_requests_requested_user',
foreignKey: {
name: 'requested_userId',
},
constraints: false,
});
db.users.hasMany(db.mix_sessions, {
as: 'mix_sessions_requested_user',
foreignKey: {
name: 'requested_userId',
},
constraints: false,
});
db.users.hasMany(db.mastering_sessions, {
as: 'mastering_sessions_requested_user',
foreignKey: {
name: 'requested_userId',
},
constraints: false,
});
db.users.hasMany(db.exports, {
as: 'exports_requested_user',
foreignKey: {
name: 'requested_userId',
},
constraints: false,
});
db.users.hasMany(db.marketplace_listings, {
as: 'marketplace_listings_seller_user',
foreignKey: {
name: 'seller_userId',
},
constraints: false,
});
db.users.hasMany(db.marketplace_orders, {
as: 'marketplace_orders_buyer_user',
foreignKey: {
name: 'buyer_userId',
},
constraints: false,
});
//end loop
db.users.belongsTo(db.roles, {
as: 'app_role',
foreignKey: {
name: 'app_roleId',
},
constraints: false,
});
db.users.belongsTo(db.organizations, {
as: 'organizations',
foreignKey: {
name: 'organizationsId',
},
constraints: false,
});
db.users.hasMany(db.file, {
as: 'avatar',
foreignKey: 'belongsToId',
constraints: false,
scope: {
belongsTo: db.users.getTableName(),
belongsToColumn: 'avatar',
},
});
db.users.belongsTo(db.users, {
as: 'createdBy',
});
db.users.belongsTo(db.users, {
as: 'updatedBy',
});
};
users.beforeCreate((users, options) => {
users = trimStringFields(users);
if (users.provider !== providers.LOCAL && Object.values(providers).indexOf(users.provider) > -1) {
users.emailVerified = true;
if (!users.password) {
const password = crypto
.randomBytes(20)
.toString('hex');
const hashedPassword = bcrypt.hashSync(
password,
config.bcrypt.saltRounds,
);
users.password = hashedPassword
}
}
});
users.beforeUpdate((users, options) => {
users = trimStringFields(users);
});
return users;
};
function trimStringFields(users) {
users.email = users.email.trim();
users.firstName = users.firstName
? users.firstName.trim()
: null;
users.lastName = users.lastName
? users.lastName.trim()
: null;
return users;
}