cleanup

src/server/Server.ts

@@ -0,0 +1,97 @@
+import express from 'express';
+import http from 'http';
+import { WebSocketServer, WebSocket } from 'ws';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { logger } from '../util/logger.js';
+import { statusRoutes } from './routes/status.js';
+import { controlRoutes } from './routes/control.js';
+import { linkRoutes } from './routes/links.js';
+import { debugRoutes } from './routes/debug.js';
+import type { Bot } from '../bot/Bot.js';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+
+export class Server {
+  private app = express();
+  private server: http.Server;
+  private wss: WebSocketServer;
+  private clients: Set<WebSocket> = new Set();
+  private bot: Bot;
+
+  constructor(bot: Bot, private port: number = 3000) {
+    this.bot = bot;
+    this.app.use(express.json());
+
+    this.app.get('/', (_req, res) => {
+      res.sendFile(path.join(__dirname, '..', '..', 'src', 'server', 'index.html'));
+    });
+
+    // Mount routes
+    this.app.use('/api', statusRoutes(bot));
+    this.app.use('/api', controlRoutes(bot));
+    this.app.use('/api/links', linkRoutes(bot));
+    this.app.use('/api/debug', debugRoutes(bot, this));
+
+    this.server = http.createServer(this.app);
+    this.wss = new WebSocketServer({ server: this.server });
+
+    this.wss.on('connection', (ws) => {
+      this.clients.add(ws);
+      ws.send(JSON.stringify({ type: 'status', data: bot.getStatus() }));
+      ws.on('close', () => this.clients.delete(ws));
+    });
+
+    // Subscribe to bot events
+    bot.on('status-update', () => this.broadcastStatus());
+    bot.on('log', (level: string, message: string) => this.broadcastLog(level, message));
+  }
+
+  broadcastStatus(): void {
+    const msg = JSON.stringify({ type: 'status', data: this.bot.getStatus() });
+    for (const client of this.clients) {
+      if (client.readyState === WebSocket.OPEN) {
+        client.send(msg);
+      }
+    }
+  }
+
+  broadcastLog(level: string, message: string): void {
+    const msg = JSON.stringify({
+      type: 'log',
+      data: { level, message, time: new Date().toISOString() },
+    });
+    for (const client of this.clients) {
+      if (client.readyState === WebSocket.OPEN) {
+        client.send(msg);
+      }
+    }
+  }
+
+  broadcastDebug(action: string, data: Record<string, unknown>): void {
+    const msg = JSON.stringify({ type: 'debug', data: { action, ...data } });
+    for (const client of this.clients) {
+      if (client.readyState === WebSocket.OPEN) {
+        client.send(msg);
+      }
+    }
+  }
+
+  async start(): Promise<void> {
+    return new Promise((resolve) => {
+      this.server.listen(this.port, () => {
+        logger.info({ port: this.port }, `Dashboard running at http://localhost:${this.port}`);
+        resolve();
+      });
+    });
+  }
+
+  async stop(): Promise<void> {
+    for (const client of this.clients) {
+      client.close();
+    }
+    return new Promise((resolve) => {
+      this.server.close(() => resolve());
+    });
+  }
+}