twing/dist/cjs/lib/extension/core/functions/random.js

First-class Twig engine for Node.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.randomSynchronously = exports.random = void 0;
const iconv_1 = require("../../../helpers/iconv");
const is_traversable_1 = require("../../../helpers/is-traversable");
const iterator_to_array_1 = require("../../../helpers/iterator-to-array");
const runes = require('runes');
const mt_rand = require('locutus/php/math/mt_rand');
const array_rand = require('locutus/php/array/array_rand');
/**
 * Returns a random value depending on the supplied parameter type:
 * - a random item from a Traversable or array
 * - a random character from a string
 * - a random integer between 0 and the integer parameter.
 *
 * @param executionContext
 * @param {*} values The values to pick a random item from
 * @param {number} max Maximum value used when values is an integer
 *
 * @throws TwingErrorRuntime when values is an empty array (does not apply to an empty string which is returned as is)
 *
 * @returns {Promise<any>} A random value from the given sequence
 */
const random = (executionContext, values, max) => {
    const { environment } = executionContext;
    const { charset } = environment;
    let _do = () => {
        if (values === null) {
            return max === null ? mt_rand() : mt_rand(0, max);
        }
        if (typeof values === 'number') {
            let min;
            if (max === null) {
                if (values < 0) {
                    max = 0;
                    min = values;
                }
                else {
                    max = values;
                    min = 0;
                }
            }
            else {
                min = values;
            }
            return mt_rand(min, max);
        }
        if (typeof values === 'string') {
            values = Buffer.from(values);
        }
        if (Buffer.isBuffer(values)) {
            if (values.toString() === '') {
                return '';
            }
            if (charset !== 'UTF-8') {
                values = (0, iconv_1.iconv)(charset, 'UTF-8', values);
            }
            // unicode split
            values = runes(values.toString());
            if (charset !== 'UTF-8') {
                values = values.map((value) => {
                    return (0, iconv_1.iconv)('UTF-8', charset, Buffer.from(value)).toString();
                });
            }
        }
        else if ((0, is_traversable_1.isTraversable)(values)) {
            values = (0, iterator_to_array_1.iteratorToArray)(values);
        }
        if (!Array.isArray(values)) {
            return values;
        }
        if (values.length < 1) {
            return Promise.reject(new Error('The random function cannot pick from an empty array.'));
        }
        return values[array_rand(values, 1)];
    };
    return Promise.resolve(_do());
};
exports.random = random;
const randomSynchronously = (executionContext, values, max) => {
    const { environment } = executionContext;
    const { charset } = environment;
    if (values === null) {
        return max === null ? mt_rand() : mt_rand(0, max);
    }
    if (typeof values === 'number') {
        let min;
        if (max === null) {
            if (values < 0) {
                max = 0;
                min = values;
            }
            else {
                max = values;
                min = 0;
            }
        }
        else {
            min = values;
        }
        return mt_rand(min, max);
    }
    if (typeof values === 'string') {
        values = Buffer.from(values);
    }
    if (Buffer.isBuffer(values)) {
        if (values.toString() === '') {
            return '';
        }
        if (charset !== 'UTF-8') {
            values = (0, iconv_1.iconv)(charset, 'UTF-8', values);
        }
        // unicode split
        values = runes(values.toString());
        if (charset !== 'UTF-8') {
            values = values.map((value) => {
                return (0, iconv_1.iconv)('UTF-8', charset, Buffer.from(value)).toString();
            });
        }
    }
    else if ((0, is_traversable_1.isTraversable)(values)) {
        values = (0, iterator_to_array_1.iteratorToArray)(values);
    }
    if (!Array.isArray(values)) {
        return values;
    }
    if (values.length < 1) {
        throw new Error('The random function cannot pick from an empty array.');
    }
    return values[array_rand(values, 1)];
};
exports.randomSynchronously = randomSynchronously;