package cn.besteam.modules.medical.controller;

import java.io.IOException;
import java.net.URLEncoder;
import java.util.*;
import java.util.stream.Collectors;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import cn.besteam.common.aspect.annotation.OpenApi;
import cn.besteam.common.constant.CommonConstant;
import cn.besteam.common.constant.DepartmentEnum;
import cn.besteam.modules.medical.entity.MwDetectionIndex;
import cn.besteam.modules.shiro.authc.util.EnvUtils;
import cn.besteam.modules.system.entity.SysDepartment;
import cn.besteam.modules.system.service.ISysDepartmentService;
import cn.hutool.poi.excel.ExcelUtil;
import cn.hutool.poi.excel.ExcelWriter;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import cn.besteam.common.api.vo.Result;
import cn.besteam.common.aspect.annotation.AutoLog;
import cn.besteam.common.util.oConvertUtils;
import cn.besteam.modules.medical.entity.MwEquipment;
import cn.besteam.modules.medical.service.IMwEquipmentService;
import lombok.Data;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;

 /**
 * @Title: Controller
 * @Description: 设备
 * @author： besteam
 * @date：   2022-09-04
 * @version： V1.0
 */
@RestController
@RequestMapping("/medical/mwEquipment")
@Slf4j
@Api(description = "设备")
public class MwEquipmentController {

	@Autowired
	private IMwEquipmentService mwEquipmentService;

	 @Autowired
	 private ISysDepartmentService sysDepartmentService;
	
	/**
	  * 分页列表查询
	 * @param mwEquipment
	 * @param pageNo
	 * @param pageSize
	 * @param req
	 * @return
	 */
	@GetMapping(value = "/list")
	@ApiOperation(value = "获取设备数据列表", notes = "获取所有设备数据列表", produces = "application/json")
	public Result<IPage<MwEquipment>> queryPageList(MwEquipment mwEquipment, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, HttpServletRequest req) {

		Result<IPage<MwEquipment>> result = new Result<IPage<MwEquipment>>();
		QueryWrapper<MwEquipment> queryWrapper = new QueryWrapper<MwEquipment>(mwEquipment);
		Page<MwEquipment> page = new Page<MwEquipment>(pageNo, pageSize);
		//排序逻辑 处理
		String column = req.getParameter("column");
		String order = req.getParameter("order");
		if(oConvertUtils.isNotEmpty(column) && oConvertUtils.isNotEmpty(order)) {
			if("asc".equals(order)) {
				queryWrapper.orderByAsc(oConvertUtils.camelToUnderline(column));
			}else {
				queryWrapper.orderByDesc(oConvertUtils.camelToUnderline(column));
			}
		}
		queryWrapper.eq("del_flag",CommonConstant.DEL_FLAG_0);
		if(oConvertUtils.isNotEmpty(mwEquipment.getName()) ){
			queryWrapper.like("name",mwEquipment.getName());
		}
		if(oConvertUtils.isNotEmpty(mwEquipment.getCode()) ){
			queryWrapper.like("code",mwEquipment.getCode());
		}
		if(oConvertUtils.isNotEmpty(mwEquipment.getHospitalId()) ){
			queryWrapper.in("hospital_id",this.getSearchHospitalIdList(Integer.parseInt(mwEquipment.getHospitalId().toString())));
			//queryWrapper.apply("  (FIND_IN_SET("+mwEquipment.getHospitalId()+",hospital_id))");
		}
		IPage<MwEquipment> pageList = mwEquipmentService.page(page, queryWrapper);
		List<MwEquipment> mwEquipmentList = pageList.getRecords();
		for(MwEquipment mwEquipment1:mwEquipmentList){
			mwEquipment1.setHospitalIdList(sysDepartmentService.getParentIdList(Integer.parseInt(mwEquipment.getHospitalId().toString())));
		}
		pageList.setRecords(mwEquipmentList);
		result.setSuccess(true);
		result.setResult(pageList);
		return result;
	}
	
	/**
	  *   添加
	 * @param mwEquipment
	 * @return
	 */
	@PostMapping(value = "/add")
	@AutoLog(value = "添加设备")
	@ApiOperation(value = "添加设备", notes = "添加设备", produces = "application/json")
	public Result<MwEquipment> add(@RequestBody MwEquipment mwEquipment) {

		Result<MwEquipment> result = new Result<MwEquipment>();
		try {
			SysDepartment sysDepartment = sysDepartmentService.getHospital(Integer.parseInt(mwEquipment.getHospitalId().toString()));
			mwEquipment.setHospitalName(sysDepartment.getDepartName());
			mwEquipmentService.save(mwEquipment);
			result.success("添加成功！");
		} catch (Exception e) {
			log.error("", e);
			result.error500("操作失败");
		}
		return result;
	}
	
	/**
	  *  编辑
	 * @param mwEquipment
	 * @return
	 */
	@PostMapping(value = "/edit")
	@AutoLog(value = "编辑设备")
	@ApiOperation(value = "编辑设备", notes = "编辑设备", produces = "application/json")
	public Result<MwEquipment> edit(@RequestBody MwEquipment mwEquipment) {

		Result<MwEquipment> result = new Result<MwEquipment>();
		MwEquipment mwEquipmentEntity = mwEquipmentService.getById(mwEquipment.getId());
		if(mwEquipmentEntity==null) {
			result.error500("未找到对应实体");
		}else {
			SysDepartment sysDepartment = sysDepartmentService.getHospital(Integer.parseInt(mwEquipment.getHospitalId().toString()));
			mwEquipment.setHospitalName(sysDepartment.getDepartName());
			boolean ok = mwEquipmentService.updateById(mwEquipment);
			//TODO 返回false说明什么？
			if(ok) {
				result.success("修改成功!");
			}
		}
		
		return result;
	}

	 @PostMapping(value = "/updateStatus")
	 @AutoLog(value = "禁用/启用")
	 @ApiOperation(value = "禁用/启用", notes = "禁用/启用", produces = "application/json")
	 public Result<MwEquipment> updateStatus(@RequestBody MwEquipment mwEquipment) {

		 Result<MwEquipment> result = new Result<MwEquipment>();
		 MwEquipment mwEquipmentEntity = mwEquipmentService.getById(mwEquipment.getId());
		 if(mwEquipmentEntity==null) {
			 result.error500("未找到对应实体");
		 }else {
			 mwEquipmentEntity.setStatus(mwEquipmentEntity.getStatus()==1?2:1);
			 boolean ok = mwEquipmentService.updateById(mwEquipmentEntity);
			 if(ok) {
				 result.success("修改成功!");
			 }
		 }

		 return result;
	 }

	/**
	  *   通过id删除
	 * @param mwEquipment
	 * @return
	 */
	@PostMapping(value = "/delete")
	@AutoLog(value = "删除设备")
	@ApiOperation(value = "删除设备", notes = "删除设备", produces = "application/json")
	public Result<MwEquipment> delete(@RequestBody MwEquipment mwEquipment) {

		Result<MwEquipment> result = new Result<MwEquipment>();
		MwEquipment mwEquipmentEntity = mwEquipmentService.getById(mwEquipment.getId());
		if(mwEquipmentEntity==null) {
			result.error500("未找到对应实体");
		}else {
			mwEquipmentEntity.setDelFlag(1);
			boolean ok = mwEquipmentService.updateById(mwEquipmentEntity);
			if(ok) {
				result.success("删除成功!");
			}
		}
		
		return result;
	}
	
	/**
	  *  批量删除
	 * @param ids
	 * @return
	 */
	@PostMapping(value = "/deleteBatch")
	@AutoLog(value = "批量删除设备")
	@ApiOperation(value = "批量删除设备", notes = "批量删除设备", produces = "application/json")
	public Result<MwEquipment> deleteBatch(@RequestParam(name="ids",required=true) String ids) {

		Result<MwEquipment> result = new Result<MwEquipment>();
		if(ids==null || "".equals(ids.trim())) {
			result.error500("参数不识别！");
		}else {
			this.mwEquipmentService.removeByIds(Arrays.asList(ids.split(",")));
			result.success("删除成功!");
		}
		return result;
	}
	
	/**
	  * 通过id查询
	 * @param id
	 * @return
	 */
	@GetMapping(value = "/queryById")
	@ApiOperation(value = "通过id查询设备", notes = "通过id查询设备", produces = "application/json")
	public Result<MwEquipment> queryById(@RequestParam(name="id",required=true) String id) {

		Result<MwEquipment> result = new Result<MwEquipment>();
		MwEquipment mwEquipment = mwEquipmentService.getById(id);
		if(mwEquipment==null) {
			result.error500("未找到对应实体");
		}else {
			mwEquipment.setHospitalIdList(sysDepartmentService.getParentIdList(Integer.parseInt(mwEquipment.getHospitalId().toString())));
			result.setResult(mwEquipment);
			result.setSuccess(true);
		}
		return result;
	}
	 /*=====设备导出--开始=======*/
	 private static Map<String, String> headerMapList = new LinkedHashMap<>();

	 static {
		 headerMapList.put("hospitalName", "医院名称");
		 headerMapList.put("name", "设备名称");
		 headerMapList.put("code", "设备编号");
		 headerMapList.put("detectionIndexName", "监测指标");
	 }

	 @Data
	 private static class MwEquipmentExportListVO {

		 private String name;

		 private String code;

		 private String hospitalName;

		 private String detectionIndexName;

	 }

	 @GetMapping("/listExport")
	 @OpenApi
	 @ApiOperation(value = "设备导出", notes = "设备导出", produces = "application/json")
	 public void listExport(HttpServletResponse response, MwEquipment mwEquipment) throws IOException {

		 QueryWrapper<MwEquipment> queryWrapper = new QueryWrapper<MwEquipment>(mwEquipment);

		 queryWrapper.eq("del_flag",CommonConstant.DEL_FLAG_0);
		 if(oConvertUtils.isNotEmpty(mwEquipment.getName()) ){
			 queryWrapper.like("name",mwEquipment.getName());
		 }
		 if(oConvertUtils.isNotEmpty(mwEquipment.getCode()) ){
			 queryWrapper.like("code",mwEquipment.getCode());
		 }
		 if(oConvertUtils.isNotEmpty(mwEquipment.getHospitalId()) ){
			 queryWrapper.in("hospital_id",this.getSearchHospitalIdList(Integer.parseInt(mwEquipment.getHospitalId().toString())));
		 }
		 List<MwEquipment> list = mwEquipmentService.list( queryWrapper);
		 List<MwEquipmentController.MwEquipmentExportListVO> voList = new ArrayList<>();
		 for (MwEquipment collectRecordVO : list) {
			 MwEquipmentController.MwEquipmentExportListVO vo=new MwEquipmentController.MwEquipmentExportListVO();
			 BeanUtils.copyProperties(collectRecordVO, vo);
			 voList.add(vo);
		 }
		 ExcelWriter writer = ExcelUtil.getWriter(true);
		 writer.renameSheet("设备管理");
		 writer.setColumnWidth(0, 30);
		 writer.setColumnWidth(1, 30);
		 writer.setColumnWidth(2, 30);
		 writer.setColumnWidth(3, 30);
		 writer.setHeaderAlias(headerMapList);
		 writer.write(voList);
		 //
		 String fileName = "设备管理";
		 response.setContentType("application/octet-stream;charset=utf-8");
		 response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8") + ".xlsx");
		 writer.flush(response.getOutputStream());
		 writer.close();

	 }
	 /*=====设备导出--结尾=======*/

	 private List<Integer> getSearchHospitalIdList(Integer hospitalId) {

		 if (hospitalId == null) {
			 return sysDepartmentService.listMyHospital(EnvUtils.getUser().getId()).stream().map(SysDepartment::getId).collect(Collectors.toList());
		 } else {
			 SysDepartment sysDepartment = sysDepartmentService.getById(hospitalId);
			 if (sysDepartment == null) {
				 return new ArrayList<>();
			 }
			 List<SysDepartment> childrenList = sysDepartmentService.list(new QueryWrapper<SysDepartment>().lambda().likeRight(SysDepartment::getTreeCode, sysDepartment.getTreeCode()).eq(SysDepartment::getOrgType, DepartmentEnum.OrgTypeEnum.HOSPITAL.getType()));
			 if (CollectionUtils.isEmpty(childrenList)) {
				 return new ArrayList<>();
			 }
			 return childrenList.stream().map(SysDepartment::getId).collect(Collectors.toList());
		 }
	 }
}