named imports

Uniswap · Nov 28, 2023 · 3f0e6a7 · 3f0e6a7
1 parent 9fb0ab4
commit 3f0e6a7
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/contracts/interfaces/IQuoter.sol b/contracts/interfaces/IQuoter.sol
@@ -14,6 +14,7 @@ interface IQuoter {
     error InvalidQuoteTypeInRevert();
     error InvalidLockAcquiredSender();
     error InvalidQuoteBatchParams();
+    error NotSelf();
     error UnexpectedRevertBytes();
 
     struct PoolDeltas {

diff --git a/contracts/lens/Quoter.sol b/contracts/lens/Quoter.sol
@@ -10,7 +10,17 @@ import {BalanceDelta} from "@uniswap/v4-core/contracts/types/BalanceDelta.sol";
 import {Currency} from "@uniswap/v4-core/contracts/types/Currency.sol";
 import {PoolKey} from "@uniswap/v4-core/contracts/types/PoolKey.sol";
 import {PoolIdLibrary} from "@uniswap/v4-core/contracts/types/PoolId.sol";
-import "../libraries/SwapIntention.sol";
+import {
+    SwapType,
+    SwapInfo,
+    SwapIntention,
+    ExactInputSingleParams,
+    ExactInputSingleBatchParams,
+    ExactInputParams,
+    ExactOutputSingleParams,
+    ExactOutputSingleBatchParams,
+    ExactOutputParams
+} from "../libraries/SwapIntention.sol";
 import {IQuoter} from "../interfaces/IQuoter.sol";
 import {PoolTicksCounter} from "../libraries/PoolTicksCounter.sol";
 
@@ -31,6 +41,12 @@ contract Quoter is IQuoter {
     /// @dev int128[2] + sqrtPriceX96After padded to 32bytes + intializeTicksLoaded padded to 32bytes
     uint256 internal constant MINIMUM_VALID_REASON_LENGTH = 96;
 
+    /// @dev Only this address may call this function
+    modifier selfOnly() {
+        if (msg.sender != address(this)) revert NotSelf();
+        _;
+    }
+
     constructor(address _poolManager) {
         manager = IPoolManager(_poolManager);
     }