tiesheng-springboot/springboot-login/src/main/java/com/tiesheng/login/controller/LoginController.java

package com.tiesheng.login.controller;

import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import com.alibaba.fastjson.JSON;
import com.tiesheng.annotation.token.TokenIgnore;
import com.tiesheng.login.config.token.bean.TokenBean;
import com.tiesheng.login.pojos.CodeExtraDTO;
import com.tiesheng.login.pojos.DoLoginInfo;
import com.tiesheng.login.pojos.UniqueIndexDTO;
import com.tiesheng.login.pojos.WxminiLoginVo;
import com.tiesheng.login.service.TieshengLoginConfigurer;
import com.tiesheng.platform.config.ding.PlatformDingConfig;
import com.tiesheng.platform.config.ding.bean.DingJsapiSignature;
import com.tiesheng.platform.config.ding.bean.DingUserInfo;
import com.tiesheng.platform.config.wxmini.PlatformWxminiConfig;
import com.tiesheng.platform.config.wxmp.PlatformWxmpConfig;
import com.tiesheng.platform.config.wxmp.bean.WxConfigBean;
import com.tiesheng.platform.config.wxmp.bean.WxJsapiSignature;
import com.tiesheng.platform.config.wxmp.bean.WxUserInfo;
import com.tiesheng.util.config.GlobalConfig;
import com.tiesheng.util.exception.ApiException;
import com.tiesheng.util.pojos.ApiResp;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.Map;

/**
 * @author hao
 */
@RestController
@RequestMapping("/login")
@TokenIgnore
public class LoginController {


    @Autowired
    PlatformWxmpConfig platformWxmpConfig;
    @Autowired
    PlatformDingConfig platformDingConfig;
    @Autowired
    PlatformWxminiConfig platformWxminiConfig;
    @Autowired
    TieshengLoginConfigurer tieshengLoginConfigurer;
    @Autowired
    GlobalConfig globalConfig;


    /**
     * 唯一值登录方式，如学号，手机号等
     *
     * @return
     */
    @GetMapping("/unique/redirect")
    public void uniqueIndex(UniqueIndexDTO dto, HttpServletResponse response) {
        TokenBean tokenBean = tieshengLoginConfigurer.doLogin(new DoLoginInfo("unique_index_web",
                dto.getNo(), "web", dto.getTo(), dto.getInfo()));
        tieshengLoginConfigurer.onLoginRedirect(tokenBean, dto.getTo(), dto.getExtra(), response);
    }


    /**
     * 授权登录
     *
     * @param dto
     * @return
     */
    @PostMapping("/unique/index")
    public ApiResp<String> uniqueIndex(@RequestBody UniqueIndexDTO dto) {
        TokenBean tokenBean = tieshengLoginConfigurer.doLogin(new DoLoginInfo("unique_index_web",
                dto.getNo(), "web", dto.getTo(), dto.getInfo()));
        if (tokenBean == null) {
            throw new ApiException("登录失败");
        }
        return ApiResp.respOK(tokenBean.toToken());
    }


    ///////////////////////////////////////////////////////////////////////////
    // 钉钉相关授权
    ///////////////////////////////////////////////////////////////////////////

    /**
     * 钉钉授权登录
     *
     * @param service
     * @param extra
     * @param response
     */
    @GetMapping("/ding/index/{service}")
    public void dingIndex(@PathVariable String service, String extra, HttpServletResponse response) {
        if (StrUtil.isEmpty(extra)) {
            extra = "";
        }
        Map<String, String> map = MapUtil.newHashMap();
        map.put("corpId", platformDingConfig.getConfigBean(service).getCorpId());
        map.put("service", service);
        map.put("extra", extra);
        String query = URLUtil.buildQuery(map, Charset.defaultCharset());
        String configUrl = globalConfig.buildPath("/ding/index.html?" + query);
        try {
            response.sendRedirect(configUrl);
        } catch (IOException ignored) {
        }
    }


    /**
     * 钉钉授权回调
     *
     * @param service
     */
    @RequestMapping("/ding/oauth2/{service}")
    public void dingOauth2(@PathVariable String service, CodeExtraDTO dto, HttpServletResponse response) {
        String ddUserId = platformDingConfig.getUserIdByCode(service, dto.getCode());
        DingUserInfo dingUserInfo = platformDingConfig.topapiV2UserGet(service, ddUserId);
        TokenBean tokenBean = tieshengLoginConfigurer.doLogin(new DoLoginInfo(dingUserInfo.getAppId(),
                dingUserInfo.getUserid(), "ding", dto.getTo(),
                JSON.toJSONString(dingUserInfo)));
        tieshengLoginConfigurer.onLoginRedirect(tokenBean, dto.getTo(), dto.getExtra(), response);
    }


    /**
     * 钉钉授权jssdk
     *
     * @param url
     * @return
     */
    @GetMapping("/ding/jssdk/{service}")
    public ApiResp<DingJsapiSignature> dingJssdk(@PathVariable String service, String url) {
        DingJsapiSignature jsapiSignature = platformDingConfig.createJsapiSignature(service, url);
        return ApiResp.respOK(jsapiSignature);
    }


    ///////////////////////////////////////////////////////////////////////////
    // 微信相关
    ///////////////////////////////////////////////////////////////////////////


    /**
     * 微信授权登录
     *
     * @return
     */
    @RequestMapping("/wxmp/index/{service}")
    public void wxmpIndex(@PathVariable String service, String extra, HttpServletResponse response) throws IOException {
        if (StrUtil.isEmpty(extra)) {
            extra = "";
        }
        String configUrl = globalConfig.buildPath("/auth/wxmp/oauth2/" + service + "?extra=" + extra);
        String authorizationUrl = platformWxmpConfig.buildAuthorizationUrl(service, configUrl, "snsapi_userinfo");
        response.sendRedirect(authorizationUrl);
    }


    /**
     * 微信授权回调
     */
    @RequestMapping("/wxmp/oauth2/{service}")
    public void wxmpOauth2(@PathVariable String service, CodeExtraDTO dto, HttpServletResponse response) {
        WxUserInfo wxUserInfo = platformWxmpConfig.getOAuth2AccessToken(service, dto.getCode());
        TokenBean tokenBean = tieshengLoginConfigurer.doLogin(new DoLoginInfo(wxUserInfo.getAppId(),
                wxUserInfo.getOpenid(), "wxmp", dto.getTo(), JSON.toJSONString(wxUserInfo)));
        tieshengLoginConfigurer.onLoginRedirect(tokenBean, dto.getTo(), dto.getExtra(), response);
    }


    /**
     * 微信授权jssdk
     *
     * @param url
     * @return
     */
    @GetMapping("/wxmp/jssdk/{service}")
    public ApiResp<WxJsapiSignature> wxmpJssdk(@PathVariable String service, String url) {
        WxJsapiSignature jsapiSignature = platformWxmpConfig.createJsapiSignature(service, url);
        return ApiResp.respOK(jsapiSignature);
    }

    ///////////////////////////////////////////////////////////////////////////
    // 小程序相关
    ///////////////////////////////////////////////////////////////////////////

    /**
     * 微信小程序登录
     *
     * @return
     */
    @RequestMapping("/wxmini/index/{service}")
    public ApiResp<WxminiLoginVo> wxminiIndex(@PathVariable String service, String code) {
        String openid = platformWxminiConfig.jscode2session(service, code);
        WxConfigBean configBean = platformWxminiConfig.getConfigBean(service);
        TokenBean tokenBean = tieshengLoginConfigurer.doLogin(new DoLoginInfo(configBean.getAppId(),
                openid, "wxmini", "mini", "{}"));

        WxminiLoginVo loginVo = new WxminiLoginVo();
        loginVo.setOpenid(openid);
        if (tokenBean != null) {
            loginVo.setToken(tokenBean.toToken());
        }
        return ApiResp.respOK(loginVo);
    }


}