@darkhunt-security/endpoint-codex 0.9.8 → 0.9.9

package/.codex-plugin/plugin.json

@@ -1,6 +1,6 @@
 {
   "name": "darkhunt-guard",
-  "version": "0.9.8",
+  "version": "0.9.9",
   "hooks": "./hooks/hooks.json",
   "description": "Darkhunt endpoint plugin for Codex CLI \u2014 full session trace capture and PreToolUse guardrails.",
   "author": {

package/dist/bin/backfill.mjs

@@ -18303,6 +18303,10 @@ function summarize(output) {
     return text;
   return `${text.slice(0, TOOL_SUMMARY_LIMIT)}\u2026 [truncated, ${text.length} chars]`;
 }
+function attachmentLabel(record) {
+  const size = record.bytes ? `, ${Math.round(record.bytes / 1024)} KB` : "";
+  return `[${record.mediaType ?? "attachment"}${size}]`;
+}
 var RESUMED_USER_TEXT_LIMIT = 4e3;
 var SessionEmitter = class {
   client;
@@ -18324,8 +18328,12 @@ var SessionEmitter = class {
   /** True while consecutive user records still belong to the same turn. */
   userTurnOpen = false;
   pendingAttachments = [];
+  /** Attachments seen before the trace could be opened knowing the prompt. */
+  deferred = [];
   pendingUsage;
   outputText = [];
+  /** The last thing the assistant said in this pass — the trace's result. */
+  lastAssistantText;
   count = 0;
   constructor(client, options) {
     this.client = client;
@@ -18376,6 +18384,28 @@ var SessionEmitter = class {
       return;
     }
     this.count++;
+    if (record.kind === "attachment")
+      this.pendingAttachments.push(attachmentLabel(record));
+    if (record.kind === "user_message")
+      this.absorbUserTurn(record);
+    if (!this.trace && record.kind === "attachment") {
+      this.deferred.push(record);
+      return;
+    }
+    this.flushDeferred();
+    this.place(record);
+  }
+  /** Open the trace on the turn held back, then replay it. No-op once the trace exists. */
+  flushDeferred() {
+    if (this.deferred.length === 0)
+      return;
+    const held = this.deferred.splice(0);
+    this.ensureTrace(held[0].ts);
+    for (const record of held)
+      this.place(record);
+  }
+  /** Route one record onto its owning trace and emit whatever it is worth. */
+  place(record) {
     const root = this.ensureTrace(record.ts);
     const parentTrace = record.parentUuid ? this.owners.get(record.parentUuid) : void 0;
     let owner = parentTrace ?? root;
@@ -18385,28 +18415,21 @@ var SessionEmitter = class {
         sessionId: this.options.sessionId,
         handoffFrom: [root.handoffToken()],
         startTime: record.ts,
+        // A subagent's first record is the task it was given, and it has already been
+        // absorbed by the time this fires — so the child trace opens with its brief
+        // rather than empty.
+        ...this.lastUserText ? { input: this.lastUserText } : {},
         ...this.options.userId !== void 0 ? { userId: this.options.userId } : {}
       }));
       this.sidechains.push(owner);
     }
     this.owners.set(record.uuid, owner);
     switch (record.kind) {
-      case "user_message": {
+      case "user_message":
         this.endGeneration();
-        const part = [this.pendingAttachments.join(" "), record.text].filter(Boolean).join(" ");
-        this.pendingAttachments = [];
-        if (this.userTurnOpen) {
-          this.lastUserText = [this.lastUserText, part].filter(Boolean).join(" ");
-        } else {
-          this.lastUserText = part;
-          this.userTurnOpen = true;
-          this.pendingToolResults = [];
-        }
         break;
-      }
       case "attachment": {
-        const label = `[${record.mediaType ?? "attachment"}${record.bytes ? `, ${Math.round(record.bytes / 1024)} KB` : ""}]`;
-        this.pendingAttachments.push(label);
+        const label = attachmentLabel(record);
         this.seeded(["attachment", record.uuid], () => owner.span("attachment", {
           startTime: record.ts,
           metadata: {
@@ -18463,8 +18486,26 @@ var SessionEmitter = class {
         break;
     }
   }
+  /**
+   * Fold a user record into the turn in flight.
+   *
+   * Split out of the `user_message` case so it can run before the trace is opened —
+   * see the call site in {@link ingest}.
+   */
+  absorbUserTurn(record) {
+    const part = [this.pendingAttachments.join(" "), record.text].filter(Boolean).join(" ");
+    this.pendingAttachments = [];
+    if (this.userTurnOpen) {
+      this.lastUserText = [this.lastUserText, part].filter(Boolean).join(" ");
+    } else {
+      this.lastUserText = part;
+      this.userTurnOpen = true;
+      this.pendingToolResults = [];
+    }
+  }
   /** Ends everything still open and flushes. Safe to call twice. */
   async finish() {
+    this.flushDeferred();
     this.endGeneration();
     for (const span of this.toolSpans.values())
       span.end();
@@ -18475,7 +18516,10 @@ var SessionEmitter = class {
     if (this.trace) {
       this.trace.update({
         metadata: this.meta,
-        ...this.title !== void 0 ? { name: this.title } : {}
+        ...this.title !== void 0 ? { name: this.title } : {},
+        // The pass's result. Paired with the `input` set when the trace opened, this is
+        // what stops the consumer inferring the trace's I/O from a descendant.
+        ...this.lastAssistantText !== void 0 ? { output: this.lastAssistantText } : {}
       });
       this.trace.end();
       this.trace = void 0;
@@ -18531,6 +18575,10 @@ var SessionEmitter = class {
         name: this.title ?? this.defaultTraceName(),
         sessionId: this.grouping ?? this.options.sessionId,
         startTime: ts,
+        // What the agent was asked. `TraceArgs.input` is construction-only — the SDK's
+        // `update()` carries output alone — which is why `ingest` absorbs the user turn
+        // before it calls this.
+        ...this.lastUserText ? { input: this.lastUserText } : {},
         tags: [this.options.vendor, ...this.options.agentId ? ["subagent"] : []],
         metadata: {
           ...this.meta,
@@ -18596,6 +18644,8 @@ var SessionEmitter = class {
     if (!this.generation)
       return;
     const output = this.outputText.join("\n");
+    if (output)
+      this.lastAssistantText = output;
     this.generation.end({
       ...output ? { outputMessages: [{ role: "assistant", content: output }] } : {},
       ...this.model !== void 0 ? { model: this.model } : {},

package/dist/bin/forwarder.mjs

@@ -18303,6 +18303,10 @@ function summarize(output) {
     return text;
   return `${text.slice(0, TOOL_SUMMARY_LIMIT)}\u2026 [truncated, ${text.length} chars]`;
 }
+function attachmentLabel(record) {
+  const size = record.bytes ? `, ${Math.round(record.bytes / 1024)} KB` : "";
+  return `[${record.mediaType ?? "attachment"}${size}]`;
+}
 var RESUMED_USER_TEXT_LIMIT = 4e3;
 var SessionEmitter = class {
   client;
@@ -18324,8 +18328,12 @@ var SessionEmitter = class {
   /** True while consecutive user records still belong to the same turn. */
   userTurnOpen = false;
   pendingAttachments = [];
+  /** Attachments seen before the trace could be opened knowing the prompt. */
+  deferred = [];
   pendingUsage;
   outputText = [];
+  /** The last thing the assistant said in this pass — the trace's result. */
+  lastAssistantText;
   count = 0;
   constructor(client, options) {
     this.client = client;
@@ -18376,6 +18384,28 @@ var SessionEmitter = class {
       return;
     }
     this.count++;
+    if (record.kind === "attachment")
+      this.pendingAttachments.push(attachmentLabel(record));
+    if (record.kind === "user_message")
+      this.absorbUserTurn(record);
+    if (!this.trace && record.kind === "attachment") {
+      this.deferred.push(record);
+      return;
+    }
+    this.flushDeferred();
+    this.place(record);
+  }
+  /** Open the trace on the turn held back, then replay it. No-op once the trace exists. */
+  flushDeferred() {
+    if (this.deferred.length === 0)
+      return;
+    const held = this.deferred.splice(0);
+    this.ensureTrace(held[0].ts);
+    for (const record of held)
+      this.place(record);
+  }
+  /** Route one record onto its owning trace and emit whatever it is worth. */
+  place(record) {
     const root = this.ensureTrace(record.ts);
     const parentTrace = record.parentUuid ? this.owners.get(record.parentUuid) : void 0;
     let owner = parentTrace ?? root;
@@ -18385,28 +18415,21 @@ var SessionEmitter = class {
         sessionId: this.options.sessionId,
         handoffFrom: [root.handoffToken()],
         startTime: record.ts,
+        // A subagent's first record is the task it was given, and it has already been
+        // absorbed by the time this fires — so the child trace opens with its brief
+        // rather than empty.
+        ...this.lastUserText ? { input: this.lastUserText } : {},
         ...this.options.userId !== void 0 ? { userId: this.options.userId } : {}
       }));
       this.sidechains.push(owner);
     }
     this.owners.set(record.uuid, owner);
     switch (record.kind) {
-      case "user_message": {
+      case "user_message":
         this.endGeneration();
-        const part = [this.pendingAttachments.join(" "), record.text].filter(Boolean).join(" ");
-        this.pendingAttachments = [];
-        if (this.userTurnOpen) {
-          this.lastUserText = [this.lastUserText, part].filter(Boolean).join(" ");
-        } else {
-          this.lastUserText = part;
-          this.userTurnOpen = true;
-          this.pendingToolResults = [];
-        }
         break;
-      }
       case "attachment": {
-        const label = `[${record.mediaType ?? "attachment"}${record.bytes ? `, ${Math.round(record.bytes / 1024)} KB` : ""}]`;
-        this.pendingAttachments.push(label);
+        const label = attachmentLabel(record);
         this.seeded(["attachment", record.uuid], () => owner.span("attachment", {
           startTime: record.ts,
           metadata: {
@@ -18463,8 +18486,26 @@ var SessionEmitter = class {
         break;
     }
   }
+  /**
+   * Fold a user record into the turn in flight.
+   *
+   * Split out of the `user_message` case so it can run before the trace is opened —
+   * see the call site in {@link ingest}.
+   */
+  absorbUserTurn(record) {
+    const part = [this.pendingAttachments.join(" "), record.text].filter(Boolean).join(" ");
+    this.pendingAttachments = [];
+    if (this.userTurnOpen) {
+      this.lastUserText = [this.lastUserText, part].filter(Boolean).join(" ");
+    } else {
+      this.lastUserText = part;
+      this.userTurnOpen = true;
+      this.pendingToolResults = [];
+    }
+  }
   /** Ends everything still open and flushes. Safe to call twice. */
   async finish() {
+    this.flushDeferred();
     this.endGeneration();
     for (const span of this.toolSpans.values())
       span.end();
@@ -18475,7 +18516,10 @@ var SessionEmitter = class {
     if (this.trace) {
       this.trace.update({
         metadata: this.meta,
-        ...this.title !== void 0 ? { name: this.title } : {}
+        ...this.title !== void 0 ? { name: this.title } : {},
+        // The pass's result. Paired with the `input` set when the trace opened, this is
+        // what stops the consumer inferring the trace's I/O from a descendant.
+        ...this.lastAssistantText !== void 0 ? { output: this.lastAssistantText } : {}
       });
       this.trace.end();
       this.trace = void 0;
@@ -18531,6 +18575,10 @@ var SessionEmitter = class {
         name: this.title ?? this.defaultTraceName(),
         sessionId: this.grouping ?? this.options.sessionId,
         startTime: ts,
+        // What the agent was asked. `TraceArgs.input` is construction-only — the SDK's
+        // `update()` carries output alone — which is why `ingest` absorbs the user turn
+        // before it calls this.
+        ...this.lastUserText ? { input: this.lastUserText } : {},
         tags: [this.options.vendor, ...this.options.agentId ? ["subagent"] : []],
         metadata: {
           ...this.meta,
@@ -18596,6 +18644,8 @@ var SessionEmitter = class {
     if (!this.generation)
       return;
     const output = this.outputText.join("\n");
+    if (output)
+      this.lastAssistantText = output;
     this.generation.end({
       ...output ? { outputMessages: [{ role: "assistant", content: output }] } : {},
       ...this.model !== void 0 ? { model: this.model } : {},

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@darkhunt-security/endpoint-codex",
-  "version": "0.9.8",
+  "version": "0.9.9",
   "type": "module",
   "description": "Darkhunt endpoint adapter for Codex CLI: hook codec, rollout transcript mapper, plugin manifest.",
   "bin": {