mobbdev 1.2.0 → 1.2.3

package/README.md

@@ -175,6 +175,24 @@ Here is a simple example of a command line that will run Bugsy in your pipeline:
 npx mobbdev analyze --ci --scan-file $SAST_RESULTS_FILENAME --repo $CI_PROJECT_URL --ref $CI_COMMIT_REF_NAME --api-key $MOBB_API_KEY
 ```
 
+### Polling Mode
+
+By default, Bugsy uses WebSocket connections for real-time status updates during analysis. However, some proxy environments or firewalls may block WebSocket connections. In these cases, you can use the `--polling` flag to use HTTP polling instead:
+
+```shell
+npx mobbdev analyze --scan-file report.json --repo https://github.com/org/repo --polling
+```
+
+**When to use `--polling`:**
+- Your network blocks WebSocket connections (ws:// or wss://)
+- You're behind a corporate proxy that doesn't support WebSocket
+- You experience connection timeouts with the default WebSocket mode
+
+**Polling characteristics:**
+- Polling interval: 5 seconds
+- Timeout: 30 minutes
+- Slightly higher latency compared to WebSocket mode, but works in restricted environments
+
 ## Contribution
 
 Install the dependencies and run the tests:

package/dist/args/commands/upload_ai_blame.d.mts

@@ -21,7 +21,7 @@ type Logger = {
     error: (msg: string, data?: unknown) => void;
 };
 declare const PromptItemZ: z.ZodObject<{
-    type: z.ZodEnum<["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING"]>;
+    type: z.ZodEnum<["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING", "MCP_TOOL_CALL"]>;
     attachedFiles: z.ZodOptional<z.ZodArray<z.ZodObject<{
         relativePath: z.ZodString;
         startLine: z.ZodOptional<z.ZodNumber>;
@@ -50,30 +50,28 @@ declare const PromptItemZ: z.ZodObject<{
         result: z.ZodString;
         rawArguments: z.ZodOptional<z.ZodString>;
         accepted: z.ZodOptional<z.ZodBoolean>;
+        mcpServer: z.ZodOptional<z.ZodString>;
+        mcpToolName: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
     }, {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
-    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
-    tool?: {
-        name: string;
-        parameters: string;
-        result: string;
-        accepted?: boolean | undefined;
-        rawArguments?: string | undefined;
-    } | undefined;
+    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING" | "MCP_TOOL_CALL";
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -82,17 +80,19 @@ declare const PromptItemZ: z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
-}, {
-    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
+    text?: string | undefined;
     tool?: {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
     } | undefined;
+}, {
+    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING" | "MCP_TOOL_CALL";
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -101,10 +101,20 @@ declare const PromptItemZ: z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
+    text?: string | undefined;
+    tool?: {
+        name: string;
+        parameters: string;
+        result: string;
+        rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
+    } | undefined;
 }>;
 type PromptItem = z.infer<typeof PromptItemZ>;
 declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
-    type: z.ZodEnum<["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING"]>;
+    type: z.ZodEnum<["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING", "MCP_TOOL_CALL"]>;
     attachedFiles: z.ZodOptional<z.ZodArray<z.ZodObject<{
         relativePath: z.ZodString;
         startLine: z.ZodOptional<z.ZodNumber>;
@@ -133,30 +143,28 @@ declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
         result: z.ZodString;
         rawArguments: z.ZodOptional<z.ZodString>;
         accepted: z.ZodOptional<z.ZodBoolean>;
+        mcpServer: z.ZodOptional<z.ZodString>;
+        mcpToolName: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
     }, {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
-    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
-    tool?: {
-        name: string;
-        parameters: string;
-        result: string;
-        accepted?: boolean | undefined;
-        rawArguments?: string | undefined;
-    } | undefined;
+    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING" | "MCP_TOOL_CALL";
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -165,17 +173,19 @@ declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
-}, {
-    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING";
+    text?: string | undefined;
     tool?: {
         name: string;
         parameters: string;
         result: string;
-        accepted?: boolean | undefined;
         rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
     } | undefined;
+}, {
+    type: "USER_PROMPT" | "AI_RESPONSE" | "TOOL_EXECUTION" | "AI_THINKING" | "MCP_TOOL_CALL";
     date?: Date | undefined;
-    text?: string | undefined;
     attachedFiles?: {
         relativePath: string;
         startLine?: number | undefined;
@@ -184,6 +194,16 @@ declare const PromptItemArrayZ: z.ZodArray<z.ZodObject<{
         inputCount: number;
         outputCount: number;
     } | undefined;
+    text?: string | undefined;
+    tool?: {
+        name: string;
+        parameters: string;
+        result: string;
+        rawArguments?: string | undefined;
+        accepted?: boolean | undefined;
+        mcpServer?: string | undefined;
+        mcpToolName?: string | undefined;
+    } | undefined;
 }>, "many">;
 type PromptItemArray = z.infer<typeof PromptItemArrayZ>;
 type UploadAiBlameOptions = {

package/dist/args/commands/upload_ai_blame.mjs

@@ -6401,6 +6401,61 @@ var GQLClient = class {
       }
     );
   }
+  async pollForAnalysisState(params) {
+    const { analysisId, callbackStates, callback, timeoutInMs } = params;
+    const startTime = Date.now();
+    const maxDuration = timeoutInMs ?? 30 * 60 * 1e3;
+    const pollingIntervalSec = REPORT_STATE_CHECK_DELAY / 1e3;
+    debug6(
+      `[pollForAnalysisState] Starting polling for analysis ${analysisId}, target states: ${callbackStates.join(", ")}, interval: ${pollingIntervalSec}s`
+    );
+    let isPolling = true;
+    let pollCount = 0;
+    while (isPolling) {
+      pollCount++;
+      const elapsedSec = Math.round((Date.now() - startTime) / 1e3);
+      if (Date.now() - startTime > maxDuration) {
+        debug6(
+          `[pollForAnalysisState] Timeout expired after ${pollCount} polls (${elapsedSec}s)`
+        );
+        throw new ReportDigestError(
+          `Polling timeout expired for analysis: ${analysisId} with timeout: ${maxDuration}ms`,
+          `Analysis timed out after ${Math.round(maxDuration / 6e4)} minutes. Please try again or check the Mobb platform for status.`
+        );
+      }
+      debug6(
+        `[pollForAnalysisState] Poll #${pollCount} (elapsed: ${elapsedSec}s) - fetching analysis state...`
+      );
+      const analysis = await this.getAnalysis(analysisId);
+      debug6(
+        `[pollForAnalysisState] Poll #${pollCount} - current state: ${analysis.state}`
+      );
+      if (!analysis.state || analysis.state === "Failed" /* Failed */) {
+        const errorMessage = analysis.failReason || `Analysis failed with id: ${analysis.id}`;
+        debug6(`[pollForAnalysisState] Analysis failed: ${errorMessage}`);
+        throw new ReportDigestError(errorMessage, analysis.failReason ?? "");
+      }
+      if (callbackStates.includes(analysis.state)) {
+        debug6(
+          `[pollForAnalysisState] Target state reached: ${analysis.state} after ${pollCount} polls (${elapsedSec}s)`
+        );
+        await callback(analysis.id);
+        isPolling = false;
+        return {
+          analysis: {
+            id: analysis.id,
+            state: analysis.state,
+            failReason: analysis.failReason
+          }
+        };
+      }
+      debug6(
+        `[pollForAnalysisState] State ${analysis.state} not in target states, waiting ${pollingIntervalSec}s before next poll...`
+      );
+      await sleep(REPORT_STATE_CHECK_DELAY);
+    }
+    throw new Error(`Unexpected end of polling for analysis: ${analysisId}`);
+  }
   async getFixReportsByRepoUrl({ repoUrl }) {
     const res = await this._clientSdk.GetFixReportsByRepoUrl({
       repoUrl
@@ -7061,7 +7116,14 @@ var defaultLogger = {
   }
 };
 var PromptItemZ = z26.object({
-  type: z26.enum(["USER_PROMPT", "AI_RESPONSE", "TOOL_EXECUTION", "AI_THINKING"]),
+  type: z26.enum([
+    "USER_PROMPT",
+    "AI_RESPONSE",
+    "TOOL_EXECUTION",
+    "AI_THINKING",
+    "MCP_TOOL_CALL"
+    // MCP (Model Context Protocol) tool invocation
+  ]),
   attachedFiles: z26.array(
     z26.object({
       relativePath: z26.string(),
@@ -7079,7 +7141,12 @@ var PromptItemZ = z26.object({
     parameters: z26.string(),
     result: z26.string(),
     rawArguments: z26.string().optional(),
-    accepted: z26.boolean().optional()
+    accepted: z26.boolean().optional(),
+    // MCP-specific fields (only populated for MCP_TOOL_CALL type)
+    mcpServer: z26.string().optional(),
+    // MCP server name (e.g., "datadog", "mobb-mcp")
+    mcpToolName: z26.string().optional()
+    // MCP tool name without prefix (e.g., "scan_and_fix_vulnerabilities")
   }).optional()
 });
 var PromptItemArrayZ = z26.array(PromptItemZ);