hust-epi/whepi-web/src/main/java/com/bofeng/wx/controller/ScanController.java

package com.bofeng.wx.controller;

import com.bofeng.dao.*;
import com.bofeng.entity.*;
import com.bofeng.service.HomeService;
import com.bofeng.service.ScanService;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageConfig;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.yvan.mvc.Pd;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.Cleanup;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

@Api("进出扫码")
@RestController
public class ScanController {

    // 获取code
    private static final String oauth2 = "https://open.weixin.qq.com/connect/oauth2/authorize?appid=";
    private static final String moreUrl = "&response_type=code&scope=snsapi_userinfo&state=STATE#wechat_redirect";
    @Value("${yvan.wechat.appID}")
    private String appId;
    @Value("${yvan.wechat.domain}")
    private String domain;

    @Autowired
    private HomeService homeService;
    @Autowired
    private UserRoleMapper userRoleMapper;
    @Autowired
    private UserOpenMapper userOpenMapper;
    @Autowired
    private UptownHomeMapper uptownHomeMapper;
    @Autowired
    private UptownUnitMapper uptownUnitMapper;
    @Autowired
    private UptownDoorMapper uptownDoorMapper;
    @Autowired
    private OutScanMapper outScanMapper;
    @Autowired
    private OutScanConfMapper outScanConfMapper;
    @Autowired
    private MsReportMapper msReportMapper;
    @Autowired
    private ScanService scanService;
    @Autowired
    private OutScanEstateMapper outScanEstateMapper;
    @Autowired
    private OutUserMapper outUserMapper;

    /**
     * @param doorId 小区大门ID
     * @param inType 1进入;2外出
     * @param request
     * @param model
     * @return
     */
    @ApiOperation("进出扫码页面")
    @GetMapping("/user/scan.html")
    public ModelAndView scan(HttpServletRequest request, ModelMap model, @RequestParam(value = "doorId", required = false, defaultValue = "0") Long doorId,
                             @RequestParam(value = "inType", required = false, defaultValue = "0") Integer inType) {
        UserOpen userOpen = homeService.getUserOpenByCode(request.getParameter("code"));
        List<UserRole> list = userRoleMapper.getUserRoleByUserId(userOpen.getUserId());
//        List<UserRole> list = userRoleMapper.getUserRoleByUserId(1L);
//        UserOpen userOpen = userOpenMapper.selectByUserId(1L);
        model.put("user", userOpen);
        model.put("user_id", "\"" + userOpen.getUserId() + "\"");
        if (list.size() == 0) {
            return new ModelAndView("/user/home.ftl", model);
        } else {
            UptownDoor uptownDoor = uptownDoorMapper.getUptownDoorById(doorId);
            List<UptownHome> uptownHomes =  uptownHomeMapper.getUptownHomeByUserId(userOpen.getUserId());
            UptownUnit unit = uptownUnitMapper.getUptownUnitByUser(userOpen.getUserId());
            model.put("door", uptownDoor.getUptownName() + uptownDoor.getDoorName());
            model.put("doorplate", unit.getRidgepole() + "栋" + unit.getUnit() + "单元" +unit.getDoorplate() + "号");
            model.put("name", uptownHomes.get(0).getLinkman());
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            model.put("now", sdf.format(new Date()));
            model.put("inType", inType);
            if (!unit.getUptownId().equals(uptownDoor.getUptownId())) {
                model.put("into", "禁止" + getMsg(inType));
                model.put("collor", "red");
                model.put("errorMsg", "非本小区居民，禁止" + getMsg(inType) + "。");
                return new ModelAndView("/user/scan.ftl", model);
            }
            OutScan o = outScanMapper.getOutScanLately(userOpen.getUserId(), inType);
            if (o != null && compareTime(o.getDate(),1)) {
                model.put("collor", "orange");
                model.put("into", "允许" + getMsg(inType));
                model.put("errorMsg", "您最近一分钟已经扫码，本次扫码重复。");
                return new ModelAndView("/user/scan.ftl", model);
            }
            if (inType == 1) {
                OutScan outScan = outScanMapper.getOutScanByUserlimit(userOpen.getUserId());
                if (outScan != null && new Date(outScan.getDate().getMillis() + 3600L * 1000 * 24).compareTo(new Date()) == -1) {
                    model.put("collor", "orange");
                    model.put("into", "允许进入");
                    model.put("errorMsg", "您在外驻留时间超过1日，请尽快上报您的健康状况，防止在外长时间逗留。");
                    scanService.saveOutScan(userOpen.getUserId(), inType, uptownDoor.getUptownId(), uptownDoor.getDoorId());
                    return new ModelAndView("/user/scan.ftl", model);
                }
            }
            if (inType == 2) {
                OutUser outUser = outUserMapper.getOutUserByUserId(userOpen.getUserId());
                if (outUser != null) {
                    model.put("collor", "green");
                    model.put("into", "允许通行");
                    model.put("errorMsg", "外出请佩戴口罩，避免近距离接触，保生产同时注意自身安全。");
                    scanService.saveOutScan(userOpen.getUserId(), inType, uptownDoor.getUptownId(), uptownDoor.getDoorId());
                    return new ModelAndView("/user/scan.ftl", model);
                }
                OutScanConf outScanConf = outScanConfMapper.getOutScanConfByUptown(uptownDoor.getUptownId());
                OutScan outScan = outScanMapper.getOutScanThreeday(userOpen.getUserId(), 2, outScanConf.getGoOutFre());
                if (outScan != null) {
                    model.put("collor", "red");
                    model.put("into", "禁止" + getMsg(inType));
                    model.put("errorMsg", "距您上次外出时间未满"+outScanConf.getGoOutFre()+"日，禁止外出，请配合。上次外出登记时间 "+sdf.format(new Date(outScan.getDate().getMillis()))+"。");
                    return new ModelAndView("/user/scan.ftl", model);
                }
                MsReport msReport = msReportMapper.selectMsReportToday(userOpen.getUserId());
                if (msReport != null && msReport.getMsStatus() == 2) {
                    model.put("collor", "red");
                    model.put("into", "禁止通行");
                    model.put("errorMsg", "健康日报提示您存在健康问题，禁止外出，请配合。");
                    return new ModelAndView("/user/scan.ftl", model);
                }
                List<MsReport> msReports = msReportMapper.selectMsReportLate(userOpen.getUserId());
                if (msReports.size() == 0) {
                    model.put("collor", "red");
                    model.put("into", "禁止" + getMsg(inType));
                    model.put("errorMsg", "您最近三日未上报健康日报，请说明原因，配合监测人员。");
                    return new ModelAndView("/user/scan.ftl", model);
                } else if (msReports.size() < 3) {
                    model.put("collor", "orange");
                    model.put("into", "可以" + getMsg(inType));
                    model.put("errorMsg", "您最近三日健康日报漏报，请说明原因并及时填报，配合监测人员。");
                    scanService.saveOutScan(userOpen.getUserId(), inType, uptownDoor.getUptownId(), uptownDoor.getDoorId());
                    return new ModelAndView("/user/scan.ftl", model);
                }
            }
            model.put("collor", "green");
            model.put("into", "允许" + getMsg(inType));
            if (inType == 2) {
                model.put("errorMsg", "外出请佩戴口罩，避免近距离接触，保生产同时注意自身安全。");
            } else if (inType == 1) {
                model.put("errorMsg", "欢迎归来，注意防护。");
            }
            scanService.saveOutScan(userOpen.getUserId(), inType, uptownDoor.getUptownId(), uptownDoor.getDoorId());
            return new ModelAndView("/user/scan.ftl", model);
        }
    }

    @ApiOperation("巡检人员检查扫码信息")
    @GetMapping("/user/scanEstate.html")
    public ModelAndView scanEstate(HttpServletRequest request, ModelMap model) {
        UserOpen userOpen = homeService.getUserOpenByCode(request.getParameter("code"));
        List<UserRole> list = userRoleMapper.getUserRoleByUserId(userOpen.getUserId());
//        List<UserRole> list = userRoleMapper.getUserRoleByUserId(1L);
//        UserOpen userOpen = userOpenMapper.selectByUserId(1L);
        model.put("user", userOpen);
        model.put("user_id", "\"" + userOpen.getUserId() + "\"");
        if (list.size() == 0) {
            return new ModelAndView("/user/home.ftl", model);
        } else {
            List<UptownHome> uptownHomes =  uptownHomeMapper.getUptownHomeByUserId(userOpen.getUserId());
            UptownUnit unit = uptownUnitMapper.getUptownUnitByUser(userOpen.getUserId());
            model.put("doorplate", unit.getRidgepole() + "栋" + unit.getUnit() + "单元" +unit.getDoorplate() + "号");
            model.put("name", uptownHomes.get(0).getLinkman());
            SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            model.put("now", format.format(new Date()));
            OutScanEstate o = outScanEstateMapper.getOutScanEstatLately(userOpen.getUserId());
            if (o != null && compareTime(o.getScanDate(),1)) {
                model.put("collor", "orange");
                model.put("into", "允许通行");
                model.put("errorMsg", "您最近一分钟已经扫码，本次扫码重复。");
                return new ModelAndView("/user/scanEstate.ftl", model);
            }
            OutUser outUser = outUserMapper.getOutUserByUserId(userOpen.getUserId());
            if (outUser != null) {
                scanService.saveOutScanEstate(userOpen.getUserId());
                model.put("collor", "green");
                model.put("into", "允许通行");
                model.put("errorMsg", "外出请佩戴口罩，避免近距离接触，保生产同时注意自身安全。");
                return new ModelAndView("/user/scanEstate.ftl", model);
            }
            List<MsReport> msReports = msReportMapper.selectMsReportLate(userOpen.getUserId());
            if (msReports.size() != 3) {
                model.put("collor", "red");
                model.put("into", "禁止通行");
                model.put("errorMsg", "您近期有未提交健康日报，无法判断您健康状态，请尽快返回住处。");
                return new ModelAndView("/user/scanEstate.ftl", model);
            }
            MsReport msReport = msReportMapper.selectMsReportToday(userOpen.getUserId());
            if (msReport != null && msReport.getMsStatus() == 2) {
                model.put("collor", "red");
                model.put("into", "禁止通行");
                model.put("errorMsg", "您近日提交的健康日报显示您健康存在问题，请尽快返回住处。");
                return new ModelAndView("/user/scanEstate.ftl", model);
            }
            OutScan outScan = outScanMapper.getOutScanToday(userOpen.getUserId());
            if (outScan == null) {
                model.put("collor", "red");
                model.put("into", "禁止通行");
                model.put("errorMsg", "您今日没有正常出门记录，为自身及家人安全，请赶紧回家。");
                return new ModelAndView("/user/scanEstate.ftl", model);
            }
            List<OutScanEstate> list1 = outScanEstateMapper.getOutScanEstateTodayByUser(userOpen.getUserId());
            int conut = list1.size() + 1;
            if (list1.size() < 2) {
                model.put("collor", "green");
                model.put("into", "允许通行");
                if (list1.size() == 0) {
                    model.put("errorMsg", "您今日已"+conut+"次被问讯，请说明外出原因，尽快回家。");
                } else {
                    model.put("errorMsg", "您今日已"+conut+"次被问讯，请说明外出原因，尽快回家。问询时间："+queryTime(list1)+"。");
                }
            } else {
                model.put("collor", "red");
                model.put("into", "禁止通行");
                model.put("errorMsg", "您今日已被巡检人员问讯超过2次，为自身及家人安全，请赶紧回家。问询时间："+queryTime(list1)+"。");
            }
            scanService.saveOutScanEstate(userOpen.getUserId());
//            model.put("img", "/scan/scanEstate_qr.png");
            return new ModelAndView("/user/scanEstate.ftl", model);
        }
    }

    /**
     * 判断传入的时间和现在相差是否在i分钟内
     * @param date
     * @param i
     * @return
     */
    public Boolean compareTime(DateTime date, Integer i) {
        Date date1 = new Date(date.getMillis());
        if ((date1.getTime() + 1000L * 60 * i) > System.currentTimeMillis()) {
            return true;
        } else {
            return false;
        }
    }

    public String getMsg(Integer inType) {
        if (inType == 1) {
            return "进入";
        } else if (inType == 2) {
            return "外出";
        }
        return "";
    }

    public String queryTime (List<OutScanEstate> list) {
        SimpleDateFormat format = new SimpleDateFormat("HH:mm");
        String time = "";
        for (OutScanEstate o : list) {
            String date = format.format(new Date(o.getScanDate().getMillis()));
            time += date + "、";
        }
        return time.substring(0, time.length() - 1);
    }

//    /**
//     * 生成进出门二维码
//     * @param doorId 小区大门ID
//     * @param inType 1进入;2外出
//     * @param response
//     * @throws IOException
//     * @throws WriterException
//     */
//    @GetMapping(value = "/scan/scan_qr.png", produces = MediaType.IMAGE_PNG_VALUE)
//    public void scanQrImg(@Pd(name = "doorId") Long doorId, @Pd(name = "inType") Integer inType,
//                            HttpServletResponse response) throws IOException, WriterException {
//        @Cleanup ServletOutputStream outputStream = response.getOutputStream();
//        String msg = URLEncoder.encode( domain + "user/scan.html?doorId=" + doorId + "&inType=" + inType,"utf-8");
//        BitMatrix matrix = new MultiFormatWriter().encode(oauth2 + appId + "&redirect_uri=" +
//                msg + moreUrl, BarcodeFormat.QR_CODE, 256, 256);
//        MatrixToImageWriter.writeToStream(matrix, MediaType.IMAGE_PNG.getSubtype(), outputStream, new MatrixToImageConfig());
//    }
//
//    /**
//     * 生成校园巡逻二维码
//     * @throws IOException
//     * @throws WriterException
//     */
//    @GetMapping(value = "/scan/scanEstate_qr.png", produces = MediaType.IMAGE_PNG_VALUE)
//    public void scanEstateQrImg(HttpServletResponse response) throws IOException, WriterException {
//        @Cleanup ServletOutputStream outputStream = response.getOutputStream();
//        String msg = URLEncoder.encode( domain + "user/scanEstate.html","utf-8");
//        BitMatrix matrix = new MultiFormatWriter().encode(oauth2 + appId + "&redirect_uri=" +
//                msg + moreUrl, BarcodeFormat.QR_CODE, 256, 256);
//        MatrixToImageWriter.writeToStream(matrix, MediaType.IMAGE_PNG.getSubtype(), outputStream, new MatrixToImageConfig());
//    }
}