@prbe.ai/electron-sdk 0.1.8 → 0.1.10

package/dist/index.mjs

@@ -122,7 +122,7 @@ var PRBEAgentState = class extends EventEmitter {
   activeCRs = /* @__PURE__ */ new Map();
   completedCRs = [];
   // Tracked tickets
-  trackedTicketIDs = [];
+  trackedSessionIDs = [];
   ticketInfo = [];
   // Computed
   get hasActiveWork() {
@@ -350,8 +350,8 @@ var PRBEAgentState = class extends EventEmitter {
     this.emit("status" /* STATUS */);
   }
   // ---------- Tickets ----------
-  updateTrackedTicketIDs(ids) {
-    this.trackedTicketIDs = ids;
+  updateTrackedSessionIDs(ids) {
+    this.trackedSessionIDs = ids;
     this.emit("tickets-changed" /* TICKETS_CHANGED */, ids);
     this.emit("status" /* STATUS */);
   }
@@ -629,7 +629,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 +643,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 +734,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 +747,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 +766,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 +777,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 +797,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 +835,7 @@ var FindFilesTool = class {
     }
     regexStr += "$";
     try {
-      return new RegExp(regexStr).test(filename);
+      return new RegExp(regexStr, caseSensitive ? "" : "i").test(filename);
     } catch {
       return false;
     }
@@ -1740,13 +1744,13 @@ var PRBEAgent = class _PRBEAgent {
     savePersistedData(this.persistedData);
     return newID;
   }
-  get trackedTicketIDs() {
-    return this.persistedData.ticketIds ?? [];
+  get trackedSessionIDs() {
+    return this.persistedData.sessionIds ?? [];
   }
-  set trackedTicketIDs(ids) {
-    this.persistedData.ticketIds = ids;
+  set trackedSessionIDs(ids) {
+    this.persistedData.sessionIds = ids;
     savePersistedData(this.persistedData);
-    this.state.updateTrackedTicketIDs(ids);
+    this.state.updateTrackedSessionIDs(ids);
     this.syncPolling(ids.length > 0);
   }
   get respondedCRIDs() {
@@ -1756,19 +1760,19 @@ var PRBEAgent = class _PRBEAgent {
     this.persistedData.respondedCRIds = Array.from(ids);
     savePersistedData(this.persistedData);
   }
-  syncPolling(hasTickets) {
-    if (this.config.backgroundPolling && hasTickets) {
+  syncPolling(hasSessions) {
+    if (this.config.backgroundPolling && hasSessions) {
       if (this.pollingTimer === null) {
         this.startPolling();
       }
-    } else if (!hasTickets) {
+    } else if (!hasSessions) {
       this.stopPolling();
     }
   }
-  addTrackedTicket(id) {
-    const ids = this.trackedTicketIDs;
+  addTrackedSession(id) {
+    const ids = this.trackedSessionIDs;
     if (!ids.includes(id)) {
-      this.trackedTicketIDs = [...ids, id];
+      this.trackedSessionIDs = [...ids, id];
     }
   }
   // ---------- Constructor ----------
@@ -1826,9 +1830,9 @@ var PRBEAgent = class _PRBEAgent {
     if (config.ipcMain) {
       this.hookRendererLogs(config.ipcMain, config.rendererLogChannel ?? "prbe-renderer-log");
     }
-    const existingTickets = this.trackedTicketIDs;
-    if (existingTickets.length > 0) {
-      this.trackedTicketIDs = existingTickets;
+    const existingSessions = this.trackedSessionIDs;
+    if (existingSessions.length > 0) {
+      this.trackedSessionIDs = existingSessions;
     }
   }
   // ---------- Log integration ----------
@@ -1959,8 +1963,8 @@ var PRBEAgent = class _PRBEAgent {
     );
     this.currentInvestigationSource = "user" /* USER */;
     this.currentCRId = null;
-    if (result?.ticketId) {
-      this.addTrackedTicket(result.ticketId);
+    if (result?.sessionId) {
+      this.addTrackedSession(result.sessionId);
     } else if (!result) {
       if (this.state.isInvestigating) {
         const message = this.userCancelled ? "Investigation cancelled" : "Investigation ended unexpectedly";
@@ -1992,33 +1996,37 @@ var PRBEAgent = class _PRBEAgent {
   }
   /**
    * Poll the backend for context requests on tracked tickets.
+   * Resolves session IDs → ticket IDs first, then polls with ticket IDs.
    */
   async poll() {
-    const ticketIDs = this.trackedTicketIDs;
-    if (ticketIDs.length === 0) return null;
-    const request = {
-      agent_id: this.agentID,
-      ticket_ids: ticketIDs
-    };
+    const sessionIDs = this.trackedSessionIDs;
+    if (sessionIDs.length === 0) return null;
     try {
+      const resolved = await this.resolveSessions(sessionIDs);
+      const returnedSessionIDs = new Set(
+        resolved.tickets.flatMap((t) => t.session_ids)
+      );
+      const resolvedSessionIDs = new Set(
+        resolved.tickets.filter((t) => t.status === "resolved").flatMap((t) => t.session_ids)
+      );
+      const survivingSessions = sessionIDs.filter(
+        (id) => returnedSessionIDs.has(id) && !resolvedSessionIDs.has(id)
+      );
+      if (survivingSessions.length !== sessionIDs.length) {
+        this.trackedSessionIDs = survivingSessions;
+      }
+      const ticketIDs = resolved.tickets.filter((t) => t.status !== "resolved").map((t) => t.ticket_id);
+      if (ticketIDs.length === 0) return { tickets: [] };
+      const request = {
+        agent_id: this.agentID,
+        ticket_ids: ticketIDs
+      };
       const response = await this.post(
         "/api/agent/poll",
         request
       );
-      const returnedIDs = new Set(response.tickets.map((t) => t.ticket_id));
-      const orphaned = ticketIDs.filter((id) => !returnedIDs.has(id));
-      if (orphaned.length > 0) {
-        this.trackedTicketIDs = this.trackedTicketIDs.filter(
-          (id) => !orphaned.includes(id)
-        );
-      }
       const knownCRIDs = /* @__PURE__ */ new Set();
       for (const ticket of response.tickets) {
-        if (ticket.status === "resolved") {
-          this.trackedTicketIDs = this.trackedTicketIDs.filter(
-            (id) => id !== ticket.ticket_id
-          );
-        }
         for (const cr of ticket.context_requests) {
           knownCRIDs.add(cr.id);
           if (!cr.is_active || this.respondedCRIDs.has(cr.id)) continue;
@@ -2044,13 +2052,17 @@ var PRBEAgent = class _PRBEAgent {
     }
   }
   /**
-   * Fetch ticket display info for all tracked tickets.
+   * Fetch ticket display info for all tracked sessions.
+   * Resolves session IDs → ticket IDs first, then fetches ticket info.
    */
   async fetchTicketInfo() {
-    const ticketIDs = this.trackedTicketIDs;
-    if (ticketIDs.length === 0) return [];
-    const request = { ticket_ids: ticketIDs };
+    const sessionIDs = this.trackedSessionIDs;
+    if (sessionIDs.length === 0) return [];
     try {
+      const resolved = await this.resolveSessions(sessionIDs);
+      const ticketIDs = resolved.tickets.map((t) => t.ticket_id);
+      if (ticketIDs.length === 0) return [];
+      const request = { ticket_ids: ticketIDs };
       const response = await this.post(
         "/api/agent/tickets",
         request
@@ -2068,7 +2080,7 @@ var PRBEAgent = class _PRBEAgent {
     }
   }
   resumePolling() {
-    if (this.trackedTicketIDs.length === 0) return;
+    if (this.trackedSessionIDs.length === 0) return;
     this.startPolling();
   }
   /**
@@ -2091,7 +2103,7 @@ var PRBEAgent = class _PRBEAgent {
     this.state.completedInvestigations = [];
     this.state.activeCRs.clear();
     this.state.completedCRs = [];
-    this.state.trackedTicketIDs = [];
+    this.state.trackedSessionIDs = [];
     this.state.ticketInfo = [];
     this.state.emit("status" /* STATUS */);
   }
@@ -2150,8 +2162,8 @@ var PRBEAgent = class _PRBEAgent {
     );
     this.currentInvestigationSource = "user" /* USER */;
     this.currentCRId = null;
-    if (result?.ticketId) {
-      this.addTrackedTicket(result.ticketId);
+    if (result?.sessionId) {
+      this.addTrackedSession(result.sessionId);
     }
   }
   // ---------- WebSocket Investigation ----------
@@ -2328,6 +2340,7 @@ var PRBEAgent = class _PRBEAgent {
             const report = msg.content ?? "";
             const userSummary = msg.metadata?.["user_summary"] ?? "";
             const ticketId2 = msg.metadata?.["ticket_id"];
+            const sessionId = msg.metadata?.["session_id"];
             emit({
               type: "completed" /* COMPLETED */,
               report,
@@ -2335,7 +2348,7 @@ var PRBEAgent = class _PRBEAgent {
               ticketId: ticketId2
             });
             ws.close(1e3, "Complete");
-            finish({ report, userSummary, ticketId: ticketId2 });
+            finish({ report, userSummary, ticketId: ticketId2, sessionId });
             break;
           }
           case "error" /* ERROR */:
@@ -2446,12 +2459,23 @@ var PRBEAgent = class _PRBEAgent {
     this.stopPolling();
     this.pollingTimer = setInterval(() => {
       void this.poll().then(() => {
-        if (this.trackedTicketIDs.length === 0) {
+        if (this.trackedSessionIDs.length === 0) {
           this.stopPolling();
         }
       });
     }, this.config.pollingInterval);
   }
+  // ---------- Session Resolution ----------
+  async resolveSessions(sessionIDs) {
+    const request = {
+      agent_id: this.agentID,
+      session_ids: sessionIDs
+    };
+    return this.post(
+      "/api/agent/resolve-sessions",
+      request
+    );
+  }
   // ---------- Networking ----------
   /**
    * Abort all in-flight fetch requests to prevent orphaned DNS lookups
@@ -2506,7 +2530,7 @@ var DEFAULT_PRBE_STATE = {
   completedInvestigations: [],
   activeCRs: [],
   completedCRs: [],
-  trackedTicketIDs: [],
+  trackedSessionIDs: [],
   ticketInfo: [],
   hasActiveWork: false
 };
@@ -2550,7 +2574,7 @@ function serializePRBEState(state) {
     })),
     activeCRs: Array.from(state.activeCRs.values()).map(serializeCR),
     completedCRs: state.completedCRs.map(serializeCR),
-    trackedTicketIDs: state.trackedTicketIDs,
+    trackedSessionIDs: state.trackedSessionIDs,
     ticketInfo: state.ticketInfo,
     hasActiveWork: state.hasActiveWork
   };