@mojir/lits 2.0.8 → 2.0.10

package/dist/src/builtin/index.d.ts

@@ -29,8 +29,11 @@ declare const specialExpressions: {
     readonly '??': import("./interface").BuiltinSpecialExpression<import("../interface").Any, import("./specialExpressions/qq").QqNode>;
 };
 export type SpecialExpressionName = keyof typeof specialExpressions;
+export type CommonSpecialExpressionName = keyof Pick<typeof specialExpressions, '??' | '??' | 'and' | 'comment' | 'cond' | 'declared?' | 'do' | 'if' | 'if-not' | 'or' | 'time!' | 'when' | 'when-not' | 'throw'>;
 export type BuiltinSpecialExpressions = typeof specialExpressions;
 export type BuiltinSpecialExpression = typeof specialExpressions[SpecialExpressionName];
+export type BuiltinCommonSpecialExpression = typeof specialExpressions[CommonSpecialExpressionName];
+export type GenericCommonSpecialExpressionNode<T extends CommonSpecialExpressionName> = ReturnType<typeof specialExpressions[T]['parse']>;
 export type SpecialExpressionNode = ReturnType<typeof specialExpressions[SpecialExpressionName]['parse']>;
 export declare const builtin: Builtin;
 export declare const normalExpressionKeys: string[];

package/dist/src/builtin/interface.d.ts

@@ -9,10 +9,11 @@ import type { BuiltinSpecialExpressions, SpecialExpressionNode } from '.';
 export type NormalExpressionEvaluator<T> = (params: Arr, sourceCodeInfo: SourceCodeInfo | undefined, contextStack: ContextStack, { executeFunction }: {
     executeFunction: ExecuteFunction;
 }) => T;
-type ValidateNode = (node: NormalExpressionNode) => void;
+type ValidateNormalExpressionNode = (node: NormalExpressionNode) => void;
+type ValidateSpecialExpressionNode = (node: SpecialExpressionNode) => void;
 interface BuiltinNormalExpression<T> {
     evaluate: NormalExpressionEvaluator<T>;
-    validate?: ValidateNode;
+    validate?: ValidateNormalExpressionNode;
 }
 export interface ParserHelpers {
     parseExpression: ParseExpression;
@@ -30,6 +31,7 @@ interface EvaluateHelpers {
 export interface BuiltinSpecialExpression<T, N extends SpecialExpressionNode> {
     parse: (tokenStream: TokenStream, parseState: ParseState, firstToken: Token, parsers: ParserHelpers) => N;
     evaluate: (node: N, contextStack: ContextStack, helpers: EvaluateHelpers) => T;
+    validateParameterCount: ValidateSpecialExpressionNode;
     findUnresolvedIdentifiers: (node: N, contextStack: ContextStack, params: {
         findUnresolvedIdentifiers: FindUnresolvedIdentifiers;
         builtin: Builtin;

package/dist/src/builtin/specialExpressions/commonParser.d.ts

@@ -0,0 +1,4 @@
+import type { CommonSpecialExpressionName, GenericCommonSpecialExpressionNode } from '..';
+import type { Any } from '../../interface';
+import type { BuiltinSpecialExpression } from '../interface';
+export declare function getCommonParser<T extends CommonSpecialExpressionName>(name: T): BuiltinSpecialExpression<Any, GenericCommonSpecialExpressionNode<T>>['parse'];

package/dist/src/parser/AlgebraicParser.d.ts

@@ -11,7 +11,8 @@ export declare class AlgebraicParser {
     private parseObject;
     private parseArray;
     private parseFunctionCall;
-    parseLamdaFunction(): AstNode | null;
+    parseLambdaFunction(): AstNode | null;
+    private parseShorthandLamdaFunction;
     private isAtEnd;
     private peek;
 }

package/dist/testFramework.esm.js

@@ -3124,7 +3124,7 @@ var mathNormalExpression = {
     },
 };
 
-var version = "2.0.8";
+var version = "2.0.10";
 
 var uuidTemplate = 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx';
 var xyRegexp = /[xy]/g;
@@ -4339,6 +4339,7 @@ var andSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -4366,42 +4367,30 @@ var andSpecialExpression = {
     },
 };
 
-var commentSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseToken = _a.parseToken;
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'comment',
-            p: [],
-            token: undefined,
-        };
-        var tkn = asToken(tokenStream.tokens[parseState.position]);
-        while (!isRParenToken(tkn)) {
-            node.p.push(parseToken(tokenStream, parseState));
-            tkn = asToken(tokenStream.tokens[parseState.position]);
-        }
-        parseState.position += 1;
-        node.token = getTokenDebugData(firstToken) && firstToken;
-        return node;
-    },
-    evaluate: function () { return null; },
-    findUnresolvedIdentifiers: function () { return new Set(); },
-};
-
-var condSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
+function getCommonParser(name) {
+    return function (tokenStream, parseState, firstToken, _a) {
         var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
         var params = parseTokensUntilClosingBracket(tokenStream, parseState);
         assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
+        return {
             t: AstNodeType.SpecialExpression,
-            n: 'cond',
+            n: name,
             p: params,
             token: getTokenDebugData(firstToken) && firstToken,
         };
-        assertEvenNumberOfParams(node);
-        return node;
-    },
+    };
+}
+
+var commentSpecialExpression = {
+    parse: getCommonParser('comment'),
+    validateParameterCount: function () { return undefined; },
+    evaluate: function () { return null; },
+    findUnresolvedIdentifiers: function () { return new Set(); },
+};
+
+var condSpecialExpression = {
+    parse: getCommonParser('cond'),
+    validateParameterCount: function (node) { return assertEvenNumberOfParams(node); },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -4430,19 +4419,8 @@ var condSpecialExpression = {
 };
 
 var declaredSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'declared?',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams(1, node);
-        return node;
-    },
+    parse: getCommonParser('declared?'),
+    validateParameterCount: function (node) { return assertNumberOfParams(1, node); },
     evaluate: function (node, contextStack) {
         var lookUpResult = contextStack.lookUp(node.p[0]);
         return lookUpResult !== null;
@@ -4487,9 +4465,9 @@ var defSpecialExpression = {
             token: getTokenDebugData(firstToken) && firstToken,
         };
         assertSymbolNode(node.p[0], (_b = getTokenDebugData(node.token)) === null || _b === void 0 ? void 0 : _b.sourceCodeInfo);
-        assertNumberOfParams(2, node);
         return node;
     },
+    validateParameterCount: function (node) { return assertNumberOfParams(2, node); },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var evaluateAstNode = _a.evaluateAstNode, builtin = _a.builtin;
@@ -4525,9 +4503,9 @@ var defsSpecialExpression = {
             p: params,
             token: getTokenDebugData(firstToken) && firstToken,
         };
-        assertNumberOfParams(2, node);
         return node;
     },
+    validateParameterCount: function (node) { return assertNumberOfParams(2, node); },
     evaluate: function (node, contextStack, _a) {
         var _b, _c;
         var evaluateAstNode = _a.evaluateAstNode, builtin = _a.builtin;
@@ -4555,23 +4533,8 @@ var defsSpecialExpression = {
 };
 
 var doSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseToken = _a.parseToken;
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'do',
-            p: [],
-            token: undefined,
-        };
-        var tkn = asToken(tokenStream.tokens[parseState.position]);
-        while (!isRParenToken(tkn)) {
-            node.p.push(parseToken(tokenStream, parseState));
-            tkn = asToken(tokenStream.tokens[parseState.position]);
-        }
-        parseState.position += 1;
-        node.token = getTokenDebugData(firstToken) && firstToken;
-        return node;
-    },
+    parse: getCommonParser('do'),
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -4648,6 +4611,7 @@ var defnSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var _c, _d;
@@ -4689,6 +4653,7 @@ var defnsSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var _c, _d;
@@ -4732,6 +4697,7 @@ var fnSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var _c;
@@ -4963,19 +4929,8 @@ function parseFunctionArguments(tokenStream, parseState, parsers) {
 }
 
 var ifSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'if',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams({ min: 2, max: 3 }, node);
-        return node;
-    },
+    parse: getCommonParser('if'),
+    validateParameterCount: function (node) { return assertNumberOfParams({ min: 2, max: 3 }, node); },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5014,9 +4969,9 @@ var ifLetSpecialExpression = {
             p: params,
             token: getTokenDebugData(firstToken) && firstToken,
         };
-        assertNumberOfParams({ min: 1, max: 2 }, node);
         return node;
     },
+    validateParameterCount: function (node) { return assertNumberOfParams({ min: 1, max: 2 }, node); },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5046,19 +5001,8 @@ var ifLetSpecialExpression = {
 };
 
 var ifNotSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'if-not',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams({ min: 2, max: 3 }, node);
-        return node;
-    },
+    parse: getCommonParser('if-not'),
+    validateParameterCount: function (node) { return assertNumberOfParams({ min: 2, max: 3 }, node); },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5095,6 +5039,7 @@ var letSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b, e_2, _c;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5166,6 +5111,7 @@ var loopSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5429,9 +5375,9 @@ var forSpecialExpression = {
             p: params,
             token: getTokenDebugData(firstToken) && firstToken,
         };
-        assertNumberOfParams(1, node);
         return node;
     },
+    validateParameterCount: function (node) { return assertNumberOfParams(1, node); },
     evaluate: function (node, contextStack, helpers) { return evaluateLoop(true, node, contextStack, helpers.evaluateAstNode); },
     findUnresolvedIdentifiers: function (node, contextStack, _a) {
         var findUnresolvedIdentifiers = _a.findUnresolvedIdentifiers, builtin = _a.builtin;
@@ -5451,9 +5397,9 @@ var doseqSpecialExpression = {
             p: params,
             token: getTokenDebugData(firstToken) && firstToken,
         };
-        assertNumberOfParams(1, node);
         return node;
     },
+    validateParameterCount: function (node) { return assertNumberOfParams(1, node); },
     evaluate: function (node, contextStack, helpers) {
         evaluateLoop(false, node, contextStack, helpers.evaluateAstNode);
         return null;
@@ -5465,18 +5411,8 @@ var doseqSpecialExpression = {
 };
 
 var orSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'or',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        return node;
-    },
+    parse: getCommonParser('or'),
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5505,19 +5441,8 @@ var orSpecialExpression = {
 };
 
 var qqSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: '??',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams({ min: 1, max: 2 }, node);
-        return node;
-    },
+    parse: getCommonParser('??'),
+    validateParameterCount: function (node) { return assertNumberOfParams({ min: 1, max: 2 }, node); },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5549,6 +5474,7 @@ var recurSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var evaluateAstNode = _a.evaluateAstNode;
         var params = node.p.map(function (paramNode) { return evaluateAstNode(paramNode, contextStack); });
@@ -5561,19 +5487,8 @@ var recurSpecialExpression = {
 };
 
 var throwSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'throw',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams(1, node);
-        return node;
-    },
+    parse: getCommonParser('throw'),
+    validateParameterCount: function (node) { return assertNumberOfParams(1, node); },
     evaluate: function (node, contextStack, _a) {
         var _b, _c;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5589,19 +5504,8 @@ var throwSpecialExpression = {
 };
 
 var timeSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'time!',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams(1, node);
-        return node;
-    },
+    parse: getCommonParser('time!'),
+    validateParameterCount: function (node) { return assertNumberOfParams(1, node); },
     evaluate: function (node, contextStack, _a) {
         var evaluateAstNode = _a.evaluateAstNode;
         var param = node.p[0];
@@ -5642,9 +5546,9 @@ var trySpecialExpression = {
             e: error,
             token: getTokenDebugData(firstToken) && firstToken,
         };
-        assertNumberOfParams(1, node);
         return node;
     },
+    validateParameterCount: function (node) { return assertNumberOfParams(1, node); },
     evaluate: function (node, contextStack, _a) {
         var _b;
         var _c;
@@ -5674,19 +5578,8 @@ var trySpecialExpression = {
 };
 
 var whenSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'when',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams({ min: 1 }, node);
-        return node;
-    },
+    parse: getCommonParser('when'),
+    validateParameterCount: function (node) { return assertNumberOfParams({ min: 1 }, node); },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var _c;
@@ -5736,6 +5629,7 @@ var whenFirstSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var _c;
@@ -5797,6 +5691,7 @@ var whenLetSpecialExpression = {
         };
         return node;
     },
+    validateParameterCount: function () { return undefined; },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var evaluateAstNode = _a.evaluateAstNode;
@@ -5835,19 +5730,8 @@ var whenLetSpecialExpression = {
 };
 
 var whenNotSpecialExpression = {
-    parse: function (tokenStream, parseState, firstToken, _a) {
-        var parseTokensUntilClosingBracket = _a.parseTokensUntilClosingBracket;
-        var params = parseTokensUntilClosingBracket(tokenStream, parseState);
-        assertRParenToken(tokenStream.tokens[parseState.position++]);
-        var node = {
-            t: AstNodeType.SpecialExpression,
-            n: 'when-not',
-            p: params,
-            token: getTokenDebugData(firstToken) && firstToken,
-        };
-        assertNumberOfParams({ min: 1 }, node);
-        return node;
-    },
+    parse: getCommonParser('when-not'),
+    validateParameterCount: function (node) { return assertNumberOfParams({ min: 1 }, node); },
     evaluate: function (node, contextStack, _a) {
         var e_1, _b;
         var _c;
@@ -7075,6 +6959,7 @@ function parseString(tokenStream, parseState) {
 }
 
 var exponentiationPrecedence = 9;
+var placeholderRegexp$1 = /^\$([1-9]\d?)?$/;
 function getPrecedence(operator) {
     var operatorSign = operator[1];
     switch (operatorSign) {
@@ -7307,7 +7192,7 @@ var AlgebraicParser = /** @class */ (function () {
         // Parentheses
         if (isLParenToken(token)) {
             var positionBefore = this.parseState.position;
-            var lamdaFunction = this.parseLamdaFunction();
+            var lamdaFunction = this.parseLambdaFunction();
             if (lamdaFunction) {
                 return lamdaFunction;
             }
@@ -7328,6 +7213,9 @@ var AlgebraicParser = /** @class */ (function () {
                 var operand = this.parseOperand();
                 return fromUnaryAlgebraicToAstNode(token, operand);
             }
+            else if (operatorName === '=>') {
+                return this.parseShorthandLamdaFunction();
+            }
             else {
                 throw new Error("Unknown unary operator: ".concat(operatorName));
             }
@@ -7443,7 +7331,7 @@ var AlgebraicParser = /** @class */ (function () {
         };
     };
     AlgebraicParser.prototype.parseFunctionCall = function (symbol) {
-        var _a, _b;
+        var _a, _b, _c;
         var params = [];
         this.advance();
         while (!this.isAtEnd() && !isRParenToken(this.peek())) {
@@ -7475,13 +7363,17 @@ var AlgebraicParser = /** @class */ (function () {
                     case 'when':
                     case 'when-not':
                     case 'do':
-                    case 'throw':
-                        return {
+                    case 'time!':
+                    case 'throw': {
+                        var node_1 = {
                             t: AstNodeType.SpecialExpression,
                             n: name_1,
                             p: params,
                             token: getTokenDebugData(symbol.token) && symbol.token,
                         };
+                        builtin.specialExpressions[node_1.n].validateParameterCount(node_1);
+                        return node_1;
+                    }
                     case 'def':
                     case 'defs':
                     case 'let':
@@ -7494,7 +7386,6 @@ var AlgebraicParser = /** @class */ (function () {
                     case 'try':
                     case 'recur':
                     case 'loop':
-                    case 'time!':
                     case 'doseq':
                     case 'for':
                         throw new Error("Special expression ".concat(name_1, " is not available in algebraic notation"));
@@ -7502,7 +7393,12 @@ var AlgebraicParser = /** @class */ (function () {
                         throw new Error("Unknown special expression: ".concat(name_1));
                 }
             }
-            return createNamedNormalExpressionNode(symbol.v, params, symbol.token);
+            var node = createNamedNormalExpressionNode(symbol.v, params, symbol.token);
+            var builtinExpression = builtin.normalExpressions[node.n];
+            if (builtinExpression) {
+                (_c = builtinExpression.validate) === null || _c === void 0 ? void 0 : _c.call(builtinExpression, __assign(__assign({}, node), { p: withoutCommentNodes(node.p) }));
+            }
+            return node;
         }
         else {
             return {
@@ -7513,7 +7409,7 @@ var AlgebraicParser = /** @class */ (function () {
             };
         }
     };
-    AlgebraicParser.prototype.parseLamdaFunction = function () {
+    AlgebraicParser.prototype.parseLambdaFunction = function () {
         var _a;
         var firstToken = this.peek();
         this.advance();
@@ -7572,6 +7468,62 @@ var AlgebraicParser = /** @class */ (function () {
             token: getTokenDebugData(firstToken) && firstToken,
         };
     };
+    AlgebraicParser.prototype.parseShorthandLamdaFunction = function () {
+        var _a, _b, _c, _d;
+        var firstToken = this.peek();
+        this.advance();
+        var startPos = this.parseState.position;
+        var exprNode = this.parseExpression();
+        var endPos = this.parseState.position - 1;
+        var arity = 0;
+        var percent1 = 'NOT_SET'; // referring to argument bindings. % = NAKED, %1, %2, %3, etc = WITH_1
+        for (var pos = startPos; pos <= endPos; pos += 1) {
+            var tkn = this.tokenStream.tokens[pos];
+            if (isA_SymbolToken(tkn)) {
+                var match = placeholderRegexp$1.exec(tkn[1]);
+                if (match) {
+                    var number = (_a = match[1]) !== null && _a !== void 0 ? _a : '1';
+                    if (number === '1') {
+                        var mixedPercent1 = (!match[1] && percent1 === 'WITH_1') || (match[1] && percent1 === 'NAKED');
+                        if (mixedPercent1)
+                            throw new LitsError('Please make up your mind, either use $ or $1', (_b = getTokenDebugData(firstToken)) === null || _b === void 0 ? void 0 : _b.sourceCodeInfo);
+                        percent1 = match[1] ? 'WITH_1' : 'NAKED';
+                    }
+                    arity = Math.max(arity, Number(number));
+                    if (arity > 20)
+                        throw new LitsError('Can\'t specify more than 20 arguments', (_c = getTokenDebugData(firstToken)) === null || _c === void 0 ? void 0 : _c.sourceCodeInfo);
+                }
+            }
+            if (isA_OperatorToken(tkn, '=>')) {
+                throw new LitsError('Nested shortcut functions are not allowed', (_d = getTokenDebugData(firstToken)) === null || _d === void 0 ? void 0 : _d.sourceCodeInfo);
+            }
+        }
+        var mandatoryArguments = [];
+        for (var i = 1; i <= arity; i += 1) {
+            if (i === 1 && percent1 === 'NAKED')
+                mandatoryArguments.push('$');
+            else
+                mandatoryArguments.push("$".concat(i));
+        }
+        var args = {
+            b: [],
+            m: mandatoryArguments,
+        };
+        var node = {
+            t: AstNodeType.SpecialExpression,
+            n: 'fn',
+            p: [],
+            o: [
+                {
+                    as: args,
+                    b: [exprNode],
+                    a: args.m.length,
+                },
+            ],
+            token: getTokenDebugData(firstToken) && firstToken,
+        };
+        return node;
+    };
     AlgebraicParser.prototype.isAtEnd = function () {
         return this.parseState.position >= this.tokenStream.tokens.length;
     };
@@ -7815,7 +7767,7 @@ function parseSpecialExpression(tokenStream, parseState) {
     var firstToken = asLParenToken(tokenStream.tokens[parseState.position++]);
     var nameToken = asP_SymbolToken(tokenStream.tokens[parseState.position++]);
     var expressionName = nameToken[1];
-    var parse = asNonUndefined(builtin.specialExpressions[expressionName], (_a = getTokenDebugData(nameToken)) === null || _a === void 0 ? void 0 : _a.sourceCodeInfo).parse;
+    var _b = asNonUndefined(builtin.specialExpressions[expressionName], (_a = getTokenDebugData(nameToken)) === null || _a === void 0 ? void 0 : _a.sourceCodeInfo), parse = _b.parse, validateParameterCount = _b.validateParameterCount;
     var node = parse(tokenStream, parseState, firstToken, {
         parseExpression: parseExpression,
         parseTokensUntilClosingBracket: parseTokensUntilClosingBracket,
@@ -7824,6 +7776,7 @@ function parseSpecialExpression(tokenStream, parseState) {
         parseBindings: parseBindings,
         parseArgument: parseArgument,
     });
+    validateParameterCount(node);
     return node;
 }
 function parsePolishToken(tokenStream, parseState) {