From 5b074c7b140d4a40230d54e9b500e29d8311b3a6 Mon Sep 17 00:00:00 2001
From: mingda <leimingda@pku.edu.cn>
Date: Thu, 30 Jul 2020 01:23:49 +0800
Subject: [PATCH] return capital change in deposit

---
 contracts/impl/LiquidityProvider.sol | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/contracts/impl/LiquidityProvider.sol b/contracts/impl/LiquidityProvider.sol
index 7bb2600..338f22e 100644
--- a/contracts/impl/LiquidityProvider.sol
+++ b/contracts/impl/LiquidityProvider.sol
@@ -64,16 +64,16 @@ contract LiquidityProvider is Storage, Pricing, Settlement {
         return withdrawBaseTo(msg.sender, amount);
     }
 
-    function depositBase(uint256 amount) external {
-        depositBaseTo(msg.sender, amount);
+    function depositBase(uint256 amount) external returns (uint256) {
+        return depositBaseTo(msg.sender, amount);
     }
 
     function withdrawQuote(uint256 amount) external returns (uint256) {
         return withdrawQuoteTo(msg.sender, amount);
     }
 
-    function depositQuote(uint256 amount) external {
-        depositQuoteTo(msg.sender, amount);
+    function depositQuote(uint256 amount) external returns (uint256) {
+        return depositQuoteTo(msg.sender, amount);
     }
 
     function withdrawAllBase() external returns (uint256) {
@@ -90,6 +90,7 @@ contract LiquidityProvider is Storage, Pricing, Settlement {
         public
         preventReentrant
         depositQuoteAllowed
+        returns (uint256)
     {
         (, uint256 quoteTarget) = getExpectedTarget();
         uint256 totalQuoteCapital = getTotalQuoteCapital();
@@ -107,9 +108,15 @@ contract LiquidityProvider is Storage, Pricing, Settlement {
         _TARGET_QUOTE_TOKEN_AMOUNT_ = _TARGET_QUOTE_TOKEN_AMOUNT_.add(amount);
 
         emit Deposit(msg.sender, to, false, amount, capital);
+        return capital;
     }
 
-    function depositBaseTo(address to, uint256 amount) public preventReentrant depositBaseAllowed {
+    function depositBaseTo(address to, uint256 amount)
+        public
+        preventReentrant
+        depositBaseAllowed
+        returns (uint256)
+    {
         (uint256 baseTarget, ) = getExpectedTarget();
         uint256 totalBaseCapital = getTotalBaseCapital();
         uint256 capital = amount;
@@ -126,6 +133,7 @@ contract LiquidityProvider is Storage, Pricing, Settlement {
         _TARGET_BASE_TOKEN_AMOUNT_ = _TARGET_BASE_TOKEN_AMOUNT_.add(amount);
 
         emit Deposit(msg.sender, to, true, amount, capital);
+        return capital;
     }
 
     // ============ Withdraw Functions ============