fm.pattern.tokamak.sdk.dsl.ClientDSL Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tokamak-sdk Show documentation
The newest version!
package fm.pattern.tokamak.sdk.dsl;

import static org.apache.commons.lang3.RandomStringUtils.randomAlphanumeric;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

import fm.pattern.tokamak.sdk.AudiencesClient;
import fm.pattern.tokamak.sdk.AuthoritiesClient;
import fm.pattern.tokamak.sdk.ClientsClient;
import fm.pattern.tokamak.sdk.GrantTypesClient;
import fm.pattern.tokamak.sdk.JwtClientProperties;
import fm.pattern.tokamak.sdk.ScopesClient;
import fm.pattern.tokamak.sdk.commons.Result;
import fm.pattern.tokamak.sdk.model.AudienceRepresentation;
import fm.pattern.tokamak.sdk.model.AuthorityRepresentation;
import fm.pattern.tokamak.sdk.model.ClientRepresentation;
import fm.pattern.tokamak.sdk.model.GrantTypeRepresentation;
import fm.pattern.tokamak.sdk.model.ScopeRepresentation;

public class ClientDSL extends AbstractDSL {

    private ClientsClient client = new ClientsClient(JwtClientProperties.getEndpoint());
    private ScopesClient scopesClient = new ScopesClient(JwtClientProperties.getEndpoint());
    private GrantTypesClient grantTypesClient = new GrantTypesClient(JwtClientProperties.getEndpoint());
    private AuthoritiesClient authoritiesClient = new AuthoritiesClient(JwtClientProperties.getEndpoint());
    private AudiencesClient audiencesClient = new AudiencesClient(JwtClientProperties.getEndpoint());

    private String clientId = "cli_" + randomAlphanumeric(15);
    private String clientSecret = "pwd_" + randomAlphanumeric(15);

    private Integer accessTokenValiditySeconds = 600;
    private Integer refreshTokenValiditySeconds = 6000;

    private Set scopes = new HashSet();
    private Set grantTypes = new HashSet();
    private Set authorities = new HashSet();
    private Set audiences = new HashSet();

    private Set scopesLookup = new HashSet();
    private Set grantTypesLookup = new HashSet();
    private Set authoritiesLookup = new HashSet();
    private Set audiencesLookup = new HashSet();

    public static ClientDSL client() {
        return new ClientDSL();
    }

    private ClientDSL() {

    }

    public ClientDSL withClientId(String clientId) {
        this.clientId = clientId;
        return this;
    }

    public ClientDSL withClientSecret(String clientSecret) {
        this.clientSecret = clientSecret;
        return this;
    }

    public ClientDSL withAuthorities(AuthorityRepresentation... authorities) {
        this.authorities.clear();
        this.authorities.addAll(Arrays.asList(authorities));
        return this;
    }

    public ClientDSL withAuthorities(String... authorities) {
        this.authoritiesLookup.clear();
        this.authoritiesLookup.addAll(Arrays.asList(authorities));
        return this;
    }

    public ClientDSL withAudiences(AudienceRepresentation... audiences) {
        this.audiences.clear();
        this.audiences.addAll(Arrays.asList(audiences));
        return this;
    }

    public ClientDSL withAudiences(String... audiences) {
        this.audiencesLookup.clear();
        this.audiencesLookup.addAll(Arrays.asList(audiences));
        return this;
    }

    public ClientDSL withScopes(ScopeRepresentation... scopes) {
        this.scopes.clear();
        this.scopes.addAll(Arrays.asList(scopes));
        return this;
    }

    public ClientDSL withScopes(String... scopes) {
        this.scopesLookup.clear();
        this.scopesLookup.addAll(Arrays.asList(scopes));
        return this;
    }

    public ClientDSL withGrantTypes(GrantTypeRepresentation... grantTypes) {
        this.grantTypes.clear();
        this.grantTypes.addAll(Arrays.asList(grantTypes));
        return this;
    }

    public ClientDSL withGrantTypes(String... grantTypes) {
        this.grantTypesLookup.clear();
        this.grantTypesLookup.addAll(Arrays.asList(grantTypes));
        return this;
    }

    public ClientDSL withAccessTokenValiditySeconds(Integer accessTokenValiditySeconds) {
        this.accessTokenValiditySeconds = accessTokenValiditySeconds;
        return this;
    }

    public ClientDSL withRefreshTokenValiditySeconds(Integer refreshTokenValiditySeconds) {
        this.refreshTokenValiditySeconds = refreshTokenValiditySeconds;
        return this;
    }

    public ClientRepresentation build() {
        ClientRepresentation representation = create();
        if (!shouldPersist()) {
            return representation;
        }

        Result response = client.create(representation, super.getToken().getAccessToken());
        if (response.getResponseCode() != 201) {
            throw new IllegalStateException("Unable to create client, response from server: " + response.getErrors().toString());
        }

        return response.getInstance();
    }

    private ClientRepresentation create() {
        resolve();

        ClientRepresentation client = new ClientRepresentation();
        client.setClientId(clientId);
        client.setClientSecret(clientSecret);
        client.setScopes(scopes);
        client.setGrantTypes(grantTypes);
        client.setAuthorities(authorities);
        client.setAudiences(audiences);
        client.setAccessTokenValiditySeconds(accessTokenValiditySeconds);
        client.setRefreshTokenValiditySeconds(refreshTokenValiditySeconds);
        return client;
    }

    private void resolve() {
        if (super.getToken() == null) {
            return;
        }

        for (String name : scopesLookup) {
            Result result = scopesClient.findByName(name, super.getToken().getAccessToken());
            if (result.accepted()) {
                scopes.add(result.getInstance());
            }
        }

        for (String name : grantTypesLookup) {
            Result result = grantTypesClient.findByName(name, super.getToken().getAccessToken());
            if (result.accepted()) {
                grantTypes.add(result.getInstance());
            }
        }

        for (String name : authoritiesLookup) {
            Result result = authoritiesClient.findByName(name, super.getToken().getAccessToken());
            if (result.accepted()) {
                authorities.add(result.getInstance());
            }
        }

        for (String name : audiencesLookup) {
            Result result = audiencesClient.findByName(name, super.getToken().getAccessToken());
            if (result.accepted()) {
                audiences.add(result.getInstance());
            }
        }
    }

}