@elaraai/east 0.0.1-beta.3 → 0.0.1-beta.30

package/dist/src/expr/array.js

@@ -2,7 +2,10 @@ import { get_location } from "../location.js";
 import { ArrayType, BooleanType, FunctionType, IntegerType, NullType, StringType, isSubtype, isTypeEqual, printType, NeverType, OptionType, isDataType, StructType, VariantType, SetType, FloatType, DictType } from "../types.js";
 import { valueOrExprToAst, valueOrExprToAstTyped } from "./ast.js";
 import { AstSymbol, Expr, FactorySymbol, TypeSymbol } from "./expr.js";
+import { equal, notEqual } from "./block.js";
 import { none, some } from "../containers/variant.js";
+import { CsvSerializeConfigType, csvSerializeOptionsToValue } from "../serialization/csv.js";
+import { BlobType } from "../types.js";
 /**
  * Expression representing mutable array values and operations.
  *
@@ -56,7 +59,32 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            location: get_location(),
+            builtin: "ArraySize",
+            type_parameters: [this.value_type],
+            arguments: [this[AstSymbol]],
+        });
+    }
+    /**
+     * Returns the length of the array (an alias for size).
+     *
+     * @returns An IntegerExpr representing the number of elements
+     *
+     * @example
+     * ```ts
+     * const getLength = East.function([ArrayType(IntegerType)], IntegerType, ($, arr) => {
+     *   $.return(arr.length());
+     * });
+     * const compiled = East.compile(getLength.toIR(), []);
+     * compiled([1n, 2n, 3n]);  // 3n
+     * compiled([]);            // 0n
+     * ```
+     */
+    length() {
+        return this[FactorySymbol]({
+            ast_type: "Builtin",
+            type: IntegerType,
+            location: get_location(),
             builtin: "ArraySize",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -84,7 +112,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BooleanType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayHas",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -123,7 +151,7 @@ export class ArrayExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                location: get_location(),
                 builtin: "ArrayGetOrDefault",
                 type_parameters: [this.value_type],
                 arguments: [this[AstSymbol], keyAst, defaultFnAst],
@@ -133,7 +161,57 @@ export class ArrayExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                location: get_location(),
+                builtin: "ArrayGet",
+                type_parameters: [this.value_type],
+                arguments: [this[AstSymbol], keyAst],
+            });
+        }
+    }
+    /**
+     * Gets the element at the specified index (alias for get).
+     *
+     * @param key - The zero-based index
+     * @param defaultFn - Optional function to provide a default value for out-of-bounds indices
+     * @returns An expression of the array's element type
+     *
+     * @throws East runtime error if index is out of bounds and no defaultFn is provided
+     *
+     * @example
+     * ```ts
+     * const getElement = East.function([ArrayType(IntegerType), IntegerType], IntegerType, ($, arr, index) => {
+     *   $.return(arr.get(index));
+     * });
+     * const compiled = East.compile(getElement.toIR(), []);
+     * compiled([10n, 20n, 30n], 1n);  // 20n
+     * // compiled([10n, 20n, 30n], 5n) would throw error
+     *
+     * // With default value
+     * const getOrDefault = East.function([ArrayType(IntegerType), IntegerType], IntegerType, ($, arr, index) => {
+     *   $.return(arr.get(index, ($, i) => -1n));
+     * });
+     * compiled = East.compile(getOrDefault.toIR(), []);
+     * compiled([10n, 20n, 30n], 5n);  // -1n (out of bounds)
+     * ```
+     */
+    at(key, defaultFn) {
+        const keyAst = valueOrExprToAstTyped(key, IntegerType);
+        if (defaultFn !== undefined) {
+            const defaultFnAst = valueOrExprToAstTyped(defaultFn, FunctionType([IntegerType], this.value_type));
+            return this[FactorySymbol]({
+                ast_type: "Builtin",
+                type: this.value_type,
+                location: get_location(),
+                builtin: "ArrayGetOrDefault",
+                type_parameters: [this.value_type],
+                arguments: [this[AstSymbol], keyAst, defaultFnAst],
+            });
+        }
+        else {
+            return this[FactorySymbol]({
+                ast_type: "Builtin",
+                type: this.value_type,
+                location: get_location(),
                 builtin: "ArrayGet",
                 type_parameters: [this.value_type],
                 arguments: [this[AstSymbol], keyAst],
@@ -161,7 +239,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: OptionType(this.value_type),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayTryGet",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -192,7 +270,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayUpdate",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keyAst, valueAst],
@@ -205,7 +283,7 @@ export class ArrayExpr extends Expr {
      * @see {@link update} for simply replacing the value.
      */
     merge(key, value, updateFn) {
-        const location = get_location(2);
+        const location = get_location();
         const keyAst = valueOrExprToAstTyped(key, IntegerType);
         const value2Type = value[TypeSymbol];
         const updateFnExpr = Expr.from(updateFn, FunctionType([value2Type, this.value_type, IntegerType], this.value_type));
@@ -239,7 +317,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayPushLast",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], valueAst],
@@ -265,7 +343,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayPopLast",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -292,7 +370,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayPushFirst",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], valueAst],
@@ -318,7 +396,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayPopFirst",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -345,7 +423,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayAppend",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], arrayExpr[AstSymbol]],
@@ -373,7 +451,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayPrepend",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], arrayExpr[AstSymbol]],
@@ -421,7 +499,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayMergeAll",
             type_parameters: [this.value_type, value2Type],
             arguments: [this[AstSymbol], array[AstSymbol], mergeFnExpr[AstSymbol]],
@@ -446,7 +524,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayClear",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -502,7 +580,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArraySortInPlace",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], byExpr[AstSymbol]],
@@ -557,7 +635,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArraySort",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], byExpr[AstSymbol]],
@@ -584,7 +662,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayReverseInPlace",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -613,7 +691,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayReverse",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -662,7 +740,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BooleanType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayIsSorted",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], byExpr[AstSymbol]],
@@ -719,7 +797,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFindSortedFirst",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], valueAst, byExpr[AstSymbol]],
@@ -764,7 +842,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFindSortedLast",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], valueAst, byExpr[AstSymbol]],
@@ -808,7 +886,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: StructType({ start: IntegerType, end: IntegerType }),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFindSortedRange",
             type_parameters: [this.value_type, projectedType],
             arguments: [this[AstSymbol], valueAst, byExpr[AstSymbol]],
@@ -840,7 +918,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArraySlice",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], startExpr[AstSymbol], endExpr[AstSymbol]],
@@ -869,7 +947,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayConcat",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], otherExpr[AstSymbol]],
@@ -907,7 +985,7 @@ export class ArrayExpr extends Expr {
         const keysExpr = Expr.from(keys, ArrayType(IntegerType));
         let default_function_ast;
         if (onMissing === undefined) {
-            const location = get_location(2);
+            const location = get_location();
             const default_function = Expr.function([IntegerType], this.value_type, ($, key) => $.error(Expr.str `Cannot get key ${key} from array`, location));
             default_function_ast = Expr.ast(default_function);
         }
@@ -918,7 +996,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayGetKeys",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol], keysExpr[AstSymbol], default_function_ast],
@@ -953,7 +1031,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayForEach",
             type_parameters: [this.value_type, returnType],
             arguments: [this[AstSymbol], fn[AstSymbol]],
@@ -984,7 +1062,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayCopy",
             type_parameters: [this.value_type],
             arguments: [this[AstSymbol]],
@@ -1010,7 +1088,7 @@ export class ArrayExpr extends Expr {
                 return this[FactorySymbol]({
                     ast_type: "Builtin",
                     type: ArrayType(output_type),
-                    location: get_location(2),
+                    location: get_location(),
                     builtin: "ArrayMap",
                     type_parameters: [this.value_type, output_type],
                     arguments: [this[AstSymbol], fn[AstSymbol]],
@@ -1082,7 +1160,7 @@ export class ArrayExpr extends Expr {
             }
             return this[FactorySymbol]({
                 ast_type: "Builtin",
-                location: get_location(2),
+                location: get_location(),
                 type: this[TypeSymbol],
                 builtin: "ArrayFilter",
                 type_parameters: [this.value_type],
@@ -1106,7 +1184,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: ArrayType(someType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFilterMap",
             type_parameters: [this.value_type, someType],
             arguments: [this[AstSymbol], fnAst],
@@ -1125,7 +1203,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: returnType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFirstMap",
             type_parameters: [this.value_type, someType],
             arguments: [this[AstSymbol], fnAst],
@@ -1212,7 +1290,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: returnType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFold",
             type_parameters: [this.value_type, returnType],
             arguments: [this[AstSymbol], initAst, combineAst],
@@ -1225,7 +1303,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: mapType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayMapReduce",
             type_parameters: [this.value_type, mapType],
             arguments: [this[AstSymbol], mapAst, combineAst],
@@ -1515,7 +1593,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: StringType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayStringJoin",
             type_parameters: [],
             arguments: [this[AstSymbol], stringAst],
@@ -1527,7 +1605,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: SetType(keyType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayToSet",
             type_parameters: [this.value_type, keyType],
             arguments: [this[AstSymbol], keyFnAst],
@@ -1540,7 +1618,7 @@ export class ArrayExpr extends Expr {
         const valueType = valueFnAst.type.output;
         let onConflictAst;
         if (onConflictFn === undefined) {
-            const location = get_location(2);
+            const location = get_location();
             const onConflictFunction = Expr.function([valueType, valueType, keyType], valueType, ($, existing, value, key) => $.error(Expr.str `Cannot insert duplicate key ${key} into dict`, location));
             onConflictAst = Expr.ast(onConflictFunction);
         }
@@ -1550,7 +1628,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: DictType(keyType, valueType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayToDict",
             type_parameters: [this.value_type, keyType, valueType],
             arguments: [this[AstSymbol], keyFnAst, valueFnAst, onConflictAst],
@@ -1566,7 +1644,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: ArrayType(elementType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFlattenToArray",
             type_parameters: [this.value_type, elementType],
             arguments: [this[AstSymbol], fnAst],
@@ -1582,7 +1660,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: SetType(elementType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFlattenToSet",
             type_parameters: [this.value_type, elementType],
             arguments: [this[AstSymbol], fnAst],
@@ -1598,7 +1676,7 @@ export class ArrayExpr extends Expr {
         const valueType = returnType.value;
         let onConflictAst;
         if (onConflictFn === undefined) {
-            const location = get_location(2);
+            const location = get_location();
             const onConflictFunction = Expr.function([valueType, valueType, keyType], valueType, ($, existing, value, key) => $.error(Expr.str `Cannot insert duplicate key ${key} into dict`, location));
             onConflictAst = Expr.ast(onConflictFunction);
         }
@@ -1608,7 +1686,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: DictType(keyType, valueType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayFlattenToDict",
             type_parameters: [this.value_type, keyType, valueType],
             arguments: [this[AstSymbol], fnAst, onConflictAst],
@@ -1624,7 +1702,7 @@ export class ArrayExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: DictType(keyType, initType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "ArrayGroupFold",
             type_parameters: [this.value_type, keyType, initType],
             arguments: [this[AstSymbol], keyFnAst, initFnAst, reduceFnAst],
@@ -1801,5 +1879,87 @@ export class ArrayExpr extends Expr {
             }));
         }
     }
+    /**
+     * Encodes this array of structs as CSV data.
+     *
+     * @param options - CSV serialization options
+     * @returns A BlobExpr containing the encoded CSV data
+     *
+     * @remarks Only works on arrays of structs with primitive or optional primitive fields.
+     *
+     * @example
+     * ```ts
+     * const PersonType = StructType({ name: StringType, age: IntegerType });
+     *
+     * const toCsv = East.function([ArrayType(PersonType)], BlobType, ($, people) => {
+     *   $.return(people.encodeCsv({ delimiter: ',' }));
+     * });
+     * const compiled = East.compile(toCsv.toIR(), []);
+     * const blob = compiled([{ name: "Alice", age: 30n }, { name: "Bob", age: 25n }]);
+     * new TextDecoder().decode(blob);  // "name,age\r\nAlice,30\r\nBob,25"
+     * ```
+     */
+    encodeCsv(options) {
+        // Convert options to East config value
+        const configValue = csvSerializeOptionsToValue(options);
+        const configAst = valueOrExprToAstTyped(configValue, CsvSerializeConfigType);
+        return this[FactorySymbol]({
+            ast_type: "Builtin",
+            type: BlobType,
+            location: get_location(),
+            builtin: "ArrayEncodeCsv",
+            type_parameters: [this.value_type, CsvSerializeConfigType],
+            arguments: [this[AstSymbol], configAst],
+        });
+    }
+    /**
+     * Checks if this array equals another array (deep comparison).
+     *
+     * @param other - The array to compare against
+     * @returns A BooleanExpr that is true if the arrays are deeply equal
+     *
+     * @example
+     * ```ts
+     * const isEqual = East.function([ArrayType(IntegerType), ArrayType(IntegerType)], BooleanType, ($, a, b) => {
+     *   $.return(a.equals(b));
+     * });
+     * const compiled = East.compile(isEqual.toIR(), []);
+     * compiled([1n, 2n, 3n], [1n, 2n, 3n]);  // true
+     * compiled([1n, 2n], [1n, 2n, 3n]);      // false
+     * ```
+     */
+    equals(other) {
+        return equal(this, other);
+    }
+    /**
+     * Checks if this array does not equal another array.
+     *
+     * @param other - The array to compare against
+     * @returns A BooleanExpr that is true if the arrays are not equal
+     *
+     * @example
+     * ```ts
+     * const isNotEqual = East.function([ArrayType(IntegerType), ArrayType(IntegerType)], BooleanType, ($, a, b) => {
+     *   $.return(a.notEquals(b));
+     * });
+     * const compiled = East.compile(isNotEqual.toIR(), []);
+     * compiled([1n, 2n], [1n, 2n, 3n]);      // true
+     * compiled([1n, 2n, 3n], [1n, 2n, 3n]);  // false
+     * ```
+     */
+    notEquals(other) {
+        return notEqual(this, other);
+    }
+    // ============================================================================
+    // Aliases for comparison operations
+    // ============================================================================
+    /** Alias for {@link equals} */
+    eq = this.equals;
+    /** Alias for {@link equals} */
+    equal = this.equals;
+    /** Alias for {@link notEquals} */
+    ne = this.notEquals;
+    /** Alias for {@link notEquals} */
+    notEqual = this.notEquals;
 }
 //# sourceMappingURL=array.js.map