heyujie
2021-06-07 8f8155aa4c83f7c2577ac123add550766b6a7ce3
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
"use strict";
 
var isObject            = require("type/object/is")
  , ensureNaturalNumber = require("type/natural-number/ensure")
  , ensureString        = require("type/string/ensure");
 
var generated = Object.create(null), random = Math.random, uniqTryLimit = 100;
 
var getChunk = function () { return random().toString(36).slice(2); };
 
var getString = function (length, charset) {
    var str;
    if (charset) {
        var charsetLength = charset.length;
        str = "";
        for (var i = 0; i < length; ++i) {
            str += charset.charAt(Math.floor(Math.random() * charsetLength));
        }
        return str;
    }
    str = getChunk();
    if (length === null) return str;
    while (str.length < length) str += getChunk();
    return str.slice(0, length);
};
 
module.exports = function (/* options */) {
    var options = arguments[0];
    if (!isObject(options)) options = {};
    var length = ensureNaturalNumber(options.length, { "default": 10 })
      , isUnique = options.isUnique
      , charset = ensureString(options.charset, { isOptional: true });
 
    var str = getString(length, charset);
    if (isUnique) {
        var count = 0;
        while (generated[str]) {
            if (++count === uniqTryLimit) {
                throw new Error(
                    "Cannot generate random string.\n" +
                        "String.random is not designed to effectively generate many short and " +
                        "unique random strings"
                );
            }
            str = getString(length);
        }
        generated[str] = true;
    }
    return str;
};