/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package org.springblade.modules.energy.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import com.google.common.collect.Lists;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.dj.entity.Dj;
import org.springblade.modules.dj.service.IDjService;
import org.springblade.modules.dj.vo.DjVO;
import org.springblade.modules.dj.wrapper.DjWrapper;
import org.springblade.modules.energy.entity.Energy;
import org.springblade.modules.energy.entity.EnergyTree;
import org.springblade.modules.energy.service.IEnergyService;
import org.springblade.modules.energy.service.IEnergyTreeService;
import org.springblade.modules.energy.vo.EnergyVO;
import org.springblade.modules.energy.wrapper.EnergyWrapper;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 *  控制器
 *
 * @author BladeX
 * @since 2020-07-19
 */
@RestController
@AllArgsConstructor
@RequestMapping("/energy")
@Api(value = "", tags = "接口")
public class EnergyController extends BladeController {

	private final IEnergyService energyService;
	private final IEnergyTreeService energyTreeService;

	/**
	 * 详情
	 */
	@GetMapping("/detail")
	@ApiOperationSupport(order = 1)
	@ApiOperation(value = "详情", notes = "传入dj")
	public R<EnergyVO> detail(Energy energy) {
		Energy detail = energyService.getOne(Condition.getQueryWrapper(energy));
		return R.data(EnergyWrapper.build().entityVO(detail));
	}

	/**
	 * 分页
	 */
	@GetMapping("/list")
	@ApiOperationSupport(order = 2)
	@ApiOperation(value = "分页", notes = "传入dj")
	public R<IPage<EnergyVO>> list(Energy energy, Query query) {
		IPage<Energy> pages = energyService.page(Condition.getPage(query), Condition.getQueryWrapper(energy));
		return R.data(EnergyWrapper.build().pageVO(pages));
	}


	/**
	 * 新增
	 */
	@PostMapping("/save")
	@ApiOperationSupport(order = 4)
	@ApiOperation(value = "新增", notes = "传入dj")
	public R save(@Valid @RequestBody Energy energy) {
		List<String> list = energyService.selectEnergyTree(energy.getIdentification());
		if (list==null){
			EnergyTree energyTree= new EnergyTree();
			energyTree.setIdentification(energy.getIdentification());
			energyTreeService.save(energyTree);
		}
		return R.status(energyService.save(energy));
	}

	/**
	 * 修改
	 */
	@PostMapping("/update")
	@ApiOperationSupport(order = 5)
	@ApiOperation(value = "修改", notes = "传入dj")
	public R update(@Valid @RequestBody Energy energy) {
		return R.status(energyService.updateById(energy));
	}

	/**
	 * 新增或修改
	 */
	@PostMapping("/submit")
	@ApiOperationSupport(order = 6)
	@ApiOperation(value = "新增或修改", notes = "传入dj")
	public R submit(@Valid @RequestBody Energy energy) {
		return R.status(energyService.saveOrUpdate(energy));
	}


	/**
	 * 删除
	 */
	@PostMapping("/remove")
	@ApiOperationSupport(order = 8)
	@ApiOperation(value = "删除", notes = "传入ids")
	public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) {
		return R.status(energyService.removeByIds(Func.toLongList(ids)));
	}


	/**
	 * 能量统计
	 * @return
	 */
	@PostMapping("/selectTreeNum")
	public R selectTreeNum() {
		Date date=new Date();
		DateFormat format=new SimpleDateFormat("yyyy-MM-dd");
		String time=format.format(date);
		List<String> list = energyService.selectTreeNum(time);
		List<List<String>> lists = splitList(list, 3);
		return R.data(lists);
	}

	/**
	 * 拾取能量
	 * @param id
	 * @return
	 */
	@PostMapping("/updateEnergyId")
	public R updateEnergyId(String id,String identification,int num) {

		energyService.updateEnergyId(id,identification);
		//能量树
		String nums = energyTreeService.selectNum(identification);
		Integer a=Integer.valueOf(nums);
		Integer n=Integer.valueOf(num);
		int i1 = n + a;
		String i = String.valueOf(i1);
		energyTreeService.updateEnergyTreeId(i,identification);
		return R.success("拾取成功");
	}


	private List<List<String>> splitList(List<String> list , int groupSize){
		int length = list.size();
		// 计算可以分成多少组
		int num = ( length + groupSize - 1 )/groupSize ; // TODO
		List<List<String>> newList = new ArrayList<>(num);
		for (int i = 0; i < num; i++) {
			// 开始位置
			int fromIndex = i * groupSize;
			// 结束位置
			int toIndex = (i+1) * groupSize < length ? ( i+1 ) * groupSize : length ;
			newList.add(list.subList(fromIndex,toIndex)) ;
		}
		return  newList ;
	}

	/**
	 * 每天23点自动拾取能量
	 *
	 */
	@PostMapping("/pickup")
	public void pickup(String ids,String time,String identification) {
		String[] split = ids.split(",");
		String strArrays = "";
		for (int i = 0; i < split.length; i++) {
			strArrays += "'" + split[i] + "',";
		}
		String code = strArrays.substring(0, strArrays.length() - 1);
		energyService.updateEnergyType(code,time);
		//能量树
		String nums = energyTreeService.selectNum(identification);
		Integer a=Integer.valueOf(nums);
		int length = split.length;
		int i = a + length;
		String num=String.valueOf(i);
		energyTreeService.updateEnergyTreeId(num,identification);
	}

}