#!/usr/bin/env node
/**
* MCP Server with HTTP/SSE Transport
* This is for containerized deployment
*/
import express from 'express';
import { SSEServerTransport } from '@modelcontextprotocol/sdk/server/sse.js';
import * as dotenv from 'dotenv';
import { createMcpServer, ServerConfig } from './server.js';
// Load environment variables
dotenv.config();
// Validate required environment variables
if (!process.env.KIBANA_URL) {
console.error('Error: KIBANA_URL environment variable is required');
process.exit(1);
}
if (
!process.env.KIBANA_API_KEY &&
!(process.env.KIBANA_USERNAME && process.env.KIBANA_PASSWORD)
) {
console.error(
'Error: Either KIBANA_API_KEY or KIBANA_USERNAME/KIBANA_PASSWORD must be provided'
);
process.exit(1);
}
// Server configuration
const PORT = parseInt(process.env.HTTP_PORT || '3000', 10);
// Validate port number
if (isNaN(PORT) || PORT < 1 || PORT > 65535) {
console.error('Error: HTTP_PORT must be a valid port number (1-65535)');
process.exit(1);
}
const config: ServerConfig = {
kibana: {
url: process.env.KIBANA_URL,
apiKey: process.env.KIBANA_API_KEY,
username: process.env.KIBANA_USERNAME,
password: process.env.KIBANA_PASSWORD,
},
logLevel: (process.env.LOG_LEVEL as any) || 'info',
};
// Create Express app
const app = express();
app.use(express.json());
// Health check endpoint
app.get('/health', (req, res) => {
res.status(200).json({
status: 'healthy',
service: 'kibana-mcp-server',
version: '0.1.0',
timestamp: new Date().toISOString(),
});
});
// Server info endpoint
app.get('/info', (req, res) => {
res.json({
name: 'kibana-mcp-server',
version: '0.1.0',
description: 'Model Context Protocol server for Kibana integration',
capabilities: ['resources', 'tools'],
transport: 'sse',
});
});
// SSE endpoint for MCP communication
app.get('/sse', async (req, res) => {
console.log('New SSE connection established');
// Create a new MCP server instance for this connection
const mcpServer = createMcpServer(config);
const transport = new SSEServerTransport('/message', res);
try {
await mcpServer.connect(transport);
} catch (error) {
console.error('Failed to establish SSE connection:', error);
res.status(500).send('Failed to establish SSE connection');
return;
}
// Handle connection close
req.on('close', async () => {
console.log('SSE connection closed');
try {
await mcpServer.close();
} catch (error) {
console.error('Error closing MCP server:', error);
}
});
});
// Message endpoint for client requests
app.post('/message', async (req, res) => {
// This endpoint is handled by the SSE transport
// Just send a 200 response
res.status(200).send();
});
// Error handling middleware
app.use(
(
err: Error,
req: express.Request,
res: express.Response,
next: express.NextFunction
) => {
console.error('Express error:', err);
res.status(500).json({
error: 'Internal server error',
message: err.message,
});
}
);
// 404 handler
app.use((req, res) => {
res.status(404).json({
error: 'Not found',
path: req.path,
});
});
// Start server
const server = app.listen(PORT, '0.0.0.0', () => {
console.log(`🚀 Kibana MCP Server running on http://0.0.0.0:${PORT}`);
console.log(`📊 Connected to Kibana: ${config.kibana.url}`);
console.log(`🔗 SSE endpoint: http://0.0.0.0:${PORT}/sse`);
console.log(`❤️ Health check: http://0.0.0.0:${PORT}/health`);
});
// Graceful shutdown
const shutdown = async () => {
console.log('Shutting down HTTP server...');
server.close(() => {
console.log('HTTP server closed');
process.exit(0);
});
// Force close after 10 seconds
setTimeout(() => {
console.error('Forced shutdown after timeout');
process.exit(1);
}, 10000);
};
process.on('SIGINT', shutdown);
process.on('SIGTERM', shutdown);
