@prbe.ai/electron-sdk 0.1.6 → 0.1.8

package/dist/index.mjs

@@ -1,7 +1,7 @@
 // src/agent.ts
-import * as fs2 from "fs";
-import * as path4 from "path";
-import * as os from "os";
+import * as fs3 from "fs";
+import * as path5 from "path";
+import * as os2 from "os";
 import { randomUUID as randomUUID5 } from "crypto";
 
 // src/models.ts
@@ -180,7 +180,7 @@ var PRBEAgentState = class extends EventEmitter {
       this.emit("status" /* STATUS */);
     }
   }
-  completeInvestigation(report, summary) {
+  completeInvestigation(report, summary, ticketId) {
     if (this.events.length > 0) {
       this.events[this.events.length - 1].isCompleted = true;
     }
@@ -190,6 +190,9 @@ var PRBEAgentState = class extends EventEmitter {
       query: this.currentQuery,
       report,
       summary,
+      ticketId,
+      events: [...this.events],
+      resolvedInteractions: [...this.resolvedInteractions],
       completedAt: /* @__PURE__ */ new Date()
     });
     this.report = report;
@@ -265,11 +268,12 @@ var PRBEAgentState = class extends EventEmitter {
     }
   }
   // ---------- CR state mutations ----------
-  beginCR(id, query, slug) {
+  beginCR(id, query, slug, ticketId) {
     const cr = {
       id,
       query,
       slug,
+      ticketId,
       events: [],
       resolvedInteractions: [],
       isRunning: true,
@@ -395,8 +399,13 @@ var PRBEToolRegistry = class {
 var PRBEClosureTool = class {
   declaration;
   handler;
-  constructor(name, description, parameters, handler) {
-    this.declaration = { name, description, parameters };
+  constructor(name, description, parameters, handler, options) {
+    this.declaration = {
+      name,
+      description,
+      parameters,
+      ...options?.interactive ? { interactive: true } : {}
+    };
     this.handler = handler;
   }
   async execute(args) {
@@ -1499,20 +1508,196 @@ var BashExecuteTool = class {
   }
 };
 
-// src/agent.ts
-function getPersistencePath() {
+// src/history.ts
+import * as fs2 from "fs";
+import * as path4 from "path";
+import * as os from "os";
+import * as crypto from "crypto";
+var HKDF_SALT = Buffer.from("prbe-history-encryption-salt", "utf-8");
+var HKDF_INFO = Buffer.from("prbe-history-v1", "utf-8");
+function getAppDataDir() {
   const appData = process.env["APPDATA"] || (process.platform === "darwin" ? path4.join(os.homedir(), "Library", "Application Support") : path4.join(os.homedir(), ".local", "share"));
-  const dir = path4.join(appData, "prbe-agent");
+  return path4.join(appData, "prbe-agent");
+}
+function getHistoryDir() {
+  const dir = path4.join(getAppDataDir(), "history" /* HISTORY_DIR */);
   if (!fs2.existsSync(dir)) {
     fs2.mkdirSync(dir, { recursive: true });
   }
-  return path4.join(dir, "agent-state.json");
+  return dir;
+}
+function deriveKey(apiKey) {
+  return Buffer.from(
+    crypto.hkdfSync(
+      "sha256",
+      Buffer.from(apiKey, "utf-8"),
+      HKDF_SALT,
+      HKDF_INFO,
+      32 /* KEY_LENGTH */
+    )
+  );
+}
+function encrypt(plaintext, key) {
+  const iv = crypto.randomBytes(12 /* IV_LENGTH */);
+  const cipher = crypto.createCipheriv(
+    "aes-256-gcm" /* ALGORITHM */,
+    key,
+    iv,
+    { authTagLength: 16 /* AUTH_TAG_LENGTH */ }
+  );
+  const encrypted = Buffer.concat([cipher.update(plaintext, "utf-8"), cipher.final()]);
+  const authTag = cipher.getAuthTag();
+  return Buffer.concat([iv, authTag, encrypted]);
+}
+function decrypt(data, key) {
+  const iv = data.subarray(0, 12 /* IV_LENGTH */);
+  const authTag = data.subarray(
+    12 /* IV_LENGTH */,
+    12 /* IV_LENGTH */ + 16 /* AUTH_TAG_LENGTH */
+  );
+  const ciphertext = data.subarray(
+    12 /* IV_LENGTH */ + 16 /* AUTH_TAG_LENGTH */
+  );
+  const decipher = crypto.createDecipheriv(
+    "aes-256-gcm" /* ALGORITHM */,
+    key,
+    iv,
+    { authTagLength: 16 /* AUTH_TAG_LENGTH */ }
+  );
+  decipher.setAuthTag(authTag);
+  return decipher.update(ciphertext) + decipher.final("utf-8");
+}
+var HistoryStore = class {
+  key;
+  constructor(apiKey) {
+    this.key = deriveKey(apiKey);
+  }
+  load() {
+    const investigations = [];
+    const crs = [];
+    try {
+      const dir = getHistoryDir();
+      let files;
+      try {
+        files = fs2.readdirSync(dir);
+      } catch {
+        return { investigations, crs };
+      }
+      for (const filename of files) {
+        try {
+          const filePath = path4.join(dir, filename);
+          const raw = fs2.readFileSync(filePath);
+          const json = decrypt(raw, this.key);
+          if (filename.startsWith("inv-") && filename.endsWith(".json")) {
+            const item = JSON.parse(json);
+            investigations.push({
+              ...item,
+              completedAt: new Date(item.completedAt)
+            });
+          } else if (filename.startsWith("cr-") && filename.endsWith(".json")) {
+            const item = JSON.parse(json);
+            crs.push({
+              ...item,
+              startedAt: new Date(item.startedAt),
+              resolvedInteractions: item.resolvedInteractions ?? []
+            });
+          }
+        } catch {
+          console.warn(`[PRBEAgent] Skipping unreadable history file: ${filename}`);
+        }
+      }
+    } catch {
+    }
+    investigations.sort(
+      (a, b) => b.completedAt.getTime() - a.completedAt.getTime()
+    );
+    crs.sort((a, b) => b.startedAt.getTime() - a.startedAt.getTime());
+    return { investigations, crs };
+  }
+  save(investigations, crs) {
+    try {
+      const dir = getHistoryDir();
+      const desiredFiles = /* @__PURE__ */ new Set();
+      for (const inv of investigations) {
+        const filename = `inv-${inv.id}.json`;
+        desiredFiles.add(filename);
+        const data = {
+          id: inv.id,
+          query: inv.query,
+          report: inv.report,
+          summary: inv.summary,
+          ticketId: inv.ticketId,
+          events: inv.events,
+          resolvedInteractions: inv.resolvedInteractions,
+          completedAt: inv.completedAt.toISOString()
+        };
+        fs2.writeFileSync(
+          path4.join(dir, filename),
+          encrypt(JSON.stringify(data), this.key)
+        );
+      }
+      for (const cr of crs) {
+        const filename = `cr-${cr.id}.json`;
+        desiredFiles.add(filename);
+        const data = {
+          id: cr.id,
+          query: cr.query,
+          slug: cr.slug,
+          ticketId: cr.ticketId,
+          events: cr.events,
+          isRunning: cr.isRunning,
+          isCompleted: cr.isCompleted,
+          isFailed: cr.isFailed,
+          report: cr.report,
+          summary: cr.summary,
+          errorMessage: cr.errorMessage,
+          startedAt: cr.startedAt.toISOString(),
+          pendingInteraction: cr.pendingInteraction,
+          resolvedInteractions: cr.resolvedInteractions ?? []
+        };
+        fs2.writeFileSync(
+          path4.join(dir, filename),
+          encrypt(JSON.stringify(data), this.key)
+        );
+      }
+      try {
+        const existing = fs2.readdirSync(dir);
+        for (const filename of existing) {
+          if (!desiredFiles.has(filename)) {
+            fs2.unlinkSync(path4.join(dir, filename));
+          }
+        }
+      } catch {
+      }
+    } catch {
+      console.error("[PRBEAgent] Failed to save investigation history");
+    }
+  }
+  static clear() {
+    try {
+      const dir = path4.join(getAppDataDir(), "history" /* HISTORY_DIR */);
+      if (fs2.existsSync(dir)) {
+        fs2.rmSync(dir, { recursive: true, force: true });
+      }
+    } catch {
+    }
+  }
+};
+
+// src/agent.ts
+function getPersistencePath() {
+  const appData = process.env["APPDATA"] || (process.platform === "darwin" ? path5.join(os2.homedir(), "Library", "Application Support") : path5.join(os2.homedir(), ".local", "share"));
+  const dir = path5.join(appData, "prbe-agent");
+  if (!fs3.existsSync(dir)) {
+    fs3.mkdirSync(dir, { recursive: true });
+  }
+  return path5.join(dir, "agent-state.json");
 }
 function loadPersistedData() {
   try {
     const filePath = getPersistencePath();
-    if (fs2.existsSync(filePath)) {
-      const raw = fs2.readFileSync(filePath, "utf-8");
+    if (fs3.existsSync(filePath)) {
+      const raw = fs3.readFileSync(filePath, "utf-8");
       return JSON.parse(raw);
     }
   } catch {
@@ -1522,7 +1707,7 @@ function loadPersistedData() {
 function savePersistedData(data) {
   try {
     const filePath = getPersistencePath();
-    fs2.writeFileSync(filePath, JSON.stringify(data, null, 2), "utf-8");
+    fs3.writeFileSync(filePath, JSON.stringify(data, null, 2), "utf-8");
   } catch {
     console.error("[PRBEAgent] Failed to save persisted data");
   }
@@ -1542,6 +1727,7 @@ var PRBEAgent = class _PRBEAgent {
   fetchAbortController = null;
   currentInvestigationSource = "user" /* USER */;
   currentCRId = null;
+  historyStore;
   /** Files flagged during the current tool call — uploaded immediately after the tool returns. */
   pendingFlaggedFiles = [];
   // ---------- Persistence ----------
@@ -1603,6 +1789,10 @@ var PRBEAgent = class _PRBEAgent {
     this.state = new PRBEAgentState();
     this.logCapture = new PRBELogCapture(this.config.maxLogEntries);
     this.persistedData = loadPersistedData();
+    this.historyStore = new HistoryStore(this.config.apiKey);
+    const history = this.historyStore.load();
+    this.state.completedInvestigations = history.investigations;
+    this.state.completedCRs = history.crs;
     const roots = this.config.autoApprovedDirs;
     const requester = this.interactionHandler ? this : void 0;
     const grantedPaths = this.grantedPaths;
@@ -1723,9 +1913,9 @@ var PRBEAgent = class _PRBEAgent {
   /**
    * Register a custom tool that the middleware can invoke during investigations.
    */
-  registerTool(name, description, parameters, handler) {
+  registerTool(name, description, parameters, handler, options) {
     this.registry.register(
-      new PRBEClosureTool(name, description, parameters, handler)
+      new PRBEClosureTool(name, description, parameters, handler, options)
     );
   }
   /**
@@ -1753,7 +1943,8 @@ var PRBEAgent = class _PRBEAgent {
           this.state.appendEvent("Thinking", status.text);
           break;
         case "completed" /* COMPLETED */:
-          this.state.completeInvestigation(status.report, status.userSummary);
+          this.state.completeInvestigation(status.report, status.userSummary, status.ticketId);
+          this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
         case "error" /* ERROR */:
           this.state.failInvestigation(status.message);
@@ -1895,6 +2086,7 @@ var PRBEAgent = class _PRBEAgent {
     this.stopPolling();
     this.persistedData = {};
     savePersistedData(this.persistedData);
+    HistoryStore.clear();
     this.state.resetInvestigation();
     this.state.completedInvestigations = [];
     this.state.activeCRs.clear();
@@ -1909,9 +2101,10 @@ var PRBEAgent = class _PRBEAgent {
   static clearPersistedData() {
     try {
       const filePath = getPersistencePath();
-      if (fs2.existsSync(filePath)) {
-        fs2.unlinkSync(filePath);
+      if (fs3.existsSync(filePath)) {
+        fs3.unlinkSync(filePath);
       }
+      HistoryStore.clear();
     } catch {
     }
   }
@@ -1920,7 +2113,7 @@ var PRBEAgent = class _PRBEAgent {
     const crID = cr.id;
     this.currentInvestigationSource = "context_request" /* CONTEXT_REQUEST */;
     this.currentCRId = crID;
-    this.state.beginCR(crID, cr.query, cr.slug ?? void 0);
+    this.state.beginCR(crID, cr.query, cr.slug ?? void 0, ticketId);
     const emitter = (status) => {
       switch (status.type) {
         case "started" /* STARTED */:
@@ -1939,9 +2132,11 @@ var PRBEAgent = class _PRBEAgent {
           break;
         case "completed" /* COMPLETED */:
           this.state.completeCR(crID, status.report, status.userSummary);
+          this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
         case "error" /* ERROR */:
           this.state.failCR(crID, status.message);
+          this.historyStore.save(this.state.completedInvestigations, this.state.completedCRs);
           break;
       }
     };
@@ -2080,7 +2275,7 @@ var PRBEAgent = class _PRBEAgent {
               const uploadPrefix = this.extractUploadPrefix(uploadBaseUrl);
               const uploadedRefs = [];
               for (const file of this.pendingFlaggedFiles) {
-                const filename = path4.basename(file.originalPath);
+                const filename = path5.basename(file.originalPath);
                 const safeName = encodeURIComponent(filename);
                 const storagePath = `${uploadPrefix}/${safeName}`;
                 uploadedRefs.push({
@@ -2136,7 +2331,8 @@ var PRBEAgent = class _PRBEAgent {
             emit({
               type: "completed" /* COMPLETED */,
               report,
-              userSummary
+              userSummary,
+              ticketId: ticketId2
             });
             ws.close(1e3, "Complete");
             finish({ report, userSummary, ticketId: ticketId2 });
@@ -2319,6 +2515,7 @@ function serializeCR(cr) {
     id: cr.id,
     query: cr.query,
     slug: cr.slug,
+    ticketId: cr.ticketId,
     events: cr.events,
     isRunning: cr.isRunning,
     isCompleted: cr.isCompleted,
@@ -2346,6 +2543,9 @@ function serializePRBEState(state) {
       query: inv.query,
       report: inv.report,
       summary: inv.summary,
+      ticketId: inv.ticketId,
+      events: inv.events,
+      resolvedInteractions: inv.resolvedInteractions,
       completedAt: inv.completedAt.toISOString()
     })),
     activeCRs: Array.from(state.activeCRs.values()).map(serializeCR),