@mojir/lits 2.1.2 → 2.1.3

package/dist/index.js

@@ -95,9 +95,15 @@ function getCodeMarker(sourceCodeInfo) {
 }
 
 function getLitsErrorMessage(message, sourceCodeInfo) {
-    var filePathLine = (sourceCodeInfo === null || sourceCodeInfo === void 0 ? void 0 : sourceCodeInfo.filePath) ? "\n".concat(sourceCodeInfo.filePath) : '';
-    var codeLine = (sourceCodeInfo === null || sourceCodeInfo === void 0 ? void 0 : sourceCodeInfo.code) ? "\n".concat(sourceCodeInfo.code) : '';
-    var codeMarker = sourceCodeInfo && codeLine ? "\n".concat(getCodeMarker(sourceCodeInfo)) : '';
+    if (!sourceCodeInfo) {
+        return message;
+    }
+    var location = "".concat(sourceCodeInfo.position.line, ":").concat(sourceCodeInfo.position.column);
+    var filePathLine = sourceCodeInfo.filePath
+        ? "\n".concat(sourceCodeInfo.filePath, ":").concat(location)
+        : "\nLocation ".concat(location);
+    var codeLine = "\n".concat(sourceCodeInfo.code);
+    var codeMarker = "\n".concat(getCodeMarker(sourceCodeInfo));
     return "".concat(message).concat(filePathLine).concat(codeLine).concat(codeMarker);
 }
 var RecurSignal = /** @class */ (function (_super) {
@@ -210,6 +216,7 @@ var functionTypes = [
     'SomePred',
     'Fnull',
     'Builtin',
+    'SpecialBuiltin',
     'NativeJsFunction',
 ];
 var functionTypeSet = new Set(functionTypes);
@@ -3671,6 +3678,26 @@ var andSpecialExpression = {
         }
         return value;
     },
+    evaluateAsNormalExpression: function (params, sourceCodeInfo) {
+        var e_2, _a;
+        var value = true;
+        try {
+            for (var params_1 = __values(params), params_1_1 = params_1.next(); !params_1_1.done; params_1_1 = params_1.next()) {
+                var param = params_1_1.value;
+                value = asAny(param, sourceCodeInfo);
+                if (!value)
+                    break;
+            }
+        }
+        catch (e_2_1) { e_2 = { error: e_2_1 }; }
+        finally {
+            try {
+                if (params_1_1 && !params_1_1.done && (_a = params_1.return)) _a.call(params_1);
+            }
+            finally { if (e_2) throw e_2.error; }
+        }
+        return value;
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols(node[1][1], contextStack, builtin, evaluateNode);
@@ -3955,7 +3982,7 @@ var defSpecialExpression = {
         var value = bindingNode[1][1];
         var bindingValue = evaluateNode(value, contextStack);
         var values = evalueateBindingNodeValues(target, bindingValue, function (Node) { return evaluateNode(Node, contextStack); });
-        contextStack.exportValues(values);
+        contextStack.exportValues(values, target[2]);
         return null;
     },
     getUndefinedSymbols: function (node, contextStack, _a) {
@@ -3967,7 +3994,7 @@ var defSpecialExpression = {
         walkDefaults(target, function (defaultNode) {
             addToSet(bindingResult, getUndefinedSymbols([defaultNode], contextStack, builtin, evaluateNode));
         });
-        contextStack.addValues(getAllBindingTargetNames(target));
+        contextStack.addValues(getAllBindingTargetNames(target), target[2]);
         return bindingResult;
     },
 };
@@ -4110,14 +4137,14 @@ var functionSpecialExpression = {
             _b.name = functionSymbol[1],
             _b.evaluatedfunction = evaluatedFunction,
             _b);
-        contextStack.addValues((_c = {}, _c[functionSymbol[1]] = litsFunction, _c));
+        contextStack.addValues((_c = {}, _c[functionSymbol[1]] = litsFunction, _c), functionSymbol[2]);
         return null;
     },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var _b, _c;
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         var functionName = node[1][1][1];
-        contextStack.addValues((_b = {}, _b[functionName] = true, _b));
+        contextStack.addValues((_b = {}, _b[functionName] = true, _b), node[1][1][2]);
         var newContext = (_c = {}, _c[functionName] = { value: true }, _c);
         return getFunctionUnresolvedSymbols(node[1][2], contextStack, getUndefinedSymbols, builtin, evaluateNode, newContext);
     },
@@ -4138,7 +4165,7 @@ var defnSpecialExpression = {
             _b.name = functionSymbol[1],
             _b.evaluatedfunction = evaluatedFunctionOverloades,
             _b);
-        contextStack.exportValues((_c = {}, _c[functionSymbol[1]] = litsFunction, _c));
+        contextStack.exportValues((_c = {}, _c[functionSymbol[1]] = litsFunction, _c), functionSymbol[2]);
         return null;
     },
     getUndefinedSymbols: function (node, contextStack, _a) {
@@ -4146,7 +4173,7 @@ var defnSpecialExpression = {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         var functionName = node[1][1][1];
         var fn = node[1][2];
-        contextStack.exportValues((_b = {}, _b[functionName] = true, _b));
+        contextStack.exportValues((_b = {}, _b[functionName] = true, _b), node[1][1][2]);
         var newContext = (_c = {}, _c[functionName] = { value: true }, _c);
         return getFunctionUnresolvedSymbols(fn, contextStack, getUndefinedSymbols, builtin, evaluateNode, newContext);
     },
@@ -4258,7 +4285,7 @@ var letSpecialExpression = {
         var value = bindingNode[1][1];
         var bindingValue = evaluateNode(value, contextStack);
         var values = evalueateBindingNodeValues(target, bindingValue, function (Node) { return evaluateNode(Node, contextStack); });
-        contextStack.addValues(values);
+        contextStack.addValues(values, target[2]);
         return null;
     },
     getUndefinedSymbols: function (node, contextStack, _a) {
@@ -4270,7 +4297,7 @@ var letSpecialExpression = {
         walkDefaults(target, function (defaultNode) {
             addToSet(bindingResult, getUndefinedSymbols([defaultNode], contextStack, builtin, evaluateNode));
         });
-        contextStack.addValues(getAllBindingTargetNames(target));
+        contextStack.addValues(getAllBindingTargetNames(target), target[2]);
         return bindingResult;
     },
 };
@@ -4541,6 +4568,26 @@ var orSpecialExpression = {
         }
         return value;
     },
+    evaluateAsNormalExpression: function (params, sourceCodeInfo) {
+        var e_2, _a;
+        var value = false;
+        try {
+            for (var params_1 = __values(params), params_1_1 = params_1.next(); !params_1_1.done; params_1_1 = params_1.next()) {
+                var param = params_1_1.value;
+                value = asAny(param, sourceCodeInfo);
+                if (value)
+                    break;
+            }
+        }
+        catch (e_2_1) { e_2 = { error: e_2_1 }; }
+        finally {
+            try {
+                if (params_1_1 && !params_1_1.done && (_a = params_1.return)) _a.call(params_1);
+            }
+            finally { if (e_2) throw e_2.error; }
+        }
+        return value;
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols(node[1][1], contextStack, builtin, evaluateNode);
@@ -4559,6 +4606,11 @@ var qqSpecialExpression = {
         var firstResult = evaluateNode(firstNode, contextStack);
         return firstResult !== null && firstResult !== void 0 ? firstResult : (secondNode ? evaluateNode(secondNode, contextStack) : null);
     },
+    evaluateAsNormalExpression: function (params, sourceCodeInfo) {
+        var firstParam = asAny(params[0], sourceCodeInfo);
+        var secondParam = params[1] !== undefined ? asAny(params[1], sourceCodeInfo) : null;
+        return firstParam !== null && firstParam !== void 0 ? firstParam : secondParam;
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols(node[1][1].filter(function (n) { return !!n; }), contextStack, builtin, evaluateNode);
@@ -4573,6 +4625,9 @@ var recurSpecialExpression = {
         var evaluatedParams = params.map(function (paramNode) { return evaluateNode(paramNode, contextStack); });
         throw new RecurSignal(evaluatedParams);
     },
+    evaluateAsNormalExpression: function (params) {
+        throw new RecurSignal(params);
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols(node[1][1], contextStack, builtin, evaluateNode);
@@ -4588,6 +4643,12 @@ var throwSpecialExpression = {
         });
         throw new UserDefinedError(message, node[2]);
     },
+    evaluateAsNormalExpression: function (params, sourceCodeInfo) {
+        var message = asString(params[0], sourceCodeInfo, {
+            nonEmpty: true,
+        });
+        throw new UserDefinedError(message, undefined);
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols([node[1][1]], contextStack, builtin, evaluateNode);
@@ -4655,6 +4716,24 @@ var arraySpecialExpression = {
         }
         return result;
     },
+    evaluateAsNormalExpression: function (params, sourceCodeInfo) {
+        var e_2, _a;
+        var result = [];
+        try {
+            for (var params_1 = __values(params), params_1_1 = params_1.next(); !params_1_1.done; params_1_1 = params_1.next()) {
+                var param = params_1_1.value;
+                result.push(asAny(param, sourceCodeInfo));
+            }
+        }
+        catch (e_2_1) { e_2 = { error: e_2_1 }; }
+        finally {
+            try {
+                if (params_1_1 && !params_1_1.done && (_a = params_1.return)) _a.call(params_1);
+            }
+            finally { if (e_2) throw e_2.error; }
+        }
+        return result;
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols(node[1][1], contextStack, builtin, evaluateNode);
@@ -4686,6 +4765,16 @@ var objectSpecialExpression = {
         }
         return result;
     },
+    evaluateAsNormalExpression: function (params, sourceCodeInfo) {
+        var result = {};
+        for (var i = 0; i < params.length; i += 2) {
+            var key = params[i];
+            var value = params[i + 1];
+            assertString(key, sourceCodeInfo);
+            result[key] = value !== null && value !== void 0 ? value : null;
+        }
+        return result;
+    },
     getUndefinedSymbols: function (node, contextStack, _a) {
         var getUndefinedSymbols = _a.getUndefinedSymbols, builtin = _a.builtin, evaluateNode = _a.evaluateNode;
         return getUndefinedSymbols(node[1][1], contextStack, builtin, evaluateNode);
@@ -4811,7 +4900,7 @@ function findUnresolvedSymbolsInNode(node, contextStack, builtin, evaluateNode)
             return findUnresolvedSymbolsInNode(node[1], contextStack, builtin, evaluateNode);
         /* v8 ignore next 2 */
         default:
-            throw new Error("Unhandled node type: ".concat(nodeType));
+            throw new LitsError("Unhandled node type: ".concat(nodeType), node[2]);
     }
 }
 
@@ -5019,6 +5108,16 @@ var functionExecutors = {
         var normalExpression = asNonUndefined(allNormalExpressions[fn.normalBuitinSymbolType], sourceCodeInfo);
         return normalExpression.evaluate(params, sourceCodeInfo, contextStack, { executeFunction: executeFunction });
     },
+    SpecialBuiltin: function (fn, params, sourceCodeInfo, contextStack, _a) {
+        var executeFunction = _a.executeFunction;
+        var specialExpression = asNonUndefined(specialExpressions[fn.specialBuiltinSymbolType], sourceCodeInfo);
+        if (specialExpression.evaluateAsNormalExpression) {
+            return specialExpression.evaluateAsNormalExpression(params, sourceCodeInfo, contextStack, { executeFunction: executeFunction });
+        }
+        else {
+            throw new LitsError("Special builtin function ".concat(fn.specialBuiltinSymbolType, " is not supported as normal expression."), sourceCodeInfo);
+        }
+    },
 };
 
 function evaluate(ast, contextStack) {
@@ -5074,7 +5173,21 @@ function evaluateReservedSymbol(node) {
 function evaluateNormalExpression(node, contextStack) {
     var sourceCodeInfo = node[2];
     var paramNodes = node[1][1];
-    var params = paramNodes.map(function (paramNode) { return evaluateNode(paramNode, contextStack); });
+    var params = [];
+    paramNodes.forEach(function (paramNode) {
+        if (isSpreadNode(paramNode)) {
+            var spreadValue = evaluateNode(paramNode[1], contextStack);
+            if (Array.isArray(spreadValue)) {
+                params.push.apply(params, __spreadArray([], __read(spreadValue), false));
+            }
+            else {
+                throw new LitsError("Spread operator requires an array, got ".concat(valueToString(paramNode)), paramNode[2]);
+            }
+        }
+        else {
+            params.push(evaluateNode(paramNode, contextStack));
+        }
+    });
     if (isNormalExpressionNodeWithName(node)) {
         var nameSymbol = node[1][0];
         if (isNormalBuiltinSymbolNode(nameSymbol)) {
@@ -5176,19 +5289,19 @@ var ContextStackImpl = /** @class */ (function () {
         var contexts = [{}, context];
         return new ContextStackImpl({ contexts: contexts });
     };
-    ContextStackImpl.prototype.exportValues = function (values) {
+    ContextStackImpl.prototype.exportValues = function (values, sourceCodeInfo) {
         var e_1, _a;
         try {
             for (var _b = __values(Object.entries(values)), _c = _b.next(); !_c.done; _c = _b.next()) {
                 var _d = __read(_c.value, 2), name_1 = _d[0], value = _d[1];
                 if (this.globalContext[name_1]) {
-                    throw new Error("Cannot redefine exported value \"".concat(name_1, "\""));
+                    throw new LitsError("Cannot redefine exported value \"".concat(name_1, "\""), sourceCodeInfo);
                 }
                 if (specialExpressionKeys.includes(name_1)) {
-                    throw new Error("Cannot shadow special expression \"".concat(name_1, "\""));
+                    throw new LitsError("Cannot shadow special expression \"".concat(name_1, "\""), sourceCodeInfo);
                 }
                 if (normalExpressionKeys.includes(name_1)) {
-                    throw new Error("Cannot shadow builtin function \"".concat(name_1, "\""));
+                    throw new LitsError("Cannot shadow builtin function \"".concat(name_1, "\""), sourceCodeInfo);
                 }
                 this.globalContext[name_1] = { value: value };
             }
@@ -5200,22 +5313,22 @@ var ContextStackImpl = /** @class */ (function () {
             }
             finally { if (e_1) throw e_1.error; }
         }
-        this.addValues(values);
+        this.addValues(values, sourceCodeInfo);
     };
-    ContextStackImpl.prototype.addValues = function (values) {
+    ContextStackImpl.prototype.addValues = function (values, sourceCodeInfo) {
         var e_2, _a;
         var currentContext = this.contexts[0];
         try {
             for (var _b = __values(Object.entries(values)), _c = _b.next(); !_c.done; _c = _b.next()) {
                 var _d = __read(_c.value, 2), name_2 = _d[0], value = _d[1];
                 if (currentContext[name_2]) {
-                    throw new Error("Cannot redefine value \"".concat(name_2, "\""));
+                    throw new LitsError("Cannot redefine value \"".concat(name_2, "\""), sourceCodeInfo);
                 }
                 if (specialExpressionKeys.includes(name_2)) {
-                    throw new Error("Cannot shadow special expression \"".concat(name_2, "\""));
+                    throw new LitsError("Cannot shadow special expression \"".concat(name_2, "\""), sourceCodeInfo);
                 }
                 if (normalExpressionKeys.includes(name_2)) {
-                    throw new Error("Cannot shadow builtin function \"".concat(name_2, "\""));
+                    throw new LitsError("Cannot shadow builtin function \"".concat(name_2, "\""), sourceCodeInfo);
                 }
                 currentContext[name_2] = { value: toAny(value) };
             }
@@ -5295,18 +5408,36 @@ var ContextStackImpl = /** @class */ (function () {
         return null;
     };
     ContextStackImpl.prototype.evaluateSymbol = function (node) {
-        var _a;
+        var _a, _b;
         if (isSpecialBuiltinSymbolNode(node)) {
-            throw new Error('Special builtin symbols should not be evaluated');
+            var functionType = node[1];
+            switch (functionType) {
+                case specialExpressionTypes['&&']:
+                case specialExpressionTypes['||']:
+                case specialExpressionTypes.array:
+                case specialExpressionTypes.object:
+                case specialExpressionTypes['defined?']:
+                case specialExpressionTypes.recur:
+                case specialExpressionTypes.throw:
+                case specialExpressionTypes['??']:
+                    return _a = {},
+                        _a[FUNCTION_SYMBOL] = true,
+                        _a.functionType = 'SpecialBuiltin',
+                        _a.specialBuiltinSymbolType = functionType,
+                        _a.sourceCodeInfo = node[2],
+                        _a;
+                default:
+                    throw new LitsError("Unknown special builtin symbol type: ".concat(functionType), node[2]);
+            }
         }
         if (isNormalBuiltinSymbolNode(node)) {
             var type = node[1];
-            return _a = {},
-                _a[FUNCTION_SYMBOL] = true,
-                _a.functionType = 'Builtin',
-                _a.normalBuitinSymbolType = type,
-                _a.sourceCodeInfo = node[2],
-                _a;
+            return _b = {},
+                _b[FUNCTION_SYMBOL] = true,
+                _b.functionType = 'Builtin',
+                _b.normalBuitinSymbolType = type,
+                _b.sourceCodeInfo = node[2],
+                _b;
         }
         var lookUpResult = this.lookUp(node);
         if (isContextEntry(lookUpResult))
@@ -5978,7 +6109,7 @@ function withSourceCodeInfo(node, sourceCodeInfo) {
     }
     return node;
 }
-function getPrecedence(operatorSign) {
+function getPrecedence(operatorSign, sourceCodeInfo) {
     switch (operatorSign) {
         case '**': // exponentiation
             return exponentiationPrecedence;
@@ -6017,7 +6148,7 @@ function getPrecedence(operatorSign) {
         // leave room for binaryFunctionalOperatorPrecedence = 1
         /* v8 ignore next 2 */
         default:
-            throw new Error("Unknown binary operator: ".concat(operatorSign));
+            throw new LitsError("Unknown binary operator: ".concat(operatorSign), sourceCodeInfo);
     }
 }
 function createNamedNormalExpressionNode(symbolNode, params, sourceCodeInfo) {
@@ -6150,7 +6281,7 @@ var Parser = /** @class */ (function () {
         while (!this.isAtExpressionEnd()) {
             if (isA_BinaryOperatorToken(operator)) {
                 var name_1 = operator[1];
-                var newPrecedece = getPrecedence(name_1);
+                var newPrecedece = getPrecedence(name_1, operator[2]);
                 if (newPrecedece <= precedence
                     // ** (exponentiation) is right associative
                     && !(newPrecedece === exponentiationPrecedence && precedence === exponentiationPrecedence)) {
@@ -6230,7 +6361,7 @@ var Parser = /** @class */ (function () {
             this.advance();
             var expression = this.parseExpression();
             if (!isRParenToken(this.peek())) {
-                throw new Error('Expected closing parenthesis');
+                throw new LitsError('Expected closing parenthesis', this.peek()[2]);
             }
             this.advance();
             return expression;
@@ -6352,7 +6483,13 @@ var Parser = /** @class */ (function () {
         this.advance();
         var params = [];
         while (!this.isAtEnd() && !isRParenToken(this.peek())) {
-            params.push(this.parseExpression());
+            if (isOperatorToken(this.peek(), '...')) {
+                this.advance();
+                params.push(withSourceCodeInfo([NodeTypes.Spread, this.parseExpression()], this.peek()[2]));
+            }
+            else {
+                params.push(this.parseExpression());
+            }
             var nextToken = this.peek();
             if (!isOperatorToken(nextToken, ',') && !isRParenToken(nextToken)) {
                 throw new LitsError('Expected comma or closing parenthesis', this.peek()[2]);
@@ -6408,7 +6545,7 @@ var Parser = /** @class */ (function () {
                     throw new LitsError("".concat(type, " is not allowed"), symbol[2]);
                 /* v8 ignore next 2 */
                 default:
-                    throw new Error("Unknown special expression: ".concat(type));
+                    throw new LitsError("Unknown special expression: ".concat(type), symbol[2]);
             }
         }
         else if (isNormalBuiltinSymbolNode(symbol) || isNormalBuiltinSymbolNode(symbol)) {