/*
 *      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 exam,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  exam, 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.exam.controller;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.alibaba.excel.write.metadata.fill.FillConfig;
import com.alibaba.excel.write.metadata.fill.FillWrapper;
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.apache.commons.codec.Charsets;
import org.springblade.core.boot.ctrl.BladeController;
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.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.FTP.FtpUtil;
import org.springblade.modules.apply.entity.Apply;
import org.springblade.modules.exam.entity.ExamPaper;
import org.springblade.modules.exam.entity.ExamSubjectChoices;
import org.springblade.modules.exam.excel.ExportExamScoreExcel;
import org.springblade.modules.exam.service.ExamPaperService;
import org.springblade.modules.exam.vo.*;
import org.springblade.modules.exam.wrapper.ExamPaperWrapper;
import org.springblade.modules.system.entity.User;
import org.springblade.modules.system.service.IUserService;
import org.springblade.modules.system.service.MyAsyncService;
import org.springblade.modules.training.entity.TrainExam;
import org.springblade.modules.training.entity.TrainingRegistration;
import org.springblade.modules.training.service.TrainingRegistrationService;
import org.springframework.core.io.ClassPathResource;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 控制器
 *
 * @author Chill
 */
@RestController
@RequestMapping("/exampaper")
@AllArgsConstructor
public class ExamPaperController extends BladeController {

	private final ExamPaperService examPaperService;


	private final TrainingRegistrationService trainingRegistrationService;


	private final IUserService userService;
	private final MyAsyncService myAsyncService;


	/**
	 * 详情
	 */
	@GetMapping("/detail")
	@ApiOperationSupport(order = 1)
	@ApiOperation(value = "详情", notes = "传入exam")
	public R<ExamPaperVO> detail(ExamPaper exam) {
		ExamPaper detail = examPaperService.getOne(Condition.getQueryWrapper(exam));
		return R.data(ExamPaperWrapper.build().entityVO(detail));
	}

	/**
	 * 分页
	 */
	@GetMapping("/list")
	@ApiOperationSupport(order = 2)
	@ApiOperation(value = "分页", notes = "传入exam")
	public R<IPage<ExamPaperVO>> list(@ApiIgnore @RequestParam Map<String, Object> exam, Query query) {
		IPage<ExamPaper> pages = examPaperService.page(Condition.getPage(query), Condition.getQueryWrapper(exam, ExamPaper.class));
		return R.data(ExamPaperWrapper.build().pageVO(pages));
	}

	/**
	 * 分页
	 */
	@GetMapping("/pages")
	public R<IPage<ExamPaperVO>> pages(ExamPaperVO exam, Query query) {
		IPage<ExamPaperVO> pages = examPaperService.selectExamPaperPages(Condition.getPage(query), exam);
		return R.data(pages);
	}


	/**
	 * 自定义查询
	 *
	 * @param query       分页条件
	 * @param examPaperVO 查询参数
	 * @return
	 */
	@GetMapping("/page")
	public R<IPage<ExamPaperVO>> page(ExamPaperVO examPaperVO, Query query) {
		IPage<ExamPaperVO> pages = examPaperService.selectExamPaperPage(Condition.getPage(query), examPaperVO);
		return R.data(pages);
	}


	/**
	 * 新增
	 */
	@PostMapping("/save")
	@ApiOperationSupport(order = 4)
	@ApiOperation(value = "新增", notes = "传入exam")
	public R save(@RequestBody ExamPaper examPaper) {
		boolean save = examPaperService.save(examPaper);
		String format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(examPaper.getStartTime());
		if (save) {
			String s = "insert into ksxt_exam(id,exam_name,exam_type,start_time) " +
				"values(" + "'" + examPaper.getId() + "'" + "," + "'" + examPaper.getExamName() +
				"'" + "," + "'" + examPaper.getExamType() +
				"'" + "," + "'" + format + "'" + ")";
			//FtpUtil.sqlFileUpload(s);
			myAsyncService.dataSync(s);
		}
		return R.status(save);
	}

	/**
	 * 修改
	 */
	@PostMapping("/update")
	@ApiOperationSupport(order = 5)
	@ApiOperation(value = "修改", notes = "传入exam")
	public R update(@RequestBody ExamPaper exam) {
		boolean b = examPaperService.updateById(exam);
		if (b) {
			String s1 =
				"update ksxt_exam set exam_name = " + "'" + exam.getExamName() + "'" +
					",start_time = " + "'" + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(exam.getStartTime()) + "'" +
					" " + "where id = " + "'" + exam.getId() + "'";
			//FtpUtil.sqlFileUpload(s1);
			myAsyncService.dataSync(s1);
		}
		return R.status(b);
	}

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

	/**
	 * 删除
	 */
	@PostMapping("/remove")
	@ApiOperationSupport(order = 7)
	@ApiOperation(value = "逻辑删除", notes = "传入exam")
	public R remove(@ApiParam(value = "主键集合") @RequestParam String ids) {
		//内网删除
		List<Long> list = Func.toLongList(ids);
		list.forEach(id -> {
			//内网同步
			String s1 = "delete from ksxt_exam where id = " + "'" + id + "'";
			//FtpUtil.sqlFileUpload(s1);
			myAsyncService.dataSync(s1);
		});
		boolean temp = examPaperService.removeByIds(Func.toLongList(ids));
		return R.status(temp);
	}

	/**
	 * 考试题目绑定
	 */
	@GetMapping("/setPagerSubject")
	@ApiOperationSupport(order = 8)
	@ApiOperation(value = "考试题目绑定", notes = "传入exam")
	public R setPagerSubject(@ApiParam(value = "主键集合") ExamPaper exam) {
		return R.status(true);
	}

	/**
	 * 查询考试题目
	 */
	@GetMapping("/queryPagerSubject")
	@ApiOperationSupport(order = 8)
	@ApiOperation(value = "考试题目绑定", notes = "传入exam")
	public R<ExamPaperSubjectVO> queryPagerSubject(@ApiParam(value = "主键集合") ExamPaper exam) {
		ExamPaperSubjectVO subject = examPaperService.queryPagerSubject(exam);
		return R.data(subject);
	}

	/**
	 * 随机生成考试题目
	 */
	@GetMapping("/queryRandomSubject")
	@ApiOperationSupport(order = 8)
	@ApiOperation(value = "考试题目绑定", notes = "传入exam")
	public R<List> queryRandomSubject(@ApiParam(value = "主键集合") ExamPaperVO exam) {
		List<ExamSubjectChoicesVO> subject = examPaperService.queryRandomSubject(exam);
		return R.data(subject);
	}


	/**
	 * 返回当前时间
	 */
	@GetMapping("/getdate")
	@ApiOperationSupport(order = 9)
	public R getdate() {
		Date date = new Date();
		String strDateFormat = "yyyy-MM-dd HH:mm:ss";
		SimpleDateFormat sdf = new SimpleDateFormat(strDateFormat);
		return R.data(sdf.format(date));
	}

	/**
	 * 自定义树
	 *
	 * @param examPaper
	 * @return
	 */
	@GetMapping("/page-tree")
	public R pageTree(ExamPaper examPaper) {
		List<ExamPaper> pages = examPaperService.selectExamPaperPageTree(examPaper);
		return R.data(pages);
	}

	/**
	 * 解除绑定
	 */
	@GetMapping("/UnbindSubject")
	public R UnbindSubject(String paperid, String subjectid) {
		return R.status(examPaperService.UnbindSubject(paperid, subjectid));
	}


	/**
	 * 查询考生考试信息
	 *
	 * @param examPaper 用户id
	 * @return
	 */
	@PostMapping("/getExamDetail")
	public R getExamDetail(@RequestBody ExamPaperVO examPaper) {
		System.out.println("examPaper = " + examPaper);
		return R.data(examPaperService.getExamDetail(examPaper));
	}

	/**
	 * 考试名单确认
	 *
	 * @param
	 * @return
	 */
	@PostMapping("/applyConfirm")
	public R applyConfirm(@RequestBody ExamPaper exam) {
		exam.setConfirmTime(new Date());
		boolean status = examPaperService.updateById(exam);
		if (status) {
			//内网数据同步
			String s1 =
				"update ksxt_exam set confirm_user = " + "'" + exam.getConfirmUser() + "'"
					+ ",confirm_time = " + "'" + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(exam.getConfirmTime()) + "'"
					+ " " + "where id = " + "'" + exam.getId() + "'";
			//FtpUtil.sqlFileUpload(s1);
			myAsyncService.dataSync(s1);
		}
		//返回数据
		return R.data(status);
	}


	/**
	 * 审核修改
	 *
	 * @param examPaper 报名培训申请信息对象
	 */
	@PostMapping("/update-audit")
	public R updateAudit(@RequestBody ExamPaper examPaper) {
		//查询之前的审核状态，如果审核状态没有发生改变，则不会去修改审核信息
		ExamPaper paper = examPaperService.getById(examPaper.getId());
		boolean status = false;
		if (null != paper.getAuditStatus()) {
			//审核状态发生改变
//			if (!paper.getAuditStatus().equals(examPaper.getAuditStatus())){
			examPaper.setAuditTime(new Date());
			//修改考试审核状态
			status = examPaperService.updateById(examPaper);
			if (status) {
				//数据同步
				String s1 =
					"update ksxt_exam set audit_status = " + "'" + examPaper.getAuditStatus() + "'"
						+ ",audit_time = " + "'" + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(examPaper.getAuditTime()) + "'"
						+ " " + "where id = " + "'" + examPaper.getId() + "'";
				//FtpUtil.sqlFileUpload(s1);
				myAsyncService.dataSync(s1);
			}
			//查询当前考试已报名的人员培训报名数据集合
			List<TrainingRegistration> trainingRegistrations = examPaperService.getTrainList(examPaper.getId());
//		System.out.println("trainingRegistrations = " + trainingRegistrations);
			if (trainingRegistrations.size() > 0) {
				trainingRegistrations.forEach(trainingRegistration -> {
					//审核通过，则生成准考证号
					if (examPaper.getAuditStatus() == 1) {
						trainingRegistration.setAuditStatus(1);
						//查询当前用户审核状态，如果异常，则审核不通过
						User user = userService.getById(trainingRegistration.getUserId());
						if (null != user) {
							if (null != user.getExaminationType() && !user.getExaminationType().equals("")) {
								if (user.getExaminationType().equals("1")) {
									//审核不通过
									trainingRegistration.setAuditStatus(2);
								}
							}
						}
						if (null == trainingRegistration.getCandidateNo() || trainingRegistration.getCandidateNo().equals("")) {
							//生成准考证号
							trainingRegistration.setCandidateNo(getCandidateNo(examPaper));
						}
					}
					//审核不通过
					if (examPaper.getAuditStatus() == 2) {
						trainingRegistration.setAuditStatus(2);
					}
					//待审核
					if (examPaper.getAuditStatus() == 3) {
						trainingRegistration.setAuditStatus(3);
					}
					//修改培训报名信息
					boolean b = trainingRegistrationService.updateById(trainingRegistration);
					if (b) {
						//数据同步
						String s2 =
							"update sys_training_registration set candidate_no = " + "'" + trainingRegistration.getCandidateNo() + "'"
								+ ",audit_status = " + "'" + trainingRegistration.getAuditStatus() + "'"
								+ " " + "where id = " + "'" + trainingRegistration.getId() + "'";
						//FtpUtil.sqlFileUpload(s2);
						myAsyncService.dataSync(s2);
					}
				});
			}
			return R.status(status);
//			}
		}
		//返回
		return R.status(status);
	}

	/**
	 * 生成准考证号码
	 *
	 * @param examPaper 考试报名信息对象
	 */
	private String getCandidateNo(ExamPaper examPaper) {
		//获取考试信息
		if (null != examPaper.getStartTime()) {
			String format = new SimpleDateFormat("yyyy-MM-dd").format(examPaper.getStartTime());
			String year = format.substring(2, 4);
			String quarter = null;
			String months = null;
			int month = Integer.parseInt(format.substring(5, 7));
			int day = Integer.parseInt(format.substring(8, 10));
			String days = null;
			if (month > 0 && month <= 3) {
				quarter = "C";
			}
			if (month > 3 && month <= 6) {
				quarter = "X";
			}
			if (month > 6 && month <= 9) {
				quarter = "Q";
			}
			if (month > 9 && month <= 12) {
				quarter = "D";
			}
			if (month <= 9) {
				months = "0" + month;
			} else {
				months = String.valueOf(month);
			}
			if (day <= 9) {
				days = "0" + day;
			} else {
				days = "" + day;
			}
			String type = null;
			if (examPaper.getExamType() == 1) {
				type = "z";
			}
			if (examPaper.getExamType() == 2) {
				type = "m";
			}
			//获取考试名称前缀,去除数字，字母
//			String examName
//				= examPaper.getExamName().replaceAll("\\s*", "").replaceAll("[^(\\u4e00-\\u9fa5)]", "").substring(0,1);

			//前缀 = 年的最后两位  + 月份（两位） + 考试名称（中文拼音）首字母（去除数字，字母） + 考试类型 + 季度拼音首字母大写（春季就是 C）
//			String result = year
//							+ months
//							+ toFirstChar(examName).toUpperCase()
//							+ examPaper.getExamType()
//							+ quarter;

			//前缀 = 年的最后两位  + 月份（两位） + 日 （两位） + 考试类型 正式考试  z   模拟考试   m
			String pre = year
				+ months
				+ days
				+ type;
			//查询是当前前缀已生成准考证号码最大的一位
			int count = examPaperService.getCandidateNoCount(pre);
			String result = null;
			//格式化
			DecimalFormat decimalFormat = new DecimalFormat("0000");
			count++;
			result = pre + (decimalFormat.format(count));
			//返回
			return result;
		}
		return null;
	}


	/**
	 * 考试中页面刷新,1查询所有的答题信息，2查询所有的已答信息  3其他信息
	 *
	 * @param examPaperVO 必须包含 scoreId
	 * @return
	 */
	@GetMapping("/getExamRefreshInfo")
	public R getExamRefreshInfo(ExamPaperVO examPaperVO) {
		return R.data(examPaperService.getExamRefreshInfo(examPaperVO));
	}

	/**
	 * 查询考试统计信息
	 *
	 * @param examPaperVO
	 * @return
	 */
	@GetMapping("/getExamStatistics")
	public R getExamStatistics(ExamPaperVO examPaperVO, Query query) {
		IPage<ExamStatisticsVO> pages = examPaperService.getExamStatistics(Condition.getPage(query), examPaperVO);
		return R.data(pages);
	}

	/**
	 * 根据考试id 统计查询考试信息
	 *
	 * @param examPaperVO
	 * @return
	 */
	@GetMapping("/getExamInfoByExamId")
	public R getExamInfoByExamId(ExamStatisticsVO examPaperVO, Query query) {
		IPage<ExamStatisticsVO> pages = examPaperService.getExamInfoByExamId(Condition.getPage(query), examPaperVO);
		return R.data(pages);
	}

	/**
	 * 导出考试统计情况信息
	 *
	 * @param examScoreVO 需要传三个参数  1：培训学校名称，2:培训学校id 3:日期  4：考试类型，培训考试
	 */
	@GetMapping("export-examStatistics")
	public void exportExamStatistics(HttpServletResponse response, ExamScoreVO examScoreVO) throws Exception {
		try {
			//获取考试情况统计数据
			List<ExamStatisticsVO> list1 = examPaperService.getExamInfoByExamIdList(examScoreVO);

			//获取考试成绩明细
			List<ExamScoreVO> list2 = examPaperService.getExamScoreList(examScoreVO);

			//模板路径
			String templateFileName = "xls" + File.separator + "examStatistics.xlsx";
			//读取模板输入流
			InputStream resourceAsStream = this.getClass().getClassLoader().getResourceAsStream(templateFileName);

			//设置文件名称
			String fileName = URLEncoder.encode(examScoreVO.getCompanyName() + examScoreVO.getExamDate() + "考试情况数据" + DateUtil.time(), Charsets.UTF_8.name());
			//设置文件类型
			response.setContentType("application/vnd.ms-excel");
			//设置编码格式
			response.setCharacterEncoding("utf-8");
			response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");

			//创建excel
			ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).withTemplate(resourceAsStream).build();
			//创建sheet
			WriteSheet writeSheet = EasyExcel.writerSheet().build();
			FillConfig fillConfig = FillConfig.builder().forceNewRow(Boolean.TRUE).build();

			// 如果有多个list 模板上必须有{前缀.} 这里的前缀就是 a，然后多个list必须用 FillWrapper包裹
			excelWriter.fill(new FillWrapper("a", list1), fillConfig, writeSheet);
			excelWriter.fill(new FillWrapper("b", list2), fillConfig, writeSheet);

			//其他数据写入 map
			Map map = new HashMap<>();
			//计算总的报名人数，异常人数，缺考人数，合格人数，
			int znum = list1.stream().mapToInt(ExamStatisticsVO::getZnum).sum();
			int ycnum = list1.stream().mapToInt(ExamStatisticsVO::getYcnum).sum();
			int qknum = list1.stream().mapToInt(ExamStatisticsVO::getQknum).sum();
			int yknum = list1.stream().mapToInt(ExamStatisticsVO::getYknum).sum();
			map.put("znumTotal", znum);
			map.put("ycnumTotal", ycnum);
			map.put("qknumTotal", qknum);
			map.put("yknumTotal", yknum);

			//计算占比，格式化小数
			DecimalFormat df = new DecimalFormat("0.00");
			//返回的是String类型,缺考人数占比
			map.put("absentProportionTotal", df.format((float) qknum / znum * 100));

			//合格人数占比
			DecimalFormat df1 = new DecimalFormat("0.00");
			map.put("qualifiedProportionTotal", df1.format((float) yknum / znum * 100));

			excelWriter.fill(map, writeSheet);

			//关闭流
			excelWriter.finish();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

}