import express, { Request, Response } from 'express';
import { randomUUID } from 'crypto';
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
import { validateToken, getConfig } from './auth/middleware.js';
import { ToolRegistry } from './tools/index.js';
import { RUN_WORKOUT_SYNTAX_RESOURCE } from './resources/run-workout-syntax.js';
import { CYCLING_WORKOUT_SYNTAX_RESOURCE } from './resources/cycling-workout-syntax.js';
export interface ServerOptions {
port: number;
}
export async function createServer(options: ServerOptions): Promise<express.Express> {
const app = express();
app.use(express.json());
const config = getConfig();
// Create tool registry with API clients (shared across connections)
const toolRegistry = new ToolRegistry({
intervals: config.intervals,
whoop: config.whoop,
trainerroad: config.trainerRoad,
});
console.log('Tool registry created');
// Store active transports and servers by sessionId
const sessions: Record<string, { transport: StreamableHTTPServerTransport; server: McpServer }> = {};
// Health check endpoint (no auth required)
app.get('/health', (_req: Request, res: Response) => {
res.json({ status: 'healthy', timestamp: new Date().toISOString() });
});
// Root redirect to GitHub
app.get('/', (_req: Request, res: Response) => {
res.redirect(302, 'https://github.com/gesteves/domestique');
});
// OAuth callback page for Whoop authorization
app.get('/callback', (req: Request, res: Response) => {
const code = req.query.code as string | undefined;
const error = req.query.error as string | undefined;
const errorDescription = req.query.error_description as string | undefined;
let content: string;
let statusClass: string;
let icon: string;
if (error) {
// OAuth error
statusClass = 'text-red-600';
icon = `<svg class="w-12 h-12 mx-auto mb-4 text-red-500" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M12 9v2m0 4h.01m-6.938 4h13.856c1.54 0 2.502-1.667 1.732-3L13.732 4c-.77-1.333-2.694-1.333-3.464 0L3.34 16c-.77 1.333.192 3 1.732 3z"></path></svg>`;
content = `
<h2 class="text-xl font-semibold ${statusClass} mb-2">Authorization Failed</h2>
<p class="text-gray-600 mb-4">${errorDescription || error}</p>
<p class="text-sm text-gray-500">Please close this window and try authorizing again.</p>
`;
} else if (code) {
// Success - show the code
statusClass = 'text-green-600';
icon = `<svg class="w-12 h-12 mx-auto mb-4 text-green-500" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M9 12l2 2 4-4m6 2a9 9 0 11-18 0 9 9 0 0118 0z"></path></svg>`;
content = `
<h2 class="text-xl font-semibold ${statusClass} mb-4">Authorization Successful</h2>
<p class="text-gray-600 mb-2">Your authorization code:</p>
<div class="flex mb-6">
<input
type="text"
id="code"
value="${code}"
readonly
class="flex-1 px-3 py-2 border border-gray-300 rounded-l-md bg-gray-50 font-mono text-sm focus:outline-none focus:ring-2 focus:ring-blue-500"
/>
<button
onclick="copyCode()"
id="copyBtn"
class="px-4 py-2 bg-blue-600 text-white rounded-r-md hover:bg-blue-700 focus:outline-none focus:ring-2 focus:ring-blue-500 transition-colors"
>
Copy
</button>
</div>
<div class="text-left text-sm text-gray-600">
<p class="font-medium mb-2">Next steps:</p>
<ol class="list-decimal list-inside space-y-1">
<li>Copy the code above</li>
<li>Return to your terminal</li>
<li>Paste the code when prompted</li>
<li>Close this window</li>
</ol>
</div>
`;
} else {
// No code or error - direct visit
statusClass = 'text-gray-600';
icon = `<svg class="w-12 h-12 mx-auto mb-4 text-gray-400" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M13 16h-1v-4h-1m1-4h.01M21 12a9 9 0 11-18 0 9 9 0 0118 0z"></path></svg>`;
content = `
<h2 class="text-xl font-semibold ${statusClass} mb-2">No Authorization Code</h2>
<p class="text-gray-600 mb-4">This page is used to receive OAuth authorization codes from Whoop.</p>
<p class="text-sm text-gray-500">To authorize, run <code class="bg-gray-100 px-1 rounded">npm run whoop:auth</code> and follow the instructions.</p>
`;
}
const html = `<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Whoop Authorization - Domestique</title>
<script src="https://cdn.tailwindcss.com"></script>
</head>
<body class="min-h-screen bg-gray-100 flex items-center justify-center p-4">
<div class="bg-white rounded-lg shadow-lg p-8 max-w-md w-full text-center">
${icon}
${content}
</div>
<script>
function copyCode() {
const codeInput = document.getElementById('code');
const copyBtn = document.getElementById('copyBtn');
if (codeInput) {
navigator.clipboard.writeText(codeInput.value).then(() => {
copyBtn.textContent = 'Copied!';
copyBtn.classList.remove('bg-blue-600', 'hover:bg-blue-700');
copyBtn.classList.add('bg-green-600');
setTimeout(() => {
copyBtn.textContent = 'Copy';
copyBtn.classList.remove('bg-green-600');
copyBtn.classList.add('bg-blue-600', 'hover:bg-blue-700');
}, 2000);
});
}
}
</script>
</body>
</html>`;
res.type('html').send(html);
});
// Notify all connected clients that tools have changed
// Useful after deployments that add/modify/remove tools
app.post('/admin/notify-tools-changed', validateToken, async (_req: Request, res: Response) => {
const sessionIds = Object.keys(sessions);
let notified = 0;
for (const sessionId of sessionIds) {
const { server } = sessions[sessionId];
try {
await server.sendToolListChanged();
notified++;
} catch (error) {
console.error(`Failed to notify session ${sessionId}:`, error);
}
}
console.log(`Notified ${notified}/${sessionIds.length} sessions of tool list change`);
res.json({
success: true,
sessions_notified: notified,
total_sessions: sessionIds.length,
});
});
// MCP endpoint - handles all Streamable HTTP requests
app.all('/mcp', validateToken, async (req: Request, res: Response) => {
// Check for existing session
const sessionId = req.headers['mcp-session-id'] as string | undefined;
// If we have an existing session, use it
if (sessionId && sessions[sessionId]) {
const { transport } = sessions[sessionId];
try {
await transport.handleRequest(req, res, req.body);
} catch (error) {
console.error('Error handling MCP request:', error);
if (!res.headersSent) {
res.status(500).json({ error: 'Internal server error' });
}
}
return;
}
// For new sessions (initialization), create a new server and transport
const mcpServer = new McpServer(
{ name: 'domestique', version: '1.0.0' },
{ capabilities: { tools: { listChanged: true }, resources: {}, prompts: {} } }
);
// Register tools for this connection
toolRegistry.registerTools(mcpServer);
// Register resources
mcpServer.resource(
'intervals-run-workout-syntax',
'intervals-run-workout-syntax://docs',
{
description: 'Documentation for creating structured running workouts in Intervals.icu format',
mimeType: 'text/markdown',
annotations: {
audience: ['assistant'],
},
},
async () => {
console.log('[MCP] Resource requested: intervals-run-workout-syntax');
return {
contents: [
{
uri: 'intervals-run-workout-syntax://docs',
mimeType: 'text/markdown',
text: RUN_WORKOUT_SYNTAX_RESOURCE,
},
],
};
}
);
mcpServer.resource(
'intervals-cycling-workout-syntax',
'intervals-cycling-workout-syntax://docs',
{
description: 'Documentation for creating structured cycling workouts in Intervals.icu format',
mimeType: 'text/markdown',
annotations: {
audience: ['assistant'],
},
},
async () => {
console.log('[MCP] Resource requested: intervals-cycling-workout-syntax');
return {
contents: [
{
uri: 'intervals-cycling-workout-syntax://docs',
mimeType: 'text/markdown',
text: CYCLING_WORKOUT_SYNTAX_RESOURCE,
},
],
};
}
);
// Register prompts
mcpServer.registerPrompt(
'daily_summary',
{
title: 'Daily Summary',
description:
'Get a complete overview of your fitness status today including recovery, strain, workouts, and fitness metrics',
},
async () => {
console.log('[MCP] Prompt requested: daily_summary');
return {
messages: [
{
role: 'user',
content: {
type: 'text',
text: 'Give me my daily fitness summary for today.',
},
},
],
};
}
);
const transport = new StreamableHTTPServerTransport({
sessionIdGenerator: () => randomUUID(),
onsessioninitialized: (newSessionId) => {
console.log(`Session initialized: ${newSessionId}`);
sessions[newSessionId] = { transport, server: mcpServer };
},
onsessionclosed: (closedSessionId) => {
console.log(`Session closed: ${closedSessionId}`);
delete sessions[closedSessionId];
},
});
// Connect the server to the transport
await mcpServer.connect(transport);
// Handle the request
try {
await transport.handleRequest(req, res, req.body);
} catch (error) {
console.error('Error handling MCP request:', error);
if (!res.headersSent) {
res.status(500).json({ error: 'Internal server error' });
}
}
});
// Handle session termination via DELETE
app.delete('/mcp', validateToken, async (req: Request, res: Response) => {
const sessionId = req.headers['mcp-session-id'] as string | undefined;
if (!sessionId || !sessions[sessionId]) {
res.status(404).json({ error: 'Session not found' });
return;
}
const { transport, server } = sessions[sessionId];
try {
await transport.close();
await server.close();
delete sessions[sessionId];
console.log(`Session terminated: ${sessionId}`);
res.status(204).send();
} catch (error) {
console.error('Error terminating session:', error);
res.status(500).json({ error: 'Internal server error' });
}
});
// Error handling middleware
app.use((err: Error, _req: Request, res: Response, _next: express.NextFunction) => {
console.error('Server error:', err);
res.status(500).json({ error: 'Internal server error' });
});
return app;
}
export async function startServer(options: ServerOptions): Promise<void> {
const app = await createServer(options);
app.listen(options.port, () => {
console.log(`Domestique MCP server running on port ${options.port}`);
console.log(`Health check: http://localhost:${options.port}/health`);
console.log(`MCP endpoint: http://localhost:${options.port}/mcp`);
});
}
