@mojir/lits 2.1.21 → 2.1.23

package/dist/cli/cli.js

@@ -92,7 +92,7 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
     return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
 };
 
-var version = "2.1.21";
+var version = "2.1.23";
 
 function getCodeMarker(sourceCodeInfo) {
     if (!sourceCodeInfo.position || !sourceCodeInfo.code)
@@ -4548,6 +4548,98 @@ function applyPlaceholders(templateString, placeholders, sourceCodeInfo) {
     return templateString;
 }
 
+function paramCountAccepts(paramsCount, nbrOfParams) {
+    if (typeof paramsCount === 'number') {
+        return paramsCount === nbrOfParams;
+    }
+    var min = paramsCount.min, max = paramsCount.max, even = paramsCount.even, odd = paramsCount.odd;
+    if (even && nbrOfParams % 2 !== 0) {
+        return false;
+    }
+    if (odd && nbrOfParams % 2 !== 1) {
+        return false;
+    }
+    if (typeof min === 'number' && nbrOfParams < min) {
+        return false;
+    }
+    if (typeof max === 'number' && nbrOfParams > max) {
+        return false;
+    }
+    return true;
+}
+function paramCountAcceptsMin(paramsCount, nbrOfParams) {
+    if (typeof paramsCount === 'number') {
+        return nbrOfParams >= paramsCount;
+    }
+    var min = paramsCount.min;
+    if (typeof min === 'number' && nbrOfParams < min) {
+        return false;
+    }
+    return true;
+}
+function getCommonParamCountFromFunctions(params) {
+    return params.reduce(function (acc, param) {
+        if (acc === null) {
+            return null;
+        }
+        var paramCount = (typeof param === 'number' || isColl(param)) ? 1 : param.paramCount;
+        if (typeof acc === 'number' && typeof paramCount === 'number') {
+            return acc === paramCount ? acc : null;
+        }
+        if (typeof paramCount === 'number') {
+            if (paramCountAccepts(acc, paramCount)) {
+                return paramCount;
+            }
+            return null;
+        }
+        if (typeof acc === 'number') {
+            if (paramCountAccepts(paramCount, acc)) {
+                return acc;
+            }
+            return null;
+        }
+        var aMin = paramCount.min, aMax = paramCount.max, aEven = paramCount.even, aOdd = paramCount.odd;
+        var bMin = acc.min, bMax = acc.max, bEven = acc.even, bOdd = acc.odd;
+        var min = typeof aMin === 'number' && typeof bMin === 'number'
+            ? Math.max(aMin, bMin)
+            : typeof aMin === 'number' ? aMin : typeof bMin === 'number' ? bMin : undefined;
+        var max = typeof aMax === 'number' && typeof bMax === 'number'
+            ? Math.min(aMax, bMax)
+            : typeof aMax === 'number' ? aMax : typeof bMax === 'number' ? bMax : undefined;
+        var even = aEven !== null && aEven !== void 0 ? aEven : bEven;
+        var odd = aOdd !== null && aOdd !== void 0 ? aOdd : bOdd;
+        if (even && odd) {
+            return null;
+        }
+        if (even) {
+            if (typeof min === 'number' && min % 2 !== 0) {
+                min += 1;
+            }
+            if (typeof max === 'number' && max % 2 !== 0) {
+                max -= 1;
+            }
+        }
+        if (odd) {
+            if (typeof min === 'number' && min % 2 === 0) {
+                min += 1;
+            }
+            if (typeof max === 'number' && max % 2 === 0) {
+                max -= 1;
+            }
+        }
+        if (typeof min === 'number' && typeof max === 'number' && min > max) {
+            return null;
+        }
+        if (typeof min === 'number' && min === max) {
+            return min;
+        }
+        return { min: min, max: max, even: even, odd: odd };
+    }, {});
+}
+function getParamCountFromFunction(param) {
+    return (typeof param === 'number' || isColl(param)) ? 1 : param.paramCount;
+}
+
 var functionalNormalExpression = {
     '|>': {
         evaluate: function (_a, sourceCodeInfo, contextStack, _b) {
@@ -4581,11 +4673,13 @@ var functionalNormalExpression = {
     'comp': {
         evaluate: function (params, sourceCodeInfo) {
             var _a;
+            params.forEach(function (param) { return assertFunctionLike(param, sourceCodeInfo); });
             return _a = {},
                 _a[FUNCTION_SYMBOL] = true,
                 _a.sourceCodeInfo = sourceCodeInfo,
                 _a.functionType = 'Comp',
                 _a.params = params,
+                _a.paramCount = params.length > 0 ? getParamCountFromFunction(params.at(-1)) : 1,
                 _a;
         },
         paramCount: {},
@@ -4599,6 +4693,7 @@ var functionalNormalExpression = {
                 _b.sourceCodeInfo = sourceCodeInfo,
                 _b.functionType = 'Constantly',
                 _b.value = toAny(value),
+                _b.paramCount = {},
                 _b;
         },
         paramCount: 1,
@@ -4606,11 +4701,17 @@ var functionalNormalExpression = {
     'juxt': {
         evaluate: function (params, sourceCodeInfo) {
             var _a;
+            params.forEach(function (param) { return assertFunctionLike(param, sourceCodeInfo); });
+            var paramCount = getCommonParamCountFromFunctions(params);
+            if (paramCount === null) {
+                throw new LitsError('All functions must accept the same number of arguments', sourceCodeInfo);
+            }
             return _a = {},
                 _a[FUNCTION_SYMBOL] = true,
                 _a.sourceCodeInfo = sourceCodeInfo,
                 _a.functionType = 'Juxt',
                 _a.params = params,
+                _a.paramCount = paramCount,
                 _a;
         },
         paramCount: { min: 1 },
@@ -4619,11 +4720,13 @@ var functionalNormalExpression = {
         evaluate: function (_a, sourceCodeInfo) {
             var _b;
             var _c = __read(_a, 1), fn = _c[0];
+            var fun = asFunctionLike(fn, sourceCodeInfo);
             return _b = {},
                 _b[FUNCTION_SYMBOL] = true,
                 _b.sourceCodeInfo = sourceCodeInfo,
                 _b.functionType = 'Complement',
-                _b.function = asFunctionLike(fn, sourceCodeInfo),
+                _b.function = fun,
+                _b.paramCount = getParamCountFromFunction(fun),
                 _b;
         },
         paramCount: 1,
@@ -4636,6 +4739,7 @@ var functionalNormalExpression = {
                 _a.sourceCodeInfo = sourceCodeInfo,
                 _a.functionType = 'EveryPred',
                 _a.params = params,
+                _a.paramCount = 1,
                 _a;
         },
         paramCount: { min: 1 },
@@ -4648,6 +4752,7 @@ var functionalNormalExpression = {
                 _a.sourceCodeInfo = sourceCodeInfo,
                 _a.functionType = 'SomePred',
                 _a.params = params,
+                _a.paramCount = 1,
                 _a;
         },
         paramCount: { min: 1 },
@@ -4656,12 +4761,14 @@ var functionalNormalExpression = {
         evaluate: function (_a, sourceCodeInfo) {
             var _b;
             var _c = __read(_a), fn = _c[0], params = _c.slice(1);
+            var fun = asFunctionLike(fn, sourceCodeInfo);
             return _b = {},
                 _b[FUNCTION_SYMBOL] = true,
                 _b.sourceCodeInfo = sourceCodeInfo,
                 _b.functionType = 'Fnull',
-                _b.function = asFunctionLike(fn, sourceCodeInfo),
+                _b.function = fun,
                 _b.params = params,
+                _b.paramCount = getParamCountFromFunction(fun),
                 _b;
         },
         paramCount: { min: 2 },
@@ -11736,12 +11843,16 @@ var functionSpecialExpression = {
         assertUserDefinedSymbolNode(functionSymbol, node[2]);
         assertNameNotDefined(functionSymbol[1], contextStack, builtin, node[2]);
         var evaluatedFunction = evaluateFunction(fn, contextStack, builtin, getUndefinedSymbols, evaluateNode);
+        var min = evaluatedFunction[0].filter(function (arg) { return arg[0] !== bindingTargetTypes.rest && arg[1][1] === undefined; }).length;
+        var max = evaluatedFunction[0].some(function (arg) { return arg[0] === bindingTargetTypes.rest; }) ? undefined : evaluatedFunction[0].length;
+        var paramCount = min === max ? min : { min: min, max: max };
         var litsFunction = (_b = {},
             _b[FUNCTION_SYMBOL] = true,
             _b.sourceCodeInfo = node[2],
             _b.functionType = 'UserDefined',
             _b.name = functionSymbol[1],
             _b.evaluatedfunction = evaluatedFunction,
+            _b.paramCount = paramCount,
             _b);
         contextStack.addValues((_c = {}, _c[functionSymbol[1]] = litsFunction, _c), functionSymbol[2]);
         return litsFunction;
@@ -11763,13 +11874,16 @@ var defnSpecialExpression = {
         var _d = __read(node[1], 3), functionSymbol = _d[1], fn = _d[2];
         assertUserDefinedSymbolNode(functionSymbol, node[2]);
         assertNameNotDefined(functionSymbol[1], contextStack, builtin, node[2]);
-        var evaluatedFunctionOverloades = evaluateFunction(fn, contextStack, builtin, getUndefinedSymbols, evaluateNode);
+        var evaluatedFunction = evaluateFunction(fn, contextStack, builtin, getUndefinedSymbols, evaluateNode);
+        var min = evaluatedFunction[0].filter(function (arg) { return arg[0] !== bindingTargetTypes.rest && arg[1][1] === undefined; }).length;
+        var paramCount = { min: min };
         var litsFunction = (_b = {},
             _b[FUNCTION_SYMBOL] = true,
             _b.sourceCodeInfo = node[2],
             _b.functionType = 'UserDefined',
             _b.name = functionSymbol[1],
-            _b.evaluatedfunction = evaluatedFunctionOverloades,
+            _b.evaluatedfunction = evaluatedFunction,
+            _b.paramCount = paramCount,
             _b);
         contextStack.exportValues((_c = {}, _c[functionSymbol[1]] = litsFunction, _c), functionSymbol[2]);
         return litsFunction;
@@ -11791,12 +11905,16 @@ var fnSpecialExpression = {
         var builtin = _a.builtin, getUndefinedSymbols = _a.getUndefinedSymbols, evaluateNode = _a.evaluateNode;
         var fn = node[1][1];
         var evaluatedFunction = evaluateFunction(fn, contextStack, builtin, getUndefinedSymbols, evaluateNode);
+        var min = evaluatedFunction[0].filter(function (arg) { return arg[0] !== bindingTargetTypes.rest && arg[1][1] === undefined; }).length;
+        var max = evaluatedFunction[0].some(function (arg) { return arg[0] === bindingTargetTypes.rest; }) ? undefined : evaluatedFunction[0].length;
+        var paramCount = min === max ? min : { min: min, max: max };
         var litsFunction = (_b = {},
             _b[FUNCTION_SYMBOL] = true,
             _b.sourceCodeInfo = node[2],
             _b.functionType = 'UserDefined',
             _b.name = undefined,
             _b.evaluatedfunction = evaluatedFunction,
+            _b.paramCount = paramCount,
             _b);
         return litsFunction;
     },
@@ -12412,12 +12530,6 @@ new Set(specialExpressionKeys);
 // TODO, remove
 // console.log('builtin', [...specialExpressionKeys, ...normalExpressionKeys].length)
 
-function checkParams(evaluatedFunction, nbrOfParams, sourceCodeInfo) {
-    var minArity = evaluatedFunction[0].filter(function (arg) { return arg[0] !== bindingTargetTypes.rest && arg[1][1] === undefined; }).length;
-    if (nbrOfParams < minArity) {
-        throw new LitsError("Unexpected number of arguments. Expected at least ".concat(minArity, ", got ").concat(nbrOfParams, "."), sourceCodeInfo);
-    }
-}
 var functionExecutors = {
     NativeJsFunction: function (fn, params, sourceCodeInfo) {
         var _a;
@@ -12437,7 +12549,10 @@ var functionExecutors = {
         var evaluateNode = _a.evaluateNode;
         var _loop_1 = function () {
             var e_1, _b;
-            checkParams(fn.evaluatedfunction, params.length, sourceCodeInfo);
+            if (!paramCountAcceptsMin(fn.paramCount, params.length)) {
+                throw new LitsError("Expected ".concat(fn.paramCount, " arguments, got ").concat(params.length, "."), sourceCodeInfo);
+            }
+            // checkParams(fn.evaluatedfunction, params.length, sourceCodeInfo)
             var evaluatedFunction = fn.evaluatedfunction;
             var args = evaluatedFunction[0];
             var nbrOfNonRestArgs = args.filter(function (arg) { return arg[0] !== bindingTargetTypes.rest; }).length;
@@ -12734,6 +12849,7 @@ function evaluateNormalExpression(node, contextStack) {
                 _a.params = params,
                 _a.placeholders = placeholders,
                 _a.sourceCodeInfo = sourceCodeInfo,
+                _a.paramCount = placeholders.length,
                 _a);
             return partialFunction;
         }
@@ -12756,11 +12872,12 @@ function evaluateNormalExpression(node, contextStack) {
         if (placeholders.length > 0) {
             var partialFunction = (_b = {},
                 _b[FUNCTION_SYMBOL] = true,
-                _b.function = asFunctionLike(fn, sourceCodeInfo),
+                _b.function = fn,
                 _b.functionType = 'Partial',
                 _b.params = params,
                 _b.placeholders = placeholders,
                 _b.sourceCodeInfo = sourceCodeInfo,
+                _b.paramCount = placeholders.length,
                 _b);
             return partialFunction;
         }
@@ -12982,11 +13099,13 @@ var ContextStackImpl = /** @class */ (function () {
         }
         if (isNormalBuiltinSymbolNode(node)) {
             var type = node[1];
+            var normalExpression = allNormalExpressions[type];
             return _b = {},
                 _b[FUNCTION_SYMBOL] = true,
                 _b.functionType = 'Builtin',
                 _b.normalBuitinSymbolType = type,
                 _b.sourceCodeInfo = node[2],
+                _b.paramCount = normalExpression.paramCount,
                 _b;
         }
         var lookUpResult = this.lookUp(node);
@@ -13008,7 +13127,8 @@ function createContextStack(params) {
         nativeJsFunctions: params.jsFunctions
             && Object.entries(params.jsFunctions).reduce(function (acc, _a) {
                 var _b;
-                var _c = __read(_a, 2), name = _c[0], jsFunction = _c[1];
+                var _c;
+                var _d = __read(_a, 2), name = _d[0], jsFunction = _d[1];
                 if (specialExpressionKeys.includes(name)) {
                     console.warn("Cannot shadow special expression \"".concat(name, "\", ignoring."));
                     return acc;
@@ -13023,6 +13143,7 @@ function createContextStack(params) {
                         name: name
                     },
                     _b[FUNCTION_SYMBOL] = true,
+                    _b.paramCount = (_c = jsFunction.paramCount) !== null && _c !== void 0 ? _c : {},
                     _b);
                 return acc;
             }, {}),

package/dist/cli/src/Lits/Lits.d.ts

@@ -3,6 +3,7 @@ import type { Any } from '../interface';
 import type { Ast, LitsFunction } from '../parser/types';
 import type { TokenStream } from '../tokenizer/tokenize';
 import { AutoCompleter } from '../AutoCompleter/AutoCompleter';
+import type { ParamCount } from '../builtin/interface';
 import { Cache } from './Cache';
 export interface LitsRuntimeInfo {
     astCache: Cache | null;
@@ -11,6 +12,7 @@ export interface LitsRuntimeInfo {
 }
 export interface JsFunction {
     fn: (...args: any[]) => unknown;
+    paramCount?: ParamCount;
 }
 export interface ContextParams {
     globalContext?: Context;

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

@@ -5,7 +5,7 @@ import type { Any, Arr } from '../interface';
 import type { SpecialExpressionNode } from '../parser/types';
 import type { SourceCodeInfo } from '../tokenizer/token';
 import type { SpecialExpressions } from '.';
-export type Count = number | {
+export type ParamCount = number | {
     min?: number;
     max?: number;
     even?: true;
@@ -16,7 +16,7 @@ export type NormalExpressionEvaluator<T> = (params: Arr, sourceCodeInfo: SourceC
 }) => T;
 export interface BuiltinNormalExpression<T> {
     evaluate: NormalExpressionEvaluator<T>;
-    paramCount: Count;
+    paramCount: ParamCount;
     aliases?: string[];
 }
 export type BuiltinNormalExpressions = Record<string, BuiltinNormalExpression<Any>>;
@@ -29,7 +29,7 @@ export interface EvaluateHelpers {
 export interface BuiltinSpecialExpression<T, N extends SpecialExpressionNode> {
     evaluate: (node: N, contextStack: ContextStack, helpers: EvaluateHelpers) => T;
     evaluateAsNormalExpression?: NormalExpressionEvaluator<T>;
-    paramCount: Count;
+    paramCount: ParamCount;
     getUndefinedSymbols: (node: N, contextStack: ContextStack, params: {
         getUndefinedSymbols: GetUndefinedSymbols;
         builtin: Builtin;

package/dist/cli/src/builtin/specialExpressions/functions.d.ts

@@ -1,4 +1,4 @@
-import type { LitsFunction, SpecialExpressionNode, SymbolNode } from '../../parser/types';
+import { type LitsFunction, type SpecialExpressionNode, type SymbolNode } from '../../parser/types';
 import type { BuiltinSpecialExpression } from '../interface';
 import type { Function } from '../utils';
 import type { specialExpressionTypes } from '../specialExpressionTypes';

package/dist/cli/src/parser/types.d.ts

@@ -1,5 +1,6 @@
 import type { JsFunction } from '../Lits/Lits';
 import type { SpecialExpressionType } from '../builtin';
+import type { ParamCount } from '../builtin/interface';
 import type { specialExpressionTypes } from '../builtin/specialExpressionTypes';
 import type { FunctionType, NodeType, NodeTypes } from '../constants/constants';
 import type { Context } from '../evaluator/interface';
@@ -15,6 +16,7 @@ interface GenericLitsFunction {
     [FUNCTION_SYMBOL]: true;
     sourceCodeInfo?: SourceCodeInfo;
     functionType: FunctionType;
+    paramCount: ParamCount;
 }
 export interface RegularExpression {
     [REGEXP_SYMBOL]: true;

package/dist/cli/src/typeGuards/index.d.ts

@@ -1,12 +1,12 @@
-import type { Count } from '../builtin/interface';
+import type { ParamCount } from '../builtin/interface';
 import type { UnknownRecord } from '../interface';
 import type { NormalExpressionNodeWithName } from '../parser/types';
 import type { SourceCodeInfo } from '../tokenizer/token';
-export declare function assertNumberOfParams(count: Count, node: NormalExpressionNodeWithName): void;
+export declare function assertNumberOfParams(count: ParamCount, node: NormalExpressionNodeWithName): void;
 export declare function isNonUndefined<T>(value: T | undefined): value is T;
 export declare function asNonUndefined<T>(value: T | undefined, sourceCodeInfo?: SourceCodeInfo): T;
 export declare function assertNonUndefined<T>(value: T | undefined, sourceCodeInfo?: SourceCodeInfo): asserts value is T;
 export declare function isUnknownRecord(value: unknown): value is Record<string, unknown>;
 export declare function assertUnknownRecord(value: unknown, sourceCodeInfo?: SourceCodeInfo): asserts value is UnknownRecord;
 export declare function asUnknownRecord(value: unknown, sourceCodeInfo?: SourceCodeInfo): UnknownRecord;
-export declare function canBeOperator(count: Count): boolean;
+export declare function canBeOperator(count: ParamCount): boolean;

package/dist/cli/src/utils/index.d.ts

@@ -1,5 +1,4 @@
 import type { Any, Coll } from '../interface';
-import type { NativeJsFunction } from '../parser/types';
 import type { SourceCodeInfo } from '../tokenizer/token';
 export declare function collHasKey(coll: unknown, key: string | number): boolean;
 export declare function compare<T extends string | number>(a: T, b: T, sourceCodeInfo: SourceCodeInfo | undefined): number;
@@ -7,7 +6,6 @@ export declare function deepEqual(a: unknown, b: unknown, sourceCodeInfo?: Sourc
 export declare function toNonNegativeInteger(num: number): number;
 export declare function toAny(value: unknown): Any;
 export declare function cloneColl<T extends Coll>(value: T): T;
-export declare function createNativeJsFunction(fn: (...args: any[]) => unknown, name?: string): NativeJsFunction;
 export declare function joinSets<T>(...results: Set<T>[]): Set<T>;
 export declare function addToSet<T>(target: Set<T>, source: Set<T>): void;
 export declare const EPSILON = 1e-10;

package/dist/cli/src/utils/paramCount.d.ts

@@ -0,0 +1,7 @@
+import type { ParamCount } from '../builtin/interface';
+import type { FunctionLike, NormalExpressionNodeWithName } from '../parser/types';
+export declare function paramCountAccepts(paramsCount: ParamCount, nbrOfParams: number): boolean;
+export declare function paramCountAcceptsMin(paramsCount: ParamCount, nbrOfParams: number): boolean;
+export declare function getCommonParamCountFromFunctions(params: FunctionLike[]): ParamCount | null;
+export declare function getParamCountFromFunction(param: FunctionLike): ParamCount;
+export declare function assertNumberOfParams(count: ParamCount, node: NormalExpressionNodeWithName): void;