fix: Refactor package.json for updated entry points

Author: robot-ux

package.json

@@ -3,18 +3,18 @@
   "version": "0.0.1",
   "description": "A MCP server for BNB Chain that supports BSC, opBNB, Greenfield, and other popular EVM-compatible networks.",
   "scripts": {
-    "start": "node dist/server/stdio.js",
-    "start:sse": "node dist/server/sse.js",
-    "dev": "bun --watch run src/server/stdio.ts",
-    "dev:sse": "bun --watch run src/server/sse.ts",
-    "build": "bun build src/**/*.ts --outdir dist --target node --format cjs",
+    "start": "node dist/index.js",
+    "start:sse": "bun run start --sse",
+    "dev": "bun --watch run src/index.ts",
+    "dev:sse": "bun run dev --sse",
+    "build": "bun build src/*.ts --outdir dist --target node --format cjs",
     "test": "npx @modelcontextprotocol/inspector bun dev",
     "prepare": "husky",
     "format": "prettier --write ."
   },
-  "main": "dist/server/stdio.js",
+  "main": "dist/index.js",
   "bin": {
-    "bnbchain-mcp": "./dist/cli/index.js"
+    "bnbchain-mcp": "./dist/index.js"
   },
   "files": [
     "dist"

src/cli/index.ts

@@ -0,0 +1,29 @@
+#!/usr/bin/env node
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp"
+
+import { startSSEServer } from "./server/sse"
+import { startStdioServer } from "./server/stdio"
+import logger from "./utils/logger"
+
+const args = process.argv.slice(2)
+const sseMode = args.includes("--sse") || args.includes("-s")
+
+async function main() {
+  let server: McpServer
+  if (sseMode) {
+    server = await startSSEServer()
+  } else {
+    server = await startStdioServer()
+  }
+
+  const handleShutdown = async () => {
+    logger.debug("BNBChain MCP Server shutting down")
+    await server.close()
+    process.exit(0)
+  }
+  // Handle process termination
+  process.on("SIGINT", handleShutdown)
+  process.on("SIGTERM", handleShutdown)
+}
+
+main()

src/index.ts

@@ -8,58 +8,68 @@ import type { Request, Response } from "express"
 import Logger from "@/utils/logger"
 import { startServer } from "./base"
 
-const app = express()
-const server = startServer()
-app.use(cors())
+export const startSSEServer = async () => {
+  try {
+    const app = express()
+    const server = startServer()
+    app.use(cors())
 
-// Log the current log level on startup
-Logger.info(`Starting sse server with log level: ${Logger.getLevel()}`)
+    // Log the current log level on startup
+    Logger.info(`Starting sse server with log level: ${Logger.getLevel()}`)
 
-// to support multiple simultaneous connections we have a lookup object from
-// sessionId to transport
-const transports: { [sessionId: string]: SSEServerTransport } = {}
+    // to support multiple simultaneous connections we have a lookup object from
+    // sessionId to transport
+    const transports: { [sessionId: string]: SSEServerTransport } = {}
 
-app.get("/sse", async (_: Request, res: Response) => {
-  const transport = new SSEServerTransport("/messages", res)
-  transports[transport.sessionId] = transport
-  Logger.info("New SSE connection established", {
-    sessionId: transport.sessionId
-  })
+    app.get("/sse", async (_: Request, res: Response) => {
+      const transport = new SSEServerTransport("/messages", res)
+      transports[transport.sessionId] = transport
+      Logger.info("New SSE connection established", {
+        sessionId: transport.sessionId
+      })
 
-  res.on("close", () => {
-    Logger.info("SSE connection closed", { sessionId: transport.sessionId })
-    delete transports[transport.sessionId]
-  })
+      res.on("close", () => {
+        Logger.info("SSE connection closed", { sessionId: transport.sessionId })
+        delete transports[transport.sessionId]
+      })
 
-  try {
-    await server.connect(transport)
-  } catch (error) {
-    Logger.error("Error connecting transport", {
-      sessionId: transport.sessionId,
-      error
+      try {
+        await server.connect(transport)
+      } catch (error) {
+        Logger.error("Error connecting transport", {
+          sessionId: transport.sessionId,
+          error
+        })
+      }
     })
-  }
-})
 
-app.post("/messages", async (req: Request, res: Response) => {
-  const sessionId = req.query.sessionId as string
-  const transport = transports[sessionId]
+    app.post("/messages", async (req: Request, res: Response) => {
+      const sessionId = req.query.sessionId as string
+      const transport = transports[sessionId]
 
-  if (transport) {
-    Logger.debug("Handling message", { sessionId, body: req.body })
-    try {
-      await transport.handlePostMessage(req, res)
-    } catch (error) {
-      Logger.error("Error handling message", { sessionId, error })
-      res.status(500).send("Internal server error")
-    }
-  } else {
-    Logger.warn("No transport found for session", { sessionId })
-    res.status(400).send("No transport found for sessionId")
-  }
-})
+      if (transport) {
+        Logger.debug("Handling message", { sessionId, body: req.body })
+        try {
+          await transport.handlePostMessage(req, res)
+        } catch (error) {
+          Logger.error("Error handling message", { sessionId, error })
+          res.status(500).send("Internal server error")
+        }
+      } else {
+        Logger.warn("No transport found for session", { sessionId })
+        res.status(400).send("No transport found for sessionId")
+      }
+    })
 
-const PORT = process.env.PORT || 3001
-app.listen(PORT, () => {
-  Logger.info(`BNBChain MCP SSE Server is running on http://localhost:${PORT}`)
-})
+    const PORT = process.env.PORT || 3001
+    app.listen(PORT, () => {
+      Logger.info(
+        `BNBChain MCP SSE Server is running on http://localhost:${PORT}`
+      )
+    })
+    return server
+  } catch (error) {
+    Logger.error("Error starting BNBChain MCP SSE Server:", error)
+    process.exit(1)
+  }
+}

src/server/sse.ts

@@ -13,10 +13,9 @@ export const startStdioServer = async () => {
     Logger.info("BNBChain MCP Server running on stdio mode")
 
     await server.connect(transport)
+    return server
   } catch (error) {
-    Logger.error("Error starting BNBChain MCP server:", error)
+    Logger.error("Error starting BNBChain MCP Stdio server:", error)
     process.exit(1)
   }
 }
-
-startStdioServer()