refactor: lp_shared to use snekmate.

velodrome-finance · Dec 20, 2024 · b2addcd · b2addcd
1 parent 5dcf377
commit b2addcd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 17 deletions.
diff --git a/ape-config.yaml b/ape-config.yaml
@@ -4,9 +4,10 @@ vyper:
   evm_version: cancun
 
 dependencies:
-  - name: snekmate
-    github: pcaversaccio/snekmate
-    version: 0.1.0
+  - pypi: snekmate
+    config_override:
+      base_path: src
+      contracts_folder: snekmate
 
 networks:
   custom:

diff --git a/contracts/modules/lp_shared.vy b/contracts/modules/lp_shared.vy
@@ -4,6 +4,8 @@
 # @title Velodrome Finance LP Module
 # @author Velodrome Finance
 
+from snekmate.utils import create2_address
+
 MAX_FACTORIES: constant(uint256) = 10
 MAX_POOLS: constant(uint256) = 2000
 MAX_ITERATIONS: constant(uint256) = 8000
@@ -202,17 +204,4 @@ def _root_lp_address(
       )
     )
 
-
-  data: bytes32 = keccak256(
-    concat(
-      0xFF,
-      convert(_factory, bytes20),
-      salt,
-      init_hash
-    )
-  )
-
-  return convert(
-    convert(data, uint256) & convert(max_value(uint160), uint256),
-    address
-  )
+  return create2_address._compute_address(salt, init_hash, _factory)