pi-one-ui 0.7.0 → 0.7.1

package/extensions/{features → app/runtime}/flush-docked-bash.ts

@@ -4,7 +4,10 @@
  * 在 handleBashCommand 收尾补一次 flush。
  */
 import { InteractiveMode } from "@earendil-works/pi-coding-agent";
-import { FLUSH_DOCKED_BASH_PATCH, patchRegistry } from "../tools/patch-keys.ts";
+import {
+  FLUSH_DOCKED_BASH_PATCH,
+  patchRegistry,
+} from "../../tools/patch-keys.ts";
 
 type Patch = {
   active: boolean;
@@ -13,7 +16,7 @@ type Patch = {
   installed: (...args: any[]) => unknown;
 };
 
-export function installFlushDockedBash(): void {
+export function installFlushDockedBash(): () => void {
   const prototype = InteractiveMode.prototype as any;
   const previous = patchRegistry.get<Patch>(FLUSH_DOCKED_BASH_PATCH);
   if (previous) previous.active = false;
@@ -33,4 +36,11 @@ export function installFlushDockedBash(): void {
   };
   prototype.handleBashCommand = patch.installed;
   patchRegistry.install(FLUSH_DOCKED_BASH_PATCH, patch);
+  return () => {
+    patch.active = false;
+    if (prototype.handleBashCommand === patch.installed) {
+      prototype.handleBashCommand = patch.original;
+    }
+    patchRegistry.dispose(FLUSH_DOCKED_BASH_PATCH, patch);
+  };
 }

package/extensions/app/runtime/render-scheduler.ts

@@ -1,56 +1,60 @@
 export type RenderRequest = (force: boolean) => void;
 export type RenderSchedule = (run: () => void) => void;
 
-/**
- * Coalesces render requests emitted by multiple layouts in one turn.
- */
+/** 一个运行时对应一个 Pi TUI；组件提供同一界面的刷新入口。 */
 export class RenderScheduler {
-  private readonly requestRender: RenderRequest;
-  private readonly schedule: RenderSchedule;
+  private readonly renderers = new Map<object, RenderRequest>();
+  private generation = 0;
   private scheduled = false;
   private forced = false;
   private disposed = false;
 
-  /**
-   * Creates a scheduler for a concrete render callback.
-   *
-   * @param requestRender Callback that renders the active TUI.
-   * @param schedule Microtask scheduler used to coalesce requests.
-   */
   constructor(
-    requestRender: RenderRequest,
-    schedule: RenderSchedule = queueMicrotask,
-  ) {
-    this.requestRender = requestRender;
-    this.schedule = schedule;
+    private readonly requestRender?: RenderRequest,
+    private readonly schedule: RenderSchedule = queueMicrotask,
+  ) {}
+
+  register(owner: object, render: RenderRequest | undefined): void {
+    if (render) {
+      this.renderers.set(owner, render);
+    } else {
+      this.renderers.delete(owner);
+    }
   }
 
-  /**
-   * Queues one render and preserves forced redraw priority.
-   *
-   * @param force Whether the pending render must be a full redraw.
-   */
   request(force = false): void {
-    if (this.disposed) return;
+    if (this.disposed) {
+      return;
+    }
     this.forced ||= force;
-    if (this.scheduled) return;
-
+    if (this.scheduled) {
+      return;
+    }
     this.scheduled = true;
+    const generation = this.generation;
     this.schedule(() => {
+      if (this.disposed || generation !== this.generation) {
+        return;
+      }
       this.scheduled = false;
-      if (this.disposed) return;
       const forced = this.forced;
       this.forced = false;
-      this.requestRender(forced);
+      const render = this.requestRender ?? this.renderers.values().next().value;
+      render?.(forced);
     });
   }
 
-  /**
-   * Cancels future callbacks and makes later requests no-ops.
-   */
-  dispose(): void {
-    this.disposed = true;
+  /** 清除当前 session 的入口与请求，旧回调不能消费新 session 的请求。 */
+  reset(): void {
+    this.generation += 1;
     this.scheduled = false;
     this.forced = false;
+    this.disposed = false;
+    this.renderers.clear();
+  }
+
+  dispose(): void {
+    this.reset();
+    this.disposed = true;
   }
 }