feat: implement setupOnInitialized function for server initialization checks and tracking

Author: ruturaj-browserstack

src/config.ts

@@ -1,12 +1,3 @@
-if (
-  !process.env.BROWSERSTACK_ACCESS_KEY ||
-  !process.env.BROWSERSTACK_USERNAME
-) {
-  throw new Error(
-    "Unable to start MCP server. Please set the BROWSERSTACK_ACCESS_KEY and BROWSERSTACK_USERNAME environment variables. Go to https://www.browserstack.com/accounts/profile/details to access them",
-  );
-}
-
 export class Config {
   constructor(
     public readonly browserstackUsername: string,

src/index.ts

@@ -13,7 +13,7 @@ import addTestManagementTools from "./tools/testmanagement.js";
 import addAppAutomationTools from "./tools/appautomate.js";
 import addFailureLogsTools from "./tools/getFailureLogs.js";
 import addAutomateTools from "./tools/automate.js";
-import { trackMCP } from "./lib/instrumentation.js";
+import { setupOnInitialized } from "./oninitialized.js";
 
 function registerTools(server: McpServer) {
   addSDKTools(server);
@@ -32,6 +32,8 @@ const server: McpServer = new McpServer({
   version: packageJson.version,
 });
 
+setupOnInitialized(server);
+
 registerTools(server);
 
 async function main() {
@@ -43,9 +45,6 @@ async function main() {
   // Start receiving messages on stdin and sending messages on stdout
   const transport = new StdioServerTransport();
   await server.connect(transport);
-
-  logger.info("MCP server started successfully");
-  trackMCP("started", server.server.getClientVersion()!);
 }
 
 main().catch(console.error);

src/oninitialized.ts

@@ -0,0 +1,24 @@
+import config from "./config.js";
+import { trackMCP } from "./lib/instrumentation.js";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+
+export function setupOnInitialized(server: McpServer) {
+  const nodeVersion = process.versions.node;
+
+  // Check for Node.js version
+  if (nodeVersion < "18.0.0") {
+    throw new Error(
+      "Node version is outdated. Please upgrade to 18.0.0 or later.",
+    );
+  }
+
+  // Check for BrowserStack credentials
+  if (!config.browserstackUsername || !config.browserstackAccessKey) {
+    throw new Error(
+      "BrowserStack credentials are missing. Please provide a valid username and access key.",
+    );
+  }
+  server.server.oninitialized = () => {
+    trackMCP("started", server.server.getClientVersion()!);
+  };
+}