main.star

# Import remote code from another package using an absolute import
ethereum_package = import_module("github.com/ethpandaops/ethereum-package/main.star")
service_utils = import_module("./service_utils.star")
shared_utils = import_module("./shared_utils.star")
contract_deployer = import_module("./contract_deployer.star")
keys = import_module("./keys.star")


def run(plan, args={}):
    ethereum_args = parse_ethereum_package_args(plan, args)
    args = parse_args(plan, args)

    # Run the Ethereum package first
    ethereum_output = ethereum_package.run(plan, ethereum_args)

    el_context = ethereum_output.all_participants[0].el_context
    http_rpc_url = el_context.rpc_http_url
    ws_url = el_context.ws_url

    pre_funded_account = ethereum_output.pre_funded_accounts[0]
    private_key = pre_funded_account.private_key
    deployer_address = pre_funded_account.address

    data = {
        "http_rpc_url": http_rpc_url,
        "ws_rpc_url": ws_url,
        "deployer_private_key": "0x" + private_key,
        "deployer_address": deployer_address,
    }
    plan.print("Initial data: " + json.indent(json.encode(data)))

    # Append fields that will be populated later
    data.update({"services": {}, "keys": {}, "addresses": {}})

    context = struct(
        artifacts=args.artifacts,
        services={},
        ethereum=ethereum_output,
        data=data,
    )

    keys.generate_all_keys(plan, context, args.keys)

    for deployment in args.deployments:
        contract_deployer.deploy(plan, context, deployment)

    for service in args.services:
        service_utils.add_service(plan, service, context)

    shared_utils.ensure_all_generated(plan, context, context.artifacts)

    return context


def parse_args(plan, args):
    artifacts = args.get("artifacts", {})
    keys = args.get("keys", [])
    deployments = args.get("deployments", [])
    services = args.get("services", [])

    # Mark artifacts that have static files as generated
    # TODO: support mixed artifacts
    for artifact_name, artifact in artifacts.items():
        files = artifact.get("files", {})
        if any(["static_file" in file for file in files.values()]):
            artifacts[artifact_name]["generated"] = True

    return struct(
        artifacts=artifacts,
        keys=keys,
        deployments=deployments,
        services=services,
    )


# We pin the images to a specific digest to ensure the package is not broken by upstream changes
DEFAULT_PARTICIPANT = {
    "el_type": "besu",
    # hyperledger/besu:latest from 2024-11-25
    "el_image": "hyperledger/besu@sha256:644f31577d06f0076375fb4a92805e30038b8dee2b25dda4dd3a843f79ccca65",
    "cl_type": "lighthouse",
    # ethpandaops/lighthouse:stable from 2024-11-25
    "cl_image": "ethpandaops/lighthouse@sha256:924fe8fb9505595b5689d86c18ced5ebfc43817d5d96811977e5d37dbab07dba",
}


def parse_ethereum_package_args(plan, args):
    ethereum_args = dict(args.get("ethereum_package", {}))
    participants = ethereum_args.get("participants", [{"el_type": "besu"}])

    # If there are no supported clients in first participant, add one
    if len(participants) == 0 or participants[0].get("el_type") != "besu":
        plan.print("WARNING: no 'besu' client as first participant. Adding one...")
        participants = [DEFAULT_PARTICIPANT] + participants

    ethereum_args["participants"] = participants
    return ethereum_args