From a1d31018d3241d0b4cf59d11413e95feddad6025 Mon Sep 17 00:00:00 2001
From: beacon <942112422@qq.com>
Date: Mon, 30 Dec 2019 21:16:48 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E9=A3=9E=E4=B9=A6=E6=8E=88?=
 =?UTF-8?q?=E6=9D=83=E7=99=BB=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../zhyd/oauth/config/AuthDefaultSource.java  |  23 +++-
 .../zhyd/oauth/request/AuthFeishuRequest.java | 106 ++++++++++++++++++
 2 files changed, 128 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/me/zhyd/oauth/request/AuthFeishuRequest.java

diff --git a/src/main/java/me/zhyd/oauth/config/AuthDefaultSource.java b/src/main/java/me/zhyd/oauth/config/AuthDefaultSource.java
index 09f67d8..7446802 100644
--- a/src/main/java/me/zhyd/oauth/config/AuthDefaultSource.java
+++ b/src/main/java/me/zhyd/oauth/config/AuthDefaultSource.java
@@ -723,6 +723,27 @@ public enum AuthDefaultSource implements AuthSource {
         public String userInfo() {
             return "https://api.twitter.com/1.1/users/show.json";
         }
-    }
+    },
+
+    FEISHU{
+        @Override
+        public String authorize() {
+            return "https://open.feishu.cn/connect/qrconnect/page/sso/";
+        }
+
+        @Override
+        public String accessToken() {
+            return "https://open.feishu.cn/connect/qrconnect/oauth2/access_token/";
+        }
+
+        @Override
+        public String userInfo() {
+            return "https://open.feishu.cn/connect/qrconnect/oauth2/user_info/";
+        }
 
+        @Override
+        public String refresh() {
+            return "https://open.feishu.cn/connect/qrconnect/oauth2/access_token/";
+        }
+    }
 }
diff --git a/src/main/java/me/zhyd/oauth/request/AuthFeishuRequest.java b/src/main/java/me/zhyd/oauth/request/AuthFeishuRequest.java
new file mode 100644
index 0000000..61cddd6
--- /dev/null
+++ b/src/main/java/me/zhyd/oauth/request/AuthFeishuRequest.java
@@ -0,0 +1,106 @@
+package me.zhyd.oauth.request;
+
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import me.zhyd.oauth.config.AuthConfig;
+import me.zhyd.oauth.config.AuthDefaultSource;
+import me.zhyd.oauth.enums.AuthResponseStatus;
+import me.zhyd.oauth.exception.AuthException;
+import me.zhyd.oauth.model.AuthCallback;
+import me.zhyd.oauth.model.AuthResponse;
+import me.zhyd.oauth.model.AuthToken;
+import me.zhyd.oauth.model.AuthUser;
+import me.zhyd.oauth.utils.GlobalAuthUtil;
+import me.zhyd.oauth.utils.UrlBuilder;
+
+/**
+ * @author beacon
+ * @since 1.14.0
+ */
+public class AuthFeishuRequest extends AuthDefaultRequest {
+
+    public AuthFeishuRequest(AuthConfig config) {
+        super(config, AuthDefaultSource.FEISHU);
+    }
+
+    @Override
+    protected AuthToken getAccessToken(AuthCallback authCallback) {
+        JSONObject requestObject = new JSONObject();
+        requestObject.put("app_id",config.getClientId());
+        requestObject.put("app_secret",config.getClientSecret());
+        requestObject.put("grant_type","authorization_code");
+        requestObject.put("code",authCallback.getCode());
+        HttpResponse httpResponse = HttpRequest.post(source.accessToken()).body(requestObject.toJSONString(), "application/json").execute();
+        JSONObject jsonObject = JSON.parseObject(httpResponse.body());
+        this.checkResponse(jsonObject);
+        return AuthToken.builder()
+            .accessToken(jsonObject.getString("access_token"))
+            .refreshToken(jsonObject.getString("refresh_token"))
+            .expireIn(jsonObject.getIntValue("expires_in"))
+            .tokenType(jsonObject.getString("token_type"))
+            .openId(jsonObject.getString("open_id"))
+            .build();
+
+    }
+
+    @Override
+    protected AuthUser getUserInfo(AuthToken authToken) {
+        String accessToken = authToken.getAccessToken();
+        HttpResponse userInfoResponse = HttpRequest.get(source.userInfo()).header("Authorization", "Bearer " + accessToken).execute();
+        JSONObject jsonObject = JSON.parseObject(userInfoResponse.body());
+        return AuthUser.builder()
+            .avatar(jsonObject.getString("AvatarUrl"))
+            .username(jsonObject.getString("Mobile"))
+            .email(jsonObject.getString("Email"))
+            .nickname("Name")
+            .build();
+    }
+
+    @Override
+    public AuthResponse refresh(AuthToken authToken) {
+        JSONObject requestObject = new JSONObject();
+        requestObject.put("app_id",config.getClientId());
+        requestObject.put("app_secret",config.getClientSecret());
+        requestObject.put("grant_type","refresh_token");
+        requestObject.put("refresh_token",authToken.getRefreshToken());
+        HttpResponse httpResponse = HttpRequest.post(source.refresh())
+            .body(requestObject.toJSONString(), "application/json")
+            .execute();
+        JSONObject jsonObject = JSON.parseObject(httpResponse.body());
+        this.checkResponse(jsonObject);
+        return AuthResponse.builder()
+            .code(AuthResponseStatus.SUCCESS.getCode())
+            .data(AuthToken.builder()
+                .accessToken(jsonObject.getString("access_token"))
+                .refreshToken(jsonObject.getString("refresh_token"))
+                .expireIn(jsonObject.getIntValue("expires_in"))
+                .tokenType(jsonObject.getString("token_type"))
+                .openId(jsonObject.getString("open_id"))
+                .build())
+            .build();
+
+    }
+
+    @Override
+    public String authorize(String state) {
+        return UrlBuilder.fromBaseUrl(source.authorize())
+            .queryParam("app_id",config.getClientId())
+            .queryParam("redirect_uri", GlobalAuthUtil.urlEncode(config.getRedirectUri()))
+            .queryParam("state",getRealState(state))
+            .build();
+    }
+
+
+    /**
+     * 校验响应内容是否正确
+     * @param jsonObject 响应内容
+     */
+    private void checkResponse(JSONObject jsonObject){
+        if(jsonObject.getIntValue("code") != 0){
+            throw new AuthException(jsonObject.getString("message"));
+        }
+    }
+
+}
-- 
GitLab