Support contract consumer set by self (#369)

subquery · Feb 19, 2024 · 6d9d562 · 6d9d562
1 parent a6eca6b
commit 6d9d562
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/contracts/ConsumerRegistry.sol b/contracts/ConsumerRegistry.sol
@@ -54,12 +54,14 @@ contract ConsumerRegistry is Initializable, OwnableUpgradeable {
      * @dev Consumer add controller can request access token from indexer.
      */
     function addController(address consumer, address controller) external {
-        if (_isContract(consumer)) {
-            require(consumer.supportsInterface(type(IConsumer).interfaceId), 'CR002');
-            IConsumer cConsumer = IConsumer(consumer);
-            require(cConsumer.isSigner(msg.sender), 'CR003');
-        } else {
-            require(msg.sender == consumer, 'CR001');
+        if (msg.sender != consumer) {
+            if (_isContract(consumer)) {
+                require(consumer.supportsInterface(type(IConsumer).interfaceId), 'CR002');
+                IConsumer cConsumer = IConsumer(consumer);
+                require(cConsumer.isSigner(msg.sender), 'CR003');
+            } else {
+                revert('CR001');
+            }
         }
 
         controllers[consumer][controller] = true;