# Files changed · PR \#23048 · pingdotgg/t3code [View on GitCafe](https://git.cafe/pingdotgg/t3code/pull/23048/diffs) Repository: [pingdotgg/t3code](https://git.cafe/pingdotgg/t3code) Visibility: public [refactor\(web\): composer context imports live in useComposerContextImport](https://git.cafe/pingdotgg/t3code/pull/23048.md) Version: 6 Head commit: d7b37b81c32eeb84d581993d1906bd32d907feaa Base commit: 25fa572eba1b8a200711bee4bb5b3acc7e39e5c0 Comparison base: 25fa572eba1b8a200711bee4bb5b3acc7e39e5c0 Showing up to 20 entries from this revision. If the pull request changes, reopen the first page before continuing. ## apps/web/src/components/chat/ChatComposer.tsx Status: modified; +26; −316 ```diff @@ -6,10 +6,8 @@ import { usePrimaryEnvironmentId } from "../../state/environments"; import { useRightPanelStore } from "~/rightPanelStore"; import { importPastedComposerText, readPastedComposerContext } from "../composerInlineTokenPaste"; -import { elementContextToPreviewAnnotation } from "../../lib/elementContext"; import { questionAttachmentDraftId, - countQuestionAttachments, trackOpenQuestionAttachmentDraft, useQuestionAttachmentPreparation, changeQuestionAttachmentPreparation, @@ -179,7 +177,6 @@ import { ComposerContextActionsContext, composerContextRecordsFromDraft, - uploadedContextRecordFromDraft, } from "../composerContextPresentation"; import { useOpenPrLink } from "~/lib/openPullRequestLink"; import { @@ -192,44 +189,26 @@ inlineContextReferenceReplacement, } from "~/lib/composerContextReferences"; import { - asKnownContextRecord, attachmentContextRecord, - composerContextImportLookupIds, - isSameComposerContextPayload, uploadedAttachmentContextRecord, fileContextReference, imageContextReference, previewAnnotationContextId, previewAnnotationContextRecord, - previewAnnotationFromRecord, reviewCommentContextId, reviewCommentContextReference, reviewCommentContextRecord, - reviewCommentFromRecord, - terminalContextDraftFromRecord, terminalContextReference, terminalContextRecord, threadContextRecord, threadContextReference, } from "~/lib/composerContextRecords"; -import { - copiedComposerAttachmentFile, - rememberCopiedComposerAttachments, +import { rememberCopiedComposerAttachments } from "~/lib/copiedComposerAttachments"; -} from "~/lib/copiedComposerAttachments"; import { useThreadContextDrop, threadContextDropTargetProps } from "./threadContextDrag"; import { readThreadShell } from "~/state/entities"; import { requestConfirmDialog } from "~/confirmDialog"; import { encodeComposerContextFragment } from "@t3tools/shared/composerContextClipboard"; -import type { +import type { ComposerContextRecord, ThreadPullRequestLink } from "@t3tools/contracts"; - ComposerContextClipboardFragment, - ComposerContextRecord, - ThreadPullRequestLink, -} from "@t3tools/contracts"; -import { resolveAssetUrl } from "~/assets/assetUrls"; -import { assetEnvironment } from "~/state/assets"; -import { readPreparedConnection } from "~/state/session"; -import { useAtomQueryRunner } from "~/state/use-atom-query-runner"; -import { orchestrationEnvironment } from "~/state/orchestration"; import { threadContextAttachment } from "~/lib/threadContextAttachment"; import { type ComposerCommandItem, @@ -337,7 +316,6 @@ import { useSoftwareKeyboardOpen } from "../../hooks/useSoftwareKeyboardOpen"; import { useStableCallback } from "../../hooks/useStableCallback"; import { usePanelAnimationSettings } from "../../panelAnimations"; -import { useAtomCommand } from "../../state/use-atom-command"; import { readEnvironmentScope } from "../../state/session"; import type { ReviewCommentContext } from "../../reviewCommentContext"; import { useComposerRestingTransition } from "./useComposerRestingTransition"; @@ -347,6 +325,7 @@ import { useComposerProviderModel } from "./useComposerProviderModel"; import { useComposerImageGroups } from "./useComposerImageGroups"; import { useComposerStash } from "./useComposerStash"; +import { useComposerContextImport } from "./useComposerContextImport"; /** Reads and disarms a paste-as-text deadline: the shortcut applies to one paste only. */ function consumePasteAsTextShortcut(deadline: { current: number }): boolean { @@ -355,15 +334,6 @@ return armed; } -async function downloadAttachmentBlob(url: string): Promise { - try { - const response = await fetch(url, { signal: AbortSignal.timeout(60_000) }); - return response.ok ? await response.blob() : null; - } catch { - return null; - } -} - const extendReplacementRangeForTrailingSpace = ( text: string, rangeEnd: number, @@ -1462,9 +1432,6 @@ removeComposerDraftFile(attachmentDraftTarget, fileId); }; - const addComposerDraftTerminalContexts = useComposerDraftStore( - (store) => store.addTerminalContexts, - ); const addComposerDraftReviewComment = useComposerDraftStore((store) => store.addReviewComment); const addComposerDraftPreviewAnnotation = useComposerDraftStore( (store) => store.addPreviewAnnotation, @@ -1524,287 +1491,30 @@ return fragment; }; const buildContextClipboardFragmentStable = useStableCallback(buildContextClipboardFragment); + const { - const createAssetUrl = useAtomQueryRunner(assetEnvironment.createUrl, { reportFailure: false }); - const loadThreadTranscript = useAtomCommand(orchestrationEnvironment.threadTranscript, { - reportFailure: false, + loadThreadTranscript, + countReservedAttachments, + countReservedAttachmentsStable, + importContextRecords, + importContextFragment, + importContextFragmentStable, + } = useComposerContextImport({ + environmentId, + activeThread, + activeThreadId, + composerDraftTarget, + attachmentDraftTarget, + attachmentTargetKey, + attachmentTargetKeyRef, + questionAttachmentTarget, + pendingUserInputs, + composerContextRecords, + composerImages, + composerFilesRef, + pendingImageCompressionsRef, + pendingThreadImportsRef, + addComposerFilesToDraft, }); - const countReservedAttachments = () => { - const questionRequest = pendingUserInputs[0]; - const otherQuestionKeys = - questionAttachmentTarget && questionRequest && activeThreadId - ? questionRequest.questions - .map((question) => - questionAttachmentDraftId( - environmentId, - activeThreadId, - questionRequest.requestId, - question.id, - ), - ) - .filter((key) => key !== questionAttachmentTarget) - : []; - const draft = getComposerDraft(attachmentDraftTarget); - return ( - (draft?.images.length ?? 0) + - (draft?.files.length ?? 0) + - (pendingImageCompressionsRef.current.get(attachmentTargetKey) ?? 0) + - (pendingThreadImportsRef.current.get(attachmentTargetKey) ?? 0) + - countQuestionAttachments(otherQuestionKeys) - ); - }; - const countReservedAttachmentsStable = useStableCallback(countReservedAttachments); - const runAttachmentImport = async ( - record: Extract, - localId: string, - sourceEnvironmentId: EnvironmentId, - importTargetKey: string, - ) => { - const fail = (reason: string) => { - toastManager.add({ - type: "error", - title: `Couldn't bring ${record.name} into this message`, - description: `${reason} Remove the chip or attach the file again.`, - }); - }; - let file = copiedComposerAttachmentFile(sourceEnvironmentId, record.attachmentId); - if (!file) { - const sourceConnection = readPreparedConnection(sourceEnvironmentId); - if (!sourceConnection) { - fail("The environment it came from is not connected."); - return; - } - const result = await createAssetUrl({ - environmentId: sourceEnvironmentId, - input: { resource: { _tag: "attachment", attachmentId: record.attachmentId } }, - }); - const url = - result._tag === "Success" - ? resolveAssetUrl(sourceConnection.httpBaseUrl, result.value.relativeUrl) - : null; - if (!url) { - fail("The original attachment is no longer available."); - return; - } - const blob = await downloadAttachmentBlob(url); - if (!blob) { - fail("Downloading it from the source failed."); - return; - } - file = new File([blob], record.name, { type: record.mimeType || blob.type }); - } - // The draft these bytes belong to may have been sent or switched away from while they - // downloaded. Dropping them here keeps them out of whatever draft is open now. - if (attachmentTargetKeyRef.current !== importTargetKey) return; - const replacesFileMarker = - record.kind === "file" && - composerFilesRef.current.some( - (candidate) => - composerFileNeedsReattach(candidate) && - composerFileMatchesReattachMarker(candidate, { - name: record.name, - mimeType: file.type, - sizeBytes: file.size, - }), - ); - if ( - !replacesFileMarker && - (pendingThreadImportsRef.current.get(importTargetKey) ?? 0) > 0 && - countReservedAttachments() >= PROVIDER_SEND_TURN_MAX_ATTACHMENTS - ) { - fail("The attachment limit has been reached."); - return; - } - if (record.kind === "image") { - const accepted = addComposerDraftImages( - attachmentDraftTarget, - [ - { - type: "image", - id: localId, - name: record.name, - mimeType: file.type, - sizeBytes: file.size, - previewUrl: URL.createObjectURL(file), - file, - }, - ], - { - allowDuplicates: - getComposerDraft(composerDraftTarget)?.previewAnnotations.some( - (annotation) => annotation.id === localId, - ) ?? false, - }, - ); - if (!accepted.includes(localId)) - fail("The draft rejected this attachment (duplicate or attachment limit reached)."); - } else { - const accepted = addComposerFilesToDraft([ - { - type: "file", - id: localId, - name: record.name, - mimeType: file.type, - sizeBytes: file.size, - file, - }, - ]); - if (!accepted.includes(localId)) - fail("The draft rejected this attachment (duplicate or attachment limit reached)."); - } - }; - const importAttachmentRecord = async ( - record: Extract, - localId: string, - sourceEnvironmentId: EnvironmentId, - ) => { - // The chip lands in the draft immediately while these bytes are still downloading. Count - // the transfer against its own draft so a send cannot snapshot a message whose chip has no - // attachment behind it, and so bytes for an abandoned draft never enter the next one. - const importTargetKey = attachmentTargetKey; - pendingDraftWork.begin(importTargetKey); - await runAttachmentImport(record, localId, sourceEnvironmentId, importTargetKey).finally(() => - pendingDraftWork.end(importTargetKey), - ); - }; - const importContextRecords = ( - records: ReadonlyArray, - sourceEnvironmentId: EnvironmentId | null, - threadDraftTarget = attachmentDraftTarget, - ): ReadonlyMap => { - const rewritten = new Map(); - const dependentAttachmentLocalIds = new Map(); - const skippedDependentAttachmentIds = new Set(); - // Resolve annotations before their dependent screenshot records even if a foreign - // clipboard producer emitted the records in a different order. - const orderedRecords = records.toSorted((left, right) => - left.kind === "preview-annotation" && right.kind !== "preview-annotation" - ? -1 - : right.kind === "preview-annotation" && left.kind !== "preview-annotation" - ? 1 - : 0, - ); - for (const candidate of orderedRecords) { - // Producer ids fold into context ids, so two different excerpts can collide. Only skip - // when the draft already holds the same payload; a colliding but different record is - // re-minted under a fresh id so both survive the paste. - const record = asKnownContextRecord(candidate); - if (!record) continue; - // A reminted annotation needs its screenshot under the new annotation id. - const existing = dependentAttachmentLocalIds.has(record.contextId) - ? undefined - : composerContextImportLookupIds(record).flatMap((contextId) => { - const found = composerContextRecords.get(contextId); - return found ? [found] : []; - })[0]; - // Upload completion can change the server id since copy; the local bytes still match. - if ( - sourceEnvironmentId !== null && - (record.kind === "image" || record.kind === "file") && - (existing?.kind === "image" || existing?.kind === "file") && - existing.record.file !== null && - existing.record.file === - copiedComposerAttachmentFile(sourceEnvironmentId, record.attachmentId) - ) - continue; - const existingRecord = - existing?.kind === "terminal" - ? terminalContextRecord(existing.record) - : existing?.kind === "review-comment" - ? reviewCommentContextRecord(existing.record) - : existing?.kind === "preview-annotation" - ? previewAnnotationContextRecord(existing.record, { - screenshotContextId: composerImages.some( - (image) => image.id === existing.record.id, - ) - ? existing.record.id - : undefined, - }) - : existing?.kind === "thread" - ? existing.record - : existing - ? (uploadedContextRecordFromDraft(existing) ?? undefined) - : undefined; - if ( - existingRecord && - isSameComposerContextPayload(existingRecord, record) && - (record.kind !== "thread" || threadDraftTarget === attachmentDraftTarget) - ) { - if (record.kind === "preview-annotation" && record.screenshotContextId) { - skippedDependentAttachmentIds.add(record.screenshotContextId); - } - continue; - } - const conflicts = existing !== undefined; - switch (record.kind) { - case "terminal": { - const threadId = activeThread?.id ?? activeThreadId; - if (!threadId) break; - const imported = terminalContextDraftFromRecord(record, threadId); - const draft = conflicts ? { ...imported, id: randomUUID() } : imported; - addComposerDraftTerminalContexts(composerDraftTarget, [draft], { - appendReference: false, - }); - rewritten.set(record.contextId, terminalContextReference(draft).contextId); - break; - } - case "review-comment": { - const imported = reviewCommentFromRecord(record); - const comment = conflicts ? { ...imported, id: randomUUID() } : imported; - addComposerDraftReviewComment(composerDraftTarget, comment, { - appendReference: false, - }); - rewritten.set(record.contextId, reviewCommentContextId(comment.id)); - break; - } - case "element": - case "preview-annotation": { - const imported = - record.kind === "element" - ? elementContextToPreviewAnnotation(record, randomUUID(), new Date().toISOString()) - : previewAnnotationFromRecord(record); - const annotation = conflicts ? { ...imported, id: randomUUID() } : imported; - if (record.kind === "preview-annotation" && record.screenshotContextId) { - dependentAttachmentLocalIds.set(record.screenshotContextId, annotation.id); - } - addComposerDraftPreviewAnnotation(composerDraftTarget, annotation, { - appendReference: false, - }); - rewritten.set(record.contextId, previewAnnotationContextId(annotation.id)); - break; - } - case "thread": { - // The agent can only read threads on its own server; a pasted foreign one is dropped. - if (record.environmentId !== environmentId) break; - addComposerDraftThreadContexts(threadDraftTarget, [record], { - appendReference: false, - }); - rewritten.set(record.contextId, record.contextId); - break; - } - case "image": - case "file": { - if (skippedDependentAttachmentIds.has(record.contextId)) break; - if (sourceEnvironmentId === null && !conflicts) { - rewritten.set(record.contextId, record.contextId); - break; - } - if (sourceEnvironmentId === null) break; - const localId = dependentAttachmentLocalIds.get(record.contextId) ?? randomUUID(); - rewritten.set(record.contextId, toKindScopedComposerContextId(record.kind, localId)); - void importAttachmentRecord(record, localId, sourceEnvironmentId); - break; - } - default: - break; - } - } - return rewritten; - }; - const importContextFragment = ( - fragment: ComposerContextClipboardFragment, - ): ReadonlyMap => - importContextRecords(fragment.records, fragment.source.environmentId); - const importContextFragmentStable = useStableCallback(importContextFragment); useEffect(() => { if (isLiteralPendingAnswer) return; ``` ## apps/web/src/components/chat/useComposerContextImport.ts Status: added; +398; −0 ```diff @@ -1,0 +1,398 @@ +import type { RefObject } from "react"; +import type { ChatComposerProps } from "./ChatComposer"; +import { + type ComposerContextClipboardFragment, + type ComposerContextRecord, + type EnvironmentId, + PROVIDER_SEND_TURN_MAX_ATTACHMENTS, +} from "@t3tools/contracts"; +import { + type ComposerFileAttachment, + type ComposerImageAttachment, + type DraftId, + composerFileMatchesReattachMarker, + composerFileNeedsReattach, + useComposerDraftStore, +} from "../../composerDraftStore"; +import { type PendingUserInput } from "../../session-logic"; +import { + asKnownContextRecord, + composerContextImportLookupIds, + isSameComposerContextPayload, + previewAnnotationContextId, + previewAnnotationContextRecord, + previewAnnotationFromRecord, + reviewCommentContextId, + reviewCommentContextRecord, + reviewCommentFromRecord, + terminalContextDraftFromRecord, + terminalContextRecord, + terminalContextReference, +} from "~/lib/composerContextRecords"; +import { assetEnvironment } from "~/state/assets"; +import { copiedComposerAttachmentFile } from "~/lib/copiedComposerAttachments"; +import { countQuestionAttachments, questionAttachmentDraftId } from "../../questionAttachments"; +import { elementContextToPreviewAnnotation } from "../../lib/elementContext"; +import { orchestrationEnvironment } from "~/state/orchestration"; +import { pendingDraftWork } from "./pendingDraftWork"; +import { randomUUID } from "~/lib/utils"; +import { readPreparedConnection } from "~/state/session"; +import { resolveAssetUrl } from "~/assets/assetUrls"; +import { toKindScopedComposerContextId } from "@t3tools/shared/composerContextReferences"; +import { toastManager } from "../ui/toast"; +import { + type ComposerDraftContextRecords, + uploadedContextRecordFromDraft, +} from "../composerContextPresentation"; +import { useAtomCommand } from "../../state/use-atom-command"; +import { useAtomQueryRunner } from "~/state/use-atom-query-runner"; +import { useStableCallback } from "../../hooks/useStableCallback"; + +async function downloadAttachmentBlob(url: string): Promise { + try { + const response = await fetch(url, { signal: AbortSignal.timeout(60_000) }); + return response.ok ? await response.blob() : null; + } catch { + return null; + } +} + +/** + * Imports copied composer context (terminal output, review comments, annotations, threads, + * images and files) into the open draft, and counts the attachment slots already spoken for. + */ +export function useComposerContextImport(input: { + readonly environmentId: EnvironmentId; + readonly activeThread: ChatComposerProps["activeThread"]; + readonly activeThreadId: ChatComposerProps["activeThreadId"]; + readonly composerDraftTarget: ChatComposerProps["composerDraftTarget"]; + readonly attachmentDraftTarget: ChatComposerProps["composerDraftTarget"]; + readonly attachmentTargetKey: string; + readonly attachmentTargetKeyRef: RefObject; + readonly questionAttachmentTarget: DraftId | null; + readonly pendingUserInputs: ReadonlyArray; + readonly composerContextRecords: ComposerDraftContextRecords; + readonly composerImages: ReadonlyArray; + readonly composerFilesRef: ChatComposerProps["composerFilesRef"]; + readonly pendingImageCompressionsRef: RefObject>; + readonly pendingThreadImportsRef: RefObject>; + readonly addComposerFilesToDraft: (files: ComposerFileAttachment[]) => string[]; +}) { + const { + environmentId, + activeThread, + activeThreadId, + composerDraftTarget, + attachmentDraftTarget, + attachmentTargetKey, + attachmentTargetKeyRef, + questionAttachmentTarget, + pendingUserInputs, + composerContextRecords, + composerImages, + composerFilesRef, + pendingImageCompressionsRef, + pendingThreadImportsRef, + addComposerFilesToDraft, + } = input; + const getComposerDraft = useComposerDraftStore((store) => store.getComposerDraft); + const addComposerDraftImages = useComposerDraftStore((store) => store.addImages); + const addComposerDraftTerminalContexts = useComposerDraftStore( + (store) => store.addTerminalContexts, + ); + const addComposerDraftReviewComment = useComposerDraftStore((store) => store.addReviewComment); + const addComposerDraftPreviewAnnotation = useComposerDraftStore( + (store) => store.addPreviewAnnotation, + ); + const addComposerDraftThreadContexts = useComposerDraftStore((store) => store.addThreadContexts); + const createAssetUrl = useAtomQueryRunner(assetEnvironment.createUrl, { reportFailure: false }); + const loadThreadTranscript = useAtomCommand(orchestrationEnvironment.threadTranscript, { + reportFailure: false, + }); + const countReservedAttachments = () => { + const questionRequest = pendingUserInputs[0]; + const otherQuestionKeys = + questionAttachmentTarget && questionRequest && activeThreadId + ? questionRequest.questions + .map((question) => + questionAttachmentDraftId( + environmentId, + activeThreadId, + questionRequest.requestId, + question.id, + ), + ) + .filter((key) => key !== questionAttachmentTarget) + : []; + const draft = getComposerDraft(attachmentDraftTarget); + return ( + (draft?.images.length ?? 0) + + (draft?.files.length ?? 0) + + (pendingImageCompressionsRef.current.get(attachmentTargetKey) ?? 0) + + (pendingThreadImportsRef.current.get(attachmentTargetKey) ?? 0) + + countQuestionAttachments(otherQuestionKeys) + ); + }; + const countReservedAttachmentsStable = useStableCallback(countReservedAttachments); + const runAttachmentImport = async ( + record: Extract, + localId: string, + sourceEnvironmentId: EnvironmentId, + importTargetKey: string, + ) => { + const fail = (reason: string) => { + toastManager.add({ + type: "error", + title: `Couldn't bring ${record.name} into this message`, + description: `${reason} Remove the chip or attach the file again.`, + }); + }; + let file = copiedComposerAttachmentFile(sourceEnvironmentId, record.attachmentId); + if (!file) { + const sourceConnection = readPreparedConnection(sourceEnvironmentId); + if (!sourceConnection) { + fail("The environment it came from is not connected."); + return; + } + const result = await createAssetUrl({ + environmentId: sourceEnvironmentId, + input: { resource: { _tag: "attachment", attachmentId: record.attachmentId } }, + }); + const url = + result._tag === "Success" + ? resolveAssetUrl(sourceConnection.httpBaseUrl, result.value.relativeUrl) + : null; + if (!url) { + fail("The original attachment is no longer available."); + return; + } + const blob = await downloadAttachmentBlob(url); + if (!blob) { + fail("Downloading it from the source failed."); + return; + } + file = new File([blob], record.name, { type: record.mimeType || blob.type }); + } + // The draft these bytes belong to may have been sent or switched away from while they + // downloaded. Dropping them here keeps them out of whatever draft is open now. + if (attachmentTargetKeyRef.current !== importTargetKey) return; + const replacesFileMarker = + record.kind === "file" && + composerFilesRef.current.some( + (candidate) => + composerFileNeedsReattach(candidate) && + composerFileMatchesReattachMarker(candidate, { + name: record.name, + mimeType: file.type, + sizeBytes: file.size, + }), + ); + if ( + !replacesFileMarker && + (pendingThreadImportsRef.current.get(importTargetKey) ?? 0) > 0 && + countReservedAttachments() >= PROVIDER_SEND_TURN_MAX_ATTACHMENTS + ) { + fail("The attachment limit has been reached."); + return; + } + if (record.kind === "image") { + const accepted = addComposerDraftImages( + attachmentDraftTarget, + [ + { + type: "image", + id: localId, + name: record.name, + mimeType: file.type, + sizeBytes: file.size, + previewUrl: URL.createObjectURL(file), + file, + }, + ], + { + allowDuplicates: + getComposerDraft(composerDraftTarget)?.previewAnnotations.some( + (annotation) => annotation.id === localId, + ) ?? false, + }, + ); + if (!accepted.includes(localId)) + fail("The draft rejected this attachment (duplicate or attachment limit reached)."); + } else { + const accepted = addComposerFilesToDraft([ + { + type: "file", + id: localId, + name: record.name, + mimeType: file.type, + sizeBytes: file.size, + file, + }, + ]); + if (!accepted.includes(localId)) + fail("The draft rejected this attachment (duplicate or attachment limit reached)."); + } + }; + const importAttachmentRecord = async ( + record: Extract, + localId: string, + sourceEnvironmentId: EnvironmentId, + ) => { + // The chip lands in the draft immediately while these bytes are still downloading. Count + // the transfer against its own draft so a send cannot snapshot a message whose chip has no + // attachment behind it, and so bytes for an abandoned draft never enter the next one. + const importTargetKey = attachmentTargetKey; + pendingDraftWork.begin(importTargetKey); + await runAttachmentImport(record, localId, sourceEnvironmentId, importTargetKey).finally(() => + pendingDraftWork.end(importTargetKey), + ); + }; + const importContextRecords = ( + records: ReadonlyArray, + sourceEnvironmentId: EnvironmentId | null, + threadDraftTarget = attachmentDraftTarget, + ): ReadonlyMap => { + const rewritten = new Map(); + const dependentAttachmentLocalIds = new Map(); + const skippedDependentAttachmentIds = new Set(); + // Resolve annotations before their dependent screenshot records even if a foreign + // clipboard producer emitted the records in a different order. + const orderedRecords = records.toSorted((left, right) => + left.kind === "preview-annotation" && right.kind !== "preview-annotation" + ? -1 + : right.kind === "preview-annotation" && left.kind !== "preview-annotation" + ? 1 + : 0, + ); + for (const candidate of orderedRecords) { + // Producer ids fold into context ids, so two different excerpts can collide. Only skip + // when the draft already holds the same payload; a colliding but different record is + // re-minted under a fresh id so both survive the paste. + const record = asKnownContextRecord(candidate); + if (!record) continue; + // A reminted annotation needs its screenshot under the new annotation id. + const existing = dependentAttachmentLocalIds.has(record.contextId) + ? undefined + : composerContextImportLookupIds(record).flatMap((contextId) => { + const found = composerContextRecords.get(contextId); + return found ? [found] : []; + })[0]; + // Upload completion can change the server id since copy; the local bytes still match. + if ( + sourceEnvironmentId !== null && + (record.kind === "image" || record.kind === "file") && + (existing?.kind === "image" || existing?.kind === "file") && + existing.record.file !== null && + existing.record.file === + copiedComposerAttachmentFile(sourceEnvironmentId, record.attachmentId) + ) + continue; + const existingRecord = + existing?.kind === "terminal" + ? terminalContextRecord(existing.record) + : existing?.kind === "review-comment" + ? reviewCommentContextRecord(existing.record) + : existing?.kind === "preview-annotation" + ? previewAnnotationContextRecord(existing.record, { + screenshotContextId: composerImages.some( + (image) => image.id === existing.record.id, + ) + ? existing.record.id + : undefined, + }) + : existing?.kind === "thread" + ? existing.record + : existing + ? (uploadedContextRecordFromDraft(existing) ?? undefined) + : undefined; + if ( + existingRecord && + isSameComposerContextPayload(existingRecord, record) && + (record.kind !== "thread" || threadDraftTarget === attachmentDraftTarget) + ) { + if (record.kind === "preview-annotation" && record.screenshotContextId) { + skippedDependentAttachmentIds.add(record.screenshotContextId); + } + continue; + } + const conflicts = existing !== undefined; + switch (record.kind) { + case "terminal": { + const threadId = activeThread?.id ?? activeThreadId; + if (!threadId) break; + const imported = terminalContextDraftFromRecord(record, threadId); + const draft = conflicts ? { ...imported, id: randomUUID() } : imported; + addComposerDraftTerminalContexts(composerDraftTarget, [draft], { + appendReference: false, + }); + rewritten.set(record.contextId, terminalContextReference(draft).contextId); + break; + } + case "review-comment": { + const imported = reviewCommentFromRecord(record); + const comment = conflicts ? { ...imported, id: randomUUID() } : imported; + addComposerDraftReviewComment(composerDraftTarget, comment, { + appendReference: false, + }); + rewritten.set(record.contextId, reviewCommentContextId(comment.id)); + break; + } + case "element": + case "preview-annotation": { + const imported = + record.kind === "element" + ? elementContextToPreviewAnnotation(record, randomUUID(), new Date().toISOString()) + : previewAnnotationFromRecord(record); + const annotation = conflicts ? { ...imported, id: randomUUID() } : imported; + if (record.kind === "preview-annotation" && record.screenshotContextId) { + dependentAttachmentLocalIds.set(record.screenshotContextId, annotation.id); + } + addComposerDraftPreviewAnnotation(composerDraftTarget, annotation, { + appendReference: false, + }); + rewritten.set(record.contextId, previewAnnotationContextId(annotation.id)); + break; + } + case "thread": { + // The agent can only read threads on its own server; a pasted foreign one is dropped. + if (record.environmentId !== environmentId) break; + addComposerDraftThreadContexts(threadDraftTarget, [record], { + appendReference: false, + }); + rewritten.set(record.contextId, record.contextId); + break; + } + case "image": + case "file": { + if (skippedDependentAttachmentIds.has(record.contextId)) break; + if (sourceEnvironmentId === null && !conflicts) { + rewritten.set(record.contextId, record.contextId); + break; + } + if (sourceEnvironmentId === null) break; + const localId = dependentAttachmentLocalIds.get(record.contextId) ?? randomUUID(); + rewritten.set(record.contextId, toKindScopedComposerContextId(record.kind, localId)); + void importAttachmentRecord(record, localId, sourceEnvironmentId); + break; + } + default: + break; + } + } + return rewritten; + }; + const importContextFragment = ( + fragment: ComposerContextClipboardFragment, + ): ReadonlyMap => + importContextRecords(fragment.records, fragment.source.environmentId); + const importContextFragmentStable = useStableCallback(importContextFragment); + + return { + loadThreadTranscript, + countReservedAttachments, + countReservedAttachmentsStable, + importContextRecords, + importContextFragment, + importContextFragmentStable, + }; +} ```