shamjs/lib/random/pick.js

Mock data generator.

var integer = require('./integer')
var shuffle = require('lodash.shuffle');
module.exports = function pick (data, count) {
    if (typeof count === 'number') {
        if (count > data.length) {
            count = data.length
        }
        if (count < 2) {
            console.warn('node_modules/shamjs/lib/random/pick.js: pick(data, count) count must > 1')
            count = 2
        }
        var type = 'array'
        if (typeof data === 'string') {
            data = data.split('')
            type = 'string'
        }
        var result = shuffle(data).slice(0, count)
        if (type === 'string') {
            result = result.join('')
        }
        return result
    }
    else {
        return data[integer(data.length-1)]
    }
}