From 856fb596dac0af52420904ff54b98649e75a318e Mon Sep 17 00:00:00 2001
From: Piotr Dyraga <piotr.dyraga@keep.network>
Date: Fri, 23 Sep 2022 16:23:22 +0200
Subject: [PATCH] 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.
---
 solidity/random-beacon/tasks/initialize.ts    | 16 +++++++++++++-
 .../tasks/utils/add_beta_operator.ts          | 22 +++++++++++++++++++
 solidity/random-beacon/tasks/utils/index.ts   |  1 +
 3 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 solidity/random-beacon/tasks/utils/add_beta_operator.ts

diff --git a/solidity/random-beacon/tasks/initialize.ts b/solidity/random-beacon/tasks/initialize.ts
index 34c289e425..07cc32f520 100644
--- a/solidity/random-beacon/tasks/initialize.ts
+++ b/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)
+  })
diff --git a/solidity/random-beacon/tasks/utils/add_beta_operator.ts b/solidity/random-beacon/tasks/utils/add_beta_operator.ts
new file mode 100644
index 0000000000..6f5ee8c971
--- /dev/null
+++ b/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()
+}
diff --git a/solidity/random-beacon/tasks/utils/index.ts b/solidity/random-beacon/tasks/utils/index.ts
index 43249cb37d..46ff9b0c5d 100644
--- a/solidity/random-beacon/tasks/utils/index.ts
+++ b/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"