gorbagana-mini-games/server.js at main · SUNDRAM07/gorbagana-mini-games · GitHub

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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import WebSocket, { WebSocketServer } from 'ws';

const server = new WebSocketServer({ port: 8080 });

const users = new Map(); // userId -> { ws, userData }
const usersByInviteCode = new Map(); // inviteCode -> userId

server.on('connection', (ws) => {
    console.log('New client connected');

    ws.on('message', (message) => {
        const data = JSON.parse(message);

        switch (data.type) {
            case 'register':
                users.set(data.user.id, { ws, userData: data.user });
                usersByInviteCode.set(data.user.inviteCode, data.user.id);

                // Broadcast online users
                broadcastOnlineUsers();
                break;

            case 'sendFriendRequest':
                const targetUserId = usersByInviteCode.get(data.targetInviteCode);
                if (targetUserId && users.has(targetUserId)) {
                    const targetUser = users.get(targetUserId);
                    targetUser.ws.send(JSON.stringify({
                        type: 'friendRequest',
                        requestId: 'req_' + Math.random().toString(36).substr(2, 9),
                        from: data.from
                    }));
                }
                break;

            case 'acceptFriendRequest':
                if (users.has(data.targetUserId)) {
                    const targetUser = users.get(data.targetUserId);
                    targetUser.ws.send(JSON.stringify({
                        type: 'friendRequestAccepted',
                        user: data.user
                    }));
                }
                break;

            case 'rejectFriendRequest':
                if (users.has(data.targetUserId)) {
                    const targetUser = users.get(data.targetUserId);
                    targetUser.ws.send(JSON.stringify({
                        type: 'friendRequestRejected'
                    }));
                }
                break;

            case 'sendGameInvite':
                if (users.has(data.targetUserId)) {
                    const targetUser = users.get(data.targetUserId);
                    targetUser.ws.send(JSON.stringify({
                        type: 'gameInvite',
                        gameType: data.gameType,
                        stake: data.stake,
                        from: data.from
                    }));
                }
                break;

            case 'acceptGameInvite':
                if (users.has(data.targetUserId)) {
                    const targetUser = users.get(data.targetUserId);
                    targetUser.ws.send(JSON.stringify({
                        type: 'gameInviteAccepted',
                        roomId: 'room_' + Math.random().toString(36).substr(2, 9),
                        gameType: data.gameType,
                        stake: data.stake,
                        user: data.user
                    }));
                }
                break;

            case 'rejectGameInvite':
                if (users.has(data.targetUserId)) {
                    const targetUser = users.get(data.targetUserId);
                    targetUser.ws.send(JSON.stringify({
                        type: 'gameInviteRejected',
                        user: data.user
                    }));
                }
                break;
        }
    });

    ws.on('close', () => {
        // Remove user from online users
        for (const [userId, userData] of users.entries()) {
            if (userData.ws === ws) {
                users.delete(userId);
                usersByInviteCode.delete(userData.userData.inviteCode);
                broadcastUserOffline(userId);
                break;
            }
        }
    });
});

function broadcastOnlineUsers() {
    const onlineUsers = {};
    for (const [userId, userData] of users.entries()) {
        onlineUsers[userId] = userData.userData;
    }

    for (const [userId, userData] of users.entries()) {
        userData.ws.send(JSON.stringify({
            type: 'onlineUsers',
            users: onlineUsers
        }));
    }
}

function broadcastUserOffline(userId) {
    for (const [_, userData] of users.entries()) {
        userData.ws.send(JSON.stringify({
            type: 'userOffline',
            userId: userId
        }));
    }
}

console.log('WebSocket server running on port 8080');