From bd692c422a81c94deee3570bc4c6d45502f6ddfd Mon Sep 17 00:00:00 2001
From: saucepoint <saucepoint@protonmail.com>
Date: Thu, 14 Mar 2024 12:41:13 +0000
Subject: [PATCH] fix getSlot0 calls

---
 contracts/hooks/examples/GeomeanOracle.sol    |  4 +--
 contracts/hooks/examples/VolatilityOracle.sol | 30 ++++++++++++-------
 2 files changed, 22 insertions(+), 12 deletions(-)

diff --git a/contracts/hooks/examples/GeomeanOracle.sol b/contracts/hooks/examples/GeomeanOracle.sol
index f3e789ef..c0f1c096 100644
--- a/contracts/hooks/examples/GeomeanOracle.sol
+++ b/contracts/hooks/examples/GeomeanOracle.sol
@@ -103,7 +103,7 @@ contract GeomeanOracle is BaseHook {
     /// @dev Called before any action that potentially modifies pool price or liquidity, such as swap or modify position
     function _updatePool(PoolKey calldata key) private {
         PoolId id = key.toId();
-        (, int24 tick,) = poolManager.getSlot0(id);
+        (, int24 tick,,) = poolManager.getSlot0(id);
 
         uint128 liquidity = poolManager.getLiquidity(id);
 
@@ -156,7 +156,7 @@ contract GeomeanOracle is BaseHook {
 
         ObservationState memory state = states[id];
 
-        (, int24 tick,) = poolManager.getSlot0(id);
+        (, int24 tick,,) = poolManager.getSlot0(id);
 
         uint128 liquidity = poolManager.getLiquidity(id);
 
diff --git a/contracts/hooks/examples/VolatilityOracle.sol b/contracts/hooks/examples/VolatilityOracle.sol
index e9b98f91..29d24a6a 100644
--- a/contracts/hooks/examples/VolatilityOracle.sol
+++ b/contracts/hooks/examples/VolatilityOracle.sol
@@ -2,24 +2,23 @@
 pragma solidity ^0.8.19;
 
 import {IPoolManager} from "@uniswap/v4-core/src/interfaces/IPoolManager.sol";
-import {IDynamicFeeManager} from "@uniswap/v4-core/src/interfaces/IDynamicFeeManager.sol";
 import {Hooks} from "@uniswap/v4-core/src/libraries/Hooks.sol";
-import {FeeLibrary} from "@uniswap/v4-core/src/libraries/FeeLibrary.sol";
+import {SwapFeeLibrary} from "@uniswap/v4-core/src/libraries/SwapFeeLibrary.sol";
 import {BaseHook} from "../../BaseHook.sol";
 import {PoolKey} from "@uniswap/v4-core/src/types/PoolKey.sol";
 
-contract VolatilityOracle is BaseHook, IDynamicFeeManager {
-    using FeeLibrary for uint24;
+contract VolatilityOracle is BaseHook {
+    using SwapFeeLibrary for uint24;
 
     error MustUseDynamicFee();
 
     uint32 deployTimestamp;
 
-    function getFee(address, PoolKey calldata) external view returns (uint24) {
-        uint24 startingFee = 3000;
-        uint32 lapsed = _blockTimestamp() - deployTimestamp;
-        return startingFee + (uint24(lapsed) * 100) / 60; // 100 bps a minute
-    }
+    // function getFee(address, PoolKey calldata) external view returns (uint24) {
+    //     uint24 startingFee = 3000;
+    //     uint32 lapsed = _blockTimestamp() - deployTimestamp;
+    //     return startingFee + (uint24(lapsed) * 100) / 60; // 100 bps a minute
+    // }
 
     /// @dev For mocking
     function _blockTimestamp() internal view virtual returns (uint32) {
@@ -33,7 +32,7 @@ contract VolatilityOracle is BaseHook, IDynamicFeeManager {
     function getHookPermissions() public pure override returns (Hooks.Permissions memory) {
         return Hooks.Permissions({
             beforeInitialize: true,
-            afterInitialize: false,
+            afterInitialize: true,
             beforeAddLiquidity: false,
             beforeRemoveLiquidity: false,
             afterAddLiquidity: false,
@@ -54,4 +53,15 @@ contract VolatilityOracle is BaseHook, IDynamicFeeManager {
         if (!key.fee.isDynamicFee()) revert MustUseDynamicFee();
         return VolatilityOracle.beforeInitialize.selector;
     }
+
+    function afterInitialize(address, PoolKey calldata key, uint160, int24, bytes calldata)
+        external
+        override
+        returns (bytes4)
+    {
+        uint24 startingFee = 3000;
+        uint32 lapsed = _blockTimestamp() - deployTimestamp;
+        uint24 fee = startingFee + (uint24(lapsed) * 100) / 60; // 100 bps a minute
+        poolManager.updateDynamicSwapFee(key, fee); // initial fee 0.30%
+    }
 }