@prbe.ai/electron-sdk 0.1.10 → 0.1.13

package/dist/index.mjs

@@ -40,6 +40,7 @@ var ToolName = /* @__PURE__ */ ((ToolName2) => {
   ToolName2["CLIENT_FLAG_APP_LOGS"] = "client_flag_app_logs";
   ToolName2["CLIENT_ASK_USER"] = "client_ask_user";
   ToolName2["CLIENT_BASH_EXECUTE"] = "client_bash_execute";
+  ToolName2["CLIENT_MESSAGE_USER"] = "client_message_user";
   return ToolName2;
 })(ToolName || {});
 var PRBEAgentConfigKey = /* @__PURE__ */ ((PRBEAgentConfigKey3) => {
@@ -104,6 +105,7 @@ var PRBEStateEvent = /* @__PURE__ */ ((PRBEStateEvent2) => {
   PRBEStateEvent2["TICKET_INFO"] = "ticket-info";
   PRBEStateEvent2["INTERACTION_REQUESTED"] = "interaction-requested";
   PRBEStateEvent2["INTERACTION_RESOLVED"] = "interaction-resolved";
+  PRBEStateEvent2["AGENT_MESSAGE"] = "agent-message";
   return PRBEStateEvent2;
 })(PRBEStateEvent || {});
 var PRBEAgentState = class extends EventEmitter {
@@ -116,6 +118,7 @@ var PRBEAgentState = class extends EventEmitter {
   investigationError;
   pendingInteraction;
   resolvedInteractions = [];
+  agentMessage;
   // Completed user investigations (history)
   completedInvestigations = [];
   // Background context requests
@@ -143,6 +146,7 @@ var PRBEAgentState = class extends EventEmitter {
     this.summary = "";
     this.currentQuery = query;
     this.investigationError = void 0;
+    this.agentMessage = void 0;
     this.emit("status" /* STATUS */);
   }
   resetInvestigation() {
@@ -154,6 +158,7 @@ var PRBEAgentState = class extends EventEmitter {
     this.currentQuery = "";
     this.investigationError = void 0;
     this.pendingInteraction = void 0;
+    this.agentMessage = void 0;
     this.emit("status" /* STATUS */);
   }
   appendEvent(label, detail, completed = false) {
@@ -260,6 +265,18 @@ var PRBEAgentState = class extends EventEmitter {
     this.emit("interaction-resolved" /* INTERACTION_RESOLVED */);
     this.emit("status" /* STATUS */);
   }
+  setAgentMessage(message) {
+    this.agentMessage = message;
+    this.emit("agent-message" /* AGENT_MESSAGE */, { message });
+    this.emit("status" /* STATUS */);
+  }
+  setCRAgentMessage(crID, message) {
+    const cr = this.activeCRs.get(crID);
+    if (!cr) return;
+    cr.agentMessage = message;
+    this.emit("agent-message" /* AGENT_MESSAGE */, { message });
+    this.emit("status" /* STATUS */);
+  }
   toggleExpansion(eventId) {
     const event = this.events.find((e) => e.id === eventId);
     if (event) {
@@ -1294,70 +1311,136 @@ var AskUserTool = class {
 import { exec } from "child_process";
 import { randomUUID as randomUUID4 } from "crypto";
 import * as path3 from "path";
-var IS_WINDOWS = process.platform === "win32";
+
+// src/tools/safe-commands.ts
 var UNIX_SAFE_COMMANDS = /* @__PURE__ */ new Set([
+  // Filesystem (read-only)
   "ls",
   "cat",
   "head",
   "tail",
-  "grep",
-  "rg",
   "find",
-  "which",
-  "whoami",
-  "pwd",
-  "echo",
-  "printf",
-  "wc",
-  "sort",
-  "uniq",
-  "diff",
   "file",
   "stat",
   "du",
   "df",
+  "wc",
+  "sort",
+  "uniq",
+  "diff",
+  "tree",
+  // Search
+  "grep",
+  "rg",
+  "which",
+  // Text
+  "echo",
+  "printf",
+  // Identity / environment
+  "whoami",
+  "pwd",
+  "id",
+  "hostname",
   "uname",
   "env",
   "printenv",
+  "locale",
+  // System info (read-only reporters)
   "date",
-  "id",
-  "hostname",
+  "uptime",
+  "free",
   "ps",
   "top",
-  "uptime",
-  "free"
+  // macOS-specific (read-only only)
+  "sw_vers",
+  // macOS version
+  "system_profiler",
+  // hardware/software info
+  "mdls",
+  // Spotlight metadata for a file
+  "mdfind",
+  // Spotlight search
+  "lsof",
+  // list open files
+  "ioreg",
+  // I/O registry (hardware tree)
+  "log",
+  // macOS unified log (show, stream — read-only)
+  // Linux-specific (read-only reporters)
+  "lsb_release",
+  // distro info
+  "lscpu",
+  // CPU info
+  "lsblk",
+  // block devices
+  "lspci",
+  // PCI devices
+  "lsusb",
+  // USB devices
+  "lsmem",
+  // memory ranges
+  "dmidecode",
+  // BIOS/hardware info
+  "ss",
+  // socket statistics
+  "journalctl"
+  // systemd logs (read-only)
 ]);
 var WINDOWS_SAFE_COMMANDS = /* @__PURE__ */ new Set([
+  // CMD — filesystem (read-only)
   "dir",
   "type",
+  "tree",
+  "more",
   "findstr",
   "where",
+  // CMD — system info (read-only reporters)
+  "systeminfo",
+  "hostname",
+  "ver",
+  "vol",
+  "date",
+  "set",
+  "path",
+  "chcp",
   "whoami",
   "echo",
   "sort",
   "fc",
-  "hostname",
-  "date",
-  "systeminfo",
+  // CMD — processes
   "tasklist",
-  "set",
-  "ver",
-  "vol",
-  "tree",
-  // PowerShell cmdlets (when shell is PowerShell)
+  "driverquery",
+  // CMD — network (read-only)
+  "netstat",
+  // CMD — policy (read-only)
+  "gpresult",
+  // PowerShell — Get-* cmdlets (strictly read-only by convention)
   "get-childitem",
   "get-content",
+  "get-item",
+  "test-path",
+  "resolve-path",
+  "get-itemproperty",
   "select-string",
-  "get-process",
-  "get-date",
+  "measure-object",
   "get-host",
   "get-computerinfo",
+  "get-date",
   "get-volume",
-  "test-path",
-  "resolve-path",
-  "measure-object"
+  "get-disk",
+  "get-partition",
+  "get-process",
+  "get-service",
+  "get-netadapter",
+  "get-nettcpconnection",
+  "get-netipaddress",
+  "get-netroute",
+  "get-hotfix",
+  "get-eventlog",
+  "get-winevent",
+  "get-wmiobject",
+  "get-ciminstance"
 ]);
-var SAFE_COMMANDS = IS_WINDOWS ? WINDOWS_SAFE_COMMANDS : UNIX_SAFE_COMMANDS;
 var SAFE_COMMAND_PREFIXES = /* @__PURE__ */ new Set([
   "git status",
   "git log",
@@ -1369,6 +1452,11 @@ var SAFE_COMMAND_PREFIXES = /* @__PURE__ */ new Set([
   "npm list",
   "npm ls"
 ]);
+var IS_WINDOWS = process.platform === "win32";
+var SAFE_COMMANDS = IS_WINDOWS ? WINDOWS_SAFE_COMMANDS : UNIX_SAFE_COMMANDS;
+
+// src/tools/bash.ts
+var IS_WINDOWS2 = process.platform === "win32";
 var MAX_OUTPUT_BYTES = 100 * 1024;
 var DEFAULT_TIMEOUT_MS = 3e4;
 var MAX_TIMEOUT_MS = 12e4;
@@ -1446,7 +1534,7 @@ var BashExecuteTool = class {
           maxBuffer: MAX_OUTPUT_BYTES,
           env: process.env,
           // On Windows, use PowerShell for more consistent behavior
-          ...IS_WINDOWS ? { shell: "powershell.exe" } : {}
+          ...IS_WINDOWS2 ? { shell: "powershell.exe" } : {}
         },
         (error, stdout, stderr) => {
           let output = "";
@@ -1483,9 +1571,9 @@ var BashExecuteTool = class {
         return true;
       }
     }
-    const firstCommand = IS_WINDOWS ? trimmed.split(/[|;]/, 1)[0].trim() : trimmed.split(/[|;&]/, 1)[0].trim();
+    const firstCommand = IS_WINDOWS2 ? trimmed.split(/[|;]/, 1)[0].trim() : trimmed.split(/[|;&]/, 1)[0].trim();
     const baseCommand = firstCommand.split(/\s+/, 1)[0];
-    const commandName = IS_WINDOWS ? baseCommand.toLowerCase() : path3.basename(baseCommand);
+    const commandName = IS_WINDOWS2 ? baseCommand.toLowerCase() : path3.basename(baseCommand);
     if (SAFE_COMMANDS.has(commandName)) {
       return this.areAllPipeSegmentsSafe(trimmed);
     }
@@ -1495,7 +1583,7 @@ var BashExecuteTool = class {
    * For piped commands, check that every segment uses a safe command.
    */
   areAllPipeSegmentsSafe(command) {
-    if (IS_WINDOWS) {
+    if (IS_WINDOWS2) {
       if (/;/.test(command) && command.split("|").length <= 1) return false;
     } else {
       if (/[;&]|&&|\|\|/.test(command)) return false;
@@ -1503,7 +1591,7 @@ var BashExecuteTool = class {
     const segments = command.split("|").map((s) => s.trim());
     for (const segment of segments) {
       const baseCommand = segment.split(/\s+/, 1)[0];
-      const commandName = IS_WINDOWS ? baseCommand.toLowerCase() : path3.basename(baseCommand);
+      const commandName = IS_WINDOWS2 ? baseCommand.toLowerCase() : path3.basename(baseCommand);
       if (!SAFE_COMMANDS.has(commandName)) {
         return false;
       }
@@ -1753,13 +1841,6 @@ var PRBEAgent = class _PRBEAgent {
     this.state.updateTrackedSessionIDs(ids);
     this.syncPolling(ids.length > 0);
   }
-  get respondedCRIDs() {
-    return new Set(this.persistedData.respondedCRIds ?? []);
-  }
-  set respondedCRIDs(ids) {
-    this.persistedData.respondedCRIds = Array.from(ids);
-    savePersistedData(this.persistedData);
-  }
   syncPolling(hasSessions) {
     if (this.config.backgroundPolling && hasSessions) {
       if (this.pollingTimer === null) {
@@ -1821,6 +1902,22 @@ var PRBEAgent = class _PRBEAgent {
       this.registry.register(new AskUserTool(requester));
       this.registry.register(new BashExecuteTool(requester, roots, grantedPaths));
     }
+    this.registry.register(
+      new PRBEClosureTool(
+        "client_message_user",
+        "Send a message to the user.",
+        [{ name: "message", type: "STRING" /* STRING */, description: "Message for the user", required: true }],
+        async (args) => {
+          const message = args["message"] ?? "";
+          if (this.currentInvestigationSource === "context_request" /* CONTEXT_REQUEST */ && this.currentCRId) {
+            this.state.setCRAgentMessage(this.currentCRId, message);
+          } else {
+            this.state.setAgentMessage(message);
+          }
+          return "Message delivered.";
+        }
+      )
+    );
     if (this.config.captureConsole) {
       this.logCapture.startCapturing();
     }
@@ -2025,27 +2122,12 @@ var PRBEAgent = class _PRBEAgent {
         "/api/agent/poll",
         request
       );
-      const knownCRIDs = /* @__PURE__ */ new Set();
       for (const ticket of response.tickets) {
         for (const cr of ticket.context_requests) {
-          knownCRIDs.add(cr.id);
-          if (!cr.is_active || this.respondedCRIDs.has(cr.id)) continue;
+          if (!cr.is_active) continue;
           await this.investigateForCR(cr, ticket.ticket_id);
-          const ids = this.respondedCRIDs;
-          ids.add(cr.id);
-          this.respondedCRIDs = ids;
         }
       }
-      const currentRespondedIDs = this.respondedCRIDs;
-      const stale = new Set(
-        [...currentRespondedIDs].filter((id) => !knownCRIDs.has(id))
-      );
-      if (stale.size > 0) {
-        const pruned = new Set(
-          [...currentRespondedIDs].filter((id) => !stale.has(id))
-        );
-        this.respondedCRIDs = pruned;
-      }
       return response;
     } catch {
       return null;
@@ -2208,7 +2290,10 @@ var PRBEAgent = class _PRBEAgent {
         }));
         const startMetadata = {
           agent_id: this.agentID,
-          custom_tools: toolDeclarations
+          custom_tools: toolDeclarations,
+          platform: os2.platform(),
+          os_version: os2.release(),
+          arch: os2.arch()
         };
         if (contextRequestID) {
           startMetadata["context_request_id"] = contextRequestID;
@@ -2547,6 +2632,7 @@ function serializeCR(cr) {
     report: cr.report,
     summary: cr.summary,
     errorMessage: cr.errorMessage,
+    agentMessage: cr.agentMessage,
     startedAt: cr.startedAt.toISOString(),
     pendingInteraction: cr.pendingInteraction,
     resolvedInteractions: cr.resolvedInteractions ?? []
@@ -2562,6 +2648,7 @@ function serializePRBEState(state) {
     investigationError: state.investigationError,
     pendingInteraction: state.pendingInteraction,
     resolvedInteractions: state.resolvedInteractions,
+    agentMessage: state.agentMessage,
     completedInvestigations: state.completedInvestigations.map((inv) => ({
       id: inv.id,
       query: inv.query,