Skip to content

Commit e3224d2

Browse files
committed
fix: description for cms
1 parent 390429a commit e3224d2

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

contracts/data/DataCompressor_2_1.sol

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@ import {PoolService} from "@gearbox-protocol/core-v2/contracts/pool/PoolService.
2525

2626
import {IVersion} from "@gearbox-protocol/core-v2/contracts/interfaces/IVersion.sol";
2727

28+
import {ACLNonReentrantTrait} from "@gearbox-protocol/core-v3/contracts/traits/ACLNonReentrantTrait.sol";
29+
2830
import {IAddressProvider} from "@gearbox-protocol/core-v2/contracts/interfaces/IAddressProvider.sol";
2931
import {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

Comments
 (0)