53 lines
1.2 KiB
JavaScript
53 lines
1.2 KiB
JavaScript
/**
|
|
* Secure random string generator with custom alphabet.
|
|
*
|
|
* Alphabet must contain 256 symbols or less. Otherwise, the generator
|
|
* will not be secure.
|
|
*
|
|
* @param {generator} random The random bytes generator.
|
|
* @param {string} alphabet Symbols to be used in new random string.
|
|
* @param {size} size The number of symbols in new random string.
|
|
*
|
|
* @return {string} Random string.
|
|
*
|
|
* @example
|
|
* const format = require('nanoid/format')
|
|
*
|
|
* function random (size) {
|
|
* const result = []
|
|
* for (let i = 0; i < size; i++) {
|
|
* result.push(randomByte())
|
|
* }
|
|
* return result
|
|
* }
|
|
*
|
|
* format(random, "abcdef", 5) //=> "fbaef"
|
|
*
|
|
* @name format
|
|
* @function
|
|
*/
|
|
module.exports = function (random, alphabet, size) {
|
|
var mask = (2 << 31 - Math.clz32((alphabet.length - 1) | 1)) - 1
|
|
var step = Math.ceil(1.6 * mask * size / alphabet.length)
|
|
var id = ''
|
|
size = +size
|
|
|
|
while (true) {
|
|
var bytes = random(step)
|
|
for (var i = step; i--;) {
|
|
var byte = bytes[i] & mask
|
|
var alpha = alphabet[byte]
|
|
if (alpha) {
|
|
id += alpha
|
|
if (id.length === size) return id
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @callback generator
|
|
* @param {number} bytes The number of bytes to generate.
|
|
* @return {number[]} Random bytes.
|
|
*/
|