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

package/dist/src/expr/dict.js

@@ -1,6 +1,7 @@
 import { get_location } from "../location.js";
 import { DictType, BooleanType, FunctionType, IntegerType, NullType, OptionType, SetType, NeverType, VariantType, printType, FloatType, isTypeEqual, ArrayType } from "../types.js";
 import { valueOrExprToAst, valueOrExprToAstTyped } from "./ast.js";
+import { equal, notEqual } from "./block.js";
 import { AstSymbol, Expr, FactorySymbol, TypeSymbol } from "./expr.js";
 import { none, some } from "../containers/variant.js";
 /**
@@ -61,7 +62,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: IntegerType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictSize",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol]],
@@ -89,7 +90,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BooleanType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictHas",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -135,7 +136,7 @@ export class DictExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                location: get_location(),
                 builtin: "DictGet",
                 type_parameters: [this.key_type, this.value_type],
                 arguments: [this[AstSymbol], keyAst],
@@ -147,7 +148,7 @@ export class DictExpr extends Expr {
             return this[FactorySymbol]({
                 ast_type: "Builtin",
                 type: this.value_type,
-                location: get_location(2),
+                location: get_location(),
                 builtin: "DictGetOrDefault",
                 type_parameters: [this.key_type, this.value_type],
                 arguments: [this[AstSymbol], keyAst, onMissingAst],
@@ -192,7 +193,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: OptionType(this.value_type),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictTryGet",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -229,7 +230,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictInsert",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst, valueAst],
@@ -265,7 +266,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictInsertOrUpdate",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst, valueAst],
@@ -302,7 +303,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictUpdate",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst, valueAst],
@@ -358,7 +359,7 @@ export class DictExpr extends Expr {
         let initialExpr;
         if (initialFn === undefined) {
             // Default: create function that throws error
-            initialExpr = Expr.function([this.key_type], this.value_type, ($, key) => $.error(Expr.str `Key ${key} not found in dictionary`, get_location(2)));
+            initialExpr = Expr.function([this.key_type], this.value_type, ($, key) => $.error(Expr.str `Key ${key} not found in dictionary`, get_location()));
         }
         else {
             initialExpr = Expr.from(initialFn, FunctionType([this.key_type], this.value_type));
@@ -366,7 +367,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictMerge",
             type_parameters: [this.key_type, this.value_type, value2Type],
             arguments: [this[AstSymbol], keyAst, valueAst, Expr.ast(updateFnExpr), Expr.ast(initialExpr)],
@@ -403,7 +404,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictGetOrInsert",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst, Expr.ast(defaultValueExpr)],
@@ -438,7 +439,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictDelete",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -469,7 +470,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: BooleanType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictTryDelete",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -502,7 +503,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictPop",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst],
@@ -537,7 +538,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this.value_type,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictSwap",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keyAst, valueAst],
@@ -565,7 +566,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictClear",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol]],
@@ -602,7 +603,7 @@ export class DictExpr extends Expr {
         let mergerAst;
         if (mergeFn === undefined) {
             // Default: replace existing value with new value (ignore key and existing, return new)
-            const mergerExpr = Expr.function([this.value_type, this.value_type, this.key_type], this.value_type, ($, _v1, _v2, k) => $.error(Expr.str `Key ${k} exists in both dictionaries`, get_location(2)));
+            const mergerExpr = Expr.function([this.value_type, this.value_type, this.key_type], this.value_type, ($, _v1, _v2, k) => $.error(Expr.str `Key ${k} exists in both dictionaries`, get_location()));
             mergerAst = Expr.ast(mergerExpr);
         }
         else {
@@ -612,7 +613,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictUnionInPlace",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], Expr.ast(dict2Expr), mergerAst],
@@ -659,7 +660,7 @@ export class DictExpr extends Expr {
         let initialExpr;
         if (initialFn === undefined) {
             // Default: create function that throws error
-            initialExpr = Expr.function([this.key_type], this.value_type, ($, key) => $.error(Expr.str `Key ${key} not found in dictionary`, get_location(2)));
+            initialExpr = Expr.function([this.key_type], this.value_type, ($, key) => $.error(Expr.str `Key ${key} not found in dictionary`, get_location()));
         }
         else {
             initialExpr = Expr.from(initialFn, FunctionType([this.key_type], this.value_type));
@@ -667,7 +668,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictMergeAll",
             type_parameters: [this.key_type, this.value_type, value2Type],
             arguments: [this[AstSymbol], dict2Ast, Expr.ast(mergerExpr), Expr.ast(initialExpr)],
@@ -694,7 +695,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: SetType(this.key_type),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictKeys",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol]],
@@ -738,7 +739,7 @@ export class DictExpr extends Expr {
         let onMissingAst;
         if (onMissing === undefined) {
             // Default: throw error with key information if key doesn't exist
-            const defaultFunction = Expr.function([this.key_type], this.value_type, ($, key) => $.error(Expr.str `Key ${key} not found in dictionary`, get_location(2)));
+            const defaultFunction = Expr.function([this.key_type], this.value_type, ($, key) => $.error(Expr.str `Key ${key} not found in dictionary`, get_location()));
             onMissingAst = Expr.ast(defaultFunction);
         }
         else {
@@ -748,7 +749,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictGetKeys",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], keysAst, onMissingAst],
@@ -788,7 +789,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: NullType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictForEach",
             type_parameters: [this.key_type, this.value_type, returnType],
             arguments: [this[AstSymbol], fnAst],
@@ -814,7 +815,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictCopy",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol]],
@@ -833,7 +834,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: DictType(this.key_type, returnType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictMap",
             type_parameters: [this.key_type, this.value_type, returnType],
             arguments: [this[AstSymbol], Expr.ast(fnExpr)],
@@ -863,7 +864,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictFilter",
             type_parameters: [this.key_type, this.value_type],
             arguments: [this[AstSymbol], fnAst],
@@ -889,7 +890,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: this[TypeSymbol],
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictFilterMap",
             type_parameters: [this.key_type, this.value_type, someType],
             arguments: [this[AstSymbol], fnAst],
@@ -912,7 +913,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: ArrayType(returnType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictToArray",
             type_parameters: [this.key_type, this.value_type, returnType],
             arguments: [this[AstSymbol], Expr.ast(fnExpr)],
@@ -939,7 +940,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: SetType(keyType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictToSet",
             type_parameters: [this.key_type, this.value_type, keyType],
             arguments: [this[AstSymbol], keyFnAst],
@@ -952,7 +953,7 @@ export class DictExpr 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);
         }
@@ -962,7 +963,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: DictType(keyType, valueType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictToDict",
             type_parameters: [this.key_type, this.value_type, keyType, valueType],
             arguments: [this[AstSymbol], keyFnAst, valueFnAst, onConflictAst],
@@ -978,7 +979,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: ArrayType(elementType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictFlattenToArray",
             type_parameters: [this.key_type, this.value_type, elementType],
             arguments: [this[AstSymbol], fnAst],
@@ -994,7 +995,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: SetType(elementType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictFlattenToSet",
             type_parameters: [this.key_type, this.value_type, elementType],
             arguments: [this[AstSymbol], fnAst],
@@ -1010,7 +1011,7 @@ export class DictExpr 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);
         }
@@ -1020,7 +1021,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: DictType(keyType, valueType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictFlattenToDict",
             type_parameters: [this.key_type, this.value_type, keyType, valueType],
             arguments: [this[AstSymbol], fnAst, onConflictAst],
@@ -1035,7 +1036,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: DictType(keyType, initType),
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictGroupFold",
             type_parameters: [this.key_type, this.value_type, keyType, initType],
             arguments: [this[AstSymbol], keyFnAst, initFnAst, reduceFnAst],
@@ -1239,7 +1240,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: returnType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictReduce",
             type_parameters: [this.key_type, this.value_type, returnType],
             arguments: [this[AstSymbol], fnAst, initAst],
@@ -1252,7 +1253,7 @@ export class DictExpr extends Expr {
         return this[FactorySymbol]({
             ast_type: "Builtin",
             type: mapType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictMapReduce",
             type_parameters: [this.key_type, this.value_type, mapType],
             arguments: [this[AstSymbol], mapAst, combineAst],
@@ -1271,7 +1272,7 @@ export class DictExpr extends Expr {
         return Expr.fromAst({
             ast_type: "Builtin",
             type: returnType,
-            location: get_location(2),
+            location: get_location(),
             builtin: "DictFirstMap",
             type_parameters: [this.key_type, this.value_type, someType],
             arguments: [this[AstSymbol], fnAst],
@@ -1313,20 +1314,20 @@ export class DictExpr extends Expr {
             const valueParam = {
                 ast_type: "Variable",
                 type: this.value_type,
-                location: get_location(2),
+                location: get_location(),
                 mutable: false,
             };
             const keyParam = {
                 ast_type: "Variable",
                 type: this.key_type,
-                location: get_location(2),
+                location: get_location(),
                 mutable: false,
             };
             // Check if boolean value is NOT true, then return some(null) to stop, otherwise return none to continue
             const notCondition = {
                 ast_type: "Builtin",
                 type: BooleanType,
-                location: get_location(2),
+                location: get_location(),
                 builtin: "BooleanNot",
                 type_parameters: [],
                 arguments: [valueParam]
@@ -1334,23 +1335,23 @@ export class DictExpr extends Expr {
             const checkFnAst = {
                 ast_type: "Function",
                 type: FunctionType([this.value_type, this.key_type], optionType),
-                location: get_location(2),
+                location: get_location(),
                 parameters: [valueParam, keyParam],
                 body: {
                     ast_type: "IfElse",
                     type: optionType,
-                    location: get_location(2),
+                    location: get_location(),
                     ifs: [{
                             predicate: notCondition,
-                            body: { ast_type: "Variant", type: optionType, location: get_location(2), case: "some", value: { ast_type: "Value", type: NullType, location: get_location(2), value: null } }
+                            body: { ast_type: "Variant", type: optionType, location: get_location(), case: "some", value: { ast_type: "Value", type: NullType, location: get_location(), value: null } }
                         }],
-                    else_body: { ast_type: "Variant", type: optionType, location: get_location(2), case: "none", value: { ast_type: "Value", type: NullType, location: get_location(2), value: null } }
+                    else_body: { ast_type: "Variant", type: optionType, location: get_location(), case: "none", value: { ast_type: "Value", type: NullType, location: get_location(), value: null } }
                 }
             };
             const result = Expr.fromAst({
                 ast_type: "Builtin",
                 type: optionType,
-                location: get_location(2),
+                location: get_location(),
                 builtin: "DictFirstMap",
                 type_parameters: [this.key_type, this.value_type, NullType],
                 arguments: [this[AstSymbol], checkFnAst],
@@ -1400,36 +1401,36 @@ export class DictExpr extends Expr {
             const valueParam = {
                 ast_type: "Variable",
                 type: this.value_type,
-                location: get_location(2),
+                location: get_location(),
                 mutable: false,
             };
             const keyParam = {
                 ast_type: "Variable",
                 type: this.key_type,
-                location: get_location(2),
+                location: get_location(),
                 mutable: false,
             };
             // Check if boolean value is true, then return some(null) to stop, otherwise return none to continue
             const checkFnAst = {
                 ast_type: "Function",
                 type: FunctionType([this.value_type, this.key_type], optionType),
-                location: get_location(2),
+                location: get_location(),
                 parameters: [valueParam, keyParam],
                 body: {
                     ast_type: "IfElse",
                     type: optionType,
-                    location: get_location(2),
+                    location: get_location(),
                     ifs: [{
                             predicate: valueParam,
-                            body: { ast_type: "Variant", type: optionType, location: get_location(2), case: "some", value: { ast_type: "Value", type: NullType, location: get_location(2), value: null } }
+                            body: { ast_type: "Variant", type: optionType, location: get_location(), case: "some", value: { ast_type: "Value", type: NullType, location: get_location(), value: null } }
                         }],
-                    else_body: { ast_type: "Variant", type: optionType, location: get_location(2), case: "none", value: { ast_type: "Value", type: NullType, location: get_location(2), value: null } }
+                    else_body: { ast_type: "Variant", type: optionType, location: get_location(), case: "none", value: { ast_type: "Value", type: NullType, location: get_location(), value: null } }
                 }
             };
             const result = Expr.fromAst({
                 ast_type: "Builtin",
                 type: optionType,
-                location: get_location(2),
+                location: get_location(),
                 builtin: "DictFirstMap",
                 type_parameters: [this.key_type, this.value_type, NullType],
                 arguments: [this[AstSymbol], checkFnAst],
@@ -1488,5 +1489,54 @@ export class DictExpr extends Expr {
             }
         }
     }
+    /**
+     * Checks if this dictionary equals another dictionary (same key-value pairs).
+     *
+     * @param other - The dictionary to compare against
+     * @returns A BooleanExpr that is true if the dictionaries are equal
+     *
+     * @example
+     * ```ts
+     * const isEqual = East.function([DictType(StringType, IntegerType), DictType(StringType, IntegerType)], BooleanType, ($, a, b) => {
+     *   $.return(a.equals(b));
+     * });
+     * const compiled = East.compile(isEqual.toIR(), []);
+     * compiled(new Map([["a", 1n], ["b", 2n]]), new Map([["a", 1n], ["b", 2n]]));  // true
+     * compiled(new Map([["a", 1n]]), new Map([["a", 1n], ["b", 2n]]));            // false
+     * ```
+     */
+    equals(other) {
+        return equal(this, other);
+    }
+    /**
+     * Checks if this dictionary does not equal another dictionary.
+     *
+     * @param other - The dictionary to compare against
+     * @returns A BooleanExpr that is true if the dictionaries are not equal
+     *
+     * @example
+     * ```ts
+     * const isNotEqual = East.function([DictType(StringType, IntegerType), DictType(StringType, IntegerType)], BooleanType, ($, a, b) => {
+     *   $.return(a.notEquals(b));
+     * });
+     * const compiled = East.compile(isNotEqual.toIR(), []);
+     * compiled(new Map([["a", 1n]]), new Map([["a", 1n], ["b", 2n]]));            // true
+     * compiled(new Map([["a", 1n], ["b", 2n]]), new Map([["a", 1n], ["b", 2n]]));  // 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=dict.js.map