diff --git a/solidity/ecdsa/tasks/initialize.ts b/solidity/ecdsa/tasks/initialize.ts index d032a1ff54..a7f14f1592 100644 --- a/solidity/ecdsa/tasks/initialize.ts +++ b/solidity/ecdsa/tasks/initialize.ts @@ -4,16 +4,19 @@ import { TASK_AUTHORIZE, TASK_REGISTER, TASK_INITIALIZE_STAKING, + TASK_ADD_BETA_OPERATOR } from "@keep-network/random-beacon/export/tasks/initialize" import { authorize, register, + addBetaOperator, } from "@keep-network/random-beacon/export/tasks/utils" // Tasks for the ECDSA application. const TASK_INITIALIZE_ECDSA = `${TASK_INITIALIZE}:ecdsa` const TASK_AUTHORIZE_ECDSA = `${TASK_AUTHORIZE}:ecdsa` const TASK_REGISTER_ECDSA = `${TASK_REGISTER}:ecdsa` +const TASK_ADD_BETA_OPERATOR_ECDSA = `${TASK_ADD_BETA_OPERATOR}:ecdsa` task( TASK_INITIALIZE, @@ -39,6 +42,7 @@ task(TASK_INITIALIZE_ECDSA, "Initializes operator for ECDSA") .setAction(async (args, hre) => { await hre.run(TASK_AUTHORIZE_ECDSA, args) await hre.run(TASK_REGISTER_ECDSA, args) + await hre.run(TASK_ADD_BETA_OPERATOR_ECDSA, args) }) task(TASK_AUTHORIZE_ECDSA, "Sets authorization for ECDSA") @@ -71,3 +75,11 @@ task( .setAction(async (args, hre) => { await register(hre, "WalletRegistry", args.provider, args.operator) }) + + task( + TASK_ADD_BETA_OPERATOR_ECDSA, + "Adds an operator to the set of beta operators in ECDSA" + ).addParam("operator", "Operator Address", undefined, types.string) + .setAction(async (args, hre) => { + await addBetaOperator(hre, "EcdsaSortitionPool", args.operator) + })