fix(vscode): open each apply patch file

This commit is contained in:
Nabeel Siddiqui
2026-08-01 13:58:48 -04:00
parent 9f60375cac
commit 8eee5b27cd
3 changed files with 105 additions and 23 deletions
@@ -53,7 +53,7 @@ import {
} from './taskToolModel';
import { areRenderRelevantPartsEqual } from '../renderCompare';
import { useI18n } from '@/lib/i18n';
import { getDiffPatchEntries, getPatchText, getPrimaryToolPath, type DiffPatchEntry } from './toolDiffUtils';
import { getApplyPatchFilePath, getDiffPatchEntries, getPatchText, getPrimaryToolPath, type DiffPatchEntry } from './toolDiffUtils';
import { isEmbeddedSessionChat } from '@/components/layout/contextPanelEmbeddedChat';
import { useStreamingTextThrottle } from '../../hooks/useStreamingTextThrottle';
import { getStreamingOutputAppend, getToolOutput } from './toolOutput';
@@ -80,6 +80,7 @@ const getMultiFileDescription = (
metadata: Record<string, unknown> | undefined,
animate = true,
showFileIcons = true,
onFileClick?: (file: Record<string, unknown>, event: React.MouseEvent<HTMLButtonElement>) => void,
): React.ReactNode => {
const files = Array.isArray(metadata?.files) ? metadata?.files : [];
if (files.length <= 1) return null;
@@ -103,10 +104,11 @@ const getMultiFileDescription = (
return base + incoming;
};
const entriesByPath = new Map<string, { path: string; name: string; added: number | null; removed: number | null }>();
const entriesByPath = new Map<string, { file: Record<string, unknown>; path: string; name: string; added: number | null; removed: number | null }>();
for (const file of files) {
const fileObj = file as { relativePath?: string; filePath?: string; additions?: unknown; deletions?: unknown };
if (!file || typeof file !== 'object') continue;
const fileObj = file as Record<string, unknown> & { relativePath?: string; filePath?: string; additions?: unknown; deletions?: unknown };
const filePath = fileObj.relativePath || fileObj.filePath || '';
if (!filePath) continue;
const fileName = filePath.split('/').pop() || filePath;
@@ -120,7 +122,7 @@ const getMultiFileDescription = (
continue;
}
entriesByPath.set(filePath, { path: filePath, name: fileName, added, removed });
entriesByPath.set(filePath, { file: fileObj, path: filePath, name: fileName, added, removed });
}
const entries = Array.from(entriesByPath.values());
@@ -129,8 +131,8 @@ const getMultiFileDescription = (
<>
{entries.map((entry) => {
const hasPerFileDiff = entry.added !== null || entry.removed !== null;
return (
<span key={entry.path} className={cn('inline-flex min-w-0 max-w-full items-center gap-1', TOOL_ROW_DESCRIPTION_CLASS)} style={{ color: 'var(--tools-description)' }}>
const content = (
<>
{showFileIcons ? <FileTypeIcon filePath={entry.path} className="h-3.5 w-3.5" /> : null}
<Text
variant={animate ? 'generate-effect' : 'static'}
@@ -147,6 +149,23 @@ const getMultiFileDescription = (
<span style={{ color: 'var(--status-error)' }}>-{entry.removed ?? 0}</span>
</span>
) : null}
</>
);
const canOpen = onFileClick && entry.file.type !== 'delete' && getApplyPatchFilePath(entry.file);
return canOpen ? (
<button
key={entry.path}
type="button"
className={cn('inline-flex min-w-0 max-w-full items-center gap-1 rounded-sm text-left hover:text-foreground focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring', TOOL_ROW_DESCRIPTION_CLASS)}
style={{ color: 'var(--tools-description)' }}
onClick={(event) => onFileClick(entry.file, event)}
onKeyDown={(event) => event.stopPropagation()}
>
{content}
</button>
) : (
<span key={entry.path} className={cn('inline-flex min-w-0 max-w-full items-center gap-1', TOOL_ROW_DESCRIPTION_CLASS)} style={{ color: 'var(--tools-description)' }}>
{content}
</span>
);
})}
@@ -1633,9 +1652,7 @@ const ToolExpandedContent: React.FC<ToolExpandedContentProps> = React.memo(({
);
const renderResultContent = () => {
const getEntryAbsolutePath = (entry: DiffPatchEntry) => (
entry.title.startsWith('/') ? entry.title : `${currentDirectory}/${entry.title}`.replace(/\/+/g, '/')
);
const getEntryAbsolutePath = (entry: DiffPatchEntry) => toAbsoluteFilePath(currentDirectory, entry.filePath ?? entry.title);
const openEntryFile = (entry: DiffPatchEntry, event: React.MouseEvent<HTMLButtonElement>) => {
event.stopPropagation();
const line = extractFirstChangedLineFromDiff(entry.patch);
@@ -2291,6 +2308,24 @@ const ToolPartContent: React.FC<ToolPartProps> = ({
}, [descriptionPath, normalizedPartTool, stateWithData, input]);
const runtime = React.useContext(RuntimeAPIContext);
const openApplyPatchFile = (file: Record<string, unknown>, event: React.MouseEvent<HTMLButtonElement>) => {
const filePath = getApplyPatchFilePath(file);
if (!runtime?.editor || !filePath || file.type === 'delete') {
return;
}
event.stopPropagation();
const patch = getPatchText(file.patch) ?? getPatchText(file.diff);
const targetLine = patch ? extractFirstChangedLineFromDiff(patch) : undefined;
const absolutePath = toAbsoluteFilePath(currentDirectory, filePath);
if (runtime.runtime.isVSCode && patch) {
const label = `${getRelativePath(absolutePath, currentDirectory)} (changes)`;
void runtime.editor.openDiff('', absolutePath, label, { line: targetLine, patch });
return;
}
void runtime.editor.openFile(absolutePath, targetLine);
};
const handleMainClick = (e: { stopPropagation: () => void }) => {
if (isTaskTool || !runtime?.editor) {
onToggle(part.id);
@@ -2403,7 +2438,7 @@ const ToolPartContent: React.FC<ToolPartProps> = ({
>
{displayName}
</MinDurationShineText>
{getMultiFileDescription(metadata, animateTailText, showToolFileIcons)}
{getMultiFileDescription(metadata, animateTailText, showToolFileIcons, runtime?.editor ? openApplyPatchFile : undefined)}
</>
) : (
<>
@@ -1,6 +1,6 @@
import { describe, expect, test } from 'bun:test';
import { getDiffPatchEntries, getPrimaryToolPath, getRenderablePatchInfo } from './toolDiffUtils';
import { getApplyPatchFilePath, getDiffPatchEntries, getPrimaryToolPath, getRenderablePatchInfo } from './toolDiffUtils';
const identity = (path: string) => path;
@@ -35,6 +35,18 @@ describe('toolDiffUtils', () => {
})).toBe('src/file.ts');
});
test('resolves each apply_patch file independently', () => {
expect(getApplyPatchFilePath({
filePath: '/workspace/project/src/first.ts',
relativePath: 'workspace/project/src/first.ts',
})).toBe('/workspace/project/src/first.ts');
expect(getApplyPatchFilePath({
filePath: '/workspace/project/src/old.ts',
movePath: '/workspace/project/src/second.ts',
relativePath: 'src/second.ts',
})).toBe('/workspace/project/src/second.ts');
});
test('treats raw apply_patch envelopes as text, not visual diffs', () => {
const entries = getDiffPatchEntries(undefined, [
'*** Begin Patch',
@@ -87,6 +99,27 @@ describe('toolDiffUtils', () => {
expect(entries[0]?.title).toBe('src/file.ts');
});
test('keeps the authoritative path for every metadata file entry', () => {
const patch = [
'--- a/src/file.ts',
'+++ b/src/file.ts',
'@@ -1 +1 @@',
'-old',
'+new',
].join('\n');
const entries = getDiffPatchEntries({
files: [
{ filePath: '/workspace/project/src/first.ts', relativePath: 'src/first.ts', patch },
{ filePath: '/workspace/project/src/second.ts', relativePath: 'src/second.ts', patch },
],
}, undefined, identity);
expect(entries.map((entry) => entry.filePath)).toEqual([
'/workspace/project/src/first.ts',
'/workspace/project/src/second.ts',
]);
});
test('synthesizes headers for valid headerless hunks', () => {
const entries = getDiffPatchEntries(undefined, [
'@@ -1 +1 @@',
@@ -3,6 +3,7 @@ import { parsePatchFiles } from '@pierre/diffs';
export type DiffPatchEntry = {
id: string;
title: string;
filePath?: string;
patch: string;
renderMode: 'diff' | 'text';
};
@@ -140,6 +141,20 @@ export const getPatchText = (value: unknown): string | undefined => {
return undefined;
};
export const getApplyPatchFilePath = (file: unknown): string | null => {
if (!isRecord(file)) {
return null;
}
return typeof file.movePath === 'string'
? file.movePath
: typeof file.filePath === 'string'
? file.filePath
: typeof file.relativePath === 'string'
? file.relativePath
: null;
};
export const getPrimaryToolPath = (
toolName: string,
input: Record<string, unknown> | undefined,
@@ -147,17 +162,15 @@ export const getPrimaryToolPath = (
): string | null => {
if (toolName === 'apply_patch') {
const files = Array.isArray(metadata?.files) ? metadata.files : [];
const first = files.find((entry) => isRecord(entry) && entry.type !== 'delete');
if (!isRecord(first)) {
return null;
for (const file of files) {
if (isRecord(file) && file.type !== 'delete') {
const filePath = getApplyPatchFilePath(file);
if (filePath) {
return filePath;
}
}
}
return typeof first.movePath === 'string'
? first.movePath
: typeof first.filePath === 'string'
? first.filePath
: typeof first.relativePath === 'string'
? first.relativePath
: null;
return null;
}
if (toolName === 'edit' || toolName === 'multiedit') {
@@ -334,7 +347,7 @@ const getPatchEntriesFromText = (
}];
};
const getFilePatch = (file: unknown): { patch: string; title: string } | null => {
const getFilePatch = (file: unknown): { filePath?: string; patch: string; title: string } | null => {
if (!isRecord(file)) {
return null;
}
@@ -351,6 +364,7 @@ const getFilePatch = (file: unknown): { patch: string; title: string } | null =>
: '';
return {
filePath: getApplyPatchFilePath(file) ?? undefined,
patch,
title: rawPath,
};
@@ -372,7 +386,7 @@ export const getDiffPatchEntries = (
filePatch.title || `File ${index + 1}`,
`file-${index}`,
resolveTitle,
);
).map((entry) => ({ ...entry, filePath: filePatch.filePath }));
});
if (fileEntries.length > 0) {