server.js

#!/usr/bin/env node
// server.js — Bridge Server 入口

const http = require('http');
const fs = require('fs');
const path = require('path');
const crypto = require('crypto');

const { ConnectionRegistry } = require('./lib/server/registry');
const { WebSocketHub } = require('./lib/server/ws-hub');
const { Router } = require('./lib/server/router');

// 加载配置
const configPath = path.join(__dirname, 'config.json');
let config;
try {
  config = JSON.parse(fs.readFileSync(configPath, 'utf8'));
} catch (e) {
  console.error('[server] 未找到 config.json，请先复制 config.example.json 并填写配置：');
  console.error('[server]   cp config.example.json config.json');
  process.exit(1);
}

// 自动生成 token（如果未配置）
if (!config.bridge.token) {
  config.bridge.token = crypto.randomBytes(24).toString('hex');
  fs.writeFileSync(configPath, JSON.stringify(config, null, 2) + '\n');
  console.error('[server] 已自动生成 access token 并写入 config.json');
}

// 初始化组件
const registry = new ConnectionRegistry();
const wsHub = new WebSocketHub({
  registry,
  port: config.bridge.port,
  host: config.bridge.host,
  heartbeatInterval: config.bridge.heartbeatInterval,
  heartbeatTimeout: config.bridge.heartbeatTimeout,
  heartbeatMaxFailures: config.bridge.heartbeatMaxFailures,
  maxPayload: config.bridge.maxBodySize,
});
const router = new Router({
  registry,
  wsHub,
  requestTimeout: config.bridge.requestTimeout || 30000,
  token: config.bridge.token,
  maxBodySize: config.bridge.maxBodySize,
});

// 创建 HTTP Server
const httpServer = http.createServer((req, res) => {
  router.handle(req, res);
});

// 将 WebSocket attach 到 HTTP Server（共用端口）
wsHub.attach(httpServer);

// 启动监听
const bridgeHost = config.bridge.host;
const bridgePort = config.bridge.port;
httpServer.listen(bridgePort, bridgeHost, () => {
  console.error(`[server] Bridge Server ready — http://${bridgeHost}:${bridgePort}`);
  console.error(`[server] Health:  http://${bridgeHost}:${bridgePort}/api/health`);
  console.error(`[server] Status:  http://${bridgeHost}:${bridgePort}/api/status`);
  console.error(`[server] WebSocket: ws://${bridgeHost}:${bridgePort}/ws`);
  console.error('[server] Waiting for Tampermonkey scripts to connect...');
});

// 优雅退出
async function shutdown() {
  console.error('\n[server] Shutting down...');
  httpServer.close();
  await wsHub.stop();
  process.exit(0);
}

process.on('SIGINT', shutdown);
process.on('SIGTERM', shutdown);