package com.taoke.autopay.controller; import com.taoke.autopay.dto.WXAppInfoDto; import com.taoke.autopay.entity.KeyOrder; import com.taoke.autopay.entity.SystemConfigKeyEnum; import com.taoke.autopay.entity.WxUserInfo; import com.taoke.autopay.exception.KeyOrderException; import com.taoke.autopay.exception.WxOrderCountException; import com.taoke.autopay.factory.OrderFactory; import com.taoke.autopay.service.KeyOrderService; import com.taoke.autopay.service.SystemConfigService; import com.taoke.autopay.service.WxUserOrderCountService; import com.taoke.autopay.service.WxUserService; import com.taoke.autopay.utils.*; import net.sf.json.JSONObject; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; import org.yeshi.utils.UrlUtils; import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpSession; import java.io.IOException; import java.io.PrintWriter; import java.util.Date; import java.util.List; @Controller @RequestMapping("webapi") public class WebApiController { Logger logger = LoggerFactory.getLogger(WebApiController.class); Logger wxLogger = LoggerFactory.getLogger("wxLogger"); @Resource private KeyOrderService keyOrderService; @Resource private WxUserService wxUserService; @Resource private SystemConfigService systemConfigService; @ResponseBody @RequestMapping(value = "submitKey") public String submitKey(String key) { if(1>0){ return JsonUtil.loadFalseResult(0, "接口已关闭"); } if (StringUtil.isNullOrEmpty(key)) { return JsonUtil.loadFalseResult(0, "请上传key"); } List urllist = UrlUtils.parseUrlsFromText(key); if (urllist.isEmpty() || !urllist.get(0).contains("ur.alipay.com")) { return JsonUtil.loadFalseResult("支付宝口令不正确"); } try { KeyOrder order = keyOrderService.addKeyOrder(key, null, TimeUtil.getGernalTime(System.currentTimeMillis(), "yyyyMMdd")); Long uid = keyOrderService.getCanDistributeUid(); if (uid != null) { KeyOrder orderUpdate = new KeyOrder(); orderUpdate.setId(order.getId()); orderUpdate.setDistributeClientUid(uid); orderUpdate.setDistributeTime(new Date()); keyOrderService.update(orderUpdate); } return JsonUtil.loadTrueResult(""); } catch (KeyOrderException e) { e.printStackTrace(); return JsonUtil.loadFalseResult(e.getMessage()); } catch (WxOrderCountException e) { return JsonUtil.loadFalseResult("今日超过最大提交次数"); } } private void addKey(String key, Long wxUid) throws KeyOrderException, WxOrderCountException { KeyOrder order = keyOrderService.addKeyOrder(key, wxUid, TimeUtil.getGernalTime(System.currentTimeMillis(), "yyyyMMdd")); Long uid = keyOrderService.getCanDistributeUid(); if (uid != null) { KeyOrder orderUpdate = new KeyOrder(); orderUpdate.setId(order.getId()); orderUpdate.setDistributeClientUid(uid); orderUpdate.setDistributeTime(new Date()); keyOrderService.update(orderUpdate); } } @ResponseBody @RequestMapping(value = "submitKeyV2") public String submitKeyV2(String key, HttpSession session) { WxUserInfo user = (WxUserInfo) session.getAttribute(Constant.SESSION_KEY_USER); if (StringUtil.isNullOrEmpty(key)) { return JsonUtil.loadFalseResult(0, "请上传key"); } List urllist = UrlUtils.parseUrlsFromText(key); if (urllist.isEmpty() || !urllist.get(0).contains("ur.alipay.com")) { return JsonUtil.loadFalseResult("支付宝口令不正确"); } if (user == null) { // 先保存KEY // SESSION_KEY_TEMP_ALIPAY_KEY session.setAttribute(Constant.SESSION_KEY_TEMP_ALIPAY_KEY, key); wxLogger.info("微信没有授权:" + session.getId()); String redictLink = systemConfigService.getValueCache(SystemConfigKeyEnum.WX_REDIRECT_LINK); if (StringUtil.isNullOrEmpty(redictLink)) { return JsonUtil.loadFalseResult("无法获取到授权链接"); } // 没有登录,返回登录链接 JSONObject root = new JSONObject(); root.put("link", redictLink); return JsonUtil.loadTrueResult(Constant.RESULT_CODE_NEED_LOGIN, root); } wxLogger.info("微信有授权:" + session.getId()); try { addKey(key, user.getId()); return JsonUtil.loadTrueResult(""); } catch (KeyOrderException e) { e.printStackTrace(); return JsonUtil.loadFalseResult(e.getMessage()); } catch (WxOrderCountException e) { return JsonUtil.loadFalseResult(e.getMessage()); } } @RequestMapping(value = "wxLogin") public void wxLogin(String code, String state, HttpServletResponse response, HttpSession session) throws IOException { // 根据code获取openid wxLogger.info("微信授权回调:{} code-{}", session.getId(), code); String failLink = systemConfigService.getValueCache(SystemConfigKeyEnum.WX_LOGIN_FAIL_LINK); try { WXAppInfoDto wxApp = systemConfigService.getWxAppInfoCache(); String successLink = systemConfigService.getValueCache(SystemConfigKeyEnum.WX_LOGIN_SUCCESS_LINK); WxApiUtil.WXAccessTokenInfo tokenInfo = WxApiUtil.getAcessTokenInfo(code, wxApp); if (tokenInfo != null && !StringUtil.isNullOrEmpty(tokenInfo.getOpenid())) { WxUserInfo user = wxUserService.login(tokenInfo.getOpenid()); session.setAttribute(Constant.SESSION_KEY_USER, user); wxLogger.info("微信保存用户信息:{} id-{}", session.getId(), user.getId()); String alipayKey = (String) session.getAttribute(Constant.SESSION_KEY_TEMP_ALIPAY_KEY); wxLogger.info("从session读取到key:{}", alipayKey); if (alipayKey != null) { addKey(alipayKey, user.getId()); } response.sendRedirect(successLink); return; } } catch (Exception e) { e.printStackTrace(); wxLogger.error("授权失败:{}", e.getMessage()); } response.sendRedirect(failLink); } }