utils.py

import dataclasses
import functools
import typing

import pycardano
import uplc.ast as uplc_ast
from pycardano import PlutusData
from uplc import eval as uplc_eval

from opshin import DEFAULT_CONFIG
from opshin.builder import _compile


@dataclasses.dataclass
class Unit(PlutusData):
    CONSTR_ID = 0


def eval_uplc(
    source_code: str,
    *args: typing.Union[pycardano.Datum, uplc_ast.Constant],
    contract_file: str = "<unknown>",
    validator_function_name="validator",
    config=DEFAULT_CONFIG,
):
    code = _compile(
        source_code,
        *args,
        contract_file=contract_file,
        validator_function_name=validator_function_name,
        config=config,
    )
    ret = uplc_eval(code).result
    if isinstance(ret, Exception):
        raise ret
    return ret


def eval_uplc_value(
    source_code: str,
    *args: typing.Union[pycardano.Datum, uplc_ast.Constant],
    contract_file: str = "<unknown>",
    validator_function_name="validator",
    config=DEFAULT_CONFIG,
):
    return eval_uplc(
        source_code,
        *args,
        contract_file=contract_file,
        validator_function_name=validator_function_name,
        config=config,
    ).value