/
array_pad.js
34 lines (30 loc) · 1003 Bytes
/
array_pad.js
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
module.exports = function array_pad(input, padSize, padValue) {
// discuss at: https://locutus.io/php/array_pad/
// original by: Waldo Malqui Silva (https://waldo.malqui.info)
// example 1: array_pad([ 7, 8, 9 ], 2, 'a')
// returns 1: [ 7, 8, 9]
// example 2: array_pad([ 7, 8, 9 ], 5, 'a')
// returns 2: [ 7, 8, 9, 'a', 'a']
// example 3: array_pad([ 7, 8, 9 ], 5, 2)
// returns 3: [ 7, 8, 9, 2, 2]
// example 4: array_pad([ 7, 8, 9 ], -5, 'a')
// returns 4: [ 'a', 'a', 7, 8, 9 ]
let pad = []
const newArray = []
let newLength
let diff = 0
let i = 0
if (Object.prototype.toString.call(input) === '[object Array]' && !isNaN(padSize)) {
newLength = padSize < 0 ? padSize * -1 : padSize
diff = newLength - input.length
if (diff > 0) {
for (i = 0; i < diff; i++) {
newArray[i] = padValue
}
pad = padSize < 0 ? newArray.concat(input) : input.concat(newArray)
} else {
pad = input
}
}
return pad
}