Add operator != msg.sender. Add setOperator to cli

Author: davekay100

cli/commands/contracts/staking.ts

@@ -124,6 +124,13 @@ export const setDelegationParameters = async (
     ...[indexingRewardCut, queryFeeCut, cooldownBlocks],
   )
 }
+export const setOperator = async (cli: CLIEnvironment, cliArgs: CLIArgs): Promise<void> => {
+  const operator = cliArgs.operator
+  const allowed = cliArgs.allowed
+  const staking = cli.contracts.Staking
+  logger.log(`Setting operator ${operator} to ${allowed} for account ${cli.walletAddress}`)
+  await sendTransaction(cli.wallet, staking, 'setOperator', ...[operator, allowed])
+}
 
 export const stakingCommand = {
   command: 'staking',
@@ -359,12 +366,32 @@ export const stakingCommand = {
             .option('cooldownBlocks', {
               description: 'Period that need to pass to update delegation parameters',
               type: 'number',
+            })
+        },
+        handler: async (argv: CLIArgs): Promise<void> => {
+          return setDelegationParameters(await loadEnv(argv), argv)
+        },
+      })
+      .command({
+        command: 'setOperator',
+        describe: 'Set the operator for a graph account',
+        builder: (yargs: Argv) => {
+          return yargs
+            .option('operator', {
+              description: 'Address of the operator',
+              type: 'string',
+              requiresArg: true,
+              demandOption: true,
+            })
+            .option('allowed', {
+              description: 'Set to true to be an operator, false to revoke',
+              type: 'boolean',
               requiresArg: true,
               demandOption: true,
             })
         },
         handler: async (argv: CLIArgs): Promise<void> => {
-          return setDelegationParameters(await loadEnv(argv), argv)
+          return setOperator(await loadEnv(argv), argv)
         },
       })
   },

contracts/staking/Staking.sol

@@ -493,6 +493,7 @@ contract Staking is StakingV1Storage, GraphUpgradeable, IStaking {
      * @param _allowed Whether authorized or not
      */
     function setOperator(address _operator, bool _allowed) external override {
+        require(_operator != msg.sender, "operator != sender");
         operatorAuth[msg.sender][_operator] = _allowed;
         emit SetOperator(msg.sender, _operator, _allowed);
     }