package org.springblade.modules.securitypaper.controller;

import com.alibaba.excel.EasyExcel;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import org.apache.commons.codec.Charsets;
import org.springblade.common.excel.RowWriteHandler;
import org.springblade.core.excel.util.ExcelUtil;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.securitypaper.entity.SecurityPaper;
import org.springblade.modules.securitypaper.excel.SecurityPaperExcel;
import org.springblade.modules.securitypaper.excel.SecurityPaperImporter;
import org.springblade.modules.securitypaper.service.SecurityPaperService;
import org.springblade.modules.system.excel.SecurityExcel;
import org.springblade.modules.system.excel.SecurityImporter;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;

/**
 * @author zhongrj
 * @time 2021-12-25
 * @desc 保安员证管理控制层
 */
@RestController
@AllArgsConstructor
@RequestMapping("/securityPaper")
public class SecurityPaperController {

	private final SecurityPaperService securityPaperService;


//	/**
//	 * 自定义分页
//	 * @param query page,size
//	 * @param securityPaper 保安员证管理信息对象
//	 */
//	@GetMapping("/page")
//	public R<IPage<SecurityPaperVo>> page(SecurityPaperVo securityPaper, Query query) {
//		IPage<SecurityPaperVo> pages = securityPaperService.selectSecurityPaperPage(Condition.getPage(query), securityPaper);
//		return R.data(pages);
//	}

	/**
	 * 新增
	 * @param securityPaper 保安员证管理信息对象
	 */
	@PostMapping("/save")
	@ApiOperation(value = "新增", notes = "传入securityPaper")
	public R save(@RequestBody SecurityPaper securityPaper){
		return R.data(securityPaperService.save(securityPaper));
	}


	/**
	 * 修改
	 * @param securityPaper 保安员证管理信息对象
	 */
	@PostMapping("/update")
	public R update(@RequestBody SecurityPaper securityPaper){
		return R.status(securityPaperService.updateById(securityPaper));
	}

	/**
	 * 新增或修改
	 * @param securityPaper 保安员证管理信息对象
	 */
	@PostMapping("/submit")
	public R submit(@RequestBody SecurityPaper securityPaper){
		return R.data(securityPaperService.saveOrUpdate(securityPaper));
	}

	/**
	 * 删除
	 * @param ids 保安员证管理信息ids 数组
	 */
	@PostMapping("/remove")
	public R remove(@ApiParam(value = "主键集合") @RequestParam String ids) {
		return R.status(securityPaperService.removeByIds(Func.toLongList(ids)));
	}

	/**
	 * 详情
	 * @param securityPaper 保安员证管理信息对象
	 */
	@GetMapping("/detail")
	@ApiOperation(value = "详情", notes = "传入securityPaper")
	public R<SecurityPaper> detail(SecurityPaper securityPaper) {
		SecurityPaper detail = securityPaperService.getOne(Condition.getQueryWrapper(securityPaper));
		return R.data(detail);
	}


	/**
	 * 导入保安员证数据
	 */
	@PostMapping("import-security-paper")
	public R importSecurityPaper(MultipartFile file, Integer isCovered, String deptId) {
		SecurityPaperImporter securityImporter = new SecurityPaperImporter(securityPaperService, false,deptId);
		ExcelUtil.save(file, securityImporter, SecurityPaperExcel.class);
		return R.success("操作成功");
	}

	/**
	 * 导出模板
	 */
	@GetMapping("export-template-security-paper")
	@ApiOperation(value = "导出模板")
	public void exportSecurity(HttpServletResponse response) throws IOException {
		List<SecurityPaperExcel> list = new ArrayList<>();
		SecurityPaperExcel securityExcel = new SecurityPaperExcel();
		securityExcel.setDeptId("xxx保安公司");
		securityExcel.setRealName("张三");
		securityExcel.setPhone("12345678901");
		securityExcel.setCardid("360XXX19XXXXXX****");
		securityExcel.setRegistered("江西省南昌市******");
		list.add(securityExcel);

		SecurityPaperExcel securityExcel1 = new SecurityPaperExcel();
		securityExcel1.setDeptId("xxx保安公司");
		securityExcel1.setRealName("李四");
		securityExcel1.setPhone("12345678901");
		securityExcel1.setCardid("360XXX19XXXXXX****");
		securityExcel1.setRegistered("江西省南昌市******");
		list.add(securityExcel1);
		String fileName = null;
		try {
			response.setContentType("application/vnd.ms-excel");
			response.setCharacterEncoding(org.apache.commons.codec.Charsets.UTF_8.name());
			fileName = URLEncoder.encode("保安员导入数据模板-制证" + DateUtil.time(), Charsets.UTF_8.name());
			response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
			//修改单元格格式为文本格式
			EasyExcel.write(response.getOutputStream(), SecurityPaperExcel.class).sheet("保安员数据表").registerWriteHandler(new RowWriteHandler()).doWrite(list);
		} catch (Throwable var6) {
			throw var6;
		}
	}

}