86 lines
2.2 KiB
TypeScript
86 lines
2.2 KiB
TypeScript
|
import express from 'express';
|
||
|
|
import cors from 'cors';
|
||
|
|
import helmet from 'helmet';
|
||
|
|
import morgan from 'morgan';
|
||
|
|
import dotenv from 'dotenv';
|
||
|
|
import http from 'http';
|
||
|
|
import { WebSocketServer } from 'ws';
|
||
|
|
import { BrainyData } from '@soulcraft/brainy';
|
||
|
|
import { setupRoutes } from './routes';
|
||
|
|
import { initializeBrainy } from './services/brainyService';
|
||
|
|
import { setupWebSocketHandlers } from './websocket';
|
||
|
|
|
||
|
|
// Load environment variables
|
||
|
|
dotenv.config();
|
||
|
|
|
||
|
|
// Initialize Express app
|
||
|
|
const app = express();
|
||
|
|
const port = process.env.PORT || 3000;
|
||
|
|
|
||
|
|
// Middleware
|
||
|
|
app.use(helmet()); // Security headers
|
||
|
|
app.use(cors()); // Enable CORS
|
||
|
|
app.use(morgan('combined')); // Logging
|
||
|
|
app.use(express.json()); // Parse JSON bodies
|
||
|
|
|
||
|
|
// Initialize Brainy
|
||
|
|
let brainyInstance: BrainyData;
|
||
|
|
|
||
|
|
// Health check endpoint
|
||
|
|
app.get('/health', (req, res) => {
|
||
|
|
res.status(200).json({ status: 'ok' });
|
||
|
|
});
|
||
|
|
|
||
|
|
// Setup API routes
|
||
|
|
app.use('/api', (req, res, next) => {
|
||
|
|
// Attach brainy instance to request
|
||
|
|
(req as any).brainy = brainyInstance;
|
||
|
|
next();
|
||
|
|
}, setupRoutes());
|
||
|
|
|
||
|
|
// Start the server
|
||
|
|
async function startServer() {
|
||
|
|
try {
|
||
|
|
// Initialize Brainy
|
||
|
|
brainyInstance = await initializeBrainy();
|
||
|
|
console.log('Brainy initialized successfully');
|
||
|
|
|
||
|
|
// Create HTTP server
|
||
|
|
const server = http.createServer(app);
|
||
|
|
|
||
|
|
// Create WebSocket server
|
||
|
|
const wss = new WebSocketServer({ server });
|
||
|
|
|
||
|
|
// Setup WebSocket handlers
|
||
|
|
setupWebSocketHandlers(wss, brainyInstance);
|
||
|
|
console.log('WebSocket server initialized');
|
||
|
|
|
||
|
|
// Start HTTP server
|
||
|
|
server.listen(port, () => {
|
||
|
|
console.log(`Server running on port ${port}`);
|
||
|
|
console.log(`WebSocket server running on ws://localhost:${port}`);
|
||
|
|
});
|
||
|
|
} catch (error) {
|
||
|
|
console.error('Failed to start server:', error);
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Handle graceful shutdown
|
||
|
|
process.on('SIGTERM', async () => {
|
||
|
|
console.log('SIGTERM received, shutting down gracefully');
|
||
|
|
// Cleanup Brainy resources
|
||
|
|
await brainyInstance?.clear();
|
||
|
|
process.exit(0);
|
||
|
|
});
|
||
|
|
|
||
|
|
process.on('SIGINT', async () => {
|
||
|
|
console.log('SIGINT received, shutting down gracefully');
|
||
|
|
// Cleanup Brainy resources
|
||
|
|
await brainyInstance?.clear();
|
||
|
|
process.exit(0);
|
||
|
|
});
|
||
|
|
|
||
|
|
// Start the server
|
||
|
|
startServer();
|