import { nwc } from "@getalby/sdk";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
import { Express } from "express";
import { createMCPServer } from "./mcp_server.js";
import { getConnectionSecretFromBearerAuth } from "./auth.js";
export function addSSEEndpoints(app: Express) {
// Store transports for each session type
// SSE is deprecated so we do not put much effort in here (e.g. cleaning up unused sessions)
const sessions: Record<
string,
{
server: McpServer;
transport: SSEServerTransport;
}
> = {};
app.get("/sse", (req, res) => {
const nostrWalletConnectUrl = getConnectionSecretFromBearerAuth(
req.header("Authorization")
);
if (!nostrWalletConnectUrl) {
res
.status(400)
.send("Bearer auth with NWC connection secret not provided");
return;
}
const client = new nwc.NWCClient({
nostrWalletConnectUrl,
});
const transport = new SSEServerTransport("/messages", res);
const server = createMCPServer(client);
sessions[transport.sessionId] = {
server,
transport,
};
server.connect(transport);
});
app.post("/messages", (req, res) => {
const sessionId = req.query.sessionId as string;
const session = sessions[sessionId];
if (session) {
session.transport.handlePostMessage(req, res);
} else {
res.status(400).send("No transport found for sessionId");
}
});
}
