OpenFrontIO/src/server/Server.ts

import express, { json } from 'express';
import http from 'http';
import { WebSocketServer } from 'ws';
import path from 'path';
import { fileURLToPath } from 'url';
import { GameManager } from './GameManager';
import { ClientMessage, ClientMessageSchema, GameRecord, GameRecordSchema, Turn, TurnSchema } from '../core/Schemas';
import { getConfig } from '../core/configuration/Config';
import { LogSeverity, slog } from './StructuredLog';
import { Client } from './Client';
import { GamePhase, GameServer } from './GameServer';
import { archive } from './Archive';

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

const app = express();
const server = http.createServer(app);
const wss = new WebSocketServer({ server });

// Serve static files from the 'out' directory
app.use(express.static(path.join(__dirname, '../../out')));
app.use(express.json())

const gm = new GameManager(getConfig())

// New GET endpoint to list lobbies
app.get('/lobbies', (req, res) => {
    const now = Date.now()
    res.json({
        lobbies: gm.gamesByPhase(GamePhase.Lobby)
            .filter(g => g.isPublic)
            .map(g => ({ id: g.id, msUntilStart: g.startTime() - now, numClients: g.numClients() }))
            .sort((a, b) => a.msUntilStart - b.msUntilStart),
    });
});

app.post('/private_lobby', (req, res) => {
    const id = gm.createPrivateGame()
    console.log('creating private lobby with id ${id}')
    res.json({
        id: id
    });
});

app.post('/archive_singleplayer_game', (req, res) => {
    try {
        const gameRecord: GameRecord = req.body
        const clientIP = req.ip || req.socket.remoteAddress || 'unknown';  // Added this line

        if (!gameRecord) {
            console.log('game record not found in request')
            res.status(404).json({ error: 'Game record not found' });
            return;
        }
        gameRecord.players.forEach(p => p.ip = clientIP)
        GameRecordSchema.parse(gameRecord);
        archive(gameRecord)
        res.json({
            success: true,
        });
    } catch (error) {
        slog('complete_single_player_game_record', 'Failed to complete game record', { error }, LogSeverity.ERROR);
        res.status(400).json({ error: 'Invalid game record format' });
    }
})

app.post('/start_private_lobby/:id', (req, res) => {
    console.log(`starting private lobby with id ${req.params.id}`)
    gm.startPrivateGame(req.params.id)
});

app.put('/private_lobby/:id', (req, res) => {
    const lobbyID = req.params.id
    gm.updateGameConfig(lobbyID, { gameMap: req.body.gameMap, difficulty: req.body.difficulty })
});

app.get('/lobby/:id/exists', (req, res) => {
    const lobbyId = req.params.id;
    console.log(`checking lobby ${lobbyId} exists`)
    const lobbyExists = gm.hasActiveGame(lobbyId);

    res.json({
        exists: lobbyExists
    });
});


app.get('/private_lobby/:id', (req, res) => {
    res.json({
        hi: '5'
    });
});

wss.on('connection', (ws, req) => {
    ws.on('message', (message: string) => {
        const clientMsg: ClientMessage = ClientMessageSchema.parse(JSON.parse(message))
        slog('websocket_msg', 'server received websocket message', clientMsg, LogSeverity.DEBUG)
        if (clientMsg.type == "join") {
            const forwarded = req.headers['x-forwarded-for']
            const ip = Array.isArray(forwarded)
                ? forwarded[0]  // Get the first IP if it's an array
                : forwarded || req.socket.remoteAddress;

            gm.addClient(new Client(clientMsg.clientID, ip, ws), clientMsg.gameID, clientMsg.lastTurn)
        }
        // TODO: send error message
    })
});

function runGame() {
    setInterval(() => tick(), 1000);
}

function tick() {
    gm.tick()
}

const PORT = process.env.PORT || 3000;
console.log(`Server will try to run on http://localhost:${PORT}`);

server.listen(PORT, () => {
    console.log(`Server is running on http://localhost:${PORT}`);
});

runGame()