@prbe.ai/electron-sdk 0.1.18 → 0.1.21

package/dist/index.mjs

@@ -1,5 +1,5 @@
 // package.json
-var version = "0.1.18";
+var version = "0.1.21";
 
 // src/agent.ts
 import * as fs2 from "fs";
@@ -25,6 +25,9 @@ var WSMessageType = /* @__PURE__ */ ((WSMessageType2) => {
   WSMessageType2["COMPLETE"] = "complete";
   WSMessageType2["ERROR"] = "error";
   WSMessageType2["PING"] = "ping";
+  WSMessageType2["PRIVACY_SANITIZING"] = "privacy_sanitizing";
+  WSMessageType2["PRIVACY_REVIEW"] = "privacy_review";
+  WSMessageType2["PRIVACY_REVIEW_RESPONSE"] = "privacy_review_response";
   return WSMessageType2;
 })(WSMessageType || {});
 var ConversationRole = /* @__PURE__ */ ((ConversationRole3) => {
@@ -51,6 +54,8 @@ var ToolName = /* @__PURE__ */ ((ToolName2) => {
   ToolName2["CLIENT_ASK_USER"] = "client_ask_user";
   ToolName2["CLIENT_BASH_EXECUTE"] = "client_bash_execute";
   ToolName2["CLIENT_MESSAGE_USER"] = "client_message_user";
+  ToolName2["SDK_MESSAGE_USER"] = "sdk_message_user";
+  ToolName2["SDK_ASK_USER"] = "sdk_ask_user";
   return ToolName2;
 })(ToolName || {});
 var UserIdentifierType = /* @__PURE__ */ ((UserIdentifierType2) => {
@@ -82,6 +87,7 @@ var PRBEAgentStatusType = /* @__PURE__ */ ((PRBEAgentStatusType2) => {
   PRBEAgentStatusType2["COMPLETED"] = "completed";
   PRBEAgentStatusType2["ERROR"] = "error";
   PRBEAgentStatusType2["AWAITING_INTERACTION"] = "awaiting_interaction";
+  PRBEAgentStatusType2["PRIVACY_SANITIZING"] = "privacy_sanitizing";
   return PRBEAgentStatusType2;
 })(PRBEAgentStatusType || {});
 var PRBEAgentErrorType = /* @__PURE__ */ ((PRBEAgentErrorType2) => {
@@ -121,7 +127,9 @@ var InvestigationConnection = class {
       if (!raw) return;
       try {
         const msg = JSON.parse(raw);
-        this.onMessage(msg);
+        Promise.resolve(this.onMessage(msg)).catch((err) => {
+          this.onError(`Message handler error: ${err instanceof Error ? err.message : String(err)}`);
+        });
       } catch {
       }
     };
@@ -147,10 +155,14 @@ var InvestigationConnection = class {
       return false;
     }
   }
-  sendConversationMessage(content) {
+  sendConversationMessage(content, role, label) {
+    const metadata = {};
+    if (role) metadata["role"] = role;
+    if (label) metadata["label"] = label;
     this.send({
       type: "conversation_message" /* CONVERSATION_MESSAGE */,
-      content
+      content,
+      ...Object.keys(metadata).length > 0 ? { metadata } : {}
     });
   }
   sendToolResult(callId, toolName, result, metadata) {
@@ -213,6 +225,7 @@ var PRBEAgentState = class extends EventEmitter {
   resolvedInteractions = [];
   agentMessage;
   conversationHistory = [];
+  isPrivacySanitizing = false;
   // Completed user investigations (history)
   completedInvestigations = [];
   // Background investigations (context requests, external requests, etc.)
@@ -310,6 +323,7 @@ var PRBEAgentState = class extends EventEmitter {
     });
     this.report = report;
     this.isInvestigating = false;
+    this.isPrivacySanitizing = false;
     this.emit("complete" /* COMPLETE */, { report });
     this.emit("status" /* STATUS */);
   }
@@ -317,9 +331,14 @@ var PRBEAgentState = class extends EventEmitter {
     this.appendEvent(`Error: ${message}`);
     this.investigationError = message;
     this.isInvestigating = false;
+    this.isPrivacySanitizing = false;
     this.emit("error" /* ERROR */, { message });
     this.emit("status" /* STATUS */);
   }
+  setPrivacySanitizing(value) {
+    this.isPrivacySanitizing = value;
+    this.emit("status" /* STATUS */);
+  }
   // ---------- Interaction state mutations ----------
   setPendingInteraction(payload) {
     this.pendingInteraction = payload;
@@ -500,6 +519,7 @@ var InteractionType = /* @__PURE__ */ ((InteractionType2) => {
   InteractionType2["ASK_QUESTION"] = "ask_question";
   InteractionType2["REQUEST_PERMISSION"] = "request_permission";
   InteractionType2["REQUEST_PATH_ACCESS"] = "request_path_access";
+  InteractionType2["REVIEW_SANITIZED_OUTPUT"] = "review_sanitized_output";
   return InteractionType2;
 })(InteractionType || {});
 var InvestigationSource = /* @__PURE__ */ ((InvestigationSource2) => {
@@ -1384,7 +1404,7 @@ var AskUserTool = class {
   }
   get declaration() {
     return {
-      name: "client_ask_user" /* CLIENT_ASK_USER */,
+      name: "sdk_ask_user" /* SDK_ASK_USER */,
       description: "Ask the user a question and wait for their response. Use this when you need clarification or additional information from the user to continue the investigation.",
       interactive: true,
       parameters: [
@@ -1413,7 +1433,8 @@ var AskUserTool = class {
     const response = await this.requester.requestUserInteraction({
       type: "ask_question" /* ASK_QUESTION */,
       interactionId: randomUUID3(),
-      question: reason
+      question,
+      context: reason
     });
     const askResponse = response;
     return askResponse.answer;
@@ -1623,6 +1644,10 @@ var BashExecuteTool = class {
         return `Error: working directory '${cwdArg}' is outside auto-approved directories`;
       }
       cwd = resolved;
+    } else if (this.autoApprovedDirs.length > 0) {
+      cwd = this.autoApprovedDirs[0];
+    } else {
+      return "Error: no approved directories configured and no working directory specified";
     }
     const isSafe = this.isCommandSafe(command);
     if (!isSafe) {
@@ -1818,7 +1843,7 @@ var PRBEAgent = class _PRBEAgent {
     }
     this.registry.register(
       new PRBEClosureTool(
-        "client_message_user",
+        "sdk_message_user" /* SDK_MESSAGE_USER */,
         "Send a message to the user.",
         [{ name: "message", type: "STRING" /* STRING */, description: "Message for the user", required: true }],
         async (args) => {
@@ -1886,8 +1911,8 @@ var PRBEAgent = class _PRBEAgent {
   get investigationSource() {
     return this.currentInvestigationSource;
   }
-  sendConversationMessage(content) {
-    this.activeConnection?.sendConversationMessage(content);
+  sendConversationMessage(content, role, label) {
+    this.activeConnection?.sendConversationMessage(content, role, label);
   }
   async requestUserInteraction(payload) {
     if (!this.interactionHandler) {
@@ -1896,6 +1921,19 @@ var PRBEAgent = class _PRBEAgent {
         "No interaction handler configured"
       );
     }
+    if (payload.type === "request_permission" /* REQUEST_PERMISSION */) {
+      const p = payload;
+      const content = p.reason ? `${p.command}
+
+${p.reason}` : p.command;
+      this.sendConversationMessage(content, "agent" /* Agent */, p.action);
+    } else if (payload.type === "request_path_access" /* REQUEST_PATH_ACCESS */) {
+      const p = payload;
+      const content = p.reason ? `${p.path}
+
+${p.reason}` : p.path;
+      this.sendConversationMessage(content, "agent" /* Agent */, "Request path access");
+    }
     if (this.currentInvestigationSource !== "user" /* USER */ && this.currentBackgroundId) {
       this.state.setBackgroundPendingInteraction(this.currentBackgroundId, payload);
     } else {
@@ -2041,6 +2079,14 @@ var PRBEAgent = class _PRBEAgent {
             this.state.failInvestigation(status.message);
           }
           break;
+        case "privacy_sanitizing" /* PRIVACY_SANITIZING */:
+          if (isBackground && opts.sourceId) {
+            this.state.appendBackgroundEvent(opts.sourceId, "Sanitizing data for privacy review...");
+          } else {
+            this.state.appendEvent("Sanitizing data for privacy review...");
+          }
+          this.state.setPrivacySanitizing(true);
+          break;
       }
     };
     const result = await this.connectToProxy(
@@ -2133,6 +2179,30 @@ var PRBEAgent = class _PRBEAgent {
       return null;
     }
   }
+  async fetchSanitizedFile(url) {
+    const res = await fetch(url, {
+      headers: { "X-API-Key": this.config.apiKey }
+    });
+    if (!res.ok) throw new Error(`Failed to fetch sanitized file: ${res.status}`);
+    const buffer = Buffer.from(await res.arrayBuffer());
+    if (!_PRBEAgent.isTextContent(buffer)) {
+      return { isText: false, url };
+    }
+    return { isText: true, content: buffer.toString("utf-8"), url };
+  }
+  /**
+   * Detect whether a buffer contains text by checking for null bytes
+   * and verifying the content is valid UTF-8.
+   */
+  static isTextContent(buffer) {
+    const checkLength = Math.min(buffer.length, 8192);
+    for (let i = 0; i < checkLength; i++) {
+      if (buffer[i] === 0) return false;
+    }
+    const decoded = buffer.toString("utf-8");
+    const reEncoded = Buffer.from(decoded, "utf-8");
+    return buffer.length === reEncoded.length;
+  }
   stopPolling() {
     if (this.pollingTimer !== null) {
       clearInterval(this.pollingTimer);
@@ -2222,8 +2292,8 @@ var PRBEAgent = class _PRBEAgent {
             uploadBaseUrl = url;
           }),
           (message) => {
-            if (!isCancelled()) emit({ type: "error" /* ERROR */, message });
-            finish(null);
+            if (!resolved && !isCancelled()) emit({ type: "error" /* ERROR */, message });
+            if (!resolved) finish(null);
           },
           () => {
             if (!resolved) {
@@ -2256,6 +2326,8 @@ var PRBEAgent = class _PRBEAgent {
         }));
         const startMetadata = {
           agent_id: this.agentID,
+          sdk_name: "electron",
+          supports_chat: true,
           custom_tools: toolDeclarations,
           platform: os.platform(),
           os_version: os.release(),
@@ -2304,7 +2376,12 @@ var PRBEAgent = class _PRBEAgent {
         const args = this.extractArgs(msg.metadata);
         emit({ type: "tool_call" /* TOOL_CALL */, name: toolName, label: msg.content ?? `Running ${toolName}` });
         this.pendingFlaggedFiles = [];
-        const toolResult = redactPII(await this.registry.execute(toolName, args));
+        let toolResult;
+        try {
+          toolResult = redactPII(await this.registry.execute(toolName, args));
+        } catch (err) {
+          toolResult = `Error: Tool execution failed: ${err instanceof Error ? err.message : String(err)}`;
+        }
         emit({ type: "observation" /* OBSERVATION */, text: toolResult.substring(0, 200) });
         let resultMetadata;
         const uploadBaseUrl = getUploadBaseUrl();
@@ -2366,6 +2443,41 @@ var PRBEAgent = class _PRBEAgent {
         }
         break;
       }
+      case "privacy_sanitizing" /* PRIVACY_SANITIZING */: {
+        emit({ type: "privacy_sanitizing" /* PRIVACY_SANITIZING */ });
+        break;
+      }
+      case "privacy_review" /* PRIVACY_REVIEW */: {
+        const sanitizedText = msg.content ?? "";
+        const issues = msg.metadata?.issues ?? [];
+        const files = msg.metadata?.files ?? [];
+        const summary = msg.metadata?.summary ?? "";
+        const payload = {
+          type: "review_sanitized_output" /* REVIEW_SANITIZED_OUTPUT */,
+          interactionId: randomUUID5(),
+          sanitizedAnalysis: sanitizedText,
+          files,
+          summary,
+          issues
+        };
+        try {
+          const response = await this.requestUserInteraction(payload);
+          const reviewResponse = response;
+          conn.send({
+            type: "privacy_review_response" /* PRIVACY_REVIEW_RESPONSE */,
+            metadata: {
+              approved: reviewResponse.approved,
+              ...reviewResponse.editedText ? { editedText: reviewResponse.editedText } : {}
+            }
+          });
+        } catch {
+          conn.send({
+            type: "privacy_review_response" /* PRIVACY_REVIEW_RESPONSE */,
+            metadata: { approved: false }
+          });
+        }
+        break;
+      }
       case "ping" /* PING */:
         conn.sendPong();
         break;
@@ -2468,6 +2580,7 @@ var PRBEAgent = class _PRBEAgent {
 // src/serialization.ts
 var DEFAULT_PRBE_STATE = {
   isInvestigating: false,
+  isPrivacySanitizing: false,
   events: [],
   report: "",
   summary: "",
@@ -2506,6 +2619,7 @@ function serializeBackgroundInvestigation(bg) {
 function serializePRBEState(state) {
   return {
     isInvestigating: state.isInvestigating,
+    isPrivacySanitizing: state.isPrivacySanitizing,
     events: state.events,
     report: state.report,
     summary: state.summary,