@diegopetrucci/pi-brrr 0.1.5 → 0.1.7

package/.pi-fleet-tested-version

@@ -1 +1 @@
-0.80.3
+0.80.6

package/README.md

@@ -32,10 +32,10 @@ Then reload pi:
 
 Config files are merged, with project config overriding global config:
 
-- `~/.pi/agent/extensions/brrr.json`
-- `<project>/.pi/brrr.json`
+- `~/<pi-config-dir>/agent/extensions/brrr.json`
+- `<project>/<pi-config-dir>/brrr.json`
 
-Project config is only read after Pi reports that the project is trusted.
+Here `<pi-config-dir>` is Pi's runtime config directory name (`CONFIG_DIR_NAME`; `.pi` by default). Project config is only read after Pi reports that the project is trusted.
 
 The default config expects your webhook in `BRRR_WEBHOOK_URL`:
 

package/index.ts

@@ -5,18 +5,17 @@
  * for more input.
  *
  * Config files (project overrides global):
- * - ~/.pi/agent/extensions/brrr.json
- * - <cwd>/.pi/brrr.json, when the project is trusted
+ * - ~/<pi-config-dir>/agent/extensions/brrr.json
+ * - <cwd>/<pi-config-dir>/brrr.json, when the project is trusted
  */
 
 import { execFile } from "node:child_process";
 import { existsSync, readFileSync } from "node:fs";
 import { basename, join } from "node:path";
 import { promisify } from "node:util";
-import { getAgentDir, type ExtensionAPI, type ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import { CONFIG_DIR_NAME, getAgentDir, type ExtensionAPI, type ExtensionCommandContext, type SessionEntry } from "@earendil-works/pi-coding-agent";
 
 const execFileAsync = promisify(execFile);
-const CONFIG_DIR_NAME = ".pi";
 
 interface BrrrConfig {
 	enabled: boolean;
@@ -180,6 +179,25 @@ function lastAssistantMessage(messages: readonly unknown[]): string | undefined
 	return undefined;
 }
 
+type SessionMessageSource = {
+	buildContextEntries(): SessionEntry[];
+};
+
+function sessionMessages(sessionManager: SessionMessageSource | undefined): readonly unknown[] {
+	if (!sessionManager) return [];
+
+	return sessionManager
+		.buildContextEntries()
+		.flatMap((entry: SessionEntry) => (entry.type === "message" ? [entry.message] : []));
+}
+
+function resolveAssistantMessage(
+	sessionManager: SessionMessageSource | undefined,
+	fallbackMessages: readonly unknown[],
+): string | undefined {
+	return lastAssistantMessage(sessionMessages(sessionManager)) ?? lastAssistantMessage(fallbackMessages);
+}
+
 function truncateMessage(value: string): string {
 	const trimmed = value.trim();
 	if (trimmed.length <= 800) return trimmed;
@@ -237,7 +255,7 @@ export default function brrrExtension(pi: ExtensionAPI) {
 		},
 	});
 
-	pi.on("agent_end", async (event, ctx) => {
+	pi.on("agent_settled", async (event, ctx) => {
 		const config = loadConfig(ctx);
 		if (!config.enabled) return;
 		if (config.onlyWhenInteractive && !ctx.hasUI) return;
@@ -246,7 +264,9 @@ export default function brrrExtension(pi: ExtensionAPI) {
 		if (!webhook || !isBrrrWebhookUrl(webhook)) return;
 		if (await shouldSkipForIdleThreshold(config.idleSeconds)) return;
 
-		const assistantMessage = config.includeLastAssistantMessage ? lastAssistantMessage(event.messages as readonly unknown[]) : undefined;
+		const assistantMessage = config.includeLastAssistantMessage
+			? resolveAssistantMessage(ctx.sessionManager, (event as { messages?: readonly unknown[] }).messages ?? [])
+			: undefined;
 		const message = truncateMessage(assistantMessage || formatTemplate(config.message, ctx.cwd));
 		const result = await sendBrrr(webhook, {
 			title: formatTemplate(config.title, ctx.cwd),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-brrr",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "A pi extension that sends brrr push notifications when pi is ready for input.",
   "keywords": [
     "pi-package",