using Binance.TradeRobot.Model.Base;

namespace Binance.TradeRobot.Model.Dto
{
    public class UserResponse : Db.User
    {
        /// <summary>
        /// 总资产(本金+收益)
        /// </summary>
        public decimal TotalAssets { get { return CostAmount + Profit; } }

        /// <summary>
        /// 本金比例
        /// </summary>
        public decimal CostRatio { get; set; }

        /// <summary>
        /// 分红比例
        /// </summary>
        public decimal DividendRatio { get; set; }

        /// <summary>
        /// 用户资金改变算法
        /// </summary>
        /// <param name="capitalChangeType"></param>
        /// <param name="changeAmount"></param>
        /// <param name="priorityAddCost">true:优先增加本金 false:优先增加利润</param>
        public void ChangeAmount(Enums.CapitalChangeType capitalChangeType, decimal changeAmount, bool priorityAddCost)
        {
            if (capitalChangeType == Enums.CapitalChangeType.Add)
            {
                if (priorityAddCost)
                    CostAmount += changeAmount;
                else
                    Profit += changeAmount;
            }
            else if (capitalChangeType == Enums.CapitalChangeType.Reduce)
            {
                if (Profit > 0)
                {
                    if (Profit >= changeAmount)
                        Profit -= changeAmount;  //收益足够提现,只扣收益
                    else
                    {
                        var lessChangeAmount = changeAmount;  //收益不足提现,先扣收益,不足部分再扣本金
                        lessChangeAmount -= Profit;
                        Profit = 0;
                        CostAmount -= lessChangeAmount;
                    }
                }
                else
                {
                    CostAmount -= changeAmount;
                }
            }
        }
    }
}