mobbdev 1.1.14 → 1.1.16

package/dist/args/commands/upload_ai_blame.mjs

@@ -100,7 +100,7 @@ import z26 from "zod";
 import crypto from "crypto";
 import os from "os";
 import chalk2 from "chalk";
-import Debug7 from "debug";
+import Debug6 from "debug";
 import open from "open";
 
 // src/constants.ts
@@ -816,6 +816,57 @@ var UploadS3BucketInfoDocument = `
   }
 }
     `;
+var AnalyzeCommitForExtensionAiBlameDocument = `
+    mutation AnalyzeCommitForExtensionAIBlame($repositoryURL: String!, $commitSha: String!, $organizationId: String!) {
+  analyzeCommitForAIBlame(
+    repositoryURL: $repositoryURL
+    commitSha: $commitSha
+    organizationId: $organizationId
+  ) {
+    __typename
+    ... on ProcessAIBlameFinalResult {
+      status
+      inferencesProcessed
+      attributionsCreated
+      attributions {
+        id
+        aiBlameCommitId
+        aiBlameInferenceId
+        filePath
+        lineNumber
+        model
+        toolName
+        commitSha
+        inferenceType
+      }
+    }
+    ... on ProcessAIBlameErrorResult {
+      status
+      error
+    }
+    ... on ProcessAIBlameRequestedResult {
+      status
+      requestIds
+    }
+  }
+}
+    `;
+var GetAiBlameInferenceDocument = `
+    query GetAIBlameInference($aiBlameInferenceIds: [uuid!]) {
+  ai_blame_inference(where: {id: {_in: $aiBlameInferenceIds}}) {
+    id
+    type
+    aiResponseAt
+  }
+}
+    `;
+var GetAiBlameAttributionPromptDocument = `
+    query GetAIBlameAttributionPrompt($aiBlameAttributionId: String!) {
+  getAIBlameInferenceData(aiBlameAttributionId: $aiBlameAttributionId) {
+    promptUrl
+  }
+}
+    `;
 var UploadAiBlameInferencesInitDocument = `
     mutation UploadAIBlameInferencesInit($sessions: [AIBlameInferenceInitInput!]!) {
   uploadAIBlameInferencesInit(sessions: $sessions) {
@@ -1109,6 +1160,15 @@ function getSdk(client, withWrapper = defaultWrapper) {
     uploadS3BucketInfo(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: UploadS3BucketInfoDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "uploadS3BucketInfo", "mutation", variables);
     },
+    AnalyzeCommitForExtensionAIBlame(variables, requestHeaders, signal) {
+      return withWrapper((wrappedRequestHeaders) => client.request({ document: AnalyzeCommitForExtensionAiBlameDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "AnalyzeCommitForExtensionAIBlame", "mutation", variables);
+    },
+    GetAIBlameInference(variables, requestHeaders, signal) {
+      return withWrapper((wrappedRequestHeaders) => client.request({ document: GetAiBlameInferenceDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetAIBlameInference", "query", variables);
+    },
+    GetAIBlameAttributionPrompt(variables, requestHeaders, signal) {
+      return withWrapper((wrappedRequestHeaders) => client.request({ document: GetAiBlameAttributionPromptDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "GetAIBlameAttributionPrompt", "query", variables);
+    },
     UploadAIBlameInferencesInit(variables, requestHeaders, signal) {
       return withWrapper((wrappedRequestHeaders) => client.request({ document: UploadAiBlameInferencesInitDocument, variables, requestHeaders: { ...requestHeaders, ...wrappedRequestHeaders }, signal }), "UploadAIBlameInferencesInit", "mutation", variables);
     },
@@ -2171,10 +2231,9 @@ var VUL_REPORT_DIGEST_TIMEOUT_MS = 1e3 * 60 * 30;
 
 // src/features/analysis/graphql/gql.ts
 import fetchOrig from "cross-fetch";
-import Debug6 from "debug";
+import Debug5 from "debug";
 import { GraphQLClient } from "graphql-request";
-import { HttpProxyAgent } from "http-proxy-agent";
-import { HttpsProxyAgent as HttpsProxyAgent2 } from "https-proxy-agent";
+import { HttpsProxyAgent } from "https-proxy-agent";
 import { v4 as uuidv4 } from "uuid";
 
 // src/mcp/core/Errors.ts
@@ -2286,6 +2345,131 @@ var sleep = (ms = 2e3) => new Promise((r) => setTimeout(r, ms));
 var CliError = class extends Error {
 };
 
+// src/utils/subscribe/subscribe.ts
+import { createClient } from "graphql-ws";
+import WebsocketNode from "isomorphic-ws";
+
+// src/utils/subscribe/graphql.ts
+function getGraphQlHeaders(options) {
+  const headers = {
+    "Content-Type": "application/json"
+  };
+  if ("type" in options) {
+    if (options.type === "apiKey") {
+      headers["x-mobb-key"] = options.apiKey;
+    } else if (options.type === "token") {
+      headers["Authorization"] = `Bearer ${options.token}`;
+    }
+    return headers;
+  }
+  if ("isAdmin" in options && options.isAdmin) {
+    headers["x-hasura-access-key"] = options.adminPassword;
+  } else if ("isApiKey" in options && options.isApiKey) {
+    headers["x-mobb-key"] = options.apiKey;
+  } else if ("accessToken" in options) {
+    headers["Authorization"] = `Bearer ${options.accessToken}`;
+  }
+  return headers;
+}
+
+// src/utils/subscribe/subscribe.ts
+var DEFAULT_API_URL2 = "https://api.mobb.ai/v1/graphql";
+var SUBSCRIPTION_TIMEOUT_MS = 30 * 60 * 1e3;
+function createWSClient(options) {
+  const url = options.url || (process.env["API_URL"] || DEFAULT_API_URL2).replace("http", "ws");
+  const websocketImpl = options.websocket || (typeof WebSocket !== "undefined" ? WebSocket : WebsocketNode);
+  const CustomWebSocket = options.proxyAgent ? (
+    // biome-ignore lint/suspicious/noExplicitAny: Dynamic WebSocket extension requires any cast for cross-platform compatibility
+    class extends websocketImpl {
+      constructor(address, protocols) {
+        super(address, protocols, { agent: options.proxyAgent });
+      }
+    }
+  ) : websocketImpl;
+  return createClient({
+    //this is needed to prevent AWS from killing the connection
+    //currently our load balancer has a 29s idle timeout
+    keepAlive: 1e4,
+    url,
+    webSocketImpl: CustomWebSocket,
+    connectionParams: () => {
+      return {
+        headers: getGraphQlHeaders(options)
+      };
+    }
+  });
+}
+function subscribe(query, variables, callback, wsClientOptions) {
+  return new Promise((resolve, reject) => {
+    let timer = null;
+    let settled = false;
+    const { timeoutInMs = SUBSCRIPTION_TIMEOUT_MS } = wsClientOptions;
+    const client = createWSClient(wsClientOptions);
+    let unsubscribe = () => {
+      return;
+    };
+    function cleanup() {
+      try {
+        unsubscribe();
+      } catch {
+      }
+      if (timer) {
+        clearTimeout(timer);
+        timer = null;
+      }
+    }
+    function finalizeResolve(data) {
+      if (settled) {
+        return;
+      }
+      settled = true;
+      cleanup();
+      resolve(data);
+    }
+    function finalizeReject(error) {
+      if (settled) {
+        return;
+      }
+      settled = true;
+      cleanup();
+      reject(error);
+    }
+    unsubscribe = client.subscribe(
+      { query, variables },
+      {
+        next: (data) => {
+          if (!data.data) {
+            finalizeReject(
+              new Error(
+                `Broken data object from graphQL subscribe: ${JSON.stringify(
+                  data
+                )} for query: ${query}`
+              )
+            );
+          } else {
+            callback(finalizeResolve, finalizeReject, data.data);
+          }
+        },
+        error: (error) => {
+          finalizeReject(error);
+        },
+        complete: () => {
+          return;
+        }
+      }
+    );
+    if (typeof timeoutInMs === "number") {
+      timer = setTimeout(() => {
+        finalizeReject(
+          new Error(
+            `Timeout expired for graphQL subscribe query: ${query} with timeout: ${timeoutInMs}`
+          )
+        );
+      }, timeoutInMs);
+    }
+  });
+}
+
 // src/features/analysis/scm/utils/index.ts
 import { z as z15 } from "zod";
 
@@ -4404,107 +4588,6 @@ var debug5 = Debug4("scm:gitlab");
 // src/features/analysis/scm/scmFactory.ts
 import { z as z24 } from "zod";
 
-// src/features/analysis/graphql/subscribe.ts
-import Debug5 from "debug";
-import { createClient } from "graphql-ws";
-import { HttpsProxyAgent } from "https-proxy-agent";
-import WebSocket from "ws";
-var DEFAULT_API_URL2 = "https://api.mobb.ai/v1/graphql";
-var debug6 = Debug5("mobbdev:subscribe");
-var SUBSCRIPTION_TIMEOUT_MS = 30 * 60 * 1e3;
-function createWSClient(options) {
-  const proxy = options.url.startsWith("wss://") && process.env["HTTPS_PROXY"] ? new HttpsProxyAgent(process.env["HTTPS_PROXY"]) : options.url.startsWith("ws://") && process.env["HTTP_PROXY"] ? new HttpsProxyAgent(process.env["HTTP_PROXY"]) : null;
-  debug6(
-    `Using proxy: ${proxy ? "yes" : "no"} with url: ${options.url} and with proxy: ${process.env["HTTP_PROXY"]} for the websocket connection`
-  );
-  const CustomWebSocket = class extends WebSocket {
-    constructor(address, protocols) {
-      super(
-        address,
-        protocols,
-        proxy ? { agent: proxy } : void 0
-      );
-    }
-  };
-  return createClient({
-    //this is needed to prevent AWS from killing the connection
-    //currently our load balancer has a 29s idle timeout
-    keepAlive: 1e4,
-    url: options.url,
-    webSocketImpl: proxy ? CustomWebSocket : options.websocket || WebSocket,
-    connectionParams: () => {
-      return {
-        headers: options.type === "apiKey" ? {
-          [API_KEY_HEADER_NAME]: options.apiKey
-        } : { authorization: `Bearer ${options.token}` }
-      };
-    }
-  });
-}
-function subscribe(query, variables, callback, wsClientOptions) {
-  return new Promise((resolve, reject) => {
-    let timer = null;
-    const { timeoutInMs = SUBSCRIPTION_TIMEOUT_MS } = wsClientOptions;
-    const API_URL2 = process.env["API_URL"] || DEFAULT_API_URL2;
-    const client = createWSClient({
-      ...wsClientOptions,
-      websocket: WebSocket,
-      url: API_URL2.replace("http", "ws")
-    });
-    const unsubscribe = client.subscribe(
-      { query, variables },
-      {
-        next: (data) => {
-          function callbackResolve(data2) {
-            unsubscribe();
-            if (timer) {
-              clearTimeout(timer);
-            }
-            resolve(data2);
-          }
-          function callbackReject(data2) {
-            unsubscribe();
-            if (timer) {
-              clearTimeout(timer);
-            }
-            reject(data2);
-          }
-          if (!data.data) {
-            reject(
-              new Error(
-                `Broken data object from graphQL subscribe: ${JSON.stringify(
-                  data
-                )} for query: ${query}`
-              )
-            );
-          } else {
-            callback(callbackResolve, callbackReject, data.data);
-          }
-        },
-        error: (error) => {
-          if (timer) {
-            clearTimeout(timer);
-          }
-          reject(error);
-        },
-        complete: () => {
-          return;
-        }
-      }
-    );
-    if (typeof timeoutInMs === "number") {
-      timer = setTimeout(() => {
-        unsubscribe();
-        reject(
-          new Error(
-            `Timeout expired for graphQL subscribe query: ${query} with timeout: ${timeoutInMs}`
-          )
-        );
-      }, timeoutInMs);
-    }
-  });
-}
-
 // src/features/analysis/graphql/types.ts
 import { z as z25 } from "zod";
 var VulnerabilityReportIssueCodeNodeZ = z25.object({
@@ -4567,7 +4650,7 @@ var GetVulByNodesMetadataZ = z25.object({
 });
 
 // src/features/analysis/graphql/gql.ts
-var debug7 = Debug6("mobbdev:gql");
+var debug6 = Debug5("mobbdev:gql");
 var API_KEY_HEADER_NAME = "x-mobb-key";
 var REPORT_STATE_CHECK_DELAY = 5 * 1e3;
 function getProxyAgent(url) {
@@ -4575,14 +4658,14 @@ function getProxyAgent(url) {
     const parsedUrl = new URL(url);
     const isHttp = parsedUrl.protocol === "http:";
     const isHttps = parsedUrl.protocol === "https:";
-    const proxy = isHttps ? HTTPS_PROXY : isHttp ? HTTP_PROXY : null;
+    const proxy = isHttps ? HTTPS_PROXY || HTTP_PROXY : isHttp ? HTTP_PROXY : null;
     if (proxy) {
-      debug7("Using proxy %s", proxy);
-      debug7("Proxy agent %o", proxy);
-      return isHttps ? new HttpsProxyAgent2(proxy) : new HttpProxyAgent(proxy);
+      debug6("Using proxy %s", proxy);
+      debug6("Proxy agent %o", proxy);
+      return new HttpsProxyAgent(proxy);
     }
   } catch (err) {
-    debug7(`Skipping proxy for ${url}. Reason: ${err.message}`);
+    debug6(`Skipping proxy for ${url}. Reason: ${err.message}`);
   }
   return void 0;
 }
@@ -4597,7 +4680,7 @@ var fetchWithProxy = (url, options = {}) => {
       });
     }
   } catch (err) {
-    debug7(`Skipping proxy for ${url}. Reason: ${err.message}`);
+    debug6(`Skipping proxy for ${url}. Reason: ${err.message}`);
   }
   return fetchOrig(url, options);
 };
@@ -4606,7 +4689,7 @@ var GQLClient = class {
     __publicField(this, "_client");
     __publicField(this, "_clientSdk");
     __publicField(this, "_auth");
-    debug7(`init with  ${args}`);
+    debug6(`init with  ${args}`);
     this._auth = args;
     this._client = new GraphQLClient(API_URL, {
       headers: args.type === "apiKey" ? { [API_KEY_HEADER_NAME]: args.apiKey || "" } : {
@@ -4615,7 +4698,7 @@ var GQLClient = class {
       fetch: fetchWithProxy,
       requestMiddleware: (request) => {
         const requestId = uuidv4();
-        debug7(
+        debug6(
           `sending API request with id: ${requestId} and with request: ${request.body}`
         );
         return {
@@ -4652,7 +4735,7 @@ var GQLClient = class {
       await this.getUserInfo();
     } catch (e) {
       if (e?.toString().startsWith("FetchError")) {
-        debug7("verify connection failed %o", e);
+        debug6("verify connection failed %o", e);
         return false;
       }
     }
@@ -4664,7 +4747,7 @@ var GQLClient = class {
     try {
       info = await this.getUserInfo();
     } catch (e) {
-      debug7("verify token failed %o", e);
+      debug6("verify token failed %o", e);
       return false;
     }
     return info?.email || true;
@@ -4725,7 +4808,7 @@ var GQLClient = class {
     try {
       await this._clientSdk.CreateCommunityUser();
     } catch (e) {
-      debug7("create community user failed %o", e);
+      debug6("create community user failed %o", e);
     }
   }
   async updateScmToken(args) {
@@ -4892,11 +4975,13 @@ var GQLClient = class {
       this._auth.type === "apiKey" ? {
         apiKey: this._auth.apiKey,
         type: "apiKey",
-        timeoutInMs: params.timeoutInMs
+        timeoutInMs: params.timeoutInMs,
+        proxyAgent: getProxyAgent(API_URL)
       } : {
         token: this._auth.token,
         type: "token",
-        timeoutInMs: params.timeoutInMs
+        timeoutInMs: params.timeoutInMs,
+        proxyAgent: getProxyAgent(API_URL)
       }
     );
   }
@@ -4949,6 +5034,12 @@ var GQLClient = class {
   async finalizeAIBlameInferencesUploadRaw(variables) {
     return await this._clientSdk.FinalizeAIBlameInferencesUpload(variables);
   }
+  async analyzeCommitForExtensionAIBlame(variables) {
+    return await this._clientSdk.AnalyzeCommitForExtensionAIBlame(variables);
+  }
+  async getAIBlameAttributionPrompt(variables) {
+    return await this._clientSdk.GetAIBlameAttributionPrompt(variables);
+  }
 };
 
 // src/utils/ConfigStoreService.ts
@@ -4971,7 +5062,7 @@ function getConfigStore() {
 var configStore = getConfigStore();
 
 // src/commands/handleMobbLogin.ts
-var debug8 = Debug7("mobbdev:commands");
+var debug7 = Debug6("mobbdev:commands");
 var LOGIN_MAX_WAIT = 10 * 60 * 1e3;
 var LOGIN_CHECK_DELAY = 5 * 1e3;
 var webLoginUrl = `${WEB_APP_URL}/cli-login`;
@@ -5050,9 +5141,9 @@ async function handleMobbLogin({
     });
     loginSpinner.spin();
     if (encryptedApiToken) {
-      debug8("encrypted API token received %s", encryptedApiToken);
+      debug7("encrypted API token received %s", encryptedApiToken);
       newApiToken = crypto.privateDecrypt(privateKey, Buffer.from(encryptedApiToken, "base64")).toString("utf-8");
-      debug8("API token decrypted");
+      debug7("API token decrypted");
       break;
     }
     await sleep(LOGIN_CHECK_DELAY);
@@ -5066,7 +5157,7 @@ async function handleMobbLogin({
   const newGqlClient = new GQLClient({ apiKey: newApiToken, type: "apiKey" });
   const loginSuccess = await newGqlClient.validateUserToken();
   if (loginSuccess) {
-    debug8(`set api token ${newApiToken}`);
+    debug7(`set api token ${newApiToken}`);
     configStore.set("apiToken", newApiToken);
     loginSpinner.success({
       text: `\u{1F513} Login to Mobb successful! ${typeof loginSpinner === "string" ? `Logged in as ${loginSuccess}` : ""}`
@@ -5081,9 +5172,9 @@ async function handleMobbLogin({
 }
 
 // src/features/analysis/upload-file.ts
-import Debug8 from "debug";
+import Debug7 from "debug";
 import fetch3, { File, fileFrom, FormData } from "node-fetch";
-var debug9 = Debug8("mobbdev:upload-file");
+var debug8 = Debug7("mobbdev:upload-file");
 async function uploadFile({
   file,
   url,
@@ -5096,9 +5187,9 @@ async function uploadFile({
   logInfo(`FileUpload: upload file start ${url}`);
   logInfo(`FileUpload: upload fields`, uploadFields);
   logInfo(`FileUpload: upload key ${uploadKey}`);
-  debug9("upload file start %s", url);
-  debug9("upload fields %o", uploadFields);
-  debug9("upload key %s", uploadKey);
+  debug8("upload file start %s", url);
+  debug8("upload fields %o", uploadFields);
+  debug8("upload key %s", uploadKey);
   const form = new FormData();
   Object.entries(uploadFields).forEach(([key, value]) => {
     form.append(key, value);
@@ -5107,11 +5198,11 @@ async function uploadFile({
     form.append("key", uploadKey);
   }
   if (typeof file === "string") {
-    debug9("upload file from path %s", file);
+    debug8("upload file from path %s", file);
     logInfo(`FileUpload: upload file from path ${file}`);
     form.append("file", await fileFrom(file));
   } else {
-    debug9("upload file from buffer");
+    debug8("upload file from buffer");
     logInfo(`FileUpload: upload file from buffer`);
     form.append("file", new File([new Uint8Array(file)], "file"));
   }
@@ -5122,11 +5213,11 @@ async function uploadFile({
     agent
   });
   if (!response.ok) {
-    debug9("error from S3 %s %s", response.body, response.status);
+    debug8("error from S3 %s %s", response.body, response.status);
     logInfo(`FileUpload: error from S3 ${response.body} ${response.status}`);
     throw new Error(`Failed to upload the file: ${response.status}`);
   }
-  debug9("upload file done");
+  debug8("upload file done");
   logInfo(`FileUpload: upload file done`);
 }