mirror of https://github.com/vuejs/core.git
65 lines
2.0 KiB
TypeScript
65 lines
2.0 KiB
TypeScript
const range: number = 2
|
|
|
|
export function generateCodeFrame(
|
|
source: string,
|
|
start = 0,
|
|
end: number = source.length,
|
|
): string {
|
|
// Ensure start and end are within the source length
|
|
start = Math.max(0, Math.min(start, source.length))
|
|
end = Math.max(0, Math.min(end, source.length))
|
|
|
|
if (start > end) return ''
|
|
|
|
// Split the content into individual lines but capture the newline sequence
|
|
// that separated each line. This is important because the actual sequence is
|
|
// needed to properly take into account the full line length for offset
|
|
// comparison
|
|
let lines = source.split(/(\r?\n)/)
|
|
|
|
// Separate the lines and newline sequences into separate arrays for easier referencing
|
|
const newlineSequences = lines.filter((_, idx) => idx % 2 === 1)
|
|
lines = lines.filter((_, idx) => idx % 2 === 0)
|
|
|
|
let count = 0
|
|
const res: string[] = []
|
|
for (let i = 0; i < lines.length; i++) {
|
|
count +=
|
|
lines[i].length +
|
|
((newlineSequences[i] && newlineSequences[i].length) || 0)
|
|
if (count >= start) {
|
|
for (let j = i - range; j <= i + range || end > count; j++) {
|
|
if (j < 0 || j >= lines.length) continue
|
|
const line = j + 1
|
|
res.push(
|
|
`${line}${' '.repeat(Math.max(3 - String(line).length, 0))}| ${
|
|
lines[j]
|
|
}`,
|
|
)
|
|
const lineLength = lines[j].length
|
|
const newLineSeqLength =
|
|
(newlineSequences[j] && newlineSequences[j].length) || 0
|
|
|
|
if (j === i) {
|
|
// push underline
|
|
const pad = start - (count - (lineLength + newLineSeqLength))
|
|
const length = Math.max(
|
|
1,
|
|
end > count ? lineLength - pad : end - start,
|
|
)
|
|
res.push(` | ` + ' '.repeat(pad) + '^'.repeat(length))
|
|
} else if (j > i) {
|
|
if (end > count) {
|
|
const length = Math.max(Math.min(end - count, lineLength), 1)
|
|
res.push(` | ` + '^'.repeat(length))
|
|
}
|
|
|
|
count += lineLength + newLineSeqLength
|
|
}
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return res.join('\n')
|
|
}
|