added setter for IOO vault and included into borrow check

mysofinance · Mar 31, 2024 · f0caf0c · f0caf0c
1 parent 6e3044d
commit f0caf0c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/contracts/test/TestnetTokenManager.sol b/contracts/test/TestnetTokenManager.sol
@@ -23,7 +23,8 @@ contract TestnetTokenManager is ERC20, Ownable2Step, IMysoTokenManager {
     uint256 public totalMysoLoanAmount;
     address public mysoIOOVault;
 
-    // Token Manager TODO; maybe allow for any vault owner to track loan token amount for vault and any token
+    // TODO: mapping oracleAddr -> vaultAddr -> tokenAddr -> loanAmount + flag for being turned tracked
+    // This will allow for other IOOs to use a custom oracle with auto-updating price for loan amount if desired
 
     constructor() ERC20("TYSO", "TYSO") {
         _decimals = 18;
@@ -45,7 +46,7 @@ contract TestnetTokenManager is ERC20, Ownable2Step, IMysoTokenManager {
             _mint(loan.borrower, _borrowerReward);
             _mint(lenderVault, _lenderReward);
         }
-        if (loan.loanToken == MYSO_TOKEN) {
+        if (loan.loanToken == MYSO_TOKEN && lenderVault == mysoIOOVault) {
             totalMysoLoanAmount += loan.initLoanAmount;
         }
     }
@@ -124,6 +125,11 @@ contract TestnetTokenManager is ERC20, Ownable2Step, IMysoTokenManager {
         _vaultCreationReward = vaultCreationReward;
     }
 
+    function setIOOVault(address _mysoIOOVault) external {
+        _checkOwner();
+        mysoIOOVault = _mysoIOOVault;
+    }
+
     function transferOwnership(address _newOwnerProposal) public override {
         _checkOwner();
         if (