Hardhat task to add an operator to beta operators set for the random beacon

After adding chaosnet feature to the sortition pool, we need to add an operator
to the beta operators set during the operator initialization.

It is done here: hardhat task doing it is provided and it is attached to the
initialize task.

Author: pdyraga

solidity/random-beacon/tasks/initialize.ts

@@ -6,6 +6,7 @@ import {
   stake,
   authorize,
   register,
+  addBetaOperator,
 } from "./utils"
 
 // Main task executing all child tasks.
@@ -19,10 +20,12 @@ export const TASK_STAKE = "stake"
 // Name prefix that should be used in tasks implementation for specific application.
 export const TASK_AUTHORIZE = "authorize"
 export const TASK_REGISTER = "register"
+export const TASK_ADD_BETA_OPERATOR = "add_beta_operator"
 // Subtask for the Random Beacon application.
 const TASK_INITIALIZE_BEACON = `${TASK_INITIALIZE}:beacon`
 const TASK_AUTHORIZE_BEACON = `${TASK_AUTHORIZE}:beacon`
 const TASK_REGISTER_BEACON = `${TASK_REGISTER}:beacon`
+const TASK_ADD_BETA_OPERATOR_BEACON = `${TASK_ADD_BETA_OPERATOR}:beacon`
 
 task(
   TASK_INITIALIZE,
@@ -45,6 +48,8 @@ task(
     await hre.run(TASK_INITIALIZE_STAKING, args)
     // Initialize Beacon
     await hre.run(TASK_INITIALIZE_BEACON, args)
+    // Set the operator as a beta operator
+    await hre.run(TASK_ADD_BETA_OPERATOR_BEACON, args)
   })
 
 task(TASK_INITIALIZE_STAKING, "Initializes staking for a service provider")
@@ -124,7 +129,16 @@ task(
   "Registers an operator for a staking provider in Beacon"
 )
   .addParam("provider", "Staking Provider", undefined, types.string)
-  .addParam("operator", "Staking Operator", undefined, types.string)
+  .addParam("operator", "Operator Address", undefined, types.string)
   .setAction(async (args, hre) => {
     await register(hre, "RandomBeacon", args.provider, args.operator)
   })
+
+task(
+  TASK_ADD_BETA_OPERATOR_BEACON,
+  "Adds an operator to the set of beta operators in Beacon"
+)
+  .addParam("operator", "Operator Address", undefined, types.string)
+  .setAction(async (args, hre) => {
+    await addBetaOperator(hre, "BeaconSortitionPool", args.operator)
+  })

solidity/random-beacon/tasks/utils/add_beta_operator.ts

@@ -0,0 +1,22 @@
+/* eslint-disable no-console */
+import type { HardhatRuntimeEnvironment } from "hardhat/types"
+
+// eslint-disable-next-line import/prefer-default-export
+export async function addBetaOperator(
+  hre: HardhatRuntimeEnvironment,
+  sortitionPoolDeploymentName: string,
+  operator: string
+): Promise<void> {
+  const { ethers, helpers } = hre
+  const sortitionPool = await helpers.contracts.getContract(
+    sortitionPoolDeploymentName
+  )
+  const chaosnetOwner = await sortitionPool.chaosnetOwner()
+
+  console.log(`Adding ${operator} to the set of beta operators...`)
+  await (
+    await sortitionPool
+      .connect(await ethers.getSigner(chaosnetOwner))
+      .addBetaOperators([operator])
+  ).wait()
+}

solidity/random-beacon/tasks/utils/index.ts

@@ -3,3 +3,4 @@ export * from "./ether"
 export * from "./mint"
 export * from "./register"
 export * from "./stake"
+export * from "./add_beta_operator"