122 lines
4 KiB
JavaScript
122 lines
4 KiB
JavaScript
const express = require('express');
|
|
const http = require('http');
|
|
const { Server } = require('socket.io');
|
|
const { Chess } = require('chess.js');
|
|
const { v4: uuidv4 } = require('uuid');
|
|
|
|
const app = express();
|
|
const server = http.createServer(app);
|
|
const io = new Server(server, { cors: { origin: '*' } });
|
|
|
|
app.use(express.json());
|
|
app.use(express.static('public'));
|
|
|
|
// In-memory rooms store. For production replace with persistent store.
|
|
// room = { id, chess: Chess, players: [{id, socketId, color}], status }
|
|
const rooms = new Map();
|
|
|
|
app.post('/rooms', (req, res) => {
|
|
const id = uuidv4().slice(0, 8);
|
|
const chess = new Chess();
|
|
rooms.set(id, { id, chess, players: [], status: 'waiting' });
|
|
res.json({ roomId: id });
|
|
});
|
|
|
|
app.get('/rooms/:id', (req, res) => {
|
|
const room = rooms.get(req.params.id);
|
|
if (!room) return res.status(404).json({ error: 'room not found' });
|
|
res.json({ id: room.id, fen: room.chess.fen(), players: room.players.map(p => ({ id: p.id, color: p.color })), status: room.status });
|
|
});
|
|
|
|
io.on('connection', (socket) => {
|
|
console.log('socket connected', socket.id);
|
|
|
|
socket.on('room:join', ({ roomId, playerId }, cb) => {
|
|
const room = rooms.get(roomId);
|
|
if (!room) return cb && cb({ error: 'room not found' });
|
|
if (room.players.length >= 2) return cb && cb({ error: 'room full' });
|
|
|
|
const assignedId = playerId || uuidv4();
|
|
const color = room.players.length === 0 ? 'white' : 'black';
|
|
room.players.push({ id: assignedId, socketId: socket.id, color });
|
|
socket.join(roomId);
|
|
socket.data.roomId = roomId;
|
|
socket.data.playerId = assignedId;
|
|
|
|
|
|
io.to(roomId).emit('room:update', {
|
|
fen: room.chess.fen(),
|
|
players: room.players.map(p => ({ id: p.id, color: p.color })),
|
|
status: room.status
|
|
});
|
|
|
|
cb && cb({ ok: true, color, roomId, playerId: assignedId });
|
|
});
|
|
|
|
socket.on('game:move', ({ roomId, from, to, promotion }, cb) => {
|
|
const room = rooms.get(roomId);
|
|
if (!room) return cb && cb({ error: 'room not found' });
|
|
|
|
const chess = room.chess;
|
|
|
|
// Validate move
|
|
const move = chess.move({ from, to, promotion });
|
|
if (!move) return cb && cb({ error: 'invalid move' });
|
|
|
|
io.to(roomId).emit('move:moved', { move, fen: chess.fen() });
|
|
|
|
if (chess.isGameOver()) {
|
|
room.status = 'finished';
|
|
let result = 'draw';
|
|
if (chess.isCheckmate()) result = 'checkmate';
|
|
else if (chess.isStalemate()) result = 'stalemate';
|
|
io.to(roomId).emit('game:over', { result, fen: chess.fen() });
|
|
}
|
|
|
|
cb && cb({ ok: true, move, fen: chess.fen() });
|
|
});
|
|
|
|
// Host can start the game explicitly
|
|
socket.on('game:start', ({ roomId }, cb) => {
|
|
const room = rooms.get(roomId);
|
|
if (!room) return cb && cb({ error: 'room not found' });
|
|
|
|
// verify sender is the host (first player)
|
|
const senderId = socket.data.playerId;
|
|
if (!senderId) return cb && cb({ error: 'not joined' });
|
|
if (room.players.length < 2) return cb && cb({ error: 'need 2 players to start' });
|
|
const host = room.players[0];
|
|
if (!host || host.id !== senderId) return cb && cb({ error: 'only host can start' });
|
|
|
|
room.status = 'playing';
|
|
io.to(roomId).emit('game:started', { roomId, fen: room.chess.fen() });
|
|
io.to(roomId).emit('room:update', {
|
|
fen: room.chess.fen(),
|
|
players: room.players.map(p => ({ id: p.id, color: p.color })),
|
|
status: room.status
|
|
});
|
|
|
|
cb && cb({ ok: true });
|
|
});
|
|
|
|
socket.on('disconnect', () => {
|
|
const roomId = socket.data.roomId;
|
|
if (!roomId) return;
|
|
const room = rooms.get(roomId);
|
|
if (!room) return;
|
|
|
|
room.players = room.players.filter(p => p.socketId !== socket.id);
|
|
if (room.players.length < 2 && room.status === 'playing') room.status = 'waiting';
|
|
|
|
io.to(roomId).emit('room:update', {
|
|
fen: room.chess.fen(),
|
|
players: room.players.map(p => ({ id: p.id, color: p.color })),
|
|
status: room.status
|
|
});
|
|
});
|
|
});
|
|
|
|
const PORT = process.env.PORT || 3000;
|
|
server.listen(PORT, () => {
|
|
console.log(`ChessNut server listening on port ${PORT}`);
|
|
});
|