@timurproko/a1 0.1.8-dev.ece8cef → 0.2.1-dev.590

package/dist/app/session-shell/prompt-chips.js

@@ -0,0 +1,499 @@
+import { randomBytes } from "node:crypto";
+import path from "node:path";
+import { canonicalizeClipboardImage } from "./clipboard-image.js";
+import { assertImageEncodedSize, assertPromptImages, canonicalPromptChipMatches, ImageAttachmentError, replaceCanonicalPromptChips, } from "../../contracts/owned-ui/index.js";
+import { ImagePreparationClient } from "./image-preparation-client.js";
+import { preparePasteText } from "./paste-text-preparation.js";
+import { pathChipTag } from "./path-chip-presentation.js";
+import { PastePreparationClient } from "./paste-preparation-client.js";
+const URL_SUBSTRING_PATTERN = /https?:\/\/[^\s\u0000-\u001f\u007f\]]+/giu;
+const IMAGE_CHIP_IDENTIFIER_PATTERN = /^\[📷 screenshot-([a-f0-9]+)(?:-resized)?\]$/u;
+const URL_DISPLAY_LENGTH = 40;
+function imageChipIdentifier(tag) {
+    const match = IMAGE_CHIP_IDENTIFIER_PATTERN.exec(tag);
+    return match === null ? null : match[1] ?? null;
+}
+/** Owns semantic chips and bounded pending paste references; cancels background work on reset or disposal. */
+export class PromptChipStore {
+    #chips = new Map();
+    // Invariant: clearing the editor never recycles a recoverable text chip's identity.
+    #textCounter = 0;
+    #pending = new Map();
+    #preparation = new ImagePreparationClient();
+    #stopping = new Set();
+    #isolated;
+    #provisionalOwners = new Map();
+    #ownedChipTags = new Map();
+    constructor(options = {}) {
+        this.#isolated = options.isolated
+            ? new PastePreparationClient({ ...options.preparation, ...(options.onEvent === undefined ? {} : { onEvent: options.onEvent }) })
+            : undefined;
+    }
+    /** Forks the spare paste helper so the session's first paste does not pay for helper startup. */
+    warm() { this.#isolated?.warm(); }
+    beginPaste(currentText, read, onError, onImage = () => { }) {
+        if (this.#isolated)
+            return this.#beginIsolatedPaste(currentText, typeof read === "function" ? { kind: "provided", read } : read, onError, onImage);
+        if (typeof read !== "function")
+            throw new Error("Clipboard sources require isolated paste preparation");
+        const id = randomBytes(5).toString("hex");
+        // Invariant: reserve paste identity immediately, but only display a screenshot after identifying an image.
+        const marker = `[📷 screenshot-${id}]`;
+        const job = this.#preparation.start(async (signal) => {
+            const content = await read(signal);
+            entry.kind = content?.kind === "image" ? "image" : "text";
+            if (content?.kind === "image") {
+                onImage();
+                if (this.#imageCount(currentText) >= 8)
+                    throw new ImageAttachmentError("image-count");
+            }
+            return content;
+        });
+        const entry = { marker, job, references: 0, kind: "unknown", completion: job.result.then(content => {
+                if (content === null)
+                    return "";
+                if (content.kind === "image") {
+                    return this.#addPreparedImage(content, id);
+                }
+                return this.transformPastedContent(content);
+            }).catch(error => {
+                entry.error = error instanceof ImageAttachmentError ? error : new ImageAttachmentError("image-codec");
+                if (entry.error.code !== "image-canceled" && entry.references === 0)
+                    onError(entry.error);
+                return marker.replace("screenshot-", "failed-");
+            }).then(replacement => {
+                entry.replacement = replacement;
+                return replacement;
+            }) };
+        this.#pending.set(marker, entry);
+        return { marker, result: entry.completion };
+    }
+    #beginIsolatedPaste(currentText, source, onError, onImage) {
+        const id = randomBytes(5).toString("hex");
+        const marker = `[📷 screenshot-${id}]`;
+        let entry;
+        const owner = Symbol("paste");
+        this.#ownedChipTags.set(owner, new Set());
+        const job = this.#isolated.start(source, async (value, signal) => {
+            if (signal.aborted)
+                throw new ImageAttachmentError("image-canceled");
+            if (value === null)
+                return "";
+            if (value.kind === "image")
+                return this.#addPreparedImage(value, id, owner);
+            entry.kind = "text";
+            return this.#adoptPreparedText(value, signal, owner);
+        }, () => {
+            entry.kind = "image";
+            onImage();
+            if (this.#imageCount(currentText) >= 8)
+                throw new ImageAttachmentError("image-count");
+        }, () => { entry.kind = "text"; });
+        entry = { marker, job, references: 0, kind: "unknown", completion: job.result.catch(error => {
+                this.#finishChipOwnership(owner, false);
+                entry.error = error instanceof ImageAttachmentError ? error : new ImageAttachmentError("paste-unavailable");
+                // Compatibility: typed image validation can fail before transfer identifies the payload to this store.
+                if (entry.kind === "unknown" && entry.error.code.startsWith("image-") && entry.error.code !== "image-canceled")
+                    entry.kind = "image";
+                if (entry.error.code !== "image-canceled" && entry.references === 0)
+                    onError(entry.error);
+                return entry.kind === "image" ? marker.replace("screenshot-", "failed-") : "";
+            }).then(replacement => { entry.replacement = replacement; return replacement; }) };
+        entry.onComplete = () => {
+            this.#finishChipOwnership(owner, job.isCurrent() && entry.error === undefined);
+            job.complete();
+        };
+        this.#pending.set(marker, entry);
+        return { marker, result: entry.completion, complete: entry.onComplete, isCurrent: () => job.isCurrent() };
+    }
+    async #adoptPreparedText(value, signal, owner) {
+        if (value.kind !== "paths")
+            return this.#recordPreparedText(value, owner);
+        let text = "";
+        for (let start = 0; start < value.paths.length; start += 32) {
+            if (signal.aborted)
+                throw new ImageAttachmentError("image-canceled");
+            text += this.#recordPreparedText({ kind: "paths", paths: value.paths.slice(start, start + 32) }, owner);
+            await new Promise(resolve => setImmediate(resolve));
+        }
+        return text;
+    }
+    hasPending(text) {
+        return [...this.#pending.values()].some(item => text.includes(item.marker) && item.replacement === undefined);
+    }
+    async waitForPastes(text, signal, readDraft = () => "") {
+        const entries = [...this.#pending.values()].filter(item => text.includes(item.marker));
+        for (const entry of entries)
+            entry.references++;
+        try {
+            await new Promise((resolve, reject) => {
+                const abort = () => reject(new ImageAttachmentError("image-canceled"));
+                if (signal.aborted) {
+                    abort();
+                    return;
+                }
+                signal.addEventListener("abort", abort, { once: true });
+                void Promise.all(entries.map(item => item.completion)).then(() => {
+                    signal.removeEventListener("abort", abort);
+                    resolve();
+                }, reject);
+            });
+            return this.prepareSubmission(text);
+        }
+        finally {
+            for (const entry of entries) {
+                entry.references--;
+                if (entry.replacement !== undefined)
+                    entry.onComplete?.();
+                if (signal.aborted && entry.references === 0 && !readDraft().includes(entry.marker))
+                    entry.job.cancel();
+            }
+        }
+    }
+    reconcileDraft(text) {
+        for (const entry of this.#pending.values()) {
+            if (entry.references === 0 && !text.includes(entry.marker) && !text.includes(entry.marker.replace("screenshot-", "failed-"))) {
+                entry.job.cancel();
+                if (entry.replacement !== undefined) {
+                    entry.onComplete?.();
+                    this.#pending.delete(entry.marker);
+                }
+            }
+        }
+    }
+    resetPastes(text) {
+        let remaining = text;
+        for (const entry of this.#pending.values())
+            if (entry.replacement === undefined)
+                remaining = remaining.replaceAll(entry.marker, "");
+        this.#isolated?.reset();
+        const stopped = this.#preparation.dispose();
+        this.#stopping.add(stopped);
+        void stopped.finally(() => this.#stopping.delete(stopped));
+        this.#preparation = new ImagePreparationClient();
+        for (const entry of this.#pending.values())
+            if (entry.replacement === undefined)
+                entry.job.cancel();
+        return remaining;
+    }
+    async dispose() {
+        await Promise.all([this.#preparation.dispose(), this.#isolated?.dispose(), ...this.#stopping]);
+        this.#chips.clear();
+        this.#pending.clear();
+        this.#provisionalOwners.clear();
+        this.#ownedChipTags.clear();
+    }
+    #imageCount(text) {
+        const tags = new Set([...this.#chips.values()].filter(chip => chip.kind === "image" && text.includes(chip.tag)).map(chip => chip.tag));
+        for (const entry of this.#pending.values()) {
+            if (entry.kind !== "text" && (text.includes(entry.marker) || text.includes(entry.marker.replace("screenshot-", "failed-")))) {
+                // Invariant: a ready screenshot can retain its pending label but still occupies only one slot.
+                tags.add(entry.replacement ?? entry.marker);
+            }
+        }
+        return tags.size;
+    }
+    #addPreparedImage(image, id, owner) {
+        const tag = `[📷 screenshot-${id}${image.transformed ? "-resized" : ""}]`;
+        const attachment = Object.freeze({ type: "image", data: image.data, mimeType: image.mimeType });
+        assertPromptImages([attachment]);
+        this.#chips.set(tag, { kind: "image", tag, image: attachment });
+        this.#claimChip(tag, owner, true);
+        return tag;
+    }
+    transformPastedContent(content, currentText = "") {
+        if (content.kind === "image") {
+            assertImageEncodedSize(content.data);
+            const image = canonicalizeClipboardImage(content);
+            if (image === null)
+                return "";
+            assertPromptImages([...this.prepareSubmission(currentText).images, { type: "image", ...image }]);
+            const id = randomBytes(5).toString("hex");
+            const tag = `[📷 screenshot-${id}]`;
+            this.#chips.set(tag, {
+                kind: "image",
+                tag,
+                image: { type: "image", data: image.data, mimeType: image.mimeType },
+            });
+            return tag;
+        }
+        const text = content.text;
+        if (this.#chips.has(text.trim())) {
+            this.#claimChip(text.trim(), undefined, false);
+            return text.trim();
+        }
+        return this.#recordPreparedText(preparePasteText(text));
+    }
+    #recordPreparedText(paste, owner) {
+        if (paste.kind === "url")
+            return this.#recordUnique({ kind: "url", tag: `[🔗 ${paste.label}]`, label: paste.label, url: paste.url }, owner);
+        if (paste.kind === "text") {
+            if (paste.label === undefined) {
+                const tag = paste.text.trim();
+                if (this.#chips.has(tag)) {
+                    this.#claimChip(tag, owner, false);
+                    return tag;
+                }
+                return paste.text;
+            }
+            const tag = `[paste #${++this.#textCounter} ${paste.label}]`;
+            this.#chips.set(tag, Object.freeze({ kind: "text", tag, text: paste.text }));
+            this.#claimChip(tag, owner, true);
+            return tag;
+        }
+        return paste.paths.map(item => this.#recordUnique({ kind: item.kind, tag: pathChipTag(item), path: item.fullPath }, owner)).join("");
+    }
+    /** Hide provisional clipboard identities until the read identifies an actual image. */
+    hiddenRanges(line) {
+        const ranges = [];
+        for (const entry of this.#pending.values()) {
+            if (entry.kind === "image")
+                continue;
+            let from = 0;
+            while ((from = line.indexOf(entry.marker, from)) >= 0) {
+                ranges.push({ start: from, end: from + entry.marker.length });
+                from += entry.marker.length;
+            }
+        }
+        return ranges.sort((left, right) => left.start - right.start);
+    }
+    atomicRanges(line) {
+        return canonicalPromptChipMatches(line)
+            .filter(match => !match.text.startsWith("[paste #") || this.#chips.has(match.text))
+            .map(match => ({ start: match.start, end: match.end }));
+    }
+    hyperlinkRanges(text) {
+        const ranges = [];
+        for (const chip of this.#chips.values()) {
+            if (chip.kind !== "url")
+                continue;
+            const labelOffset = chip.tag.indexOf(chip.label);
+            if (labelOffset < 0)
+                continue;
+            let searchFrom = 0;
+            while (searchFrom <= text.length - chip.tag.length) {
+                const tagStart = text.indexOf(chip.tag, searchFrom);
+                if (tagStart < 0)
+                    break;
+                const start = tagStart + labelOffset;
+                ranges.push({ start, end: start + chip.label.length, target: chip.url });
+                searchFrom = tagStart + chip.tag.length;
+            }
+        }
+        return ranges.sort((left, right) => right.start - left.start);
+    }
+    expandCopiedText(text) {
+        return this.#replaceResolvable(text, false).text;
+    }
+    /**
+     * Recall-safe history text. Text / URL / file / folder chips expand to their resolved value
+     * so a restart can reuse them without the originating chip cache. Image chip tags are
+     * preserved verbatim so the sidecar-backed rehydration path can re-render them into live
+     * chips carrying the original attachment bytes.
+     */
+    prepareHistoryText(text) {
+        return this.#replaceResolvable(text, false, false).text.trim();
+    }
+    /**
+     * Extract the image chip attachments referenced by an editor draft in occurrence order,
+     * paired with their sidecar identifier. Callers use this to persist image sidecars before a
+     * history row commits, so a fresh process can rehydrate the same chip layout. Unresolved
+     * pending pastes and non-image chips are skipped; caller is responsible for waiting on
+     * pending readiness before this point.
+     */
+    imageChipAttachments(text) {
+        const results = [];
+        const seen = new Set();
+        for (const match of canonicalPromptChipMatches(text)) {
+            const tag = match.text;
+            if (seen.has(tag))
+                continue;
+            const chip = this.#chips.get(tag);
+            if (chip === undefined || chip.kind !== "image")
+                continue;
+            const identifier = imageChipIdentifier(tag);
+            if (identifier === null)
+                continue;
+            seen.add(tag);
+            results.push({ id: identifier, tag, image: chip.image });
+        }
+        return results;
+    }
+    /**
+     * Rehydrate a durable recall value into an editor-ready draft: image chip tags are looked up
+     * against a caller-supplied sidecar resolver, and detectable URL / file / folder / large text
+     * substrings are re-registered as atomic chips through the same identity rules the paste path
+     * uses. Image chips whose sidecar returns null are silently stripped without a placeholder.
+     *
+     * Every chip returned is registered in the session `#chips` map so atomic-range,
+     * hyperlink-range, and submission-expansion paths behave identically to a freshly typed draft.
+     * Callers should treat the returned string as programmatic editor text, not another draft to
+     * scan.
+     */
+    rehydrateHistoryText(text, resolveImage) {
+        if (!text)
+            return text;
+        // Rationale: resolve image chip tags in-place before further classification. Sidecar hits
+        // register a live image chip; misses silently strip the tag (no placeholder, no notice) per
+        // the durability contract.
+        const resolvedImages = replaceCanonicalPromptChips(text, match => {
+            const identifier = imageChipIdentifier(match.text);
+            if (identifier === null)
+                return match.text;
+            const attachment = resolveImage(identifier);
+            if (attachment === null)
+                return "";
+            this.#chips.set(match.text, { kind: "image", tag: match.text, image: attachment });
+            return match.text;
+        });
+        // Rationale: split around any surviving image tags so paste-time classification runs on the
+        // intervening prose exactly as it would on a fresh clipboard payload, plus a substring URL
+        // scan so embedded links become atomic chips even in the middle of surrounding text.
+        const segments = [];
+        let cursor = 0;
+        for (const match of canonicalPromptChipMatches(resolvedImages)) {
+            if (match.start > cursor)
+                segments.push(resolvedImages.slice(cursor, match.start));
+            segments.push(match.text);
+            cursor = match.end;
+        }
+        if (cursor < resolvedImages.length)
+            segments.push(resolvedImages.slice(cursor));
+        return segments.map(segment => {
+            if (segment.length === 0 || canonicalPromptChipMatches(segment).length > 0)
+                return segment;
+            // Invariant: preserve original whitespace framing so the concatenated recall value
+            // remains stable across rehydration.
+            const leading = segment.match(/^\s+/)?.[0] ?? "";
+            const trailing = segment.match(/\s+$/)?.[0] ?? "";
+            const body = segment.slice(leading.length, segment.length - trailing.length);
+            if (body.length === 0)
+                return segment;
+            // Rationale: substring URL rehydration reuses paste-time URL classification and identity
+            // rules, producing a URL chip whenever a bare http(s) URL appears in the prose. Whole-text
+            // classification runs afterwards to catch path / large-paste candidates that consume the
+            // entire remaining body.
+            const withUrlChips = this.#rehydrateUrlSubstrings(body);
+            if (withUrlChips !== body)
+                return `${leading}${withUrlChips}${trailing}`;
+            const classified = this.transformPastedContent({ kind: "text", text: body });
+            return `${leading}${classified}${trailing}`;
+        }).join("");
+    }
+    #rehydrateUrlSubstrings(body) {
+        let output = "";
+        let cursor = 0;
+        URL_SUBSTRING_PATTERN.lastIndex = 0;
+        for (;;) {
+            const match = URL_SUBSTRING_PATTERN.exec(body);
+            if (match === null)
+                break;
+            const url = match[0];
+            if (match.index > cursor)
+                output += body.slice(cursor, match.index);
+            const label = url.length <= URL_DISPLAY_LENGTH ? url : `${url.slice(0, URL_DISPLAY_LENGTH)}…`;
+            output += this.#recordUnique({ kind: "url", tag: `[🔗 ${label}]`, label, url });
+            cursor = match.index + url.length;
+        }
+        URL_SUBSTRING_PATTERN.lastIndex = 0;
+        if (cursor === 0)
+            return body;
+        if (cursor < body.length)
+            output += body.slice(cursor);
+        return output;
+    }
+    prepareSubmission(text) {
+        const expanded = this.#replaceResolvable(text, true);
+        return { text: expanded.text, images: expanded.images };
+    }
+    #replaceResolvable(text, includeImages, omitImages = false) {
+        const images = [];
+        const seenImages = new Set();
+        const resolveChip = (tag) => {
+            const chip = this.#chips.get(tag);
+            if (chip === undefined)
+                return tag;
+            if (chip.kind === "text")
+                return chip.text;
+            if (chip.kind === "image") {
+                if (includeImages && !seenImages.has(tag)) {
+                    images.push(chip.image);
+                    seenImages.add(tag);
+                }
+                return omitImages ? "" : tag;
+            }
+            return chip.kind === "url" ? chip.url : chip.path;
+        };
+        // Invariant: scan only draft tokens (and resolved reservation tokens), never emitted payloads.
+        const expanded = replaceCanonicalPromptChips(text, match => {
+            const tag = match.text;
+            const entry = this.#pending.get(tag) ?? this.#pending.get(tag.replace("failed-", "screenshot-"));
+            if (entry === undefined)
+                return resolveChip(tag);
+            if (includeImages && entry.error !== undefined)
+                throw entry.error;
+            if (includeImages && entry.replacement === undefined)
+                throw new ImageAttachmentError("image-pending");
+            return entry.replacement === undefined ? tag : replaceCanonicalPromptChips(entry.replacement, replacement => resolveChip(replacement.text));
+        });
+        return { text: expanded, images };
+    }
+    #claimChip(tag, owner, created) {
+        if (owner === undefined) {
+            this.#provisionalOwners.delete(tag);
+            return;
+        }
+        let owners = this.#provisionalOwners.get(tag);
+        if (!owners && !created)
+            return; // Invariant: existing committed chips must survive canceled later pastes.
+        if (!owners) {
+            owners = new Set();
+            this.#provisionalOwners.set(tag, owners);
+        }
+        owners.add(owner);
+        this.#ownedChipTags.get(owner)?.add(tag);
+    }
+    #finishChipOwnership(owner, commit) {
+        for (const tag of this.#ownedChipTags.get(owner) ?? []) {
+            const owners = this.#provisionalOwners.get(tag);
+            if (!owners?.has(owner))
+                continue;
+            if (commit)
+                this.#provisionalOwners.delete(tag);
+            else {
+                owners.delete(owner);
+                if (owners.size === 0) {
+                    this.#provisionalOwners.delete(tag);
+                    this.#chips.delete(tag);
+                }
+            }
+        }
+        this.#ownedChipTags.delete(owner);
+    }
+    #recordUnique(chip, owner) {
+        const existing = this.#chips.get(chip.tag);
+        if (existing === undefined || sameChipValue(existing, chip)) {
+            this.#chips.set(chip.tag, chip);
+            this.#claimChip(chip.tag, owner, existing === undefined);
+            return chip.tag;
+        }
+        const suffix = randomBytes(2).toString("hex");
+        const uniqueTag = `${chip.tag.slice(0, -1)} #${suffix}]`;
+        const unique = { ...chip, tag: uniqueTag };
+        this.#chips.set(uniqueTag, unique);
+        this.#claimChip(uniqueTag, owner, true);
+        return uniqueTag;
+    }
+}
+function sameChipValue(left, right) {
+    if (left.kind !== right.kind)
+        return false;
+    if (left.kind === "url" && right.kind === "url")
+        return left.url === right.url;
+    if ((left.kind === "file" || left.kind === "folder") && (right.kind === "file" || right.kind === "folder")) {
+        return path.normalize(left.path) === path.normalize(right.path);
+    }
+    if (left.kind === "image" && right.kind === "image")
+        return left.image.data === right.image.data;
+    return false;
+}

package/dist/app/session-shell/prompt-history-controller.d.ts

@@ -0,0 +1,36 @@
+import { type PromptHistoryKind, type PromptHistoryPort } from "../../contracts/owned-ui/index.js";
+import type { PiShellEditorPort } from "../../integrations/pi/components/index.js";
+/** Merges bounded recall snapshots; the editor alone owns browsing, draft, cursor, and undo state. */
+export declare class PromptHistoryController {
+    #private;
+    readonly options: {
+        editor: PiShellEditorPort;
+        store: PromptHistoryPort;
+        limit: number;
+        fallback: readonly string[];
+        active(): boolean;
+        render(): void;
+        /** Applied to every recall text before it enters the editor recall list. Callers use this
+         * to re-classify durable chip content back into atomic chips; the identity map returned by
+         * default keeps historical behavior for callers that opt out. */
+        rehydrate?: (text: string) => string;
+    };
+    constructor(options: {
+        editor: PiShellEditorPort;
+        store: PromptHistoryPort;
+        limit: number;
+        fallback: readonly string[];
+        active(): boolean;
+        render(): void;
+        /** Applied to every recall text before it enters the editor recall list. Callers use this
+         * to re-classify durable chip content back into atomic chips; the identity map returned by
+         * default keeps historical behavior for callers that opt out. */
+        rehydrate?: (text: string) => string;
+    });
+    start(): void;
+    capture(text: string, kind: PromptHistoryKind, cwd: string, sessionId: string): void;
+    rememberRecovery(text: string): void;
+    synchronize(): void;
+    reset(fallback: readonly string[]): void;
+    close(): Promise<boolean>;
+}