@fitchmultz/pi-subagents 0.49.1 → 0.51.0

package/dist/agent/observe-transcript.js

@@ -0,0 +1,382 @@
+import { recordAt, errorMessage } from "../shared/unknown.js";
+import { actionText, actionHints } from "../shared/tui-action.js";
+import { ConversationJournal, plainText } from "./journal-tail.js";
+import { Card } from "./observe-card.js";
+import { Container, Text } from "./observe-layout.js";
+/** One mounted conversation page. Frozen cards are never mutated while the reader is above live. */
+export class ObserverTranscript extends Container {
+    journal;
+    positions = [];
+    cards = new Map();
+    pages = new Map();
+    expanded = new Set();
+    before;
+    start = 0;
+    stop = 0;
+    entries = [];
+    selected;
+    height = 0;
+    mounted = [];
+    waiting = false;
+    revision = "";
+    display;
+    changed;
+    loaded;
+    paused = false;
+    hydration;
+    active;
+    jobs = new Map();
+    pending = new Map();
+    version = 0;
+    constructor(options) {
+        super();
+        this.journal = new ConversationJournal(options.file, options.sessionId);
+        this.display = options;
+        this.changed = options.changed;
+        this.loaded = options.loaded;
+    }
+    observe(client) {
+        const native = this.journal.native;
+        native.refresh();
+        this.waiting = !native.indexed;
+        if (client && !client.settle((id) => native.byId.has(id))) {
+            this.waiting = true;
+        }
+    }
+    refresh(client) {
+        this.observe(client);
+        const revision = `${this.journal.native.records.length}:${client?.revision ?? -1}:${this.before ?? ""}`;
+        if (revision === this.revision) {
+            return false;
+        }
+        this.revision = revision;
+        return true;
+    }
+    sync(client) {
+        if (!this.refresh(client)) {
+            return false;
+        }
+        const entries = this.journal.entries(client?.leaf);
+        this.entries = entries;
+        const stop = this.boundary(entries);
+        const start = Math.max(0, stop - 24);
+        this.start = start;
+        this.stop = stop;
+        const selected = entries.slice(start, stop);
+        this.clear();
+        if (start > 0) {
+            this.historyRow("↑ Load earlier activity", -1);
+        }
+        this.mounted = [];
+        for (const entry of selected.toReversed()) {
+            this.mounted.unshift(this.card(entry, this.before === undefined && entry === selected.at(-1)));
+        }
+        for (const card of this.mounted) {
+            this.addChild(card);
+        }
+        if (this.before === undefined) {
+            this.live(client);
+        }
+        if (this.before !== undefined) {
+            this.historyRow("↓ Load later activity", 1);
+        }
+        this.evict();
+        return true;
+    }
+    pause() {
+        this.paused = true;
+        this.active?.controller.abort();
+        this.jobs.clear();
+        this.pending.clear();
+    }
+    resume() {
+        if (!this.paused) {
+            return;
+        }
+        this.paused = false;
+        for (const card of this.mounted) {
+            const entry = this.journal.native.byId.get(card.id);
+            const request = this.pages.get(card.id) ?? { requested: 0 };
+            if (entry && (!card.ready || request.requested !== card.page)) {
+                this.load(entry, card, request);
+            }
+        }
+    }
+    card(entry, latest) {
+        const id = entry.value.id;
+        let card = this.cards.get(id);
+        if (!card) {
+            const call = recordAt(entry.value, "message")?.toolCallId;
+            const live = typeof call === "string" ? `tool:${call}` : "";
+            card = this.cards.get(live) ?? new Card(id);
+            this.cards.delete(live);
+            card.id = id;
+            if (!card.tool && !card.assistant) {
+                card.content.addChild(new Text("Loading published content…", 0, 0));
+            }
+            this.cards.set(id, card);
+        }
+        if (!card.ready) {
+            this.load(entry, card, this.pages.get(id) ?? { requested: latest ? -1 : 0 });
+        }
+        return card;
+    }
+    current({ card, request, controller }) {
+        return (!controller.signal.aborted &&
+            this.cards.get(card.id) === card &&
+            this.pending.get(card.id) === request);
+    }
+    load(entry, card, request) {
+        if (this.paused || this.pending.get(card.id) === request) {
+            return;
+        }
+        this.pending.set(card.id, request);
+        this.pages.set(card.id, request);
+        this.jobs.set(card.id, { entry, card, request, controller: new AbortController() });
+        if (this.active?.card === card) {
+            this.active.controller.abort();
+        }
+        this.drain();
+    }
+    drain() {
+        const job = this.jobs.values().next().value;
+        if (this.hydration || this.paused || !job) {
+            return;
+        }
+        this.jobs.delete(job.card.id);
+        const { card } = job;
+        this.active = job;
+        this.hydration = this.hydrate(job)
+            .catch((error) => {
+            if (!this.current(job)) {
+                return;
+            }
+            this.pending.delete(card.id);
+            card.footer.clear();
+            card.footer.addChild(new Text(`Content unavailable: ${plainText(errorMessage(error))}`, 0, 0));
+            card.invalidate();
+            this.version++;
+            this.loaded();
+        })
+            .finally(() => {
+            this.hydration = undefined;
+            this.active = undefined;
+            this.drain();
+        });
+    }
+    async hydrate(job) {
+        if (!this.current(job)) {
+            return;
+        }
+        const { entry, card, request, controller } = job;
+        const options = { signal: controller.signal, hideThinking: this.display.hideThinking };
+        let page = await this.journal.page(entry, Math.max(0, request.requested), options);
+        if (request.requested < 0 && page.pages > 1) {
+            page = await this.journal.page(entry, page.pages - 1, options);
+        }
+        const call = typeof page.message.toolCallId === "string" && !card.tool
+            ? await this.journal.call(page.message.toolCallId, card.id, controller.signal)
+            : undefined;
+        if (!this.current(job)) {
+            return;
+        }
+        this.durable(card.id, page, call?.arguments);
+        this.pages.set(card.id, { requested: page.page });
+        this.pending.delete(card.id);
+        this.version++;
+        if (request.navigation === undefined) {
+            this.loaded();
+        }
+        else {
+            this.changed({ kind: "content", id: card.id, direction: request.navigation });
+        }
+    }
+    evict() {
+        const mounted = new Set(this.mounted.map((card) => card.id));
+        for (const id of this.cards.keys()) {
+            if (!mounted.has(id)) {
+                this.cards.delete(id);
+                this.jobs.delete(id);
+                this.pending.delete(id);
+            }
+        }
+        if (this.active && !mounted.has(this.active.card.id)) {
+            this.active.controller.abort();
+        }
+        for (const choices of [this.pages, this.expanded]) {
+            for (const id of choices.keys()) {
+                if (choices.size > 48 && !mounted.has(id)) {
+                    choices.delete(id);
+                }
+            }
+        }
+    }
+    durable(id, page, args) {
+        const card = this.cards.get(id);
+        if (!card) {
+            return;
+        }
+        card.expanded ||= this.expanded.has(id);
+        card.publish(page, args, this.display);
+        if (page.pages > 1) {
+            card.footer.addChild(new Text(`Content page ${page.page + 1}/${page.pages}`, 0, 0));
+            card.footer.addChild(actionHints([-1, 1].map((direction) => ({
+                text: direction < 0 ? "Previous page" : "Next page",
+                action: () => {
+                    this.changePage(id, direction);
+                },
+            }))));
+        }
+        card.invalidate();
+    }
+    changePage(id, direction) {
+        const card = this.cards.get(id);
+        const entry = this.journal.native.byId.get(id);
+        if (!card || !entry) {
+            return false;
+        }
+        const page = Math.max(0, Math.min(card.pages - 1, card.page + direction));
+        if (page === card.page) {
+            return false;
+        }
+        const request = { requested: page, navigation: direction };
+        this.pages.set(id, request);
+        this.changed({ kind: "pending" });
+        this.load(entry, card, request);
+        return true;
+    }
+    contentPage(line, height, direction) {
+        const positions = direction < 0 ? this.positions.toReversed() : this.positions;
+        return positions.some((item) => {
+            const visible = item.end > line && item.start < line + height;
+            const boundary = direction < 0 ? line <= item.start : line + height >= item.end;
+            return visible && boundary && this.changePage(item.id, direction);
+        });
+    }
+    live(client) {
+        if (!client) {
+            return;
+        }
+        const message = client.message;
+        if (message) {
+            const id = `live:${message.key}`;
+            let card = this.cards.get(id);
+            if (!card) {
+                card = new Card(id);
+                this.cards.set(id, card);
+            }
+            card.stream(message, this.display.hideThinking);
+            this.mounted.push(card);
+            this.addChild(card);
+        }
+        this.pendingCalls(client);
+        for (const value of client.tools.values()) {
+            this.liveTool(value);
+        }
+    }
+    pendingCalls(client) {
+        for (const block of client.message?.blocks.filter(Boolean) ?? []) {
+            if (block.type === "toolCall" &&
+                block.id !== undefined &&
+                block.name !== undefined &&
+                !client.tools.has(block.id)) {
+                this.liveTool({ id: block.id, name: block.name, args: block.args }, true);
+            }
+        }
+    }
+    liveTool(value, preparing = false) {
+        const id = `tool:${value.id}`;
+        let card = this.cards.get(id);
+        if (!card) {
+            card = new Card(id);
+            card.expanded = this.expanded.has(id);
+            this.cards.set(id, card);
+        }
+        card.updateTool(value, this.display, preparing);
+        this.mounted.push(card);
+        this.addChild(card);
+    }
+    boundary(entries) {
+        const found = entries.findIndex((entry) => entry.value.id === this.before);
+        this.before = found < 0 ? undefined : this.before;
+        return this.before === undefined || found < 0 ? entries.length : found;
+    }
+    historyRow(label, direction) {
+        this.addChild(actionText(label, () => {
+            this.changed({ kind: "history", direction });
+        }));
+    }
+    history(direction) {
+        if ((direction < 0 && this.start === 0) || (direction > 0 && this.before === undefined)) {
+            return;
+        }
+        const anchor = direction < 0
+            ? this.mounted.at(0)
+            : this.mounted.findLast((card) => this.journal.native.byId.has(card.id));
+        const stop = direction < 0 ? this.start + 1 : this.stop + 23;
+        this.before = this.entries[stop]?.value.id;
+        return anchor?.id;
+    }
+    latest() {
+        this.before = undefined;
+        const entry = this.entries.at(-1);
+        if (!entry) {
+            return;
+        }
+        const card = this.cards.get(entry.value.id);
+        if (card?.ready === true && card.page === card.pages - 1) {
+            return;
+        }
+        const request = { requested: -1 };
+        this.pages.set(entry.value.id, request);
+        if (card) {
+            this.load(entry, card, request);
+        }
+    }
+    select(line, height) {
+        const position = this.positions.find((item) => item.end > line &&
+            item.start < line + height &&
+            this.cards.get(item.id)?.tool !== undefined);
+        this.selected = position?.id;
+        let changed = false;
+        for (const card of this.mounted) {
+            const selected = card.id === this.selected;
+            if (card.selected !== selected) {
+                card.selected = selected;
+                card.invalidate();
+                changed = true;
+            }
+        }
+        return changed;
+    }
+    toggle() {
+        this.cards.get(this.selected ?? "")?.toggle();
+    }
+    anchor(line) {
+        const position = this.positions.find((item) => item.end > line);
+        return position && { id: position.id, offset: Math.max(0, line - position.start) };
+    }
+    line(anchor) {
+        const position = this.positions.find((item) => item.id === anchor.id);
+        return (anchor.end === true ? (position?.end ?? 0) : (position?.start ?? 0)) + anchor.offset;
+    }
+    render(width) {
+        this.positions.length = 0;
+        let start = 0;
+        for (const child of this.children) {
+            const height = child.render(width).length;
+            if (child instanceof Card) {
+                this.positions.push({ id: child.id, start, end: start + height });
+                if (child.expanded) {
+                    this.expanded.add(child.id);
+                }
+                else {
+                    this.expanded.delete(child.id);
+                }
+            }
+            start += height;
+        }
+        this.height = start;
+        return super.render(width);
+    }
+}