Spaces:
Sleeping
Sleeping
File size: 1,615 Bytes
be5030f |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
'use strict'
const arrayify = require('array-back')
const testValue = require('test-value')
/**
* Find and either replace or remove items from an array.
*
* @module find-replace
* @example
* > findReplace = require('find-replace')
*
* > findReplace([ 1, 2, 3], 2, 'two')
* [ 1, 'two', 3 ]
*
* > findReplace([ 1, 2, 3], 2, [ 'two', 'zwei' ])
* [ 1, [ 'two', 'zwei' ], 3 ]
*
* > findReplace([ 1, 2, 3], 2, 'two', 'zwei')
* [ 1, 'two', 'zwei', 3 ]
*
* > findReplace([ 1, 2, 3], 2) // no replacement, so remove
* [ 1, 3 ]
*/
module.exports = findReplace
/**
* @param {array} - the input array
* @param {valueTest} - a [test-value](https://github.com/75lb/test-value) query to match the value you're looking for
* @param [replaceWith] {...any} - If specified, found values will be replaced with these values, else removed.
* @returns {array}
* @alias module:find-replace
*/
function findReplace (array, valueTest) {
const found = []
const replaceWiths = arrayify(arguments)
replaceWiths.splice(0, 2)
arrayify(array).forEach((value, index) => {
let expanded = []
replaceWiths.forEach(replaceWith => {
if (typeof replaceWith === 'function') {
expanded = expanded.concat(replaceWith(value))
} else {
expanded.push(replaceWith)
}
})
if (testValue(value, valueTest)) {
found.push({
index: index,
replaceWithValue: expanded
})
}
})
found.reverse().forEach(item => {
const spliceArgs = [ item.index, 1 ].concat(item.replaceWithValue)
array.splice.apply(array, spliceArgs)
})
return array
}
|