@maibornwolff/alt-core-js/lib/variableInjection.js

"use strict";
var __assign = (this && this.__assign) || function () {
    __assign = Object.assign || function(t) {
        for (var s, i = 1, n = arguments.length; i < n; i++) {
            s = arguments[i];
            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
                t[p] = s[p];
        }
        return t;
    };
    return __assign.apply(this, arguments);
};
Object.defineProperty(exports, "__esModule", { value: true });
var vm_1 = require("vm");
var logging_1 = require("./logging");
var nodeGlobals_1 = require("./nodeGlobals");
function injectVariableAccessAndEvaluate(expression, scenarioVariables) {
    var regex = /{{(\w*)}}/g;
    var variables = {};
    var expressionWithVariables = searchForMatchingStrings(regex, expression).reduce(function (previousString, variable) {
        variables[variable] = scenarioVariables.get(variable);
        var searchValue = "{{" + variable + "}}";
        return previousString.replace(searchValue, variable);
    }, expression);
    return vm_1.runInNewContext(expressionWithVariables, __assign(__assign({}, nodeGlobals_1.nodeGlobals), variables));
}
exports.injectVariableAccessAndEvaluate = injectVariableAccessAndEvaluate;
function injectVarsToString(str, scenarioVariables, ctx) {
    var regex = /{{(\w*)}}/g;
    return searchForMatchingStrings(regex, str).reduce(function (previousString, variable) {
        var replaceValue = scenarioVariables.get(variable);
        if (replaceValue !== undefined) {
            var searchValue = "{{" + variable + "}}";
            logging_1.getLogger(ctx.scenario).debug("Replacing '" + searchValue + "' with '" + replaceValue + "'", ctx);
            return previousString.replace(searchValue, "" + replaceValue);
        }
        logging_1.getLogger(ctx.scenario).debug("Not able to replace {{" + variable + "}} because no variable with that name found!", ctx);
        return previousString;
    }, str);
}
function injectEvalAndVarsToString(str, scenarioVariables, ctx) {
    var afterEvalToString = injectEvaluationToString(str, ctx, scenarioVariables);
    var afterVarInjection = injectVarsToString(afterEvalToString, scenarioVariables, ctx);
    var _a = injectEvaluationToNumber(afterVarInjection, ctx, scenarioVariables), afterEvalToNumber = _a[0], foundNumericExpression = _a[1];
    // if the string contains only number description, that can be converted, then return a number, in other case return a string
    if (foundNumericExpression &&
        afterEvalToNumber === (+afterEvalToNumber).toString()) {
        return +afterEvalToNumber;
    }
    return afterEvalToNumber;
}
exports.injectEvalAndVarsToString = injectEvalAndVarsToString;
function injectEvalAndVarsToMap(keyValueMap, scenarioVariables, ctx) {
    var copy = keyValueMap instanceof Array ? [] : {};
    Object.assign(copy, keyValueMap);
    for (var _i = 0, _a = Object.entries(copy); _i < _a.length; _i++) {
        var mapEntry = _a[_i];
        var key = mapEntry[0];
        var value = mapEntry[1];
        if (value instanceof Object) {
            // contains nested values
            copy[key] = injectEvalAndVarsToMap(value, scenarioVariables, ctx);
        }
        else if (typeof value === 'string') {
            copy[key] = injectEvalAndVarsToString(value, scenarioVariables, ctx);
        }
    }
    return copy;
}
exports.injectEvalAndVarsToMap = injectEvalAndVarsToMap;
function searchForMatchingStrings(regex, str) {
    var regexCopy = regex;
    var m;
    var matchingStrings = [];
    while ((m = regexCopy.exec(str)) !== null) {
        if (m.index === regexCopy.lastIndex)
            regexCopy.lastIndex += 1;
        matchingStrings.push(m[1]);
    }
    return matchingStrings;
}
function injectEvaluationToString(str, ctx, vars) {
    // the vars (scenario variable) should be left available here in order to access
    // and set them from within evaluated expressions
    var regex = /{{{(.*?)}}}/g;
    var result = str;
    searchForMatchingStrings(regex, result).forEach(function (expression) {
        var replaceValue = (function () { return eval(expression); }).call(buildExpHelpers(vars));
        if (replaceValue) {
            var searchValue = "{{{" + expression + "}}}";
            logging_1.getLogger(ctx.scenario).debug("Replacing '" + searchValue + "' with '" + replaceValue + "'", ctx);
            result = result.replace(searchValue, replaceValue);
        }
        else {
            logging_1.getLogger(ctx.scenario).debug("Not able to replace {{{" + expression + "}}} !", ctx);
        }
    });
    return result;
}
function injectEvaluationToNumber(str, ctx, vars) {
    // the vars (scenario variable) should be left available here in order to access
    // and set them from within evaluated expressions
    var regex = /<<<(.*?)>>>/g;
    var foundNumericExpression = false;
    var result = str;
    searchForMatchingStrings(regex, result).forEach(function (expression) {
        foundNumericExpression = true;
        var replaceValue = (function () { return eval(expression); }).call(buildExpHelpers(vars));
        if (replaceValue) {
            var searchValue = "<<<" + expression + ">>>";
            logging_1.getLogger(ctx.scenario).debug("Replacing '\"" + searchValue + "\"' with '" + replaceValue + "'", ctx);
            result = result.replace(searchValue, replaceValue);
        }
        else {
            logging_1.getLogger(ctx.scenario).debug("Not able to replace {{{" + expression + "}}} !", ctx);
        }
    });
    return [result, foundNumericExpression];
}
function buildExpHelpers(vars) {
    return {
        get: function (name) {
            return vars.get(name);
        },
        set: function (name, value) {
            return vars.set(name, value);
        },
        getAndInc: function (name) {
            var currentValue = vars.get(name);
            vars.set(name, currentValue + 1);
            return currentValue;
        },
        getAndIncBy: function (name, howMuch) {
            var currentValue = vars.get(name);
            vars.set(name, currentValue + howMuch);
            return currentValue;
        },
        incAndGet: function (name) {
            var targetValue = vars.get(name) + 1;
            vars.set(name, targetValue);
            return targetValue;
        },
        incByAndGet: function (name, howMuch) {
            var targetValue = vars.get(name) + howMuch;
            vars.set(name, targetValue);
            return targetValue;
        },
        datePlusMinutesIso: function (minutes) {
            return new Date(Date.now() + minutes * 60e3).toISOString();
        },
        timestampPlusMinutes: function (minutes) {
            return Date.now() + minutes * 60e3;
        },
    };
}