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;

    /**
     * @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 format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            model.put("now", format.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);
            }
            if (inType == 2) {
                OutScanConf outScanConf = outScanConfMapper.getOutScanConfByUptown(uptownDoor.getUptownId());
                OutScan outScan = outScanMapper.getOutScanByUserlimit(userOpen.getUserId());
                if (outScan != null && !compareTime(new Date(), outScan.getOutDate(), outScanConf.getGoOutFre())) {
                    model.put("collor", "red");
                    model.put("into", "禁止" + getMsg(inType));
                    model.put("errorMsg", "外出时间间隔不满三日，禁止外出，请配合，上次外出时间 "+format.format(new Date(outScan.getOutDate().getMillis()))+"。");
                    return new ModelAndView("/user/scan.ftl", model);
                }
                List<MsReport> msReports = msReportMapper.selectMsReportLate(userOpen.getUserId());
                if (msReports.size() != 3) {
                    model.put("collor", "red");
                    model.put("into", "禁止" + getMsg(inType));
                    model.put("errorMsg", "因近期您未提交健康日报，无法确认您当前健康状态，禁止外出，请配合。");
                    return new ModelAndView("/user/scan.ftl", model);
                }
            }
            model.put("collor", "green");
            model.put("into", "可以" + getMsg(inType));
            if (inType == 2) {
                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()));
            List<OutScanEstate> list1 = outScanEstateMapper.getOutScanEstateTodayByUser(userOpen.getUserId());
            int conut = list1.size() + 1;
            if (list1.size() < 2) {
                model.put("collor", "green");
                model.put("into", "允许行动");
                model.put("errorMsg", "这是您今天第" + conut + "次出行，允许同行，注意安全。");
            } else {
                model.put("collor", "red");
                model.put("into", "劝您回家");
                model.put("errorMsg", "这是您今天第" + conut + "次出行，为了您的安全，请尽快回家。");
            }
            scanService.saveOutScanEstate(userOpen.getUserId());
//            model.put("img", "/scan/scanEstate_qr.png");
            return new ModelAndView("/user/scanEstate.ftl", model);
        }
    }

    public Boolean compareTime(Date time1, DateTime time2, Integer day) {
        Date date = new Date(time2.getMillis() + 1000L*3600*24*day);
        if (changeTime(date).equals(changeTime(time1))) {
            return true;
        } else {
            return false;
        }
    }

    public String changeTime(Date date) {
        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
        return format.format(date);
    }

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

    /**
     * 生成进出门二维码
     * @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());
    }
}