chunkFiles.js

'use strict'

const debug = require('debug')('lint-staged:chunkFiles')
const normalize = require('normalize-path')
const path = require('path')

/**
 * Chunk array into sub-arrays
 * @param {Array} arr
 * @param {Number} chunkCount
 * @retuns {Array<Array>}
 */
function chunkArray(arr, chunkCount) {
  if (chunkCount === 1) return [arr]
  const chunked = []
  let position = 0
  for (let i = 0; i < chunkCount; i++) {
    const chunkLength = Math.ceil((arr.length - position) / (chunkCount - i))
    chunked.push([])
    chunked[i] = arr.slice(position, chunkLength + position)
    position += chunkLength
  }
  return chunked
}

/**
 * Chunk files into sub-arrays based on the length of the resulting argument string
 * @param {Object} opts
 * @param {Array<String>} opts.files
 * @param {String} [opts.baseDir] The optional base directory to resolve relative paths.
 * @param {number} [opts.maxArgLength] the maximum argument string length
 * @param {Boolean} [opts.relative] whether files are relative to `gitDir` or should be resolved as absolute
 * @returns {Array<Array<String>>}
 */
module.exports = function chunkFiles({ files, baseDir, maxArgLength = null, relative = false }) {
  if (!maxArgLength) {
    debug('Skip chunking files because of undefined maxArgLength')
    return [files]
  }

  const normalizedFiles = files.map(file =>
    normalize(relative || !baseDir ? file : path.resolve(baseDir, file))
  )
  const fileListLength = normalizedFiles.join(' ').length
  debug(
    `Resolved an argument string length of ${fileListLength} characters from ${normalizedFiles.length} files`
  )
  const chunkCount = Math.min(Math.ceil(fileListLength / maxArgLength), normalizedFiles.length)
  debug(`Creating ${chunkCount} chunks for maxArgLength of ${maxArgLength}`)
  return chunkArray(files, chunkCount)
}