@harmoniclabs/pebble 0.2.0 → 0.3.0

package/dist/compiler/tir/program/stdScope/stdScope.d.ts

@@ -11,8 +11,15 @@ export declare const string_t: TirStringT;
 export declare const bytes_t: TirBytesT;
 export declare const bool_t: TirBoolT;
 export declare const data_t: TirDataT;
+export declare const valueMapLovelacesName: string;
+export declare const valueMapAmountOfName: string;
 export declare const valueLovelacesName: string;
 export declare const valueAmountOfName: string;
+export declare const valueInsertCoinName: string;
+export declare const valueUnionName: string;
+export declare const valueContainsName: string;
+export declare const valueScaleName: string;
+export declare const valueToDataName: string;
 export declare const getCredentialHashFuncName: string;
 export declare function populateStdScope(program: TypedProgram): void;
 export declare function populatePreludeScope(program: TypedProgram): void;

package/dist/compiler/tir/program/stdScope/stdScope.js

@@ -6,6 +6,8 @@ import { TirDataT } from "../../types/TirNativeType/native/data.js";
 import { TirIntT } from "../../types/TirNativeType/native/int.js";
 import { TirLinearMapT } from "../../types/TirNativeType/native/linearMap.js";
 import { TirListT } from "../../types/TirNativeType/native/list.js";
+import { TirArrayT } from "../../types/TirNativeType/native/array.js";
+import { TirValueT } from "../../types/TirNativeType/native/value.js";
 import { TirDataOptT } from "../../types/TirNativeType/native/Optional/data.js";
 import { TirSopOptT } from "../../types/TirNativeType/native/Optional/sop.js";
 import { TirStringT } from "../../types/TirNativeType/native/string.js";
@@ -18,14 +20,23 @@ import { IRNative } from "../../../../IR/IRNodes/IRNative/index.js";
 import { IRFunc } from "../../../../IR/IRNodes/IRFunc.js";
 import { IRVar } from "../../../../IR/IRNodes/IRVar.js";
 import { _ir_apps } from "../../../../IR/IRNodes/IRApp.js";
+import { IRConst } from "../../../../IR/IRNodes/IRConst.js";
 export const void_t = new TirVoidT();
 export const int_t = new TirIntT();
 export const string_t = new TirStringT();
 export const bytes_t = new TirBytesT();
 export const bool_t = new TirBoolT();
 export const data_t = new TirDataT();
-export const valueLovelacesName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "sortedValueLovelaces";
-export const valueAmountOfName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "amountOfValue";
+export const valueMapLovelacesName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "sortedValueLovelaces";
+export const valueMapAmountOfName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "amountOfValue";
+// Native Value (ConstTyTag.value); methods route to the value builtins.
+export const valueLovelacesName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueLovelaces";
+export const valueAmountOfName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueAmountOf";
+export const valueInsertCoinName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueInsertCoin";
+export const valueUnionName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueUnion";
+export const valueContainsName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueContains";
+export const valueScaleName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueScale";
+export const valueToDataName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "valueToData";
 export const getCredentialHashFuncName = PEBBLE_INTERNAL_IDENTIFIER_PREFIX + "getCredentialHash";
 export function populateStdScope(program) {
     const stdScope = program.stdScope;
@@ -40,6 +51,24 @@ export function populateStdScope(program) {
     _defineStdUnambigous(bytes_t);
     _defineStdUnambigous(string_t);
     _defineStdUnambigous(data_t);
+    // Register native Value (ConstTyTag.value) in program.types so the prelude
+    // alias can wrap it. The user-facing `Value` name is defined in the
+    // prelude scope (with methods routed to the value builtins).
+    {
+        const value_native_t = new TirValueT();
+        program.types.set(value_native_t.toTirTypeKey(), value_native_t);
+    }
+    const array_name = TirArrayT.toTirTypeKey();
+    program.defineGenericType(array_name, 1, ([arg]) => new TirArrayT(arg));
+    stdScope.defineType(array_name, {
+        sopTirName: array_name,
+        dataTirName: array_name,
+        allTirNames: new Set([
+            array_name
+        ]),
+        methodsNames: new Map(),
+        isGeneric: true
+    });
     const opt_data_name = TirDataOptT.toTirTypeKey();
     program.defineGenericType(opt_data_name, 1, ([arg]) => new TirDataOptT(arg));
     const opt_sop_name = TirSopOptT.toTirTypeKey();
@@ -542,29 +571,29 @@ export function populatePreludeScope(program) {
         payment: credential_t,
         stake: opt_stakeCredential_t
     }, onlyData);
-    // type Value = LinearMap<PolicyId, LinearMap<TokenName, int>>
+    // type ValueMap = LinearMap<PolicyId, LinearMap<TokenName, int>>
     const map_tokenName_int_t = program.getAppliedGeneric(TirLinearMapT.toTirTypeKey(), [tokenName_t, int_t]);
     if (!map_tokenName_int_t)
         throw new Error("expected map_tokenName_int_t");
     const map_policyId_map_tokenName_int_t = program.getAppliedGeneric(TirLinearMapT.toTirTypeKey(), [policyId_t, map_tokenName_int_t]);
     if (!map_policyId_map_tokenName_int_t)
         throw new Error("expected map_policyId_map_tokenName_int_t");
-    const value_t = _defineUnambigousAlias("Value", map_policyId_map_tokenName_int_t, new Map([
+    const valueMap_t = _defineUnambigousAlias("ValueMap", map_policyId_map_tokenName_int_t, new Map([
         [
             "lovelaces",
-            valueLovelacesName
+            valueMapLovelacesName
         ],
         [
             "amountOf",
-            valueAmountOfName
+            valueMapAmountOfName
         ],
     ]));
-    preludeScope.program.functions.set(valueLovelacesName, new TirInlineClosedIR(new TirFuncT([value_t], int_t), (ctx) => IRNative._sortedValueLovelaces, SourceRange.unknown));
-    // Value.amountOf( policy: PolicyId, name: bytes ): int
+    preludeScope.program.functions.set(valueMapLovelacesName, new TirInlineClosedIR(new TirFuncT([valueMap_t], int_t), (ctx) => IRNative._sortedValueLovelaces, SourceRange.unknown));
+    // ValueMap.amountOf( policy: PolicyId, name: bytes ): int
     // The IR native _amountOfValue is curried as: (isPolicy)(value)(isTokenName) => int
     // where isPolicy and isTokenName are equality-check predicates.
     // This wrapper adapts (self, policy, tokenName) => _amountOfValue(p => equalsByteString(p, policy))(self)(tn => equalsByteString(tn, tokenName))
-    preludeScope.program.functions.set(valueAmountOfName, new TirInlineClosedIR(new TirFuncT([value_t, policyId_t, bytes_t], int_t), (ctx) => {
+    preludeScope.program.functions.set(valueMapAmountOfName, new TirInlineClosedIR(new TirFuncT([valueMap_t, policyId_t, bytes_t], int_t), (ctx) => {
         const self = Symbol("amtOf_self");
         const policy = Symbol("amtOf_policy");
         const tokenName = Symbol("amtOf_tokenName");
@@ -578,37 +607,51 @@ export function populatePreludeScope(program) {
         // isTokenName predicate: \tn -> equalsByteString(tn, tokenName)
         _ir_apps(IRNative.equalsByteString, new IRVar(tokenName))));
     }, SourceRange.unknown));
-    /* // TODO
-    untagged struct FlatValueEntry {
-        policy: PolicyId,
-        name: bytes,
-        amount: int
-    }
-
-    type FlatValue = List<FlatValueEntry>;
-
-    type Value implements {
-        amountOf( policy: PolicyId, name: bytes ): int
-        {
-            return native.valueAmountOf( this, policy, name );
-        }
-        lovelaces(): int
-        {
-            return this.amountOf( #, # );
-        }
-        flatten(): FlatValue
-        {
-            return native.flattenValue( this );
-        }
-    }
-
-    type FlatValue implements {
-        unflatten(): Value
-        {
-            return native.unflattenValue( this );
-        }
-    }
-    */
+    // Native `Value` (ConstTyTag.value). Methods route to the value builtins.
+    const value_native_t = program.types.get(TirValueT.toTirTypeKey());
+    if (!value_native_t)
+        throw new Error("expected native Value type registered");
+    const value_t = _defineUnambigousAlias("Value", value_native_t, new Map([
+        ["lovelaces", valueLovelacesName],
+        ["amountOf", valueAmountOfName],
+        ["insert", valueInsertCoinName],
+        ["union", valueUnionName],
+        ["contains", valueContainsName],
+        ["scale", valueScaleName],
+        ["toData", valueToDataName],
+    ]));
+    // Value.lovelaces(): int  -- lookupCoin "" "" self
+    preludeScope.program.functions.set(valueLovelacesName, new TirInlineClosedIR(new TirFuncT([value_t], int_t), (ctx) => {
+        const self = Symbol("value_lovelaces_self");
+        return new IRFunc([self], _ir_apps(IRNative.lookupCoin, IRConst.bytes(new Uint8Array(0)), IRConst.bytes(new Uint8Array(0)), new IRVar(self)));
+    }, SourceRange.unknown));
+    // Value.amountOf( policy, name ): int  -- lookupCoin policy name self
+    preludeScope.program.functions.set(valueAmountOfName, new TirInlineClosedIR(new TirFuncT([value_t, policyId_t, bytes_t], int_t), (ctx) => {
+        const self = Symbol("value_amountOf_self");
+        const policy = Symbol("value_amountOf_policy");
+        const tokenName = Symbol("value_amountOf_tokenName");
+        return new IRFunc([self, policy, tokenName], _ir_apps(IRNative.lookupCoin, new IRVar(policy), new IRVar(tokenName), new IRVar(self)));
+    }, SourceRange.unknown));
+    // Value.insert( policy, name, amount ): Value -- insertCoin policy name amount self
+    preludeScope.program.functions.set(valueInsertCoinName, new TirInlineClosedIR(new TirFuncT([value_t, policyId_t, bytes_t, int_t], value_t), (ctx) => {
+        const self = Symbol("value_insert_self");
+        const policy = Symbol("value_insert_policy");
+        const tokenName = Symbol("value_insert_tokenName");
+        const amount = Symbol("value_insert_amount");
+        return new IRFunc([self, policy, tokenName, amount], _ir_apps(IRNative.insertCoin, new IRVar(policy), new IRVar(tokenName), new IRVar(amount), new IRVar(self)));
+    }, SourceRange.unknown));
+    // Value.union( other ): Value -- unionValue self other
+    preludeScope.program.functions.set(valueUnionName, new TirInlineClosedIR(new TirFuncT([value_t, value_t], value_t), (ctx) => IRNative.unionValue, SourceRange.unknown));
+    // Value.contains( other ): bool -- valueContains self other
+    preludeScope.program.functions.set(valueContainsName, new TirInlineClosedIR(new TirFuncT([value_t, value_t], bool_t), (ctx) => IRNative.valueContains, SourceRange.unknown));
+    // Value.scale( factor ): Value -- scaleValue factor self
+    preludeScope.program.functions.set(valueScaleName, new TirInlineClosedIR(new TirFuncT([value_t, int_t], value_t), (ctx) => {
+        const self = Symbol("value_scale_self");
+        const factor = Symbol("value_scale_factor");
+        return new IRFunc([self, factor], _ir_apps(IRNative.scaleValue, new IRVar(factor), new IRVar(self)));
+    }, SourceRange.unknown));
+    // Value.toData(): data -- valueData self
+    preludeScope.program.functions.set(valueToDataName, new TirInlineClosedIR(new TirFuncT([value_t], data_t), (ctx) => IRNative.valueData, SourceRange.unknown));
     // struct OutputDatum {
     //     NoDatum {}
     //     DatumHash { hash: Hash32 }

package/dist/compiler/tir/types/TirEnumType.d.ts

@@ -0,0 +1,21 @@
+import { ConstType } from "@harmoniclabs/uplc";
+import { AstFuncName, TirFuncName } from "../../AstCompiler/scope/AstScope.js";
+import { ITirType, TirType } from "./TirType.js";
+export declare class TirEnumType implements ITirType {
+    readonly name: string;
+    readonly fileUid: string;
+    readonly members: readonly string[];
+    readonly methodNamesPtr: Map<AstFuncName, TirFuncName>;
+    constructor(name: string, fileUid: string, members: readonly string[], methodNamesPtr: Map<AstFuncName, TirFuncName>);
+    indexOf(member: string): number;
+    isConcrete(): boolean;
+    hasDataEncoding(): boolean;
+    toString(): string;
+    toAstName(): string;
+    toTirTypeKey(): string;
+    toConcreteTirTypeName(): string;
+    toUplcConstType(): ConstType;
+    clone(): TirEnumType;
+}
+export declare function isTirEnumType(thing: any): thing is TirEnumType;
+export declare function getEnumType(type: TirType | undefined): TirEnumType | undefined;

package/dist/compiler/tir/types/TirEnumType.js

@@ -0,0 +1,36 @@
+import { isObject } from "@harmoniclabs/obj-utils";
+import { constT } from "@harmoniclabs/uplc";
+import { TirAliasType } from "./TirAliasType.js";
+export class TirEnumType {
+    name;
+    fileUid;
+    members;
+    methodNamesPtr;
+    constructor(name, fileUid, members, methodNamesPtr) {
+        this.name = name;
+        this.fileUid = fileUid;
+        this.members = members;
+        this.methodNamesPtr = methodNamesPtr;
+    }
+    indexOf(member) {
+        return this.members.indexOf(member);
+    }
+    isConcrete() { return true; }
+    hasDataEncoding() { return true; }
+    toString() { return this.name; }
+    toAstName() { return this.name; }
+    toTirTypeKey() { return "enum_" + this.name + "_" + this.fileUid; }
+    toConcreteTirTypeName() { return this.toTirTypeKey(); }
+    toUplcConstType() { return constT.int; }
+    clone() {
+        return new TirEnumType(this.name, this.fileUid, this.members.slice(), this.methodNamesPtr);
+    }
+}
+export function isTirEnumType(thing) {
+    return isObject(thing) && thing instanceof TirEnumType;
+}
+export function getEnumType(type) {
+    while (type instanceof TirAliasType)
+        type = type.aliased;
+    return type instanceof TirEnumType ? type : undefined;
+}

package/dist/compiler/tir/types/TirNativeType/TirNativeType.d.ts

@@ -12,9 +12,11 @@ import { TirDataOptT } from "./native/Optional/data.js";
 import { TirSopOptT } from "./native/Optional/sop.js";
 import { TirStringT } from "./native/string.js";
 import { TirVoidT } from "./native/void.js";
-export type TirNamedDestructableNativeType = TirDataT | TirDataOptT<TirType> | TirSopOptT<TirType> | TirListT<TirType> | TirLinearMapT<TirType, TirType>;
+import { TirArrayT } from "./native/array.js";
+import { TirValueT } from "./native/value.js";
+export type TirNamedDestructableNativeType = TirDataT | TirDataOptT<TirType> | TirSopOptT<TirType> | TirListT<TirType> | TirArrayT<TirType> | TirLinearMapT<TirType, TirType>;
 export declare function isTirNamedDestructableNativeType(t: any): t is TirNamedDestructableNativeType;
-export type TirNativeType = TirVoidT | TirBoolT | TirIntT | TirBytesT | TirStringT | TirDataT | TirDataOptT<TirType> | TirSopOptT<TirType> | TirListT<TirType> | TirLinearMapT<TirType, TirType> | TirLinearMapEntryT<TirType, TirType> | TirFuncT | TirUnConstrDataResultT | TirPairDataT | TirBlsG1T | TirBlsG2T | TirMlResultT;
+export type TirNativeType = TirVoidT | TirBoolT | TirIntT | TirBytesT | TirStringT | TirDataT | TirDataOptT<TirType> | TirSopOptT<TirType> | TirListT<TirType> | TirArrayT<TirType> | TirValueT | TirLinearMapT<TirType, TirType> | TirLinearMapEntryT<TirType, TirType> | TirFuncT | TirUnConstrDataResultT | TirPairDataT | TirBlsG1T | TirBlsG2T | TirMlResultT;
 export declare function isTirNativeType(t: any): t is TirNativeType;
 export declare class TirUnConstrDataResultT implements ITirType {
     constructor();

package/dist/compiler/tir/types/TirNativeType/TirNativeType.js

@@ -11,11 +11,14 @@ import { TirDataOptT } from "./native/Optional/data.js";
 import { TirSopOptT } from "./native/Optional/sop.js";
 import { TirStringT } from "./native/string.js";
 import { TirVoidT } from "./native/void.js";
+import { TirArrayT } from "./native/array.js";
+import { TirValueT } from "./native/value.js";
 export function isTirNamedDestructableNativeType(t) {
     return (t instanceof TirDataT
         || t instanceof TirDataOptT
         || t instanceof TirSopOptT
         || t instanceof TirListT
+        || t instanceof TirArrayT
         || t instanceof TirLinearMapT);
 }
 export function isTirNativeType(t) {
@@ -28,6 +31,8 @@ export function isTirNativeType(t) {
         || t instanceof TirDataOptT
         || t instanceof TirSopOptT
         || t instanceof TirListT
+        || t instanceof TirArrayT
+        || t instanceof TirValueT
         || t instanceof TirLinearMapT
         || t instanceof TirLinearMapEntryT
         || t instanceof TirFuncT // =>

package/dist/compiler/tir/types/TirNativeType/native/array.d.ts

@@ -0,0 +1,16 @@
+import { ConstType } from "@harmoniclabs/uplc";
+import { TirType, ITirType } from "../../TirType.js";
+export declare class TirArrayT<T extends TirType = TirType> implements ITirType {
+    readonly typeArg: T;
+    constructor(typeArg: T);
+    hasDataEncoding(): boolean;
+    static toTirTypeKey(): string;
+    toTirTypeKey(): string;
+    toConcreteTirTypeName(): string;
+    toString(): string;
+    toAstName(): string;
+    private _isConcrete;
+    isConcrete(): boolean;
+    clone(): TirArrayT<T>;
+    toUplcConstType(): ConstType;
+}

package/dist/compiler/tir/types/TirNativeType/native/array.js

@@ -0,0 +1,38 @@
+import { constT } from "@harmoniclabs/uplc";
+import { getAppliedTirTypeName } from "../../../program/TypedProgram.js";
+export class TirArrayT {
+    typeArg;
+    constructor(typeArg) {
+        this.typeArg = typeArg;
+    }
+    hasDataEncoding() { return this.typeArg.hasDataEncoding(); }
+    static toTirTypeKey() {
+        return "Array";
+    }
+    toTirTypeKey() {
+        return TirArrayT.toTirTypeKey();
+    }
+    toConcreteTirTypeName() {
+        return getAppliedTirTypeName(this.toTirTypeKey(), [this.typeArg.toConcreteTirTypeName()]);
+    }
+    toString() {
+        return `${this.toTirTypeKey()}<${this.typeArg.toString()}>`;
+    }
+    toAstName() {
+        return this.toTirTypeKey();
+    }
+    _isConcrete = undefined;
+    isConcrete() {
+        if (typeof this._isConcrete !== "boolean")
+            this._isConcrete = this.typeArg.isConcrete();
+        return this._isConcrete ?? false;
+    }
+    clone() {
+        const result = new TirArrayT(this.typeArg.clone());
+        result._isConcrete = this._isConcrete;
+        return result;
+    }
+    toUplcConstType() {
+        return constT.arrayOf(this.typeArg.toUplcConstType());
+    }
+}

package/dist/compiler/tir/types/TirNativeType/native/index.d.ts

@@ -1,4 +1,5 @@
 export * from "./Optional/index.js";
+export * from "./array.js";
 export * from "./bool.js";
 export * from "./bytes.js";
 export * from "./data.js";
@@ -8,4 +9,5 @@ export * from "./linearMap.js";
 export * from "./linearMapEntry.js";
 export * from "./list.js";
 export * from "./string.js";
+export * from "./value.js";
 export * from "./void.js";

package/dist/compiler/tir/types/TirNativeType/native/index.js

@@ -1,4 +1,5 @@
 export * from "./Optional/index.js";
+export * from "./array.js";
 export * from "./bool.js";
 export * from "./bytes.js";
 export * from "./data.js";
@@ -8,4 +9,5 @@ export * from "./linearMap.js";
 export * from "./linearMapEntry.js";
 export * from "./list.js";
 export * from "./string.js";
+export * from "./value.js";
 export * from "./void.js";

package/dist/compiler/tir/types/TirNativeType/native/value.d.ts

@@ -0,0 +1,18 @@
+import { ConstType } from "@harmoniclabs/uplc";
+import { ITirType } from "../../TirType.js";
+export declare class TirValueT implements ITirType {
+    clone(): TirValueT;
+    isConcrete(): boolean;
+    toString(): string;
+    toAstName(): string;
+    static toTirTypeKey(): string;
+    toTirTypeKey(): string;
+    toConcreteTirTypeName(): string;
+    /**
+     * Native `Value` is bidirectionally reachable from `data` via
+     * `unValueData` / `valueData`, so it qualifies as data-encoded for
+     * the purposes of `data struct` field types.
+     */
+    hasDataEncoding(): boolean;
+    toUplcConstType(): ConstType;
+}

package/dist/compiler/tir/types/TirNativeType/native/value.js

@@ -0,0 +1,17 @@
+import { constT } from "@harmoniclabs/uplc";
+export class TirValueT {
+    clone() { return new TirValueT(); }
+    isConcrete() { return true; }
+    toString() { return "Value"; }
+    toAstName() { return "Value"; }
+    static toTirTypeKey() { return "Value"; }
+    toTirTypeKey() { return TirValueT.toTirTypeKey(); }
+    toConcreteTirTypeName() { return this.toTirTypeKey(); }
+    /**
+     * Native `Value` is bidirectionally reachable from `data` via
+     * `unValueData` / `valueData`, so it qualifies as data-encoded for
+     * the purposes of `data struct` field types.
+     */
+    hasDataEncoding() { return true; }
+    toUplcConstType() { return constT.value; }
+}

package/dist/compiler/tir/types/TirStructType.js

@@ -27,7 +27,12 @@ export class TirDataStructType {
         this.constructors = constructors;
         this.methodNamesPtr = methodNamesPtr;
         this.untagged = untagged;
-        this.untagged = false; // always false for now
+        // `untagged === true` requires a single constructor — its runtime
+        // form is `listData(fields)` instead of `constrData(idx, fields)`.
+        if (untagged && constructors.length !== 1) {
+            throw new Error("untagged data struct must have exactly one constructor; got "
+                + constructors.length);
+        }
         this.narrowedFromParentCtorIdxs = narrowedFromParentCtorIdxs;
     }
     hasDataEncoding() { return true; }

package/dist/compiler/tir/types/TirType.d.ts

@@ -1,5 +1,6 @@
 import { TirAliasType } from "./TirAliasType.js";
 import { TirStructType } from "./TirStructType.js";
+import { TirEnumType } from "./TirEnumType.js";
 import { TirNamedDestructableNativeType, TirNativeType } from "./TirNativeType/TirNativeType.js";
 import { TirTypeParam } from "./TirTypeParam.js";
 import { ConstType } from "@harmoniclabs/uplc";
@@ -28,7 +29,7 @@ export interface ITirType {
 }
 export type TirType = TirNativeType | TirCustomType | TirTypeParam;
 export declare function isTirType(thing: any): thing is TirType;
-export type TirCustomType = TirAliasType | TirStructType;
+export type TirCustomType = TirAliasType | TirStructType | TirEnumType;
 export declare function isTirCustomType(thing: any): thing is TirCustomType;
-export type TirNamedDestructableType = TirNamedDestructableNativeType | TirStructType;
+export type TirNamedDestructableType = TirNamedDestructableNativeType | TirStructType | TirEnumType;
 export declare function isTirNamedDestructableType(thing: any): thing is TirNamedDestructableType;

package/dist/compiler/tir/types/TirType.js

@@ -1,6 +1,7 @@
 import { isObject } from "@harmoniclabs/obj-utils";
 import { TirAliasType } from "./TirAliasType.js";
 import { isTirStructType } from "./TirStructType.js";
+import { isTirEnumType } from "./TirEnumType.js";
 import { isTirNamedDestructableNativeType, isTirNativeType } from "./TirNativeType/TirNativeType.js";
 import { TirTypeParam } from "./TirTypeParam.js";
 export function isTirType(thing) {
@@ -10,9 +11,11 @@ export function isTirType(thing) {
 }
 export function isTirCustomType(thing) {
     return isObject(thing) && (thing instanceof TirAliasType
-        || isTirStructType(thing));
+        || isTirStructType(thing)
+        || isTirEnumType(thing));
 }
 export function isTirNamedDestructableType(thing) {
     return isObject(thing) && (isTirStructType(thing)
+        || isTirEnumType(thing)
         || isTirNamedDestructableNativeType(thing));
 }

package/dist/compiler/tir/types/utils/canAssignTo.js

@@ -5,6 +5,8 @@ import { TirBytesT } from "../TirNativeType/native/bytes.js";
 import { TirDataT } from "../TirNativeType/native/data.js";
 import { TirFuncT } from "../TirNativeType/native/function.js";
 import { TirIntT } from "../TirNativeType/native/int.js";
+import { TirArrayT } from "../TirNativeType/native/array.js";
+import { TirValueT } from "../TirNativeType/native/value.js";
 import { TirLinearMapT } from "../TirNativeType/native/linearMap.js";
 import { TirLinearMapEntryT } from "../TirNativeType/native/linearMapEntry.js";
 import { TirListT } from "../TirNativeType/native/list.js";
@@ -13,6 +15,7 @@ import { TirSopOptT } from "../TirNativeType/native/Optional/sop.js";
 import { TirStringT } from "../TirNativeType/native/string.js";
 import { TirVoidT } from "../TirNativeType/native/void.js";
 import { isTirStructType, TirDataStructType, TirSoPStructType } from "../TirStructType.js";
+import { TirEnumType } from "../TirEnumType.js";
 import { isTirNamedDestructableType } from "../TirType.js";
 import { TirTypeParam } from "../TirTypeParam.js";
 import { canCastToData } from "./canCastTo.js";
@@ -108,6 +111,10 @@ function uncheckedGetCanAssign(a, b, symbols) {
     if (b instanceof TirIntT) {
         if (a instanceof TirIntT)
             return CanAssign.Yes;
+        // enum runtime representation is a plain int, so an enum value can
+        // flow directly into any int operation.
+        if (a instanceof TirEnumType)
+            return CanAssign.Yes;
         if (a instanceof TirDataT)
             return CanAssign.RequiresExplicitCast;
         return CanAssign.No;
@@ -218,6 +225,19 @@ function uncheckedGetCanAssign(a, b, symbols) {
     }
     if (isTirStructType(a))
         return CanAssign.No;
+    if (b instanceof TirEnumType) {
+        if (a instanceof TirEnumType
+            && a.toTirTypeKey() === b.toTirTypeKey())
+            return CanAssign.Yes;
+        // `data` (e.g. datum/redeemer) can be decoded to an enum with an
+        // explicit cast; plain `int` cannot — there is no way to statically
+        // verify it is a valid tag.
+        if (a instanceof TirDataT)
+            return CanAssign.RequiresExplicitCast;
+        return CanAssign.No;
+    }
+    if (a instanceof TirEnumType)
+        return CanAssign.No;
     if (b instanceof TirFuncT) {
         if (!(a instanceof TirFuncT
             && a.argTypes.length === b.argTypes.length))
@@ -239,6 +259,14 @@ function uncheckedGetCanAssign(a, b, symbols) {
         return a instanceof TirBlsG2T ? CanAssign.Yes : CanAssign.No;
     if (b instanceof TirMlResultT)
         return a instanceof TirMlResultT ? CanAssign.Yes : CanAssign.No;
+    // Native Value / Array<T>
+    if (b instanceof TirValueT)
+        return a instanceof TirValueT ? CanAssign.Yes : CanAssign.No;
+    if (b instanceof TirArrayT) {
+        if (!(a instanceof TirArrayT))
+            return CanAssign.No;
+        return uncheckedGetCanAssign(a.typeArg, b.typeArg, symbols);
+    }
     const tsEnsureExhautstiveCheck = b;
     return CanAssign.No;
 }

package/dist/compiler/tir/types/utils/canCastTo.js

@@ -12,6 +12,7 @@ import { TirSopOptT } from "../TirNativeType/native/Optional/sop.js";
 import { TirStringT } from "../TirNativeType/native/string.js";
 import { TirVoidT } from "../TirNativeType/native/void.js";
 import { isTirStructType, TirDataStructType } from "../TirStructType.js";
+import { TirEnumType } from "../TirEnumType.js";
 import { CanAssign, getCanAssign } from "./canAssignTo.js";
 export function canCastTo(a, b) {
     if (!(a.isConcrete()
@@ -52,6 +53,8 @@ export function canCastTo(a, b) {
     if (b instanceof TirIntT) {
         if (a instanceof TirIntT)
             return true;
+        if (a instanceof TirEnumType)
+            return true; // enum is an int at runtime
         if (a instanceof TirDataT)
             return true;
         if (a instanceof TirBoolT)
@@ -113,6 +116,15 @@ export function canCastTo(a, b) {
         }
         return false;
     }
+    if (b instanceof TirEnumType) {
+        if (a instanceof TirEnumType)
+            return a.toTirTypeKey() === b.toTirTypeKey();
+        // only `data` (e.g. datum/redeemer decoding) can be cast to enum;
+        // plain `int` cannot — there is no way to verify it is a valid tag.
+        if (a instanceof TirDataT)
+            return true;
+        return false;
+    }
     if (b instanceof TirFuncT) {
         if (!(a instanceof TirFuncT))
             return false;
@@ -148,7 +160,8 @@ export function canCastToData(a) {
         || a instanceof TirStringT
         || a instanceof TirBoolT
         || a instanceof TirDataOptT
-        || a instanceof TirDataStructType)
+        || a instanceof TirDataStructType
+        || a instanceof TirEnumType)
         return true;
     if (a instanceof TirLinearMapT) {
         const key = canCastToData(a.keyTypeArg);

package/dist/compiler/tir/types/utils/getDeconstructableType.d.ts

@@ -1,3 +1,4 @@
+import { TirEnumType } from "../TirEnumType.js";
 import { TirDataT } from "../TirNativeType/native/data.js";
 import { TirLinearMapT } from "../TirNativeType/native/linearMap.js";
 import { TirListT } from "../TirNativeType/native/list.js";
@@ -5,5 +6,5 @@ import { TirDataOptT } from "../TirNativeType/native/Optional/data.js";
 import { TirSopOptT } from "../TirNativeType/native/Optional/sop.js";
 import { TirStructType } from "../TirStructType.js";
 import { TirType } from "../TirType.js";
-export type DeconstructableTirType = TirStructType | TirSopOptT | TirDataOptT | TirListT | TirLinearMapT | TirDataT;
+export type DeconstructableTirType = TirStructType | TirEnumType | TirSopOptT | TirDataOptT | TirListT | TirLinearMapT | TirDataT;
 export declare function getDeconstructableType(type: TirType): DeconstructableTirType | undefined;

package/dist/compiler/tir/types/utils/getDeconstructableType.js

@@ -1,4 +1,5 @@
 import { TirAliasType } from "../TirAliasType.js";
+import { TirEnumType } from "../TirEnumType.js";
 import { TirDataT } from "../TirNativeType/native/data.js";
 import { TirLinearMapT } from "../TirNativeType/native/linearMap.js";
 import { TirListT } from "../TirNativeType/native/list.js";
@@ -9,6 +10,7 @@ export function getDeconstructableType(type) {
     while (type instanceof TirAliasType)
         type = type.aliased;
     if (isTirStructType(type)
+        || type instanceof TirEnumType
         || type instanceof TirSopOptT
         || type instanceof TirDataOptT
         || type instanceof TirListT

package/dist/compiler/tir/types/utils/inferTypeArgs.js

@@ -1,3 +1,4 @@
+import { TirArrayT } from "../TirNativeType/native/array.js";
 import { TirFuncT } from "../TirNativeType/native/function.js";
 import { TirLinearMapT } from "../TirNativeType/native/linearMap.js";
 import { TirLinearMapEntryT } from "../TirNativeType/native/linearMapEntry.js";
@@ -34,6 +35,9 @@ export function inferTypeArgs(formal, actual, env) {
     if (formal instanceof TirListT && actual instanceof TirListT) {
         return inferTypeArgs(formal.typeArg, actual.typeArg, env);
     }
+    if (formal instanceof TirArrayT && actual instanceof TirArrayT) {
+        return inferTypeArgs(formal.typeArg, actual.typeArg, env);
+    }
     if (formal instanceof TirFuncT && actual instanceof TirFuncT) {
         if (formal.argTypes.length !== actual.argTypes.length)
             return false;

package/dist/compiler/tir/types/utils/normalizeEnumToInt.d.ts

@@ -0,0 +1,10 @@
+import { TirIntT } from "../TirNativeType/native/int.js";
+import { TirType } from "../TirType.js";
+/**
+ * If `t` is an enum type (or alias of one), returns the shared `int` type
+ * (since enums lower to plain ints at runtime). Otherwise returns `t`.
+ *
+ * Use this to relax type checks for binary operators that treat enums as
+ * integers (arithmetic, comparison, equality).
+ */
+export declare function normalizeEnumToInt(t: TirType, int_t: TirIntT): TirType;