package org.springblade.modules.equipage.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import org.springblade.common.utils.arg;
import org.springblade.core.excel.util.ExcelUtil;
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.equipage.entity.Equipage;
import org.springblade.modules.equipage.excel.EquipageExcel;
import org.springblade.modules.equipage.excel.EquipageImporter;
import org.springblade.modules.equipage.service.EquipageService;
import org.springblade.modules.equipage.vo.EquipageVo;
import org.springblade.modules.exam.excel.ExamScoreExcel;
import org.springblade.modules.exam.excel.ExamScoreImporter;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * @author zhongrj
 * @time 2021-07-06
 * @desc 装备管理控制层
 */
@RestController
@AllArgsConstructor
@RequestMapping("/equipage")
public class EquipageController {

	private final EquipageService equipageService;

	/**
	 * 自定义分页
	 * @param query page,size
	 * @param equipage 装备信息对象
	 */
	@GetMapping("/page")
	public R<IPage<EquipageVo>> page(EquipageVo equipage, Query query) {
		IPage<EquipageVo> pages = equipageService.selectEquipagePage(Condition.getPage(query), equipage);
		return R.data(pages);
	}

	/**
	 * 新增
	 * @param equipage 装备信息对象
	 */
	@PostMapping("/save")
	@ApiOperation(value = "新增", notes = "传入equipage")
	public R save(@RequestBody Equipage equipage) {
		return R.status(equipageService.save(equipage));
	}

	/**
	 * 修改
	 * @param equipage 装备信息对象
	 */
	@PostMapping("/update")
	public R update(@RequestBody Equipage equipage) {
		return R.status(equipageService.updateById(equipage));
	}

	/**
	 * 新增或修改
	 * @param equipage 装备信息对象
	 */
	@PostMapping("/submit")
	public R submit(@RequestBody Equipage equipage) throws Exception {
//		if (equipage.getId()==null){
//			arg arg = new arg();
//			arg.test01(arg.url+"/equipage/save",equipage);
//		}
		return R.status(equipageService.saveOrUpdate(equipage));
	}

	/**
	 * 删除
	 * @param ids 装备信息ids 数组
	 */
	@PostMapping("/remove")
	public R remove(@ApiParam(value = "主键集合") @RequestParam String ids) {
		//arg.sendPostRemoveByIds(arg.url+"/equipage/remove",ids);
		return R.status(equipageService.removeByIds(Func.toLongList(ids)));
	}

	/**
	 * 详情
	 * @param equipage 装备信息对象
	 */
	@GetMapping("/detail")
	@ApiOperation(value = "详情", notes = "传入equipage")
	public R<EquipageVo> detail(Equipage equipage) {
		EquipageVo detail = equipageService.selectEquipageInfo(equipage);
		return R.data(detail);
	}

	/**
	 * 导入装备数据
	 * @param isCovered 1 覆盖  0不覆盖
	 * @return
	 */
	@PostMapping("import-equipage")
	@ApiOperation(value = "导入装备数据", notes = "传入excel")
	public R importExamScore(MultipartFile file, Integer isCovered) {
		EquipageImporter equipageImporter = new EquipageImporter(equipageService, false);
		ExcelUtil.save(file, equipageImporter, EquipageExcel.class);
		return R.success("操作成功");
	}

	/**
	 * 导出模板
	 */
	@GetMapping("export-template")
	@ApiOperation(value = "导出模板")
	public void exportUser(HttpServletResponse response) {
		List<EquipageExcel> list = new ArrayList<>();
		ExcelUtil.export(response, "装备导入数据模板", "装备导入数据表", list, EquipageExcel.class);
	}

}