pdf-merger-js/parsePagesString.js

merge multiple PDF documents, or parts of them, to a new PDF document

/**
 * Takes a string like "1,2,3" or "1-3" and returns an Array of numbers.
 *
 * @param {string} pages
 *
 * @example ```js
 * parsePagesString('2') // [2]
 * parsePagesString('1,2,3') // [1,2,3]
 * parsePagesString('1-3') // [1,2,3]
 * parsePagesString('1to3') // [1,2,3]
 * parsePagesString('1 to 3') // [1,2,3]
 * parsePagesString('10,1-3') // [10,1,2,3]
 * parsePagesString('9,1-3,5-7') // [9,1,2,3,5,6,7]
 * ```
 */
export function parsePagesString (pages) {
  const throwError = () => {
    throw new Error([
      'Invalid parameter "pages".',
      'Must be a string like "1,2,3" or "1-3" or "1to3"',
      `Was "${pages}" instead.`
    ].join(' '))
  }

  const isRangeString = (rangeString) => {
    return rangeString.includes('-') || rangeString.toLowerCase().includes('to')
  }

  const parseRange = (rangeString) => {
    const [start, end] = rangeString.split(/-|to/).map(s => typeof s === 'string' ? parseInt(s.trim()) : s)
    return Array.from({ length: end - start + 1 }, (_, i) => start + i)
  }

  if (typeof pages !== 'string') {
    throwError()
  } else if (!pages.trim().replace(/ /g, '').match(/^(\d+|\d+-\d+|\d+to\d+)(,(\d+|\d+-\d+|\d+to\d+))*$/)) {
    // string does not fit the expected pattern
    throwError()
  } else if (pages.trim().match(/^\d+$/)) {
    // string consists of a single page-number
    return [parseInt(pages.trim())]
  } else if (pages.trim().includes(',')) {
    // string consists od a list of page-numbers and/or ranges
    return pages.split(',').flatMap(s => isRangeString(s) ? parseRange(s) : parseInt(s))
  } else if (isRangeString(pages)) {
    // string consists of a single range
    return parseRange(pages)
  }

  throwError()
}