fun.feellmoose.service.impl.OkHttpSastLinkService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sast-link-sdk Show documentation
sast-link-sdk
There is a newer version: 0.1.3
package fun.feellmoose.service.impl;

import com.fasterxml.jackson.core.type.TypeReference;
import fun.feellmoose.enums.SastLinkErrorEnum;
import fun.feellmoose.model.response.data.AccessToken;
import okhttp3.*;
import fun.feellmoose.enums.GrantType;
import fun.feellmoose.enums.SastLinkApi;
import fun.feellmoose.exception.SastLinkException;
import fun.feellmoose.model.response.SastLinkResponse;
import fun.feellmoose.model.response.data.RefreshToken;
import fun.feellmoose.model.response.data.User;
import fun.feellmoose.service.SastLinkService;
import fun.feellmoose.util.JsonUtil;

import java.io.IOException;

public final class OkHttpSastLinkService extends AbstractSastLinkService {
    private final OkHttpClient okHttpClient;
    private OkHttpSastLinkService(Builder builder) {
        super(builder);
        this.okHttpClient = builder.okHttpClient;
    }

    @Override
    public AccessToken accessToken(String code) throws SastLinkException {
        HttpUrl url = HttpUrl.get(SastLinkApi.ACCESS_TOKEN.getHttpURI(host_name));
        FormBody formBody = new FormBody.Builder()
                .add(CODE, code)
                .add(CODE_VERIFIER, code_verifier)
                .add(GRANT_TYPE, GrantType.AUTHORIZATION_CODE.name)
                .add(REDIRECT_URI, redirect_uri)
                .add(CLIENT_ID, client_id)
                .add(CLIENT_SECRET, client_secret).build();
        Request request = new Request.Builder()
                .header(CONTENT_TYPE, "multipart/form-data")
                .url(url)
                .method("POST", formBody)
                .build();
        String body = exchangeForResponseBody(request);
        SastLinkResponse response = JsonUtil.fromJson(body, new TypeReference<>() {
        });
        if (!response.isSuccess()) {
            throw new SastLinkException(response);
        }
        return response.getData();
    }

    @Override
    public RefreshToken refreshToken(String refreshToken) throws SastLinkException {
        HttpUrl url = HttpUrl.get(SastLinkApi.REFRESH.getHttpURI(host_name));
        FormBody formBody = new FormBody.Builder()
                .add(REFRESH_TOKEN, refreshToken)
                .add(GRANT_TYPE, GrantType.REFRESH_TOKEN.name).build();
        Request request = new Request.Builder()
                .header(CONTENT_TYPE, "multipart/form-data")
                .url(url)
                .method("POST", formBody)
                .build();
        String body = exchangeForResponseBody(request);
        SastLinkResponse response = JsonUtil.fromJson(body, new TypeReference<>() {
        });
        if (!response.isSuccess()) {
            throw new SastLinkException(response);
        }
        return response.getData();
    }

    @Override
    public User user(String accessToken) throws SastLinkException {
        HttpUrl url = HttpUrl.get(SastLinkApi.USER_INFO.getHttpURI(host_name));
        Request request = new Request.Builder()
                .header(AUTHORIZATION, "Bearer " + accessToken)
                .url(url)
                .get()
                .build();
        String body = exchangeForResponseBody(request);
        SastLinkResponse response = JsonUtil.fromJson(body, new TypeReference<>() {
        });
        if (!response.isSuccess()) {
            throw new SastLinkException(response);
        }
        return response.getData();
    }

    private String exchangeForResponseBody(Request request) {
        String body;
        try {
            ResponseBody responseBody = okHttpClient.newCall(request).execute().body();
            if (responseBody == null) throw new SastLinkException(SastLinkErrorEnum.NULL_RESPONSE_BODY);
            body = responseBody.string();
            if (body.isEmpty()) throw new SastLinkException(SastLinkErrorEnum.EMPTY_RESPONSE_BODY);
        } catch (IOException e) {
            throw new SastLinkException(SastLinkErrorEnum.IO_ERROR, e);
        }
        return body;
    }

    public static class Builder extends AbstractSastLinkService.Builder {
        private OkHttpClient okHttpClient;

        public Builder setOkHttpClient(OkHttpClient okHttpClient) {
            this.okHttpClient = okHttpClient;
            return this;
        }

        @Override
        protected Builder self() {
            return this;
        }

        @Override
        public SastLinkService build() {
            super.build();
            if (okHttpClient == null) {
                okHttpClient = new OkHttpClient();
            }
            return new OkHttpSastLinkService(this);
        }
    }

}