From 20fe1625206b74b897a79d0f1bc0cb2f24524ccd Mon Sep 17 00:00:00 2001
From: marie-fourier <sobirovutkir@gmail.com>
Date: Mon, 28 Nov 2022 20:32:19 +0500
Subject: [PATCH] feat: task for adding guards

---
 extensions/guards/addGuardians.ts | 101 ++++++++++++++++++++++++++++++
 extensions/index.ts               |   1 +
 2 files changed, 102 insertions(+)
 create mode 100644 extensions/guards/addGuardians.ts

diff --git a/extensions/guards/addGuardians.ts b/extensions/guards/addGuardians.ts
new file mode 100644
index 00000000..c8b1e114
--- /dev/null
+++ b/extensions/guards/addGuardians.ts
@@ -0,0 +1,101 @@
+import { task } from "hardhat/config";
+import { GatewayV2 } from "typings/GatewayV2";
+import { fetchJson } from "ethers/lib/utils";
+
+/**
+ * USAGE
+ * npx hardhat add-guards --network [network] (--env [testnet|mainnet]) (--estimator estimator address)
+ * --env is optional, testnet by default
+ * --estimator is optional
+ * 
+ * Example:
+ * npx hardhat add-guards --network goerli --env testnet --estimator 0x0408ff3B0A2cE1a0D5c25a0D4a387332f814Bd72
+ */
+
+interface IRelayer {
+  address: string;
+}
+
+enum EtherspotEnv {
+  testnet = "testnet",
+  mainnet = "mainnet",
+}
+
+interface TaskArgs {
+  env: EtherspotEnv;
+  estimator: string;
+}
+
+const EtherspotEnvUrls = {
+  [EtherspotEnv.testnet]: "https://qa-etherspot.pillarproject.io",
+  [EtherspotEnv.mainnet]: "https://etherspot.pillarproject.io",
+};
+
+const TASK_ADD_GUARDS = "add-guards";
+
+task(TASK_ADD_GUARDS, "Add gw2 guardians")
+  .addParam("env", "Environment", EtherspotEnv.testnet)
+  .addOptionalParam("estimator", "Estimator adderss for the chain")
+  .setAction(async (args: TaskArgs, hre) => {
+    if (!EtherspotEnvUrls[args.env]) {
+      return console.error('Invalid env');
+    }
+    if (args.estimator && !hre.ethers.utils.isAddress(args.estimator)) {
+      return console.error('Invalid estimator');
+    }
+
+    const { network, ethers } = hre;
+    const gw2: GatewayV2 = await ethers.getContract("GatewayV2");
+
+    const relayers: IRelayer[] = await fetchJson(
+      EtherspotEnvUrls[args.env],
+      JSON.stringify({
+        operationName: null,
+        variables: {},
+        query: `{
+        gatewaySenders(chainId: ${network.config.chainId}) {
+          items {
+            address,
+          }
+        }
+      }`,
+      }),
+    ).then(({ data }: any) => {
+      return data?.gatewaySenders?.items;
+    });
+
+    if (!relayers || !relayers.length) {
+      console.log("Relayers are empty");
+      return;
+    }
+    if (args.estimator) {
+      relayers.push({
+        address: args.estimator,
+      });
+    }
+
+    for (const relayer of relayers) {
+      const isGuardian = await gw2.isGuardian(relayer.address);
+      console.log(
+        `Relayer ${relayer.address} ${isGuardian ? "is" : "is not"} a guardian`,
+      );
+
+      if (!isGuardian) {
+        try {
+          console.log(`Adding a guardian...`);
+          const tx = await gw2.addGuardian(relayer.address);
+          console.log(`Waiting for the tx to settle`);
+          await tx.wait();
+          const isGuardian = await gw2.isGuardian(relayer.address);
+          console.log(
+            `Relayer ${relayer.address} ${
+              isGuardian ? "is" : "is not"
+            } a guardian`,
+          );
+        } catch (err) {
+          console.log("Tx failed", err);
+          return;
+        }
+      }
+    }
+  });
diff --git a/extensions/index.ts b/extensions/index.ts
index 8f1c05d7..4507b9c0 100644
--- a/extensions/index.ts
+++ b/extensions/index.ts
@@ -5,6 +5,7 @@ import './deploy-diamonds';
 import './test';
 import './typings';
 import './verify-all';
+import './guards/addGuardians';
 
 export * from './constants';
 export * from './utils';