diff --git a/backend/models/User.js b/backend/models/User.js index eb506ed5..be084301 100644 --- a/backend/models/User.js +++ b/backend/models/User.js @@ -31,4 +31,12 @@ UserSchema.methods.comparePassword = async function (enteredPassword) { return bcrypt.compare(enteredPassword, this.password); }; +UserSchema.methods.toSafeObject = function () { + return { + id: this._id, + username: this.username, + email: this.email, + }; +}; + module.exports = mongoose.model("User", UserSchema); \ No newline at end of file diff --git a/backend/routes/auth.js b/backend/routes/auth.js index 7c2cda78..d6162b73 100644 --- a/backend/routes/auth.js +++ b/backend/routes/auth.js @@ -32,7 +32,7 @@ router.post("/signup", validateRequest(signupSchema), async (req, res) => { // Login route router.post("/login", validateRequest(loginSchema), passport.authenticate('local'), (req, res) => { - res.status(200).json( { message: 'Login successful', user: req.user } ); + res.status(200).json({ message: 'Login successful', user: req.user.toSafeObject() }); }); // Logout route