Alex Newman
9cb4b9d02a
- Introduced BaseRouteHandler class to centralize error handling and response management. - Updated SettingsRoutes to use wrapHandler for automatic error logging and response. - Refactored ViewerRoutes to extend BaseRouteHandler and utilize wrapHandler for health check and UI serving. - Enhanced error handling in SettingsRoutes and ViewerRoutes for better maintainability and readability.
80 lines
2.5 KiB
TypeScript
80 lines
2.5 KiB
TypeScript
/**
|
|
* Viewer Routes
|
|
*
|
|
* Handles health check, viewer UI, and SSE stream endpoints.
|
|
* These are used by the web viewer UI at http://localhost:37777
|
|
*/
|
|
|
|
import express, { Request, Response } from 'express';
|
|
import path from 'path';
|
|
import { readFileSync } from 'fs';
|
|
import { getPackageRoot } from '../../../../shared/paths.js';
|
|
import { SSEBroadcaster } from '../../SSEBroadcaster.js';
|
|
import { DatabaseManager } from '../../DatabaseManager.js';
|
|
import { SessionManager } from '../../SessionManager.js';
|
|
import { BaseRouteHandler } from '../BaseRouteHandler.js';
|
|
|
|
export class ViewerRoutes extends BaseRouteHandler {
|
|
constructor(
|
|
private sseBroadcaster: SSEBroadcaster,
|
|
private dbManager: DatabaseManager,
|
|
private sessionManager: SessionManager
|
|
) {
|
|
super();
|
|
}
|
|
|
|
setupRoutes(app: express.Application): void {
|
|
app.get('/health', this.handleHealth.bind(this));
|
|
app.get('/', this.handleViewerUI.bind(this));
|
|
app.get('/stream', this.handleSSEStream.bind(this));
|
|
}
|
|
|
|
/**
|
|
* Health check endpoint
|
|
*/
|
|
private handleHealth = this.wrapHandler((req: Request, res: Response): void => {
|
|
res.json({ status: 'ok', timestamp: Date.now() });
|
|
});
|
|
|
|
/**
|
|
* Serve viewer UI
|
|
*/
|
|
private handleViewerUI = this.wrapHandler((req: Request, res: Response): void => {
|
|
const packageRoot = getPackageRoot();
|
|
const viewerPath = path.join(packageRoot, 'plugin', 'ui', 'viewer.html');
|
|
const html = readFileSync(viewerPath, 'utf-8');
|
|
res.setHeader('Content-Type', 'text/html');
|
|
res.send(html);
|
|
});
|
|
|
|
/**
|
|
* SSE stream endpoint
|
|
*/
|
|
private handleSSEStream = this.wrapHandler((req: Request, res: Response): void => {
|
|
// Setup SSE headers
|
|
res.setHeader('Content-Type', 'text/event-stream');
|
|
res.setHeader('Cache-Control', 'no-cache');
|
|
res.setHeader('Connection', 'keep-alive');
|
|
|
|
// Add client to broadcaster
|
|
this.sseBroadcaster.addClient(res);
|
|
|
|
// Send initial_load event with projects list
|
|
const allProjects = this.dbManager.getSessionStore().getAllProjects();
|
|
this.sseBroadcaster.broadcast({
|
|
type: 'initial_load',
|
|
projects: allProjects,
|
|
timestamp: Date.now()
|
|
});
|
|
|
|
// Send initial processing status (based on queue depth + active generators)
|
|
const isProcessing = this.sessionManager.isAnySessionProcessing();
|
|
const queueDepth = this.sessionManager.getTotalActiveWork(); // Includes queued + actively processing
|
|
this.sseBroadcaster.broadcast({
|
|
type: 'processing_status',
|
|
isProcessing,
|
|
queueDepth
|
|
});
|
|
});
|
|
}
|