All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.gravitee.am.gateway.service.impl.ClientServiceImpl Maven / Gradle / Ivy

There is a newer version: 2.1.1
Show newest version
/**
 * Copyright (C) 2015 The Gravitee team (http://gravitee.io)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.gravitee.am.gateway.service.impl;

import io.gravitee.am.gateway.service.ClientService;
import io.gravitee.am.gateway.service.IdentityProviderService;
import io.gravitee.am.gateway.service.exception.ClientAlreadyExistsException;
import io.gravitee.am.gateway.service.exception.ClientNotFoundException;
import io.gravitee.am.gateway.service.exception.TechnicalManagementException;
import io.gravitee.am.gateway.service.model.NewClient;
import io.gravitee.am.gateway.service.model.TopClient;
import io.gravitee.am.gateway.service.model.TotalClient;
import io.gravitee.am.gateway.service.model.UpdateClient;
import io.gravitee.am.model.Client;
import io.gravitee.am.model.common.Page;
import io.gravitee.am.repository.exceptions.TechnicalException;
import io.gravitee.am.repository.management.api.ClientRepository;
import io.gravitee.am.repository.oauth2.api.TokenRepository;
import io.gravitee.common.utils.UUID;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.*;
import java.util.stream.Collectors;

/**
 * @author David BRASSELY (david.brassely at graviteesource.com)
 * @author GraviteeSource Team
 */
@Component
public class ClientServiceImpl implements ClientService {

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

    @Autowired
    private ClientRepository clientRepository;

    @Autowired
    private IdentityProviderService identityProviderService;

    @Autowired
    private TokenRepository tokenRepository;

    @Override
    public Client findById(String id) {
        try {
            LOGGER.debug("Find client by ID: {}", id);
            Optional clientOpt = clientRepository.findById(id);

            if (!clientOpt.isPresent()) {
                throw new ClientNotFoundException(id);
            }

            Client client = clientOpt.get();
            // Send an empty array in case of no grant types
            if (client.getAuthorizedGrantTypes() == null) {
                client.setAuthorizedGrantTypes(Collections.emptyList());
            }

            return client;
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find a client using its ID: {}", id, ex);
            throw new TechnicalManagementException(
                    String.format("An error occurs while trying to find a client using its ID: %s", id), ex);
        }
    }

    @Override
    public Client findByDomainAndClientId(String domain, String clientId) {
        try {
            Optional clientOpt = clientRepository.findByClientIdAndDomain(clientId, domain);
            if (!clientOpt.isPresent()) {
                throw new ClientNotFoundException(clientId);
            }

            return clientOpt.get();
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find client by domain: {} and client id: {}", domain, clientId, ex);
            throw new TechnicalManagementException(
                    String.format("An error occurs while trying to find client by domain: %s and client id: %s", domain, clientId), ex);
        }
    }

    @Override
    public Set findByDomain(String domain) {
        try {
            LOGGER.debug("Find clients by domain", domain);
            return clientRepository.findByDomain(domain);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find clients by domain: {}", domain, ex);
            throw new TechnicalManagementException(
                    String.format("An error occurs while trying to find clients by domain: %s", domain), ex);
        }
    }

    @Override
    public Page findByDomain(String domain, int page, int size) {
        try {
            LOGGER.debug("Find clients by domain", domain);
            return clientRepository.findByDomain(domain, page, size);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find clients by domain: {}", domain, ex);
            throw new TechnicalManagementException(
                    String.format("An error occurs while trying to find clients by domain: %s", domain), ex);
        }
    }

    @Override
    public Set findByIdentityProvider(String identityProvider) {
        try {
            LOGGER.debug("Find clients by identity provider : {}", identityProvider);
            return clientRepository.findByIdentityProvider(identityProvider);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find clients by identity provider", ex);
            throw new TechnicalManagementException("An error occurs while trying to find clients by identity provider", ex);
        }
    }

    @Override
    public Set findByCertificate(String certificate) {
        try {
            LOGGER.debug("Find clients by certificate : {}", certificate);
            return clientRepository.findByCertificate(certificate);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find clients by certificate", ex);
            throw new TechnicalManagementException("An error occurs while trying to find clients by certificate", ex);
        }
    }

    @Override
    public Set findAll() {
        try {
            LOGGER.debug("Find clients");
            return clientRepository.findAll();
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find clients", ex);
            throw new TechnicalManagementException("An error occurs while trying to find clients", ex);
        }
    }

    @Override
    public Page findAll(int page, int size) {
        try {
            LOGGER.debug("Find clients");
            return clientRepository.findAll(page, size);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find clients", ex);
            throw new TechnicalManagementException("An error occurs while trying to find clients", ex);
        }
    }

    @Override
    public Set findTopClients() {
        try {
            LOGGER.debug("Find top clients");
            Set clients = clientRepository.findAll();
            return clients.parallelStream().map(c -> {
                TopClient client = new TopClient();
                client.setClient(c);
                client.setAccessTokens(tokenRepository.findTokensByClientId(c.getClientId()).size());
                return client;
            }).filter(topClient -> topClient.getAccessTokens() > 0).collect(Collectors.toSet());
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find top clients", ex);
            throw new TechnicalManagementException("An error occurs while trying to find top clients", ex);
        }
    }

    @Override
    public Set findTopClientsByDomain(String domain) {
        try {
            LOGGER.debug("Find top clients by domain: {}", domain);
            Set clients = clientRepository.findByDomain(domain);
            return clients.parallelStream().map(c -> {
                TopClient client = new TopClient();
                client.setClient(c);
                client.setAccessTokens(tokenRepository.findTokensByClientId(c.getClientId()).size());
                return client;
            }).filter(topClient -> topClient.getAccessTokens() > 0).collect(Collectors.toSet());
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find top clients by domain", ex);
            throw new TechnicalManagementException("An error occurs while trying to find top clients by domain", ex);
        }
    }

    @Override
    public TotalClient findTotalClientsByDomain(String domain) {
        try {
            LOGGER.debug("Find total clients by domain: {}", domain);
            TotalClient totalClient = new TotalClient();
            totalClient.setTotalClients(clientRepository.countByDomain(domain));
            return totalClient;
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find total clients by domain: {}", domain, ex);
            throw new TechnicalManagementException(
                    String.format("An error occurs while trying to find total clients by domain: %s", domain), ex);
        }
    }

    @Override
    public TotalClient findTotalClients() {
        try {
            LOGGER.debug("Find total client");
            TotalClient totalClient = new TotalClient();
            totalClient.setTotalClients(clientRepository.count());
            return totalClient;
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to find total clients", ex);
            throw new TechnicalManagementException("An error occurs while trying to find total clients", ex);
        }
    }

    @Override
    public Client create(String domain, NewClient newClient) {
        try {
            LOGGER.debug("Create a new client {} for domain {}", newClient, domain);

            Optional clientOpt = clientRepository.findByClientIdAndDomain(newClient.getClientId(), domain);
            if (clientOpt.isPresent()) {
                throw new ClientAlreadyExistsException(newClient.getClientId(), domain);
            }

            Client client = new Client();
            client.setId(UUID.toString(UUID.random()));
            client.setClientId(newClient.getClientId());
            if (newClient.getClientSecret() == null || newClient.getClientSecret().trim().isEmpty()) {
                client.setClientSecret(UUID.toString(UUID.random()));
            } else {
                client.setClientSecret(newClient.getClientSecret());
            }
            client.setDomain(domain);
            client.setAccessTokenValiditySeconds(Client.DEFAULT_ACCESS_TOKEN_VALIDITY_SECONDS);
            client.setRefreshTokenValiditySeconds(Client.DEFAULT_REFRESH_TOKEN_VALIDITY_SECONDS);
            client.setIdTokenValiditySeconds(Client.DEFAULT_ID_TOKEN_VALIDITY_SECONDS);
            client.setAuthorizedGrantTypes(Client.AUTHORIZED_GRANT_TYPES);
            client.setEnabled(true);
            client.setCreatedAt(new Date());
            client.setUpdatedAt(client.getCreatedAt());

            return clientRepository.create(client);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to create a client", ex);
            throw new TechnicalManagementException("An error occurs while trying to create a client", ex);
        }
    }

    @Override
    public Client update(String domain, String id, UpdateClient updateClient) {
        try {
            LOGGER.debug("Update a client {} for domain {}", id, domain);

            Optional clientOpt = clientRepository.findById(id);
            if (!clientOpt.isPresent()) {
                throw new ClientNotFoundException(id);
            }

            // Check associated identity providers
            Set identities = updateClient.getIdentities();
            if (identities != null) {
                identities.forEach(identityProviderId -> identityProviderService.findById(identityProviderId));
            }

            Client client = clientOpt.get();
            client.setScopes(updateClient.getScopes());
            client.setAutoApproveScopes(updateClient.getAutoApproveScopes());
            client.setAccessTokenValiditySeconds(updateClient.getAccessTokenValiditySeconds());
            client.setRefreshTokenValiditySeconds(updateClient.getRefreshTokenValiditySeconds());
            client.setAuthorizedGrantTypes(updateClient.getAuthorizedGrantTypes());
            client.setRedirectUris(updateClient.getRedirectUris());
            client.setEnabled(updateClient.isEnabled());
            client.setIdentities(updateClient.getIdentities());
            client.setIdTokenValiditySeconds(updateClient.getIdTokenValiditySeconds());
            client.setIdTokenCustomClaims(updateClient.getIdTokenCustomClaims());
            client.setCertificate(updateClient.getCertificate());
            client.setEnhanceScopesWithUserPermissions(updateClient.isEnhanceScopesWithUserPermissions());
            client.setUpdatedAt(new Date());

            return clientRepository.update(client);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to update a client", ex);
            throw new TechnicalManagementException("An error occurs while trying to update a client", ex);
        }
    }

    @Override
    public void delete(String clientId) {
        try {
            LOGGER.debug("Delete client {}", clientId);

            Optional optClient = clientRepository.findById(clientId);
            if (! optClient.isPresent()) {
                throw new ClientNotFoundException(clientId);
            }

            clientRepository.delete(clientId);
        } catch (TechnicalException ex) {
            LOGGER.error("An error occurs while trying to delete client: {}", clientId, ex);
            throw new TechnicalManagementException(
                    String.format("An error occurs while trying to delete client: %s", clientId), ex);
        }
    }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy