@prbe.ai/electron-sdk 0.1.9 → 0.1.12

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) {
@@ -629,7 +646,8 @@ var SearchContentTool = class {
         { name: "pattern", type: "STRING" /* STRING */, description: "Regex pattern", required: true },
         { name: "path", type: "STRING" /* STRING */, description: "File or directory to search", required: true },
         { name: "context_lines", type: "INTEGER" /* INTEGER */, description: "Context lines (default 2)", required: false },
-        { name: "max_results", type: "INTEGER" /* INTEGER */, description: "Max results (default 50)", required: false }
+        { name: "max_results", type: "INTEGER" /* INTEGER */, description: "Max results (default 50)", required: false },
+        { name: "case_sensitive", type: "BOOLEAN" /* BOOLEAN */, description: "Case-sensitive search (default false)", required: false }
       ]
     };
   }
@@ -642,9 +660,10 @@ var SearchContentTool = class {
     if (!resolved) return `Error: ${resolveErr}`;
     const contextLines = typeof args["context_lines"] === "number" ? args["context_lines"] : 2;
     const maxResults = typeof args["max_results"] === "number" ? args["max_results"] : 50;
+    const caseSensitive = args["case_sensitive"] === true;
     let regex;
     try {
-      regex = new RegExp(pattern);
+      regex = new RegExp(pattern, caseSensitive ? "" : "i");
     } catch {
       return `Error: invalid regex pattern '${pattern}'`;
     }
@@ -732,7 +751,8 @@ var FindFilesTool = class {
       parameters: [
         { name: "pattern", type: "STRING" /* STRING */, description: "Glob pattern", required: true },
         { name: "path", type: "STRING" /* STRING */, description: "Directory to search", required: true },
-        { name: "max_results", type: "INTEGER" /* INTEGER */, description: "Max results (default 50)", required: false }
+        { name: "max_results", type: "INTEGER" /* INTEGER */, description: "Max results (default 50)", required: false },
+        { name: "case_sensitive", type: "BOOLEAN" /* BOOLEAN */, description: "Case-sensitive matching (default false)", required: false }
       ]
     };
   }
@@ -744,8 +764,9 @@ var FindFilesTool = class {
     const [resolved, resolveErr] = await resolvePath(pathStr, this.autoApprovedDirs, this.requester, this.grantedPaths);
     if (!resolved) return `Error: ${resolveErr}`;
     const maxResults = typeof args["max_results"] === "number" ? args["max_results"] : 50;
+    const caseSensitive = args["case_sensitive"] === true;
     const matches = [];
-    this.walkAndMatch(resolved, pattern, matches);
+    this.walkAndMatch(resolved, pattern, matches, caseSensitive);
     matches.sort((a, b) => b.modified.getTime() - a.modified.getTime());
     const limited = matches.slice(0, maxResults);
     if (limited.length === 0) {
@@ -762,7 +783,7 @@ var FindFilesTool = class {
     }
     return result;
   }
-  walkAndMatch(dirPath, pattern, matches) {
+  walkAndMatch(dirPath, pattern, matches, caseSensitive) {
     let entries;
     try {
       entries = fs.readdirSync(dirPath, { withFileTypes: true });
@@ -773,9 +794,9 @@ var FindFilesTool = class {
       if (entry.name.startsWith(".")) continue;
       const fullPath = path2.join(dirPath, entry.name);
       if (entry.isDirectory()) {
-        this.walkAndMatch(fullPath, pattern, matches);
+        this.walkAndMatch(fullPath, pattern, matches, caseSensitive);
       } else if (entry.isFile()) {
-        if (this.globMatch(entry.name, pattern)) {
+        if (this.globMatch(entry.name, pattern, caseSensitive)) {
           try {
             const stat = fs.statSync(fullPath);
             matches.push({
@@ -793,7 +814,7 @@ var FindFilesTool = class {
    * Simple glob matching: supports *, ?, and character classes [...].
    * Converts glob to regex and tests against the filename.
    */
-  globMatch(filename, pattern) {
+  globMatch(filename, pattern, caseSensitive) {
     let regexStr = "^";
     for (let i = 0; i < pattern.length; i++) {
       const c = pattern[i];
@@ -831,7 +852,7 @@ var FindFilesTool = class {
     }
     regexStr += "$";
     try {
-      return new RegExp(regexStr).test(filename);
+      return new RegExp(regexStr, caseSensitive ? "" : "i").test(filename);
     } catch {
       return false;
     }
@@ -1290,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",
@@ -1365,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;
@@ -1442,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 = "";
@@ -1479,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);
     }
@@ -1491,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;
@@ -1499,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;
       }
@@ -1817,6 +1909,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();
     }
@@ -2204,7 +2312,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;
@@ -2543,6 +2654,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 ?? []
@@ -2558,6 +2670,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,