[hebaov2] Add resetGuardians method in SmartWallet (#2400)

* [hebaov2] Add resetGuardians method in SmartWallet

* fix feedback

Author: kongliangzhong

packages/hebao_v2/contracts/base/SmartWallet.sol

@@ -215,6 +215,24 @@ contract SmartWallet is ERC1271
         wallet.removeGuardianWA(DOMAIN_SEPARATOR, approval, guardian);
     }
 
+     function resetGuardians(
+         address[] calldata newGuardians
+         )
+         external
+         onlyFromWalletOrOwnerWhenUnlocked
+     {
+         wallet.resetGuardians(newGuardians);
+     }
+
+     function resetGuardiansWA(
+         Approval  calldata approval,
+         address[] calldata newGuardians
+         )
+         external
+     {
+         wallet.resetGuardiansWA(DOMAIN_SEPARATOR, approval, newGuardians);
+     }
+
      function isGuardian(address addr, bool includePendingAddition)
          public
          view

packages/hebao_v2/contracts/base/libwallet/GuardianLib.sol

@@ -28,6 +28,9 @@ library GuardianLib
     bytes32 public constant REMOVE_GUARDIAN_TYPEHASH = keccak256(
         "removeGuardian(address wallet,uint256 validUntil,address guardian)"
     );
+    bytes32 public constant RESET_GUARDIANS_TYPEHASH = keccak256(
+        "resetGuardians(address wallet,uint256 validUntil,address[] guardians)"
+    );
 
     event GuardianAdded   (address guardian, uint effectiveTime);
     event GuardianRemoved (address guardian, uint effectiveTime);
@@ -108,6 +111,48 @@ library GuardianLib
         _removeGuardian(wallet, guardian, 0, true);
     }
 
+    function resetGuardians(
+        Wallet    storage  wallet,
+        address[] calldata newGuardians
+        )
+        external
+    {
+        Guardian[] memory allGuardians = guardians(wallet, true);
+        for (uint i = 0; i < allGuardians.length; i++) {
+            _removeGuardian(wallet, allGuardians[i].addr, GUARDIAN_PENDING_PERIOD, false);
+        }
+
+        for (uint j = 0; j < newGuardians.length; j++) {
+            _addGuardian(wallet, newGuardians[j], GUARDIAN_PENDING_PERIOD, false);
+        }
+    }
+
+    function resetGuardiansWA(
+        Wallet    storage  wallet,
+        bytes32            domainSeperator,
+        Approval  calldata approval,
+        address[] calldata newGuardians
+        )
+        external
+    {
+        wallet.verifyApproval(
+            domainSeperator,
+            SigRequirement.MAJORITY_OWNER_REQUIRED,
+            approval,
+            abi.encode(
+                RESET_GUARDIANS_TYPEHASH,
+                approval.wallet,
+                approval.validUntil,
+                keccak256(abi.encodePacked(newGuardians))
+            )
+        );
+
+        removeAllGuardians(wallet);
+        for (uint i = 0; i < newGuardians.length; i++) {
+            _addGuardian(wallet, newGuardians[i], 0, true);
+        }
+    }
+
     function requireMajority(
         Wallet         storage wallet,
         address[]      memory  signers,