pi-agent-extensions 0.2.0 → 0.2.1

package/extensions/whimsical/index.ts

@@ -1,4 +1,4 @@
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { BOLLYWOOD_MESSAGES, CONTEXT_MESSAGES, PI_TIPS, WHIMSICAL_VERBS, GOODBYE_MESSAGES } from "./messages.js";
 
 type WhimsyMode = 'chaos' | 'classic' | 'bollywood' | 'geek';
@@ -86,30 +86,14 @@ export default function (pi: ExtensionAPI) {
   });
 
   // Register /exit and /bye
-  const performExit = (ctx: ExtensionContext) => {
-    // 1. Try official shutdown
-    ctx.shutdown();
-
-    // 2. Set a watchdog to force exit if graceful shutdown hangs
-    setTimeout(() => {
-      // Force manual cleanup in case Pi didn't get to it
-      if (process.stdin.isTTY) {
-        process.stdin.setRawMode(false);
-      }
-      // Reset cursor, disable mouse tracking, restore buffer
-      process.stdout.write('\x1b[?25h\x1b[?1000l\x1b[?1002l\x1b[?1015l\x1b[?1006l\x1b[?1049l');
-      
-      // Force kill
-      process.exit(0);
-    }, 200);
-  };
-
   pi.registerCommand("exit", {
     description: "Exit Pi with a whimsical goodbye",
     handler: async (_args, ctx) => {
       const msg = GOODBYE_MESSAGES[Math.floor(Math.random() * GOODBYE_MESSAGES.length)];
-      console.log(`\n👋 ${msg}\n`);
-      performExit(ctx);
+      if (ctx.hasUI) {
+        ctx.ui.notify(`👋 ${msg}`, "info");
+      }
+      ctx.shutdown();
     },
   });
 
@@ -117,8 +101,10 @@ export default function (pi: ExtensionAPI) {
     description: "Exit Pi with a whimsical goodbye (alias)",
     handler: async (_args, ctx) => {
       const msg = GOODBYE_MESSAGES[Math.floor(Math.random() * GOODBYE_MESSAGES.length)];
-      console.log(`\n👋 ${msg}\n`);
-      performExit(ctx);
+      if (ctx.hasUI) {
+        ctx.ui.notify(`👋 ${msg}`, "info");
+      }
+      ctx.shutdown();
     },
   });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-extensions",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Collection of extensions for pi coding agent",
   "type": "module",
   "repository": {