package com.huimv.receive.controller; import cn.hutool.core.util.ObjectUtil; import cn.hutool.http.HttpUtil; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.huimv.receive.common.exception.ExceptionEnum; import com.huimv.receive.common.exception.MiException; import com.huimv.receive.common.token.TokenSign; import com.huimv.receive.common.utils.*; import com.huimv.receive.entity.BillGate; import com.huimv.receive.entity.SysAccountMultilevel; import com.huimv.receive.entity.dto.WeCahtVo; import com.huimv.receive.mapper.BillGateMapper; import com.huimv.receive.service.IBaseJobService; import com.huimv.receive.service.ILoginService; import com.huimv.receive.service.ISysAccountMultilevelService; import com.huimv.receive.session.AccessToken; import com.huimv.receive.timer.test.TestWeight; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.*; import org.springframework.web.client.RestTemplate; import org.springframework.web.multipart.MultipartFile; import javax.imageio.ImageIO; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.awt.image.BufferedImage; import java.io.File; import java.io.IOException; import java.util.Date; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.concurrent.TimeUnit; @RestController @RequestMapping(value = "/admin/my") @CrossOrigin public class LoginController { @Autowired ILoginService iLoginService; @Autowired private ISysAccountMultilevelService accountMultilevelService; @Autowired private IBaseJobService baseJobService; @Autowired private UploadImage uploadImage; @Autowired private BillGateMapper billGateMapper; @PostMapping("/saveDiBang") public String logout(@RequestBody Map> map ) { List dibang = map.get("dibang"); for (TestWeight testWeight : dibang) { System.out.println(testWeight); BillGate billGate = billGateMapper.selectOne(new QueryWrapper().eq("stdno",testWeight.getFStdNo())); if (ObjectUtil.isEmpty(billGate)) { List billGate1 = billGateMapper.selectList(new QueryWrapper().eq("car_num", testWeight.getFCarNo()) .between("check_date", testWeight.getBeginTime(), testWeight.getEndTime()).eq("check_location",testWeight.getDeviceCode())); if (ObjectUtil.isNotEmpty(billGate1)) { billGate = billGate1.get(0); billGate.setGross(testWeight.getGross() + ""); billGate.setTare(testWeight.getMoney() + ""); billGate.setNet(testWeight.getNet() + ""); billGate.setStdno(testWeight.getFStdNo()); billGateMapper.insert(billGate); } } } return ""; } @RequestMapping("/logout") public Result logout(@RequestParam(name = "accountName", required = true) String accountName, @RequestParam(name = "password", required = true) String password) { return iLoginService.logout(); } @PostMapping("/test") public Result test(@RequestParam(name = "picture1")MultipartFile picture1 ) throws IOException { uploadImage.getVideoCom(picture1); return iLoginService.logout(); } @Autowired private RedisTemplate redisTemplate; @GetMapping("/send") public void createImg(HttpServletRequest request, HttpServletResponse response) { try { //设置响应类型,告诉浏览器输出的内容为图片 response.setContentType("image/jpeg"); //设置响应头信息，告诉浏览器不要缓存此内容 response.setHeader("Pragma", "No-cache"); response.setHeader("Cache-Control", "no-cache"); response.setDateHeader("Expire", 0); VerifyUtil randomValidateCode = new VerifyUtil(); //输出验证码图片 String code = randomValidateCode.getRandcode(request, response); System.out.println(code); //将生成的随机验证码存放到redis中 String remoteHost = request.getRemoteHost(); redisTemplate.opsForValue().set(remoteHost, code, 300, TimeUnit.SECONDS); } catch (Exception e) { System.out.println("获取验证码异常:" + e); throw new MiException(ExceptionEnum.VERIFCATION_FAID); } } //登录-多牧场 @PostMapping(value = "/loginMultilevel") public Result loginMultilevel(HttpServletRequest req, @RequestBody Map map) { String accountName = map.get("accountName"); String password = map.get("password"); // String code = map.get("code"); String remoteHost = req.getRemoteHost(); // String redisCode = redisTemplate.opsForValue().get(remoteHost); // if (redisCode == null) { // return new Result(10005,"验证码过期,点击刷新",false); // } // if (!redisCode.equals(code.toUpperCase())) { // return new Result(10006,"验证码错误,登录失败",false); // } // 未对密码加密 Result result = iLoginService.loginMultilevel(accountName, password); if (result.getCode() == 10000) { SysAccountMultilevel accountEntity = (SysAccountMultilevel) result.getData(); Integer id = accountEntity.getId(); IpTools.getRemoteHost(req); AccessToken accessToken = new AccessToken(); accessToken.setCreated(new Date()); accessToken.setAccountName(accountName); accessToken.setFarmId(accountEntity.getFarmIds()); String token = TokenSign.sign(accountName, id, accountEntity.getLastFarmId().toString(), null, null); accessToken.setToken(token); accessToken.setId(id); accessToken.setLoginIp(remoteHost); accessToken.setType(accountEntity.getType()); Integer lastFarmId = accountEntity.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(accountEntity.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setUserName(accountEntity.getUserName()); return new Result(ResultCode.SUCCESS, accessToken); } return result; } @PostMapping(value = "/switchUserType") public Result switchUserType(HttpServletRequest req, @RequestBody Map map) { Integer userId = TokenSign.getMemberIdByJwtToken(req); String type = map.get("type"); String remoteHost = req.getRemoteHost(); // 未对密码加密 SysAccountMultilevel accountEntity = accountMultilevelService.getById(userId); accountEntity.setType(Integer.parseInt(type)); accountMultilevelService.updateById(accountEntity); accountEntity.setJobName(baseJobService.getById(type).getJobName()); Integer id = accountEntity.getId(); IpTools.getRemoteHost(req); AccessToken accessToken = new AccessToken(); accessToken.setCreated(new Date()); accessToken.setAccountName(accountEntity.getAccount()); accessToken.setFarmId(accountEntity.getFarmIds()); String token = TokenSign.sign(accountEntity.getUserName(), id, accountEntity.getFarmIds(), accountEntity.getWorkLocationType()+"", accountEntity.getType()); accessToken.setToken(token); accessToken.setId(id); accessToken.setLoginIp(remoteHost); accessToken.setType(accountEntity.getType()); Integer lastFarmId = accountEntity.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(accountEntity.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setUserName(accountEntity.getUserName()); accessToken.setSysAccountMultilevel(accountEntity); return new Result(ResultCode.SUCCESS, accessToken); } //测试-token @PostMapping(value = "/getToken") public String getToken(HttpServletRequest req) { String token = TokenSign.sign("superadmin", 1, null, null, null); return token; } @PostMapping(value = "/loginMultilevelClent") public Result loginMultilevelClent(HttpServletRequest req, @RequestBody Map map) { String accountName = map.get("accountName"); String password = map.get("password"); // String code = map.get("code"); String remoteHost = req.getRemoteHost(); // String redisCode = redisTemplate.opsForValue().get(remoteHost); // if (redisCode == null) { // return new Result(10005,"验证码过期,点击刷新",false); // } // if (!redisCode.equals(code.toUpperCase())) { // return new Result(10006,"验证码错误,登录失败",false); // } // 未对密码加密 Result result = iLoginService.loginMultilevelClient(accountName, password); if (result.getCode() == 10000) { SysAccountMultilevel accountEntity = (SysAccountMultilevel) result.getData(); Integer id = accountEntity.getId(); IpTools.getRemoteHost(req); AccessToken accessToken = new AccessToken(); accessToken.setCreated(new Date()); accessToken.setAccountName(accountName); accessToken.setFarmId(accountEntity.getFarmIds()); Integer lastFarmId1 = accountEntity.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId1)){ lastFarmId1 = accountEntity.getFarmId(); } String token = TokenSign.sign(accountName, id,lastFarmId1.toString() , accountEntity.getWorkLocationType(),accountEntity.getType()); accessToken.setToken(token); accessToken.setId(id); accessToken.setLoginIp(remoteHost); accessToken.setType(accountEntity.getType()); Integer lastFarmId = accountEntity.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(accountEntity.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setUserName(accountEntity.getUserName()); return new Result(ResultCode.SUCCESS, accessToken); } return result; } @RequestMapping("/logoutClient") public Result logoutClient(HttpServletRequest req, @RequestBody Map map) { String accountName = map.get("accountName"); String password = map.get("password"); String remoteHost = req.getRemoteHost(); Result result = iLoginService.loginMultilevel(accountName, password); if (result.getCode() == 10000) { SysAccountMultilevel data = (SysAccountMultilevel) result.getData(); AccessToken accessToken = new AccessToken(); accessToken.setSysAccountMultilevel(data); accessToken.setAccountName(accountName); accessToken.setUserName(data.getUserName()); accessToken.setFarmId(data.getFarmIds()); accessToken.setType(data.getType()); accessToken.setId(data.getId()); String token = TokenSign.sign(data.getUserName(), data.getId(), data.getLastFarmId().toString(), data.getWorkLocationType()+"", data.getType()); accessToken.setToken(token); Integer lastFarmId = data.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(data.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setLoginIp(remoteHost); return new Result(ResultCode.SUCCESS, accessToken); } else { return result; } } /* @RequestMapping("/logoutWeChat") public Result logoutWeChat(HttpServletRequest req, @RequestBody Map map) { String remark = map.get("remark"); QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.eq("account", remark); SysAccountMultilevel one = accountMultilevelService.getOne(queryWrapper); AccessToken accessToken = new AccessToken(); if (ObjectUtil.isEmpty(one)) { SysAccountMultilevel accountMultilevel = new SysAccountMultilevel(); accountMultilevel.setAccount(remark); accountMultilevel.setFarmIds("25"); accountMultilevel.setColor(""); accountMultilevel.setMode(true); accountMultilevel.setUserName("访客"); accountMultilevel.setType(7); accountMultilevelService.save(accountMultilevel); QueryWrapper queryWrapper1 = new QueryWrapper<>(); queryWrapper1.eq("account", remark); SysAccountMultilevel one1 = accountMultilevelService.getOne(queryWrapper1); accessToken.setAccountName(remark); accessToken.setUserName("访客"); accessToken.setFarmId("25"); accessToken.setType(7); accessToken.setId(one1.getId()); String token = TokenSign.sign(remark, one1.getId(), one1.getFarmIds(), null,7); accessToken.setToken(token); } else { accessToken.setAccountName(one.getUserName()); accessToken.setUserName("访客"); accessToken.setFarmId("25"); accessToken.setType(7); accessToken.setId(one.getId()); String token = TokenSign.sign(remark, one.getId(), one.getFarmIds(), one.getWorkLocationId(),7); accessToken.setToken(token); } return new Result(ResultCode.SUCCESS,accessToken); * String accountName = map.get("accountName"); String password = map.get("password"); String remoteHost = req.getRemoteHost(); Result result = iLoginService.loginMultilevel(accountName, password); if (result.getCode() == 10000) { SysAccountMultilevel data = (SysAccountMultilevel) result.getData(); AccessToken accessToken = new AccessToken(); accessToken.setAccountName(accountName); accessToken.setUserName(data.getUserName()); accessToken.setFarmId(data.getFarmIds()); accessToken.setType(data.getType()); accessToken.setId(data.getId()); String token = TokenSign.sign(accountName,data.getId(),data.getFarmIds(),data.getWorkLocationId()); accessToken.setToken(token); Integer lastFarmId = data.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)){ lastFarmId = Integer.parseInt(data.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setLoginIp(remoteHost); return new Result(ResultCode.SUCCESS,accessToken); } else { return result; }*//* }*/ @GetMapping("/logoutWeChat") public Result logoutWeChat(HttpServletRequest req, @RequestParam String jsCode,@RequestParam String phone) { String remoteHost = req.getRemoteHost(); Map paramsMap = new HashMap<>(); paramsMap.put("appid", "wx221c66f51278c05f"); paramsMap.put("secret", "b553cf2e2de240ac93db6b6c8d135f55"); paramsMap.put("js_code", jsCode); paramsMap.put("grant_type", "authorization_code"); String result = HttpUtil.get("https://api.weixin.qq.com/sns/jscode2session", paramsMap); System.out.println(result); JSONObject jsonObject = JSON.parseObject(result); System.out.println(jsonObject); String openid = (String) jsonObject.get("openid"); String sessionKey = (String) jsonObject.get("session_key"); System.out.println("拿到了openid" + openid); System.out.println("拿到了sessionKey" + sessionKey); QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.eq("wechat_id", openid); SysAccountMultilevel data = accountMultilevelService.getOne(queryWrapper); WeCahtVo weCahtVo = new WeCahtVo(); weCahtVo.setOpenId(openid); weCahtVo.setSessionKey(sessionKey); if (ObjectUtil.isEmpty(data)) { QueryWrapper queryWrapper1 = new QueryWrapper<>(); queryWrapper1.eq("phone", phone); SysAccountMultilevel data1 = accountMultilevelService.getOne(queryWrapper1); if (ObjectUtil.isEmpty(data1)) { return new Result(ResultCode.FAIL, weCahtVo); } else { if ("".equals(data1.getWechatId()) || null == data1.getWechatId()) { //手机号已经代注册 data1.setWechatId(openid); accountMultilevelService.updateById(data1); AccessToken accessToken = new AccessToken(); if (data1.getType() == 7) { accessToken.setIsAdmin(0); } else { accessToken.setIsAdmin(1); } data1.setType(7); accessToken.setSysAccountMultilevel(data1); accessToken.setAccountName(data1.getAccount()); accessToken.setUserName(data1.getUserName()); accessToken.setFarmId(data1.getFarmIds()); accessToken.setType(7); accessToken.setId(data1.getId()); String token = TokenSign.sign(data1.getUserName(), data1.getId(), data1.getLastFarmId().toString(), data1.getWorkLocationType()+"", 7); accessToken.setToken(token); Integer lastFarmId = data1.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(data1.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setLoginIp(remoteHost); return new Result(ResultCode.SUCCESS, accessToken); } else { //返回错误，输入的手机号已经进行过微信登录 return new Result(ResultCode.UNAUTHENTICATED, "输入的手机号已被他人进行过微信登录！"); } } } else { if (phone.equals(data.getPhone())) { AccessToken accessToken = new AccessToken(); if (data.getType() == 7) { accessToken.setIsAdmin(0); } else { accessToken.setIsAdmin(1); } data.setType(7); accessToken.setSysAccountMultilevel(data); accessToken.setAccountName(data.getAccount()); accessToken.setUserName(data.getUserName()); accessToken.setFarmId(data.getFarmIds()); accessToken.setType(7); accessToken.setId(data.getId()); String token = TokenSign.sign(data.getUserName(), data.getId(), data.getLastFarmId().toString(), data.getWorkLocationType()+"", 7); accessToken.setToken(token); Integer lastFarmId = data.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(data.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setLoginIp(remoteHost); return new Result(ResultCode.SUCCESS, accessToken); }else { return new Result(ResultCode.UNAUTHENTICATED, "输入的手机号已被他人进行过微信登录！"); } } } @GetMapping("/logWeChatGetPhone") public Result logWeChatGetPhone(HttpServletRequest req, @RequestParam String jsCode) { String access_token =""; if ( !redisTemplate.hasKey("weChatGetPhone") || redisTemplate.getExpire("weChatGetPhone") ==null || redisTemplate.getExpire("weChatGetPhone") <=0){ String result = HttpUtil.get("https://api.weixin.qq.com/cgi-bin/token?appid=wx221c66f51278c05f&secret=b553cf2e2de240ac93db6b6c8d135f55&grant_type=client_credential"); System.out.println(result); JSONObject jsonObject = JSON.parseObject(result); access_token = (String) jsonObject.get("access_token"); redisTemplate.opsForValue().set("weChatGetPhone",access_token); redisTemplate.expire("weChatGetPhone",7200,TimeUnit.SECONDS); }else { access_token = redisTemplate.opsForValue().get("weChatGetPhone"); } RestTemplate restTemplate = new RestTemplate(); Map paramsMap = new HashMap<>(); paramsMap.put("code", jsCode); ResponseEntity stringResponseEntity = restTemplate.postForEntity("https://api.weixin.qq.com/wxa/business/getuserphonenumber?access_token=" + access_token, paramsMap, String.class); System.out.println(stringResponseEntity.getBody()); return new Result(ResultCode.SUCCESS,JSON.parseObject(stringResponseEntity.getBody())); } @PostMapping("/register2") public String register2( @RequestParam(value = "video", required = false) MultipartFile video) throws IOException { return uploadImage.getVideoCom(video); } @PostMapping("/register") public Result addAdmission(HttpServletRequest req, @RequestParam("userName") String userName, @RequestParam("phone") String phone, @RequestParam("workName") String workName, @RequestParam("papersType") String papersType, @RequestParam("papersCode") String papersCode, @RequestParam(value = "imgUrl", required = false) MultipartFile imgUrl, @RequestParam("openId") String openId) throws IOException { String remoteHost = req.getRemoteHost(); MultipartFile multipartFile = ImageSizeUtil.byte2Base64StringFun(imgUrl); boolean result = uploadImage.checkFileSize(multipartFile.getSize(), 200, "K"); if (!result) { return new Result(10001, "上传个人照片太大，请重新上传", false); } QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.eq("type",7).and(i->i.eq("phone", phone).or().eq("papers_code", papersCode)); if (ObjectUtil.isNotEmpty(accountMultilevelService.getOne(queryWrapper))) { return new Result(10001, "该手机号或身份证号已被使用！", false); } if (!PhoneNumberValidator.isValidPhoneNumber(phone)) { return new Result(10001, "手机号格式错误！", false); } if (papersType.equals("0")) { // if (!IDCardValidator.isValidIDCard(papersCode)) { // return new Result(10001, "身份证号格式错误!", false); // } } SysAccountMultilevel sysAccountMultilevel = new SysAccountMultilevel(); sysAccountMultilevel.setUserName(userName); sysAccountMultilevel.setPapersCode(papersCode); sysAccountMultilevel.setPhone(phone); sysAccountMultilevel.setAccount(phone); sysAccountMultilevel.setPid(0); sysAccountMultilevel.setPids("0"); sysAccountMultilevel.setPassword("123456"); sysAccountMultilevel.setWorkName(workName); sysAccountMultilevel.setAccountStatus(1); sysAccountMultilevel.setFarmIds("25"); sysAccountMultilevel.setLastFarmId(25); sysAccountMultilevel.setPapersType(papersType); sysAccountMultilevel.setWechatId(openId); sysAccountMultilevel.setType(7); String imageCom = uploadImage.getImageCom(multipartFile); if (imageCom.equals("上传失败")) { return new Result(10001, "图片上传失败！", false); } sysAccountMultilevel.setImgUrl(imageCom); accountMultilevelService.save(sysAccountMultilevel); AccessToken accessToken = new AccessToken(); accessToken.setSysAccountMultilevel(sysAccountMultilevel); accessToken.setAccountName(sysAccountMultilevel.getAccount()); accessToken.setUserName(sysAccountMultilevel.getUserName()); accessToken.setFarmId(sysAccountMultilevel.getFarmIds()); accessToken.setType(sysAccountMultilevel.getType()); accessToken.setId(sysAccountMultilevel.getId()); String token = TokenSign.sign(sysAccountMultilevel.getUserName(), sysAccountMultilevel.getId(), sysAccountMultilevel.getLastFarmId().toString(), sysAccountMultilevel.getWorkLocationType()+"", sysAccountMultilevel.getType()); accessToken.setToken(token); Integer lastFarmId = sysAccountMultilevel.getLastFarmId(); if (ObjectUtil.isEmpty(lastFarmId)) { lastFarmId = Integer.parseInt(sysAccountMultilevel.getFarmIds().split(",")[0]); } accessToken.setLastFarmId(lastFarmId); accessToken.setLoginIp(remoteHost); return new Result(ResultCode.SUCCESS, accessToken); } }