@@ -25,6 +25,8 @@ import {PoolService} from "@gearbox-protocol/core-v2/contracts/pool/PoolService.
2525
2626import {IVersion} from "@gearbox-protocol/core-v2/contracts/interfaces/IVersion.sol " ;
2727
28+ import {ACLNonReentrantTrait} from "@gearbox-protocol/core-v3/contracts/traits/ACLNonReentrantTrait.sol " ;
29+
2830import {IAddressProvider} from "@gearbox-protocol/core-v2/contracts/interfaces/IAddressProvider.sol " ;
2931import {IDataCompressorV2_10} from "../interfaces/IDataCompressorV2_10.sol " ;
3032
@@ -39,11 +41,21 @@ import {LinearInterestModelHelper} from "./LinearInterestModelHelper.sol";
3941/// @title Data compressor 2.1.
4042/// @notice Collects data from various contracts for use in the dApp
4143/// Do not use for data from data compressor for state-changing functions
42- contract DataCompressorV2_10 is IDataCompressorV2_10 , ContractsRegisterTrait , LinearInterestModelHelper {
44+ contract DataCompressorV2_10 is
45+ IDataCompressorV2_10 ,
46+ ACLNonReentrantTrait ,
47+ ContractsRegisterTrait ,
48+ LinearInterestModelHelper
49+ {
4350 // Contract version
4451 uint256 public constant version = 2_10 ;
4552
46- constructor (address _addressProvider ) ContractsRegisterTrait (_addressProvider) {}
53+ mapping (address => string ) public cmDescriptions;
54+
55+ constructor (address _addressProvider )
56+ ACLNonReentrantTrait (_addressProvider)
57+ ContractsRegisterTrait (_addressProvider)
58+ {}
4759
4860 /// @dev Returns CreditAccountData for all opened accounts for particular borrower
4961 /// @param borrower Borrower address
@@ -172,6 +184,7 @@ contract DataCompressorV2_10 is IDataCompressorV2_10, ContractsRegisterTrait, Li
172184
173185 result.addr = _creditManager;
174186 result.cfVersion = ver;
187+ result.description = cmDescriptions[_creditManager];
175188
176189 result.underlying = creditManagerV2.underlying ();
177190
@@ -390,4 +403,8 @@ contract DataCompressorV2_10 is IDataCompressorV2_10, ContractsRegisterTrait, Li
390403 }
391404 }
392405 }
406+
407+ function setCreditManagerDescription (address _cm , string calldata description ) external controllerOnly {
408+ cmDescriptions[_cm] = description;
409+ }
393410}
0 commit comments