tw.yukina.notion.sdk.client.api.ApiClientFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of notion-sdk-java Show documentation
Notion in Java
There is a newer version: 1.0.1
package tw.yukina.notion.sdk.client.api;

import com.fasterxml.jackson.databind.*;
import com.fasterxml.jackson.databind.module.SimpleModule;
import okhttp3.OkHttpClient;
import org.jetbrains.annotations.NotNull;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import tw.yukina.notion.sdk.client.api.exception.*;
import tw.yukina.notion.sdk.client.api.proxy.ApiClientHandler;
import tw.yukina.notion.sdk.model.deserializer.ZonedDateTimeDeserializer;
import tw.yukina.notion.sdk.model.serializer.ZonedDateTimeSerializer;
import tw.yukina.notion.sdk.support.NotionExceptionWrapper;

import java.lang.reflect.Proxy;
import java.time.ZonedDateTime;

@SuppressWarnings("UnusedReturnValue")
public class ApiClientFactory {

    private String token;

    private static final String NOTION_VERSION = "2021-08-16";

    private static final Logger LOGGER = LoggerFactory.getLogger(ApiClientFactory.class);

    private final SimpleModule objectMapperModule;

    private final ObjectMapper objectMapper;

    private final OkHttpClient okHttpClient;

    private final NotionExceptionWrapper notionExceptionWrapper;

    public ApiClientFactory(){
        this.objectMapperModule = new SimpleModule();
        this.objectMapper = new ObjectMapper();
        this.okHttpClient = new OkHttpClient();
        this.notionExceptionWrapper = new NotionExceptionWrapper();
    }

    public ApiClientFactory setToken(String token){
        this.token = token;
        return this;
    }

    public ApiClientFactory applyDefaultSetting(){
        addDefaultObjectMapperModule();
        addDefaultObjectMapperConfigure();
        addDefaultExceptionDefine();
        return this;
    }

    public ApiClient build(){
        LOGGER.info("Building Notion Api client in Api version " + NOTION_VERSION);
        ApiClientImpl apiClient = new ApiClientImpl(token);
        settingClientInstance(apiClient);
        return proxyApiClient(apiClient);
    }

    public ApiClient buildWithoutProxy(){
        LOGGER.info("Building Notion Api client in Api version " + NOTION_VERSION + " and without proxy");
        ApiClientImpl apiClient = new ApiClientImpl(token);
        settingClientInstance(apiClient);
        return apiClient;
    }

    private ApiClient proxyApiClient(ApiClientImpl apiClientImpl){
        ApiClientHandler apiClientHandler = new ApiClientHandler(apiClientImpl, notionExceptionWrapper);

        return  (ApiClient) Proxy.newProxyInstance(this.getClass().getClassLoader(),
                new Class[]{ApiClient.class}, apiClientHandler);
    }

    public void settingClientInstance(@NotNull ApiClientImpl apiClient){
        apiClient.setNotionVersion(NOTION_VERSION);
        this.objectMapper.registerModule(objectMapperModule);

        apiClient.setObjectMapperModule(objectMapperModule);
        apiClient.setObjectMapper(objectMapper);
        apiClient.setOkHttpClient(okHttpClient);
    }

    public ApiClientFactory addDefaultObjectMapperModule(){
        addDeserializer(ZonedDateTime.class, new ZonedDateTimeDeserializer());
        addSerializer(ZonedDateTime.class, new ZonedDateTimeSerializer());
        return this;
    }

    public ApiClientFactory addDefaultObjectMapperConfigure(){
        objectMapperConfigure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        objectMapperConfigure(SerializationFeature.FAIL_ON_EMPTY_BEANS, false);
        return this;
    }


    public ApiClientFactory objectMapperConfigure(@NotNull DeserializationFeature feature, boolean state){
        LOGGER.info("Changing ObjectMapper state of " + feature.name() + " to " + state);
        objectMapper.configure(feature, state);
        return this;
    }

    public ApiClientFactory objectMapperConfigure(@NotNull SerializationFeature feature, boolean state){
        LOGGER.info("Changing ObjectMapper state of " + feature.name() + " to " + state);
        objectMapper.configure(feature, state);
        return this;
    }

    public ApiClientFactory addDefaultExceptionDefine(){
        notionExceptionWrapper.getExceptionDefine().put("invalid_json", NotionJsonDecodeException.class);
        notionExceptionWrapper.getExceptionDefine().put("invalid_request_url", NotionBadRequestSettingException.class);
        notionExceptionWrapper.getExceptionDefine().put("invalid_request", NotionBadRequestSettingException.class);
        notionExceptionWrapper.getExceptionDefine().put("validation_error", NotonValidationFailureException.class);
        notionExceptionWrapper.getExceptionDefine().put("missing_version", NotionBadRequestSettingException.class);
        notionExceptionWrapper.getExceptionDefine().put("unauthorized", NotionUnauthorizedException.class);
        notionExceptionWrapper.getExceptionDefine().put("restricted_resource", NotionRestrictedResourceException.class);
        notionExceptionWrapper.getExceptionDefine().put("object_not_found", NotionObjectNotFoundException.class);
        notionExceptionWrapper.getExceptionDefine().put("conflict_error", NotionDataConflictException.class);
        notionExceptionWrapper.getExceptionDefine().put("rate_limited", NotionRateLimitedException.class);
        notionExceptionWrapper.getExceptionDefine().put("internal_server_error", NotionUnavailableException.class);
        notionExceptionWrapper.getExceptionDefine().put("service_unavailable", NotionUnavailableException.class);
        notionExceptionWrapper.getExceptionDefine().put("database_connection_unavailable", NotionUnavailableException.class);
        return this;
    }

    public  ApiClientFactory addDeserializer(@NotNull Class type, @NotNull JsonDeserializer deser){
        LOGGER.info("Adding "+ deser.getClass().getName() +" deserializer to handle " + type.getSimpleName() + " type");
        this.objectMapperModule.addDeserializer(type, deser);
        return this;
    }

    public  ApiClientFactory addSerializer(@NotNull Class type, @NotNull JsonSerializer ser){
        LOGGER.info("Adding "+ ser.getClass().getName() +" serializer to handle " + type.getSimpleName() + " type");
        this.objectMapperModule.addSerializer(type, ser);
        return this;
    }
}