chore: check for v3 vault

yearn · Oct 15, 2024 · 33953b5 · 33953b5
1 parent 30ff05d
commit 33953b5
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/AprOracle/AprOracle.sol b/src/AprOracle/AprOracle.sol
@@ -83,8 +83,13 @@ contract AprOracle is Governance {
         if (oracle != address(0)) {
             return IOracle(oracle).aprAfterDebtChange(_strategy, _debtChange);
         } else {
-            // This will revert for non v3 vaults.
-            return getExpectedApr(_strategy, _debtChange);
+            // If the vault is a v3 vault, we can get default to the expected apr.
+            try IVault(_strategy).fullProfitUnlockDate() returns (uint256) {
+                return getExpectedApr(_strategy, _debtChange);
+            } catch {
+                // Else just return 0.
+                return 0;
+            }
         }
     }