feat: refactor API URLs to use dynamic base URL retrieval for improved flexibility

Author: tech-sushant

src/tools/testmanagement-utils/add-test-result.ts

@@ -3,6 +3,7 @@ import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { z } from "zod";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Schema for adding a test result to a test run.
@@ -37,7 +38,8 @@ export async function addTestResult(
 ): Promise<CallToolResult> {
   try {
     const args = AddTestResultSchema.parse(rawArgs);
-    const url = `https://test-management.browserstack.com/api/v2/projects/${encodeURIComponent(
+    const tmBaseUrl = await getTMBaseURL();
+    const url = `${tmBaseUrl}/api/v2/projects/${encodeURIComponent(
       args.project_identifier,
     )}/test-runs/${encodeURIComponent(args.test_run_id)}/results`;
 

src/tools/testmanagement-utils/create-lca-steps.ts

@@ -8,6 +8,7 @@ import {
 import { pollLCAStatus } from "./poll-lca-status.js";
 import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Schema for creating LCA steps for a test case
@@ -81,7 +82,8 @@ export async function createLCASteps(
       config,
     );
 
-    const url = `https://test-management.browserstack.com/api/v1/projects/${projectId}/test-cases/${testCaseId}/lcnc`;
+    const tmBaseUrl = await getTMBaseURL();
+    const url = `${tmBaseUrl}/api/v1/projects/${projectId}/test-cases/${testCaseId}/lcnc`;
 
     const payload = {
       base_url: args.base_url,

src/tools/testmanagement-utils/create-project-folder.ts

@@ -5,6 +5,7 @@ import { formatAxiosError } from "../../lib/error.js";
 import { projectIdentifierToId } from "../testmanagement-utils/TCG-utils/api.js";
 import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 // Schema for combined project/folder creation
 export const CreateProjFoldSchema = z.object({
@@ -65,8 +66,9 @@ export async function createProjectOrFolder(
     try {
       const authString = getBrowserStackAuth(config);
       const [username, password] = authString.split(":");
+      const tmBaseUrl = await getTMBaseURL();
       const res = await apiClient.post({
-        url: "https://test-management.browserstack.com/api/v2/projects",
+        url: `${tmBaseUrl}/api/v2/projects`,
         headers: {
           "Content-Type": "application/json",
           Authorization:
@@ -95,8 +97,9 @@ export async function createProjectOrFolder(
     if (!projId)
       throw new Error("Cannot create folder without project_identifier.");
     try {
+      const tmBaseUrl = await getTMBaseURL();
       const res = await apiClient.post({
-        url: `https://test-management.browserstack.com/api/v2/projects/${encodeURIComponent(
+        url: `${tmBaseUrl}/api/v2/projects/${encodeURIComponent(
           projId,
         )}/folders`,
         headers: {

src/tools/testmanagement-utils/list-testcases.ts

@@ -4,6 +4,7 @@ import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { formatAxiosError } from "../../lib/error.js";
 import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Schema for listing test cases with optional filters.
@@ -49,7 +50,8 @@ export async function listTestCases(
     if (args.priority) params.append("priority", args.priority);
     if (args.p !== undefined) params.append("p", args.p.toString());
 
-    const url = `https://test-management.browserstack.com/api/v2/projects/${encodeURIComponent(
+    const tmBaseUrl = await getTMBaseURL();
+    const url = `${tmBaseUrl}/api/v2/projects/${encodeURIComponent(
       args.project_identifier,
     )}/test-cases?${params.toString()}`;
 

src/tools/testmanagement-utils/list-testruns.ts

@@ -4,6 +4,7 @@ import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { formatAxiosError } from "../../lib/error.js";
 import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Schema for listing test runs with optional filters.
@@ -35,8 +36,9 @@ export async function listTestRuns(
       params.set("run_state", args.run_state);
     }
 
+    const tmBaseUrl = await getTMBaseURL();
     const url =
-      `https://test-management.browserstack.com/api/v2/projects/${encodeURIComponent(
+      `${tmBaseUrl}/api/v2/projects/${encodeURIComponent(
         args.project_identifier,
       )}/test-runs?` + params.toString();
 

src/tools/testmanagement-utils/poll-lca-status.ts

@@ -1,6 +1,7 @@
 import { apiClient } from "../../lib/apiClient.js";
 import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Interface for the test case response structure
@@ -39,7 +40,8 @@ export async function pollLCAStatus(
   pollIntervalMs: number = 10 * 1000, // 10 seconds interval
   config: BrowserStackConfig,
 ): Promise<{ resource_path: string; status: string } | null> {
-  const url = `https://test-management.browserstack.com/api/v1/projects/${projectId}/folder/${folderId}/test-cases/${testCaseId}`;
+  const tmBaseUrl = await getTMBaseURL();
+  const url = `${tmBaseUrl}/api/v1/projects/${projectId}/folder/${folderId}/test-cases/${testCaseId}`;
 
   const startTime = Date.now();
 

src/tools/testmanagement-utils/testcase-from-file.ts

@@ -14,6 +14,7 @@ import { signedUrlMap } from "../../lib/inmemory-store.js";
 import logger from "../../logger.js";
 import { projectIdentifierToId } from "./TCG-utils/api.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 export async function createTestCasesFromFile(
   args: CreateTestCasesFromFileArgs,
@@ -87,7 +88,8 @@ export async function createTestCasesFromFile(
 
   signedUrlMap.delete(args.documentId);
 
-  const dashboardURL = `https://test-management.browserstack.com/projects/${args.projectReferenceId}/folder/${args.folderId}/test-cases`;
+  const tmBaseUrl = await getTMBaseURL();
+  const dashboardURL = `${tmBaseUrl}/projects/${args.projectReferenceId}/folder/${args.folderId}/test-cases`;
 
   return {
     content: [

src/tools/testmanagement-utils/update-testrun.ts

@@ -4,6 +4,7 @@ import { z } from "zod";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { formatAxiosError } from "../../lib/error.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Schema for updating a test run with partial fields.
@@ -40,7 +41,8 @@ export async function updateTestRun(
 ): Promise<CallToolResult> {
   try {
     const body = { test_run: args.test_run };
-    const url = `https://test-management.browserstack.com/api/v2/projects/${encodeURIComponent(
+    const tmBaseUrl = await getTMBaseURL();
+    const url = `${tmBaseUrl}/api/v2/projects/${encodeURIComponent(
       args.project_identifier,
     )}/test-runs/${encodeURIComponent(args.test_run_id)}/update`;
 

src/tools/testmanagement-utils/upload-file.ts

@@ -9,6 +9,7 @@ import { getBrowserStackAuth } from "../../lib/get-auth.js";
 import { signedUrlMap } from "../../lib/inmemory-store.js";
 import { projectIdentifierToId } from "./TCG-utils/api.js";
 import { BrowserStackConfig } from "../../lib/types.js";
+import { getTMBaseURL } from "../../lib/tm-base-url.js";
 
 /**
  * Schema for the upload file tool
@@ -56,7 +57,8 @@ export async function uploadFile(
     const formData = new FormData();
     formData.append("attachments[]", fs.createReadStream(file_path));
 
-    const uploadUrl = `https://test-management.browserstack.com/api/v1/projects/${projectIdResponse}/generic/attachments/ai_uploads`;
+    const tmBaseUrl = await getTMBaseURL();
+    const uploadUrl = `${tmBaseUrl}/api/v1/projects/${projectIdResponse}/generic/attachments/ai_uploads`;
 
     const response = await apiClient.post({
       url: uploadUrl,