cn.herodotus.engine.access.justauth.processor.JustAuthProcessor Maven / Gradle / Ivy
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2020-2030 郑庚伟 ZHENGGENGWEI (码匠君), Licensed under the AGPL License
*
* This file is part of Herodotus Engine.
*
* Herodotus Engine is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published
* by the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Herodotus Engine is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see .
*/
package cn.herodotus.engine.access.justauth.processor;
import cn.herodotus.engine.access.core.exception.AccessConfigErrorException;
import cn.herodotus.engine.access.core.exception.IllegalAccessSourceException;
import cn.herodotus.engine.access.justauth.properties.JustAuthProperties;
import cn.herodotus.engine.access.justauth.stamp.JustAuthStateStampManager;
import me.zhyd.oauth.config.AuthConfig;
import me.zhyd.oauth.config.AuthDefaultSource;
import me.zhyd.oauth.request.*;
import me.zhyd.oauth.utils.AuthStateUtils;
import org.apache.commons.collections4.MapUtils;
import org.apache.commons.lang3.ObjectUtils;
import org.dromara.hutool.core.util.EnumUtil;
import org.jetbrains.annotations.NotNull;
import java.util.Map;
import java.util.stream.Collectors;
/**
* Description: JustAuth请求的生成器
*
* @author : gengwei.zheng
* @date : 2021/5/22 11:23
*/
public class JustAuthProcessor {
private JustAuthProperties justAuthProperties;
private JustAuthStateStampManager justAuthStateStampManager;
public void setJustAuthProperties(JustAuthProperties justAuthProperties) {
this.justAuthProperties = justAuthProperties;
}
public void setJustAuthStateRedisCache(JustAuthStateStampManager justAuthStateStampManager) {
this.justAuthStateStampManager = justAuthStateStampManager;
}
private JustAuthStateStampManager getJustAuthStateRedisCache() {
return justAuthStateStampManager;
}
public AuthRequest getAuthRequest(String source) {
AuthDefaultSource authDefaultSource = parseAuthDefaultSource(source);
AuthConfig authConfig = getAuthConfig(authDefaultSource);
return getAuthRequest(authDefaultSource, authConfig);
}
public AuthRequest getAuthRequest(String source, AuthConfig authConfig) {
AuthDefaultSource authDefaultSource = parseAuthDefaultSource(source);
return getAuthRequest(authDefaultSource, authConfig);
}
/**
* 返回带state参数的授权url，授权回调时会带上这个state
*
* @param source 第三方登录的类别 {@link AuthDefaultSource}
* @return 返回授权地址
*/
public String getAuthorizeUrl(String source) {
AuthRequest authRequest = this.getAuthRequest(source);
return authRequest.authorize(AuthStateUtils.createState());
}
public String getAuthorizeUrl(String source, AuthConfig authConfig) {
AuthRequest authRequest = this.getAuthRequest(source, authConfig);
return authRequest.authorize(AuthStateUtils.createState());
}
public Map getAuthorizeUrls() {
Map configs = getConfigs();
return configs.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> getAuthorizeUrl(entry.getKey(), entry.getValue())));
}
@NotNull
private Map getConfigs() {
Map configs = justAuthProperties.getConfigs();
if(MapUtils.isEmpty(configs)) {
throw new AccessConfigErrorException();
}
return configs;
}
@NotNull
private AuthConfig getAuthConfig(AuthDefaultSource authDefaultSource) {
Map configs = getConfigs();
AuthConfig authConfig = configs.get(authDefaultSource.name());
// 找不到对应关系，直接返回空
if (ObjectUtils.isEmpty(authConfig)) {
throw new AccessConfigErrorException();
}
return authConfig;
}
private static AuthDefaultSource parseAuthDefaultSource(String source) {
AuthDefaultSource authDefaultSource;
try {
authDefaultSource = EnumUtil.fromString(AuthDefaultSource.class, source.toUpperCase());
} catch (IllegalArgumentException e) {
throw new IllegalAccessSourceException();
}
return authDefaultSource;
}
private AuthRequest getAuthRequest(AuthDefaultSource authDefaultSource, AuthConfig authConfig) {
switch (authDefaultSource) {
case GITHUB:
return new AuthGithubRequest(authConfig, this.getJustAuthStateRedisCache());
case WEIBO:
return new AuthWeiboRequest(authConfig, this.getJustAuthStateRedisCache());
case GITEE:
return new AuthGiteeRequest(authConfig, this.getJustAuthStateRedisCache());
case DINGTALK:
return new AuthDingTalkRequest(authConfig, this.getJustAuthStateRedisCache());
case BAIDU:
return new AuthBaiduRequest(authConfig, this.getJustAuthStateRedisCache());
case CSDN:
return new AuthCsdnRequest(authConfig, this.getJustAuthStateRedisCache());
case CODING:
return new AuthCodingRequest(authConfig, this.getJustAuthStateRedisCache());
case OSCHINA:
return new AuthOschinaRequest(authConfig, this.getJustAuthStateRedisCache());
case ALIPAY:
return new AuthAlipayRequest(authConfig, this.getJustAuthStateRedisCache());
case QQ:
return new AuthQqRequest(authConfig, this.getJustAuthStateRedisCache());
case WECHAT_MP:
return new AuthWeChatMpRequest(authConfig, this.getJustAuthStateRedisCache());
case WECHAT_OPEN:
return new AuthWeChatOpenRequest(authConfig, this.getJustAuthStateRedisCache());
case WECHAT_ENTERPRISE:
return new AuthWeChatEnterpriseQrcodeRequest(authConfig, this.getJustAuthStateRedisCache());
case WECHAT_ENTERPRISE_WEB:
return new AuthWeChatEnterpriseWebRequest(authConfig, this.getJustAuthStateRedisCache());
case TAOBAO:
return new AuthTaobaoRequest(authConfig, this.getJustAuthStateRedisCache());
case GOOGLE:
return new AuthGoogleRequest(authConfig, this.getJustAuthStateRedisCache());
case FACEBOOK:
return new AuthFacebookRequest(authConfig, this.getJustAuthStateRedisCache());
case DOUYIN:
return new AuthDouyinRequest(authConfig, this.getJustAuthStateRedisCache());
case LINKEDIN:
return new AuthLinkedinRequest(authConfig, this.getJustAuthStateRedisCache());
case MICROSOFT:
return new AuthMicrosoftRequest(authConfig, this.getJustAuthStateRedisCache());
case MI:
return new AuthMiRequest(authConfig, this.getJustAuthStateRedisCache());
case TOUTIAO:
return new AuthToutiaoRequest(authConfig, this.getJustAuthStateRedisCache());
case TEAMBITION:
return new AuthTeambitionRequest(authConfig, this.getJustAuthStateRedisCache());
case RENREN:
return new AuthRenrenRequest(authConfig, this.getJustAuthStateRedisCache());
case PINTEREST:
return new AuthPinterestRequest(authConfig, this.getJustAuthStateRedisCache());
case STACK_OVERFLOW:
return new AuthStackOverflowRequest(authConfig, this.getJustAuthStateRedisCache());
case HUAWEI:
return new AuthHuaweiRequest(authConfig, this.getJustAuthStateRedisCache());
case GITLAB:
return new AuthGitlabRequest(authConfig, this.getJustAuthStateRedisCache());
case KUJIALE:
return new AuthKujialeRequest(authConfig, this.getJustAuthStateRedisCache());
case ELEME:
return new AuthElemeRequest(authConfig, this.getJustAuthStateRedisCache());
case MEITUAN:
return new AuthMeituanRequest(authConfig, this.getJustAuthStateRedisCache());
case TWITTER:
return new AuthTwitterRequest(authConfig, this.getJustAuthStateRedisCache());
case FEISHU:
return new AuthFeishuRequest(authConfig, this.getJustAuthStateRedisCache());
case JD:
return new AuthJdRequest(authConfig, this.getJustAuthStateRedisCache());
case ALIYUN:
return new AuthAliyunRequest(authConfig, this.getJustAuthStateRedisCache());
case XMLY:
return new AuthXmlyRequest(authConfig, this.getJustAuthStateRedisCache());
default:
return null;
}
}
}