Function generateCodeFrame

packages/shared/src/codeframe.ts:3–64

(
  source: string,
  start = 0,
  end: number = source.length,
)

const range: number = 2

export function generateCodeFrame(
  source: string,
  start = 0,
  end: number = source.length,
): string {
  class="cm">// 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 class="st">''

  class="cm">// Split the content into individual lines but capture the newline sequence
  class="cm">// that separated each line. This is important because the actual sequence is
  class="cm">// needed to properly take into account the full line length for offset
  class="cm">// comparison
  let lines = source.split(/(\r?\n)/)

  class="cm">// 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}${class="st">' '.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) {
          class="cm">// push underline
          const pad = start - (count - (lineLength + newLineSeqLength))
          const length = Math.max(
            1,
            end > count ? lineLength - pad : end - start,
          )
          res.push(`   |  ` + class="st">' '.repeat(pad) + class="st">'^'.repeat(length))
        } else if (j > i) {
          if (end > count) {
            const length = Math.max(Math.min(end - count, lineLength), 1)
            res.push(`   |  ` + class="st">'^'.repeat(length))
          }

          count += lineLength + newLineSeqLength
        }
      }
      break