package org.springblade.modules.recovery.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
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.farm.entity.Farm;
import org.springblade.modules.farm.entity.FarmingRecord;
import org.springblade.modules.farm.service.FarmingRecordService;
import org.springblade.modules.farmplant.entity.FarmProductStock;
import org.springblade.modules.farmplant.entity.Strain;
import org.springblade.modules.farmplant.service.FarmProductStockService;
import org.springblade.modules.farmplant.service.StrainService;
import org.springblade.modules.recovery.entity.Recovery;
import org.springblade.modules.recovery.service.RecoveryService;
import org.springblade.modules.recovery.vo.RecoveryVO;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;

import javax.validation.Valid;
import java.math.BigDecimal;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * 农事记录控制器
 *
 * @author zhongrj
 * @since 2022-05-18
 */
@RestController
@AllArgsConstructor
@RequestMapping("/recovery")
public class RecoveryController extends BladeController {

	private final RecoveryService recoveryService;

	private final FarmingRecordService farmingRecordService;

	private final FarmProductStockService farmProductStockService;

	private final StrainService strainService;

	/**
	 * 详情
	 */
	@GetMapping("/detail")
	@ApiOperationSupport(order = 1)
	@ApiOperation(value = "详情", notes = "传入recovery")
	public R<Recovery> detail(Recovery recovery) {
		Recovery detail = recoveryService.getOne(Condition.getQueryWrapper(recovery));
		return R.data(detail);
	}

	/**
	 * 分页
	 */
	@GetMapping("/list")
	@ApiOperationSupport(order = 2)
	@ApiOperation(value = "分页", notes = "传入recovery")
	public R<IPage<Recovery>> list(Recovery recovery, Query query) {
		IPage<Recovery> pages = recoveryService.page(Condition.getPage(query), Condition.getQueryWrapper(recovery));
		return R.data(pages);
	}

	/**
	 * 自定义分页
	 */
	@GetMapping("/page")
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "分页", notes = "传入recovery")
	public R<IPage<RecoveryVO>> page(RecoveryVO recovery, Query query) {
		IPage<RecoveryVO> pages = recoveryService.selectRecoveryPage(Condition.getPage(query), recovery);
		return R.data(pages);
	}

	/**
	 * 新增
	 */
	@PostMapping("/save")
	@ApiOperationSupport(order = 4)
	@ApiOperation(value = "新增", notes = "传入recovery")
	@Transactional(rollbackFor = Exception.class)
	public R save(@Valid @RequestBody Recovery recovery) {
		recovery.setCreateTime(new Date());
		//新增
		boolean save = recoveryService.save(recovery);
		if (save) {
			//同时生成农事记录和库存记录
			FarmingRecord record = new FarmingRecord();
			record.setCreateTime(new Date());
			record.setJobWay(recovery.getJobWay());
			record.setTime(recovery.getTime());
			//采收
			record.setType("12");
			record.setLandId(recovery.getLandId());
			record.setDeptId(recovery.getDeptId());
			record.setTenantId(recovery.getTenantId());
			record.setStrainId(recovery.getStrainId());
			record.setOperator(recovery.getOperator());
			Strain strain = strainService.getById(recovery.getStrainId());
			record.setContent("品种: " + strain.getStrainName() + ", 重量：" + recovery.getWeight());
			//新增
			farmingRecordService.save(record);

			//库存
			//先查询是否有该农产品的库存
			FarmProductStock stock = new FarmProductStock();
//			stock.setFarmPlantId(recovery.getFarmPlantId());
			stock.setStrainId(recovery.getStrainId());
			FarmProductStock stock1 = farmProductStockService.getOne(new QueryWrapper<>(stock));
			if (null != stock1) {
				//更新库存
				//计算库存
				double old = Double.parseDouble(stock1.getWeight());
				double now = Double.parseDouble(recovery.getWeight());
				double addNum = add(old, now);
				stock1.setWeight(String.valueOf(addNum));

				//总采收
				double olds = Double.parseDouble(stock1.getRecovery());
				double addNums = add(olds, now);
				stock1.setRecovery(String.valueOf(addNums));
				//更新
				farmProductStockService.updateById(stock1);
			} else {
				//新增库存
				FarmProductStock productStock = new FarmProductStock();
				productStock.setCreateTime(new Date());
				productStock.setLeaves(recovery.getLeaves());
				productStock.setOperator(recovery.getOperator());
				productStock.setTime(recovery.getTime());
				productStock.setStrainId(recovery.getStrainId());
				productStock.setWeight(recovery.getWeight());
				productStock.setRecovery(recovery.getWeight());
				productStock.setFarmPlantId(recovery.getFarmPlantId());
				productStock.setTenantId(recovery.getTenantId());
				productStock.setDeptId(recovery.getDeptId());
				//新增操作
				farmProductStockService.save(productStock);
			}
		}
		return R.status(save);
	}


	/**
	 * double 相加
	 *
	 * @param d1
	 * @param d2
	 * @return
	 */
	private double add(double d1, double d2) {
		// 进行加法运算
		BigDecimal b1 = new BigDecimal(d1);
		BigDecimal b2 = new BigDecimal(d2);
		return b1.add(b2).doubleValue();
	}

	/**
	 * 修改
	 */
	@PostMapping("/update")
	@ApiOperationSupport(order = 5)
	@ApiOperation(value = "修改", notes = "传入recovery")
	public R update(@Valid @RequestBody Recovery recovery) {
		//更新并返回
		return R.status(recoveryService.updateById(recovery));
	}

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


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


	/**
	 * 小程序采收详情列表自定义分页
	 */
	@GetMapping("/pagec")
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "分页", notes = "传入recovery")
	public R<IPage<RecoveryVO>> pagec(RecoveryVO recovery, Query query) {
		IPage<RecoveryVO> pages = recoveryService.selectRecoveryPage(Condition.getPage(query), recovery);
		for (int i = 0; i < pages.getRecords().size(); i++) {
			String jobWay = pages.getRecords().get(i).getJobWay();
			if (jobWay.equals("0")) {
				pages.getRecords().get(i).setLx("人工");
			} else {
				pages.getRecords().get(i).setLx("机械");
			}

		}
		return R.data(pages);
	}
}