fix: v2 v3 engine rates import generator

bgd-labs · Oct 25, 2023 · 11ef02c · 11ef02c
1 parent 60aa9c8
commit 11ef02c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/generator/features/rateUpdates.ts b/generator/features/rateUpdates.ts
@@ -94,7 +94,7 @@ export const rateUpdatesV2: FeatureModule<RateStrategyUpdate[]> = {
             .map(
               (cfg, ix) => `rateStrategies[${ix}] = IAaveV2ConfigEngine.RateStrategyUpdate({
                 asset: ${cfg.asset},
-                params: Rates.RateStrategyParams({
+                params: IV2RateStrategyFactory.RateStrategyParams({
                   optimalUtilizationRate: ${cfg.params.optimalUtilizationRate},
                   baseVariableBorrowRate: ${cfg.params.baseVariableBorrowRate},
                   variableRateSlope1: ${cfg.params.variableRateSlope1},
@@ -149,7 +149,7 @@ export const rateUpdatesV3: FeatureModule<RateStrategyUpdate[]> = {
             .map(
               (cfg, ix) => `rateStrategies[${ix}] = IAaveV3ConfigEngine.RateStrategyUpdate({
                   asset: ${cfg.asset},
-                  params: Rates.RateStrategyParams({
+                  params: IV3RateStrategyFactory.RateStrategyParams({
                     optimalUsageRatio: ${cfg.params.optimalUtilizationRate},
                     baseVariableBorrowRate: ${cfg.params.baseVariableBorrowRate},
                     variableRateSlope1: ${cfg.params.variableRateSlope1},

diff --git a/generator/utils/importsResolver.ts b/generator/utils/importsResolver.ts
@@ -83,8 +83,8 @@ export function prefixWithImports(code: string) {
   if (findMatch(code, 'IAaveV3ConfigEngine')) {
     imports += `import {IAaveV3ConfigEngine} from 'aave-helpers/v3-config-engine/IAaveV3ConfigEngine.sol';\n`;
   }
-  if (findMatch(code, 'Rates')) {
-    imports += `import {IV3RateStrategyFactory as Rates} from 'aave-helpers/v3-config-engine/IV3RateStrategyFactory.sol';\n`;
+  if (findMatch(code, 'IV3RateStrategyFactory')) {
+    imports += `import {IV3RateStrategyFactory} from 'aave-helpers/v3-config-engine/IV3RateStrategyFactory.sol';\n`;
   }
   // v2 config engine imports
   if (findMatch(code, 'IAaveV2ConfigEngine')) {