File size: 1,682 Bytes
11f4e50 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | import mongoose, { Schema, Document } from 'mongoose';
import bcrypt from 'bcryptjs';
export interface IUser extends Document {
email: string;
password: string;
subscription: 'free' | 'starter' | 'creator' | 'studio' | 'custom';
videosGenerated: number;
presets: mongoose.Types.ObjectId[];
role: 'user' | 'admin';
createdAt: Date;
comparePassword(candidatePassword: string): Promise<boolean>;
}
const UserSchema = new Schema<IUser>({
email: {
type: String,
required: true,
unique: true,
lowercase: true,
trim: true,
},
password: {
type: String,
required: true,
minlength: 8,
},
subscription: {
type: String,
enum: ['free', 'starter', 'creator', 'studio', 'custom'],
default: 'free',
},
videosGenerated: {
type: Number,
default: 0,
},
presets: [{
type: Schema.Types.ObjectId,
ref: 'Preset',
}],
role: {
type: String,
enum: ['user', 'admin'],
default: 'user',
},
createdAt: {
type: Date,
default: Date.now,
},
});
UserSchema.pre('save', async function (next) {
if (!this.isModified('password')) return next();
const salt = await bcrypt.genSalt(12);
this.password = await bcrypt.hash(this.password, salt);
next();
});
UserSchema.methods.comparePassword = async function (
candidatePassword: string
): Promise<boolean> {
return bcrypt.compare(candidatePassword, this.password);
};
export const User = mongoose.model<IUser>('User', UserSchema);
|