server.js

const WebSocket = require('ws');
const jwt = require('jsonwebtoken');
const { text } = require('express');
require('dotenv').config();

const PORT = process.env.PORT || 5000;
const wss = new WebSocket.Server({ port: PORT });
const clients = new Map();
const userPastes = {};

wss.on('connection', (ws, req) => {
    console.log('Client connected!');

    const urlParams = new URLSearchParams(req.url.slice(1));
    const token = urlParams.get('token');

    if (!token) {
        ws.send(JSON.stringify({ status: 1, msg: 'ERROR: missing token' }));
        ws.close();
        return;
    }

    jwt.verify(token, process.env.JWT_SECRET, (err, decoded) => {
        if (err) {
            console.log('Invalid JWT: ', err.message);
            ws.send(JSON.stringify({ status: 1, msg: 'Error invalid or expired token' }));
            ws.close();
            return;
        }

        const userId = decoded.id;
        console.log('Authenticated user: ', userId);

        clients.set(ws, userId);

        ws.send(JSON.stringify({
            type: 'init',
            text: userPastes[userId] || ''
        }));

        ws.on('message', (msg) => {
            console.log(`Message from ${userId}:`, msg.toString());

            let parsed;
            try {
                parsed = JSON.parse(msg);
            } catch {
                ws.send(JSON.stringify({ type: 'error', message: 'Invalid message' }));
                return;
            }

            if (parsed.type == 'update') {
                userPastes[userId] = parsed.text;

                for (const [client, id] of clients.entries()) {
                    if (id === userId && client.readyState === WebSocket.OPEN) {
                        client.send(JSON.stringify({
                            type: 'update',
                            text: parsed.text
                        }));
                    }
                }
            }
        });

        ws.on('close', () => {
            console.log(`User ${userId} disconnected`);
            clients.delete(ws);
        })
    });
});