Refactoring SDK Tool

src/lib/inmemory-store.ts

@@ -1 +1,2 @@
 export const signedUrlMap = new Map<string, object>();
+export const testFilePathsMap = new Map<string, string[]>();

src/server-factory.ts

@@ -7,6 +7,8 @@ const require = createRequire(import.meta.url);
 const packageJson = require("../package.json");
 import logger from "./logger.js";
 import addSDKTools from "./tools/bstack-sdk.js";
+import addPercyTools from "./tools/percy-sdk.js";
+import addSimulatePercyChangeTool from "./tools/simulate-percy.js";
 import addBrowserLiveTools from "./tools/live.js";
 import addAccessibilityTools from "./tools/accessibility.js";
 import addTestManagementTools from "./tools/testmanagement.js";
@@ -15,6 +17,8 @@ import addFailureLogsTools from "./tools/get-failure-logs.js";
 import addAutomateTools from "./tools/automate.js";
 import addSelfHealTools from "./tools/selfheal.js";
 import addAppLiveTools from "./tools/applive.js";
+import addListTestFilesTool from "./tools/list-test-files.js";
+import addPercySnapshotTools from "./tools/add-percy-snapshots.js";
 import { setupOnInitialized } from "./oninitialized.js";
 import { BrowserStackConfig } from "./lib/types.js";
 
@@ -48,13 +52,17 @@ export class BrowserStackMcpServer {
     const toolAdders = [
       addAccessibilityTools,
       addSDKTools,
+      addPercyTools,
+      addSimulatePercyChangeTool,
       addAppLiveTools,
       addBrowserLiveTools,
       addTestManagementTools,
       addAppAutomationTools,
       addFailureLogsTools,
       addAutomateTools,
       addSelfHealTools,
+      addListTestFilesTool,
+      addPercySnapshotTools,
     ];
 
     toolAdders.forEach((adder) => {

src/tools/add-percy-snapshots.ts

@@ -0,0 +1,48 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { testFilePathsMap } from "../lib/inmemory-store.js";
+import { UpdateTestFileWithInstructionsParams } from "./percy-snapshot-utils/constants.js";
+import { updateFileAndStep } from "./percy-snapshot-utils/utils.js";
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { percyWebSetupInstructions } from "../tools/sdk-utils/percy-web/handler.js";
+
+async function updateTestsWithPercyCommands(args: {
+  uuid: string;
+  index: number;
+}): Promise<CallToolResult> {
+  const { uuid, index } = args;
+  const filePaths = testFilePathsMap.get(uuid);
+
+  if (!filePaths) {
+    throw new Error(`No test files found in memory for UUID: ${uuid}`);
+  }
+  if (index < 0 || index >= filePaths.length) {
+    throw new Error(
+      `Invalid index: ${index}. There are ${filePaths.length} files for UUID: ${uuid}`,
+    );
+  }
+  const result = await updateFileAndStep(
+    filePaths[index],
+    index,
+    filePaths.length,
+    percyWebSetupInstructions,
+  );
+
+  return {
+    content: result,
+  };
+}
+
+export default function addPercySnapshotTools(server: McpServer) {
+  const tools: Record<string, any> = {};
+
+  tools.addPercySnapshotCommands = server.tool(
+    "addPercySnapshotCommands",
+    "Adds Percy snapshot commands to the specified test files.",
+    UpdateTestFileWithInstructionsParams,
+    async (args) => {
+      return await updateTestsWithPercyCommands(args);
+    },
+  );
+
+  return tools;
+}

src/tools/bstack-sdk.ts

@@ -1,254 +1,25 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-import { z } from "zod";
-import { trackMCP } from "../lib/instrumentation.js";
-import { getSDKPrefixCommand } from "./sdk-utils/commands.js";
-
-import {
-  SDKSupportedBrowserAutomationFramework,
-  SDKSupportedLanguage,
-  SDKSupportedTestingFramework,
-  SDKSupportedLanguageEnum,
-  SDKSupportedBrowserAutomationFrameworkEnum,
-  SDKSupportedTestingFrameworkEnum,
-} from "./sdk-utils/types.js";
-
-import {
-  generateBrowserStackYMLInstructions,
-  getInstructionsForProjectConfiguration,
-  formatInstructionsWithNumbers,
-} from "./sdk-utils/instructions.js";
-
-import {
-  formatPercyInstructions,
-  getPercyInstructions,
-} from "./sdk-utils/percy/instructions.js";
-import { getBrowserStackAuth } from "../lib/get-auth.js";
 import { BrowserStackConfig } from "../lib/types.js";
+import { RunTestsOnBrowserStackParamsShape } from "./sdk-utils/common/schema.js";
+import { runTestsOnBrowserStackHandler } from "./sdk-utils/handler.js";
+import { RUN_ON_BROWSERSTACK_DESCRIPTION } from "./sdk-utils/common/constants.js";
 
-/**
- * BrowserStack SDK hooks into your test framework to seamlessly run tests on BrowserStack.
- * This tool gives instructions to setup a browserstack.yml file in the project root and installs the necessary dependencies.
- */
-export async function bootstrapProjectWithSDK({
-  detectedBrowserAutomationFramework,
-  detectedTestingFramework,
-  detectedLanguage,
-  desiredPlatforms,
-  enablePercy,
-  config,
-}: {
-  detectedBrowserAutomationFramework: SDKSupportedBrowserAutomationFramework;
-  detectedTestingFramework: SDKSupportedTestingFramework;
-  detectedLanguage: SDKSupportedLanguage;
-  desiredPlatforms: string[];
-  enablePercy: boolean;
-  config: BrowserStackConfig;
-}): Promise<CallToolResult> {
-  // Get credentials from config
-  const authString = getBrowserStackAuth(config);
-  const [username, accessKey] = authString.split(":");
-
-  // Handle frameworks with unique setup instructions that don't use browserstack.yml
-  if (
-    detectedBrowserAutomationFramework === "cypress" ||
-    detectedTestingFramework === "webdriverio"
-  ) {
-    let combinedInstructions = getInstructionsForProjectConfiguration(
-      detectedBrowserAutomationFramework,
-      detectedTestingFramework,
-      detectedLanguage,
-      username,
-      accessKey,
-    );
-
-    if (enablePercy) {
-      const percyInstructions = getPercyInstructions(
-        detectedLanguage,
-        detectedBrowserAutomationFramework,
-        detectedTestingFramework,
-      );
-
-      if (percyInstructions) {
-        combinedInstructions +=
-          "\n\n" + formatPercyInstructions(percyInstructions);
-      } else {
-        throw new Error(
-          `Percy is currently not supported through MCP for ${detectedLanguage} with ${detectedTestingFramework}. If you want to run the test cases without Percy, disable Percy and run it again.`,
-        );
-      }
-    }
-
-    // Apply consistent formatting for all configurations
-    return formatFinalInstructions(combinedInstructions);
-  }
-
-  // Handle default flow using browserstack.yml
-  const sdkSetupCommand = getSDKPrefixCommand(
-    detectedLanguage,
-    detectedTestingFramework,
-    username,
-    accessKey,
-  );
-
-  const ymlInstructions = generateBrowserStackYMLInstructions(
-    desiredPlatforms,
-    enablePercy,
-  );
-
-  const instructionsForProjectConfiguration =
-    getInstructionsForProjectConfiguration(
-      detectedBrowserAutomationFramework,
-      detectedTestingFramework,
-      detectedLanguage,
-      username,
-      accessKey,
-    );
-
-  let combinedInstructions = "";
-
-  // Step 1: Add SDK setup command
-  if (sdkSetupCommand) {
-    combinedInstructions += sdkSetupCommand;
-  }
-
-  // Step 2: Add browserstack.yml setup
-  if (ymlInstructions) {
-    combinedInstructions += "\n\n---STEP---\n" + ymlInstructions;
-  }
-
-  // Step 3: Add language/framework-specific setup
-  if (instructionsForProjectConfiguration) {
-    combinedInstructions += "\n\n" + instructionsForProjectConfiguration;
-  }
-
-  // Step 4: Add Percy setup if applicable
-  if (enablePercy) {
-    const percyInstructions = getPercyInstructions(
-      detectedLanguage,
-      detectedBrowserAutomationFramework,
-      detectedTestingFramework,
-    );
-
-    if (percyInstructions) {
-      combinedInstructions +=
-        "\n\n" + formatPercyInstructions(percyInstructions);
-    } else {
-      throw new Error(
-        `Percy is currently not supported through MCP for ${detectedLanguage} with ${detectedTestingFramework}. If you want to run the test cases without Percy, disable Percy and run it again.`,
-      );
-    }
-  }
-
-  // Apply consistent formatting for all configurations
-  return formatFinalInstructions(combinedInstructions);
-}
-
-// Helper function to apply consistent formatting to all instruction types
-function formatFinalInstructions(combinedInstructions: string): CallToolResult {
-  const fullInstructions = `⚠️ IMPORTANT: DO NOT SKIP ANY STEP
-  All the setup steps described in this file MUST be executed regardless of any existing configuration or setup.
-  This ensures proper BrowserStack SDK setup.
-
-  ${formatInstructionsWithNumbers(combinedInstructions)}`;
-
-  return {
-    content: [
-      {
-        type: "text",
-        text: fullInstructions,
-        isError: false,
-      },
-    ],
-  };
-}
-
-export default function addSDKTools(
+export function registerRunBrowserStackTestsTool(
   server: McpServer,
   config: BrowserStackConfig,
 ) {
   const tools: Record<string, any> = {};
 
   tools.setupBrowserStackAutomateTests = server.tool(
     "setupBrowserStackAutomateTests",
-    "Set up and run automated web-based tests on BrowserStack using the BrowserStack SDK. Use for functional or integration tests on BrowserStack, with optional Percy visual testing for supported frameworks. Example prompts: run this test on browserstack; run this test on browserstack with Percy; set up this project for browserstack with Percy. Integrate BrowserStack SDK into your project",
-    {
-      detectedBrowserAutomationFramework: z
-        .nativeEnum(SDKSupportedBrowserAutomationFrameworkEnum)
-        .describe(
-          "The automation framework configured in the project. Example: 'playwright', 'selenium'",
-        ),
-
-      detectedTestingFramework: z
-        .nativeEnum(SDKSupportedTestingFrameworkEnum)
-        .describe(
-          "The testing framework used in the project. Be precise with framework selection Example: 'webdriverio', 'jest', 'pytest', 'junit4', 'junit5', 'mocha'",
-        ),
-
-      detectedLanguage: z
-        .nativeEnum(SDKSupportedLanguageEnum)
-        .describe(
-          "The programming language used in the project. Example: 'nodejs', 'python', 'java', 'csharp'",
-        ),
-
-      desiredPlatforms: z
-        .array(z.enum(["windows", "macos", "android", "ios"]))
-        .describe(
-          "The platforms the user wants to test on. Always ask this to the user, do not try to infer this.",
-        ),
-
-      enablePercy: z
-        .boolean()
-        .optional()
-        .default(false)
-        .describe(
-          "Set to true if the user wants to enable Percy for visual testing. Defaults to false.",
-        ),
-    },
-
+    RUN_ON_BROWSERSTACK_DESCRIPTION,
+    RunTestsOnBrowserStackParamsShape,
     async (args) => {
-      try {
-        trackMCP(
-          "runTestsOnBrowserStack",
-          server.server.getClientVersion()!,
-          undefined,
-          config,
-        );
-
-        return await bootstrapProjectWithSDK({
-          detectedBrowserAutomationFramework:
-            args.detectedBrowserAutomationFramework as SDKSupportedBrowserAutomationFramework,
-
-          detectedTestingFramework:
-            args.detectedTestingFramework as SDKSupportedTestingFramework,
-
-          detectedLanguage: args.detectedLanguage as SDKSupportedLanguage,
-
-          desiredPlatforms: args.desiredPlatforms,
-          enablePercy: args.enablePercy,
-          config,
-        });
-      } catch (error) {
-        trackMCP(
-          "runTestsOnBrowserStack",
-          server.server.getClientVersion()!,
-          error,
-          config,
-        );
-
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Failed to bootstrap project with BrowserStack SDK. Error: ${error}. Please open an issue on GitHub if the problem persists`,
-              isError: true,
-            },
-          ],
-          isError: true,
-        };
-      }
+      return runTestsOnBrowserStackHandler(args, config);
     },
   );
 
   return tools;
 }
+
+export default registerRunBrowserStackTestsTool;