File size: 2,997 Bytes
7a1ad33 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | /**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import * as Diff from 'diff';
import type { DiffStat } from './tools.js';
const DEFAULT_STRUCTURED_PATCH_OPTS: Diff.StructuredPatchOptionsNonabortable = {
context: 3,
ignoreWhitespace: false,
};
export const DEFAULT_DIFF_OPTIONS: Diff.CreatePatchOptionsNonabortable = {
context: 3,
ignoreWhitespace: false,
};
export function getDiffStat(
fileName: string,
oldStr: string,
aiStr: string,
userStr: string,
): DiffStat {
const getStats = (patch: Diff.StructuredPatch) => {
let addedLines = 0;
let removedLines = 0;
let addedChars = 0;
let removedChars = 0;
patch.hunks.forEach((hunk: Diff.StructuredPatchHunk) => {
hunk.lines.forEach((line: string) => {
if (line.startsWith('+')) {
addedLines++;
addedChars += line.length - 1;
} else if (line.startsWith('-')) {
removedLines++;
removedChars += line.length - 1;
}
});
});
return { addedLines, removedLines, addedChars, removedChars };
};
const modelPatch = Diff.structuredPatch(
fileName,
fileName,
oldStr,
aiStr,
'Current',
'Proposed',
DEFAULT_STRUCTURED_PATCH_OPTS,
);
const modelStats = getStats(modelPatch);
const userPatch = Diff.structuredPatch(
fileName,
fileName,
aiStr,
userStr,
'Proposed',
'User',
DEFAULT_STRUCTURED_PATCH_OPTS,
);
const userStats = getStats(userPatch);
return {
model_added_lines: modelStats.addedLines,
model_removed_lines: modelStats.removedLines,
model_added_chars: modelStats.addedChars,
model_removed_chars: modelStats.removedChars,
user_added_lines: userStats.addedLines,
user_removed_lines: userStats.removedLines,
user_added_chars: userStats.addedChars,
user_removed_chars: userStats.removedChars,
};
}
/**
* Extracts line and character stats from a unified diff patch string.
* This is useful for reconstructing stats for rejected or errored operations
* where the full strings may no longer be easily accessible.
*/
export function getDiffStatFromPatch(patch: string): DiffStat {
let addedLines = 0;
let removedLines = 0;
let addedChars = 0;
let removedChars = 0;
const lines = patch.split('\n');
for (const line of lines) {
// Only count lines that are additions or removals,
// excluding the diff headers (--- and +++) and metadata (\)
if (line.startsWith('+') && !line.startsWith('+++')) {
addedLines++;
addedChars += line.length - 1;
} else if (line.startsWith('-') && !line.startsWith('---')) {
removedLines++;
removedChars += line.length - 1;
}
}
return {
model_added_lines: addedLines,
model_removed_lines: removedLines,
model_added_chars: addedChars,
model_removed_chars: removedChars,
user_added_lines: 0,
user_removed_lines: 0,
user_added_chars: 0,
user_removed_chars: 0,
};
}
|