package com.yami.trading.admin.controller.etf;

import cn.hutool.core.lang.Tuple;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.yami.trading.bean.data.domain.Depth;
import com.yami.trading.bean.data.domain.Kline;
import com.yami.trading.bean.data.domain.Realtime;
import com.yami.trading.bean.data.domain.RobotModel;
import com.yami.trading.bean.etf.domain.EtfKLine;
import com.yami.trading.bean.etf.domain.EtfSecKLine;
import com.yami.trading.bean.etf.mapstruct.EtfSecKLineWrapper;
import com.yami.trading.bean.item.domain.Item;
import com.yami.trading.bean.item.dto.ItemDTO;
import com.yami.trading.bean.item.query.ItemQuery;
import com.yami.trading.common.domain.Result;

import javax.validation.Valid;

import com.google.common.collect.Lists;
import com.yami.trading.common.query.QueryWrapperGenerator;

import com.yami.trading.dao.etf.mapper.EtfSecKLineMapper;
import com.yami.trading.service.etf.EtfSecKLineService;
import com.yami.trading.service.etf.MarketService;
import com.yami.trading.service.item.ItemService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.http.ResponseEntity;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.yami.trading.bean.etf.domain.KlineConfig;

import com.yami.trading.bean.etf.dto.KlineConfigDTO;
import com.yami.trading.bean.etf.mapstruct.KlineConfigWrapper;
import com.yami.trading.service.etf.KlineConfigService;
import com.yami.trading.bean.etf.query.KlineConfigQuery;

import java.math.BigDecimal;
import java.util.*;


/**
 * etfK线图配置表Controller
 *
 * @author lucas
 * @version 2023-05-03
 */

@Api(tags = "债劵ETF_币对管理")
@RestController
@RequestMapping(value = "/etf/klineConfig")
public class KlineConfigController {

    @Autowired
    private KlineConfigService klineConfigService;

    @Autowired
    private KlineConfigWrapper klineConfigWrapper;

    @Autowired
    private EtfSecKLineWrapper etfSecKLineWrapper;

    @Autowired
    private ItemService itemService;

    @Autowired
    MarketService marketService;

    @Autowired
    EtfSecKLineService etfSecKLineService;

    /**
     * etfK线图配置表列表数据
     */
    @ApiOperation(value = "查询etfK线图配置表列表数据")
    @GetMapping("list")
    public Result<IPage<ItemDTO>> list(ItemQuery itemQuery, Page<ItemDTO> page) throws Exception {
        QueryWrapper queryWrapper = QueryWrapperGenerator.buildQueryCondition(itemQuery, ItemQuery.class);
        IPage<ItemDTO> result = itemService.findPage(page, queryWrapper);
        return Result.ok(result);
    }

    /**
     * 添加交易对
     */
    @ApiOperation(value = "添加交易对")
    @PostMapping("addItem")
    public Result<String> addItem(@Valid @RequestBody Item item) throws Exception {
        //新增或编辑表单保存
        itemService.saveOrUpdate(item);
        return Result.ok("添加交易对成功");
    }


    /**
     * 根据Id获取etfK线图配置表数据
     */
    @ApiOperation(value = "根据Id获取etfK线图配置表数据")
    @GetMapping("queryById")
    public Result<KlineConfigDTO> queryById(String id) {
        return Result.ok(klineConfigWrapper.toDTO(klineConfigService.getById(id)));
    }

    /**
     * 根据Id获取etfK线图配置表数据
     */
    @ApiOperation(value = "查询etfK线图配置列表")
    @GetMapping("pageList")
    public Result<IPage<KlineConfigDTO>> pageList(KlineConfigQuery query, Page<KlineConfig> page) throws Exception {
        IPage<KlineConfigDTO> list;
        if (Objects.isNull(query.getSymbol())) {
            list = klineConfigService.page(page, new LambdaQueryWrapper<KlineConfig>().orderBy(true, false, KlineConfig::getOpenTimeTs)).convert(klineConfigWrapper::toDTO);
        } else {
            list = klineConfigService.page(page, new LambdaQueryWrapper<>(new KlineConfig()).eq(KlineConfig::getSymbol, query.getSymbol()).orderBy(true, false, KlineConfig::getOpenTimeTs)).convert(klineConfigWrapper::toDTO);
        }
        list.getRecords().forEach(r -> {
            Item bySymbol = itemService.findBySymbol(r.getSymbol());
            if (bySymbol != null) {
                r.setSymbolName(bySymbol.getName());
            }
        });
        return Result.ok(list);
    }

    /**
     * 保存etfK线图配置表
     */
    @ApiOperation(value = "保存etfK线图配置表")
    @PostMapping("save")
    public Result<String> save(@RequestBody KlineConfigDTO klineConfigDTO) {
        Tuple startAndEnd = klineConfigService.getStartAndEnd(klineConfigDTO.getOpenTimeTs());
        klineConfigDTO.setOpenTimeTs(startAndEnd.get(0));
        klineConfigDTO.setCloseTimeTs(startAndEnd.get(1));

        if (etfSecKLineService.count(new LambdaQueryWrapper<>(new EtfSecKLine()).eq(EtfSecKLine::getSymbol, klineConfigDTO.getSymbol()).between(EtfSecKLine::getTs, klineConfigDTO.getOpenTimeTs(), klineConfigDTO.getCloseTimeTs())) > 0) {
            return Result.failed("今天数据已存在，无法添加");
        }

        //新增或编辑表单保存
        KlineConfig klineConfig = klineConfigWrapper.toEntity(klineConfigDTO);
        klineConfig.setCreateTime(new Date());

        List<Kline> klines = klineConfigService.generateSecKLine(klineConfigDTO);
        if (klines != null) {
            List<EtfSecKLine> etfSecKLines = etfSecKLineWrapper.toDTO(klines);
            etfSecKLineService.saveBatch(etfSecKLines, 10000);
        }

        klineConfigService.saveOrUpdate(klineConfig);
        marketService.clear();
        return Result.ok("保存etfK线图配置表成功");
    }


    /**
     * 删除etfK线图配置表
     */
    @ApiOperation(value = "删除etfK线图配置表")
    @DeleteMapping("delete")
    public Result<String> delete(String ids) {
        List<String> list = Arrays.asList(ids.split(","));

        klineConfigService.deleteByRobotModelId(list);
        return Result.ok("删除etfK线图配置表成功");
    }


    /**
     * 删除etfK线图配置表
     */
    @ApiOperation(value = "初始化ETF当天K线图,分钟级别")
    @PostMapping("init")
    public Result<EtfKLine> init(@RequestBody KlineConfigDTO klineConfigDTO) {
        Item symbol = itemService.findBySymbol(klineConfigDTO.getSymbol());
        EtfKLine model = klineConfigService.queryKLine(klineConfigDTO);
        List<Kline> klineList = model.getKlineList();
        List<Kline> result = new ArrayList<>();
        // 抽象成15分钟图，方便前端显示
        List<List<Kline>> partition = Lists.partition(klineList, 15);
        for (List<Kline> list1Min : partition) {
            Double high = list1Min.get(0).getHigh().doubleValue();
            Double low = list1Min.get(0).getLow().doubleValue();
            for (Kline kline : list1Min) {
                if (high <= kline.getHigh().doubleValue()) {
                    high = kline.getHigh().doubleValue();
                }
                if (low >= kline.getLow().doubleValue()) {
                    low = kline.getLow().doubleValue();
                }
            }
            int lastIndex = list1Min.size() - 1;
            Kline kline = new Kline();
            kline.setSymbol(klineConfigDTO.getSymbol());
            kline.setTs(list1Min.get(lastIndex).getTs());
            kline.setSymbol(klineConfigDTO.getSymbol());
            kline.setOpen(list1Min.get(0).getOpen().setScale(symbol.getDecimals(), BigDecimal.ROUND_DOWN));
            kline.setHigh(new BigDecimal(high).setScale(symbol.getDecimals(), BigDecimal.ROUND_DOWN));
            kline.setLow(new BigDecimal(low).setScale(symbol.getDecimals(), BigDecimal.ROUND_DOWN));
            kline.setClose(list1Min.get(lastIndex).getClose().setScale(symbol.getDecimals(), BigDecimal.ROUND_DOWN));
            kline.setPeriod(Kline.PERIOD_15MIN);
            // 格式化小数点位
            // klineOneTop.formatPoint(kline);
            BigDecimal sumAmount = klineList.stream()
                    .map(Kline::getAmount)
                    .filter(Objects::nonNull)
                    .reduce(BigDecimal.ZERO, BigDecimal::add);
            BigDecimal sumVolume = klineList.stream()
                    .map(Kline::getVolume)
                    .filter(Objects::nonNull)
                    .reduce(BigDecimal.ZERO, BigDecimal::add);
            kline.setAmount(sumAmount);
            kline.setVolume(sumVolume);
            result.add(kline);
        }
        model.setKlineList(result);
        return Result.ok(model);
    }

    /**
     * 根据Id获取etfK线图配置表数据
     */
    @ApiOperation(value = "获取每秒k线图")
    @GetMapping("secKline")
    public Result<Realtime> secKline(String symbol) {
        return Result.ok(marketService.queryRealtime(symbol));
    }


    /**
     * 根据Id获取etfK线图配置表数据
     */
    @ApiOperation(value = "获取每秒深度")
    @GetMapping("secDepth")
    public Result<Depth> secDepth(String symbol) {
        return Result.ok(marketService.queryDepth(symbol));
    }


    /**
     * 根据Id获取etfK线图配置表数据
     */
    @ApiOperation(value = "加倍深度")
    @GetMapping("accelerate")
    public Result accelerate(@RequestParam(value = "symbol") String symbol, @RequestParam(value = "enlarge") Double
            enlarge, @RequestParam(value = "enable") Boolean enable) {
        Map<String, Double> accelerate = marketService.getAccelerate();
        if (enable) {
            accelerate.put(symbol, enlarge);
        } else {
            accelerate.remove(symbol);
        }
        return Result.succeed();
    }


}