From 79b18475bc2375782353916c9c4084861e67fba3 Mon Sep 17 00:00:00 2001
From: Lukas <lukasstrassel@googlemail.com>
Date: Thu, 18 Jan 2024 17:47:49 +0100
Subject: [PATCH] feat: add stk AAVE wstETH BPTv2 (#330)

* feat: add stk AAVE wstETH BPTv2

* fix: add migrator
---
 scripts/generator/safetyModuleGenerator.ts |  4 +++-
 src/AaveSafetyModule.sol                   | 12 ++++++++++--
 src/ts/AaveSafetyModule.ts                 | 10 ++++++++--
 3 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/scripts/generator/safetyModuleGenerator.ts b/scripts/generator/safetyModuleGenerator.ts
index cc55e06d..baa67f74 100644
--- a/scripts/generator/safetyModuleGenerator.ts
+++ b/scripts/generator/safetyModuleGenerator.ts
@@ -12,8 +12,10 @@ const SAFETY_MODULE = {
   STK_AAVE: '0x4da27a545c0c5B758a6BA100e3a049001de870f5',
   STK_ABPT: '0xa1116930326D21fB917d5A27F1E9943A9595fb47',
   STK_ABPT_ORACLE: '0x209Ad99bd808221293d03827B86cC544bcA0023b',
-  STK_ABPT_V2_ORACLE: '0xADf86b537eF08591c2777E144322E8b0Ca7E82a7',
+  STK_ABPT_STK_AAVE_WSTETH_BPTV2_MIGRATOR: '0xecD4bd3121F9FD604ffaC631bF6d41ec12f1fafb',
   STK_GHO: '0x1a88Df1cFe15Af22B3c4c783D4e6F7F9e0C1885d',
+  STK_AAVE_WSTETH_BPTV2: '0x9eDA81C21C273a82BE9Bbc19B6A6182212068101',
+  STK_AAVE_WSTETH_BPTV2_ORACLE: '0xADf86b537eF08591c2777E144322E8b0Ca7E82a7',
 } as const;
 
 export function generateSafetyModule() {
diff --git a/src/AaveSafetyModule.sol b/src/AaveSafetyModule.sol
index 65b425be..621f1665 100644
--- a/src/AaveSafetyModule.sol
+++ b/src/AaveSafetyModule.sol
@@ -12,9 +12,17 @@ library AaveSafetyModule {
   // https://etherscan.io/address/0x209Ad99bd808221293d03827B86cC544bcA0023b
   address internal constant STK_ABPT_ORACLE = 0x209Ad99bd808221293d03827B86cC544bcA0023b;
 
-  // https://etherscan.io/address/0xADf86b537eF08591c2777E144322E8b0Ca7E82a7
-  address internal constant STK_ABPT_V2_ORACLE = 0xADf86b537eF08591c2777E144322E8b0Ca7E82a7;
+  // https://etherscan.io/address/0xecD4bd3121F9FD604ffaC631bF6d41ec12f1fafb
+  address internal constant STK_ABPT_STK_AAVE_WSTETH_BPTV2_MIGRATOR =
+    0xecD4bd3121F9FD604ffaC631bF6d41ec12f1fafb;
 
   // https://etherscan.io/address/0x1a88Df1cFe15Af22B3c4c783D4e6F7F9e0C1885d
   address internal constant STK_GHO = 0x1a88Df1cFe15Af22B3c4c783D4e6F7F9e0C1885d;
+
+  // https://etherscan.io/address/0x9eDA81C21C273a82BE9Bbc19B6A6182212068101
+  address internal constant STK_AAVE_WSTETH_BPTV2 = 0x9eDA81C21C273a82BE9Bbc19B6A6182212068101;
+
+  // https://etherscan.io/address/0xADf86b537eF08591c2777E144322E8b0Ca7E82a7
+  address internal constant STK_AAVE_WSTETH_BPTV2_ORACLE =
+    0xADf86b537eF08591c2777E144322E8b0Ca7E82a7;
 }
diff --git a/src/ts/AaveSafetyModule.ts b/src/ts/AaveSafetyModule.ts
index 4565c192..c83a6dca 100644
--- a/src/ts/AaveSafetyModule.ts
+++ b/src/ts/AaveSafetyModule.ts
@@ -8,8 +8,14 @@ export const STK_ABPT = '0xa1116930326D21fB917d5A27F1E9943A9595fb47';
 // https://etherscan.io/address/0x209Ad99bd808221293d03827B86cC544bcA0023b
 export const STK_ABPT_ORACLE = '0x209Ad99bd808221293d03827B86cC544bcA0023b';
 
-// https://etherscan.io/address/0xADf86b537eF08591c2777E144322E8b0Ca7E82a7
-export const STK_ABPT_V2_ORACLE = '0xADf86b537eF08591c2777E144322E8b0Ca7E82a7';
+// https://etherscan.io/address/0xecD4bd3121F9FD604ffaC631bF6d41ec12f1fafb
+export const STK_ABPT_STK_AAVE_WSTETH_BPTV2_MIGRATOR = '0xecD4bd3121F9FD604ffaC631bF6d41ec12f1fafb';
 
 // https://etherscan.io/address/0x1a88Df1cFe15Af22B3c4c783D4e6F7F9e0C1885d
 export const STK_GHO = '0x1a88Df1cFe15Af22B3c4c783D4e6F7F9e0C1885d';
+
+// https://etherscan.io/address/0x9eDA81C21C273a82BE9Bbc19B6A6182212068101
+export const STK_AAVE_WSTETH_BPTV2 = '0x9eDA81C21C273a82BE9Bbc19B6A6182212068101';
+
+// https://etherscan.io/address/0xADf86b537eF08591c2777E144322E8b0Ca7E82a7
+export const STK_AAVE_WSTETH_BPTV2_ORACLE = '0xADf86b537eF08591c2777E144322E8b0Ca7E82a7';