@prbe.ai/electron-sdk 0.1.9 → 0.1.12

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-CeIldtlI.mjs';
-export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolveSessionsRequest, J as ResolveSessionsResponse, K as ResolvedInteraction, L as ResolvedTicketOut, N as TicketInfoRequest, O as TicketInfoResponse, Q as TicketStatusOut, S as ToolName, U as ToolParamType, W as WSMessage, V as WSMessageType, X as redactPII, Y as serializePRBEState } from './types-CeIldtlI.mjs';
+import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-B_KS1-FJ.mjs';
+export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolveSessionsRequest, J as ResolveSessionsResponse, K as ResolvedInteraction, L as ResolvedTicketOut, N as TicketInfoRequest, O as TicketInfoResponse, Q as TicketStatusOut, S as ToolName, U as ToolParamType, W as WSMessage, V as WSMessageType, X as redactPII, Y as serializePRBEState } from './types-B_KS1-FJ.mjs';
 import 'events';
 
 /**

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-CeIldtlI.js';
-export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolveSessionsRequest, J as ResolveSessionsResponse, K as ResolvedInteraction, L as ResolvedTicketOut, N as TicketInfoRequest, O as TicketInfoResponse, Q as TicketStatusOut, S as ToolName, U as ToolParamType, W as WSMessage, V as WSMessageType, X as redactPII, Y as serializePRBEState } from './types-CeIldtlI.js';
+import { P as PRBEToolDeclaration, a as PRBEToolParameter, b as PRBEInteractionRequester, F as FlaggedFileIn, c as PRBEAgentState, d as PRBEAgentConfig, I as InvestigationSource, e as InteractionPayload, f as InteractionResponse, g as PollResponse, T as TicketInfoOut } from './types-B_KS1-FJ.js';
+export { A as API_URL, h as AskQuestionPayload, i as AskQuestionResponse, C as ContextRequestOut, D as DEFAULT_PRBE_STATE, j as InteractionType, k as InvestigationResult, M as MIDDLEWARE_URL, l as PRBEAgentConfigKey, m as PRBEAgentError, n as PRBEAgentErrorType, o as PRBEAgentStatus, p as PRBEAgentStatusType, q as PRBECRInvestigation, r as PRBECompletedInvestigation, s as PRBEInteractionHandler, t as PRBESerializedCR, u as PRBESerializedCompletedInvestigation, v as PRBESerializedState, w as PRBESerializedTicket, x as PRBEStateEvent, y as PRBEStatusEvent, z as PollRequest, R as RequestPathAccessPayload, B as RequestPathAccessResponse, E as RequestPermissionPayload, G as RequestPermissionResponse, H as ResolveSessionsRequest, J as ResolveSessionsResponse, K as ResolvedInteraction, L as ResolvedTicketOut, N as TicketInfoRequest, O as TicketInfoResponse, Q as TicketStatusOut, S as ToolName, U as ToolParamType, W as WSMessage, V as WSMessageType, X as redactPII, Y as serializePRBEState } from './types-B_KS1-FJ.js';
 import 'events';
 
 /**

package/dist/index.js

@@ -110,6 +110,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) => {
@@ -174,6 +175,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 import_events.EventEmitter {
@@ -186,6 +188,7 @@ var PRBEAgentState = class extends import_events.EventEmitter {
   investigationError;
   pendingInteraction;
   resolvedInteractions = [];
+  agentMessage;
   // Completed user investigations (history)
   completedInvestigations = [];
   // Background context requests
@@ -213,6 +216,7 @@ var PRBEAgentState = class extends import_events.EventEmitter {
     this.summary = "";
     this.currentQuery = query;
     this.investigationError = void 0;
+    this.agentMessage = void 0;
     this.emit("status" /* STATUS */);
   }
   resetInvestigation() {
@@ -224,6 +228,7 @@ var PRBEAgentState = class extends import_events.EventEmitter {
     this.currentQuery = "";
     this.investigationError = void 0;
     this.pendingInteraction = void 0;
+    this.agentMessage = void 0;
     this.emit("status" /* STATUS */);
   }
   appendEvent(label, detail, completed = false) {
@@ -330,6 +335,18 @@ var PRBEAgentState = class extends import_events.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) {
@@ -699,7 +716,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 }
       ]
     };
   }
@@ -712,9 +730,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}'`;
     }
@@ -802,7 +821,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 }
       ]
     };
   }
@@ -814,8 +834,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) {
@@ -832,7 +853,7 @@ var FindFilesTool = class {
     }
     return result;
   }
-  walkAndMatch(dirPath, pattern, matches) {
+  walkAndMatch(dirPath, pattern, matches, caseSensitive) {
     let entries;
     try {
       entries = fs.readdirSync(dirPath, { withFileTypes: true });
@@ -843,9 +864,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({
@@ -863,7 +884,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];
@@ -901,7 +922,7 @@ var FindFilesTool = class {
     }
     regexStr += "$";
     try {
-      return new RegExp(regexStr).test(filename);
+      return new RegExp(regexStr, caseSensitive ? "" : "i").test(filename);
     } catch {
       return false;
     }
@@ -1360,70 +1381,136 @@ var AskUserTool = class {
 var import_child_process = require("child_process");
 var import_crypto4 = require("crypto");
 var path3 = __toESM(require("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",
@@ -1435,6 +1522,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;
@@ -1512,7 +1604,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 = "";
@@ -1549,9 +1641,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);
     }
@@ -1561,7 +1653,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;
@@ -1569,7 +1661,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;
       }
@@ -1887,6 +1979,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();
     }
@@ -2274,7 +2382,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;
@@ -2613,6 +2724,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 ?? []
@@ -2628,6 +2740,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,