Spaces:
Sleeping
Sleeping
File size: 2,979 Bytes
154bc5b 1bc57f1 154bc5b 1bc57f1 154bc5b 1bc57f1 154bc5b 1bc57f1 154bc5b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | #!/usr/bin/env node
import { createServer } from 'node:http'
import { readFile } from 'node:fs/promises'
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js'
import { createSourceServer, SERVER_NAME, SERVER_VERSION } from './mcp-server.js'
import { seedFantasyCatalog } from './fantasy-catalog.js'
import { parseAllowedRoots } from './security.js'
async function seedManagedFantasyCatalog() {
if (process.env.FANTASY_AUTO_SEED !== 'true' || !process.env.MONGODB_URI?.trim()) return
try {
const seed = JSON.parse(
await readFile(new URL('../data/fantasy-seed.json', import.meta.url), 'utf8'),
)
const result = await seedFantasyCatalog(seed)
console.error(
`Fantasy catalog ready: ${result.entities} cards (${result.inserted} inserted, ${result.updated} refreshed)`,
)
} catch (error) {
console.error(
`Fantasy catalog initialization failed: ${error instanceof Error ? error.message : String(error)}`,
)
}
}
async function startStdio() {
const server = createSourceServer()
await server.connect(new StdioServerTransport())
console.error(`${SERVER_NAME} ${SERVER_VERSION} ready on stdio`)
}
async function startHttp() {
await seedManagedFantasyCatalog()
const port = Number(process.env.PORT ?? 8790)
const host = process.env.HOST ?? '127.0.0.1'
const mcpPath = '/mcp'
const httpServer = createServer(async (request, response) => {
const url = new URL(request.url ?? '/', `http://${request.headers.host ?? 'localhost'}`)
if (request.method === 'GET' && url.pathname === '/') {
response.writeHead(200, { 'content-type': 'application/json; charset=utf-8' })
response.end(JSON.stringify({
name: SERVER_NAME,
version: SERVER_VERSION,
status: 'ready',
readOnly: true,
roots: parseAllowedRoots(),
mcp: `http://${host}:${port}${mcpPath}`,
}))
return
}
if (url.pathname === mcpPath && request.method && ['POST', 'GET', 'DELETE'].includes(request.method)) {
const server = createSourceServer()
const transport = new StreamableHTTPServerTransport({
sessionIdGenerator: undefined,
enableJsonResponse: true,
})
response.on('close', () => {
transport.close()
server.close()
})
try {
await server.connect(transport)
await transport.handleRequest(request, response)
} catch (error) {
console.error('MCP request failed', error)
if (!response.headersSent) response.writeHead(500).end('Internal server error')
}
return
}
response.writeHead(404).end('Not Found')
})
httpServer.listen(port, host, () => {
console.error(`${SERVER_NAME} ${SERVER_VERSION}: http://${host}:${port}${mcpPath}`)
})
}
if (process.argv.includes('--http')) await startHttp()
else await startStdio()
|