File size: 2,693 Bytes
064bfd6 | 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 | import { getPastedTextRefNumLines } from 'src/history.js'
import type { PastedContent } from 'src/utils/config.js'
const TRUNCATION_THRESHOLD = 10000 // Characters before we truncate
const PREVIEW_LENGTH = 1000 // Characters to show at start and end
type TruncatedMessage = {
truncatedText: string
placeholderContent: string
}
/**
* Determines whether the input text should be truncated. If so, it adds a
* truncated text placeholder and neturns
*
* @param text The input text
* @param nextPasteId The reference id to use
* @returns The new text to display and separate placeholder content if applicable.
*/
export function maybeTruncateMessageForInput(
text: string,
nextPasteId: number,
): TruncatedMessage {
// If the text is short enough, return it as-is
if (text.length <= TRUNCATION_THRESHOLD) {
return {
truncatedText: text,
placeholderContent: '',
}
}
// Calculate how much text to keep from start and end
const startLength = Math.floor(PREVIEW_LENGTH / 2)
const endLength = Math.floor(PREVIEW_LENGTH / 2)
// Extract the portions we'll keep
const startText = text.slice(0, startLength)
const endText = text.slice(-endLength)
// Calculate the number of lines that will be truncated
const placeholderContent = text.slice(startLength, -endLength)
const truncatedLines = getPastedTextRefNumLines(placeholderContent)
// Create a placeholder reference similar to pasted text
const placeholderId = nextPasteId
const placeholderRef = formatTruncatedTextRef(placeholderId, truncatedLines)
// Combine the parts with the placeholder
const truncatedText = startText + placeholderRef + endText
return {
truncatedText,
placeholderContent,
}
}
function formatTruncatedTextRef(id: number, numLines: number): string {
return `[...Truncated text #${id} +${numLines} lines...]`
}
export function maybeTruncateInput(
input: string,
pastedContents: Record<number, PastedContent>,
): { newInput: string; newPastedContents: Record<number, PastedContent> } {
// Get the next available ID for the truncated content
const existingIds = Object.keys(pastedContents).map(Number)
const nextPasteId = existingIds.length > 0 ? Math.max(...existingIds) + 1 : 1
// Apply truncation
const { truncatedText, placeholderContent } = maybeTruncateMessageForInput(
input,
nextPasteId,
)
if (!placeholderContent) {
return { newInput: input, newPastedContents: pastedContents }
}
return {
newInput: truncatedText,
newPastedContents: {
...pastedContents,
[nextPasteId]: {
id: nextPasteId,
type: 'text',
content: placeholderContent,
},
},
}
}
|