File size: 1,518 Bytes
4888678 |
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 |
const jwt = require('jsonwebtoken');
const User = require('../models/User');
// Socket authentication middleware
const authenticateSocket = async (socket, next) => {
try {
const token = socket.handshake.auth.token;
if (!token) {
return next(new Error('Authentication error'));
}
const decoded = jwt.verify(token, process.env.JWT_SECRET);
const user = await User.findById(decoded.userId);
if (!user) {
return next(new Error('User not found'));
}
socket.userId = user._id.toString();
socket.user = user;
next();
} catch (error) {
next(new Error('Authentication error'));
}
};
const initializeSocket = (io) => {
// Authentication middleware
io.use(authenticateSocket);
io.on('connection', (socket) => {
console.log(`User ${socket.user.username} connected`);
// Join user to their personal room
socket.join(`user_${socket.userId}`);
// Handle typing events
socket.on('typing', (data) => {
socket.to(`user_${socket.userId}`).emit('userTyping', {
conversationId: data.conversationId,
isTyping: data.isTyping
});
});
// Handle disconnection
socket.on('disconnect', () => {
console.log(`User ${socket.user.username} disconnected`);
});
// Update user's last seen
socket.user.lastSeen = new Date();
socket.user.save().catch(console.error);
});
return io;
};
module.exports = {
initializeSocket
}; |