fix: init/deinit module via safe account

rhinestonewtf · Feb 21, 2024 · d7a2fe3 · d7a2fe3
1 parent 0ae520a
commit d7a2fe3
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/accounts/safe7579/src/core/HookManager.sol b/accounts/safe7579/src/core/HookManager.sol
@@ -60,12 +60,23 @@ abstract contract HookManager is ModuleManager {
             revert HookAlreadyInstalled(currentHook);
         }
         _setHook(hook);
-        IHook(hook).onInstall(data);
+
+        _execute({
+            safe: msg.sender,
+            target: hook,
+            value: 0,
+            callData: abi.encodeCall(IModule.onInstall, (data))
+        });
     }
 
     function _uninstallHook(address hook, bytes calldata data) internal virtual {
         _setHook(address(0));
-        IHook(hook).onUninstall(data);
+        _execute({
+            safe: msg.sender,
+            target: hook,
+            value: 0,
+            callData: abi.encodeCall(IModule.onUninstall, (data))
+        });
     }
 
     function _getHook(address smartAccount) internal view returns (address _hook) {