David Snelling
c641467854
### Changes: - Introduced `initializeMCPService` in `cloud-wrapper/src/index.ts` to support initializing Model Control Protocol (MCP) services. - Enabled configuration for WebSocket and REST APIs with customizable options like ports, authentication, rate limiting, and CORS. - Exported MCP modules, types, and services (`BrainyMCPAdapter`, `BrainyMCPService`, etc.) in `src/index.ts`. - Enhanced `unified.ts` to include global environment detection by populating `globalThis.__ENV__`. - Added environment-specific logging for browser, Node.js, and serverless scenarios. - Improved the structure of `environment` object and its global accessibility. ### Purpose: Implemented MCP service integration to expand Brainy's external compatibility via WebSocket and REST interfaces. Enhanced environment detection provides better global access and debugging insights, ensuring robust operation across various runtime environments.
106 lines
3.2 KiB
TypeScript
106 lines
3.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.js';
|
|
import { initializeBrainy } from './services/brainyService.js';
|
|
import { setupWebSocketHandlers } from './websocket.js';
|
|
import { initializeMCPService } from './services/mcpService.js';
|
|
|
|
// 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
|
|
const apiRouter = setupRoutes();
|
|
app.use('/api', (req, res, next) => {
|
|
// Attach brainy instance to request
|
|
(req as any).brainy = brainyInstance;
|
|
next();
|
|
}, apiRouter);
|
|
|
|
// 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');
|
|
|
|
// Initialize MCP service
|
|
const mcpWsPort = process.env.MCP_WS_PORT ? parseInt(process.env.MCP_WS_PORT, 10) : undefined;
|
|
const mcpRestPort = process.env.MCP_REST_PORT ? parseInt(process.env.MCP_REST_PORT, 10) : undefined;
|
|
|
|
if (mcpWsPort || mcpRestPort) {
|
|
initializeMCPService(brainyInstance, {
|
|
wsPort: mcpWsPort,
|
|
restPort: mcpRestPort,
|
|
enableAuth: process.env.MCP_ENABLE_AUTH === 'true',
|
|
apiKeys: process.env.MCP_API_KEYS ? process.env.MCP_API_KEYS.split(',') : undefined,
|
|
rateLimit: process.env.MCP_RATE_LIMIT_REQUESTS ? {
|
|
windowMs: parseInt(process.env.MCP_RATE_LIMIT_WINDOW_MS || '60000', 10),
|
|
maxRequests: parseInt(process.env.MCP_RATE_LIMIT_REQUESTS, 10)
|
|
} : undefined,
|
|
cors: process.env.MCP_ENABLE_CORS === 'true' ? {} : undefined
|
|
});
|
|
console.log('MCP service 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();
|