@skyramp/mcp 0.4.1 → 0.4.2-rc.2

package/build/recommendation/verifiers/statedDifference.js

@@ -1,4 +1,4 @@
-import { callSteps, isUIPlannedTest, stepRouteKey, subjectStep } from "../subjectStep.js";
+import { callSteps, isUIPlannedTest, stepRouteKey, subjectStep, } from "../subjectStep.js";
 import { STATED_DIFFERENCE_CONTRACT } from "../verifierContracts.js";
 /** What two planned tests must share before this asks how they differ: the method and
  *  path of their subject step. UI planned tests are OUT — two tests of one screen are
@@ -20,7 +20,10 @@ function nameKey(raw) {
  *  difference, and the pair's own objection is the question about it — naming the
  *  bad reference too would ask twice about one mistake. */
 function differenceStated(a, b) {
-    const states = (from, to) => (from.declarations?.differsFrom ?? []).some((entry) => nameKey(entry?.plannedTestId) === nameKey(to.plannedTestId) && (entry?.difference ?? "").trim().length > 0);
+    const states = (from, to) => (Array.isArray(from?.declarations?.differsFrom)
+        ? from.declarations.differsFrom
+        : []).some((entry) => nameKey(entry.plannedTestId) === nameKey(to?.plannedTestId) &&
+        String(entry.difference ?? "").trim().length > 0);
     return states(a, b) || states(b, a);
 }
 /** One objection per `differsFrom` entry naming no planned test in the plan. A
@@ -29,12 +32,16 @@ function differenceStated(a, b) {
  *  question the agent believes it already answered, and it would answer it again
  *  the same way. A blank reference names nothing either and is one of these. */
 function unknownPartnerObjections(plannedTests) {
-    const known = new Set(plannedTests.map((plannedTest) => nameKey(plannedTest?.plannedTestId)).filter(Boolean));
+    const known = new Set(plannedTests
+        .map((plannedTest) => nameKey(plannedTest.plannedTestId))
+        .filter(Boolean));
     const seen = new Set();
     const objections = [];
     for (const plannedTest of plannedTests) {
-        for (const entry of plannedTest?.declarations?.differsFrom ?? []) {
-            const raw = entry?.plannedTestId;
+        for (const entry of plannedTest.declarations.differsFrom ?? []) {
+            const raw = entry.plannedTestId;
+            // A non-string names nothing, and objecting to it put an empty reference in
+            // the objection's own id.
             if (typeof raw !== "string")
                 continue;
             const reference = nameKey(raw);
@@ -42,7 +49,7 @@ function unknownPartnerObjections(plannedTests) {
                 continue;
             // One objection per (planned test, reference): the same bad name written by
             // two planned tests is two mistakes, and each owes its own answer.
-            const key = `${nameKey(plannedTest?.plannedTestId)}\u0000${reference}`;
+            const key = `${nameKey(plannedTest.plannedTestId)}\u0000${reference}`;
             if (seen.has(key))
                 continue;
             seen.add(key);
@@ -70,7 +77,7 @@ function unstatedSubjectObjections(plannedTests) {
         const calls = callSteps(plannedTest);
         if (calls.length === 0 || subjectStep(plannedTest))
             continue;
-        const declared = plannedTest?.declarations?.stepUnderTest;
+        const declared = plannedTest.declarations.stepUnderTest;
         objections.push({
             objectionId: `statedDifference:subjectNotStated:${plannedTest.plannedTestId}`,
             verifier: "statedDifference",
@@ -91,7 +98,7 @@ function unstatedSubjectObjections(plannedTests) {
 export const statedDifference = {
     name: "statedDifference",
     run(registration, _ctx) {
-        const plannedTests = registration.plannedTests ?? [];
+        const plannedTests = registration.plannedTests;
         /** Planned-test id -> route -> the partners on that route it has NOT explained,
          *  recorded while the pair is known to be unexplained. Deriving it afterwards
          *  from the route would list partners this planned test already explained. */

package/build/recommendation/verifiers/uiElementGrounded.js

@@ -14,7 +14,9 @@ function field(value) {
  *  letter case, and the repeated spaces an HTML render collapses — and nothing
  *  about what it says: a shortened or paraphrased name is not a copy. */
 function folded(value) {
-    return typeof value === "string" ? value.trim().replace(/\s+/g, " ").toLowerCase() : "";
+    return typeof value === "string"
+        ? value.trim().replace(/\s+/g, " ").toLowerCase()
+        : "";
 }
 function context(value) {
     return Array.isArray(value) ? value.map(folded).filter(Boolean) : [];
@@ -75,7 +77,12 @@ function addTo(index, key, element) {
         index.set(key, [element]);
 }
 function poolOf(captured) {
-    const pool = { byTestId: new Map(), byStableId: new Map(), byRow: new Map(), byName: new Map() };
+    const pool = {
+        byTestId: new Map(),
+        byStableId: new Map(),
+        byRow: new Map(),
+        byName: new Map(),
+    };
     for (const raw of captured) {
         const element = fieldsOf(raw);
         addTo(pool.byTestId, element.testId, element);
@@ -101,18 +108,28 @@ function grounded(declared, pool) {
 /** Identity for the duplicate check: two entries agreeing on every field name
  *  one element, so the second says nothing the first did not. */
 function identityKey(element) {
-    return JSON.stringify([element.role, element.name, element.testId, element.stableId, element.contextText]);
+    return JSON.stringify([
+        element.role,
+        element.name,
+        element.testId,
+        element.stableId,
+        element.contextText,
+    ]);
 }
 /** How an element reads back in an objection, in the agent's own spelling. */
 function label(raw) {
     const element = (raw && typeof raw === "object" ? raw : {});
     const parts = [
         element.role ? `role "${String(element.role).trim()}"` : "",
-        element.accessibleName ? `name "${String(element.accessibleName).trim()}"` : "",
+        element.accessibleName
+            ? `name "${String(element.accessibleName).trim()}"`
+            : "",
         element.testId ? `testId "${String(element.testId).trim()}"` : "",
         element.stableId ? `stableId "${String(element.stableId).trim()}"` : "",
     ].filter(Boolean);
-    return parts.length > 0 ? parts.join(" ") : "an entry with no role, name or identifier";
+    return parts.length > 0
+        ? parts.join(" ")
+        : "an entry with no role, name or identifier";
 }
 /** A url split into the part before the fragment and the fragment WHEN it is a
  *  route (`#/…`). An ordinary anchor comes back as an empty route and is
@@ -123,7 +140,10 @@ function splitHashRoute(raw) {
     if (hash < 0)
         return { base: value, route: "" };
     const fragment = value.slice(hash);
-    return { base: value.slice(0, hash), route: fragment.startsWith("#/") ? fragment.replace(/\/+$/, "") : "" };
+    return {
+        base: value.slice(0, hash),
+        route: fragment.startsWith("#/") ? fragment.replace(/\/+$/, "") : "",
+    };
 }
 /** The path half of a url. The ROOT stays `/` rather than folding to empty, or a
  *  page named by its host alone would have no path to match on and the check
@@ -141,16 +161,26 @@ function urlPath(raw) {
         return new URL(base).pathname.replace(/\/+$/, "") || "/";
     }
     catch {
-        return base.startsWith("/") ? base.split("?")[0].replace(/\/+$/, "") || "/" : "";
+        return base.startsWith("/")
+            ? base.split("?")[0].replace(/\/+$/, "") || "/"
+            : "";
     }
 }
 /** The spellings of the machine the run is on: a capture at `127.0.0.1` and a
  *  page url at `localhost` are the same page. */
-const LOOPBACK_HOSTS = new Set(["localhost", "127.0.0.1", "[::1]", "::1", "0.0.0.0"]);
+const LOOPBACK_HOSTS = new Set([
+    "localhost",
+    "127.0.0.1",
+    "[::1]",
+    "::1",
+    "0.0.0.0",
+]);
 /** The host half of a url, empty for a path-only one. Read so a wrong host
  *  cannot ground an element on another service's capture. */
 function urlHost(raw) {
-    const value = String(raw ?? "").trim().split("#")[0];
+    const value = String(raw ?? "")
+        .trim()
+        .split("#")[0];
     if (!value)
         return "";
     try {
@@ -175,7 +205,7 @@ function elementsForPage(captures, url) {
     const pool = captures.filter((capture) => urlPath(capture.url) === path && (!host || urlHost(capture.url) === host));
     if (pool.length === 0)
         return undefined;
-    return poolOf(pool.flatMap((capture) => (Array.isArray(capture.elements) ? capture.elements : [])));
+    return poolOf(pool.flatMap((capture) => Array.isArray(capture.elements) ? capture.elements : []));
 }
 /** How many elements this run captured, across every page. The exemption below
  *  reads this and never `captures.length`: a capture object that holds nothing
@@ -203,10 +233,12 @@ function capturedPages(captures) {
  *  for. A guard on `data-cy` or `data-qa` reaches the plan with `testId` null and
  *  nothing else that carries the pair, so this cannot recognise it. */
 function removedTestIds(ctx) {
-    const removed = Array.isArray(ctx?.removedUiElements) ? ctx.removedUiElements : [];
+    const removed = Array.isArray(ctx?.removedUiElements)
+        ? ctx.removedUiElements
+        : [];
     return new Set(removed
-        .filter((item) => field(item?.attribute).toLowerCase() === "data-testid")
-        .map((item) => field(item?.value))
+        .filter((item) => field(item.attribute).toLowerCase() === "data-testid")
+        .map((item) => field(item.value))
         .filter(Boolean));
 }
 /** An objection names everything it saw. It used to stop at eight and say "and
@@ -223,15 +255,17 @@ export const uiElementGrounded = {
     name: "uiElementGrounded",
     run(registration, ctx) {
         const objections = [];
-        const captures = Array.isArray(ctx.uiCaptures) ? ctx.uiCaptures.filter((capture) => capture && typeof capture === "object") : [];
+        const captures = Array.isArray(ctx.uiCaptures)
+            ? ctx.uiCaptures.filter((capture) => capture && typeof capture === "object")
+            : [];
         const pools = new Map();
         const removedIds = removedTestIds(ctx);
         for (const plannedTest of registration.plannedTests ?? []) {
             if (!isUIPlannedTest(plannedTest))
                 continue;
-            const raw = plannedTest.declarations?.elements?.items;
+            const raw = plannedTest.declarations.elements?.items;
             const declared = Array.isArray(raw) ? raw : [];
-            const url = String(plannedTest.declarations?.elements?.pageUrl ?? "").trim();
+            const url = String(plannedTest.declarations.elements?.pageUrl ?? "").trim();
             if (declared.length === 0) {
                 // A run that saw nothing cannot ask this: standalone and IDE use store no
                 // digest and a backend-only run takes none. The boundary is the ELEMENT

package/build/resources/analysisResources.js

@@ -2,7 +2,7 @@ import * as fs from "fs";
 import { ResourceTemplate, } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StateManager, getSessionFilePath, getRegisteredSessions, hasSessionData, getSessionData, normalizeRecommendationState, } from "../utils/AnalysisStateManager.js";
 import { logger } from "../utils/logger.js";
-import { AnalysisScope, } from "../types/RepositoryAnalysis.js";
+import { AnalysisScope } from "../types/RepositoryAnalysis.js";
 export const ANALYSIS_URI_PREFIX = "skyramp://analysis";
 /**
  * Register MCP Resources for analysis data access.
@@ -25,7 +25,9 @@ export function registerAnalysisResources(server) {
         if (hasSessionData(sessionId)) {
             const memData = getSessionData(sessionId);
             if (memData?.analysis) {
-                logger.debug("Loaded analysis from process memory for resource", { sessionId });
+                logger.debug("Loaded analysis from process memory for resource", {
+                    sessionId,
+                });
                 return memData;
             }
         }
@@ -125,7 +127,9 @@ export function registerAnalysisResources(server) {
                     {
                         uri: uri.href,
                         mimeType: "application/json",
-                        text: JSON.stringify({ message: "No branch diff context (analysis scope was full_repo)." }, null, 2),
+                        text: JSON.stringify({
+                            message: "No branch diff context (analysis scope was full_repo).",
+                        }, null, 2),
                     },
                 ],
             };

package/build/resources/progressResource.js

@@ -5,10 +5,12 @@ export function registerProgressResource(server) {
         description: "Maps Skyramp MCP tool names to testbot progress phases.",
         mimeType: "application/json",
     }, () => ({
-        contents: [{
+        contents: [
+            {
                 uri: "skyramp://progress/tool-phases",
                 mimeType: "application/json",
                 text: JSON.stringify(TOOL_PHASE_MAP),
-            }],
+            },
+        ],
     }));
 }

package/build/resources/sutSetupResource.js

@@ -4,7 +4,8 @@ import { resourceText } from "../utils/utils.js";
 import { AnalyticsService } from "../services/AnalyticsService.js";
 import { getAdaptWorkflowPrompt } from "../prompts/sut-setup/modes/adaptWorkflowPrompt.js";
 import { getDockerComposePrompt } from "../prompts/sut-setup/modes/dockerComposePrompt.js";
-import { SutSetupMode } from "../prompts/sut-setup/shared.js";
+import { SutSetupMode, } from "../prompts/sut-setup/shared.js";
+import { getLocalRepositoryInfo } from "../utils/telemetry.js";
 export function registerSutSetupResource(server) {
     logger.info("Registering SUT setup resource");
     // RFC 6570 {+rest} (reserved expansion) captures the entire query string
@@ -19,8 +20,25 @@ export function registerSutSetupResource(server) {
     }, async (uri) => {
         const param = (name, fallback) => uri.searchParams.get(name) ?? fallback;
         const mode = param("sutSetupMode", SutSetupMode.None);
+        const repositoryPath = param("repositoryPath", ".");
+        // The released action sends no `repository` on this URI at all. Read the
+        // same thing it reads for the testbot URI — the origin remote of this very
+        // checkout — so the two names can never disagree.
+        // TODO: drop the fallback once a released action sends it here.
+        let repository = (uri.searchParams.get("repository") ?? "").trim();
+        // Mode "none" returns a fixed string that names no repository.
+        if (mode !== SutSetupMode.None) {
+            if (!repository) {
+                repository =
+                    (await getLocalRepositoryInfo(repositoryPath)).repository ?? "";
+            }
+            if (!repository) {
+                return resourceText(uri.toString(), `repository is required: pass the checkout's owner/repo on the resource URI, or run where ${repositoryPath} has an origin remote that names one`);
+            }
+        }
         const args = {
-            repositoryPath: param("repositoryPath", "."),
+            repositoryPath,
+            repository,
             sutSetupMode: mode,
             sutSourceWorkflowFile: param("sutSourceWorkflowFile", ""),
             sutSourceDockerComposeFile: param("sutSourceDockerComposeFile", ""),

package/build/resources/testbotResource.js

@@ -3,6 +3,8 @@ import { logger } from "../utils/logger.js";
 import { AnalyticsService } from "../services/AnalyticsService.js";
 import { getTestbotPrompt, parseRelatedRepositories, } from "../prompts/testbot/testbot-prompts.js";
 import { readWorkspaceServices } from "../prompts/prompt-utils.js";
+import { getLocalRepositoryInfo } from "../utils/telemetry.js";
+import { parseMaxFixAttemptsArg } from "../utils/fixAttempts.js";
 export function registerTestbotResource(server) {
     logger.info("Registering testbot resource");
     // RFC 6570 {+rest} (reserved expansion) captures the entire query string
@@ -35,6 +37,17 @@ export function registerTestbotResource(server) {
         };
         const prNum = intParam("prNumber");
         const repositoryPath = param("repositoryPath", ".");
+        // Required, like the prompt's own argument. A blank one would render an
+        // empty <REPOSITORY> block and tell the agent to attribute to nothing.
+        // The released action sends `primaryRepo` only on a multi-repo run, so a
+        // single-repo run names nothing. Read the same thing the action reads —
+        // the origin remote of this very checkout — so the two can never disagree.
+        // TODO: drop the fallback once a released action sends it on every run.
+        const primaryRepo = (uri.searchParams.get("primaryRepo") ?? "").trim() ||
+            (await getLocalRepositoryInfo(repositoryPath)).repository;
+        if (!primaryRepo) {
+            throw new Error(`primaryRepo is required: pass the primary checkout's owner/repo on the resource URI, or run where ${repositoryPath} has an origin remote that names one`);
+        }
         const services = await readWorkspaceServices(repositoryPath);
         const prompt = getTestbotPrompt({
             prTitle: param("prTitle", ""),
@@ -47,11 +60,16 @@ export function registerTestbotResource(server) {
             uiCredentials: uri.searchParams.get("uiCredentials") || undefined,
             testsRepoDir: uri.searchParams.get("testsRepoDir") || undefined,
             relatedRepositories: parseRelatedRepositories(uri.searchParams.get("relatedRepositories") || undefined),
-            primaryRepo: uri.searchParams.get("primaryRepo") || undefined,
+            primaryRepo,
             // Plan-only eval lane (SKYR-3879): recommendation phase only, nothing
             // generated or executed. Same accepted spellings as the prompt schema.
             planOnly: ["true", "1"].includes(uri.searchParams.get("planOnly") ?? ""),
             language: uri.searchParams.get("language") || undefined,
+            // Fix-and-rerun attempt cap (SKYR-4460): the shared parser, so an unusable
+            // or out-of-range value reads as unset here and on the prompt alike. Named
+            // "perTest…" on the wire: the URI is in the agent's context, and a bare
+            // "maxFixAttempts" reads as a run-wide total.
+            maxFixAttempts: parseMaxFixAttemptsArg(uri.searchParams.get("perTestMaxFixAttempts") ?? undefined),
         });
         AnalyticsService.pushMCPToolEvent("skyramp_testbot_prompt", undefined, {}).catch(() => { });
         // The URI WITHOUT its query string. The agent reads this line above the

package/build/services/AnalyticsService.js

@@ -35,7 +35,9 @@ async function safePushToolEvent(entryPoint, toolName, errorMessage, params) {
         await withTimeout(pushToolEvent(entryPoint, toolName, errorMessage, params), ANALYTICS_TIMEOUT_MS, "pushToolEvent");
     }
     catch (error) {
-        logger.error("safePushToolEvent failed", { error: error instanceof Error ? error.message : String(error) });
+        logger.error("safePushToolEvent failed", {
+            error: error instanceof Error ? error.message : String(error),
+        });
         // Swallow — analytics must never crash the server.
     }
 }

package/build/services/ScenarioGenerationService.js

@@ -49,10 +49,8 @@ export class ScenarioGenerationService {
         const statusCode = params.statusCode ?? inferExpectedStatus(method);
         const requestBody = params.requestBody ||
             (method === "GET" || method === "DELETE" ? "" : "{}");
-        const responseHeaders = params.responseHeaders
-            || { "Content-Type": ["application/json"] };
-        const isJsonResponse = (responseHeaders["Content-Type"] || [])
-            .some(v => v.includes("application/json"));
+        const responseHeaders = params.responseHeaders || { "Content-Type": ["application/json"] };
+        const isJsonResponse = (responseHeaders["Content-Type"] || []).some((v) => v.includes("application/json"));
         const responseBody = params.responseBody || (isJsonResponse ? "{}" : "");
         const requestHeaders = {
             "Content-Type": ["application/json"],
@@ -76,7 +74,9 @@ export class ScenarioGenerationService {
                         if (Array.isArray(v)) {
                             const items = v
                                 .filter((item) => item !== null)
-                                .map((item) => (typeof item === "object" ? JSON.stringify(item) : String(item)));
+                                .map((item) => typeof item === "object"
+                                ? JSON.stringify(item)
+                                : String(item));
                             if (items.length === 0)
                                 continue;
                             value = items.join(",");

package/build/services/TestDiscoveryService.js

@@ -4,7 +4,15 @@ import { logger } from "../utils/logger.js";
 import { TestSource } from "../types/TestAnalysis.js";
 import { TestType } from "../types/TestTypes.js";
 import fg from "fast-glob";
-import { isDiscoveredTestFile } from "../utils/testFileClassification.js";
+import { hasTestFileName, isDiscoveredTestFile, } from "../utils/testFileClassification.js";
+/**
+ * A Jest/Vitest snapshot in the colocated CRA layout. Its name never identifies
+ * it as a test (`Button.test.tsx.snap` cannot match a `.<ext>` tail across the
+ * `.tsx.` dot), but a render change leaves it stale, so it is maintenance
+ * material. It never enters the before-run gate: `detectLanguage` on `.snap`
+ * returns "unknown".
+ */
+const COLOCATED_SNAPSHOT = /[\\/]__snapshots__[\\/][^\\/]+\.snap$/;
 export class TestDiscoveryService {
     EXCLUDED_DIRS = [
         "node_modules",
@@ -23,9 +31,23 @@ export class TestDiscoveryService {
     SKYRAMP_MARKER = "Generated by Skyramp";
     // Supported test file extensions — covers all common languages/frameworks
     SUPPORTED_EXTENSIONS = [
-        ".py", ".js", ".ts", ".tsx", ".jsx", ".java",
-        ".go", ".rb", ".kt", ".kts", ".cs", ".rs",
-        ".php", ".scala", ".swift", ".m", ".dart",
+        ".py",
+        ".js",
+        ".ts",
+        ".tsx",
+        ".jsx",
+        ".java",
+        ".go",
+        ".rb",
+        ".kt",
+        ".kts",
+        ".cs",
+        ".rs",
+        ".php",
+        ".scala",
+        ".swift",
+        ".m",
+        ".dart",
         ".snap", // Jest/Vitest snapshot files — may need updating when UI changes break them
     ];
     // Concurrency control for parallel operations
@@ -54,7 +76,9 @@ export class TestDiscoveryService {
         logger.info(`Found ${classified.skyramp.length} Skyramp test files, ${classified.external.length} external test files`);
         // Process Skyramp tests (content already cached from classification)
         const skyrampTests = await this.processFilesInBatches(classified.skyramp, false, classified.contentCache);
-        skyrampTests.forEach(t => { t.source = TestSource.Skyramp; });
+        skyrampTests.forEach((t) => {
+            t.source = TestSource.Skyramp;
+        });
         // No diff context reaches this service: its one caller passes no options, so
         // every external file is potentially relevant and the cap is what bounds the
         // read set.
@@ -122,11 +146,12 @@ export class TestDiscoveryService {
         // TEST_FILE_PATTERNS includes Skyramp-generated suffixes (_smoke, _contract, etc.)
         // so Skyramp files outside standard test directories are still discovered.
         // This avoids reading every source file in the repo just to check the marker.
-        const testCandidates = allFiles.filter(f => this.isExternalTestFile(f));
+        const testCandidates = allFiles.filter((f) => this.isExternalTestFile(f));
         logger.debug(`Found ${allFiles.length} candidate files, narrowed to ${testCandidates.length} likely test files`);
         const result = { skyramp: [], external: [] };
         const contentCache = new Map();
         const marker = this.SKYRAMP_MARKER;
+        const dropped = [];
         const batchSize = 100;
         for (let i = 0; i < testCandidates.length; i += batchSize) {
             const batch = testCandidates.slice(i, i + batchSize);
@@ -137,16 +162,27 @@ export class TestDiscoveryService {
                         result.skyramp.push(file);
                         contentCache.set(file, content);
                     }
-                    else {
+                    else if (hasTestFileName(file) || COLOCATED_SNAPSHOT.test(file)) {
                         result.external.push(file);
                         contentCache.set(file, content);
                     }
+                    else {
+                        // A candidate without the marker and without a test file name is a
+                        // helper, fixture, page object, setup file or config that only
+                        // qualified by sitting in a test directory. It cannot be run as a
+                        // test, so it is not external coverage. Skyramp's own files keep the
+                        // wide rule above -- the marker is what identifies them.
+                        dropped.push(file);
+                    }
                 }
                 catch (error) {
-                    logger.debug(`Skipping file ${file}: ${error}`);
+                    logger.warning(`Skipping file ${file}: ${error}`);
                 }
             }
         }
+        if (dropped.length > 0) {
+            logger.debug(`Dropped ${dropped.length} test-directory candidates with no test file name: ${dropped.slice(0, 10).join(", ")}`);
+        }
         logger.debug(`Classified ${result.skyramp.length} Skyramp files, ${result.external.length} external test files`);
         return { ...result, contentCache };
     }
@@ -155,7 +191,11 @@ export class TestDiscoveryService {
      * or directory placement.
      */
     isExternalTestFile(filePath) {
-        return isDiscoveredTestFile(filePath);
+        // Either rule may admit a candidate: the directory rule for a test that
+        // sits in a test tree, the name rule for one that does not. Without the
+        // second, `auth.e2e.ts` or `OrdersIT.java` outside a test directory never
+        // reaches the classification below.
+        return isDiscoveredTestFile(filePath) || hasTestFileName(filePath);
     }
     /**
      * Extract metadata from a test file
@@ -457,7 +497,11 @@ export class TestDiscoveryService {
         const languageMap = {
             ".py": "python",
             ".js": "javascript",
+            ".mjs": "javascript",
+            ".cjs": "javascript",
             ".ts": "typescript",
+            ".mts": "typescript",
+            ".cts": "typescript",
             ".tsx": "typescript",
             ".jsx": "javascript",
             ".java": "java",