com.github.qq120011676.credential.client.CredentialClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of credential-client Show documentation
credential helper
There is a newer version: 0.1.2
package com.github.qq120011676.credential.client;

import com.github.qq120011676.credential.common.entity.CredentialEntity;
import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import org.jsoup.Connection;
import org.jsoup.Jsoup;
import org.jsoup.internal.StringUtil;

import java.io.IOException;
import java.text.MessageFormat;
import java.time.Duration;
import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;

public class CredentialClient {
    private String baseUrl;

    public CredentialClient(String baseUrl) {
        this.baseUrl = baseUrl;
    }

    public CredentialEntity get() throws IOException {
        return this.get(null);
    }

    public CredentialEntity get(Duration timeout) throws IOException {
        Connection connection = Jsoup.connect(MessageFormat.format("{0}/credential/get", this.baseUrl));
        if (timeout != null) {
            connection.data("timeout", MessageFormat.format("{0}s", timeout.getSeconds()));
        }
        Connection.Response response = connection.ignoreContentType(true).ignoreHttpErrors(true).execute();
        String body = response.body();
        if (200 != response.statusCode()) {
            throw new RuntimeException(new Gson().fromJson(body,JsonObject.class).get("message").getAsString());
        }
        if (!StringUtil.isBlank(body)) {
            JsonObject jsonObject = new Gson().fromJson(body, JsonObject.class);
            CredentialEntity credential = new CredentialEntity();
            credential.setToken(jsonObject.get("token").getAsString());
            credential.setZonedDateTime(ZonedDateTime.parse(jsonObject.get("zonedDateTime").getAsString(), DateTimeFormatter.ISO_ZONED_DATE_TIME));
            JsonElement timeoutJsonElement = jsonObject.get("timeout");
            if (timeoutJsonElement != null) {
                credential.setTimeout(Duration.parse(timeoutJsonElement.getAsString()));
            }
            return credential;
        }
        return null;
    }

    public boolean check(String token) throws IOException {
        Connection.Response response = Jsoup.connect(MessageFormat.format("{0}/credential/check", this.baseUrl)).data("token", token).method(Connection.Method.POST).ignoreContentType(true).ignoreHttpErrors(true).execute();
        return analysisStatus(response);
    }

    public boolean verify(String token) throws IOException {
        Connection.Response response = Jsoup.connect(MessageFormat.format("{0}/credential/verify", this.baseUrl)).data("token", token).method(Connection.Method.POST).ignoreContentType(true).ignoreHttpErrors(true).execute();
        return analysisStatus(response);
    }

    private boolean analysisStatus(Connection.Response response) {
        String body = response.body();
        if (200 != response.statusCode()) {
            throw new RuntimeException(new Gson().fromJson(body,JsonObject.class).get("message").getAsString());
        }
        if (!StringUtil.isBlank(body)) {
            return new Gson().fromJson(body, JsonObject.class).get("status").getAsBoolean();
        }
        return false;
    }
}