package com.xcong.excoin.modules.contract.service.impl;

import cn.hutool.core.collection.CollUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xcong.excoin.common.LoginUserUtils;
import com.xcong.excoin.common.enumerates.CoinTypeEnum;
import com.xcong.excoin.common.enumerates.MemberWalletCoinEnum;
import com.xcong.excoin.common.enumerates.RabbitPriceTypeEnum;
import com.xcong.excoin.common.exception.GlobalException;
import com.xcong.excoin.common.response.Result;
import com.xcong.excoin.common.system.service.CommonService;
import com.xcong.excoin.modules.contract.dao.ContractEntrustOrderDao;
import com.xcong.excoin.modules.contract.dao.ContractHoldOrderDao;
import com.xcong.excoin.modules.contract.dao.ContractOrderDao;
import com.xcong.excoin.modules.contract.entity.ContractEntrustOrderEntity;
import com.xcong.excoin.modules.contract.entity.ContractHoldOrderEntity;
import com.xcong.excoin.modules.contract.entity.ContractOrderEntity;
import com.xcong.excoin.modules.contract.mapper.ContractEntrustOrderEntityMapper;
import com.xcong.excoin.modules.contract.parameter.dto.SubmitCloseEntrustDto;
import com.xcong.excoin.modules.contract.parameter.dto.SubmitEntrustDto;
import com.xcong.excoin.modules.contract.parameter.dto.SubmitOrderDto;
import com.xcong.excoin.modules.contract.parameter.dto.WholeCloseOrderDto;
import com.xcong.excoin.modules.contract.parameter.vo.ContractEntrustVo;
import com.xcong.excoin.modules.contract.service.ContractEntrustOrderService;
import com.xcong.excoin.modules.contract.service.ContractHoldOrderService;
import com.xcong.excoin.modules.member.dao.MemberDao;
import com.xcong.excoin.modules.member.dao.MemberWalletContractDao;
import com.xcong.excoin.modules.member.entity.MemberEntity;
import com.xcong.excoin.modules.member.entity.MemberWalletContractEntity;
import com.xcong.excoin.modules.platform.entity.PlatformTradeSettingEntity;
import com.xcong.excoin.rabbit.pricequeue.OrderModel;
import com.xcong.excoin.rabbit.producer.OrderProducer;
import com.xcong.excoin.utils.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.List;

/**
 * @author wzy
 * @date 2020-05-27
 **/
@Slf4j
@Service
public class ContractEntrustOrderServiceImpl extends ServiceImpl<ContractEntrustOrderDao, ContractEntrustOrderEntity> implements ContractEntrustOrderService {

    @Resource
    private ContractEntrustOrderDao contractEntrustOrderDao;

    @Resource
    private MemberWalletContractDao memberWalletContractDao;

    @Resource
    private RedisUtils redisUtils;

    @Resource
    private CacheSettingUtils cacheSettingUtils;

    @Resource
    private ContractOrderDao contractOrderDao;
    
    @Resource
    private MemberDao memberDao;

    @Resource
    private CommonService commonService;

    @Resource
    private OrderProducer producer;

    @Resource
    private ContractHoldOrderDao contractHoldOrderDao;

    @Resource
    private ContractHoldOrderService contractHoldOrderService;

    @Transactional(rollbackFor = Exception.class)
    @Override
    public Result addContractEntrustOrder(SubmitEntrustDto submitEntrustDto) {
        MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
        //查询是否可以点击平仓按钮
        Long memberIdLong = memberEntity.getId();
        MemberEntity selectById = memberDao.selectById(memberIdLong);
        Integer pcState = selectById.getPcState();
        if(MemberEntity.PCSTATE_N == pcState) {
        	return Result.loading("loading_type");
        }

        // 获取最新价
        BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(submitEntrustDto.getSymbol())));

        // 委托开仓
        if (submitEntrustDto.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_MORE || submitEntrustDto.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_LESS) {
            // 开多委托价不能大于当前价
            if (submitEntrustDto.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_MORE) {
                if (submitEntrustDto.getEntrustPrice().compareTo(newPrice) > -1) {
                    SubmitOrderDto submitOrderDto = new SubmitOrderDto();
                    submitOrderDto.setOrderType(submitEntrustDto.getEntrustType());
                    submitOrderDto.setSymbol(submitEntrustDto.getSymbol());
                    submitOrderDto.setSymbolCnt(submitEntrustDto.getSymbolCnt());
                    submitOrderDto.setLeverRatio(submitEntrustDto.getLeverRatio());
                    return contractHoldOrderService.submitOrder(submitOrderDto);
                }
            }

            // 开空委托价不能小于当前价
            if (submitEntrustDto.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_LESS) {
                if (submitEntrustDto.getEntrustPrice().compareTo(newPrice) < 1) {
                    SubmitOrderDto submitOrderDto = new SubmitOrderDto();
                    submitOrderDto.setOrderType(submitEntrustDto.getEntrustType());
                    submitOrderDto.setSymbol(submitEntrustDto.getSymbol());
                    submitOrderDto.setSymbolCnt(submitEntrustDto.getSymbolCnt());
                    submitOrderDto.setLeverRatio(submitEntrustDto.getLeverRatio());
                    return contractHoldOrderService.submitOrder(submitOrderDto);
                }
            }

            BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(submitEntrustDto.getSymbol());
            PlatformTradeSettingEntity tradeSettingEntity = cacheSettingUtils.getTradeSetting();

            MemberWalletContractEntity walletContract = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());

            // 保证金计算 -- 建仓价X规格X手数X（1/杠杆倍率）
            BigDecimal bondAmount = CalculateUtil.getBondAmount(submitEntrustDto.getEntrustPrice(), lotNumber, submitEntrustDto.getSymbolCnt(), submitEntrustDto.getLeverRatio());

            // 开仓手续费 建仓价*规格*手数*手续费率
            BigDecimal openFeePrice = CalculateUtil.getOpenFeePrice(submitEntrustDto.getEntrustPrice(), lotNumber, submitEntrustDto.getSymbolCnt(), tradeSettingEntity.getFeeRatio());
            log.info("手续费:{}", openFeePrice);

            // 预付款
            BigDecimal entrustTotalAmount = bondAmount.add(openFeePrice).add(openFeePrice);
            log.info("预付款:{}", entrustTotalAmount);

            if (entrustTotalAmount.compareTo(walletContract.getAvailableBalance()) > -1) {
                throw new GlobalException(MessageSourceUtils.getString("member_service_0085"));
            }

            ContractEntrustOrderEntityMapper convert = ContractEntrustOrderEntityMapper.INSTANCE;
            ContractEntrustOrderEntity entrustOrderEntity = convert.submitEntrustDtoToEntity(submitEntrustDto);
            entrustOrderEntity.setOrderNo(commonService.generateOrderNo(memberEntity.getId()));
            entrustOrderEntity.setMemberId(memberEntity.getId());
            entrustOrderEntity.setBondAmount(bondAmount.add(openFeePrice));
            entrustOrderEntity.setSymbolSku(lotNumber);
            entrustOrderEntity.setEntrustAmount(entrustTotalAmount);
            entrustOrderEntity.setPositionType(memberEntity.getContractPositionType());

            int i = contractEntrustOrderDao.insert(entrustOrderEntity);
            memberWalletContractDao.increaseWalletContractBalanceById(entrustTotalAmount.negate(), null, entrustOrderEntity.getBondAmount(), walletContract.getId());
            if (i > 0) {

                // 发送委托单队列消息
                if (submitEntrustDto.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_MORE) {
                    OrderModel model = new OrderModel(entrustOrderEntity.getId(), RabbitPriceTypeEnum.ENTRUST_OPEN_MORE.getValue(), submitEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), submitEntrustDto.getSymbol(), entrustOrderEntity.getMemberId());
                    producer.sendPriceOperate(JSONObject.toJSONString(model));

                    LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), entrustTotalAmount, walletContract.getAvailableBalance().subtract(entrustTotalAmount), submitEntrustDto.getSymbol(), "委托买涨", "买涨:" + submitEntrustDto.getSymbol());
                } else {
                    OrderModel model = new OrderModel(entrustOrderEntity.getId(), RabbitPriceTypeEnum.ENTRUST_OPEN_LESS.getValue(), submitEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), submitEntrustDto.getSymbol(), entrustOrderEntity.getMemberId());
                    producer.sendPriceOperate(JSONObject.toJSONString(model));

                    LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), entrustTotalAmount, walletContract.getAvailableBalance().subtract(entrustTotalAmount), submitEntrustDto.getSymbol(), "委托买跌", "买跌:" + submitEntrustDto.getSymbol());
                }

                return Result.ok(MessageSourceUtils.getString("result_success_msg"));
            } else {
                return Result.fail(MessageSourceUtils.getString("result_fail_msg"));
            }
        }

        return Result.fail(MessageSourceUtils.getString("result_fail_msg"));
    }

    @Override
    public Result findEntrustOrderList(String symbol) {
        MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
        List<ContractEntrustOrderEntity> list = contractEntrustOrderDao.selectEntrustOrderListByMemberIdAndSymbol(memberEntity.getId(), symbol);
        List<ContractEntrustVo> resultList = ContractEntrustOrderEntityMapper.INSTANCE.entityListToVoList(list);
        return Result.ok(resultList);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result cancelEntrustOrder(Long id) {
        MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
        //查询是否可以点击平仓按钮
        Long memberIdLong = memberEntity.getId();
        MemberEntity selectById = memberDao.selectById(memberIdLong);
        Integer pcState = selectById.getPcState();
        if(MemberEntity.PCSTATE_N == pcState) {
        	return Result.loading("loading_type");
        }

        // 查询该委托单是否为该用户所有
        ContractEntrustOrderEntity entrustOrderEntity = contractEntrustOrderDao.selectEntrustOrderByIdAndMemberId(id, memberEntity.getId());
        if (entrustOrderEntity == null) {
            return Result.fail(MessageSourceUtils.getString("entrust_order_not_exist"));
        }
        ContractOrderEntity orderEntity = ContractEntrustOrderEntityMapper.INSTANCE.entrustOrderToOrder(entrustOrderEntity);
        orderEntity.setTradeType(ContractOrderEntity.TRADE_TYPE_MARK_PRICE);
        orderEntity.setOrderStatus(ContractOrderEntity.ORDER_STATUS_CANCEL);

        if (entrustOrderEntity.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_CLOSE_MORE) {
            orderEntity.setClosingType(RabbitPriceTypeEnum.ENTRUST_OPEN_MORE.getValue());
        }

        if (entrustOrderEntity.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_CLOSE_LESS) {
            orderEntity.setClosingType(RabbitPriceTypeEnum.ENTRUST_OPEN_LESS.getValue());
        }

        int i = contractOrderDao.insert(orderEntity);

        contractEntrustOrderDao.deleteById(entrustOrderEntity.getId());

        if (i > 0) {
            MemberWalletContractEntity walletContractEntity = memberWalletContractDao.findWalletContractByMemberIdAndSymbol(memberEntity.getId(), CoinTypeEnum.USDT.name());
            BigDecimal total = entrustOrderEntity.getEntrustAmount();
            if (entrustOrderEntity.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_MORE || entrustOrderEntity.getEntrustType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_LESS) {
                memberWalletContractDao.increaseWalletContractBalanceById(total, null, entrustOrderEntity.getBondAmount().negate(), walletContractEntity.getId());

                // 插入财务流水
                LogRecordUtils.insertMemberAccountFlow(memberEntity.getId(), total, walletContractEntity.getAvailableBalance().add(total), entrustOrderEntity.getSymbol(), "撤销委托单", "撤销委托单");
            }
            if (ContractEntrustOrderEntity.POSITION_TYPE_ALL == entrustOrderEntity.getPositionType()) {
                ThreadPoolUtils.sendWholeForceClosingPrice(entrustOrderEntity.getSymbol(), memberEntity);
            }
            return Result.ok(MessageSourceUtils.getString("cancellation_success"));
        }
        return Result.fail(MessageSourceUtils.getString("cancellation_fail"));
    }

    @Override
    public List<ContractEntrustOrderEntity> selectEntrustOrderListByIds(List<Long> list) {
        return contractEntrustOrderDao.selectEntrustOrderListByIds(list);
    }

    @Override
    public Result addCloseContractEntrustOrder(SubmitCloseEntrustDto submitCloseEntrustDto) {
        MemberEntity member = LoginUserUtils.getAppLoginUser();
        //查询是否可以点击平仓按钮
        Long memberIdLong = member.getId();
        MemberEntity selectById = memberDao.selectById(memberIdLong);
        Integer pcState = selectById.getPcState();
        if(MemberEntity.PCSTATE_N == pcState) {
        	return Result.loading("loading_type");
        }
        ContractHoldOrderEntity holdOrderEntity = contractHoldOrderDao.selectHoldOrderByMemberIdAndId(member.getId(), submitCloseEntrustDto.getId());
        if (holdOrderEntity == null) {
            return Result.fail("订单不存在");
        }

        if (holdOrderEntity.getSymbolCntSale() - submitCloseEntrustDto.getSymbolCnt() < 0) {
            return Result.fail("可平张数不足");
        }

        List<ContractEntrustOrderEntity> entrustOrders = contractEntrustOrderDao.selectEntrustOrderByOrderNo(holdOrderEntity.getOrderNo());
        if (CollUtil.isNotEmpty(entrustOrders)) {
            int entrustCnt = 0;
            for (ContractEntrustOrderEntity entrustOrder : entrustOrders) {
                entrustCnt += entrustOrder.getSymbolCnt();
            }


            if (submitCloseEntrustDto.getSymbolCnt() > holdOrderEntity.getSymbolCntSale() - entrustCnt) {
                return Result.fail("当前存在多个委托，无法提交该数量委托");
            }
        }


        // 获取最新价
        BigDecimal newPrice = new BigDecimal(redisUtils.getString(CoinTypeConvert.convertToKey(holdOrderEntity.getSymbol())));

        int closeType;
        if (holdOrderEntity.getOpeningType() == ContractHoldOrderEntity.OPENING_TYPE_MORE) {
            // 开多, 若委托价低于当前价则立即成交
            if (submitCloseEntrustDto.getEntrustPrice().compareTo(newPrice) <= 0) {
                WholeCloseOrderDto wholeCloseOrderDto = new WholeCloseOrderDto();
                wholeCloseOrderDto.setCount(submitCloseEntrustDto.getSymbolCnt());
                wholeCloseOrderDto.setId(submitCloseEntrustDto.getId());
                return contractHoldOrderService.cancelHoldOrder(wholeCloseOrderDto);
            }

            closeType = ContractOrderEntity.ORDER_TYPE_CLOSE_MORE;
        } else {
            // 开空, 若委托价高于当前价则立即成交
            if (submitCloseEntrustDto.getEntrustPrice().compareTo(newPrice) >= 0) {
                WholeCloseOrderDto wholeCloseOrderDto = new WholeCloseOrderDto();
                wholeCloseOrderDto.setCount(submitCloseEntrustDto.getSymbolCnt());
                wholeCloseOrderDto.setId(submitCloseEntrustDto.getId());
                return contractHoldOrderService.cancelHoldOrder(wholeCloseOrderDto);
            }

            closeType = ContractOrderEntity.ORDER_TYPE_CLOSE_LESS;
        }

        BigDecimal lotNumber = cacheSettingUtils.getSymbolSku(holdOrderEntity.getSymbol());
        BigDecimal bondAmount = CalculateUtil.getBondAmount(holdOrderEntity.getOpeningPrice(), lotNumber, submitCloseEntrustDto.getSymbolCnt(), holdOrderEntity.getLeverRatio());

        ContractEntrustOrderEntity entrustOrder = new ContractEntrustOrderEntity();
        entrustOrder.setOrderNo(holdOrderEntity.getOrderNo());
        entrustOrder.setEntrustAmount(bondAmount);
        entrustOrder.setBondAmount(bondAmount);
        entrustOrder.setPositionType(ContractEntrustOrderEntity.POSITION_TYPE_ALL);
        entrustOrder.setMemberId(member.getId());
        entrustOrder.setEntrustPrice(submitCloseEntrustDto.getEntrustPrice());
        entrustOrder.setLeverRatio(holdOrderEntity.getLeverRatio());
        entrustOrder.setSymbolSku(holdOrderEntity.getSymbolSku());
        entrustOrder.setEntrustType(closeType);
        entrustOrder.setSymbolCnt(submitCloseEntrustDto.getSymbolCnt());
        entrustOrder.setSymbol(holdOrderEntity.getSymbol());

        int i = contractEntrustOrderDao.insert(entrustOrder);

        if (i > 0) {
            // 发送委托消息
            if (holdOrderEntity.getOpeningType() == ContractEntrustOrderEntity.ENTRUST_TYPE_OPEN_MORE) {
                OrderModel model = new OrderModel(entrustOrder.getId(), RabbitPriceTypeEnum.ENTRUST_CLOSE_MORE.getValue(), submitCloseEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol(), holdOrderEntity.getMemberId());
                producer.sendPriceOperate(JSONObject.toJSONString(model));
            } else {
                OrderModel model = new OrderModel(entrustOrder.getId(), RabbitPriceTypeEnum.ENTRUST_CLOSE_LESS.getValue(), submitCloseEntrustDto.getEntrustPrice().setScale(8, RoundingMode.HALF_UP).toPlainString(), holdOrderEntity.getSymbol(), holdOrderEntity.getMemberId());
                producer.sendPriceOperate(JSONObject.toJSONString(model));
            }
        }
        return Result.ok("委托成功");
    }
}