AuthGithubRequest.java 2.2 KB
Newer Older
智布道's avatar
智布道 已提交
1 2 3 4 5 6
package me.zhyd.oauth.request;

import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import com.alibaba.fastjson.JSONObject;
import me.zhyd.oauth.config.AuthConfig;
智布道's avatar
智布道 已提交
7
import me.zhyd.oauth.exception.AuthException;
智布道's avatar
智布道 已提交
8
import me.zhyd.oauth.model.AuthSource;
9
import me.zhyd.oauth.model.AuthToken;
智布道's avatar
智布道 已提交
10
import me.zhyd.oauth.model.AuthUser;
智布道's avatar
智布道 已提交
11
import me.zhyd.oauth.utils.GlobalAuthUtil;
智布道's avatar
智布道 已提交
12 13
import me.zhyd.oauth.utils.UrlBuilder;

智布道's avatar
智布道 已提交
14 15
import java.util.Map;

智布道's avatar
智布道 已提交
16
/**
17 18
 * Github登录
 *
智布道's avatar
智布道 已提交
19 20 21 22
 * @author yadong.zhang (yadong.zhang0415(a)gmail.com)
 * @version 1.0
 * @since 1.8
 */
智布道's avatar
智布道 已提交
23
public class AuthGithubRequest extends BaseAuthRequest {
24 25

    public AuthGithubRequest(AuthConfig config) {
26
        super(config, AuthSource.GITHUB);
智布道's avatar
智布道 已提交
27 28 29
    }

    @Override
30
    protected AuthToken getAccessToken(String code) {
31
        String accessTokenUrl = UrlBuilder.getGithubAccessTokenUrl(config.getClientId(), config.getClientSecret(), code, config.getRedirectUri());
智布道's avatar
智布道 已提交
32
        HttpResponse response = HttpRequest.post(accessTokenUrl).execute();
智布道's avatar
智布道 已提交
33 34 35 36
        Map<String, String> res = GlobalAuthUtil.parseStringToMap(response.body());
        if (res.containsKey("error")) {
            throw new AuthException(res.get("error") + ":" + res.get("error_description"));
        }
37 38 39
        return AuthToken.builder()
                .accessToken(res.get("access_token"))
                .build();
40 41 42
    }

    @Override
43 44
    protected AuthUser getUserInfo(AuthToken authToken) {
        String accessToken = authToken.getAccessToken();
45
        HttpResponse response = HttpRequest.get(UrlBuilder.getGithubUserInfoUrl(accessToken)).execute();
智布道's avatar
智布道 已提交
46 47
        String userInfo = response.body();
        JSONObject object = JSONObject.parseObject(userInfo);
48 49 50 51 52 53 54 55 56
        return AuthUser.builder()
                .username(object.getString("login"))
                .avatar(object.getString("avatar_url"))
                .blog(object.getString("blog"))
                .nickname(object.getString("name"))
                .company(object.getString("company"))
                .location(object.getString("location"))
                .email(object.getString("email"))
                .remark(object.getString("bio"))
57
                .token(authToken)
58
                .source(AuthSource.GITHUB)
智布道's avatar
智布道 已提交
59 60 61
                .build();
    }
}