83 lines
2.3 KiB
TypeScript
83 lines
2.3 KiB
TypeScript
/**
|
|
* Pi Done Notify Extension
|
|
*
|
|
* Sends a native terminal notification when Pi finishes a prompt
|
|
* and is waiting for input.
|
|
*/
|
|
|
|
import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
|
|
|
|
function windowsToastScript(title: string, body: string): string {
|
|
const type = "Windows.UI.Notifications";
|
|
const mgr = `[${type}.ToastNotificationManager, ${type}, ContentType = WindowsRuntime]`;
|
|
const template = `[${type}.ToastTemplateType]::ToastText01`;
|
|
const toast = `[${type}.ToastNotification]::new($xml)`;
|
|
return [
|
|
`${mgr} > $null`,
|
|
`$xml = [${type}.ToastNotificationManager]::GetTemplateContent(${template})`,
|
|
`$xml.GetElementsByTagName('text')[0].AppendChild($xml.CreateTextNode('${body}')) > $null`,
|
|
`[${type}.ToastNotificationManager]::CreateToastNotifier('${title}').Show(${toast})`,
|
|
].join("; ");
|
|
}
|
|
|
|
function notifyOSC777(title: string, body: string): void {
|
|
process.stdout.write(`\x1b]777;notify;${title};${body}\x07`);
|
|
}
|
|
|
|
function notifyOSC99(title: string, body: string): void {
|
|
process.stdout.write(`\x1b]99;i=1:d=1;${title}\x1b\\`);
|
|
process.stdout.write(`\x1b]99;i=1:p=body;${body}\x1b\\`);
|
|
}
|
|
|
|
const DEFAULT_DURATION_MS = 5000;
|
|
|
|
function notifyWindows(title: string, body: string): void {
|
|
const { execFile } = require("child_process");
|
|
execFile("powershell.exe", [
|
|
"-NoProfile",
|
|
"-Command",
|
|
windowsToastScript(title, body),
|
|
]);
|
|
}
|
|
|
|
function notifyDesktop(
|
|
title: string,
|
|
body: string,
|
|
iconPath: string,
|
|
durationMs = DEFAULT_DURATION_MS
|
|
): void {
|
|
const { execFile } = require("child_process");
|
|
execFile(
|
|
"notify-send",
|
|
["-t", `${durationMs}`, "-i", iconPath, title, body],
|
|
(error: Error | null) => {
|
|
if (error) {
|
|
notifyOSC777(title, body);
|
|
}
|
|
}
|
|
);
|
|
}
|
|
|
|
const ICON_PATH = "/home/thomasgl/.pi/agent/extensions/assets/pi-logo.svg";
|
|
|
|
function notify(title: string, body: string): void {
|
|
if (process.platform === "linux") {
|
|
notifyDesktop(title, body, ICON_PATH);
|
|
return;
|
|
}
|
|
if (process.env.WT_SESSION) {
|
|
notifyWindows(title, body);
|
|
} else if (process.env.KITTY_WINDOW_ID) {
|
|
notifyOSC99(title, body);
|
|
} else {
|
|
notifyOSC777(title, body);
|
|
}
|
|
}
|
|
|
|
export default function (pi: ExtensionAPI) {
|
|
pi.on("agent_end", async (_event, ctx) => {
|
|
if (!ctx.hasUI) return;
|
|
notify("Pi", "Done. Ready for input.");
|
|
});
|
|
}
|