src/server.js

require('dotenv').config();
const express = require('express');
const path = require('path');
const { WebSocketServer } = require('ws');
const http = require('http');
const Database = require('./database');
const ConnectionManager = require('./websocket');
const { verifyApiKey, requireAdminKey, requireRegularOrAdminKey, requireAnyKey, requireServerKey } = require('./auth');
const createKeysRouter = require('./routes/keys');
const createEventsRouter = require('./routes/events');
const createHealthRouter = require('./routes/health');
const createServerRouter = require('./routes/server');
const createAIRouter = require('./routes/ai');

const HOST = process.env.SERVER_HOST || '0.0.0.0';
const PORT = parseInt(process.env.SERVER_PORT || '8000');

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

const db = new Database(process.env.DATABASE_PATH || 'minecraft_ws.db');
const manager = new ConnectionManager();

app.use(express.json());

// Request Logging Middleware
app.use((req, res, next) => {
    const start = Date.now();
    const timestamp = new Date().toISOString();
    
    // Log request
    console.log(`[${timestamp}] \x1b[36m${req.method}\x1b[0m ${req.url}`);
    
    // Capture response finish
    res.on('finish', () => {
        const duration = Date.now() - start;
        const statusColor = res.statusCode >= 400 ? '\x1b[31m' : (res.statusCode >= 300 ? '\x1b[33m' : '\x1b[32m');
        console.log(`[${timestamp}] ${statusColor}${res.statusCode}\x1b[0m ${req.method} ${req.url} - ${duration}ms`);
    });
    
    next();
});

app.use('/dashboard', express.static(path.join(__dirname, '../dashboard/public')));

app.get('/', (req, res) => {
    res.json({
        message: 'Minecraft WebSocket API Server (Node.js)',
        dashboard: '/dashboard',
        websocket: 'ws://' + req.get('host') + '/ws',
        version: '1.0.0'
    });
});

app.use('/manage/keys', verifyApiKey(db), createKeysRouter(db, requireAdminKey));
app.use('/api/events', verifyApiKey(db), requireAnyKey, createEventsRouter(db, manager, requireAnyKey));
app.use('/api/server', verifyApiKey(db), createServerRouter(db, manager, requireRegularOrAdminKey));
app.use('/api/ai', verifyApiKey(db), createAIRouter(db, manager, requireAdminKey, requireAnyKey));
app.use('/health', createHealthRouter(db, manager));

server.on('upgrade', async (request, socket, head) => {
    const url = new URL(request.url, `http://${request.headers.host}`);

    if (url.pathname === '/ws') {
        const apiKey = url.searchParams.get('api_key');

        if (!apiKey) {
            socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
            socket.destroy();
            return;
        }

        const result = await db.verifyApiKey(apiKey);

        if (!result) {
            socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
            socket.destroy();
            return;
        }

        wss.handleUpgrade(request, socket, head, (ws) => {
            wss.emit('connection', ws, request, result);
        });
    } else {
        socket.destroy();
    }
});

wss.on('connection', (ws, request, apiKeyInfo) => {
    manager.connect(ws, apiKeyInfo.id);

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

            if (msg.type === 'ping') {
                ws.send(JSON.stringify({ type: 'pong' }));
            }
        } catch (error) {
        }
    });

    ws.on('close', () => {
        manager.disconnect(ws);
    });

    ws.on('error', () => {
        manager.disconnect(ws);
    });
});

async function startServer() {
    console.log('🚀 启动Minecraft WebSocket API服务器...');
    console.log('='.repeat(50));

    await db.init();

    const adminKeyInfo = await db.ensureInitialAdminKey();
    const recoveryTokenInfo = await db.ensureAdminRecoveryToken();

    if (adminKeyInfo) {
        console.log('='.repeat(60));
        console.log('重要: 已生成新的Admin Key!');
        console.log(`  名称: ${adminKeyInfo.name}`);
        console.log(`  密钥: ${adminKeyInfo.key}`);
        console.log('请复制并安全保存此密钥。');
        console.log('您需要使用它来管理API密钥。');
        console.log('如果丢失此密钥且没有其他Admin Key，您可能失去管理员访问权限。');
        console.log('='.repeat(60));
    } else {
        console.log('信息: 已找到现有Admin Key或Admin Key检查已执行。');
    }

    if (recoveryTokenInfo) {
        console.log('='.repeat(60));
        console.log('重要: 已生成 Admin 恢复令牌。');
        console.log(`  令牌: ${recoveryTokenInfo.token}`);
        console.log('请安全存储此令牌。它仅显示一次。');
        console.log('='.repeat(60));
    }

    server.listen(PORT, HOST, () => {
        const displayHost = HOST === '0.0.0.0' ? 'localhost' : HOST;
        console.log(`服务器地址: http://${displayHost}:${PORT}/dashboard`);
        console.log(`WebSocket端点: ws://${displayHost}:${PORT}/ws`);
        console.log(`健康检查: http://${displayHost}:${PORT}/health`);
        console.log('='.repeat(50));
        console.log();
        console.log('💡 首次使用提示:');
        console.log('1. 使用CLI工具创建API密钥:');
        console.log('   node cli/cli.js create-key "MyServer"');
        console.log();
        console.log('2. 生成Minecraft插件配置文件:');
        console.log('   node cli/cli.js generate-config');
        console.log();
        console.log('3. 将API密钥配置到Minecraft插件中');
        console.log('='.repeat(50));
    });
}

if (require.main === module) {
    startServer().catch(error => {
        console.error('❌ 启动失败:', error);
        process.exit(1);
    });
}

module.exports = { app, server, db, wss, startServer };