fix(ssh): align session restore with pi 0.65

Prefer persisted SSH state over env fallback, stop relying on removed session_fork hooks, and add session messages when SSH mode is toggled so the agent sees state changes in conversation context.

Author: tshu-w

config/pi/extensions/ssh.ts

@@ -369,13 +369,14 @@ function getCommandCompletions(currentState: SshState | null, prefix: string): A
   return filtered.length > 0 ? filtered : null;
 }
 
-function findPersistedState(ctx: SessionRestoreContext): SshState | null {
+function findPersistedState(ctx: SessionRestoreContext): { found: boolean; state: SshState | null } {
   const entry = ctx.sessionManager
     .getEntries()
     .filter((item) => item.type === "custom" && item.customType === ENTRY_TYPE)
     .pop();
 
-  return deserializeState(entry?.data);
+  if (!entry) return { found: false, state: null };
+  return { found: true, state: deserializeState(entry.data) };
 }
 
 export default function (pi: ExtensionAPI) {
@@ -421,7 +422,8 @@ export default function (pi: ExtensionAPI) {
   };
 
   const restoreState = (ctx: SessionRestoreContext) => {
-    activeSsh = loadStateFromEnv() ?? findPersistedState(ctx);
+    const persisted = findPersistedState(ctx);
+    activeSsh = persisted.found ? persisted.state : loadStateFromEnv();
     writeStateToEnv(activeSsh);
     updateStatus(ctx);
   };
@@ -486,24 +488,36 @@ export default function (pi: ExtensionAPI) {
 
       if (["off", "disable", "clear"].includes(trimmed)) {
         await applyState(null, ctx, { persist: true });
+        pi.sendMessage({
+          customType: "ssh-state-change",
+          content: "SSH mode disabled. All tool calls (read, write, edit, bash) and user ! commands now execute locally.",
+          display: true,
+        }, { deliverAs: "nextTurn" });
         return;
       }
 
       try {
         const nextState = await resolveSshTarget(trimmed, localCwd);
         await applyState(nextState, ctx, { persist: true });
+        pi.sendMessage({
+          customType: "ssh-state-change",
+          content: `SSH mode enabled: ${nextState.remote}:${nextState.remoteRootCwd}\nAll tool calls (read, write, edit, bash) and user ! commands now execute on this remote host.`,
+          display: true,
+        }, { deliverAs: "nextTurn" });
       } catch (error) {
         const message = error instanceof Error ? error.message : String(error);
         if (ctx.hasUI) ctx.ui.notify(`Failed to enable SSH mode: ${message}`, "error");
       }
     },
   });
 
-  pi.on("session_start", async (_event, ctx) => {
+  pi.on("session_start", async (event, ctx) => {
     const flag = pi.getFlag("ssh");
+    const shouldNotify = !event.reason || event.reason === "startup" || event.reason === "reload";
+
     if (typeof flag === "string" && flag.trim()) {
       try {
-        await applyState(await resolveSshTarget(flag, localCwd), ctx, { notify: true });
+        await applyState(await resolveSshTarget(flag, localCwd), ctx, { notify: shouldNotify });
         return;
       } catch (error) {
         const message = error instanceof Error ? error.message : String(error);
@@ -518,10 +532,6 @@ export default function (pi: ExtensionAPI) {
     restoreState(ctx);
   });
 
-  pi.on("session_fork", async (_event, ctx) => {
-    restoreState(ctx);
-  });
-
   pi.on("user_bash", () => {
     if (!getSsh()) return;
     return { operations: createRemoteBashOps(getSsh) };