• Home
• com.github.reniec-idaas
• idaas_sdk_java
• 1.1.7
• source code
• ReniecIdaasClient.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

pe.gob.reniec.pki.idaas.sdk.ReniecIdaasClient Maven / Gradle / Ivy

package pe.gob.reniec.pki.idaas.sdk;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;
import pe.gob.reniec.pki.idaas.sdk.dto.Config;
import pe.gob.reniec.pki.idaas.sdk.dto.IdToken;
import pe.gob.reniec.pki.idaas.sdk.dto.TokenResponse;
import pe.gob.reniec.pki.idaas.sdk.dto.User;
import pe.gob.reniec.pki.idaas.sdk.enums.Acr;
import pe.gob.reniec.pki.idaas.sdk.enums.Prompt;
import pe.gob.reniec.pki.idaas.sdk.enums.Scope;
import pe.gob.reniec.pki.idaas.sdk.utils.ConvertResponse;
import pe.gob.reniec.pki.idaas.sdk.utils.MySSLConnectionSocketFactory;
import pe.gob.reniec.pki.idaas.sdk.utils.UrlQueryString;

import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Base64;
import java.util.LinkedHashMap;
import java.util.List;

/**
 * @author Miguel Pazo (http://miguelpazo.com)
 */
public class ReniecIdaasClient {

    private String redirectUri = null;
    private List lstScopes = new ArrayList<>();
    private Acr acr = Acr.ONE_FACTOR;
    private Prompt prompt = null;
    private Integer maxAge = null;
    private String state = null;
    private String loginHint = null;
    private Config config;

    public ReniecIdaasClient(String configFile) throws IOException {
        ObjectMapper mapper = new ObjectMapper();
        config = mapper.readValue(new File(configFile), Config.class);
    }

    public String getLoginUrl() {
        String paramScope = "openid";

        LinkedHashMap query = new LinkedHashMap<>();

        query.put("acr_values", this.acr.getValue());
        query.put("client_id", this.config.getClientId());
        query.put("response_type", "code");
        query.put("redirect_uri", this.redirectUri);

        if (this.prompt != null) {
            query.put("prompt", this.prompt.getValue());
        }

        if (this.state != null) {
            query.put("state", this.state);
        }

        if (this.maxAge != null) {
            query.put("max_age", this.maxAge);
        }

        if (this.loginHint != null) {
            query.put("login_hint", this.loginHint);
        }

        //lstScopes
        for (Scope scope : this.lstScopes) {
            paramScope += " " + scope.getValue();
        }

        query.put("scope", paramScope);

        return this.config.getAuthUri() + "?" + UrlQueryString.getInstance().buildQuery(query);
    }

    public TokenResponse getTokens(final String code) throws IOException {
        CloseableHttpClient client = HttpClients.custom().setSSLSocketFactory(MySSLConnectionSocketFactory.getConnectionSocketFactory()).build();
        HttpPost post = new HttpPost(this.config.getTokenUri());

        post.setHeader("Content-Type", "application/x-www-form-urlencoded");

        List urlParameters = new ArrayList();
        urlParameters.add(new BasicNameValuePair("grant_type", "authorization_code"));
        urlParameters.add(new BasicNameValuePair("code", code));
        urlParameters.add(new BasicNameValuePair("redirect_uri", this.redirectUri));
        urlParameters.add(new BasicNameValuePair("client_id", this.config.getClientId()));
        urlParameters.add(new BasicNameValuePair("client_secret", this.config.getClientSecret()));

        post.setEntity(new UrlEncodedFormEntity(urlParameters, StandardCharsets.UTF_8));

        HttpResponse response = client.execute(post);
        Object object = ConvertResponse.getInstance().convert(response, TokenResponse.class);

        if (object != null) {
            TokenResponse tokenResponse = (TokenResponse) object;
            String[] parts = tokenResponse.getIdToken().split("\\.");
            ObjectMapper objectMapper = new ObjectMapper();
            IdToken idToken = objectMapper.readValue(new String(Base64.getDecoder().decode(parts[1]), "UTF-8"), IdToken.class);

            tokenResponse.setIdTokenObject(idToken);

            return tokenResponse;
        } else {
            return null;
        }
    }

    public User getUserInfo(String accessToken) throws IOException {
        CloseableHttpClient client = HttpClients.custom().setSSLSocketFactory(MySSLConnectionSocketFactory.getConnectionSocketFactory()).build();
        HttpPost post = new HttpPost(this.config.getUserInfoUri());

        post.setHeader("Content-Type", "application/x-www-form-urlencoded");
        post.setHeader("Authorization", "Bearer " + accessToken);

        HttpResponse response = client.execute(post);
        Object object = ConvertResponse.getInstance().convert(response, User.class);

        if (object != null) {
            return (User) object;
        } else {
            return null;
        }
    }

    public String getLogoutUri(String redirectPostLogout) {
        LinkedHashMap query = new LinkedHashMap<>();

        query.put("post_logout_redirect_uri", redirectPostLogout);

        return this.config.getLogoutUri() + "?" + UrlQueryString.getInstance().buildQuery(query);
    }

    public String getState() {
        return state;
    }

    public void setState(String state) {
        this.state = state;
    }

    public void setAcr(Acr acr) {
        this.acr = acr;
    }

    public void setPrompt(Prompt prompt) {
        this.prompt = prompt;
    }

    public void setMaxAge(Integer maxAge) {
        this.maxAge = maxAge;
    }

    public void addScope(Scope scope) {
        this.lstScopes.add(scope);
    }

    public void cleanScopes() {
        this.lstScopes.clear();
    }

    public void setRedirectUri(String redirectUri) {
        this.redirectUri = redirectUri;
    }

    public Config getConfig() {
        return config;
    }

    public void setLoginHint(String loginHint) {
        this.loginHint = loginHint;
    }
}